feat: set spine

yjl9903 · Feb 27, 2023 · ca7a864 · ca7a864
1 parent 38b1d0d
commit ca7a864
Show file tree

Hide file tree

Showing 4 changed files with 39 additions and 5 deletions.
diff --git a/packages/core/src/epub/epub.ts b/packages/core/src/epub/epub.ts
@@ -38,6 +38,11 @@ export class Epub {
     return this;
   }
 
+  public spine(...items: Item[]) {
+    this.opfs[0].setSpine(items);
+    return this;
+  }
+
   async bundle() {
     const { bundle } = await import('../bundle');
     return await bundle(this);

diff --git a/packages/core/src/epub/opf.ts b/packages/core/src/epub/opf.ts
@@ -101,6 +101,7 @@ export class PackageDocument {
   // --- manifest ---
   public addItem(item: Item) {
     this._items.push(item);
+    return this;
   }
 
   public items() {
@@ -115,6 +116,11 @@ export class PackageDocument {
     return this.items().map((i) => i.manifest());
   }
 
+  public setSpine(items: Item[]) {
+    this._spine.splice(0, this._spine.length, ...items.map((i) => i.itemref()));
+    return this;
+  }
+
   public spine() {
     return this._spine;
   }

diff --git a/packages/core/test/__snapshots__/bundle.test.ts.snap b/packages/core/test/__snapshots__/bundle.test.ts.snap
@@ -53,7 +53,28 @@ PackageDocument {
     "title": "Test Book",
     "type": "",
   },
-  "_spine": [],
+  "_spine": [
+    ManifestItemRef {
+      "_idref": "page1",
+      "optional": {},
+    },
+    ManifestItemRef {
+      "_idref": "page2",
+      "optional": {},
+    },
+    ManifestItemRef {
+      "_idref": "page3",
+      "optional": {},
+    },
+    ManifestItemRef {
+      "_idref": "page4",
+      "optional": {},
+    },
+    ManifestItemRef {
+      "_idref": "page5",
+      "optional": {},
+    },
+  ],
   "_toc": Html {
     "_properties": "nav",
     "content": "<html xmlns=\\"http://www.w3.org/1999/xhtml\\" xmlns:epub=\\"http://www.idpf.org/2007/ops\\" lang=\\"zh-CN\\" xml:lang=\\"zh-CN\\">

diff --git a/packages/core/test/bundle.test.ts b/packages/core/test/bundle.test.ts
@@ -36,8 +36,7 @@ describe('Bundle Epub', () => {
     });
 
     const cover = new Html('cover.xhtml', '');
-    opf.addItem(cover);
-    opf.spine().push(cover.manifest().ref());
+    epub.addItem(cover).spine(cover);
 
     const res = makePackageDocument(opf);
     expect(res).toMatchInlineSnapshot(`
@@ -93,8 +92,10 @@ describe('Bundle Epub', () => {
   </body>
 </html>`;
     const cover = new Html('cover.xhtml', content);
-    epub.addItem(cover).toc([{ text: 'cover', item: cover }]);
-    epub.main().spine().push(cover.itemref());
+    epub
+      .addItem(cover)
+      .spine(cover)
+      .toc([{ text: 'cover', item: cover }]);
 
     await epub.writeFile('.output/test.epub');
   });
@@ -118,6 +119,7 @@ describe('Bundle Epub', () => {
 
     await epub
       .addItem(item1, item2, item3, item4, item5)
+      .spine(item1, item2, item3, item4, item5)
       .toc(
         [
           { text: '1', item: item1 },