squidclaw 2.2.0 → 2.4.0

package/lib/tools/pdf.js

@@ -0,0 +1,133 @@
+/**
+ * 🦑 PDF Generator
+ * Create .pdf documents
+ */
+
+import { logger } from '../core/logger.js';
+import { mkdirSync, createWriteStream } from 'fs';
+import { join } from 'path';
+
+export class PdfGenerator {
+  constructor() {
+    this.outputDir = '/tmp/squidclaw-files';
+    mkdirSync(this.outputDir, { recursive: true });
+  }
+
+  async create(title, content, options = {}) {
+    const PDFDocument = (await import('pdfkit')).default;
+
+    const filename = title.replace(/[^a-zA-Z0-9\u0600-\u06FF ]/g, '').replace(/\s+/g, '_').slice(0, 50) + '.pdf';
+    const filepath = join(this.outputDir, filename);
+
+    return new Promise((resolve, reject) => {
+      const doc = new PDFDocument({
+        size: 'A4',
+        margins: { top: 60, bottom: 60, left: 60, right: 60 },
+        info: { Title: title, Author: 'Squidclaw AI 🦑', Creator: 'Squidclaw' },
+      });
+
+      const stream = createWriteStream(filepath);
+      doc.pipe(stream);
+
+      const colors = {
+        title: '#1a1a2e',
+        heading: '#2563eb',
+        text: '#374151',
+        accent: '#2563eb',
+        light: '#6b7280',
+        bg: '#f8fafc',
+      };
+
+      // Header accent bar
+      doc.rect(0, 0, doc.page.width, 8).fill(colors.accent);
+
+      // Title
+      doc.moveDown(1);
+      doc.fontSize(28).fillColor(colors.title).font('Helvetica-Bold').text(title, { align: 'left' });
+      
+      // Subtitle line
+      doc.moveDown(0.3);
+      doc.moveTo(60, doc.y).lineTo(200, doc.y).strokeColor(colors.accent).lineWidth(2).stroke();
+      doc.moveDown(0.5);
+
+      // Date
+      const date = new Date().toLocaleDateString('en-US', { year: 'numeric', month: 'long', day: 'numeric' });
+      doc.fontSize(10).fillColor(colors.light).font('Helvetica').text(date);
+      doc.moveDown(1.5);
+
+      // Parse and render content
+      const lines = content.split('\n');
+      for (const line of lines) {
+        const trimmed = line.trim();
+        if (!trimmed) { doc.moveDown(0.5); continue; }
+
+        // Check for page overflow
+        if (doc.y > doc.page.height - 100) {
+          doc.addPage();
+          doc.rect(0, 0, doc.page.width, 4).fill(colors.accent);
+          doc.moveDown(1);
+        }
+
+        // ## Heading
+        if (trimmed.startsWith('## ')) {
+          doc.moveDown(0.8);
+          doc.fontSize(18).fillColor(colors.heading).font('Helvetica-Bold').text(trimmed.slice(3));
+          doc.moveDown(0.2);
+          doc.moveTo(60, doc.y).lineTo(160, doc.y).strokeColor(colors.accent).lineWidth(1).stroke();
+          doc.moveDown(0.5);
+        }
+        // ### Subheading
+        else if (trimmed.startsWith('### ')) {
+          doc.moveDown(0.5);
+          doc.fontSize(14).fillColor(colors.title).font('Helvetica-Bold').text(trimmed.slice(4));
+          doc.moveDown(0.3);
+        }
+        // - Bullet
+        else if (trimmed.startsWith('- ') || trimmed.startsWith('• ') || trimmed.startsWith('* ')) {
+          const bullet = trimmed.replace(/^[-•*]\s*/, '');
+          doc.fontSize(11).fillColor(colors.text).font('Helvetica');
+          doc.text('●  ' + bullet, { indent: 15, lineGap: 4 });
+        }
+        // **Bold**
+        else if (trimmed.startsWith('**') && trimmed.endsWith('**')) {
+          doc.fontSize(12).fillColor(colors.title).font('Helvetica-Bold').text(trimmed.replace(/\*\*/g, ''));
+        }
+        // > Quote
+        else if (trimmed.startsWith('> ')) {
+          doc.moveDown(0.3);
+          const quoteText = trimmed.slice(2);
+          const qx = doc.x;
+          doc.rect(qx, doc.y, 3, 40).fill(colors.accent);
+          doc.fontSize(11).fillColor(colors.light).font('Helvetica-Oblique').text(quoteText, qx + 15, doc.y - 40 + 5, { width: 420 });
+          doc.moveDown(0.5);
+        }
+        // --- Divider
+        else if (trimmed === '---' || trimmed === '***') {
+          doc.moveDown(0.5);
+          doc.moveTo(60, doc.y).lineTo(doc.page.width - 60, doc.y).strokeColor('#e5e7eb').lineWidth(0.5).stroke();
+          doc.moveDown(0.5);
+        }
+        // Normal text
+        else {
+          doc.fontSize(11).fillColor(colors.text).font('Helvetica').text(trimmed, { lineGap: 4 });
+        }
+      }
+
+      // Footer
+      const pages = doc.bufferedPageRange();
+      for (let i = 0; i < pages.count; i++) {
+        doc.switchToPage(i);
+        doc.fontSize(8).fillColor(colors.light).font('Helvetica');
+        doc.text('Created with Squidclaw AI 🦑', 60, doc.page.height - 40, { width: doc.page.width - 120, align: 'center' });
+        doc.text(String(i + 1), 60, doc.page.height - 40, { width: doc.page.width - 120, align: 'right' });
+      }
+
+      doc.end();
+      stream.on('finish', () => {
+        logger.info('pdf', `Created: ${filepath}`);
+        resolve({ filepath, filename });
+      });
+      stream.on('error', reject);
+    });
+  }
+}