@mcpcn/markdown-converter 1.0.10

package/dist/core/PDFConverter.js

@@ -0,0 +1,284 @@
+/**
+ * PDF转换器
+ * 基于原EnhancedConverter重构，继承BaseConverter
+ */
+import pLimit from 'p-limit';
+import { BaseConverter } from './BaseConverter.js';
+import { PuppeteerBrowserPool } from './BrowserPool.js';
+import { TemplateManager } from './TemplateManager.js';
+import { MarkdownRenderer } from './MarkdownRenderer.js';
+import { PDFGenerator } from './PDFGenerator.js';
+import { MemoryCacheManager } from './CacheManager.js';
+import path from 'path';
+import { writeFile, readFile } from 'fs/promises';
+import { createRequire } from 'module';
+const require = createRequire(import.meta.url);
+export class PDFConverter extends BaseConverter {
+    browserPool;
+    templateManager;
+    markdownRenderer;
+    pdfGenerator;
+    memoryCacheManager;
+    concurrencyLimit;
+    constructor(options = {}) {
+        super({
+            cacheEnabled: options.cacheEnabled,
+            cacheDir: options.cacheDir
+        });
+        const { maxBrowsers = 3, maxConcurrent = 2, templateDir, cacheEnabled = true } = options;
+        // 初始化核心组件
+        this.browserPool = new PuppeteerBrowserPool(maxBrowsers);
+        this.templateManager = new TemplateManager(templateDir);
+        this.markdownRenderer = new MarkdownRenderer();
+        this.pdfGenerator = new PDFGenerator(this.browserPool, this.templateManager);
+        this.memoryCacheManager = cacheEnabled ? new MemoryCacheManager() : null;
+        this.concurrencyLimit = pLimit(maxConcurrent);
+        // 优雅关闭处理
+        process.on('SIGINT', () => this.cleanup());
+        process.on('SIGTERM', () => this.cleanup());
+    }
+    /**
+     * 实现基类的抽象方法
+     */
+    async convert(markdownContent, outputPath, options = {}) {
+        const startTime = Date.now();
+        try {
+            // 解析内容并获取统计信息
+            const { content, stats: parseStats } = this.parseMarkdown(markdownContent);
+            this.reportProgress({
+                stage: 'parsing',
+                progress: 10,
+                message: '开始转换PDF...'
+            }, options.progress);
+            // 检查本地文件系统缓存
+            const cacheKey = this.generateCacheKey(content, options);
+            let cachedResult = null;
+            if (this.cacheEnabled) {
+                cachedResult = await this.loadFromCache(cacheKey, 'pdf');
+                if (cachedResult) {
+                    await this.ensureOutputDir(outputPath);
+                    await writeFile(outputPath, cachedResult);
+                    const fileSize = await this.getFileSize(outputPath);
+                    this.reportProgress({
+                        stage: 'complete',
+                        progress: 100,
+                        message: '使用缓存结果'
+                    }, options.progress);
+                    return {
+                        success: true,
+                        filePath: outputPath,
+                        fileSize,
+                        stats: {
+                            processingTime: Date.now() - startTime,
+                            headings: parseStats.headings || 0,
+                            codeBlocks: parseStats.codeBlocks || 0,
+                            paragraphs: parseStats.paragraphs || 0,
+                            tables: parseStats.tables || 0,
+                            fileSize
+                        }
+                    };
+                }
+            }
+            // 内存缓存HTML处理
+            const memoryCacheKey = this.memoryCacheManager
+                ? MemoryCacheManager.generateKey('html', {
+                    markdown: content,
+                    template: options.template,
+                    markdownConfig: options.markdownConfig,
+                    watermark: options.watermark
+                })
+                : null;
+            let html;
+            if (memoryCacheKey && this.memoryCacheManager) {
+                const cached = await this.memoryCacheManager.get(memoryCacheKey);
+                if (cached) {
+                    this.reportProgress({
+                        stage: 'parsing',
+                        progress: 50,
+                        message: '从内存缓存加载HTML'
+                    }, options.progress);
+                    html = cached;
+                }
+                else {
+                    // 渲染HTML并缓存到内存
+                    this.markdownRenderer.updateConfig(options.markdownConfig || {});
+                    html = await this.markdownRenderer.render(content, options.progress);
+                    await this.memoryCacheManager.set(memoryCacheKey, html, 600000); // 10分钟TTL
+                }
+            }
+            else {
+                // 直接渲染HTML
+                this.markdownRenderer.updateConfig(options.markdownConfig || {});
+                html = await this.markdownRenderer.render(content, options.progress);
+            }
+            this.reportProgress({
+                stage: 'converting',
+                progress: 70,
+                message: '生成PDF文件'
+            }, options.progress);
+            // 生成PDF
+            const result = await this.pdfGenerator.generatePDF(html, outputPath, {
+                template: options.template || 'default',
+                pdfConfig: options.pdfConfig,
+                watermark: options.watermark,
+                progressCallback: options.progress
+            });
+            // 缓存PDF结果
+            if (this.cacheEnabled) {
+                const pdfBuffer = await readFile(result);
+                await this.saveToCache(cacheKey, 'pdf', pdfBuffer);
+            }
+            const fileSize = await this.getFileSize(result);
+            const processingTime = Date.now() - startTime;
+            this.reportProgress({
+                stage: 'complete',
+                progress: 100,
+                message: `PDF转换完成: ${this.formatFileSize(fileSize)}`
+            }, options.progress);
+            return {
+                success: true,
+                filePath: result,
+                fileSize,
+                stats: {
+                    processingTime,
+                    headings: parseStats.headings || 0,
+                    codeBlocks: parseStats.codeBlocks || 0,
+                    paragraphs: parseStats.paragraphs || 0,
+                    tables: parseStats.tables || 0,
+                    fileSize
+                }
+            };
+        }
+        catch (error) {
+            console.error('PDF转换失败:', error);
+            this.reportProgress({
+                stage: 'parsing',
+                progress: 0,
+                message: '转换失败',
+                error: error
+            }, options.progress);
+            return {
+                success: false,
+                error: error instanceof Error ? error.message : String(error)
+            };
+        }
+    }
+    /**
+     * 批量转换
+     */
+    async convertBatch(inputs, globalProgress) {
+        const results = [];
+        const errors = [];
+        let completed = 0;
+        const tasks = inputs.map((input, index) => this.concurrencyLimit(async () => {
+            try {
+                const progressCallback = input.options?.progress || ((progress) => {
+                    globalProgress?.({
+                        completed,
+                        total: inputs.length,
+                        current: `${index + 1}/${inputs.length}: ${progress.message}`,
+                        errors
+                    });
+                });
+                const result = await this.convert(input.markdown, input.outputPath, {
+                    ...input.options,
+                    progress: progressCallback
+                });
+                if (result.success && result.filePath) {
+                    results[index] = result.filePath;
+                    completed++;
+                    globalProgress?.({
+                        completed,
+                        total: inputs.length,
+                        current: `完成: ${path.basename(result.filePath)}`,
+                        errors
+                    });
+                    return result.filePath;
+                }
+                else {
+                    throw new Error(result.error || '转换失败');
+                }
+            }
+            catch (error) {
+                errors.push(error);
+                completed++;
+                globalProgress?.({
+                    completed,
+                    total: inputs.length,
+                    current: `失败: ${input.outputPath}`,
+                    errors
+                });
+                throw error;
+            }
+        }));
+        // 等待所有任务完成
+        const settledResults = await Promise.allSettled(tasks);
+        // 收集成功的结果
+        const successfulResults = [];
+        settledResults.forEach((result, index) => {
+            if (result.status === 'fulfilled' && result.value) {
+                successfulResults.push(result.value);
+            }
+        });
+        if (errors.length > 0 && successfulResults.length === 0) {
+            throw new Error(`批量转换失败: ${errors.map(e => e.message).join(', ')}`);
+        }
+        return successfulResults;
+    }
+    /**
+     * 从文件转换
+     */
+    async convertFromFile(markdownPath, outputPath, options = {}) {
+        // 验证输入文件
+        await this.validateInputFile(markdownPath);
+        const fs = await import('fs/promises');
+        const markdown = await fs.readFile(markdownPath, 'utf-8');
+        const finalOutputPath = outputPath ||
+            markdownPath.replace(/\.md$/i, '.pdf');
+        return this.convert(markdown, finalOutputPath, options);
+    }
+    /**
+     * 获取系统状态
+     */
+    getStats() {
+        const memUsage = process.memoryUsage();
+        return {
+            browserPool: this.browserPool.getStats(),
+            cache: this.memoryCacheManager?.getStats(),
+            system: {
+                memory: {
+                    used: memUsage.heapUsed,
+                    total: memUsage.heapTotal
+                },
+                cpu: process.cpuUsage().user / 1000000 // 转换为秒
+            }
+        };
+    }
+    /**
+     * 清理资源 - 重写基类方法
+     */
+    async cleanup() {
+        try {
+            await this.browserPool.cleanup();
+            if (this.memoryCacheManager) {
+                await this.memoryCacheManager.clear();
+            }
+        }
+        catch (error) {
+            console.error('清理PDF转换器资源时出错:', error);
+        }
+    }
+    /**
+     * 获取可用模板
+     */
+    getAvailableTemplates() {
+        return this.templateManager.getTemplateNames();
+    }
+    /**
+     * 获取模板详情
+     */
+    async getTemplateInfo(name) {
+        return await this.templateManager.getTemplate(name);
+    }
+}
+//# sourceMappingURL=PDFConverter.js.map

