node.js实现为PDF添加水印的示例代码

2019-11-19 12:24:26

字体：大中小

来源：转载

供稿：网友

背景

项目中有用到node做中间层，期间有个为PDF文件添加水印的需求，网上搜了不少帖子，发现有以下几点存在的问题。

1、单页PDF的话，可以用PNG图片加到PDF上，但是实际使用情况就没那么理想，必须满足多页PDF每页都加上水印的效果方可。

2、中间想过，把PDF转成图片，图片加上水印后再转回PDF，但是感觉中间步骤太麻烦，而且会面临清晰度下降的可能。

3、Python、Java有相关方案，可以实现，有想过用node去执行Python脚本去为PDF加水印，考虑到开发人员都是前端、如果以后需要维护，扩展可能就不是那么友好了。

解决方案

逛了Github和NPM，发现node-pdftk可以满足需求，不过得依赖PDFTK这个软件，进一步了解，发现PDFTK支持linux，好，直接用这个开搞。

环境

项目部署是用了docker，那就用docker来吧，比较贴合情况也省事点；如果平时在自己电脑上开发，但需要linux环境的话练手或者模拟服务器环境的话，大家也可以用一个linux镜像运行一个dokcer容器，这里就不赘述了。

实现

1、app.js

API相对较为简单，input方法中我们输入源文件，stamp方法中我们输入水印文件，out方法中指定保存的路径，这里的话我们起了一个服务，当访问3100端口的时候就去生成我们带水印的PDF。

'use strict';const express = require('express');const pdftk = require('node-pdftk');const path = require('path');const app = express();app.get('/', (req, res) => {  pdftk    .input(path.resolve(__dirname + '/../files/javascript.pdf'))    .stamp(path.resolve(__dirname + '/../files/watermark.pdf'))    .output(path.resolve(__dirname + '/../files/javascript_watermark.pdf'))    .then(buffer => {      return res.send('successful');    })    .catch(err => {      console.error(err);      return res.send('failed');    });});app.listen(3100);

2、dockerfile

FROM node:7.6.0# 创建工作目录RUN mkdir -p /usr/src/appWORKDIR /usr/src/appCOPY ./node_modules /usr/src/app/node_modulesRUN npm set registry https://registry.npm.taobao.org/# 下载安装PDFTKRUN apt-get updateRUN apt-get install -y pdftk# 使用PM2启动应用CMD ["./node_modules/pm2/bin/pm2-docker", "process.yml"]

3、效果

源文件