把HTML转成PDF的4个方案及实现方法

素颜马尾好姑娘i 2022-10-01 11:46 136阅读 0赞

> 翻译：疯狂的技术宅 原文：[blog.risingstack.com/pdf-from-ht…][blog.risingstack.com_pdf-from-ht]

**在本文中，我将展示如何使用 Node.js、Puppeteer、headless Chrome 和 Docker 从样式复杂的 React 页面生成 PDF 文档。**

背景：几个月前，一个客户要求我们开发一个功能，用户可以得到 PDF 格式的 React 页面内容。该页面基本上是患者病例的报告和数据可视化结果，其中包含许多 SVG。另外还有一些特殊的请求来操纵布局，并对 HTML 元素进行一些重新排列。因此与原始的 React 页面相比，PDF 中应该有不同的样式和额外的内容。

**由于这个任务比用简单的 CSS 规则解决要复杂得多，所以我们先探讨了可能的实现方法。我们找到了 3 个主要解决方案。这篇博文将指导你了解它们的可能性并最终实施。**

## 目录： ##

*  在客户端还是服务器端生成？
 *  方案1：从 DOM 制作屏幕截图
 *  方案2：仅使用 PDF 库
 *  最终方案3：Node.js、Puppeteer 和 Headless Chrome
    
     *  样式控制
     *  将文件发送到客户端并保存
 *  在 Docker 中使用 Puppeteer
 *  方案3 +1：CSS打印规则
 *  总结

## 在客户端还是服务器端生成？ ##

在客户端和服务器端都可以生成PDF文件。但是让后端处理它可能更有意义，因为你并不想耗尽用户浏览器可以提供的所有资源。

即便如此，我仍然会展示这两种方法的解决方案。

## 方案1：从 DOM 制作屏幕截图 ##

乍一看，这个解决方案似乎是最简单的，事实证明的确是这样，但它有其自身的局限性。如果你没有特殊需求，例如在 PDF 中选择文本或对文本进行搜索，那么这就是一种简单易用的方法。

此方法简单明了：从页面创建屏幕截图，并把它放到 PDF 文件中。非常直截了当。我们可以使用两个包来实现：

*  [Html2canvas][]，根据 DOM 生成截图
 *  [jsPdf][]，一个生成PDF的库

开始编码：

`npm install html2canvas jspdf`

