@embedpdf/plugin-annotation 2.11.1 → 2.12.1

package/dist/index.js

@@ -1,5 +1,5 @@
 import { clamp, BasePlugin, createScopedEmitter, createBehaviorEmitter } from "@embedpdf/core";
-import { PdfAnnotationSubtype, PdfAnnotationReplyType, expandRect, rectFromPoints, uuidV4, PdfAnnotationLineEnding, getRectCenter, inferRotationCenterFromRects, calculateRotatedRectAABBAroundPoint, rotateAndTranslatePoint, calculateRotatedRectAABB, rotatePointAround, rotateVertices, PdfAnnotationName, PdfVerticalAlignment, PdfTextAlignment, PdfStandardFont, PdfAnnotationBorderStyle, PdfBlendMode, PdfPermissionFlag, ignore, PdfTaskHelper, PdfErrorCode, Task, PdfActionType, PdfZoomMode, TaskStage } from "@embedpdf/models";
+import { PdfAnnotationSubtype, PdfAnnotationReplyType, expandRect, rectFromPoints, uuidV4, PdfAnnotationLineEnding, getRectCenter, inferRotationCenterFromRects, calculateRotatedRectAABBAroundPoint, rotateAndTranslatePoint, calculateRotatedRectAABB, rotatePointAround, rotateVertices, PdfAnnotationName, PdfVerticalAlignment, PdfTextAlignment, PdfStandardFont, PdfAnnotationBorderStyle, getImageMetadata, fitSizeWithin, PdfBlendMode, PdfPermissionFlag, ignore, PdfTaskHelper, PdfErrorCode, Task, PdfActionType, PdfZoomMode, TaskStage } from "@embedpdf/models";
 import { calculateRotatedRectAABB as calculateRotatedRectAABB2, calculateRotatedRectAABBAroundPoint as calculateRotatedRectAABBAroundPoint2, getRectCenter as getRectCenter2, inferRotationCenterFromRects as inferRotationCenterFromRects2, rotatePointAround as rotatePointAround2, rotateVertices as rotateVertices2 } from "@embedpdf/models";
 const ANNOTATION_PLUGIN_ID = "annotation";
 const manifest = {
@@ -2201,62 +2201,125 @@ const squareHandlerFactory = {
     };
   }
 };
