@pyreon/document 0.9.0 → 0.11.0

package/src/renderers/pdf.ts

@@ -1,10 +1,4 @@
-import type {
-  DocChild,
-  DocNode,
-  DocumentRenderer,
-  RenderOptions,
-  TableColumn,
-} from '../types'
+import type { DocChild, DocNode, DocumentRenderer, RenderOptions, TableColumn } from "../types"
 
 /**
  * PDF renderer — lazy-loads pdfmake on first use.
@@ -23,15 +17,13 @@ import type {
  */
 
 function resolveColumn(col: string | TableColumn): TableColumn {
-  return typeof col === 'string' ? { header: col } : col
+  return typeof col === "string" ? { header: col } : col
 }
 
 function getTextContent(children: DocChild[]): string {
   return children
-    .map((c) =>
-      typeof c === 'string' ? c : getTextContent((c as DocNode).children),
-    )
-    .join('')
+    .map((c) => (typeof c === "string" ? c : getTextContent((c as DocNode).children)))
+    .join("")
 }
 
 type PdfContent = Record<string, unknown> | string | PdfContent[]
@@ -58,42 +50,39 @@ const PAGE_SIZES: Record<string, { width: number; height: number }> = {
 function resolveImageSrc(
   src: string,
 ): { image: string } | { text: string; italics: true; color: string } {
-  if (src.startsWith('data:')) {
+  if (src.startsWith("data:")) {
     return { image: src }
   }
-  if (src.startsWith('http://') || src.startsWith('https://')) {
-    return { text: `[Image: ${src}]`, italics: true, color: '#999999' }
+  if (src.startsWith("http://") || src.startsWith("https://")) {
+    return { text: `[Image: ${src}]`, italics: true, color: "#999999" }
   }
   // Local path — cannot resolve in browser
-  return { text: `[Image: ${src}]`, italics: true, color: '#999999' }
+  return { text: `[Image: ${src}]`, italics: true, color: "#999999" }
 }
 
 function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
   const p = node.props
 
   switch (node.type) {
-    case 'document':
-    case 'page':
+    case "document":
+    case "page":
       return node.children
-        .map((c) => (typeof c === 'string' ? c : nodeToContent(c)))
+        .map((c) => (typeof c === "string" ? c : nodeToContent(c)))
         .filter((c): c is PdfContent => c != null)
 
-    case 'section': {
+    case "section": {
       const content = node.children
-        .map((c) => (typeof c === 'string' ? c : nodeToContent(c)))
+        .map((c) => (typeof c === "string" ? c : nodeToContent(c)))
         .filter((c): c is PdfContent => c != null)
         .flat()
 
-      if (p.direction === 'row') {
+      if (p.direction === "row") {
         return {
           columns: node.children
-            .filter((c): c is DocNode => typeof c !== 'string')
+            .filter((c): c is DocNode => typeof c !== "string")
             .map((child) => ({
               stack: [nodeToContent(child)].flat().filter(Boolean),
-              width:
-                child.props.width === '*' || !child.props.width
-                  ? '*'
-                  : child.props.width,
+              width: child.props.width === "*" || !child.props.width ? "*" : child.props.width,
             })),
           columnGap: (p.gap as number) ?? 0,
         }
@@ -102,25 +91,25 @@ function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
       return content
     }
 
-    case 'row': {
+    case "row": {
       return {
         columns: node.children
-          .filter((c): c is DocNode => typeof c !== 'string')
+          .filter((c): c is DocNode => typeof c !== "string")
           .map((child) => ({
             stack: [nodeToContent(child)].flat().filter(Boolean),
-            width: child.props.width ?? '*',
+            width: child.props.width ?? "*",
           })),
         columnGap: (p.gap as number) ?? 0,
       }
     }
 
-    case 'column':
+    case "column":
       return node.children
-        .map((c) => (typeof c === 'string' ? c : nodeToContent(c)))
+        .map((c) => (typeof c === "string" ? c : nodeToContent(c)))
         .filter((c): c is PdfContent => c != null)
         .flat()
 
-    case 'heading': {
+    case "heading": {
       const level = (p.level as number) ?? 1
       const sizes: Record<number, number> = {
         1: 24,
@@ -134,49 +123,45 @@ function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
         text: getTextContent(node.children),
         fontSize: sizes[level] ?? 18,
         bold: true,
-        color: (p.color as string) ?? '#000000',
-        alignment: (p.align as string) ?? 'left',
+        color: (p.color as string) ?? "#000000",
+        alignment: (p.align as string) ?? "left",
         margin: [0, level === 1 ? 0 : 8, 0, 8],
       }
     }
 
-    case 'text':
+    case "text":
       return {
         text: getTextContent(node.children),
         fontSize: (p.size as number) ?? 12,
-        color: (p.color as string) ?? '#333333',
+        color: (p.color as string) ?? "#333333",
         bold: p.bold ?? false,
         italics: p.italic ?? false,
-        decoration: p.underline
-          ? 'underline'
-          : p.strikethrough
-            ? 'lineThrough'
-            : undefined,
-        alignment: (p.align as string) ?? 'left',
+        decoration: p.underline ? "underline" : p.strikethrough ? "lineThrough" : undefined,
+        alignment: (p.align as string) ?? "left",
         lineHeight: (p.lineHeight as number) ?? 1.4,
         margin: [0, 0, 0, 8],
       }
 
-    case 'link':
+    case "link":
       return {
         text: getTextContent(node.children),
         link: p.href as string,
-        color: (p.color as string) ?? '#4f46e5',
-        decoration: 'underline',
+        color: (p.color as string) ?? "#4f46e5",
+        decoration: "underline",
       }
 
-    case 'image': {
+    case "image": {
       const src = p.src as string
       const resolved = resolveImageSrc(src)
 
-      if ('image' in resolved) {
+      if ("image" in resolved) {
         const result: Record<string, unknown> = {
           image: resolved.image,
           fit: [p.width ?? 500, p.height ?? 400],
           margin: [0, 0, 0, 8],
         }
-        if (p.align === 'center') result.alignment = 'center'
-        if (p.align === 'right') result.alignment = 'right'
+        if (p.align === "center") result.alignment = "center"
+        if (p.align === "right") result.alignment = "right"
         return result
       }
 
@@ -184,34 +169,30 @@ function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
       return { ...resolved, margin: [0, 0, 0, 8] }
     }
 
-    case 'table': {
-      const columns = ((p.columns ?? []) as (string | TableColumn)[]).map(
-        resolveColumn,
-      )
+    case "table": {
+      const columns = ((p.columns ?? []) as (string | TableColumn)[]).map(resolveColumn)
       const rows = (p.rows ?? []) as (string | number)[][]
-      const hs = p.headerStyle as
-        | { background?: string; color?: string }
-        | undefined
+      const hs = p.headerStyle as { background?: string; color?: string } | undefined
 
       const headerRow = columns.map((col) => ({
         text: col.header,
         bold: true,
-        fillColor: hs?.background ?? '#f5f5f5',
-        color: hs?.color ?? '#000000',
-        alignment: col.align ?? 'left',
+        fillColor: hs?.background ?? "#f5f5f5",
+        color: hs?.color ?? "#000000",
+        alignment: col.align ?? "left",
       }))
 
       const dataRows = rows.map((row, rowIdx) =>
         columns.map((col, colIdx) => ({
-          text: String(row[colIdx] ?? ''),
-          alignment: col.align ?? 'left',
-          fillColor: p.striped && rowIdx % 2 === 1 ? '#f9f9f9' : undefined,
+          text: String(row[colIdx] ?? ""),
+          alignment: col.align ?? "left",
+          fillColor: p.striped && rowIdx % 2 === 1 ? "#f9f9f9" : undefined,
         })),
       )
 
       const widths = columns.map((col) => {
-        if (!col.width) return '*'
-        if (typeof col.width === 'string' && col.width.endsWith('%')) {
+        if (!col.width) return "*"
+        if (typeof col.width === "string" && col.width.endsWith("%")) {
           return col.width
         }
         return col.width
@@ -223,83 +204,81 @@ function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
           widths,
           body: [headerRow, ...dataRows],
         },
-        layout: p.bordered ? undefined : 'lightHorizontalLines',
+        layout: p.bordered ? undefined : "lightHorizontalLines",
         unbreakable: p.keepTogether ?? false,
         margin: [0, 0, 0, 12],
       }
     }
 
-    case 'list': {
+    case "list": {
       const items = node.children
-        .filter((c): c is DocNode => typeof c !== 'string')
+        .filter((c): c is DocNode => typeof c !== "string")
         .map((item) => getTextContent(item.children))
 
-      return p.ordered
-        ? { ol: items, margin: [0, 0, 0, 8] }
-        : { ul: items, margin: [0, 0, 0, 8] }
+      return p.ordered ? { ol: items, margin: [0, 0, 0, 8] } : { ul: items, margin: [0, 0, 0, 8] }
     }
 
-    case 'list-item':
+    case "list-item":
       return getTextContent(node.children)
 
-    case 'code':
+    case "code":
       return {
         text: getTextContent(node.children),
-        font: 'Courier',
+        font: "Courier",
         fontSize: 10,
-        background: '#f5f5f5',
+        background: "#f5f5f5",
         margin: [0, 0, 0, 8],
       }
 
-    case 'page-break':
-      return { text: '', pageBreak: 'after' }
+    case "page-break":
+      return { text: "", pageBreak: "after" }
 
-    case 'divider':
+    case "divider":
       return {
         canvas: [
           {
-            type: 'line',
+            type: "line",
             x1: 0,
             y1: 0,
             x2: 515,
             y2: 0,
             lineWidth: (p.thickness as number) ?? 1,
-            lineColor: (p.color as string) ?? '#dddddd',
+            lineColor: (p.color as string) ?? "#dddddd",
           },
         ],
         margin: [0, 8, 0, 8],
       }
 
-    case 'spacer':
-      return { text: '', margin: [0, (p.height as number) ?? 12, 0, 0] }
+    case "spacer":
+      return { text: "", margin: [0, (p.height as number) ?? 12, 0, 0] }
 
-    case 'button':
+    case "button":
       return {
         text: getTextContent(node.children),
         link: p.href as string,
         bold: true,
-        color: (p.color as string) ?? '#ffffff',
-        background: (p.background as string) ?? '#4f46e5',
+        color: (p.color as string) ?? "#ffffff",
+        background: (p.background as string) ?? "#4f46e5",
         margin: [0, 8, 0, 8],
       }
 
-    case 'quote':
+    case "quote":
       return {
         table: {
-          widths: [4, '*'],
+          widths: [4, "*"],
           body: [
             [
-              { text: '', fillColor: (p.borderColor as string) ?? '#dddddd' },
+              { text: "", fillColor: (p.borderColor as string) ?? "#dddddd" },
               {
                 text: getTextContent(node.children),
                 italics: true,
-                color: '#555555',
+                color: "#555555",
                 margin: [8, 4, 0, 4],
               },
             ],
           ],
         },
-        layout: 'noBorders',
+        layout: "noBorders",
         margin: [0, 4, 0, 8],
       }
 
@@ -309,14 +288,10 @@ function nodeToContent(node: DocNode): PdfContent | PdfContent[] | null {
 }
 
 function resolveMargin(
-  margin:
-    | number
-    | [number, number]
-    | [number, number, number, number]
-    | undefined,
+  margin: number | [number, number] | [number, number, number, number] | undefined,
 ): [number, number, number, number] {
   if (margin == null) return [40, 40, 40, 40]
-  if (typeof margin === 'number') return [margin, margin, margin, margin]
+  if (typeof margin === "number") return [margin, margin, margin, margin]
   if (margin.length === 2) return [margin[1], margin[0], margin[1], margin[0]]
   return margin
 }
@@ -332,23 +307,30 @@ function renderHeaderFooter(node: DocNode | undefined): PdfContent | undefined {
   const content = nodeToContent(node)
   if (content == null) return undefined
   if (Array.isArray(content)) return { stack: content, margin: [40, 10, 40, 0] }
-  if (typeof content === 'object')
-    return { ...content, margin: [40, 10, 40, 0] }
+  if (typeof content === "object") return { ...content, margin: [40, 10, 40, 0] }
   return { text: content, margin: [40, 10, 40, 0] }
 }
 
 export const pdfRenderer: DocumentRenderer = {
   async render(node: DocNode, _options?: RenderOptions): Promise<Uint8Array> {
     // Lazy-load pdfmake — handle ESM/CJS interop
-    const pdfMakeModule = await import('pdfmake/build/pdfmake')
-    const pdfFontsModule = await import('pdfmake/build/vfs_fonts')
+    let pdfMakeModule: any
+    let pdfFontsModule: any
+    try {
+      pdfMakeModule = await import("pdfmake/build/pdfmake")
+      pdfFontsModule = await import("pdfmake/build/vfs_fonts")
+    } catch {
+      throw new Error(
+        '[@pyreon/document] PDF renderer requires "pdfmake" package. Install it: bun add pdfmake',
+      )
+    }
 
     // Resolve the actual exports (handle .default for ESM wrappers).
     // pdfmake's default export is a singleton instance of browser_extensions_pdfmake.
     // ESM interop may wrap it in an extra .default layer.
     // eslint-disable-next-line @typescript-eslint/no-explicit-any -- pdfmake types are incomplete
     let pdfMake: any = pdfMakeModule.default ?? pdfMakeModule
-    if (pdfMake.default && typeof pdfMake.default.createPdf === 'function') {
+    if (pdfMake.default && typeof pdfMake.default.createPdf === "function") {
       pdfMake = pdfMake.default
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any -- pdfmake types are incomplete
@@ -361,11 +343,10 @@ export const pdfRenderer: DocumentRenderer = {
 
     // Find page config
     const pageNode = node.children.find(
-      (c): c is DocNode => typeof c !== 'string' && c.type === 'page',
+      (c): c is DocNode => typeof c !== "string" && c.type === "page",
     )
-    const pageSize = (pageNode?.props.size as string) ?? 'A4'
-    const pageOrientation =
-      (pageNode?.props.orientation as string) ?? 'portrait'
+    const pageSize = (pageNode?.props.size as string) ?? "A4"
+    const pageOrientation = (pageNode?.props.orientation as string) ?? "portrait"
     const pageMargin = resolveMargin(
       pageNode?.props.margin as
         | number
@@ -377,22 +358,18 @@ export const pdfRenderer: DocumentRenderer = {
     const content = [nodeToContent(node)].flat().filter(Boolean) as PdfContent[]
 
     // Build header/footer from PageProps if present
-    const headerFn = renderHeaderFooter(
-      pageNode?.props.header as DocNode | undefined,
-    )
-    const footerFn = renderHeaderFooter(
-      pageNode?.props.footer as DocNode | undefined,
-    )
+    const headerFn = renderHeaderFooter(pageNode?.props.header as DocNode | undefined)
+    const footerFn = renderHeaderFooter(pageNode?.props.footer as DocNode | undefined)
 
     const docDefinition: Record<string, unknown> = {
       pageSize: PAGE_SIZES[pageSize] ?? PAGE_SIZES.A4,
       pageOrientation,
       pageMargins: pageMargin,
       info: {
-        title: (node.props.title as string) ?? '',
-        author: (node.props.author as string) ?? '',
-        subject: (node.props.subject as string) ?? '',
-        keywords: (node.props.keywords as string[])?.join(', ') ?? '',
+        title: (node.props.title as string) ?? "",
+        author: (node.props.author as string) ?? "",
+        subject: (node.props.subject as string) ?? "",
+        keywords: (node.props.keywords as string[])?.join(", ") ?? "",
       },
       content,
       defaultStyle: {