nexu-io
diff --git a/‎next/src/lib/export/__tests__/wechat.test.ts‎
Lines changed: 16 additions & 1 deletion b/‎next/src/lib/export/__tests__/wechat.test.ts‎
Lines changed: 16 additions & 1 deletion
diff --git a/‎next/src/lib/export/wechat.ts‎
Lines changed: 10 additions & 2 deletions b/‎next/src/lib/export/wechat.ts‎
Lines changed: 10 additions & 2 deletions
@@ -1,5 +1,5 @@
 import { describe, it, expect, afterEach } from "vitest";
-import { toWechatHtmlFromDocument } from "../wechat";
+import { toWechatHtmlForExport, toWechatHtmlFromDocument } from "../wechat";
 
 function parseFragment(html: string): HTMLBodyElement {
   return new DOMParser().parseFromString(`<body>${html}</body>`, "text/html")
@@ -210,4 +210,19 @@ describe("toWechatHtmlFromDocument", () => {
     expect(style).not.toContain("180px");
     expect(style).not.toContain("96px");
   });
+
+  it("keeps every deck slide when the selected iframe contains only one slide", () => {
+    const fullDeck = `<!doctype html><html><body>
+      <section class="slide" data-slide-id="1"><h1>Slide one</h1></section>
+      <section class="slide" data-slide-id="2"><h1>Slide two</h1></section>
+    </body></html>`;
+    const selectedSlide = new DOMParser().parseFromString(
+      `<!doctype html><html><body><section class="slide"><h1>Slide two</h1></section></body></html>`,
+      "text/html",
+    );
+
+    const exported = toWechatHtmlForExport(fullDeck, selectedSlide);
+    expect(exported).toContain("Slide one");
+    expect(exported).toContain("Slide two");
+  });
 });
@@ -1,6 +1,7 @@
 "use client";
 
 import juice from "juice";
+import { isDeck } from "../deck";
 import { copyHtml } from "./clipboard";
 
 /**
@@ -105,9 +106,16 @@ export function toWechatHtmlFromDocument(renderedDoc: Document): string {
   return section.outerHTML;
 }
 
+export function toWechatHtmlForExport(
+  fullHtml: string,
+  renderedDoc?: Document | null,
+ ): string {
+  if (isDeck(fullHtml)) return toWechatHtml(fullHtml);
+  return renderedDoc?.body ? toWechatHtmlFromDocument(renderedDoc) : toWechatHtml(fullHtml);
+}
+
 export async function copyToWechat(fullHtml: string, renderedDoc?: Document | null): Promise<void> {
-  const html = renderedDoc?.body ? toWechatHtmlFromDocument(renderedDoc) : toWechatHtml(fullHtml);
-  await copyHtml(html);
+  await copyHtml(toWechatHtmlForExport(fullHtml, renderedDoc));
 }
 
 function inlineComputedTree(source: Element, clone: Element, view: Window): void {