+const imageFetchCache = /* @__PURE__ */ new Map();
 const stampHandlerFactory = {
   annotationType: PdfAnnotationSubtype.STAMP,
   create(context) {
-    const { services, onCommit, getTool, pageSize, pageRotation } = context;
+    const { services, onCommit, onPreview, getTool, pageSize, pageRotation } = context;
+    let cachedBuffer = null;
+    let cachedSize = null;
+    const commitStamp = (pos, width, height, ctx) => {
+      var _a;
+      const tool = getTool();
+      if (!tool) return;
+      const rect = {
+        origin: { x: pos.x - width / 2, y: pos.y - height / 2 },
+        size: { width, height }
+      };
+      let anno = {
+        ...tool.defaults,
+        rect,
+        type: PdfAnnotationSubtype.STAMP,
+        name: tool.defaults.name ?? PdfAnnotationName.Image,
+        subject: tool.defaults.subject ?? "Stamp",
+        flags: tool.defaults.flags ?? ["print"],
+        pageIndex: context.pageIndex,
+        id: uuidV4(),
+        created: /* @__PURE__ */ new Date()
+      };
+      if ((_a = tool.behavior) == null ? void 0 : _a.insertUpright) {
+        anno = applyInsertUpright(anno, pageRotation, false);
+      }
+      anno = clampAnnotationToPage(anno, pageSize);
+      onCommit(anno, ctx);
+    };
+    const commitFromBuffer = (pos, buffer, imageSize) => {
+      const meta = getImageMetadata(buffer);
+      if (!meta || meta.mimeType === "application/pdf") return false;
+      const fitted = fitSizeWithin(meta, pageSize);
+      const width = (imageSize == null ? void 0 : imageSize.width) ?? fitted.width;
+      const height = (imageSize == null ? void 0 : imageSize.height) ?? fitted.height;
+      commitStamp(pos, width, height, { data: buffer });
+      return true;
+    };
     return {
+      onHandlerActiveStart: () => {
+        const tool = getTool();
+        const imageSrc = tool == null ? void 0 : tool.defaults.imageSrc;
+        if (!imageSrc) return;
+        let entry = imageFetchCache.get(imageSrc);
+        if (!entry) {
+          const promise = fetch(imageSrc).then((res) => res.arrayBuffer()).then((buffer) => {
+            const meta = getImageMetadata(buffer);
+            if (!meta || meta.mimeType === "application/pdf") return null;
+            const fitted = fitSizeWithin(meta, pageSize);
+            const imageSize = tool.defaults.imageSize;
+            return {
+              buffer,
+              width: (imageSize == null ? void 0 : imageSize.width) ?? fitted.width,
+              height: (imageSize == null ? void 0 : imageSize.height) ?? fitted.height
+            };
+          }).catch(() => null);
+          entry = { promise, refs: 1 };
+          imageFetchCache.set(imageSrc, entry);
+        } else {
+          entry.refs++;
+        }
+        entry.promise.then((result) => {
+          if (!result) return;
+          cachedBuffer = result.buffer;
+          cachedSize = { width: result.width, height: result.height };
+        });
+      },
+      onHandlerActiveEnd: () => {
+        const tool = getTool();
+        const imageSrc = tool == null ? void 0 : tool.defaults.imageSrc;
+        if (imageSrc) {
+          const entry = imageFetchCache.get(imageSrc);
+          if (entry && --entry.refs <= 0) {
+            imageFetchCache.delete(imageSrc);
+          }
+        }
+        cachedBuffer = null;
+        cachedSize = null;
+        onPreview(null);
+      },
+      onPointerMove: (pos) => {
+        var _a;
+        const tool = getTool();
+        if (!((_a = tool == null ? void 0 : tool.behavior) == null ? void 0 : _a.showGhost) || !cachedSize || !tool.defaults.imageSrc) return;
+        const rect = {
+          origin: { x: pos.x - cachedSize.width / 2, y: pos.y - cachedSize.height / 2 },
+          size: cachedSize
+        };
+        onPreview({
+          type: PdfAnnotationSubtype.STAMP,
+          bounds: rect,
+          data: { rect, ghostUrl: tool.defaults.imageSrc, pageRotation }
+        });
+      },
       onPointerDown: (pos) => {
         const tool = getTool();
         if (!tool) return;
         const { imageSrc, imageSize } = tool.defaults;
-        const placeStamp = (imageData, width, height) => {
-          var _a;
-          const rect = {
-            origin: { x: pos.x - width / 2, y: pos.y - height / 2 },
-            size: { width, height }
-          };
-          let anno = {
-            ...tool.defaults,
-            rect,
-            type: PdfAnnotationSubtype.STAMP,
-            name: tool.defaults.name ?? PdfAnnotationName.Image,
-            subject: tool.defaults.subject ?? "Stamp",
-            flags: tool.defaults.flags ?? ["print"],
-            pageIndex: context.pageIndex,
-            id: uuidV4(),
-            created: /* @__PURE__ */ new Date()
-          };
-          if ((_a = tool.behavior) == null ? void 0 : _a.insertUpright) {
-            anno = applyInsertUpright(anno, pageRotation, false);
-          }
-          anno = clampAnnotationToPage(anno, pageSize);
-          onCommit(anno, { imageData });
-        };
         if (imageSrc) {
-          services.processImage({
-            source: imageSrc,
-            maxWidth: pageSize.width,
-            maxHeight: pageSize.height,
-            onComplete: (result) => placeStamp(
-              result.imageData,
-              (imageSize == null ? void 0 : imageSize.width) ?? result.width,
-              (imageSize == null ? void 0 : imageSize.height) ?? result.height
-            )
-          });
+          onPreview(null);
+          if (cachedBuffer) {
+            commitFromBuffer(pos, cachedBuffer, imageSize);
+          } else {
+            fetch(imageSrc).then((res) => res.arrayBuffer()).then((buffer) => commitFromBuffer(pos, buffer, imageSize));
+          }
         } else {
           services.requestFile({
             accept: "image/png,image/jpeg",
             onFile: (file) => {
-              services.processImage({
-                source: file,
-                maxWidth: pageSize.width,
-                maxHeight: pageSize.height,
-                onComplete: (result) => placeStamp(result.imageData, result.width, result.height)
-              });
+              file.arrayBuffer().then((buffer) => commitFromBuffer(pos, buffer));
             }
           });
         }
+      },
+      onPointerLeave: () => {
+        onPreview(null);
       }
     };
   }
@@ -4295,6 +4358,7 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
       getSelectedAnnotations: () => this.getSelectedAnnotationsMethod(),
       getSelectedAnnotationIds: () => this.getSelectedAnnotationIdsMethod(),
       getAnnotationById: (id) => this.getAnnotationById(id),
+      getAnnotations: (options) => this.getAnnotationsMethod(options),
       selectAnnotation: (pageIndex, id) => this.selectAnnotation(pageIndex, id),
       toggleSelection: (pageIndex, id) => this.toggleSelectionMethod(pageIndex, id),
       addToSelection: (pageIndex, id) => this.addToSelectionMethod(pageIndex, id),
@@ -4302,12 +4366,14 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
       setSelection: (ids) => this.setSelectionMethod(ids),
       deselectAnnotation: () => this.deselectAnnotation(),
       importAnnotations: (items) => this.importAnnotations(items),
+      exportAnnotations: (options, documentId) => this.exportAnnotationsMethod(options, documentId),
       createAnnotation: (pageIndex, anno, ctx) => this.createAnnotation(pageIndex, anno, ctx),
       updateAnnotation: (pageIndex, id, patch) => this.updateAnnotation(pageIndex, id, patch),
       updateAnnotations: (patches) => this.updateAnnotationsMethod(patches),
       moveAnnotation: (pageIndex, id, position, mode, documentId) => this.moveAnnotationMethod(pageIndex, id, position, mode, documentId),
       deleteAnnotation: (pageIndex, id) => this.deleteAnnotation(pageIndex, id),
       deleteAnnotations: (annotations, documentId) => this.deleteAnnotationsMethod(annotations, documentId),
+      deleteAllAnnotations: (documentId) => this.deleteAllAnnotationsMethod(documentId),
       purgeAnnotation: (pageIndex, id, documentId) => this.purgeAnnotationMethod(pageIndex, id, documentId),
       renderAnnotation: (options) => this.renderAnnotation(options),
       getPageAppearances: (pageIndex, options, documentId) => this.getPageAppearances(pageIndex, options, documentId),
@@ -4365,6 +4431,7 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
       getSelectedAnnotations: () => this.getSelectedAnnotationsMethod(documentId),
       getSelectedAnnotationIds: () => this.getSelectedAnnotationIdsMethod(documentId),
       getAnnotationById: (id) => this.getAnnotationById(id, documentId),
+      getAnnotations: (options) => this.getAnnotationsMethod(options, documentId),
       selectAnnotation: (pageIndex, id) => this.selectAnnotation(pageIndex, id, documentId),
       toggleSelection: (pageIndex, id) => this.toggleSelectionMethod(pageIndex, id, documentId),
       addToSelection: (pageIndex, id) => this.addToSelectionMethod(pageIndex, id, documentId),
@@ -4375,12 +4442,14 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
       setActiveTool: (toolId, context) => this.setActiveTool(toolId, documentId, context),
       findToolForAnnotation: (anno) => this.findToolForAnnotation(anno),
       importAnnotations: (items) => this.importAnnotations(items, documentId),
+      exportAnnotations: (options) => this.exportAnnotationsMethod(options, documentId),
       createAnnotation: (pageIndex, anno, ctx) => this.createAnnotation(pageIndex, anno, ctx, documentId),
       updateAnnotation: (pageIndex, id, patch) => this.updateAnnotation(pageIndex, id, patch, documentId),
       updateAnnotations: (patches) => this.updateAnnotationsMethod(patches, documentId),
       moveAnnotation: (pageIndex, id, position, mode) => this.moveAnnotationMethod(pageIndex, id, position, mode, documentId),
       deleteAnnotation: (pageIndex, id) => this.deleteAnnotation(pageIndex, id, documentId),
       deleteAnnotations: (annotations) => this.deleteAnnotationsMethod(annotations, documentId),
+      deleteAllAnnotations: () => this.deleteAllAnnotationsMethod(documentId),
       purgeAnnotation: (pageIndex, id) => this.purgeAnnotationMethod(pageIndex, id, documentId),
       renderAnnotation: (options) => this.renderAnnotation(options, documentId),
       getPageAppearances: (pageIndex, options) => this.getPageAppearances(pageIndex, options, documentId),
@@ -4540,6 +4609,15 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
     const docState = this.getDocumentState(documentId);
     return getAnnotationByUid(docState, id);
   }
