zp-bee/es/bee-tree-multi-select/index.js

zp-bee，是一款基于 Dumi，由 React + TypeScript 开发的组件库 🎉。

import _slicedToArray from "@babel/runtime/helpers/esm/slicedToArray";
import "antd/es/tree-select/style";
import _TreeSelect from "antd/es/tree-select";

var __rest = this && this.__rest || function (s, e) {
  var t = {};

  for (var p in s) {
    if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  }

  if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
    if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  }
  return t;
};

import React, { useCallback, useEffect, useRef, useState, forwardRef, useImperativeHandle } from 'react';
import { createPortal } from 'react-dom';
import { getPopContainer } from '../_utils';
var TreeNode = _TreeSelect.TreeNode;
var InternalTreeMultiSelect = /*#__PURE__*/forwardRef(function (props, ref) {
  var label = props.label,
      className = props.className,
      injectLabel = props.injectLabel,
      onChange = props.onChange,
      others = __rest(props, ["label", "className", "injectLabel", "onChange"]);

  var _useState = useState(1),
      _useState2 = _slicedToArray(_useState, 2),
      maxTagCount = _useState2[0],
      setMaxTagCount = _useState2[1];

  var _useState3 = useState(false),
      _useState4 = _slicedToArray(_useState3, 2),
      showLabelPop = _useState4[0],
      setShowLabelPop = _useState4[1];

  var container = useRef(null);
  var selectRef = useRef({});
  var injectRef = useRef(null);
  var positionRef = useRef({});
  useEffect(function () {
    var _a;

    var parent = (_a = container.current) === null || _a === void 0 ? void 0 : _a.previousSibling;
    var dom = parent.querySelector('.ant-select-selector');
    var input = dom.querySelector('.ant-select-selection-search-input');
    var span = document.createElement('span');
    span.innerHTML = label;
    span.classList.add('bee-label');
    dom.prepend(span);
    var left = span.getBoundingClientRect().width + 10;
    dom.style.paddingLeft = "".concat(left, "px");
    input.style.textIndent = "".concat(left - 10, "px");
    positionRef.current = dom.getBoundingClientRect();
  }, []);
  useImperativeHandle(ref, function () {
    return {
      handleChange: onSelfChange,
      onChange: onSelfChange,
      onFocus: onFocus,
      onBlur: onBlur
    };
  });
  var onFocus = useCallback(function () {
    setMaxTagCount(undefined);
    setShowLabelPop(false);
  }, []);
  var onBlur = useCallback(function () {
    setMaxTagCount(1);
  }, []);
  var onMouseEnter = useCallback(function () {
    setShowLabelPop(true);
  }, []);
  var onMouseLeave = useCallback(function () {
    setShowLabelPop(false);
  }, []);
  var onSelfChange = useCallback(function (value, label, extra) {
    onChange && onChange(value, label, extra);
    selectRef.current = label;
  }, []);
  var _positionRef$current = positionRef.current,
      top = _positionRef$current.top,
      left = _positionRef$current.left,
      width = _positionRef$current.width;
  return /*#__PURE__*/React.createElement(React.Fragment, null, /*#__PURE__*/React.createElement(_TreeSelect, Object.assign({
    ref: injectRef,
    multiple: true,
    className: className ? className : '',
    maxTagCount: maxTagCount,
    onFocus: onFocus,
    onBlur: onBlur,
    onMouseEnter: onMouseEnter,
    onMouseLeave: onMouseLeave,
    onChange: onSelfChange
  }, others)), /*#__PURE__*/React.createElement("noscript", {
    ref: container
  }), showLabelPop && selectRef.current && selectRef.current.length > 0 && /*#__PURE__*/createPortal( /*#__PURE__*/React.createElement("div", {
    className: "bee-hover",
    style: {
      top: top + 36,
      left: left + 40,
      width: width
    }
  }, selectRef.current.map(function (item) {
    return /*#__PURE__*/React.createElement("span", {
      className: "bee-span"
    }, item, ";");
  })), getPopContainer()));
});
var BeeTreeMultiSelect = InternalTreeMultiSelect;
BeeTreeMultiSelect.MultiTreeNode = TreeNode;
export default BeeTreeMultiSelect;