pdf-smith 0.2.0 → 0.3.0

package/src/preview/preview-plugin.ts

@@ -1,4 +1,8 @@
+import fs from 'node:fs';
+import type { AddressInfo } from 'node:net';
+import path from 'node:path';
 import type { Plugin } from 'vite';
+import type { DocumentConfig } from '../config';
 
 const VIRTUAL_DOCUMENTS_ID = 'virtual:pdf-smith-documents';
 const RESOLVED_VIRTUAL_DOCUMENTS_ID = `\0${VIRTUAL_DOCUMENTS_ID}`;
@@ -7,11 +11,110 @@ interface PreviewPluginOptions {
   pkgSrcDir: string;
 }
 
+async function handleExportRequest(
+  server: import('vite').ViteDevServer,
+  slug: string,
+  res: import('node:http').ServerResponse,
+) {
+  const root = server.config.root;
+  const pagesDir = path.join(root, 'pdfs', slug, 'pages');
+
+  if (!fs.existsSync(pagesDir)) {
+    res.statusCode = 404;
+    res.setHeader('Content-Type', 'application/json');
+    res.end(JSON.stringify({ error: `Document "${slug}" not found` }));
+    return;
+  }
+
+  let config: DocumentConfig | undefined;
+  const configPath = path.join(root, 'pdfs', slug, 'config.ts');
+  if (fs.existsSync(configPath)) {
+    try {
+      const configModule = await server.ssrLoadModule(configPath);
+      config = configModule.default;
+    } catch {
+      // Config loading failed, proceed without it
+    }
+  }
+
+  let playwright: typeof import('playwright');
+  try {
+    playwright = await import('playwright');
+  } catch {
+    res.statusCode = 500;
+    res.setHeader('Content-Type', 'application/json');
+    res.end(
+      JSON.stringify({
+        error:
+          'Playwright is required for PDF export. Install it with: npm install -D playwright && npx playwright install chromium',
+      }),
+    );
+    return;
+  }
+
+  let browser: import('playwright').Browser | undefined;
+  try {
+    const address = server.httpServer?.address() as AddressInfo | null;
+    const port = address?.port;
+    if (!port) throw new Error('Could not determine server port');
+
+    browser = await playwright.chromium.launch({ headless: true });
+    const page = await browser.newPage();
+    await page.goto(`http://localhost:${port}/${slug}`, { waitUntil: 'networkidle' });
+    await page.waitForSelector('[data-pdf-smith-page]', { timeout: 30_000 });
+
+    const margin = { top: '0', bottom: '0', left: '0', right: '0' };
+    const pdfOptions: Parameters<typeof page.pdf>[0] = {
+      preferCSSPageSize: true,
+      printBackground: true,
+      margin,
+    };
+
+    if (config?.pageNumbers?.enabled) {
+      pdfOptions.displayHeaderFooter = true;
+      pdfOptions.headerTemplate = '<span></span>';
+      pdfOptions.footerTemplate =
+        config.pageNumbers.template ??
+        '<div style="font-size:10px;text-align:center;width:100%;"><span class="pageNumber"></span> / <span class="totalPages"></span></div>';
+      margin.bottom = '40px';
+    }
+
+    const buffer = await page.pdf(pdfOptions);
+    await page.close();
+
+    const filename = config?.output ?? `${slug}.pdf`;
+    res.statusCode = 200;
+    res.setHeader('Content-Type', 'application/pdf');
+    res.setHeader('Content-Disposition', `attachment; filename="${filename}"`);
+    res.end(buffer);
+  } catch (err) {
+    if (!res.headersSent) {
+      res.statusCode = 500;
+      res.setHeader('Content-Type', 'application/json');
+      res.end(JSON.stringify({ error: err instanceof Error ? err.message : 'PDF export failed' }));
+    }
+  } finally {
+    await browser?.close().catch(() => {});
+  }
+}
+
 export function pdfSmithPreviewPlugin({ pkgSrcDir }: PreviewPluginOptions): Plugin {
   return {
     name: 'pdf-smith-preview',
 
     configureServer(server) {
+      // Pre-middleware: Export API runs before Vite internals.
+      // Must be a sync function — async middleware returns a Promise that
+      // Connect ignores, so the pipeline can advance to the catch-all
+      // before the async work finishes.
+      server.middlewares.use((req, res, next) => {
+        if (req.method !== 'POST') return next();
+        const match = req.url?.match(/^\/api\/export\/([^/]+)$/);
+        if (!match) return next();
+        void handleExportRequest(server, match[1], res);
+      });
+
+      // Post-middleware: catch-all HTML handler (after Vite internals)
       return () => {
         server.middlewares.use((req, res, next) => {
           const url = req.url ?? '';
@@ -36,18 +139,35 @@ window.__vite_plugin_react_preamble_installed__ = true
   <meta charset="UTF-8" />
   <meta name="viewport" content="width=device-width, initial-scale=1.0" />
   <title>pdf-smith Preview</title>
+  <link rel="preconnect" href="https://fonts.googleapis.com">
+  <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
+  <link href="https://fonts.googleapis.com/css2?family=Fraunces:ital,opsz,wght,SOFT,WONK@0,9..144,100..900,0..100,0..1&family=Inter:opsz,wght@14..32,100..900&family=JetBrains+Mono:wght@400;500;600&display=swap" rel="stylesheet">
   <style>
+    *, *::before, *::after { box-sizing: border-box; }
+    body {
+      margin: 0;
+      font-family: 'Inter', system-ui, sans-serif;
+      -webkit-font-smoothing: antialiased;
+    }
+    .pdf-smith-search::placeholder { color: #78716C; }
     @media print {
       [data-pdf-smith-nav] { display: none !important; }
+      [data-pdf-smith-toolbar] { display: none !important; }
+      [data-pdf-smith-minimap] { display: none !important; }
       [data-pdf-smith-container] {
         margin-left: 0 !important;
         padding: 0 !important;
+        padding-top: 0 !important;
         background: none !important;
       }
-      [data-pdf-smith-container] > [data-pdf-smith-document] > div {
+      [data-pdf-smith-container] > div {
+        padding: 0 !important;
+        gap: 0 !important;
+      }
+      [data-pdf-smith-document] > div {
         margin-bottom: 0 !important;
       }
-      [data-pdf-smith-container] > [data-pdf-smith-document] > div > div:first-child {
+      [data-pdf-smith-document] > div > div:first-child {
         display: none !important;
       }
     }

package/dist/preview-plugin-QDLEMAEE.js

@@ -1,104 +0,0 @@
-// src/preview/preview-plugin.ts
-var VIRTUAL_DOCUMENTS_ID = "virtual:pdf-smith-documents";
-var RESOLVED_VIRTUAL_DOCUMENTS_ID = `\0${VIRTUAL_DOCUMENTS_ID}`;
-function pdfSmithPreviewPlugin({ pkgSrcDir }) {
-  return {
-    name: "pdf-smith-preview",
-    configureServer(server) {
-      return () => {
-        server.middlewares.use((req, res, next) => {
-          const url = req.url ?? "";
-          if (url.includes(".") || url.startsWith("/@") || url.startsWith("/node_modules/")) {
-            next();
-            return;
-          }
-          const html = `<!DOCTYPE html>
-<html lang="en">
-<head>
-  <script type="module">
-import RefreshRuntime from "/@react-refresh"
-RefreshRuntime.injectIntoGlobalHook(window)
-window.$RefreshReg$ = () => {}
-window.$RefreshSig$ = () => (type) => type
-window.__vite_plugin_react_preamble_installed__ = true
-</script>
-  <script type="module" src="/@vite/client"></script>
-  <meta charset="UTF-8" />
-  <meta name="viewport" content="width=device-width, initial-scale=1.0" />
-  <title>pdf-smith Preview</title>
-  <style>
-    @media print {
-      [data-pdf-smith-nav] { display: none !important; }
-      [data-pdf-smith-container] {
-        margin-left: 0 !important;
-        padding: 0 !important;
-        background: none !important;
-      }
-      [data-pdf-smith-container] > [data-pdf-smith-document] > div {
-        margin-bottom: 0 !important;
-      }
-      [data-pdf-smith-container] > [data-pdf-smith-document] > div > div:first-child {
-        display: none !important;
-      }
-    }
-  </style>
-</head>
-<body>
-  <div id="root"></div>
-  <script type="module" src="/@fs/${pkgSrcDir}/preview/entry.tsx"></script>
-</body>
-</html>`;
-          res.setHeader("Content-Type", "text/html");
-          res.statusCode = 200;
-          res.end(html);
-        });
-      };
-    },
-    resolveId(id) {
-      if (id === VIRTUAL_DOCUMENTS_ID) {
-        return RESOLVED_VIRTUAL_DOCUMENTS_ID;
-      }
-    },
-    load(id) {
-      if (id === RESOLVED_VIRTUAL_DOCUMENTS_ID) {
-        return `
-const pageModules = import.meta.glob('/pdfs/*/pages/**/*.tsx', { eager: true });
-const configModules = import.meta.glob('/pdfs/*/config.ts', { eager: true });
-
-export function getDocuments() {
-  const documents = {};
-
-  for (const [path, mod] of Object.entries(pageModules)) {
-    const match = path.match(/^\\/pdfs\\/([^/]+)\\/pages\\/(.+)\\.[^.]+$/);
-    if (!match) continue;
-    const [, slug, pageName] = match;
-    if (!documents[slug]) {
-      documents[slug] = { slug, pages: {}, config: undefined };
-    }
-    documents[slug].pages[pageName] = mod.default;
-  }
-
-  for (const [path, mod] of Object.entries(configModules)) {
-    const match = path.match(/^\\/pdfs\\/([^/]+)\\/config\\.ts$/);
-    if (!match) continue;
-    const slug = match[1];
-    if (documents[slug]) {
-      documents[slug].config = mod.default;
-    }
-  }
-
-  return documents;
-}
-
-export function getDocumentSlugs() {
-  return Object.keys(getDocuments());
-}
-`;
-      }
-    }
-  };
-}
-export {
-  pdfSmithPreviewPlugin
-};
-//# sourceMappingURL=preview-plugin-QDLEMAEE.js.map

package/dist/preview-plugin-QDLEMAEE.js.map

@@ -1 +0,0 @@
-{"version":3,"sources":["../src/preview/preview-plugin.ts"],"sourcesContent":["import type { Plugin } from 'vite';\n\nconst VIRTUAL_DOCUMENTS_ID = 'virtual:pdf-smith-documents';\nconst RESOLVED_VIRTUAL_DOCUMENTS_ID = `\\0${VIRTUAL_DOCUMENTS_ID}`;\n\ninterface PreviewPluginOptions {\n  pkgSrcDir: string;\n}\n\nexport function pdfSmithPreviewPlugin({ pkgSrcDir }: PreviewPluginOptions): Plugin {\n  return {\n    name: 'pdf-smith-preview',\n\n    configureServer(server) {\n      return () => {\n        server.middlewares.use((req, res, next) => {\n          const url = req.url ?? '';\n\n          // Skip file requests, Vite internals, and node_modules\n          if (url.includes('.') || url.startsWith('/@') || url.startsWith('/node_modules/')) {\n            next();\n            return;\n          }\n\n          const html = `<!DOCTYPE html>\n<html lang=\"en\">\n<head>\n  <script type=\"module\">\nimport RefreshRuntime from \"/@react-refresh\"\nRefreshRuntime.injectIntoGlobalHook(window)\nwindow.$RefreshReg$ = () => {}\nwindow.$RefreshSig$ = () => (type) => type\nwindow.__vite_plugin_react_preamble_installed__ = true\n</script>\n  <script type=\"module\" src=\"/@vite/client\"></script>\n  <meta charset=\"UTF-8\" />\n  <meta name=\"viewport\" content=\"width=device-width, initial-scale=1.0\" />\n  <title>pdf-smith Preview</title>\n  <style>\n    @media print {\n      [data-pdf-smith-nav] { display: none !important; }\n      [data-pdf-smith-container] {\n        margin-left: 0 !important;\n        padding: 0 !important;\n        background: none !important;\n      }\n      [data-pdf-smith-container] > [data-pdf-smith-document] > div {\n        margin-bottom: 0 !important;\n      }\n      [data-pdf-smith-container] > [data-pdf-smith-document] > div > div:first-child {\n        display: none !important;\n      }\n    }\n  </style>\n</head>\n<body>\n  <div id=\"root\"></div>\n  <script type=\"module\" src=\"/@fs/${pkgSrcDir}/preview/entry.tsx\"></script>\n</body>\n</html>`;\n\n          res.setHeader('Content-Type', 'text/html');\n          res.statusCode = 200;\n          res.end(html);\n        });\n      };\n    },\n\n    resolveId(id) {\n      if (id === VIRTUAL_DOCUMENTS_ID) {\n        return RESOLVED_VIRTUAL_DOCUMENTS_ID;\n      }\n    },\n\n    load(id) {\n      if (id === RESOLVED_VIRTUAL_DOCUMENTS_ID) {\n        return `\nconst pageModules = import.meta.glob('/pdfs/*/pages/**/*.tsx', { eager: true });\nconst configModules = import.meta.glob('/pdfs/*/config.ts', { eager: true });\n\nexport function getDocuments() {\n  const documents = {};\n\n  for (const [path, mod] of Object.entries(pageModules)) {\n    const match = path.match(/^\\\\/pdfs\\\\/([^/]+)\\\\/pages\\\\/(.+)\\\\.[^.]+$/);\n    if (!match) continue;\n    const [, slug, pageName] = match;\n    if (!documents[slug]) {\n      documents[slug] = { slug, pages: {}, config: undefined };\n    }\n    documents[slug].pages[pageName] = mod.default;\n  }\n\n  for (const [path, mod] of Object.entries(configModules)) {\n    const match = path.match(/^\\\\/pdfs\\\\/([^/]+)\\\\/config\\\\.ts$/);\n    if (!match) continue;\n    const slug = match[1];\n    if (documents[slug]) {\n      documents[slug].config = mod.default;\n    }\n  }\n\n  return documents;\n}\n\nexport function getDocumentSlugs() {\n  return Object.keys(getDocuments());\n}\n`;\n      }\n    },\n  };\n}\n"],"mappings":";AAEA,IAAM,uBAAuB;AAC7B,IAAM,gCAAgC,KAAK,oBAAoB;AAMxD,SAAS,sBAAsB,EAAE,UAAU,GAAiC;AACjF,SAAO;AAAA,IACL,MAAM;AAAA,IAEN,gBAAgB,QAAQ;AACtB,aAAO,MAAM;AACX,eAAO,YAAY,IAAI,CAAC,KAAK,KAAK,SAAS;AACzC,gBAAM,MAAM,IAAI,OAAO;AAGvB,cAAI,IAAI,SAAS,GAAG,KAAK,IAAI,WAAW,IAAI,KAAK,IAAI,WAAW,gBAAgB,GAAG;AACjF,iBAAK;AACL;AAAA,UACF;AAEA,gBAAM,OAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,oCAiCa,SAAS;AAAA;AAAA;AAInC,cAAI,UAAU,gBAAgB,WAAW;AACzC,cAAI,aAAa;AACjB,cAAI,IAAI,IAAI;AAAA,QACd,CAAC;AAAA,MACH;AAAA,IACF;AAAA,IAEA,UAAU,IAAI;AACZ,UAAI,OAAO,sBAAsB;AAC/B,eAAO;AAAA,MACT;AAAA,IACF;AAAA,IAEA,KAAK,IAAI;AACP,UAAI,OAAO,+BAA+B;AACxC,eAAO;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA;AAAA,MAiCT;AAAA,IACF;AAAA,EACF;AACF;","names":[]}