acud/lib/watermark/useWatermark.js

acg react 组件库

"use strict";

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
var _typeof = require("@babel/runtime/helpers/typeof");
Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.FontGap = exports.BaseSize = void 0;
exports["default"] = useWatermark;
var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));
var React = _interopRequireWildcard(require("react"));
var _utils = require("./utils");
function _interopRequireWildcard(e, t) { if ("function" == typeof WeakMap) var r = new WeakMap(), n = new WeakMap(); return (_interopRequireWildcard = function _interopRequireWildcard(e, t) { if (!t && e && e.__esModule) return e; var o, i, f = { __proto__: null, "default": e }; if (null === e || "object" != _typeof(e) && "function" != typeof e) return f; if (o = t ? n : r) { if (o.has(e)) return o.get(e); o.set(e, f); } for (var _t in e) "default" !== _t && {}.hasOwnProperty.call(e, _t) && ((i = (o = Object.defineProperty) && Object.getOwnPropertyDescriptor(e, _t)) && (i.get || i.set) ? o(f, _t, i) : f[_t] = e[_t]); return f; })(e, t); }
/**
 * Base size of the canvas, 1 for parallel layout and 2 for alternate layout
 * Only alternate layout is currently supported
 */
var BaseSize = exports.BaseSize = 2;
var FontGap = exports.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（节点）。
 */
function useWatermark(markStyle) {
  var _React$useState = React.useState(function () {
      return new Map();
    }),
    _React$useState2 = (0, _slicedToArray2["default"])(_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', (0, _utils.getStyleStr)((0, _extends2["default"])((0, _extends2["default"])((0, _extends2["default"])({}, 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];
}