@mui/x-tree-view 6.0.0-alpha.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (76) hide show
  1. package/CHANGELOG.md +3682 -0
  2. package/LICENSE +21 -0
  3. package/README.md +34 -0
  4. package/TreeItem/TreeItem.d.ts +13 -0
  5. package/TreeItem/TreeItem.js +392 -0
  6. package/TreeItem/TreeItem.types.d.ts +84 -0
  7. package/TreeItem/TreeItem.types.js +1 -0
  8. package/TreeItem/TreeItemContent.d.ts +52 -0
  9. package/TreeItem/TreeItemContent.js +101 -0
  10. package/TreeItem/index.d.ts +5 -0
  11. package/TreeItem/index.js +4 -0
  12. package/TreeItem/package.json +6 -0
  13. package/TreeItem/treeItemClasses.d.ts +23 -0
  14. package/TreeItem/treeItemClasses.js +6 -0
  15. package/TreeItem/useTreeItem.d.ts +10 -0
  16. package/TreeItem/useTreeItem.js +68 -0
  17. package/TreeView/TreeView.d.ts +14 -0
  18. package/TreeView/TreeView.js +854 -0
  19. package/TreeView/TreeView.types.d.ts +170 -0
  20. package/TreeView/TreeView.types.js +1 -0
  21. package/TreeView/TreeViewContext.d.ts +6 -0
  22. package/TreeView/TreeViewContext.js +31 -0
  23. package/TreeView/descendants.d.ts +42 -0
  24. package/TreeView/descendants.js +186 -0
  25. package/TreeView/index.d.ts +3 -0
  26. package/TreeView/index.js +3 -0
  27. package/TreeView/package.json +6 -0
  28. package/TreeView/treeViewClasses.d.ts +7 -0
  29. package/TreeView/treeViewClasses.js +6 -0
  30. package/index.d.ts +2 -0
  31. package/index.js +9 -0
  32. package/internals/models.d.ts +1 -0
  33. package/internals/models.js +1 -0
  34. package/legacy/TreeItem/TreeItem.js +391 -0
  35. package/legacy/TreeItem/TreeItem.types.js +1 -0
  36. package/legacy/TreeItem/TreeItemContent.js +98 -0
  37. package/legacy/TreeItem/index.js +4 -0
  38. package/legacy/TreeItem/treeItemClasses.js +6 -0
  39. package/legacy/TreeItem/useTreeItem.js +67 -0
  40. package/legacy/TreeView/TreeView.js +907 -0
  41. package/legacy/TreeView/TreeView.types.js +1 -0
  42. package/legacy/TreeView/TreeViewContext.js +41 -0
  43. package/legacy/TreeView/descendants.js +199 -0
  44. package/legacy/TreeView/index.js +3 -0
  45. package/legacy/TreeView/treeViewClasses.js +6 -0
  46. package/legacy/index.js +9 -0
  47. package/legacy/internals/models.js +1 -0
  48. package/modern/TreeItem/TreeItem.js +391 -0
  49. package/modern/TreeItem/TreeItem.types.js +1 -0
  50. package/modern/TreeItem/TreeItemContent.js +101 -0
  51. package/modern/TreeItem/index.js +4 -0
  52. package/modern/TreeItem/treeItemClasses.js +6 -0
  53. package/modern/TreeItem/useTreeItem.js +68 -0
  54. package/modern/TreeView/TreeView.js +853 -0
  55. package/modern/TreeView/TreeView.types.js +1 -0
  56. package/modern/TreeView/TreeViewContext.js +31 -0
  57. package/modern/TreeView/descendants.js +186 -0
  58. package/modern/TreeView/index.js +3 -0
  59. package/modern/TreeView/treeViewClasses.js +6 -0
  60. package/modern/index.js +9 -0
  61. package/modern/internals/models.js +1 -0
  62. package/node/TreeItem/TreeItem.js +400 -0
  63. package/node/TreeItem/TreeItem.types.js +5 -0
  64. package/node/TreeItem/TreeItemContent.js +109 -0
  65. package/node/TreeItem/index.js +38 -0
  66. package/node/TreeItem/treeItemClasses.js +15 -0
  67. package/node/TreeItem/useTreeItem.js +76 -0
  68. package/node/TreeView/TreeView.js +862 -0
  69. package/node/TreeView/TreeView.types.js +5 -0
  70. package/node/TreeView/TreeViewContext.js +40 -0
  71. package/node/TreeView/descendants.js +194 -0
  72. package/node/TreeView/index.js +27 -0
  73. package/node/TreeView/treeViewClasses.js +15 -0
  74. package/node/index.js +34 -0
  75. package/node/internals/models.js +5 -0
  76. package/package.json +59 -0
