node-pdf2img 0.1.1

package/bin/cli.js

@@ -0,0 +1,247 @@
+#!/usr/bin/env node
+
+/**
+ * pdf2img CLI - 命令行 PDF 转图片工具
+ *
+ * 用法：
+ *   pdf2img <input> [options]
+ *
+ * 示例：
+ *   pdf2img document.pdf -o ./output
+ *   pdf2img https://example.com/doc.pdf -o ./output
+ *   pdf2img document.pdf -p 1,2,3 -o ./output
+ *   pdf2img document.pdf --quality 90 --width 1920 -o ./output
+ *   pdf2img document.pdf --format png -o ./output  # 输出 PNG 格式
+ *   pdf2img document.pdf --cos --cos-prefix images/doc  # 上传到 COS
+ *
+ * COS 环境变量：
+ *   COS_SECRET_ID     - 腾讯云 SecretId
+ *   COS_SECRET_KEY    - 腾讯云 SecretKey
+ *   COS_BUCKET        - COS 存储桶名称
+ *   COS_REGION        - COS 地域（如 ap-guangzhou）
+ */
+
+import { program } from 'commander';
+import path from 'path';
+import fs from 'fs';
+import { fileURLToPath } from 'url';
+
+const __dirname = path.dirname(fileURLToPath(import.meta.url));
+const pkg = JSON.parse(fs.readFileSync(path.join(__dirname, '../package.json'), 'utf8'));
+
+// 动态导入 ora（ESM 模块）
+let ora;
+try {
+    ora = (await import('ora')).default;
+} catch {
+    // 如果 ora 不可用，提供一个简单的替代
+    ora = (text) => ({
+        start: () => ({ text, succeed: () => {}, fail: () => {}, stop: () => {} }),
+    });
+}
+
+// 处理 --version-info 选项（在解析前检查）
+if (process.argv.includes('--version-info')) {
+    const { isAvailable, getVersion } = await import('../src/index.js');
+    console.log(`pdf2img v${pkg.version}`);
+    console.log(`原生渲染器: ${getVersion()}`);
+    console.log(`可用: ${isAvailable() ? '是' : '否'}`);
+    process.exit(0);
+}
+
+program
+    .name('pdf2img')
+    .description('高性能 PDF 转图片工具，基于 PDFium')
+    .version(pkg.version)
+    .argument('<input>', 'PDF 文件路径或 URL')
+    .option('-o, --output <dir>', '输出目录（本地模式）', './output')
+    .option('-p, --pages <pages>', '要转换的页码（逗号分隔，如 1,2,3）')
+    .option('-w, --width <width>', '目标渲染宽度（像素）', '1920')
+    .option('-q, --quality <quality>', '图片质量（0-100，用于 webp/jpg）', '100')
+    .option('-f, --format <format>', '输出格式：webp, png, jpg', 'webp')
+    .option('--prefix <prefix>', '输出文件名前缀', 'page')
+    .option('--info', '仅显示 PDF 信息（页数）')
+    .option('--version-info', '显示原生渲染器版本')
+    .option('-v, --verbose', '详细输出')
+    // COS 相关选项
+    .option('--cos', '上传到腾讯云 COS（需配置环境变量）')
+    .option('--cos-prefix <prefix>', 'COS key 前缀（如 images/doc-123）')
+    .option('--cos-secret-id <id>', 'COS SecretId（优先使用环境变量 COS_SECRET_ID）')
+    .option('--cos-secret-key <key>', 'COS SecretKey（优先使用环境变量 COS_SECRET_KEY）')
+    .option('--cos-bucket <bucket>', 'COS 存储桶（优先使用环境变量 COS_BUCKET）')
+    .option('--cos-region <region>', 'COS 地域（优先使用环境变量 COS_REGION）')
+    .action(async (input, options) => {
+        // 设置调试模式
+        if (options.verbose) {
+            process.env.PDF2IMG_DEBUG = 'true';
+        }
+
+        // 动态导入主模块
+        const { convert, getPageCount, isAvailable, getVersion } = await import('../src/index.js');
+
+        // 显示版本信息
+        if (options.versionInfo) {
+            console.log(`pdf2img v${pkg.version}`);
+            console.log(`原生渲染器: ${getVersion()}`);
+            console.log(`可用: ${isAvailable() ? '是' : '否'}`);
+            return;
+        }
+
+        // 检查 native renderer
+        if (!isAvailable()) {
+            console.error('错误：原生渲染器不可用。');
+            console.error('请确保 PDFium 库已正确安装。');
+            process.exit(1);
+        }
+
+        // 验证格式
+        const supportedFormats = ['webp', 'png', 'jpg', 'jpeg'];
+        const format = options.format.toLowerCase();
+        if (!supportedFormats.includes(format)) {
+            console.error(`错误：不支持的格式 "${options.format}"。支持的格式：webp, png, jpg`);
+            process.exit(1);
+        }
+
+        // 检查输入
+        const isUrl = input.startsWith('http://') || input.startsWith('https://');
+        if (!isUrl && !fs.existsSync(input)) {
+            console.error(`错误：文件不存在: ${input}`);
+            process.exit(1);
+        }
+
+        // 仅显示 PDF 信息
+        if (options.info) {
+            if (isUrl) {
+                console.error('错误：--info 选项仅支持本地文件');
+                process.exit(1);
+            }
+
+            try {
+                const pageCount = getPageCount(input);
+                const stat = fs.statSync(input);
+                console.log(`文件: ${path.basename(input)}`);
+                console.log(`大小: ${(stat.size / 1024 / 1024).toFixed(2)} MB`);
+                console.log(`页数: ${pageCount}`);
+            } catch (err) {
+                console.error(`错误: ${err.message}`);
+                process.exit(1);
+            }
+            return;
+        }
+
+        // 解析页码
+        let pages = [];
+        if (options.pages) {
+            pages = options.pages.split(',').map(p => parseInt(p.trim(), 10)).filter(p => !isNaN(p) && p > 0);
+        }
+
+        // 确定输出类型和配置
+        let outputType = 'file';
+        let convertOptions = {
+            pages,
+            prefix: options.prefix,
+            targetWidth: parseInt(options.width, 10),
+            quality: parseInt(options.quality, 10),
+            format: format,
+        };
+
+        // COS 模式
+        if (options.cos) {
+            outputType = 'cos';
+
+            // 从环境变量或命令行参数获取 COS 配置
+            const cosConfig = {
+                secretId: options.cosSecretId || process.env.COS_SECRET_ID,
+                secretKey: options.cosSecretKey || process.env.COS_SECRET_KEY,
+                bucket: options.cosBucket || process.env.COS_BUCKET,
+                region: options.cosRegion || process.env.COS_REGION,
+            };
+
+            // 验证 COS 配置
+            const missingFields = [];
+            if (!cosConfig.secretId) missingFields.push('COS_SECRET_ID');
+            if (!cosConfig.secretKey) missingFields.push('COS_SECRET_KEY');
+            if (!cosConfig.bucket) missingFields.push('COS_BUCKET');
+            if (!cosConfig.region) missingFields.push('COS_REGION');
+
+            if (missingFields.length > 0) {
+                console.error('错误：COS 配置不完整，缺少以下环境变量或参数：');
+                console.error(`  ${missingFields.join(', ')}`);
+                console.error('\n请设置环境变量或使用命令行参数：');
+                console.error('  export COS_SECRET_ID=xxx');
+                console.error('  export COS_SECRET_KEY=xxx');
+                console.error('  export COS_BUCKET=xxx');
+                console.error('  export COS_REGION=ap-guangzhou');
+                process.exit(1);
+            }
+
+            convertOptions.cos = cosConfig;
+            convertOptions.cosKeyPrefix = options.cosPrefix || '';
+        } else {
+            // 本地文件模式
+            convertOptions.outputDir = options.output;
+        }
+
+        convertOptions.outputType = outputType;
+
+        // 开始转换
+        const modeText = options.cos ? '转换并上传' : '转换';
+        const spinner = ora(`正在${modeText} ${path.basename(input)}...`).start();
+
+        try {
+            const startTime = Date.now();
+
+            const result = await convert(input, convertOptions);
+
+            const duration = Date.now() - startTime;
+
+            spinner.succeed(`${modeText}完成 ${result.renderedPages}/${result.numPages} 页，格式: ${result.format.toUpperCase()}，耗时 ${duration}ms`);
+
+            // 显示结果
+            if (options.cos) {
+                console.log('\n已上传到 COS:');
+                for (const page of result.pages) {
+                    if (page.success) {
+                        console.log(`  第 ${page.pageNum} 页: ${page.width}x${page.height} -> ${page.cosKey} (${formatBytes(page.size)})`);
+                    } else {
+                        console.log(`  第 ${page.pageNum} 页: 失败 - ${page.error}`);
+                    }
+                }
+            } else {
+                console.log(`\n输出目录: ${path.resolve(options.output)}`);
+                console.log('\n页面详情:');
+                for (const page of result.pages) {
+                    if (page.success) {
+                        console.log(`  第 ${page.pageNum} 页: ${page.width}x${page.height} -> ${path.basename(page.outputPath)} (${formatBytes(page.size)})`);
+                    } else {
+                        console.log(`  第 ${page.pageNum} 页: 失败 - ${page.error}`);
+                    }
+                }
+            }
+
+            // 统计
+            const totalSize = result.pages.reduce((sum, p) => sum + (p.size || 0), 0);
+            console.log(`\n总输出大小: ${formatBytes(totalSize)}`);
+            console.log(`平均每页耗时: ${Math.round(duration / result.renderedPages)}ms`);
+
+        } catch (err) {
+            spinner.fail(`${modeText}失败: ${err.message}`);
+            if (options.verbose) {
+                console.error(err.stack);
+            }
+            process.exit(1);
+        }
+    });
+
+/**
+ * 格式化字节数
+ */
+function formatBytes(bytes) {
+    if (bytes === 0) return '0 B';
+    const k = 1024;
+    const sizes = ['B', 'KB', 'MB', 'GB'];
+    const i = Math.floor(Math.log(bytes) / Math.log(k));
+    return parseFloat((bytes / Math.pow(k, i)).toFixed(2)) + ' ' + sizes[i];
+}
+
+program.parse();

