acud/es/watermark/useWatermark.js

acg react 组件库

import _extends from "@babel/runtime/helpers/esm/extends";
import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
import * as React from 'react';
import { getStyleStr } from './utils';
/**
 * Base size of the canvas, 1 for parallel layout and 2 for alternate layout
 * Only alternate layout is currently supported
 */
export var BaseSize = 2;
export var FontGap = 3;
// Prevent external hidden elements from adding accent styles
var EmphasizedStyles = {
  visibility: 'visible !important'
};
/**
 * 获取水印组件的钩子函数，用于在组件中添加或移除水印。
 *
 * @param markStyle 给水印元素设置的样式。
 * @returns 返回一个包含三个函数的数组：appendWatermark、removeWatermark和isWatermarkEle。
 * - appendWatermark 函数用于在指定容器中添加水印元素。参数为 base64Url（水印图片的 base64 编码字符串），markWidth（水印宽度）和 container（容器）。
 * - removeWatermark 函数用于从指定容器中删除水印元素。参数为 container（容器）。
 * - isWatermarkEle 函数用于判断某个节点是否为水印元素，并返回布尔值。参数为 ele（节点）。
 */
export default function useWatermark(markStyle) {
  var _React$useState = React.useState(function () {
      return new Map();
    }),
    _React$useState2 = _slicedToArray(_React$useState, 1),
    watermarkMap = _React$useState2[0];
  var appendWatermark = function appendWatermark(base64Url, markWidth, container) {
    if (container) {
      if (!watermarkMap.get(container)) {
        var newWatermarkEle = document.createElement('div');
        watermarkMap.set(container, newWatermarkEle);
      }
      var watermarkEle = watermarkMap.get(container);
      watermarkEle.setAttribute('style', getStyleStr(_extends(_extends(_extends({}, markStyle), {
        backgroundImage: "url('".concat(base64Url, "')"),
        backgroundSize: "".concat(Math.floor(markWidth), "px"),
        // @ts-ignore
        '-webkitPrintColorAdjust': 'exact',
        printColorAdjust: 'exact'
      }), EmphasizedStyles)));
      // Prevents using the browser `Hide Element` to hide watermarks
      watermarkEle.removeAttribute('class');
      if (watermarkEle.parentElement !== container) {
        container.append(watermarkEle);
      }
    }
    return watermarkMap.get(container);
  };
  var removeWatermark = function removeWatermark(container) {
    var watermarkEle = watermarkMap.get(container);
    if (watermarkEle && container) {
      container.removeChild(watermarkEle);
    }
    watermarkMap["delete"](container);
  };
  var isWatermarkEle = function isWatermarkEle(ele) {
    return Array.from(watermarkMap.values()).includes(ele);
  };
  return [appendWatermark, removeWatermark, isWatermarkEle];
}