@oceanbase/ui
Version:
The UI library based on OceanBase Design
171 lines (168 loc) • 7.36 kB
JavaScript
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
import { SearchOutlined } from '@oceanbase/icons';
import { Divider, Empty, Form, Input, Tree } from '@oceanbase/design';
import React, { forwardRef, useEffect, useImperativeHandle, useMemo, useRef, useState } from 'react';
import { deepFilter, findFromTree, getAllOptions, nodes2treeNodes, treeNode2node } from "./util";
// 用户传入的树节点数据结构
import { jsx as _jsx } from "react/jsx-runtime";
import { jsxs as _jsxs } from "react/jsx-runtime";
export default /*#__PURE__*/forwardRef(function (_ref, ref) {
var _ref$treeData = _ref.treeData,
treeData = _ref$treeData === void 0 ? [] : _ref$treeData,
width = _ref.width,
_ref$checkable = _ref.checkable,
checkable = _ref$checkable === void 0 ? true : _ref$checkable,
titleRender = _ref.titleRender,
onChange = _ref.onChange,
_ref$followLeaf = _ref.followLeaf,
followLeaf = _ref$followLeaf === void 0 ? true : _ref$followLeaf,
height = _ref.height,
defaultExpandAll = _ref.defaultExpandAll,
loadData = _ref.loadData,
_ref$searchStyle = _ref.searchStyle,
searchStyle = _ref$searchStyle === void 0 ? {} : _ref$searchStyle;
var memoTreeData = useMemo(function () {
return nodes2treeNodes(treeData);
}, [treeData]);
var _useState = useState(memoTreeData),
_useState2 = _slicedToArray(_useState, 2),
searchData = _useState2[0],
setSearchData = _useState2[1];
// 内部计算虚拟滚动的高度
var treeWrapperDOM = useRef(null);
var _useState3 = useState(height),
_useState4 = _slicedToArray(_useState3, 2),
innerHeight = _useState4[0],
setInnerHeight = _useState4[1];
var _useState5 = useState([]),
_useState6 = _slicedToArray(_useState5, 2),
checkedKeys = _useState6[0],
setCheckedKeys = _useState6[1];
var _useState7 = useState(''),
_useState8 = _slicedToArray(_useState7, 2),
searchStr = _useState8[0],
setSearchStr = _useState8[1];
var _Form$useForm = Form.useForm(),
_Form$useForm2 = _slicedToArray(_Form$useForm, 1),
form = _Form$useForm2[0];
useEffect(function () {
if (height === undefined && treeWrapperDOM !== null && treeWrapperDOM !== void 0 && treeWrapperDOM.current) {
var _treeWrapperDOM$curre;
var containerheight = treeWrapperDOM === null || treeWrapperDOM === void 0 || (_treeWrapperDOM$curre = treeWrapperDOM.current) === null || _treeWrapperDOM$curre === void 0 ? void 0 : _treeWrapperDOM$curre.clientHeight;
setInnerHeight(containerheight);
}
}, []);
useEffect(function () {
if (!searchData || searchData.length === 0) {
setSearchData(memoTreeData);
}
onSearch({
searchKey: searchStr
});
}, [memoTreeData]);
var reset = function reset() {
form.resetFields();
setCheckedKeys([]);
};
var checkAll = function checkAll() {
var allOptions = getAllOptions(memoTreeData);
setCheckedKeys(allOptions.map(function (node) {
return node.key;
}));
onChange(allOptions);
};
// 反选
var invertSelect = function invertSelect() {
var allOptions = getAllOptions(memoTreeData);
// 收集未选择的项，用于设置反选
var notCheckedOptions = allOptions.filter(function (item) {
return !checkedKeys.includes(item.key);
});
setCheckedKeys(notCheckedOptions.map(function (item) {
return item.key;
}));
onChange(allOptions.filter(function (node) {
return !checkedKeys.includes(node.key);
}));
};
useImperativeHandle(ref, function () {
return {
reset: reset,
checkAll: checkAll,
invertSelect: invertSelect
};
});
var onSearch = function onSearch(_ref2) {
var searchKey = _ref2.searchKey;
setSearchStr(searchKey);
// 过滤掉所有叶子节点不匹配的
var next = deepFilter(memoTreeData, function (node) {
var titleIsStr = typeof node.title === 'string';
return (titleIsStr ? node.title : node.value).includes(searchKey);
// (!isLeaf(node) && node.children.length > 0)
});
setSearchData(next);
};
var onCheck = function onCheck(keys) {
var treeNodes = keys.map(function (key) {
return findFromTree(key, memoTreeData);
});
if (followLeaf) {
treeNodes = treeNodes.filter(function (node) {
return !(node !== null && node !== void 0 && node.children);
});
}
setCheckedKeys(treeNodes.map(function (node) {
return node.key;
}));
onChange(treeNodes.map(treeNode2node));
};
// 整体宽度 - input - divider
var treeHeight = innerHeight ? innerHeight - (32 + 12 * 2 + 1) : 0;
var defaultSearchStyle = {
padding: '0 12px'
};
return /*#__PURE__*/_jsxs("div", {
ref: treeWrapperDOM,
style: {
width: width,
height: height !== null && height !== void 0 ? height : '100%'
},
children: [/*#__PURE__*/_jsx(Form, {
form: form,
onValuesChange: onSearch,
style: Object.assign(defaultSearchStyle, searchStyle),
children: /*#__PURE__*/_jsx(Form.Item, {
noStyle: true,
name: "searchKey",
children: /*#__PURE__*/_jsx(Input, {
"data-testid": "search",
prefix: /*#__PURE__*/_jsx(SearchOutlined, {})
})
})
}), /*#__PURE__*/_jsx(Divider, {
style: {
margin: '12px 0'
}
}), searchData.length !== 0 ? /*#__PURE__*/_jsx(Tree, {
height: treeHeight,
showIcon: true,
checkable: checkable,
blockNode: true,
selectable: false,
checkedKeys: checkedKeys,
onCheck: onCheck,
titleRender: titleRender,
defaultExpandAll: defaultExpandAll,
treeData: form.getFieldValue('searchKey') ? searchData : memoTreeData,
loadData: loadData
}) : /*#__PURE__*/_jsx(Empty, {
image: Empty.PRESENTED_IMAGE_SIMPLE
})]
});
});