@embedpdf/plugin-redaction 2.9.1 → 2.10.0

package/dist/index.js

@@ -1,5 +1,6 @@
 import { BasePlugin, createBehaviorEmitter, refreshPages } from "@embedpdf/core";
 import { PdfAnnotationSubtype, PdfPermissionFlag, uuidV4, PdfTaskHelper, PdfErrorCode, Task } from "@embedpdf/models";
+import { defineAnnotationTool } from "@embedpdf/plugin-annotation";
 var RedactionMode = /* @__PURE__ */ ((RedactionMode2) => {
   RedactionMode2["Redact"] = "redact";
   RedactionMode2["MarqueeRedact"] = "marqueeRedact";
@@ -257,9 +258,10 @@ const redactionReducer = (state = initialState, action) => {
       return state;
   }
 };
-const redactTool = {
+const redactTool = defineAnnotationTool({
   id: "redact",
   name: "Redact",
+  categories: ["annotation", "redaction"],
   matchScore: (a) => a.type === PdfAnnotationSubtype.REDACT ? 10 : 0,
   interaction: {
     mode: RedactionMode.Redact,
@@ -293,7 +295,7 @@ const redactTool = {
   behavior: {
     useAppearanceStream: false
   }
-};
+});
 const redactTools = [redactTool];
 const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
   constructor(id, registry, config) {
@@ -325,7 +327,7 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
     } else {
       this.useAnnotationMode = false;
     }
-    if (this.useAnnotationMode) {
+    if (this.useAnnotationMode && this.annotationCapability) {
       for (const tool of redactTools) {
         this.annotationCapability.addTool(tool);
       }
@@ -680,6 +682,14 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
     }
     return state;
   }
+  isRedactTool(tool) {
+    return (tool == null ? void 0 : tool.id) === redactTool.id && tool.defaults.type === PdfAnnotationSubtype.REDACT;
+  }
+  getRedactTool() {
+    var _a;
+    const tool = (_a = this.annotationCapability) == null ? void 0 : _a.getTool("redact");
+    return this.isRedactTool(tool) ? tool : void 0;
+  }
   // ─────────────────────────────────────────────────────────
   // Annotation Mode State Sync
   // ─────────────────────────────────────────────────────────
@@ -763,6 +773,7 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
       return;
     }
     if (this.useAnnotationMode) {
+      if (!this.annotationCapability) return;
       const annoScope = this.annotationCapability.forDocument(id);
       for (const item of items) {
         const annotation = this.redactionItemToAnnotation(item);
@@ -962,9 +973,9 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
    * In legacy mode: returns hardcoded red
    */
   getPreviewStrokeColor() {
-    if (this.useAnnotationMode && this.annotationCapability) {
-      const tool = this.annotationCapability.getTool("redact");
-      return (tool == null ? void 0 : tool.defaults.strokeColor) ?? "#FF0000";
+    var _a;
+    if (this.useAnnotationMode) {
+      return ((_a = this.getRedactTool()) == null ? void 0 : _a.defaults.strokeColor) ?? "#FF0000";
     }
     return "#FF0000";
   }
@@ -1222,7 +1233,14 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
    * Annotation mode: Apply all redactions using engine.applyAllRedactions per page
    */
   applyAllRedactionsAnnotationMode(docId, doc) {
-    const annoState = this.annotationCapability.forDocument(docId).getState();
+    const annotationCapability = this.annotationCapability;
+    if (!annotationCapability) {
+      return PdfTaskHelper.reject({
+        code: PdfErrorCode.NotFound,
+        message: "Annotation capability not found"
+      });
+    }
+    const annoState = annotationCapability.forDocument(docId).getState();
     const redactAnnotationsByPage = /* @__PURE__ */ new Map();
     for (const ta of Object.values(annoState.byUid)) {
       if (ta.object.type === PdfAnnotationSubtype.REDACT) {
@@ -1244,18 +1262,18 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
     const task = new Task();
     Task.all(tasks).wait(
       () => {
-        var _a, _b;
-        const annoScope = (_a = this.annotationCapability) == null ? void 0 : _a.forDocument(docId);
+        var _a;
+        const annoScope = annotationCapability.forDocument(docId);
         const allPurgedIds = [];
         for (const [pageIndex, ids] of redactAnnotationsByPage) {
           for (const id of ids) {
-            annoScope == null ? void 0 : annoScope.purgeAnnotation(pageIndex, id);
+            annoScope.purgeAnnotation(pageIndex, id);
             this.dispatch(removePending(docId, pageIndex, id));
             allPurgedIds.push(id);
           }
         }
         if (allPurgedIds.length > 0) {
-          (_b = this.historyCapability) == null ? void 0 : _b.forDocument(docId).purgeByMetadata(
+          (_a = this.historyCapability) == null ? void 0 : _a.forDocument(docId).purgeByMetadata(
             (meta) => {
               var _a2;
               return ((_a2 = meta == null ? void 0 : meta.annotationIds) == null ? void 0 : _a2.some((id) => allPurgedIds.includes(id))) ?? false;
@@ -1287,10 +1305,10 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
    * @returns Array of annotation IDs that were created
    */
   createRedactAnnotationsFromSelection(documentId, formattedSelection, text) {
+    var _a;
     if (!this.annotationCapability) return [];
     const annoScope = this.annotationCapability.forDocument(documentId);
-    const tool = this.annotationCapability.getTool("redact");
-    const defaults = tool == null ? void 0 : tool.defaults;
+    const defaults = (_a = this.getRedactTool()) == null ? void 0 : _a.defaults;
     const annotationIds = [];
     for (const selection of formattedSelection) {
       const annotationId = uuidV4();
@@ -1354,10 +1372,10 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
    * Create a REDACT annotation from an area/marquee selection (annotation mode only)
    */
   createRedactAnnotationFromArea(documentId, pageIndex, rect) {
+    var _a;
     if (!this.annotationCapability) return;
     const annoScope = this.annotationCapability.forDocument(documentId);
-    const tool = this.annotationCapability.getTool("redact");
-    const defaults = tool == null ? void 0 : tool.defaults;
+    const defaults = (_a = this.getRedactTool()) == null ? void 0 : _a.defaults;
     const annotationId = uuidV4();
     const annotation = {
       ...defaults,
@@ -1377,8 +1395,7 @@ const _RedactionPlugin = class _RedactionPlugin extends BasePlugin {
    */
   redactionItemToAnnotation(item) {
     var _a;
-    const tool = (_a = this.annotationCapability) == null ? void 0 : _a.getTool("redact");
-    const defaults = (tool == null ? void 0 : tool.defaults) ?? {};
+    const defaults = ((_a = this.getRedactTool()) == null ? void 0 : _a.defaults) ?? {};
     return {
       ...defaults,
       id: item.id,