import html2canvas from 'html2canvas'
    import jsPdf from 'jspdf'
     
    function printPDF () {
        const domElement = document.getElementById('your-id')
        html2canvas(domElement, { onclone: (document) => {
          document.getElementById('print-button').style.visibility = 'hidden'
    }})
        .then((canvas) => {
            const img = canvas.toDataURL('image/png')
            const pdf = new jsPdf()
            pdf.addImage(imgData, 'JPEG', 0, 0, width, height)
            pdf.save('your-filename.pdf')
    })
    复制代码

就这样！

请注意 `html2canvas` 的 `onclone`方法。当你在截图之前需要操纵 DOM（例如隐藏打印按钮）时，它是非常方便的。我看到过很多使用这个包的项目。但不幸的是，这不是我们想要的，因为我们需要在后端完成对 PDF 的创建工作。

## 方案2：只使用 PDF 库 ##

NPM上有几个库，如 jsPDF（如上所述）或[PDFKit][]。他们的问题是，如果我想使用这些库，我将不得不重新调整页面结构。这肯定会损害可维护性，因为我需要将所有后续更改应用到 PDF 模板和 React 页面中。

请看下面的代码。你需要亲自手动创建 PDF 文档。你需要遍历 DOM 并找出每个元素并将其转换为 PDF 格式，这是一项繁琐的工作。必须找到一个更简单的方法。

doc = new PDFDocument
    doc.pipe fs.createWriteStream('output.pdf')
    doc.font('fonts/PalatinoBold.ttf')
       .fontSize(25)
       .text('Some text with an embedded font!', 100, 100)
     
    doc.image('path/to/image.png', {
       fit: [250, 300],
       align: 'center',
       valign: 'center'
    });
     
    doc.addPage()
       .fontSize(25)
       .text('Here is some vector graphics...', 100, 100)
     
    doc.end()
    复制代码

这段代码段来自 PDFKit 文档。但是如果你的目标是直接生成一个 PDF 文件，而不是对一个已经存在的（并且不断变化的）HTML 页面进行转换，它还是很有用的。

## 最终方案3：基于 Node.js 的 Puppeteer 和 Headless Chrome ##

什么是 [Puppeteer][]？其文档中写道：

> Puppeteer 是一个 Node 库，它提供了一个高级 API 来控制 DevTools 协议上的 Chrome 或 Chromium。 Puppeteer 默认以 headless 模式运行 Chrome 或 Chromium，但其也可以被配置为完整的（non-headless）模式运行。

它本质上是一个可以从 Node.js 运行的浏览器。如果你读过它的文档，其中首先提到的就是你可以用 Puppeteer 来**生成页面的截图和PDF**。优秀！这正是我们想要的。

先用 `npmi i puppeteer` 安装 Puppeteer，并实现我们的功能。

这是一个简单的功能，可导航到 URL 并生成站点的 PD F文件。

首先，我们启动浏览器（仅在 headless 模式下支持 PDF 生成），然后打开新页面，设置视口，并导航到提供的URL。

设置 `waitUntil:'networkidle0'` 选项意味着当至少500毫秒没有网络连接时，Puppeteer 会认为导航已完成。 （可以从 [API docs][] 获取更多信息。）

**之后，我们将 PDF 保存为变量，关闭浏览器并返回 PDF。**

注意：`page.pdf` 方法接收 `options` 对象，你可以使用 'path' 选项将文件保存到磁盘。如果未提供路径，则 PDF 将不会被保存到磁盘，而是会得到缓冲区。（稍后我将讨论如何处理它。）

如果需要先登录才能从受保护的页面生成 PDF，首先你要导航到登录页面，检查表单元素的 ID 或名称，填写它们，然后提交表单：

await page.type('#email', process.env.PDF_USER)
    await page.type('#password', process.env.PDF_PASSWORD)
    await page.click('#submit')
    复制代码

要始终将登录凭据保存在环境变量中，不要硬编码！

### 样式控制 ###

Puppeteer 也有这种样式操作的解决方案。你可以在生成 PDF 之前插入样式标记，Puppeteer 将生成具有已修改样式的文件。

await page.addStyleTag({ content: '.nav { display: none} .navbar { border: 0px} #print-button {display: none}' })
    复制代码

### 将文件发送到客户端并保存 ###

好的，现在你已经在后端生成了一个 PDF 文件。接下来做什么？

如上所述，如果你不把文件保存到磁盘，将会得到一个缓冲区。你只需要把含有适当内容类型的缓冲区发送到前端即可。

printPDF.then(pdf => {
    	res.set({ 'Content-Type': 'application/pdf', 'Content-Length': pdf.length })
    	res.send(pdf)
    复制代码

现在，你只需在浏览器向服务器发送请求即可得到生成的 PDF。

function getPDF() {
     return axios.get(`${API_URL}/your-pdf-endpoint`, {
       responseType: 'arraybuffer',
       headers: {
         'Accept': 'application/pdf'
       }
     })
    复制代码

一旦发送了请求，缓冲区的内容就应该开始下载了。最后一步是将缓冲区数据转换为 PDF 文件。

savePDF = () => {
        this.openModal(‘Loading…’) // open modal
       return getPDF() // API call
         .then((response) => {
           const blob = new Blob([response.data], {type: 'application/pdf'})
           const link = document.createElement('a')
           link.href = window.URL.createObjectURL(blob)
           link.download = `your-file-name.pdf`
           link.click()
           this.closeModal() // close modal
         })
       .catch(err => /** error handling **/)
     }
    <button onClick={this.savePDF}>Save as PDF</button>
    复制代码

就这样！如果单击“保存”按钮，那么浏览器将会保存 PDF。

## 在 Docker 中使用 Puppeteer ##

我认为这是实施中最棘手的部分 —— 所以让我帮你节省几个小时的百度时间。

官方文档指出\*“在 Docker 中使用 headless Chrome 并使其运行起来可能会非常棘手”\*。官方文档有[疑难解答][Link 1]部分，你可以找到有关用 Docker 安装 puppeteer 的所有必要信息。

如果你在 Alpine 镜像上安装 Puppeteer，请确保在看到[页面的这一部分][Link 2]时再向下滚动一点。否则你可能会忽略一个事实：你无法运行最新的 Puppeteer 版本，并且你还需要用一个标记禁用 shm ：

const browser = await puppeteer.launch({
      headless: true,
      args: ['--disable-dev-shm-usage']
    });
    复制代码

否则，Puppeteer 子进程可能会在正常启动之前耗尽内存。

## 方案 3 + 1：CSS 打印规则 ##

可能有人认为从开发人员的角度来看，简单地使用 CSS 打印规则很容易。没有 NPM 模块，只有纯 CSS。但是在跨浏览器兼容性方面，它的表现如何呢？

在选择 CSS 打印规则时，你必须在每个浏览器中测试结果，以确保它提供的布局是相同的，并且它不是100％能做到这一点。

例如，在给定元素后面插入一个 break-after 并不是一个多么高深的技术，但是你可能会惊讶的发现要[在 Firefox 中使用它][Firefox]需要使用变通方法。

除非你是一位经验丰富的 CSS 大师，在创建可打印页面方面有很多的经验，否则这可能会非常耗时。

如果你可以使打印样式表保持简单，打印规则是很好用的。

让我们来看一个例子吧。

@media print {
        .print-button {
            display: none;
        }
        
        .content div {
            break-after: always;
        }
    }
    复制代码

上面的 CSS 隐藏了打印按钮，并在每个 `div` 之后插入一个分页符，其中包含`content` 类。有一篇[很棒的文章][Link 3]总结了你可以用打印规则做什么，以及它们有什么问题，包括浏览器兼容性。

考虑到所有因素，如果你想从不那么复杂的页面生成 PDF，CSS打印规则非常有效。

## 总结 ##

让我们快速回顾前面介绍的方案，以便从 HTML 页面生成 PDF 文件：

*  **从 DOM 产生截图**：当你需要从页面创建快照时（例如创建缩略图）可能很有用，但是当你需要处理大量数据时就会有些捉襟见肘。
 *  **只用 PDF 库**：如果你打算从头开始以编程方式创建 PDF 文件，这是一个完美的解决方案。否则，你需要同时维护 HTML 和 PDF 模板，这绝对是一个禁忌。
 *  **Puppeteer**：尽管在 Docker 上工作相对困难，但它为我们的实现提供了最好的结果，而且编写代码也是最简单的。
 *  **CSS打印规则**：如果你的用户受过足够的教育，知道如何把页面内容打印到文件，并且你的页面相对简单，那么它可能是最轻松的解决方案。正如你在我们的案例中所看到的，事实并非如此。

虽然今天是愚人节，但是以上所有内容都是在真的！

### 愚人节快乐！ ###

欢迎关注京程一灯公众号：京程一灯，获取更多前端干货。

转载于:https://juejin.im/post/5ca1dc0251882543d569e075

[blog.risingstack.com_pdf-from-ht]: https://link.juejin.im?target=https%3A%2F%2Fblog.risingstack.com%2Fpdf-from-html-node-js-puppeteer%2F
[Html2canvas]: https://link.juejin.im?target=https%3A%2F%2Fhtml2canvas.hertzen.com%2F
[jsPdf]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FMrRio%2FjsPDF
[PDFKit]: https://link.juejin.im?target=https%3A%2F%2Fwww.npmjs.com%2Fpackage%2Fpdfkit
[Puppeteer]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FGoogleChrome%2Fpuppeteer
[API docs]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FGoogleChrome%2Fpuppeteer%2Fblob%2Fv1.11.0%2Fdocs%2Fapi.md
[Link 1]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FGoogleChrome%2Fpuppeteer%2Fblob%2Fmaster%2Fdocs%2Ftroubleshooting.md%23running-puppeteer-in-docker
[Link 2]: https://link.juejin.im?target=https%3A%2F%2Fgithub.com%2FGoogleChrome%2Fpuppeteer%2Fblob%2Fmaster%2Fdocs%2Ftroubleshooting.md%23running-on-alpine
[Firefox]: https://link.juejin.im?target=https%3A%2F%2Fdeveloper.mozilla.org%2Fen-US%2Fdocs%2FWeb%2FCSS%2Fbreak-after%23Browser_compatibility
[Link 3]: https://link.juejin.im?target=https%3A%2F%2Fwww.smashingmagazine.com%2F2018%2F05%2Fprint-stylesheets-in-2018%2F