Create pdf-server.js

pdf-server.js

@@ -0,0 +1,231 @@
+const express = require('express');
+const { chromium } = require('playwright-core');
+
+const app = express();
+app.use(express.json({ limit: '50mb' }));
+
+// ==================== CONFIGURATION ====================
+const PORT = 3000;
+const CHROMIUM_PATH = '/usr/bin/chromium-browser';
+
+// Formats KDP
+const FORMATS = {
+  '6x9': { width: '6in', height: '9in' },
+  '5x8': { width: '5in', height: '8in' },
+  '5.5x8.5': { width: '5.5in', height: '8.5in' },
+  '8.5x11': { width: '8.5in', height: '11in' },
+  'A4': { width: '210mm', height: '297mm' },
+  'A5': { width: '148mm', height: '210mm' },
+};
+
+// ==================== HEALTH CHECK ====================
+app.get('/health', (req, res) => {
+  res.json({ status: 'ok', service: 'pdf-server' });
+});
+
+// ==================== GÉNÉRATION PDF ====================
+app.post('/generate-pdf', async (req, res) => {
+  const startTime = Date.now();
+  let browser = null;
+
+  try {
+    const {
+      html,
+      format = '6x9',
+      margins = { top: '0.75in', bottom: '0.75in', left: '0.5in', right: '0.5in' },
+      printBackground = true,
+      waitForImages = true,
+      waitTimeout = 30000,
+    } = req.body;
+
+    if (!html) {
+      return res.status(400).json({ error: 'html requis' });
+    }
+
+    console.log(`📄 Génération PDF (format: ${format})...`);
+
+    // Lancer le navigateur
+    browser = await chromium.launch({
+      executablePath: CHROMIUM_PATH,
+      headless: true,
+      args: [
+        '--no-sandbox',
+        '--disable-setuid-sandbox',
+        '--disable-dev-shm-usage',
+        '--disable-gpu',
+        '--single-process',
+      ],
+    });
+
+    const context = await browser.newContext();
+    const page = await context.newPage();
+
+    // Charger le HTML
+    await page.setContent(html, {
+      waitUntil: 'networkidle',
+      timeout: waitTimeout,
+    });
+
+    // Attendre que les images soient chargées
+    if (waitForImages) {
+      await page.evaluate(() => {
+        return Promise.all(
+          Array.from(document.images)
+            .filter(img => !img.complete)
+            .map(img => new Promise(resolve => {
+              img.onload = img.onerror = resolve;
+              setTimeout(resolve, 5000);
+            }))
+        );
+      });
+    }
+
+    // Attendre un peu pour le rendu CSS
+    await page.waitForTimeout(500);
+
+    // Configuration du format
+    const formatConfig = FORMATS[format] || FORMATS['6x9'];
+
+    // Générer le PDF
+    const pdfBuffer = await page.pdf({
+      width: formatConfig.width,
+      height: formatConfig.height,
+      margin: margins,
+      printBackground: printBackground,
+      preferCSSPageSize: true,
+    });
+
+    await browser.close();
+    browser = null;
+
+    const duration = Date.now() - startTime;
+    console.log(`✅ PDF généré (${(pdfBuffer.length / 1024 / 1024).toFixed(2)} MB, ${duration}ms)`);
+
+    // Retourner en base64
+    res.json({
+      success: true,
+      pdf: pdfBuffer.toString('base64'),
+      size_bytes: pdfBuffer.length,
+      size_mb: (pdfBuffer.length / 1024 / 1024).toFixed(2),
+      duration_ms: duration,
+    });
+
+  } catch (error) {
+    console.error('❌ Erreur:', error.message);
+    if (browser) await browser.close();
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// ==================== SCREENSHOT ====================
+app.post('/screenshot', async (req, res) => {
+  let browser = null;
+
+  try {
+    const {
+      url,
+      html,
+      fullPage = true,
+      width = 1920,
+      height = 1080,
+    } = req.body;
+
+    if (!url && !html) {
+      return res.status(400).json({ error: 'url ou html requis' });
+    }
+
+    browser = await chromium.launch({
+      executablePath: CHROMIUM_PATH,
+      headless: true,
+      args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
+    });
+
+    const page = await browser.newPage();
+    await page.setViewportSize({ width, height });
+
+    if (url) {
+      await page.goto(url, { waitUntil: 'networkidle', timeout: 30000 });
+    } else {
+      await page.setContent(html, { waitUntil: 'networkidle' });
+    }
+
+    const screenshot = await page.screenshot({ fullPage, type: 'png' });
+    await browser.close();
+
+    res.json({
+      success: true,
+      screenshot: screenshot.toString('base64'),
+      size_bytes: screenshot.length,
+    });
+
+  } catch (error) {
+    if (browser) await browser.close();
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// ==================== SCRAPING ====================
+app.post('/scrape', async (req, res) => {
+  let browser = null;
+
+  try {
+    const {
+      url,
+      selector,
+      waitFor,
+      evaluate,
+    } = req.body;
+
+    if (!url) {
+      return res.status(400).json({ error: 'url requis' });
+    }
+
+    browser = await chromium.launch({
+      executablePath: CHROMIUM_PATH,
+      headless: true,
+      args: ['--no-sandbox', '--disable-setuid-sandbox', '--disable-dev-shm-usage'],
+    });
+
+    const page = await browser.newPage();
+    await page.goto(url, { waitUntil: 'networkidle', timeout: 30000 });
+
+    if (waitFor) {
+      await page.waitForSelector(waitFor, { timeout: 10000 });
+    }
+
+    let result = {};
+
+    // Récupérer le HTML
+    result.html = await page.content();
+    result.title = await page.title();
+    result.url = page.url();
+
+    // Si un sélecteur est spécifié
+    if (selector) {
+      result.elements = await page.$$eval(selector, els => els.map(el => ({
+        text: el.textContent?.trim(),
+        html: el.innerHTML,
+        href: el.href || null,
+        src: el.src || null,
+      })));
+    }
+
+    // Si un script personnalisé est fourni
+    if (evaluate) {
+      result.custom = await page.evaluate(evaluate);
+    }
+
+    await browser.close();
+
+    res.json({ success: true, ...result });
+
+  } catch (error) {
+    if (browser) await browser.close();
+    res.status(500).json({ error: error.message });
+  }
+});
+
+// ==================== DÉMARRAGE ====================
+app.listen(PORT, '0.0.0.0', () => {
+  console.log(`🚀 PDF Server running on port ${PORT}`);
+});