package/dist/core/PDFConverter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"PDFConverter.js","sourceRoot":"","sources":["../../src/core/PDFConverter.ts"],"names":[],"mappings":"AAAA;;;GAGG;AAEH,OAAO,MAAM,MAAM,SAAS,CAAC;AAC7B,OAAO,EAAE,aAAa,EAA2C,MAAM,oBAAoB,CAAC;AAC5F,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjD,OAAO,EAAE,kBAAkB,EAAE,MAAM,mBAAmB,CAAC;AAEvD,OAAO,IAAI,MAAM,MAAM,CAAC;AACxB,OAAO,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,aAAa,CAAC;AAClD,OAAO,EAAE,aAAa,EAAE,MAAM,QAAQ,CAAC;AACvC,MAAM,OAAO,GAAG,aAAa,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAS/C,MAAM,OAAO,YAAa,SAAQ,aAAa;IACrC,WAAW,CAAuB;IAClC,eAAe,CAAkB;IACjC,gBAAgB,CAAmB;IACnC,YAAY,CAAe;IAC3B,kBAAkB,CAA4B;IAC9C,gBAAgB,CAA2C;IAEnE,YAAY,UAMR,EAAE;QACJ,KAAK,CAAC;YACJ,YAAY,EAAE,OAAO,CAAC,YAAY;YAClC,QAAQ,EAAE,OAAO,CAAC,QAAQ;SAC3B,CAAC,CAAC;QAEH,MAAM,EACJ,WAAW,GAAG,CAAC,EACf,aAAa,GAAG,CAAC,EACjB,WAAW,EACX,YAAY,GAAG,IAAI,EACpB,GAAG,OAAO,CAAC;QAEZ,UAAU;QACV,IAAI,CAAC,WAAW,GAAG,IAAI,oBAAoB,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,CAAC,eAAe,GAAG,IAAI,eAAe,CAAC,WAAW,CAAC,CAAC;QACxD,IAAI,CAAC,gBAAgB,GAAG,IAAI,gBAAgB,EAAE,CAAC;QAC/C,IAAI,CAAC,YAAY,GAAG,IAAI,YAAY,CAAC,IAAI,CAAC,WAAW,EAAE,IAAI,CAAC,eAAe,CAAC,CAAC;QAC7E,IAAI,CAAC,kBAAkB,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,kBAAkB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;QACzE,IAAI,CAAC,gBAAgB,GAAG,MAAM,CAAC,aAAa,CAAC,CAAC;QAE9C,SAAS;QACT,OAAO,CAAC,EAAE,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;QAC3C,OAAO,CAAC,EAAE,CAAC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC;IAC9C,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO,CACX,eAAuB,EACvB,UAAkB,EAClB,UAAgC,EAAE;QAElC,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,CAAC;YACH,cAAc;YACd,MAAM,EAAE,OAAO,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAI,CAAC,aAAa,CAAC,eAAe,CAAC,CAAC;YAE3E,IAAI,CAAC,cAAc,CAAC;gBAClB,KAAK,EAAE,SAAS;gBAChB,QAAQ,EAAE,EAAE;gBACZ,OAAO,EAAE,YAAY;aACtB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YAErB,aAAa;YACb,MAAM,QAAQ,GAAG,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YACzD,IAAI,YAAY,GAAkB,IAAI,CAAC;YAEvC,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,YAAY,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;gBACzD,IAAI,YAAY,EAAE,CAAC;oBACjB,MAAM,IAAI,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC;oBACvC,MAAM,SAAS,CAAC,UAAU,EAAE,YAAY,CAAC,CAAC;oBAE1C,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,CAAC;oBACpD,IAAI,CAAC,cAAc,CAAC;wBAClB,KAAK,EAAE,UAAU;wBACjB,QAAQ,EAAE,GAAG;wBACb,OAAO,EAAE,QAAQ;qBAClB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;oBAErB,OAAO;wBACL,OAAO,EAAE,IAAI;wBACb,QAAQ,EAAE,UAAU;wBACpB,QAAQ;wBACR,KAAK,EAAE;4BACL,cAAc,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;4BACtC,QAAQ,EAAE,UAAU,CAAC,QAAQ,IAAI,CAAC;4BAClC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,CAAC;4BACtC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,CAAC;4BACtC,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,CAAC;4BAC9B,QAAQ;yBACT;qBACF,CAAC;gBACJ,CAAC;YACH,CAAC;YAED,aAAa;YACb,MAAM,cAAc,GAAG,IAAI,CAAC,kBAAkB;gBAC5C,CAAC,CAAC,kBAAkB,CAAC,WAAW,CAAC,MAAM,EAAE;oBACrC,QAAQ,EAAE,OAAO;oBACjB,QAAQ,EAAE,OAAO,CAAC,QAAQ;oBAC1B,cAAc,EAAE,OAAO,CAAC,cAAc;oBACtC,SAAS,EAAE,OAAO,CAAC,SAAS;iBAC7B,CAAC;gBACJ,CAAC,CAAC,IAAI,CAAC;YAET,IAAI,IAAY,CAAC;YACjB,IAAI,cAAc,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC9C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAS,cAAc,CAAC,CAAC;gBACzE,IAAI,MAAM,EAAE,CAAC;oBACX,IAAI,CAAC,cAAc,CAAC;wBAClB,KAAK,EAAE,SAAS;wBAChB,QAAQ,EAAE,EAAE;wBACZ,OAAO,EAAE,aAAa;qBACvB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;oBACrB,IAAI,GAAG,MAAM,CAAC;gBAChB,CAAC;qBAAM,CAAC;oBACN,eAAe;oBACf,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;oBACjE,IAAI,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;oBACrE,MAAM,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,cAAc,EAAE,IAAI,EAAE,MAAM,CAAC,CAAC,CAAC,UAAU;gBAC7E,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,WAAW;gBACX,IAAI,CAAC,gBAAgB,CAAC,YAAY,CAAC,OAAO,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC;gBACjE,IAAI,GAAG,MAAM,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,OAAO,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YACvE,CAAC;YAED,IAAI,CAAC,cAAc,CAAC;gBAClB,KAAK,EAAE,YAAY;gBACnB,QAAQ,EAAE,EAAE;gBACZ,OAAO,EAAE,SAAS;aACnB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YAErB,QAAQ;YACR,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,WAAW,CAAC,IAAI,EAAE,UAAU,EAAE;gBACnE,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,SAAS;gBACvC,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,SAAS,EAAE,OAAO,CAAC,SAAS;gBAC5B,gBAAgB,EAAE,OAAO,CAAC,QAAQ;aACnC,CAAC,CAAC;YAEH,UAAU;YACV,IAAI,IAAI,CAAC,YAAY,EAAE,CAAC;gBACtB,MAAM,SAAS,GAAG,MAAM,QAAQ,CAAC,MAAM,CAAC,CAAC;gBACzC,MAAM,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,KAAK,EAAE,SAAS,CAAC,CAAC;YACrD,CAAC;YAED,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAChD,MAAM,cAAc,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YAE9C,IAAI,CAAC,cAAc,CAAC;gBAClB,KAAK,EAAE,UAAU;gBACjB,QAAQ,EAAE,GAAG;gBACb,OAAO,EAAE,YAAY,IAAI,CAAC,cAAc,CAAC,QAAQ,CAAC,EAAE;aACrD,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YAErB,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,QAAQ,EAAE,MAAM;gBAChB,QAAQ;gBACR,KAAK,EAAE;oBACL,cAAc;oBACd,QAAQ,EAAE,UAAU,CAAC,QAAQ,IAAI,CAAC;oBAClC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,CAAC;oBACtC,UAAU,EAAE,UAAU,CAAC,UAAU,IAAI,CAAC;oBACtC,MAAM,EAAE,UAAU,CAAC,MAAM,IAAI,CAAC;oBAC9B,QAAQ;iBACT;aACF,CAAC;QAEJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;YACjC,IAAI,CAAC,cAAc,CAAC;gBAClB,KAAK,EAAE,SAAS;gBAChB,QAAQ,EAAE,CAAC;gBACX,OAAO,EAAE,MAAM;gBACf,KAAK,EAAE,KAAc;aACtB,EAAE,OAAO,CAAC,QAAQ,CAAC,CAAC;YAErB,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC;aAC9D,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAChB,MAIE,EACF,cAKU;QAEV,MAAM,OAAO,GAAa,EAAE,CAAC;QAC7B,MAAM,MAAM,GAAY,EAAE,CAAC;QAC3B,IAAI,SAAS,GAAG,CAAC,CAAC;QAElB,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE,CACxC,IAAI,CAAC,gBAAgB,CAAC,KAAK,IAAI,EAAE;YAC/B,IAAI,CAAC;gBACH,MAAM,gBAAgB,GAAG,KAAK,CAAC,OAAO,EAAE,QAAQ,IAAI,CAAC,CAAC,QAAQ,EAAE,EAAE;oBAChE,cAAc,EAAE,CAAC;wBACf,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,MAAM;wBACpB,OAAO,EAAE,GAAG,KAAK,GAAG,CAAC,IAAI,MAAM,CAAC,MAAM,KAAK,QAAQ,CAAC,OAAO,EAAE;wBAC7D,MAAM;qBACP,CAAC,CAAC;gBACL,CAAC,CAAC,CAAC;gBAEH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAC/B,KAAK,CAAC,QAAQ,EACd,KAAK,CAAC,UAAU,EAChB;oBACE,GAAG,KAAK,CAAC,OAAO;oBAChB,QAAQ,EAAE,gBAAgB;iBAC3B,CACF,CAAC;gBAEF,IAAI,MAAM,CAAC,OAAO,IAAI,MAAM,CAAC,QAAQ,EAAE,CAAC;oBACtC,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,CAAC,QAAQ,CAAC;oBACjC,SAAS,EAAE,CAAC;oBAEZ,cAAc,EAAE,CAAC;wBACf,SAAS;wBACT,KAAK,EAAE,MAAM,CAAC,MAAM;wBACpB,OAAO,EAAE,OAAO,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,QAAQ,CAAC,EAAE;wBAChD,MAAM;qBACP,CAAC,CAAC;oBAEH,OAAO,MAAM,CAAC,QAAQ,CAAC;gBACzB,CAAC;qBAAM,CAAC;oBACN,MAAM,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,IAAI,MAAM,CAAC,CAAC;gBAC1C,CAAC;YAEH,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,KAAc,CAAC,CAAC;gBAC5B,SAAS,EAAE,CAAC;gBAEZ,cAAc,EAAE,CAAC;oBACf,SAAS;oBACT,KAAK,EAAE,MAAM,CAAC,MAAM;oBACpB,OAAO,EAAE,OAAO,KAAK,CAAC,UAAU,EAAE;oBAClC,MAAM;iBACP,CAAC,CAAC;gBAEH,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QAEF,WAAW;QACX,MAAM,cAAc,GAAG,MAAM,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;QAEvD,UAAU;QACV,MAAM,iBAAiB,GAAa,EAAE,CAAC;QACvC,cAAc,CAAC,OAAO,CAAC,CAAC,MAAM,EAAE,KAAK,EAAE,EAAE;YACvC,IAAI,MAAM,CAAC,MAAM,KAAK,WAAW,IAAI,MAAM,CAAC,KAAK,EAAE,CAAC;gBAClD,iBAAiB,CAAC,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YACvC,CAAC;QACH,CAAC,CAAC,CAAC;QAEH,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,IAAI,iBAAiB,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YACxD,MAAM,IAAI,KAAK,CAAC,WAAW,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACtE,CAAC;QAED,OAAO,iBAAiB,CAAC;IAC3B,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,YAAoB,EACpB,UAAmB,EACnB,UAAgC,EAAE;QAElC,SAAS;QACT,MAAM,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,CAAC;QAE3C,MAAM,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;QACvC,MAAM,QAAQ,GAAG,MAAM,EAAE,CAAC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC,CAAC;QAE1D,MAAM,eAAe,GAAG,UAAU;YAChC,YAAY,CAAC,OAAO,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC;QAEzC,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,eAAe,EAAE,OAAO,CAAC,CAAC;IAC1D,CAAC;IAED;;OAEG;IACH,QAAQ;QAQN,MAAM,QAAQ,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;QAEvC,OAAO;YACL,WAAW,EAAE,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE;YACxC,KAAK,EAAE,IAAI,CAAC,kBAAkB,EAAE,QAAQ,EAAE;YAC1C,MAAM,EAAE;gBACN,MAAM,EAAE;oBACN,IAAI,EAAE,QAAQ,CAAC,QAAQ;oBACvB,KAAK,EAAE,QAAQ,CAAC,SAAS;iBAC1B;gBACD,GAAG,EAAE,OAAO,CAAC,QAAQ,EAAE,CAAC,IAAI,GAAG,OAAO,CAAC,OAAO;aAC/C;SACF,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,OAAO;QACX,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,CAAC;YACjC,IAAI,IAAI,CAAC,kBAAkB,EAAE,CAAC;gBAC5B,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,CAAC;YACxC,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,OAAO,CAAC,KAAK,CAAC,gBAAgB,EAAE,KAAK,CAAC,CAAC;QACzC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,qBAAqB;QACnB,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,EAAE,CAAC;IACjD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,IAAY;QAChC,OAAO,MAAM,IAAI,CAAC,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,CAAC;IACtD,CAAC;CACF"}

