antd
Version:
An enterprise-class UI design language and React components implementation
171 lines (141 loc) • 9.59 kB
JavaScript
function _typeof(obj) { "@babel/helpers - typeof"; if (typeof Symbol === "function" && typeof Symbol.iterator === "symbol") { _typeof = function _typeof(obj) { return typeof obj; }; } else { _typeof = function _typeof(obj) { return obj && typeof Symbol === "function" && obj.constructor === Symbol && obj !== Symbol.prototype ? "symbol" : typeof obj; }; } return _typeof(obj); }
function _extends() { _extends = Object.assign || function (target) { for (var i = 1; i < arguments.length; i++) { var source = arguments[i]; for (var key in source) { if (Object.prototype.hasOwnProperty.call(source, key)) { target[key] = source[key]; } } } return target; }; return _extends.apply(this, arguments); }
function _defineProperty(obj, key, value) { if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, descriptor.key, descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); return Constructor; }
function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); if (superClass) _setPrototypeOf(subClass, superClass); }
function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf || function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } return _assertThisInitialized(self); }
function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Date.prototype.toString.call(Reflect.construct(Date, [], function () {})); return true; } catch (e) { return false; } }
function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
import * as React from 'react';
import RcTreeSelect, { TreeNode, SHOW_ALL, SHOW_PARENT, SHOW_CHILD } from 'rc-tree-select';
import classNames from 'classnames';
import omit from 'omit.js';
import { ConfigConsumer } from '../config-provider';
import devWarning from '../_util/devWarning';
import getIcons from '../select/utils/iconUtil';
import renderSwitcherIcon from '../tree/utils/iconUtil';
import SizeContext from '../config-provider/SizeContext';
var TreeSelect = /*#__PURE__*/function (_React$Component) {
_inherits(TreeSelect, _React$Component);
var _super = _createSuper(TreeSelect);
function TreeSelect(props) {
var _this;
_classCallCheck(this, TreeSelect);
_this = _super.call(this, props);
_this.selectRef = /*#__PURE__*/React.createRef();
_this.renderTreeSelect = function (_ref) {
var getContextPopupContainer = _ref.getPopupContainer,
getPrefixCls = _ref.getPrefixCls,
renderEmpty = _ref.renderEmpty,
direction = _ref.direction,
virtual = _ref.virtual,
dropdownMatchSelectWidth = _ref.dropdownMatchSelectWidth;
var _this$props = _this.props,
customizePrefixCls = _this$props.prefixCls,
customizeSize = _this$props.size,
className = _this$props.className,
treeCheckable = _this$props.treeCheckable,
multiple = _this$props.multiple,
_this$props$listHeigh = _this$props.listHeight,
listHeight = _this$props$listHeigh === void 0 ? 256 : _this$props$listHeigh,
_this$props$listItemH = _this$props.listItemHeight,
listItemHeight = _this$props$listItemH === void 0 ? 26 : _this$props$listItemH,
notFoundContent = _this$props.notFoundContent,
_switcherIcon = _this$props.switcherIcon,
treeLine = _this$props.treeLine,
getPopupContainer = _this$props.getPopupContainer,
dropdownClassName = _this$props.dropdownClassName,
bordered = _this$props.bordered,
_this$props$treeIcon = _this$props.treeIcon,
treeIcon = _this$props$treeIcon === void 0 ? false : _this$props$treeIcon;
var prefixCls = getPrefixCls('select', customizePrefixCls);
var treePrefixCls = getPrefixCls('select-tree', customizePrefixCls);
var treeSelectPrefixCls = getPrefixCls('tree-select', customizePrefixCls);
var mergedDropdownClassName = classNames(dropdownClassName, "".concat(treeSelectPrefixCls, "-dropdown"), _defineProperty({}, "".concat(treeSelectPrefixCls, "-dropdown-rtl"), direction === 'rtl'));
var isMultiple = !!(treeCheckable || multiple); // ===================== Icons =====================
var _getIcons = getIcons(_extends(_extends({}, _this.props), {
multiple: isMultiple
})),
suffixIcon = _getIcons.suffixIcon,
itemIcon = _getIcons.itemIcon,
removeIcon = _getIcons.removeIcon,
clearIcon = _getIcons.clearIcon; // ===================== Empty =====================
var mergedNotFound;
if (notFoundContent !== undefined) {
mergedNotFound = notFoundContent;
} else {
mergedNotFound = renderEmpty('Select');
} // ==================== Render =====================
var selectProps = omit(_this.props, ['prefixCls', 'suffixIcon', 'itemIcon', 'removeIcon', 'clearIcon', 'switcherIcon', 'size', 'bordered']);
return /*#__PURE__*/React.createElement(SizeContext.Consumer, null, function (size) {
var _classNames2;
var mergedSize = customizeSize || size;
var mergedClassName = classNames(!customizePrefixCls && treeSelectPrefixCls, (_classNames2 = {}, _defineProperty(_classNames2, "".concat(prefixCls, "-lg"), mergedSize === 'large'), _defineProperty(_classNames2, "".concat(prefixCls, "-sm"), mergedSize === 'small'), _defineProperty(_classNames2, "".concat(prefixCls, "-rtl"), direction === 'rtl'), _defineProperty(_classNames2, "".concat(prefixCls, "-borderless"), !bordered), _classNames2), className);
return /*#__PURE__*/React.createElement(RcTreeSelect, _extends({
virtual: virtual,
dropdownMatchSelectWidth: dropdownMatchSelectWidth
}, selectProps, {
ref: _this.selectRef,
prefixCls: prefixCls,
className: mergedClassName,
listHeight: listHeight,
listItemHeight: listItemHeight,
treeCheckable: treeCheckable ? /*#__PURE__*/React.createElement("span", {
className: "".concat(prefixCls, "-tree-checkbox-inner")
}) : treeCheckable,
inputIcon: suffixIcon,
menuItemSelectedIcon: itemIcon,
removeIcon: removeIcon,
clearIcon: clearIcon,
switcherIcon: function switcherIcon(nodeProps) {
return renderSwitcherIcon(treePrefixCls, _switcherIcon, treeLine, nodeProps);
},
showTreeIcon: treeIcon,
notFoundContent: mergedNotFound,
getPopupContainer: getPopupContainer || getContextPopupContainer,
treeMotion: null,
dropdownClassName: mergedDropdownClassName
}));
});
};
devWarning(props.multiple !== false || !props.treeCheckable, 'TreeSelect', '`multiple` will alway be `true` when `treeCheckable` is true');
return _this;
}
_createClass(TreeSelect, [{
key: "focus",
value: function focus() {
if (this.selectRef.current) {
this.selectRef.current.focus();
}
}
}, {
key: "blur",
value: function blur() {
if (this.selectRef.current) {
this.selectRef.current.blur();
}
}
}, {
key: "render",
value: function render() {
return /*#__PURE__*/React.createElement(ConfigConsumer, null, this.renderTreeSelect);
}
}]);
return TreeSelect;
}(React.Component);
TreeSelect.TreeNode = TreeNode;
TreeSelect.SHOW_ALL = SHOW_ALL;
TreeSelect.SHOW_PARENT = SHOW_PARENT;
TreeSelect.SHOW_CHILD = SHOW_CHILD;
TreeSelect.defaultProps = {
transitionName: 'slide-up',
choiceTransitionName: 'zoom',
bordered: true
};
export { TreeNode };
export default TreeSelect;