前端页面添加水印

最近做的项目里都需要给页面设置水印，npm里有一些比较成熟的库可以直接用，但其实如果是简单的水印添加，自己写个工具函数也可以实现。

前端水印的实现原理主要是利用canvas标签的一些属性，实现的思路：

• 创建canvas元素
• 给canvas绘制文本，填充样式旋转角度、文字字体颜色等
• 将绘制好的canvas元素转换成base64格式的图片
• 动态创建一个div标签，相对位置属性为fixed，宽高设置为屏幕的可见大小，将上一步canvas转换的图片设置为div的背景图片并平铺满整个内部
• 最后将动态创建的div添加到body中

下面按照思路实现代码：

首先是设置水印的方法，可以传入水印的文案，这个传入的参数可以将根据实际需要进行优化，比如可以传入一个数组设置多个文案，还有水印位置大小等

/**  水印添加方法  */

const setWatermark = (str1, str2) => {
    const id = '1.23452384164.123412415'

    if (document.getElementById(id) !== null) {
      document.body.removeChild(document.getElementById(id))
    }

    const can = document.createElement('canvas')
    // 设置canvas画布大小
    can.width = 240
    can.height = 140

    const cans = can.getContext('2d')
    cans.rotate(-25 * Math.PI / 180) // 水印旋转角度
    cans.font = '1.2rem Vedana'
    cans.fillStyle = '#666'
    cans.textAlign = 'center'
    cans.textBaseline = 'Middle'
    cans.fillText(str1, can.width / 2, can.height) // 水印在画布的位置x，y轴
    cans.fillText(str2, can.width / 2, can.height + 25)

    const div = document.createElement('div')
    div.id = id
    div.style.pointerEvents = 'none'
    div.style.top = '20px'
    div.style.left = '0px'
    div.style.opacity = '0.06'
    div.style.position = 'fixed'
    div.style.zIndex = '100000'
    div.style.width = document.documentElement.clientWidth + 'px'
    div.style.height = document.documentElement.clientHeight + 'px'
    div.style.background = 'url(' + can.toDataURL('image/png') + ') left top repeat'
    document.body.appendChild(div)
    return id
  }

再来两个方法用来添加水印和移除水印

// 添加水印方法
  export const setWaterMark = (str1, str2) => {
    let id = setWatermark(str1, str2)
    if (document.getElementById(id) === null) {
      id = setWatermark(str1, str2)
    }
  }

  // 移除水印方法
  export const removeWatermark = () => {
    const id = '1.23452384164.123412415'
    if (document.getElementById(id) !== null) {
      document.body.removeChild(document.getElementById(id))
    }
  }

看看效果：

这个水印的实现其实很简单，主要就是对canvas标签的属性操作。我自己将这个实现方式编写发布了一个npm包，可自定义水印参数，这个包很小基本上能满足大部分的水印设置需求，有需要的可以下载引用，包名：watermark-package