package/dist/core/PDFGenerator.d.ts

@@ -0,0 +1,32 @@
+import { PDFConfig, ConversionProgress } from '../types/index.js';
+import { PuppeteerBrowserPool } from './BrowserPool.js';
+import { TemplateManager } from './TemplateManager.js';
+/**
+ * 高性能PDF生成器
+ * 支持并行生成、模板系统、水印等功能
+ */
+export declare class PDFGenerator {
+    private browserPool;
+    private templateManager;
+    constructor(browserPool: PuppeteerBrowserPool, templateManager: TemplateManager);
+    generatePDF(html: string, outputPath: string, options?: {
+        template?: string;
+        pdfConfig?: Partial<PDFConfig>;
+        watermark?: {
+            text: string;
+            opacity?: number;
+            fontSize?: number;
+            color?: string;
+            rotation?: number;
+        };
+        progressCallback?: (progress: ConversionProgress) => void;
+    }): Promise<string>;
+    private buildFullHtml;
+    private buildWatermarkHtml;
+    private waitForMermaidRender;
+    private buildPDFConfig;
+    private createTempHtml;
+    private getUniqueFilePath;
+    private delay;
+}
+//# sourceMappingURL=PDFGenerator.d.ts.map

package/dist/core/PDFGenerator.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"PDFGenerator.d.ts","sourceRoot":"","sources":["../../src/core/PDFGenerator.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,SAAS,EAAE,kBAAkB,EAAY,MAAM,mBAAmB,CAAC;AAC5E,OAAO,EAAE,oBAAoB,EAAE,MAAM,kBAAkB,CAAC;AACxD,OAAO,EAAE,eAAe,EAAE,MAAM,sBAAsB,CAAC;AAMvD;;;GAGG;AACH,qBAAa,YAAY;IAErB,OAAO,CAAC,WAAW;IACnB,OAAO,CAAC,eAAe;gBADf,WAAW,EAAE,oBAAoB,EACjC,eAAe,EAAE,eAAe;IAGpC,WAAW,CACf,IAAI,EAAE,MAAM,EACZ,UAAU,EAAE,MAAM,EAClB,OAAO,GAAE;QACP,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,SAAS,CAAC,EAAE,OAAO,CAAC,SAAS,CAAC,CAAC;QAC/B,SAAS,CAAC,EAAE;YACV,IAAI,EAAE,MAAM,CAAC;YACb,OAAO,CAAC,EAAE,MAAM,CAAC;YACjB,QAAQ,CAAC,EAAE,MAAM,CAAC;YAClB,KAAK,CAAC,EAAE,MAAM,CAAC;YACf,QAAQ,CAAC,EAAE,MAAM,CAAC;SACnB,CAAC;QACF,gBAAgB,CAAC,EAAE,CAAC,QAAQ,EAAE,kBAAkB,KAAK,IAAI,CAAC;KACtD,GACL,OAAO,CAAC,MAAM,CAAC;YA6HJ,aAAa;IA4L3B,OAAO,CAAC,kBAAkB;YAUZ,oBAAoB;IAgBlC,OAAO,CAAC,cAAc;YAuBR,cAAc;YAYd,iBAAiB;YAgBjB,KAAK;CAGpB"}

