@xenterprises/fastify-xpdf 1.0.0

package/src/xPDF.js

@@ -0,0 +1,126 @@
+// src/xPDF.js
+import fp from "fastify-plugin";
+import puppeteer from "puppeteer";
+import { getGeneratorMethods } from "./services/generator.js";
+import { getFormsMethods } from "./services/forms.js";
+import { getMergerMethods } from "./services/merger.js";
+
+/**
+ * xPDF Plugin for Fastify
+ * PDF generation and manipulation library with optional xStorage integration
+ *
+ * Provides methods for:
+ * - Generate PDFs from HTML/Markdown (Puppeteer-based)
+ * - Fill PDF forms and flatten them
+ * - Merge multiple PDFs
+ * - Optional automatic saving to S3-compatible storage via xStorage
+ *
+ * Usage:
+ *   await fastify.xPDF.generateFromHtml('<h1>Hello</h1>')
+ *   await fastify.xPDF.generateFromMarkdown('# Hello')
+ *   await fastify.xPDF.fillForm(pdfBuffer, { fieldName: 'value' })
+ *   await fastify.xPDF.mergePDFs([pdfBuffer1, pdfBuffer2])
+ */
+async function xPDFPlugin(fastify, options) {
+  const {
+    // Puppeteer options
+    headless = true,
+    args = ["--no-sandbox", "--disable-setuid-sandbox"],
+
+    // xStorage integration (optional)
+    useStorage = false,
+    defaultFolder = "pdfs",
+
+    // PDF generation defaults
+    format = "A4",
+    printBackground = true,
+    margin = {
+      top: "1cm",
+      right: "1cm",
+      bottom: "1cm",
+      left: "1cm",
+    },
+  } = options;
+
+  console.info("\n 📄 Starting xPDF...\n");
+
+  // Browser instance (lazy initialized)
+  let browser = null;
+
+  /**
+   * Get or initialize browser instance
+   * Implements lazy initialization and reconnection handling
+   */
+  const getBrowser = async () => {
+    try {
+      // Reconnect if existing browser disconnected
+      if (browser && !browser.isConnected()) {
+        browser = null;
+      }
+
+      if (!browser) {
+        browser = await puppeteer.launch({
+          headless,
+          args,
+        });
+      }
+
+      return browser;
+    } catch (error) {
+      fastify.log.error("Failed to initialize browser:", error);
+      throw new Error(`Failed to initialize PDF browser: ${error.message}`);
+    }
+  };
+
+  // Store configuration
+  const config = {
+    getBrowser,
+    fastify,
+    format,
+    printBackground,
+    margin,
+    useStorage,
+    defaultFolder,
+  };
+
+  // Check if xStorage is available
+  const hasXStorage = fastify.hasDecorator("xStorage");
+  if (useStorage && !hasXStorage) {
+    fastify.log.warn(
+      "xPDF: useStorage is true but xStorage plugin not registered. PDFs will not be saved to storage."
+    );
+  }
+
+  // Get all PDF methods
+  const generatorMethods = getGeneratorMethods(config);
+  const formsMethods = getFormsMethods(config);
+  const mergerMethods = getMergerMethods(config);
+
+  // Combine all methods
+  const pdfMethods = {
+    ...generatorMethods,
+    ...formsMethods,
+    ...mergerMethods,
+  };
+
+  // Decorate Fastify with xPDF namespace
+  fastify.decorate("xPDF", pdfMethods);
+
+  // Cleanup on Fastify shutdown
+  fastify.addHook("onClose", async () => {
+    if (browser) {
+      try {
+        await browser.close();
+        fastify.log.info("xPDF browser closed");
+      } catch (error) {
+        fastify.log.warn("Error closing xPDF browser:", error.message);
+      }
+    }
+  });
+
+  console.info("\n 📄 xPDF Ready!\n");
+}
+
+export default fp(xPDFPlugin, {
+  name: "xPDF",
+});