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