pass totalPages in pages api even when no query

rename seedPages -> initialPages
webrecorder · Feb 13, 2025 · c09bed6 · c09bed6
1 parent 89ddc74
commit c09bed6
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 5 deletions.
diff --git a/src/api.ts b/src/api.ts
@@ -297,6 +297,7 @@ class API {
         if (!coll) {
           return { error: "collection_not_found" };
         }
+        let total = undefined;
         if (coll.store instanceof MultiWACZ) {
           // @ts-expect-error [TODO] - TS4111 - Property '_query' comes from an index signature, so it must be accessed with ['_query'].
           const search = params._query.get("search");
@@ -312,10 +313,12 @@ class API {
               pageSize,
             );
             return { pages, total };
+          } else {
+            total = coll.store.totalPages;
           }
         }
         const pages = await coll.store.getAllPages();
-        return { pages };
+        return { pages, total };
       }
 
       case "textIndex": {

diff --git a/src/wacz/multiwacz.ts b/src/wacz/multiwacz.ts
@@ -100,6 +100,8 @@ export class MultiWACZ
 
   pagesQueryUrl = "";
 
+  totalPages?: number = undefined;
+
   preloadResources: string[] = [];
   seedPageWACZs: Map<string, Set<string>> = new Map<string, Set<string>>();
 
@@ -959,7 +961,7 @@ export class MultiWACZ
   async loadWACZFiles(
     // [TODO]
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    json: { resources: any; seedPages: any; preloadResources: any },
+    json: { resources: any; initialPages: any; preloadResources: any, totalPages: number },
     parent: WACZLoadSource = this,
   ) {
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
@@ -1009,14 +1011,18 @@ export class MultiWACZ
       }
     }
 
-    if (json.seedPages) {
+    if (json.initialPages) {
       // eslint-disable-next-line @typescript-eslint/no-unsafe-argument
-      await this.addSeedPages(json.seedPages);
+      await this.addInitialPages(json.initialPages);
+    }
+
+    if (!isNaN(json.totalPages)) {
+      this.totalPages = json.totalPages;
     }
   }
 
   // eslint-disable-next-line @typescript-eslint/no-explicit-any
-  async addSeedPages(pagesImport: Record<string, any>[]) {
+  async addInitialPages(pagesImport: Record<string, any>[]) {
     const pages: PageEntry[] = [];
     for (const {
       id,