@@ -0,0 +1,862 @@
1
+ "use strict";
2
+
3
+ var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
4
+ Object.defineProperty(exports, "__esModule", {
5
+ value: true
6
+ });
7
+ exports.TreeView = void 0;
8
+ var _objectWithoutPropertiesLoose2 = _interopRequireDefault(require("@babel/runtime/helpers/objectWithoutPropertiesLoose"));
9
+ var _extends2 = _interopRequireDefault(require("@babel/runtime/helpers/extends"));
10
+ var React = _interopRequireWildcard(require("react"));
11
+ var _propTypes = _interopRequireDefault(require("prop-types"));
12
+ var _clsx = _interopRequireDefault(require("clsx"));
13
+ var _styles = require("@mui/material/styles");
14
+ var _base = require("@mui/base");
15
+ var _useControlled = _interopRequireDefault(require("@mui/utils/useControlled"));
16
+ var _useForkRef = _interopRequireDefault(require("@mui/utils/useForkRef"));
17
+ var _ownerDocument = _interopRequireDefault(require("@mui/utils/ownerDocument"));
18
+ var _useId = _interopRequireDefault(require("@mui/utils/useId"));
19
+ var _TreeViewContext = require("./TreeViewContext");
20
+ var _descendants = require("./descendants");
21
+ var _treeViewClasses = require("./treeViewClasses");
22
+ var _jsxRuntime = require("react/jsx-runtime");
23
+ const _excluded = ["children", "className", "defaultCollapseIcon", "defaultEndIcon", "defaultExpanded", "defaultExpandIcon", "defaultParentIcon", "defaultSelected", "disabledItemsFocusable", "disableSelection", "expanded", "id", "multiSelect", "onBlur", "onFocus", "onKeyDown", "onNodeFocus", "onNodeSelect", "onNodeToggle", "selected"];
24
+ function _getRequireWildcardCache(nodeInterop) { if (typeof WeakMap !== "function") return null; var cacheBabelInterop = new WeakMap(); var cacheNodeInterop = new WeakMap(); return (_getRequireWildcardCache = function (nodeInterop) { return nodeInterop ? cacheNodeInterop : cacheBabelInterop; })(nodeInterop); }
25
+ function _interopRequireWildcard(obj, nodeInterop) { if (!nodeInterop && obj && obj.__esModule) { return obj; } if (obj === null || typeof obj !== "object" && typeof obj !== "function") { return { default: obj }; } var cache = _getRequireWildcardCache(nodeInterop); if (cache && cache.has(obj)) { return cache.get(obj); } var newObj = {}; var hasPropertyDescriptor = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var key in obj) { if (key !== "default" && Object.prototype.hasOwnProperty.call(obj, key)) { var desc = hasPropertyDescriptor ? Object.getOwnPropertyDescriptor(obj, key) : null; if (desc && (desc.get || desc.set)) { Object.defineProperty(newObj, key, desc); } else { newObj[key] = obj[key]; } } } newObj.default = obj; if (cache) { cache.set(obj, newObj); } return newObj; }
26
+ const useUtilityClasses = ownerState => {
27
+ const {
28
+ classes
29
+ } = ownerState;
30
+ const slots = {
31
+ root: ['root']
32
+ };
33
+ return (0, _base.unstable_composeClasses)(slots, _treeViewClasses.getTreeViewUtilityClass, classes);
34
+ };
35
+ const TreeViewRoot = (0, _styles.styled)('ul', {
36
+ name: 'MuiTreeView',
37
+ slot: 'Root',
38
+ overridesResolver: (props, styles) => styles.root
39
+ })({
40
+ padding: 0,
41
+ margin: 0,
42
+ listStyle: 'none',
43
+ outline: 0
44
+ });
45
+ function isPrintableCharacter(string) {
46
+ return string && string.length === 1 && string.match(/\S/);
47
+ }
48
+ function findNextFirstChar(firstChars, startIndex, char) {
49
+ for (let i = startIndex; i < firstChars.length; i += 1) {
50
+ if (char === firstChars[i]) {
51
+ return i;
52
+ }
53
+ }
54
+ return -1;
55
+ }
56
+ function noopSelection() {
57
+ return false;
58
+ }
59
+ const defaultDefaultExpanded = [];
60
+ const defaultDefaultSelected = [];
61
+
62
+ /**
63
+ *
64
+ * Demos:
65
+ *
66
+ * - [Tree View](https://mui.com/material-ui/react-tree-view/)
67
+ *
68
+ * API:
69
+ *
70
+ * - [TreeView API](https://mui.com/material-ui/api/tree-view/)
71
+ */
72
+ const TreeView = /*#__PURE__*/React.forwardRef(function TreeView(inProps, ref) {
73
+ const themeProps = (0, _styles.useThemeProps)({
74
+ props: inProps,
75
+ name: 'MuiTreeView'
76
+ });
77
+ const props = (0, _extends2.default)({}, themeProps, {
78
+ disabledItemsFocusable: themeProps.disabledItemsFocusable ?? false,
79
+ disableSelection: themeProps.disableSelection ?? false,
80
+ multiSelect: themeProps.multiSelect ?? false,
81
+ defaultExpanded: themeProps.defaultExpanded ?? defaultDefaultExpanded,
82
+ defaultSelected: themeProps.defaultSelected ?? defaultDefaultSelected
83
+ });
84
+ const {
85
+ children,
86
+ className,
87
+ defaultCollapseIcon,
88
+ defaultEndIcon,
89
+ defaultExpanded,
90
+ defaultExpandIcon,
91
+ defaultParentIcon,
92
+ defaultSelected,
93
+ disabledItemsFocusable,
94
+ disableSelection,
95
+ expanded: expandedProp,
96
+ id: idProp,
97
+ multiSelect,
98
+ onBlur,
99
+ onFocus,
100
+ onKeyDown,
101
+ onNodeFocus,
102
+ onNodeSelect,
103
+ onNodeToggle,
104
+ selected: selectedProp
105
+ } = props,
106
+ other = (0, _objectWithoutPropertiesLoose2.default)(props, _excluded);
107
+ const theme = (0, _styles.useTheme)();
108
+ const isRtl = theme.direction === 'rtl';
109
+ const classes = useUtilityClasses(props);
110
+ const treeId = (0, _useId.default)(idProp);
111
+ const treeRef = React.useRef(null);
112
+ const handleRef = (0, _useForkRef.default)(treeRef, ref);
113
+ const [focusedNodeId, setFocusedNodeId] = React.useState(null);
114
+ const nodeMap = React.useRef({});
115
+ const firstCharMap = React.useRef({});
116
+ const [expanded, setExpandedState] = (0, _useControlled.default)({
117
+ controlled: expandedProp,
118
+ default: defaultExpanded,
119
+ name: 'TreeView',
120
+ state: 'expanded'
121
+ });
122
+ const [selected, setSelectedState] = (0, _useControlled.default)({
123
+ controlled: selectedProp,
124
+ default: defaultSelected,
125
+ name: 'TreeView',
126
+ state: 'selected'
127
+ });
128
+
129
+ /*
130
+ * Status Helpers
131
+ */
132
+ const isExpanded = React.useCallback(nodeId => Array.isArray(expanded) ? expanded.indexOf(nodeId) !== -1 : false, [expanded]);
133
+ const isExpandable = React.useCallback(nodeId => nodeMap.current[nodeId] && nodeMap.current[nodeId].expandable, []);
134
+ const isSelected = React.useCallback(nodeId => Array.isArray(selected) ? selected.indexOf(nodeId) !== -1 : selected === nodeId, [selected]);
135
+ const isDisabled = React.useCallback(nodeId => {
136
+ if (nodeId == null) {
137
+ return false;
138
+ }
139
+ let node = nodeMap.current[nodeId];
140
+
141
+ // This can be called before the node has been added to the node map.
142
+ if (!node) {
143
+ return false;
144
+ }
145
+ if (node.disabled) {
146
+ return true;
147
+ }
148
+ while (node.parentId != null) {
149
+ node = nodeMap.current[node.parentId];
150
+ if (node.disabled) {
151
+ return true;
152
+ }
153
+ }
154
+ return false;
155
+ }, []);
156
+ const isFocused = nodeId => focusedNodeId === nodeId;
157
+
158
+ /*
159
+ * Child Helpers
160
+ */
161
+
162
+ // Using Object.keys -> .map to mimic Object.values we should replace with Object.values() once we stop IE11 support.
163
+ const getChildrenIds = nodeId => Object.keys(nodeMap.current).map(key => {
164
+ return nodeMap.current[key];
165
+ }).filter(node => node.parentId === nodeId).sort((a, b) => a.index - b.index).map(child => child.id);
166
+ const getNavigableChildrenIds = nodeId => {
167
+ let childrenIds = getChildrenIds(nodeId);
168
+ if (!disabledItemsFocusable) {
169
+ childrenIds = childrenIds.filter(node => !isDisabled(node));
170
+ }
171
+ return childrenIds;
172
+ };
173
+
174
+ /*
175
+ * Node Helpers
176
+ */
177
+
178
+ const getNextNode = nodeId => {
179
+ // If expanded get first child
180
+ if (isExpanded(nodeId) && getNavigableChildrenIds(nodeId).length > 0) {
181
+ return getNavigableChildrenIds(nodeId)[0];
182
+ }
183
+ let node = nodeMap.current[nodeId];
184
+ while (node != null) {
185
+ // Try to get next sibling
186
+ const siblings = getNavigableChildrenIds(node.parentId);
187
+ const nextSibling = siblings[siblings.indexOf(node.id) + 1];
188
+ if (nextSibling) {
189
+ return nextSibling;
190
+ }
191
+
192
+ // If the sibling does not exist, go up a level to the parent and try again.
193
+ node = nodeMap.current[node.parentId];
194
+ }
195
+ return null;
196
+ };
197
+ const getPreviousNode = nodeId => {
198
+ const node = nodeMap.current[nodeId];
199
+ const siblings = getNavigableChildrenIds(node.parentId);
200
+ const nodeIndex = siblings.indexOf(nodeId);
201
+ if (nodeIndex === 0) {
202
+ return node.parentId;
203
+ }
204
+ let currentNode = siblings[nodeIndex - 1];
205
+ while (isExpanded(currentNode) && getNavigableChildrenIds(currentNode).length > 0) {
206
+ currentNode = getNavigableChildrenIds(currentNode).pop();
207
+ }
208
+ return currentNode;
209
+ };
210
+ const getLastNode = () => {
211
+ let lastNode = getNavigableChildrenIds(null).pop();
212
+ while (isExpanded(lastNode)) {
213
+ lastNode = getNavigableChildrenIds(lastNode).pop();
214
+ }
215
+ return lastNode;
216
+ };
217
+ const getFirstNode = () => getNavigableChildrenIds(null)[0];
218
+ const getParent = nodeId => nodeMap.current[nodeId].parentId;
219
+
220
+ /**
221
+ * This is used to determine the start and end of a selection range so
222
+ * we can get the nodes between the two border nodes.
223
+ *
224
+ * It finds the nodes' common ancestor using
225
+ * a naive implementation of a lowest common ancestor algorithm
226
+ * (https://en.wikipedia.org/wiki/Lowest_common_ancestor).
227
+ * Then compares the ancestor's 2 children that are ancestors of nodeA and NodeB
228
+ * so we can compare their indexes to work out which node comes first in a depth first search.
229
+ * (https://en.wikipedia.org/wiki/Depth-first_search)
230
+ *
231
+ * Another way to put it is which node is shallower in a trémaux tree
232
+ * https://en.wikipedia.org/wiki/Tr%C3%A9maux_tree
233
+ */
234
+ const findOrderInTremauxTree = (nodeAId, nodeBId) => {
235
+ if (nodeAId === nodeBId) {
236
+ return [nodeAId, nodeBId];
237
+ }
238
+ const nodeA = nodeMap.current[nodeAId];
239
+ const nodeB = nodeMap.current[nodeBId];
240
+ if (nodeA.parentId === nodeB.id || nodeB.parentId === nodeA.id) {
241
+ return nodeB.parentId === nodeA.id ? [nodeA.id, nodeB.id] : [nodeB.id, nodeA.id];
242
+ }
243
+ const aFamily = [nodeA.id];
244
+ const bFamily = [nodeB.id];
245
+ let aAncestor = nodeA.parentId;
246
+ let bAncestor = nodeB.parentId;
247
+ let aAncestorIsCommon = bFamily.indexOf(aAncestor) !== -1;
248
+ let bAncestorIsCommon = aFamily.indexOf(bAncestor) !== -1;
249
+ let continueA = true;
250
+ let continueB = true;
251
+ while (!bAncestorIsCommon && !aAncestorIsCommon) {
252
+ if (continueA) {
253
+ aFamily.push(aAncestor);
254
+ aAncestorIsCommon = bFamily.indexOf(aAncestor) !== -1;
255
+ continueA = aAncestor !== null;
256
+ if (!aAncestorIsCommon && continueA) {
257
+ aAncestor = nodeMap.current[aAncestor].parentId;
258
+ }
259
+ }
260
+ if (continueB && !aAncestorIsCommon) {
261
+ bFamily.push(bAncestor);
262
+ bAncestorIsCommon = aFamily.indexOf(bAncestor) !== -1;
263
+ continueB = bAncestor !== null;
264
+ if (!bAncestorIsCommon && continueB) {
265
+ bAncestor = nodeMap.current[bAncestor].parentId;
266
+ }
267
+ }
268
+ }
269
+ const commonAncestor = aAncestorIsCommon ? aAncestor : bAncestor;
270
+ const ancestorFamily = getChildrenIds(commonAncestor);
271
+ const aSide = aFamily[aFamily.indexOf(commonAncestor) - 1];
272
+ const bSide = bFamily[bFamily.indexOf(commonAncestor) - 1];
273
+ return ancestorFamily.indexOf(aSide) < ancestorFamily.indexOf(bSide) ? [nodeAId, nodeBId] : [nodeBId, nodeAId];
274
+ };
275
+ const getNodesInRange = (nodeAId, nodeBId) => {
276
+ const [first, last] = findOrderInTremauxTree(nodeAId, nodeBId);
277
+ const nodes = [first];
278
+ let current = first;
279
+ while (current !== last) {
280
+ current = getNextNode(current);
281
+ nodes.push(current);
282
+ }
283
+ return nodes;
284
+ };
285
+
286
+ /*
287
+ * Focus Helpers
288
+ */
289
+
290
+ const focus = (event, nodeId) => {
291
+ if (nodeId) {
292
+ setFocusedNodeId(nodeId);
293
+ if (onNodeFocus) {
294
+ onNodeFocus(event, nodeId);
295
+ }
296
+ }
297
+ };
298
+ const focusNextNode = (event, nodeId) => focus(event, getNextNode(nodeId));
299
+ const focusPreviousNode = (event, nodeId) => focus(event, getPreviousNode(nodeId));
300
+ const focusFirstNode = event => focus(event, getFirstNode());
301
+ const focusLastNode = event => focus(event, getLastNode());
302
+ const focusByFirstCharacter = (event, nodeId, firstChar) => {
303
+ let start;
304
+ let index;
305
+ const lowercaseChar = firstChar.toLowerCase();
306
+ const firstCharIds = [];
307
+ const firstChars = [];
308
+ // This really only works since the ids are strings
309
+ Object.keys(firstCharMap.current).forEach(mapNodeId => {
310
+ const map = nodeMap.current[mapNodeId];
311
+ const visible = map.parentId ? isExpanded(map.parentId) : true;
312
+ const shouldBeSkipped = disabledItemsFocusable ? false : isDisabled(mapNodeId);
313
+ if (visible && !shouldBeSkipped) {
314
+ firstCharIds.push(mapNodeId);
315
+ firstChars.push(firstCharMap.current[mapNodeId]);
316
+ }
317
+ });
318
+
319
+ // Get start index for search based on position of currentItem
320
+ start = firstCharIds.indexOf(nodeId) + 1;
321
+ if (start >= firstCharIds.length) {
322
+ start = 0;
323
+ }
324
+
325
+ // Check remaining slots in the menu
326
+ index = findNextFirstChar(firstChars, start, lowercaseChar);
327
+
328
+ // If not found in remaining slots, check from beginning
329
+ if (index === -1) {
330
+ index = findNextFirstChar(firstChars, 0, lowercaseChar);
331
+ }
332
+
333
+ // If match was found...
334
+ if (index > -1) {
335
+ focus(event, firstCharIds[index]);
336
+ }
337
+ };
338
+
339
+ /*
340
+ * Expansion Helpers
341
+ */
342
+
343
+ const toggleExpansion = (event, value = focusedNodeId) => {
344
+ if (value == null) {
345
+ return;
346
+ }
347
+ let newExpanded;
348
+ if (expanded.indexOf(value) !== -1) {
349
+ newExpanded = expanded.filter(id => id !== value);
350
+ } else {
351
+ newExpanded = [value].concat(expanded);
352
+ }
353
+ if (onNodeToggle) {
354
+ onNodeToggle(event, newExpanded);
355
+ }
356
+ setExpandedState(newExpanded);
357
+ };
358
+ const expandAllSiblings = (event, nodeId) => {
359
+ const map = nodeMap.current[nodeId];
360
+ const siblings = getChildrenIds(map.parentId);
361
+ const diff = siblings.filter(child => isExpandable(child) && !isExpanded(child));
362
+ const newExpanded = expanded.concat(diff);
363
+ if (diff.length > 0) {
364
+ setExpandedState(newExpanded);
365
+ if (onNodeToggle) {
366
+ onNodeToggle(event, newExpanded);
367
+ }
368
+ }
369
+ };
370
+
371
+ /*
372
+ * Selection Helpers
373
+ */
374
+
375
+ const lastSelectedNode = React.useRef(null);
376
+ const lastSelectionWasRange = React.useRef(false);
377
+ const currentRangeSelection = React.useRef([]);
378
+ const handleRangeArrowSelect = (event, nodes) => {
379
+ let base = selected.slice();
380
+ const {
381
+ start,
382
+ next,
383
+ current
384
+ } = nodes;
385
+ if (!next || !current) {
386
+ return;
387
+ }
388
+ if (currentRangeSelection.current.indexOf(current) === -1) {
389
+ currentRangeSelection.current = [];
390
+ }
391
+ if (lastSelectionWasRange.current) {
392
+ if (currentRangeSelection.current.indexOf(next) !== -1) {
393
+ base = base.filter(id => id === start || id !== current);
394
+ currentRangeSelection.current = currentRangeSelection.current.filter(id => id === start || id !== current);
395
+ } else {
396
+ base.push(next);
397
+ currentRangeSelection.current.push(next);
398
+ }
399
+ } else {
400
+ base.push(next);
401
+ currentRangeSelection.current.push(current, next);
402
+ }
403
+ if (onNodeSelect) {
404
+ onNodeSelect(event, base);
405
+ }
406
+ setSelectedState(base);
407
+ };
408
+ const handleRangeSelect = (event, nodes) => {
409
+ let base = selected.slice();
410
+ const {
411
+ start,
412
+ end
413
+ } = nodes;
414
+ // If last selection was a range selection ignore nodes that were selected.
415
+ if (lastSelectionWasRange.current) {
416
+ base = base.filter(id => currentRangeSelection.current.indexOf(id) === -1);
417
+ }
418
+ let range = getNodesInRange(start, end);
419
+ range = range.filter(node => !isDisabled(node));
420
+ currentRangeSelection.current = range;
421
+ let newSelected = base.concat(range);
422
+ newSelected = newSelected.filter((id, i) => newSelected.indexOf(id) === i);
423
+ if (onNodeSelect) {
424
+ onNodeSelect(event, newSelected);
425
+ }
426
+ setSelectedState(newSelected);
427
+ };
428
+ const handleMultipleSelect = (event, nodeId) => {
429
+ if (!Array.isArray(selected)) {
430
+ return;
431
+ }
432
+ let newSelected;
433
+ if (selected.indexOf(nodeId) !== -1) {
434
+ newSelected = selected.filter(id => id !== nodeId);
435
+ } else {
436
+ newSelected = [nodeId].concat(selected);
437
+ }
438
+ if (onNodeSelect) {
439
+ onNodeSelect(event, newSelected);
440
+ }
441
+ setSelectedState(newSelected);
442
+ };
443
+ const handleSingleSelect = (event, nodeId) => {
444
+ const newSelected = multiSelect ? [nodeId] : nodeId;
445
+ if (onNodeSelect) {
446
+ onNodeSelect(event, newSelected);
447
+ }
448
+ setSelectedState(newSelected);
449
+ };
450
+ const selectNode = (event, nodeId, multiple = false) => {
451
+ if (nodeId) {
452
+ if (multiple) {
453
+ handleMultipleSelect(event, nodeId);
454
+ } else {
455
+ handleSingleSelect(event, nodeId);
456
+ }
457
+ lastSelectedNode.current = nodeId;
458
+ lastSelectionWasRange.current = false;
459
+ currentRangeSelection.current = [];
460
+ return true;
461
+ }
462
+ return false;
463
+ };
464
+ const selectRange = (event, nodes, stacked = false) => {
465
+ const {
466
+ start = lastSelectedNode.current,
467
+ end,
468
+ current
469
+ } = nodes;
470
+ if (stacked) {
471
+ handleRangeArrowSelect(event, {
472
+ start,
473
+ next: end,
474
+ current
475
+ });
476
+ } else if (start != null && end != null) {
477
+ handleRangeSelect(event, {
478
+ start,
479
+ end
480
+ });
481
+ }
482
+ lastSelectionWasRange.current = true;
483
+ };
484
+ const rangeSelectToFirst = (event, id) => {
485
+ if (!lastSelectedNode.current) {
486
+ lastSelectedNode.current = id;
487
+ }
488
+ const start = lastSelectionWasRange.current ? lastSelectedNode.current : id;
489
+ selectRange(event, {
490
+ start,
491
+ end: getFirstNode()
492
+ });
493
+ };
494
+ const rangeSelectToLast = (event, id) => {
495
+ if (!lastSelectedNode.current) {
496
+ lastSelectedNode.current = id;
497
+ }
498
+ const start = lastSelectionWasRange.current ? lastSelectedNode.current : id;
499
+ selectRange(event, {
500
+ start,
501
+ end: getLastNode()
502
+ });
503
+ };
504
+ const selectNextNode = (event, id) => {
505
+ if (!isDisabled(getNextNode(id))) {
506
+ selectRange(event, {
507
+ end: getNextNode(id),
508
+ current: id
509
+ }, true);
510
+ }
511
+ };
512
+ const selectPreviousNode = (event, id) => {
513
+ if (!isDisabled(getPreviousNode(id))) {
514
+ selectRange(event, {
515
+ end: getPreviousNode(id),
516
+ current: id
517
+ }, true);
518
+ }
519
+ };
520
+ const selectAllNodes = event => {
521
+ selectRange(event, {
522
+ start: getFirstNode(),
523
+ end: getLastNode()
524
+ });
525
+ };
526
+
527
+ /*
528
+ * Mapping Helpers
529
+ */
530
+ const registerNode = React.useCallback(node => {
531
+ const {
532
+ id,
533
+ index,
534
+ parentId,
535
+ expandable,
536
+ idAttribute,
537
+ disabled
538
+ } = node;
539
+ nodeMap.current[id] = {
540
+ id,
541
+ index,
542
+ parentId,
543
+ expandable,
544
+ idAttribute,
545
+ disabled
546
+ };
547
+ }, []);
548
+ const unregisterNode = React.useCallback(nodeId => {
549
+ const newMap = (0, _extends2.default)({}, nodeMap.current);
550
+ delete newMap[nodeId];
551
+ nodeMap.current = newMap;
552
+ setFocusedNodeId(oldFocusedNodeId => {
553
+ if (oldFocusedNodeId === nodeId && treeRef.current === (0, _ownerDocument.default)(treeRef.current).activeElement) {
554
+ return getChildrenIds(null)[0];
555
+ }
556
+ return oldFocusedNodeId;
557
+ });
558
+ }, []);
559
+ const mapFirstChar = React.useCallback((nodeId, firstChar) => {
560
+ firstCharMap.current[nodeId] = firstChar;
561
+ }, []);
562
+ const unMapFirstChar = React.useCallback(nodeId => {
563
+ const newMap = (0, _extends2.default)({}, firstCharMap.current);
564
+ delete newMap[nodeId];
565
+ firstCharMap.current = newMap;
566
+ }, []);
567
+
568
+ /**
569
+ * Event handlers and Navigation
570
+ */
571
+
572
+ const handleNextArrow = event => {
573
+ if (focusedNodeId != null && isExpandable(focusedNodeId)) {
574
+ if (isExpanded(focusedNodeId)) {
575
+ focusNextNode(event, focusedNodeId);
576
+ } else if (!isDisabled(focusedNodeId)) {
577
+ toggleExpansion(event);
578
+ }
579
+ }
580
+ return true;
581
+ };
582
+ const handlePreviousArrow = event => {
583
+ if (focusedNodeId == null) {
584
+ return false;
585
+ }
586
+ if (isExpanded(focusedNodeId) && !isDisabled(focusedNodeId)) {
587
+ toggleExpansion(event, focusedNodeId);
588
+ return true;
589
+ }
590
+ const parent = getParent(focusedNodeId);
591
+ if (parent) {
592
+ focus(event, parent);
593
+ return true;
594
+ }
595
+ return false;
596
+ };
597
+ const handleKeyDown = event => {
598
+ let flag = false;
599
+ const key = event.key;
600
+
601
+ // If the tree is empty there will be no focused node
602
+ if (event.altKey || event.currentTarget !== event.target || !focusedNodeId) {
603
+ return;
604
+ }
605
+ const ctrlPressed = event.ctrlKey || event.metaKey;
606
+ switch (key) {
607
+ case ' ':
608
+ if (!disableSelection && focusedNodeId != null && !isDisabled(focusedNodeId)) {
609
+ if (multiSelect && event.shiftKey) {
610
+ selectRange(event, {
611
+ end: focusedNodeId
612
+ });
613
+ flag = true;
614
+ } else if (multiSelect) {
615
+ flag = selectNode(event, focusedNodeId, true);
616
+ } else {
617
+ flag = selectNode(event, focusedNodeId);
618
+ }
619
+ }
620
+ event.stopPropagation();
621
+ break;
622
+ case 'Enter':
623
+ if (!isDisabled(focusedNodeId)) {
624
+ if (isExpandable(focusedNodeId)) {
625
+ toggleExpansion(event);
626
+ flag = true;
627
+ } else if (multiSelect) {
628
+ flag = selectNode(event, focusedNodeId, true);
629
+ } else {
630
+ flag = selectNode(event, focusedNodeId);
631
+ }
632
+ }
633
+ event.stopPropagation();
634
+ break;
635
+ case 'ArrowDown':
636
+ if (multiSelect && event.shiftKey && !disableSelection) {
637
+ selectNextNode(event, focusedNodeId);
638
+ }
639
+ focusNextNode(event, focusedNodeId);
640
+ flag = true;
641
+ break;
642
+ case 'ArrowUp':
643
+ if (multiSelect && event.shiftKey && !disableSelection) {
644
+ selectPreviousNode(event, focusedNodeId);
645
+ }
646
+ focusPreviousNode(event, focusedNodeId);
647
+ flag = true;
648
+ break;
649
+ case 'ArrowRight':
650
+ if (isRtl) {
651
+ flag = handlePreviousArrow(event);
652
+ } else {
653
+ flag = handleNextArrow(event);
654
+ }
655
+ break;
656
+ case 'ArrowLeft':
657
+ if (isRtl) {
658
+ flag = handleNextArrow(event);
659
+ } else {
660
+ flag = handlePreviousArrow(event);
661
+ }
662
+ break;
663
+ case 'Home':
664
+ if (multiSelect && ctrlPressed && event.shiftKey && !disableSelection && !isDisabled(focusedNodeId)) {
665
+ rangeSelectToFirst(event, focusedNodeId);
666
+ }
667
+ focusFirstNode(event);
668
+ flag = true;
669
+ break;
670
+ case 'End':
671
+ if (multiSelect && ctrlPressed && event.shiftKey && !disableSelection && !isDisabled(focusedNodeId)) {
672
+ rangeSelectToLast(event, focusedNodeId);
673
+ }
674
+ focusLastNode(event);
675
+ flag = true;
676
+ break;
677
+ default:
678
+ if (key === '*') {
679
+ expandAllSiblings(event, focusedNodeId);
680
+ flag = true;
681
+ } else if (multiSelect && ctrlPressed && key.toLowerCase() === 'a' && !disableSelection) {
682
+ selectAllNodes(event);
683
+ flag = true;
684
+ } else if (!ctrlPressed && !event.shiftKey && isPrintableCharacter(key)) {
685
+ focusByFirstCharacter(event, focusedNodeId, key);
686
+ flag = true;
687
+ }
688
+ }
689
+ if (flag) {
690
+ event.preventDefault();
691
+ event.stopPropagation();
692
+ }
693
+ if (onKeyDown) {
694
+ onKeyDown(event);
695
+ }
696
+ };
697
+ const handleFocus = event => {
698
+ // if the event bubbled (which is React specific) we don't want to steal focus
699
+ if (event.target === event.currentTarget) {
700
+ const firstSelected = Array.isArray(selected) ? selected[0] : selected;
701
+ focus(event, firstSelected || getNavigableChildrenIds(null)[0]);
702
+ }
703
+ if (onFocus) {
704
+ onFocus(event);
705
+ }
706
+ };
707
+ const handleBlur = event => {
708
+ setFocusedNodeId(null);
709
+ if (onBlur) {
710
+ onBlur(event);
711
+ }
712
+ };
713
+ const activeDescendant = nodeMap.current[focusedNodeId] ? nodeMap.current[focusedNodeId].idAttribute : null;
714
+ return /*#__PURE__*/(0, _jsxRuntime.jsx)(_TreeViewContext.TreeViewContext.Provider, {
715
+ // TODO: fix this lint error
716
+ // eslint-disable-next-line react/jsx-no-constructed-context-values
717
+ value: {
718
+ icons: {
719
+ defaultCollapseIcon,
720
+ defaultExpandIcon,
721
+ defaultParentIcon,
722
+ defaultEndIcon
723
+ },
724
+ focus,
725
+ toggleExpansion,
726
+ isExpanded,
727
+ isExpandable,
728
+ isFocused,
729
+ isSelected,
730
+ isDisabled,
731
+ selectNode: disableSelection ? noopSelection : selectNode,
732
+ selectRange: disableSelection ? noopSelection : selectRange,
733
+ multiSelect,
734
+ disabledItemsFocusable,
735
+ mapFirstChar,
736
+ unMapFirstChar,
737
+ registerNode,
738
+ unregisterNode,
739
+ treeId
740
+ },
741
+ children: /*#__PURE__*/(0, _jsxRuntime.jsx)(_descendants.DescendantProvider, {
742
+ children: /*#__PURE__*/(0, _jsxRuntime.jsx)(TreeViewRoot, (0, _extends2.default)({
743
+ role: "tree",
744
+ id: treeId,
745
+ "aria-activedescendant": activeDescendant ?? undefined,
746
+ "aria-multiselectable": multiSelect,
747
+ className: (0, _clsx.default)(classes.root, className),
748
+ tabIndex: 0,
749
+ onKeyDown: handleKeyDown,
750
+ onFocus: handleFocus,
751
+ onBlur: handleBlur,
752
+ ownerState: props
753
+ }, other, {
754
+ ref: handleRef,
755
+ children: children
756
+ }))
757
+ })
758
+ });
759
+ });
760
+ exports.TreeView = TreeView;
761
+ process.env.NODE_ENV !== "production" ? TreeView.propTypes = {
762
+ // ----------------------------- Warning --------------------------------
763
+ // | These PropTypes are generated from the TypeScript type definitions |
764
+ // | To update them edit the TypeScript types and run "yarn proptypes" |
765
+ // ----------------------------------------------------------------------
766
+ /**
767
+ * The content of the component.
768
+ */
769
+ children: _propTypes.default.node,
770
+ /**
771
+ * Override or extend the styles applied to the component.
772
+ */
773
+ classes: _propTypes.default.object,
774
+ /**
775
+ * className applied to the root element.
776
+ */
777
+ className: _propTypes.default.string,
778
+ /**
779
+ * The default icon used to collapse the node.
780
+ */
781
+ defaultCollapseIcon: _propTypes.default.node,
782
+ /**
783
+ * The default icon displayed next to a end node. This is applied to all
784
+ * tree nodes and can be overridden by the TreeItem `icon` prop.
785
+ */
786
+ defaultEndIcon: _propTypes.default.node,
787
+ /**
788
+ * Expanded node ids.
789
+ * Used when the item's expansion are not controlled.
790
+ * @default []
791
+ */
792
+ defaultExpanded: _propTypes.default.arrayOf(_propTypes.default.string),
793
+ /**
794
+ * The default icon used to expand the node.
795
+ */
796
+ defaultExpandIcon: _propTypes.default.node,
797
+ /**
798
+ * The default icon displayed next to a parent node. This is applied to all
799
+ * parent nodes and can be overridden by the TreeItem `icon` prop.
800
+ */
801
+ defaultParentIcon: _propTypes.default.node,
802
+ /**
803
+ * Selected node ids. (Uncontrolled)
804
+ * When `multiSelect` is true this takes an array of strings; when false (default) a string.
805
+ * @default []
806
+ */
807
+ defaultSelected: _propTypes.default.oneOfType([_propTypes.default.arrayOf(_propTypes.default.string), _propTypes.default.string]),
808
+ /**
809
+ * If `true`, will allow focus on disabled items.
810
+ * @default false
811
+ */
812
+ disabledItemsFocusable: _propTypes.default.bool,
813
+ /**
814
+ * If `true` selection is disabled.
815
+ * @default false
816
+ */
817
+ disableSelection: _propTypes.default.bool,
818
+ /**
819
+ * Expanded node ids.
820
+ * Used when the item's expansion are controlled.
821
+ */
822
+ expanded: _propTypes.default.arrayOf(_propTypes.default.string),
823
+ /**
824
+ * This prop is used to help implement the accessibility logic.
825
+ * If you don't provide this prop. It falls back to a randomly generated id.
826
+ */
827
+ id: _propTypes.default.string,
828
+ /**
829
+ * If true `ctrl` and `shift` will trigger multiselect.
830
+ * @default false
831
+ */
832
+ multiSelect: _propTypes.default.bool,
833
+ /**
834
+ * Callback fired when tree items are focused.
835
+ * @param {React.SyntheticEvent} event The event source of the callback **Warning**: This is a generic event not a focus event.
836
+ * @param {string} nodeId The id of the node focused.
837
+ * @param {string} value of the focused node.
838
+ */
839
+ onNodeFocus: _propTypes.default.func,
840
+ /**
841
+ * Callback fired when tree items are selected/unselected.
842
+ * @param {React.SyntheticEvent} event The event source of the callback
843
+ * @param {string[] | string} nodeIds Ids of the selected nodes. When `multiSelect` is true
844
+ * this is an array of strings; when false (default) a string.
845
+ */
846
+ onNodeSelect: _propTypes.default.func,
847
+ /**
848
+ * Callback fired when tree items are expanded/collapsed.
849
+ * @param {React.SyntheticEvent} event The event source of the callback.
850
+ * @param {array} nodeIds The ids of the expanded nodes.
851
+ */
852
+ onNodeToggle: _propTypes.default.func,
853
+ /**
854
+ * Selected node ids. (Controlled)
855
+ * When `multiSelect` is true this takes an array of strings; when false (default) a string.
856
+ */
857
+ selected: _propTypes.default.oneOfType([_propTypes.default.arrayOf(_propTypes.default.string), _propTypes.default.string]),
858
+ /**
859
+ * The system prop that allows defining system overrides as well as additional CSS styles.
860
+ */
861
+ sx: _propTypes.default.oneOfType([_propTypes.default.arrayOf(_propTypes.default.oneOfType([_propTypes.default.func, _propTypes.default.object, _propTypes.default.bool])), _propTypes.default.func, _propTypes.default.object])
862
+ } : void 0;