package/package.json

@@ -0,0 +1,71 @@
+{
+    "name": "node-pdf2img",
+    "version": "0.1.1",
+    "description": "High-performance PDF to image converter using PDFium native renderer",
+    "type": "module",
+    "main": "./src/index.js",
+    "types": "./src/index.d.ts",
+    "bin": {
+        "pdf2img": "./bin/cli.js"
+    },
+    "exports": {
+        ".": {
+            "import": "./src/index.js",
+            "types": "./src/index.d.ts"
+        }
+    },
+    "files": [
+        "src",
+        "bin"
+    ],
+    "scripts": {
+        "test": "node --test test/*.test.js",
+        "test:run": "node test/api.test.js",
+        "build:native": "cd ../native-renderer && npm run build",
+        "prepublishOnly": "npm run test"
+    },
+    "keywords": [
+        "pdf",
+        "image",
+        "converter",
+        "pdfium",
+        "webp",
+        "native",
+        "cli"
+    ],
+    "author": "",
+    "license": "MIT",
+    "engines": {
+        "node": ">=18.0.0"
+    },
+    "os": [
+        "linux",
+        "darwin",
+        "win32"
+    ],
+    "cpu": [
+        "x64",
+        "arm64"
+    ],
+    "dependencies": {
+        "commander": "^12.0.0",
+        "cos-nodejs-sdk-v5": "^2.14.6",
+        "dotenv": "^16.0.3",
+        "ora": "^8.0.0",
+        "p-limit": "^7.2.0",
+        "piscina": "^5.1.4",
+        "sharp": "^0.33.0",
+        "node-pdf2img-native": "file:../native-renderer"
+    },
+    "devDependencies": {
+        "@types/node": "^20.0.0"
+    },
+    "repository": {
+        "type": "git",
+        "url": "git+https://github.com/sigma-2026/node-pdf2img.git"
+    },
+    "bugs": {
+        "url": "https://github.com/sigma-2026/node-pdf2img/issues"
+    },
+    "homepage": "https://github.com/sigma-2026/node-pdf2img#readme"
+}

