@pagepocket/main-content-unit 0.12.0 → 0.13.0

package/dist/main-content-unit.d.ts

@@ -1,15 +1,15 @@
-import { Unit, type FileTree, type UnitContext, type UnitPatch, type UnitRuntime } from "@pagepocket/lib";
+import { SnapshotUnit, type UnitContext, type UnitRuntime } from "@pagepocket/lib";
 export type MainContentUnitOptions = {
     markdown?: boolean;
 };
-export declare class MainContentUnit extends Unit {
-    readonly id = "mainContent";
-    readonly kind = "extract.mainContent";
+export declare class MainContentUnit extends SnapshotUnit {
+    readonly id = "buildSnapshot";
+    readonly description = "Extracting main content";
+    readonly snapshotType = "main-content";
     private options;
     constructor(options?: MainContentUnitOptions);
-    merge(returnValue: UnitPatch, pluginContributedValue?: UnitPatch): UnitPatch;
-    run(ctx: UnitContext, _rt: UnitRuntime): Promise<{
-        files: FileTree;
+    build(ctx: UnitContext, _rt: UnitRuntime): Promise<{
+        files: import("@pagepocket/lib").FileTree;
         html: {
             htmlString: string;
             baseUrl: string;

package/dist/main-content-unit.js

@@ -1,23 +1,15 @@
-import { mergeFileTrees, Unit } from "@pagepocket/lib";
+import { SnapshotUnit } from "@pagepocket/lib";
 import { Defuddle } from "defuddle/node";
 import { buildContentTree } from "./build-content-tree.js";
-export class MainContentUnit extends Unit {
+export class MainContentUnit extends SnapshotUnit {
     constructor(options) {
         super();
-        this.id = "mainContent";
-        this.kind = "extract.mainContent";
+        this.id = "buildSnapshot";
+        this.description = "Extracting main content";
+        this.snapshotType = "main-content";
         this.options = options ?? {};
     }
-    merge(returnValue, pluginContributedValue = {}) {
-        const mergedValue = { ...returnValue, ...pluginContributedValue };
-        const returnFiles = returnValue.files;
-        const pluginFiles = pluginContributedValue.files;
-        if (!isFileTree(returnFiles) || !isFileTree(pluginFiles)) {
-            return mergedValue;
-        }
-        return { ...mergedValue, files: mergeFileTrees(returnFiles, pluginFiles) };
-    }
-    async run(ctx, _rt) {
+    async build(ctx, _rt) {
         const capture = ctx.value.capture;
         if (!capture) {
             throw new Error("MainContentUnit requires ctx.value.capture");
@@ -44,19 +36,3 @@ export class MainContentUnit extends Unit {
         return result.content;
     }
 }
-const isFileTree = (value) => {
-    if (!value || typeof value !== "object") {
-        return false;
-    }
-    if (!("root" in value)) {
-        return false;
-    }
-    const root = value.root;
-    if (!root || typeof root !== "object") {
-        return false;
-    }
-    const rootRecord = root;
-    return (rootRecord.kind === "directory" &&
-        typeof rootRecord.path === "string" &&
-        Array.isArray(rootRecord.entries));
-};

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pagepocket/main-content-unit",
-  "version": "0.12.0",
+  "version": "0.13.0",
   "description": "PagePocket unit: extract main content from captured HTML using defuddle and rewrite resource URLs to local paths",
   "type": "module",
   "main": "dist/index.js",
@@ -13,8 +13,8 @@
     "cheerio": "^1.0.0",
     "defuddle": "^0.6.0",
     "jsdom": "^26.0.0",
-    "@pagepocket/lib": "0.12.0",
-    "@pagepocket/shared": "0.12.0"
+    "@pagepocket/lib": "0.13.0",
+    "@pagepocket/shared": "0.13.0"
   },
   "devDependencies": {
     "typescript": "^5.4.5",