zp-bee/lib/bee-tree-select/index.js

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

"use strict";

var _interopRequireWildcard = require("@babel/runtime/helpers/interopRequireWildcard");

var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");

Object.defineProperty(exports, "__esModule", {
  value: true
});
exports.default = void 0;

var _slicedToArray2 = _interopRequireDefault(require("@babel/runtime/helpers/slicedToArray"));

require("antd/es/tree-select/style");

var _treeSelect = _interopRequireDefault(require("antd/es/tree-select"));

var _react = _interopRequireWildcard(require("react"));

var __rest = void 0 && (void 0).__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;
};

var TreeNode = _treeSelect.default.TreeNode;
var InternalTreeSelect = /*#__PURE__*/(0, _react.forwardRef)(function (props, ref) {
  var label = props.label,
      className = props.className,
      value = props.value,
      onChange = props.onChange,
      others = __rest(props, ["label", "className", "value", "onChange"]);

  var _useState = (0, _react.useState)(value),
      _useState2 = (0, _slicedToArray2.default)(_useState, 2),
      selfValue = _useState2[0],
      setSelfValue = _useState2[1];

  var container = (0, _react.useRef)(null);
  var injectRef = (0, _react.useRef)(null);
  (0, _react.useImperativeHandle)(ref, function () {
    var _a, _b;

    return {
      handleChange: onSelfChange,
      onChange: onSelfChange,
      onFocus: (_a = injectRef.current) === null || _a === void 0 ? void 0 : _a.focus,
      onBlur: (_b = injectRef.current) === null || _b === void 0 ? void 0 : _b.blur
    };
  });
  (0, _react.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 + 16;
    dom.style.paddingLeft = "".concat(left, "px");
    input.style.textIndent = "".concat(left - 12, "px");
  }, []);
  (0, _react.useEffect)(function () {
    setSelfValue(value);
  }, [value]);
  var onSelfChange = (0, _react.useCallback)(function (value, label, extra) {
    onChange && onChange(value, label, extra);
    setSelfValue(value);
  }, []);
  return /*#__PURE__*/_react.default.createElement(_react.default.Fragment, null, /*#__PURE__*/_react.default.createElement(_treeSelect.default, Object.assign({
    ref: injectRef,
    className: className ? className : '',
    value: selfValue,
    onChange: onSelfChange
  }, others)), /*#__PURE__*/_react.default.createElement("noscript", {
    ref: container
  }));
});
var BeeTreeSelect = InternalTreeSelect;
BeeTreeSelect.TreeNode = TreeNode;
var _default = BeeTreeSelect;
exports.default = _default;