package/dist/core/PDFGenerator.js

@@ -0,0 +1,352 @@
+import fs from 'fs/promises';
+import path from 'path';
+import crypto from 'crypto';
+import os from 'os';
+/**
+ * 高性能PDF生成器
+ * 支持并行生成、模板系统、水印等功能
+ */
+export class PDFGenerator {
+    browserPool;
+    templateManager;
+    constructor(browserPool, templateManager) {
+        this.browserPool = browserPool;
+        this.templateManager = templateManager;
+    }
+    async generatePDF(html, outputPath, options = {}) {
+        const { template = 'default', pdfConfig = {}, watermark, progressCallback } = options;
+        let browser = null;
+        let page = null;
+        try {
+            progressCallback?.({
+                stage: 'rendering',
+                progress: 0,
+                message: 'Acquiring browser instance...'
+            });
+            // 获取浏览器实例
+            browser = await this.browserPool.getBrowser();
+            page = await browser.newPage();
+            progressCallback?.({
+                stage: 'rendering',
+                progress: 10,
+                message: 'Loading template...'
+            });
+            // 加载模板
+            const templateObj = await this.templateManager.getTemplate(template);
+            if (!templateObj) {
+                throw new Error(`Template not found: ${template}`);
+            }
+            progressCallback?.({
+                stage: 'rendering',
+                progress: 20,
+                message: 'Building full HTML...'
+            });
+            // 构建完整HTML
+            const fullHtml = await this.buildFullHtml(html, templateObj, watermark);
+            // 创建临时HTML文件
+            const tempHtmlPath = await this.createTempHtml(fullHtml);
+            progressCallback?.({
+                stage: 'rendering',
+                progress: 40,
+                message: 'Configuring page...'
+            });
+            // 设置视口
+            await page.setViewport({
+                width: 1200,
+                height: 1600,
+                deviceScaleFactor: 2 // 提高清晰度
+            });
+            // 加载页面
+            await page.goto(`file://${tempHtmlPath}`, {
+                waitUntil: 'networkidle0',
+                timeout: 60000
+            });
+            progressCallback?.({
+                stage: 'rendering',
+                progress: 60,
+                message: 'Waiting for content rendering...'
+            });
+            // 等待Mermaid图表渲染
+            await this.waitForMermaidRender(page);
+            progressCallback?.({
+                stage: 'generating',
+                progress: 80,
+                message: 'Generating PDF...'
+            });
+            // 确保输出目录存在
+            await fs.mkdir(path.dirname(outputPath), { recursive: true });
+            // 获取最终输出路径（处理重名）
+            const finalOutputPath = await this.getUniqueFilePath(outputPath);
+            // 生成PDF
+            const finalPdfConfig = this.buildPDFConfig(pdfConfig, templateObj);
+            await page.pdf({
+                path: finalOutputPath,
+                ...finalPdfConfig
+            });
+            // 清理临时文件
+            await fs.unlink(tempHtmlPath).catch(() => { });
+            progressCallback?.({
+                stage: 'complete',
+                progress: 100,
+                message: 'PDF generation completed',
+                filename: finalOutputPath
+            });
+            return finalOutputPath;
+        }
+        catch (error) {
+            progressCallback?.({
+                stage: 'error',
+                progress: 0,
+                message: 'Error generating PDF',
+                error: error
+            });
+            throw error;
+        }
+        finally {
+            // 释放浏览器
+            if (browser) {
+                this.browserPool.releaseBrowser(browser);
+            }
+            if (page && !page.isClosed()) {
+                await page.close().catch(() => { });
+            }
+        }
+    }
+    async buildFullHtml(content, template, watermark) {
+        const wm = watermark
+            ? { opacity: 0.1, fontSize: 60, color: '#cccccc', rotation: -45, ...watermark }
+            : undefined;
+        const watermarkHtml = wm ? this.buildWatermarkHtml(wm) : '';
+        return `<!DOCTYPE html>
+<html lang="zh-CN">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>Generated PDF</title>
+  
+  <!-- Mermaid -->
+  <script src="https://cdn.jsdelivr.net/npm/mermaid@10/dist/mermaid.min.js"></script>
+  
+  <!-- Highlight.js CSS -->
+  <link rel="stylesheet" href="https://cdnjs.cloudflare.com/ajax/libs/highlight.js/11.9.0/styles/github.min.css">
+  
+  <!-- 模板样式 -->
+  <style>
+    ${template.css}
+    
+    /* 水印样式 */
+    ${wm ? `
+      .watermark {
+        position: fixed;
+        top: 50%;
+        left: 50%;
+        transform: translate(-50%, -50%) rotate(${wm.rotation}deg);
+        font-size: ${wm.fontSize}px;
+        color: ${wm.color};
+        opacity: ${wm.opacity};
+        pointer-events: none;
+        user-select: none;
+        z-index: 1000;
+        white-space: nowrap;
+        font-weight: bold;
+      }
+    ` : ''}
+    
+    /* 任务列表样式 */
+    .task-list-item {
+      list-style: none;
+      margin-left: -20px;
+    }
+    
+    .task-list-item input[type="checkbox"] {
+      margin-right: 8px;
+    }
+    
+    /* 警告框样式 */
+    .alert {
+      padding: 12px 16px;
+      margin: 16px 0;
+      border-radius: 4px;
+      border-left: 4px solid;
+    }
+    
+    .alert-note { border-left-color: #0969da; background-color: #f6f8fa; }
+    .alert-tip { border-left-color: #1a7f37; background-color: #f6f8fa; }
+    .alert-important { border-left-color: #8250df; background-color: #f6f8fa; }
+    .alert-warning { border-left-color: #d1242f; background-color: #fff8f8; }
+    .alert-caution { border-left-color: #bf8700; background-color: #fff8c5; }
+    
+    /* 代码块增强 */
+    .code-block-container {
+      position: relative;
+      margin: 16px 0;
+    }
+    
+    .code-language {
+      position: absolute;
+      top: 8px;
+      right: 8px;
+      background: rgba(0, 0, 0, 0.1);
+      padding: 2px 8px;
+      border-radius: 3px;
+      font-size: 12px;
+      opacity: 0.7;
+    }
+    
+    /* 表格容器 */
+    .table-container {
+      overflow-x: auto;
+      margin: 16px 0;
+    }
+    
+    .enhanced-table {
+      border-collapse: collapse;
+      width: 100%;
+    }
+    
+    /* Mermaid容器 */
+    .mermaid-container {
+      text-align: center;
+      margin: 20px 0;
+      page-break-inside: avoid;
+    }
+    
+    /* 脚注样式 */
+    .footnote {
+      font-size: 12px;
+      margin-top: 20px;
+      padding: 8px;
+      background-color: #f8f9fa;
+      border-left: 3px solid #dee2e6;
+    }
+    
+    .footnote-ref {
+      text-decoration: none;
+      font-weight: bold;
+    }
+    
+    /* 打印优化 */
+    @media print {
+      .watermark {
+        position: fixed !important;
+        top: 50% !important;
+        left: 50% !important;
+      }
+      
+      .mermaid-container,
+      .code-block-container,
+      .table-container {
+        page-break-inside: avoid;
+      }
+      
+      h1, h2, h3, h4, h5, h6 {
+        page-break-after: avoid;
+      }
+    }
+  </style>
+</head>
+<body>
+  ${watermarkHtml}
+  <div class="content">
+    ${content}
+  </div>
+  
+  <script>
+    // 初始化Mermaid
+    document.addEventListener('DOMContentLoaded', function() {
+      mermaid.initialize({
+        startOnLoad: false,
+        theme: 'default',
+        themeVariables: {
+          fontFamily: 'inherit'
+        }
+      });
+      
+      // 渲染所有Mermaid图表
+      const mermaidElements = document.querySelectorAll('.mermaid[data-processed="false"]');
+      
+      if (mermaidElements.length > 0) {
+        mermaid.run({
+          nodes: mermaidElements
+        }).then(() => {
+          // 标记为已处理
+          mermaidElements.forEach(el => el.setAttribute('data-processed', 'true'));
+        }).catch(error => {
+          console.error('Mermaid渲染失败:', error);
+          // 显示错误信息
+          mermaidElements.forEach(el => {
+            el.innerHTML = \`<div style="color: red; padding: 10px; border: 1px solid red;">
+              <strong>Mermaid渲染失败:</strong><br>
+              \${error.message}
+            </div>\`;
+            el.setAttribute('data-processed', 'error');
+          });
+        });
+      }
+    });
+  </script>
+</body>
+</html>`;
+    }
+    buildWatermarkHtml(watermark) {
+        return `<div class="watermark">${watermark.text}</div>`;
+    }
+    async waitForMermaidRender(page) {
+        // 等待Mermaid渲染完成
+        await page.waitForFunction(() => {
+            const mermaidElements = document.querySelectorAll('.mermaid');
+            return Array.from(mermaidElements).every(el => el.getAttribute('data-processed') === 'true' ||
+                el.getAttribute('data-processed') === 'error');
+        }, { timeout: 30000 }).catch(() => {
+            console.warn('Mermaid渲染超时，继续生成PDF');
+        });
+        // 额外等待确保渲染稳定
+        await this.delay(2000);
+    }
+    buildPDFConfig(config, template) {
+        const defaultConfig = {
+            format: 'a4',
+            orientation: 'portrait',
+            margin: {
+                top: '20mm',
+                right: '15mm',
+                bottom: '20mm',
+                left: '15mm'
+            },
+            printBackground: true,
+            displayHeaderFooter: false,
+            preferCSSPageSize: true,
+            scale: 1
+        };
+        return {
+            ...defaultConfig,
+            ...template.config,
+            ...config
+        };
+    }
+    async createTempHtml(html) {
+        // 使用系统临时目录而不是当前工作目录
+        const tempDir = path.join(os.tmpdir(), 'enhanced-markdown2pdf');
+        await fs.mkdir(tempDir, { recursive: true });
+        const hash = crypto.createHash('md5').update(html).digest('hex');
+        const tempPath = path.join(tempDir, `temp_${hash}.html`);
+        await fs.writeFile(tempPath, html, 'utf-8');
+        return tempPath;
+    }
+    async getUniqueFilePath(basePath) {
+        const dir = path.dirname(basePath);
+        const ext = path.extname(basePath);
+        const name = path.basename(basePath, ext);
+        let counter = 1;
+        let uniquePath = basePath;
+        while (await fs.access(uniquePath).then(() => true).catch(() => false)) {
+            uniquePath = path.join(dir, `${name}-${counter}${ext}`);
+            counter++;
+        }
+        return uniquePath;
+    }
+    async delay(ms) {
+        await new Promise((resolve) => setTimeout(resolve, ms));
+    }
+}
+//# sourceMappingURL=PDFGenerator.js.map