fork from https://github.com/mozilla/pdf.js.git

web/pdf_attachment_viewer.js

@@ -0,0 +1,202 @@
+/* Copyright 2012 Mozilla Foundation
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *     http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
+/** @typedef {import("./event_utils.js").EventBus} EventBus */
+// eslint-disable-next-line max-len
+/** @typedef {import("./download_manager.js").DownloadManager} DownloadManager */
+
+/**
+ * @import {
+ *   CatalogAttachmentContent,
+ *   CatalogAttachment,
+ * } from "../src/core/catalog.js";
+ * @import { PDFLinkService } from "./pdf_link_service.js";
+ */
+
+import { BaseTreeViewer } from "./base_tree_viewer.js";
+import { internalOpt } from "./internal_evt.js";
+import { waitOnEventOrTimeout } from "./event_utils.js";
+
+/**
+ * @typedef {Object} PDFAttachmentViewerOptions
+ * @property {HTMLDivElement} container - The viewer element.
+ * @property {EventBus} eventBus - The application event bus.
+ * @property {DownloadManager} downloadManager - The download manager.
+ * @property {PDFLinkService} linkService - Link service.
+ */
+
+/**
+ * @typedef PDFAttachmentViewerRenderParameters
+ * @property {Record<string, CatalogAttachment> | null} attachments - A lookup
+ *   table of attachment objects.
+ * @property {boolean} [keepRenderedCapability]
+ */
+
+class PDFAttachmentViewer extends BaseTreeViewer {
+  /**
+   * @param {PDFAttachmentViewerOptions} options
+   */
+  constructor(options) {
+    super(options);
+    this.downloadManager = options.downloadManager;
+    this.linkService = options.linkService;
+
+    this.eventBus.on(
+      "fileattachmentannotation",
+      this.#appendAttachment.bind(this),
+      internalOpt
+    );
+  }
+
+  reset(keepRenderedCapability = false) {
+    super.reset();
+    this._attachments = null;
+
+    if (!keepRenderedCapability) {
+      // The only situation in which the `_renderedCapability` should *not* be
+      // replaced is when appending FileAttachment annotations.
+      this._renderedCapability = Promise.withResolvers();
+    }
+    this._pendingDispatchEvent = false;
+  }
+
+  /**
+   * @protected
+   */
+  async _dispatchEvent(attachmentsCount) {
+    this._renderedCapability.resolve();
+
+    if (attachmentsCount === 0 && !this._pendingDispatchEvent) {
+      // Delay the event when no "regular" attachments exist, to allow time for
+      // parsing of any FileAttachment annotations that may be present on the
+      // *initially* rendered page; this reduces the likelihood of temporarily
+      // disabling the attachmentsView when the `PDFSidebar` handles the event.
+      this._pendingDispatchEvent = true;
+
+      await waitOnEventOrTimeout({
+        target: this.eventBus,
+        name: "annotationlayerrendered",
+        delay: 1000,
+      });
+
+      if (!this._pendingDispatchEvent) {
+        return; // There was already another `_dispatchEvent`-call`.
+      }
+    }
+    this._pendingDispatchEvent = false;
+
+    this.eventBus.dispatch("attachmentsloaded", {
+      source: this,
+      attachmentsCount,
+    });
+  }
+
+  /**
+   * @param {HTMLAnchorElement} element
+   * @param {CatalogAttachment & { attachmentId?: string }} item
+   * @returns {undefined}
+   * @protected
+   */
+  _bindLink(
+    element,
+    { attachmentId, content: fallbackContent, description, filename }
+  ) {
+    if (description) {
+      element.title = description;
+    }
+
+    const openAttachment = async () => {
+      /** @type {CatalogAttachmentContent | undefined} */
+      // Prefer lazy loading when we have an attachment id; fallbackContent is
+      // only for the annotation-append path where bytes may already be present.
+      const content = attachmentId
+        ? await this.linkService.getAttachmentContent(attachmentId)
+        : fallbackContent;
+
+      if (content) {
+        this.downloadManager.openOrDownloadData(content, filename);
+      }
+    };
+
+    element.onclick = () => {
+      openAttachment();
+      return false;
+    };
+  }
+
+  /**
+   * @param {PDFAttachmentViewerRenderParameters} params
+   */
+  render({ attachments, keepRenderedCapability = false }) {
+    if (this._attachments) {
+      this.reset(keepRenderedCapability);
+    }
+    this._attachments = attachments || null;
+
+    if (!attachments) {
+      this._dispatchEvent(/* attachmentsCount = */ 0);
+      return;
+    }
+
+    const fragment = document.createDocumentFragment();
+    const ul = document.createElement("ul");
+    fragment.append(ul);
+    let attachmentsCount = 0;
+    for (const name in attachments) {
+      const item = attachments[name];
+      const li = document.createElement("li");
+      ul.append(li);
+      const element = document.createElement("a");
+      li.append(element);
+      this._bindLink(element, {
+        ...item,
+        attachmentId: item.attachmentId ?? name,
+      });
+      element.textContent = this._normalizeTextContent(item.filename);
+
+      attachmentsCount++;
+    }
+
+    this._finishRendering(fragment, attachmentsCount);
+  }
+
+  /**
+   * Used to append FileAttachment annotations to the sidebar.
+   */
+  #appendAttachment(item) {
+    const renderedPromise = this._renderedCapability.promise;
+
+    renderedPromise.then(() => {
+      if (renderedPromise !== this._renderedCapability.promise) {
+        return; // The FileAttachment annotation belongs to a previous document.
+      }
+      const attachments = this._attachments || Object.create(null);
+
+      for (const name in attachments) {
+        if (item.filename === name) {
+          return; // Ignore the new attachment if it already exists.
+        }
+      }
+      attachments[item.filename] = item;
+
+      this.render({
+        attachments,
+        keepRenderedCapability: true,
+      });
+    });
+  }
+}
+
+export { PDFAttachmentViewer };