JS前端基于canvas给图片添加水印

前两天给个人网站添加了一个小功能，就是在文章编辑上传图片的时候自动给图片加上水印。给网页图片添加水印是个常见的功能，也是互联网内容作者保护自己版权的方法之一。本文简单记录一下借助canvas在前端实现图片添加水印的实现方法。
canvas元素其实就是一个画布，我们可以很方便地绘制一些文字、线条、图形等，它也可以将一个img标签里渲染的图片画在画布上。

在上传文件到后端的时候，使用input标签读取用户本地文件后得到的其实是一个Blob对象（更精确的说是File对象，特殊的Blob对象）；而在页面上展示一个图片使用的是img标签；绘制功能用canvas实现。添加水印的功能需要在img标签、canvas画布、Blob对象这三者之间相互转换，通过一些API可以完成这个工作：

我们可以从本地读取图片Blob，然后渲染到img标签，使用canvas绘制img内容并且绘制水印内容到画布，再将canvas内容转为Blob对象上传服务器，这样就完整实现了图片+水印的功能。

一、本地读取图像文件渲染到img标签

本地读取图片文件将会得到一个Blob对象，我们可以借助FileReader.readAsDataURL方法读取Blob的内容，并得到一个Base64编码的文件内容，可以将该内容赋值给img.src从而在浏览器上渲染出本地的图像。当然，img并非必须渲染到DOM树。读取操作是个异步操作，读取完成会触发load事件，为了便于之后的调用，我们可以用一个Promise包装这个操作，最后返回一个Promise对象。

function blobToImg (blob) {
 return new Promise((resolve, reject) => {
  let reader = new FileReader()
  reader.addEventListener('load', () => {
   let img = new Image()
   img.src = reader.result
   img.addEventListener('load', () => resolve(img))
  })
  reader.readAsDataURL(blob)
 })
}

二、将img标签内容绘制到canvas画布

调用canvas元素画布上下文对象的drawImage方法即可实现将img内容绘制到画布。

function imgToCanvas (img) {
 let canvas = document.createElement('canvas')
 canvas.width = img.width
 canvas.height = img.height
 let ctx = canvas.getContext('2d')
 ctx.drawImage(img, 0, 0)
 return canvas
}

drawImage这个方法可以传入多个参数，以定义绘制的图像的范围，这里传入的0, 0定义从图像左上角开始绘制，后面可以继续传入两个参数来定义图像的绘制终点，不过这里整个图片都要绘制到canvas，所以采用默认值即可。

三、canvas画布上绘制水印并转换为Blob对象

在图片上传的时候，我们通常采用FormData，图片文件以一个Blob对象的形式放到FormData中，所以我们需要把canvas再转为Blob以便文件上传等操作。利用HTMLCanvasElement.toBlob方法：

function watermark (canvas, text) {
 return new Promise((resolve, reject) => {
  let ctx = canvas.getContext('2d')
  // 设置填充字号和字体，样式
  ctx.font = "24px 宋体"
  ctx.fillStyle = "#FFC82C"
  // 设置右对齐
  ctx.textAlign = 'right'
  // 在指定位置绘制文字，这里指定距离右下角20坐标的地方
  ctx.fillText(text, canvas.width - 20, canvas.height - 20)
  canvas.toBlob(blob => resolve(blob))
 })
}

四、图片添加水印完整接口

将以上三个步骤结合起来，就完整地实现了从图片添加水印，下面是一个简单的使用示例：从本地选择一个图片文件，然后添加水印后，在传入的dom元素下预览添加水印后的图片。

function imgWatermark (dom, text) {
 let input = document.createElement('input')
 input.setAttribute('type', 'file')
 input.setAttribute('accept', 'image/*')
 input.onchange = async () => {
  let img = await blobToImg(input.files[0])
  let canvas = imgToCanvas(img)
  let blob = await watermark(canvas, text)
  // 此处将Blob读取到img标签，并在dom内渲染出来；如果是上传文件，可以将blob添加到FormData中
  let newImage = await blobToImg(blob)
  dom.appendChild(newImage)
 }
 input.click()
}

给页面加一个id为container的div元素，然后如下调用：

let dom = document.querySelector('#container')
imgWatermark(dom, '水印文字')

这样就完整地给图片添加了水印效果，下面看一下实际效果，你也可以在线体验。

添加水印前：

添加水印后（水印内容：“腾冲?清凉山”）：

五、总结

本文仅仅介绍了图像+水印文字的简单实现，但是涉及的一些接口其实很有用。比如有时候遇到的一个功能是头像上传的预览和剪裁，这时候你可以利用FileReader来读取文件内容预览，利用CanvasRenderingContext2D.drawImage来实现剪裁功能。关于本文涉及接口的更多详细用法，可以参照MDN文档，文章中的API都使用了链接的形式链接到了MDN。