Update server.js

server.js

@@ -227,6 +227,150 @@ app.post('/api/parse-character-png', upload.single('file'), async (req, res) =>
   }
 });
 
+// PDF生成API
+app.post('/api/generate-pdf', async (req, res) => {
+  try {
+    const { title, chapters, style } = req.body;
+
+    if (!title || !chapters) {
+      return res.status(400).json({ error: '缺少必要参数' });
+    }
+
+    // 导入必要的库
+    const { default: PDFDocument } = await import('pdfkit');
+    const { default: fetch } = await import('node-fetch');
+
+    // 创建PDF文档
+    const doc = new PDFDocument({
+      size: 'A4',
+      margins: { top: 50, bottom: 50, left: 50, right: 50 }
+    });
+
+    // 设置响应头
+    res.setHeader('Content-Type', 'application/pdf');
+    res.setHeader('Content-Disposition', `attachment; filename="${encodeURIComponent(title)}.pdf"`);
+
+    // 管道输出到响应
+    doc.pipe(res);
+
+    // 样式配置
+    const styles = {
+      modern: {
+        titleFont: 'Helvetica-Bold',
+        titleSize: 32,
+        headingFont: 'Helvetica-Bold',
+        headingSize: 20,
+        bodyFont: 'Helvetica',
+        bodySize: 12,
+        lineHeight: 1.5
+      },
+      classic: {
+        titleFont: 'Times-Bold',
+        titleSize: 36,
+        headingFont: 'Times-Bold',
+        headingSize: 22,
+        bodyFont: 'Times-Roman',
+        bodySize: 13,
+        lineHeight: 1.6
+      },
+      magazine: {
+        titleFont: 'Helvetica-Bold',
+        titleSize: 40,
+        headingFont: 'Helvetica-Bold',
+        headingSize: 24,
+        bodyFont: 'Helvetica',
+        bodySize: 11,
+        lineHeight: 1.4
+      },
+      technical: {
+        titleFont: 'Courier-Bold',
+        titleSize: 28,
+        headingFont: 'Courier-Bold',
+        headingSize: 18,
+        bodyFont: 'Courier',
+        bodySize: 10,
+        lineHeight: 1.5
+      }
+    };
+
+    const currentStyle = styles[style] || styles.modern;
+
+    // 封面
+    doc.font(currentStyle.titleFont)
+       .fontSize(currentStyle.titleSize)
+       .text(title, 50, 300, { align: 'center' });
+
+    doc.fontSize(14)
+       .font('Helvetica')
+       .text(`生成于 ${new Date().toLocaleDateString('zh-CN')}`, 50, 400, { align: 'center' });
+
+    doc.addPage();
+
+    // 生成章节
+    for (let i = 0; i < chapters.length; i++) {
+      const chapter = chapters[i];
+
+      // 章节标题
+      if (i > 0) {
+        doc.addPage();
+      }
+
+      doc.font(currentStyle.headingFont)
+         .fontSize(currentStyle.headingSize)
+         .text(chapter.title, { align: 'left' });
+
+      doc.moveDown(1);
+
+      // 章节内容
+      doc.font(currentStyle.bodyFont)
+         .fontSize(currentStyle.bodySize);
+
+      // 移除IMAGE标记
+      const cleanContent = chapter.content.replace(/\[IMAGE:.*?\]/g, '');
+      const paragraphs = cleanContent.split('\n').filter(p => p.trim());
+
+      for (const paragraph of paragraphs) {
+        doc.text(paragraph, {
+          align: 'justify',
+          lineGap: currentStyle.lineHeight * 2
+        });
+        doc.moveDown(0.5);
+      }
+
+      // 插入图片
+      if (chapter.images && chapter.images.length > 0) {
+        for (const imageUrl of chapter.images) {
+          try {
+            // 下载图片
+            const imageResponse = await fetch(imageUrl);
+            const imageBuffer = await imageResponse.buffer();
+
+            doc.moveDown(1);
+
+            // 添加图片（适配页面宽度）
+            const pageWidth = doc.page.width - 100;
+            doc.image(imageBuffer, 50, doc.y, {
+              fit: [pageWidth, 300],
+              align: 'center'
+            });
+
+            doc.moveDown(1);
+          } catch (error) {
+            console.error('添加图片失败:', error);
+          }
+        }
+      }
+    }
+
+    // 完成PDF
+    doc.end();
+
+  } catch (error) {
+    console.error('PDF生成错误:', error);
+    res.status(500).json({ error: error.message });
+  }
+});
+
 const PORT = process.env.PORT || 7860;
 
 app.listen(PORT, () => {