+  getAnnotationsMethod(options, documentId) {
+    const docState = this.getDocumentState(documentId);
+    if ((options == null ? void 0 : options.pageIndex) !== void 0) {
+      return getAnnotationsByPageIndex(docState, options.pageIndex).filter(
+        (ta) => ta !== void 0
+      );
+    }
+    return Object.values(docState.byUid);
+  }
   renderAnnotation({ pageIndex, annotation, options }, documentId) {
     const id = documentId ?? this.getActiveDocumentId();
     const docState = this.getCoreDocument(id);
@@ -4614,6 +4692,74 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
     if (!pageMap) return;
     delete pageMap[annotId];
   }
+  exportAnnotationsMethod(options, documentId) {
+    const id = documentId ?? this.getActiveDocumentId();
+    const docState = this.getDocumentState(id);
+    const coreDoc = this.getCoreDocument(id);
+    const doc = coreDoc == null ? void 0 : coreDoc.document;
+    if (!doc) {
+      return PdfTaskHelper.reject({ code: PdfErrorCode.NotFound, message: "Document not found" });
+    }
+    const pageIndices = (options == null ? void 0 : options.pageIndex) !== void 0 ? [options.pageIndex] : Object.keys(docState.pages).map(Number);
+    const entries = [];
+    for (const pi of pageIndices) {
+      const uids = docState.pages[pi] ?? [];
+      for (const uid of uids) {
+        const ta = docState.byUid[uid];
+        if (!ta || ta.commitState === "deleted") continue;
+        entries.push({
+          annotation: ta.object,
+          pageIndex: pi,
+          isStamp: ta.object.type === PdfAnnotationSubtype.STAMP
+        });
+      }
+    }
+    const stampEntries = entries.filter((e) => e.isStamp);
+    if (stampEntries.length === 0) {
+      return PdfTaskHelper.resolve(entries.map((e) => ({ annotation: e.annotation })));
+    }
+    const resultTask = PdfTaskHelper.create();
+    const appearances = /* @__PURE__ */ new Map();
+    let pending = stampEntries.length;
+    let failed = false;
+    for (const entry of stampEntries) {
+      const page = doc.pages.find((p) => p.index === entry.pageIndex);
+      if (!page) {
+        pending--;
+        continue;
+      }
+      const exportTask = this.engine.exportAnnotationAppearanceAsPdf(doc, page, entry.annotation);
+      exportTask.wait(
+        (buffer) => {
+          if (failed) return;
+          appearances.set(entry.annotation.id, buffer);
+          pending--;
+          if (pending === 0) {
+            resultTask.resolve(
+              entries.map((e) => ({
+                annotation: e.annotation,
+                ...appearances.has(e.annotation.id) ? {
+                  ctx: {
+                    data: appearances.get(e.annotation.id),
+                    mimeType: "application/pdf"
+                  }
+                } : {}
+              }))
+            );
+          }
+        },
+        (error) => {
+          if (failed) return;
+          failed = true;
+          resultTask.reject(error.reason);
+        }
+      );
+    }
+    if (pending === 0) {
+      resultTask.resolve(entries.map((e) => ({ annotation: e.annotation })));
+    }
+    return resultTask;
+  }
   importAnnotations(items, documentId) {
     const id = documentId ?? this.getActiveDocumentId();
     if (!this.isInitialLoadComplete.get(id)) {
@@ -4841,6 +4987,20 @@ const _AnnotationPlugin = class _AnnotationPlugin extends BasePlugin {
       this.deleteAnnotation(pageIndex, id, documentId);
     }
   }
+  deleteAllAnnotationsMethod(documentId) {
+    const docId = documentId ?? this.getActiveDocumentId();
+    const docState = this.getDocumentState(docId);
+    const toDelete = [];
+    for (const [pageIdx, uids] of Object.entries(docState.pages)) {
+      for (const uid of uids) {
+        const ta = docState.byUid[uid];
+        if (ta) toDelete.push({ pageIndex: Number(pageIdx), id: ta.object.id });
+      }
+    }
+    if (toDelete.length > 0) {
+      this.deleteAnnotationsMethod(toDelete, docId);
+    }
+  }
   purgeAnnotationMethod(pageIndex, id, documentId) {
     const docId = documentId ?? this.getActiveDocumentId();
     this.dispatch(purgeAnnotation(docId, pageIndex, id));