nodejs用html转成pdf

2023-05-14 22:54

短信预约 -IT技能 免费直播动态提醒

随着互联网技术的不断发展，越来越多的应用程序需要将网页内容转换为 PDF 格式，以使其更易于分享和下载。在这个过程中，Node.js 可以作为一个强大的工具来实现将 HTML 文档转换为 PDF 格式。这篇文章将介绍使用 Node.js 将 HTML 转换为 PDF 的过程。

安装必要的依赖项

在开始之前，您需要在系统中安装以下必要的软件包：

Node.js 开发环境
npm - Node.js 的包管理器
wkhtmltopdf - 一个从网页生成 PDF 的工具

Wkhtmltopdf 是一个基于 WebKit 引擎的命令行工具，它可以将 html 文档转换为 pdf 文件，同时它还支持添加水印、页眉页脚等多种特性。

在 Ubuntu 系统下，您可以通过以下命令进行 wkhtmltopdf 的安装：

sudo apt-get install wkhtmltopdf

安装依赖项

接下来，您需要安装 pdfkit 和 fs 模块，这两个模块可通过 npm 包管理器进行安装。

在命令行输入以下命令进行安装：

npm install pdfkit

npm install fs

编写代码

安装完相关依赖项后，就可以开始编写 JavaScript 代码了。您可以创建一个名为“html-to-pdf.js”的文件，写入以下代码：

const pdfkit = require('pdfkit');
const fs = require('fs');
const wkhtmltopdf = require('wkhtmltopdf');

// 生成 PDF
function generatePDF(html, options, callback) {
    // 创建文件流
    const inputStream = fs.createReadStream(html);

    // 调用 wkhtmltopdf
    wkhtmltopdf(inputStream, options, (error, stream) => {
        if (error) {
            callback(error);
        } else {
            callback(null, stream);
        }
    });
}

// 写 PDF
function writePDF(doc, path, callback) {
    const writeStream = fs.createWriteStream(path);

    writeStream.on('error', (error) => {
        callback(error);
    });

    doc.pipe(writeStream);

    doc.end();
}

// 创建 PDF
function createPDF(html, path, options, callback) {
    const doc = new pdfkit();

    generatePDF(html, options, (error, stream) => {
        if (error) {
            callback(error);
        } else {
            doc.on('pageAdded', () => {
                doc.image(stream, {
                    fit: [doc.page.width - 100, doc.page.height - 100],
                    align: 'center',
                    valign: 'center'
                });
            });

            doc.pipe(fs.createWriteStream(path));

            doc.end();

            callback(null, path);
        }
    });
}

// 导出模块
module.exports = {
    createPDF,
    writePDF,
    generatePDF
};

在该代码中，我们引入了 pdfkit、fs 和 wkhtmltopdf 模块，并编写了三个函数：generatePDF、writePDF 和 createPDF 。

generatePDF 函数的作用是将 HTML 文件转换为 PDF 格式。它需要传递三个参数：html、options 和 callback。

writePDF 函数的作用是将 PDF 文档写入文件系统中。它需要传递三个参数：doc、path 和 callback。

createPDF 函数则是整个过程的控制中心，它可以调用 generatePDF 和 writePDF 函数来完成 HTML 转 PDF 的过程。

使用 Node.js 将 HTML 转换为 PDF

现在，您已经完成了代码的编写，可以开始使用 Node.js 将 HTML 转换为 PDF。您可以在命令行中输入以下命令：

node html-to-pdf.js

或者，您也可以在自己的项目中使用该模块：

const htmlToPdf = require('./html-to-pdf');

htmlToPdf.createPDF('test.html', 'test.pdf', {
   pageSize: 'letter',
   marginTop: '0mm',
   marginBottom: '0mm',
   marginLeft: '0mm',
   marginRight: '0mm'
}, (err, result) => {
   if (err) {
      console.error(err);
      return;
   }
   console.log('File saved to: ' + result);
});

在这个例子中，我们使用 createPDF 函数来将 test.html 转换为 test.pdf，同时设置了页面大小和边距。