DirectoryTree.js 11 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. Object.defineProperty(exports, "__esModule", {
  4. value: true
  5. });
  6. exports.directoryTreeProps = exports.default = void 0;
  7. var _vue = require("vue");
  8. var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
  9. var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
  10. var _debounce = _interopRequireDefault(require("lodash/debounce"));
  11. var _FolderOpenOutlined = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/FolderOpenOutlined"));
  12. var _FolderOutlined = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/FolderOutlined"));
  13. var _FileOutlined = _interopRequireDefault(require("@ant-design/icons-vue/lib/icons/FileOutlined"));
  14. var _classNames = _interopRequireDefault(require("../_util/classNames"));
  15. var _Tree = _interopRequireWildcard(require("./Tree"));
  16. var _initDefaultProps = _interopRequireDefault(require("../_util/props-util/initDefaultProps"));
  17. var _treeUtil = require("../vc-tree/utils/treeUtil");
  18. var _util = require("../vc-tree/util");
  19. var _dictUtil = require("./utils/dictUtil");
  20. var _useConfigInject = _interopRequireDefault(require("../config-provider/hooks/useConfigInject"));
  21. var _propsUtil = require("../_util/props-util");
  22. var _type = require("../_util/type");
  23. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function (e) { return e ? t : r; })(e); }
  24. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != typeof e && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  25. var __rest = void 0 && (void 0).__rest || function (s, e) {
  26. var t = {};
  27. for (var p in s) if (Object.prototype.hasOwnProperty.call(s, p) && e.indexOf(p) < 0) t[p] = s[p];
  28. if (s != null && typeof Object.getOwnPropertySymbols === "function") for (var i = 0, p = Object.getOwnPropertySymbols(s); i < p.length; i++) {
  29. if (e.indexOf(p[i]) < 0 && Object.prototype.propertyIsEnumerable.call(s, p[i])) t[p[i]] = s[p[i]];
  30. }
  31. return t;
  32. };
  33. const directoryTreeProps = () => (0, _extends2.default)((0, _extends2.default)({}, (0, _Tree.treeProps)()), {
  34. expandAction: (0, _type.someType)([Boolean, String])
  35. });
  36. exports.directoryTreeProps = directoryTreeProps;
  37. function getIcon(props) {
  38. const {
  39. isLeaf,
  40. expanded
  41. } = props;
  42. if (isLeaf) {
  43. return (0, _vue.createVNode)(_FileOutlined.default, null, null);
  44. }
  45. return expanded ? (0, _vue.createVNode)(_FolderOpenOutlined.default, null, null) : (0, _vue.createVNode)(_FolderOutlined.default, null, null);
  46. }
  47. var _default = exports.default = (0, _vue.defineComponent)({
  48. compatConfig: {
  49. MODE: 3
  50. },
  51. name: 'ADirectoryTree',
  52. inheritAttrs: false,
  53. props: (0, _initDefaultProps.default)(directoryTreeProps(), {
  54. showIcon: true,
  55. expandAction: 'click'
  56. }),
  57. slots: Object,
  58. // emits: [
  59. // 'update:selectedKeys',
  60. // 'update:checkedKeys',
  61. // 'update:expandedKeys',
  62. // 'expand',
  63. // 'select',
  64. // 'check',
  65. // 'doubleclick',
  66. // 'dblclick',
  67. // 'click',
  68. // ],
  69. setup(props, _ref) {
  70. let {
  71. attrs,
  72. slots,
  73. emit,
  74. expose
  75. } = _ref;
  76. var _a;
  77. // convertTreeToData 兼容 a-tree-node 历史写法,未来a-tree-node移除后,删除相关代码,不要再render中调用 treeData,否则死循环
  78. const treeData = (0, _vue.ref)(props.treeData || (0, _treeUtil.convertTreeToData)((0, _propsUtil.filterEmpty)((_a = slots.default) === null || _a === void 0 ? void 0 : _a.call(slots))));
  79. (0, _vue.watch)(() => props.treeData, () => {
  80. treeData.value = props.treeData;
  81. });
  82. (0, _vue.onUpdated)(() => {
  83. (0, _vue.nextTick)(() => {
  84. var _a;
  85. if (props.treeData === undefined && slots.default) {
  86. treeData.value = (0, _treeUtil.convertTreeToData)((0, _propsUtil.filterEmpty)((_a = slots.default) === null || _a === void 0 ? void 0 : _a.call(slots)));
  87. }
  88. });
  89. });
  90. // Shift click usage
  91. const lastSelectedKey = (0, _vue.ref)();
  92. const cachedSelectedKeys = (0, _vue.ref)();
  93. const fieldNames = (0, _vue.computed)(() => (0, _treeUtil.fillFieldNames)(props.fieldNames));
  94. const treeRef = (0, _vue.ref)();
  95. const scrollTo = scroll => {
  96. var _a;
  97. (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.scrollTo(scroll);
  98. };
  99. expose({
  100. scrollTo,
  101. selectedKeys: (0, _vue.computed)(() => {
  102. var _a;
  103. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.selectedKeys;
  104. }),
  105. checkedKeys: (0, _vue.computed)(() => {
  106. var _a;
  107. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.checkedKeys;
  108. }),
  109. halfCheckedKeys: (0, _vue.computed)(() => {
  110. var _a;
  111. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.halfCheckedKeys;
  112. }),
  113. loadedKeys: (0, _vue.computed)(() => {
  114. var _a;
  115. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.loadedKeys;
  116. }),
  117. loadingKeys: (0, _vue.computed)(() => {
  118. var _a;
  119. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.loadingKeys;
  120. }),
  121. expandedKeys: (0, _vue.computed)(() => {
  122. var _a;
  123. return (_a = treeRef.value) === null || _a === void 0 ? void 0 : _a.expandedKeys;
  124. })
  125. });
  126. const getInitExpandedKeys = () => {
  127. const {
  128. keyEntities
  129. } = (0, _treeUtil.convertDataToEntities)(treeData.value, {
  130. fieldNames: fieldNames.value
  131. });
  132. let initExpandedKeys;
  133. // Expanded keys
  134. if (props.defaultExpandAll) {
  135. initExpandedKeys = Object.keys(keyEntities);
  136. } else if (props.defaultExpandParent) {
  137. initExpandedKeys = (0, _util.conductExpandParent)(props.expandedKeys || props.defaultExpandedKeys || [], keyEntities);
  138. } else {
  139. initExpandedKeys = props.expandedKeys || props.defaultExpandedKeys;
  140. }
  141. return initExpandedKeys;
  142. };
  143. const selectedKeys = (0, _vue.ref)(props.selectedKeys || props.defaultSelectedKeys || []);
  144. const expandedKeys = (0, _vue.ref)(getInitExpandedKeys());
  145. (0, _vue.watch)(() => props.selectedKeys, () => {
  146. if (props.selectedKeys !== undefined) {
  147. selectedKeys.value = props.selectedKeys;
  148. }
  149. }, {
  150. immediate: true
  151. });
  152. (0, _vue.watch)(() => props.expandedKeys, () => {
  153. if (props.expandedKeys !== undefined) {
  154. expandedKeys.value = props.expandedKeys;
  155. }
  156. }, {
  157. immediate: true
  158. });
  159. const expandFolderNode = (event, node) => {
  160. const {
  161. isLeaf
  162. } = node;
  163. if (isLeaf || event.shiftKey || event.metaKey || event.ctrlKey) {
  164. return;
  165. }
  166. // Call internal rc-tree expand function
  167. // https://github.com/ant-design/ant-design/issues/12567
  168. treeRef.value.onNodeExpand(event, node);
  169. };
  170. const onDebounceExpand = (0, _debounce.default)(expandFolderNode, 200, {
  171. leading: true
  172. });
  173. const onExpand = (keys, info) => {
  174. if (props.expandedKeys === undefined) {
  175. expandedKeys.value = keys;
  176. }
  177. // Call origin function
  178. emit('update:expandedKeys', keys);
  179. emit('expand', keys, info);
  180. };
  181. const onClick = (event, node) => {
  182. const {
  183. expandAction
  184. } = props;
  185. // Expand the tree
  186. if (expandAction === 'click') {
  187. onDebounceExpand(event, node);
  188. }
  189. emit('click', event, node);
  190. };
  191. const onDoubleClick = (event, node) => {
  192. const {
  193. expandAction
  194. } = props;
  195. // Expand the tree
  196. if (expandAction === 'dblclick' || expandAction === 'doubleclick') {
  197. onDebounceExpand(event, node);
  198. }
  199. emit('doubleclick', event, node);
  200. emit('dblclick', event, node);
  201. };
  202. const onSelect = (keys, event) => {
  203. const {
  204. multiple
  205. } = props;
  206. const {
  207. node,
  208. nativeEvent
  209. } = event;
  210. const key = node[fieldNames.value.key];
  211. // const newState: DirectoryTreeState = {};
  212. // We need wrap this event since some value is not same
  213. const newEvent = (0, _extends2.default)((0, _extends2.default)({}, event), {
  214. selected: true
  215. });
  216. // Windows / Mac single pick
  217. const ctrlPick = (nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.ctrlKey) || (nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.metaKey);
  218. const shiftPick = nativeEvent === null || nativeEvent === void 0 ? void 0 : nativeEvent.shiftKey;
  219. // Generate new selected keys
  220. let newSelectedKeys;
  221. if (multiple && ctrlPick) {
  222. // Control click
  223. newSelectedKeys = keys;
  224. lastSelectedKey.value = key;
  225. cachedSelectedKeys.value = newSelectedKeys;
  226. newEvent.selectedNodes = (0, _dictUtil.convertDirectoryKeysToNodes)(treeData.value, newSelectedKeys, fieldNames.value);
  227. } else if (multiple && shiftPick) {
  228. // Shift click
  229. newSelectedKeys = Array.from(new Set([...(cachedSelectedKeys.value || []), ...(0, _dictUtil.calcRangeKeys)({
  230. treeData: treeData.value,
  231. expandedKeys: expandedKeys.value,
  232. startKey: key,
  233. endKey: lastSelectedKey.value,
  234. fieldNames: fieldNames.value
  235. })]));
  236. newEvent.selectedNodes = (0, _dictUtil.convertDirectoryKeysToNodes)(treeData.value, newSelectedKeys, fieldNames.value);
  237. } else {
  238. // Single click
  239. newSelectedKeys = [key];
  240. lastSelectedKey.value = key;
  241. cachedSelectedKeys.value = newSelectedKeys;
  242. newEvent.selectedNodes = (0, _dictUtil.convertDirectoryKeysToNodes)(treeData.value, newSelectedKeys, fieldNames.value);
  243. }
  244. emit('update:selectedKeys', newSelectedKeys);
  245. emit('select', newSelectedKeys, newEvent);
  246. if (props.selectedKeys === undefined) {
  247. selectedKeys.value = newSelectedKeys;
  248. }
  249. };
  250. const onCheck = (checkedObjOrKeys, eventObj) => {
  251. emit('update:checkedKeys', checkedObjOrKeys);
  252. emit('check', checkedObjOrKeys, eventObj);
  253. };
  254. const {
  255. prefixCls,
  256. direction
  257. } = (0, _useConfigInject.default)('tree', props);
  258. return () => {
  259. const connectClassName = (0, _classNames.default)(`${prefixCls.value}-directory`, {
  260. [`${prefixCls.value}-directory-rtl`]: direction.value === 'rtl'
  261. }, attrs.class);
  262. const {
  263. icon = slots.icon,
  264. blockNode = true
  265. } = props,
  266. otherProps = __rest(props, ["icon", "blockNode"]);
  267. return (0, _vue.createVNode)(_Tree.default, (0, _objectSpread2.default)((0, _objectSpread2.default)((0, _objectSpread2.default)({}, attrs), {}, {
  268. "icon": icon || getIcon,
  269. "ref": treeRef,
  270. "blockNode": blockNode
  271. }, otherProps), {}, {
  272. "prefixCls": prefixCls.value,
  273. "class": connectClassName,
  274. "expandedKeys": expandedKeys.value,
  275. "selectedKeys": selectedKeys.value,
  276. "onSelect": onSelect,
  277. "onClick": onClick,
  278. "onDblclick": onDoubleClick,
  279. "onExpand": onExpand,
  280. "onCheck": onCheck
  281. }), slots);
  282. };
  283. }
  284. });