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

背景

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

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

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

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

解决方案

逛了github和npm，发现可以满足需求，不过得依赖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、效果

源文件

水印文件

添加水印后

4、地址

github仓库地址

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。