package/src/core/config.js

@@ -0,0 +1,101 @@
+/**
+ * PDF2IMG 配置
+ */
+
+// ==================== 渲染配置 ====================
+export const RENDER_CONFIG = {
+    // 目标渲染宽度（像素）
+    TARGET_RENDER_WIDTH: parseInt(process.env.TARGET_RENDER_WIDTH) || 1280,
+
+    // 图片密集型页面的目标宽度（像素）
+    IMAGE_HEAVY_TARGET_WIDTH: parseInt(process.env.IMAGE_HEAVY_TARGET_WIDTH) || 1024,
+
+    // 最大渲染缩放比例
+    MAX_RENDER_SCALE: parseFloat(process.env.MAX_RENDER_SCALE) || 4.0,
+
+    // 默认输出格式：webp, png, jpg
+    OUTPUT_FORMAT: process.env.OUTPUT_FORMAT || 'webp',
+
+    // Native Stream 阈值（字节）- 大于此值使用流式加载
+    NATIVE_STREAM_THRESHOLD: parseInt(process.env.NATIVE_STREAM_THRESHOLD) || 5 * 1024 * 1024, // 5MB
+};
+
+// ==================== 编码器配置 ====================
+export const ENCODER_CONFIG = {
+    // WebP 编码质量（0-100）
+    WEBP_QUALITY: parseInt(process.env.WEBP_QUALITY) || 80,
+    
+    // WebP 编码方法/速度（0-6，0最快，6最慢但压缩最好）
+    // 默认值 4 是速度和压缩率的最佳平衡点
+    WEBP_METHOD: parseInt(process.env.WEBP_METHOD) || 4,
+    
+    // JPEG 编码质量（0-100）
+    JPEG_QUALITY: parseInt(process.env.JPEG_QUALITY) || 85,
+    
+    // PNG 压缩级别（0-9，0不压缩，9最大压缩）
+    PNG_COMPRESSION: parseInt(process.env.PNG_COMPRESSION) || 6,
+};
+
+// ==================== 超时配置 ====================
+export const TIMEOUT_CONFIG = {
+    // 分片请求超时
+    RANGE_REQUEST_TIMEOUT: parseInt(process.env.RANGE_REQUEST_TIMEOUT) || 25000, // 25s
+
+    // 下载超时
+    DOWNLOAD_TIMEOUT: parseInt(process.env.DOWNLOAD_TIMEOUT) || 60000, // 60s
+};
+
+// ==================== 支持的输出格式 ====================
+export const SUPPORTED_FORMATS = ['webp', 'png', 'jpg', 'jpeg'];
+
+/**
+ * 合并用户配置与默认配置
+ * @param {Object} userConfig - 用户配置
+ * @returns {Object} 合并后的配置（用于原生渲染器）
+ */
+export function mergeConfig(userConfig = {}) {
+    const format = userConfig.format ?? RENDER_CONFIG.OUTPUT_FORMAT;
+    
+    return {
+        targetWidth: userConfig.targetWidth ?? RENDER_CONFIG.TARGET_RENDER_WIDTH,
+        imageHeavyWidth: userConfig.imageHeavyWidth ?? RENDER_CONFIG.IMAGE_HEAVY_TARGET_WIDTH,
+        maxScale: userConfig.maxScale ?? RENDER_CONFIG.MAX_RENDER_SCALE,
+        detectScan: userConfig.detectScan ?? true,
+        format,
+        
+        // WebP 编码配置
+        webpQuality: userConfig.webp?.quality ?? userConfig.quality ?? ENCODER_CONFIG.WEBP_QUALITY,
+        webpMethod: userConfig.webp?.method ?? ENCODER_CONFIG.WEBP_METHOD,
+        
+        // JPEG 编码配置
+        jpegQuality: userConfig.jpeg?.quality ?? userConfig.quality ?? ENCODER_CONFIG.JPEG_QUALITY,
+        
+        // PNG 编码配置
+        pngCompression: userConfig.png?.compressionLevel ?? ENCODER_CONFIG.PNG_COMPRESSION,
+    };
+}
+
+/**
+ * 获取文件扩展名
+ * @param {string} format - 格式名称
+ * @returns {string} 文件扩展名
+ */
+export function getExtension(format) {
+    if (format === 'jpeg') return 'jpg';
+    return format;
+}
+
+/**
+ * 获取 MIME 类型
+ * @param {string} format - 格式名称
+ * @returns {string} MIME 类型
+ */
+export function getMimeType(format) {
+    const mimeTypes = {
+        webp: 'image/webp',
+        png: 'image/png',
+        jpg: 'image/jpeg',
+        jpeg: 'image/jpeg',
+    };
+    return mimeTypes[format] || 'image/webp';
+}