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

背景

项目中有用到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/app
WORKDIR /usr/src/app

COPY ./node_modules /usr/src/app/node_modules

RUN npm set registry https://registry.npm.taobao.org/

# 下载安装PDFTK
RUN apt-get update
RUN apt-get install -y pdftk

# 使用PM2启动应用
CMD ["./node_modules/pm2/bin/pm2-docker", "process.yml"]

3、效果

源文件