@itwin/presentation-hierarchies-react 0.1.0

package/lib/cjs/presentation-hierarchies-react/UseSelectionHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseSelectionHandler.js","sourceRoot":"","sources":["../../../src/presentation-hierarchies-react/UseSelectionHandler.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAA0C;AAC1C,yCAA+E;AAoC/E;;;GAGG;AACH,SAAgB,mBAAmB,CAAC,KAA+B;IACjE,MAAM,EAAE,SAAS,EAAE,aAAa,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IACxD,MAAM,oBAAoB,GAAG,IAAA,cAAM,EAAqB,SAAS,CAAC,CAAC;IACnE,MAAM,KAAK,GAAG,IAAA,cAAM,EAAgB;QAClC,YAAY,EAAE,EAAE;QAChB,gBAAgB,EAAE,IAAI,GAAG,EAAE;KAC5B,CAAC,CAAC;IAEH,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,SAAS,EAAE,CAAC;YACf,OAAO;QACT,CAAC;QACD,KAAK,CAAC,OAAO,GAAG,mBAAmB,CAAC,SAAS,CAAC,CAAC;IACjD,CAAC,EAAE,CAAC,SAAS,CAAC,CAAC,CAAC;IAEhB,MAAM,YAAY,GAAG,CAAC,OAAgB,EAAE,QAAiB,EAAE,EAAE;QAC3D,MAAM,QAAQ,GAAG,CAAC,MAAe,EAAE,EAAE;YACnC,OAAO,MAAM,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,gBAAgB,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;QACjE,CAAC,CAAC;QACF,MAAM,UAAU,GAAG,QAAQ,CAAC,OAAO,CAAC,CAAC;QACrC,MAAM,WAAW,GAAG,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACvC,IAAI,UAAU,KAAK,SAAS,IAAI,WAAW,KAAK,SAAS,EAAE,CAAC;YAC1D,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,aAAa,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACxD,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,CAAC,UAAU,EAAE,WAAW,CAAC,CAAC;QACnD,OAAO,KAAK,CAAC,OAAO,CAAC,YAAY,CAAC,KAAK,CAAC,aAAa,EAAE,QAAQ,GAAG,CAAC,CAAC,CAAC;IACvE,CAAC,CAAC;IAEF,MAAM,YAAY,GAAG,CAAC,MAAc,EAAE,UAAmB,EAAE,SAAkB,EAAE,QAAiB,EAAE,EAAE;QAClG,MAAM,SAAS,GAAG,kBAAkB,CAAC,aAAa,EAAE,UAAU,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;QACrF,IAAI,SAAS,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;YAClC,OAAO;QACT,CAAC;QAED,MAAM,KAAK,GAAG,SAAS,CAAC,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,YAAY,CAAC,oBAAoB,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC;QAC3G,IAAI,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC;YAClB,OAAO;QACT,CAAC;QAED,SAAS,CAAC,MAAM,KAAK,OAAO,IAAI,CAAC,oBAAoB,CAAC,OAAO,GAAG,MAAM,CAAC,CAAC;QACxE,WAAW,CAAC,KAAK,EAAE,SAAS,CAAC,IAAI,CAAC,CAAC;IACrC,CAAC,CAAC;IAEF,MAAM,WAAW,GAAG,CAAC,MAAc,EAAE,UAAmB,EAAE,KAAgD,EAAE,EAAE;QAC5G,OAAO,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;IACzE,CAAC,CAAC;IAEF,MAAM,aAAa,GAAG,CAAC,MAAc,EAAE,UAAmB,EAAE,KAAuC,EAAE,EAAE;QACrG,IAAI,KAAK,CAAC,GAAG,KAAK,GAAG,IAAI,KAAK,CAAC,GAAG,KAAK,UAAU,IAAI,KAAK,CAAC,GAAG,KAAK,OAAO,EAAE,CAAC;YAC3E,OAAO,YAAY,CAAC,MAAM,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,CAAC;QACzE,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,EAAE,WAAW,EAAE,aAAa,EAAE,CAAC;AACxC,CAAC;AAxDD,kDAwDC;AAOD,SAAS,kBAAkB,CAAC,aAA4B,EAAE,UAAmB,EAAE,SAAkB,EAAE,QAAiB;IAClH,QAAQ,aAAa,EAAE,CAAC;QACtB,KAAK,MAAM;YACT,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,EAAE,CAAC;QAC9C,KAAK,QAAQ;YACX,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACrE,KAAK,UAAU;YACb,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,UAAU,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC;QACjE,KAAK,UAAU;YACb,OAAO,0BAA0B,CAAC,UAAU,EAAE,SAAS,EAAE,QAAQ,CAAC,CAAC;IACvE,CAAC;AACH,CAAC;AAED,SAAS,0BAA0B,CAAC,UAAmB,EAAE,SAAkB,EAAE,QAAiB;IAC5F,IAAI,SAAS,EAAE,CAAC;QACd,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,CAAC;IAC9C,CAAC;IACD,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC5C,CAAC;IAED,OAAO,EAAE,MAAM,EAAE,MAAM,EAAE,IAAI,EAAE,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS,EAAE,CAAC;AAChE,CAAC;AAED,SAAS,mBAAmB,CAAC,SAAsC;IACjE,MAAM,YAAY,GAAkB,EAAE,CAAC;IACvC,MAAM,gBAAgB,GAAwB,IAAI,GAAG,EAAE,CAAC;IAExD,MAAM,sBAAsB,GAAG,CAAC,KAAkC,EAAE,EAAE;QACpE,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;YACrB,IAAI,CAAC,IAAA,sCAA2B,EAAC,IAAI,CAAC,EAAE,CAAC;gBACvC,OAAO;YACT,CAAC;YACD,gBAAgB,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,YAAY,CAAC,MAAM,CAAC,CAAC;YACnD,YAAY,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC3B,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gBAC1D,sBAAsB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YACxC,CAAC;QACH,CAAC,CAAC,CAAC;IACL,CAAC,CAAC;IAEF,sBAAsB,CAAC,SAAS,CAAC,CAAC;IAClC,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,CAAC;AAC5C,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useEffect, useRef } from \"react\";\nimport { isPresentationHierarchyNode, PresentationTreeNode } from \"./TreeNode\";\nimport { useTree } from \"./UseTree\";\n\n/**\n * A union of different supported selection modes in a tree component:\n * - `none` - no selection is allowed,\n * - `single` - only one node can be selected at a time,\n * - `extended` - multiple nodes can be selected using shift and ctrl keys,\n * - `multiple` - multiple nodes can be selected without using shift or ctrl keys.\n *\n * @internal\n */\nexport type SelectionMode = \"none\" | \"single\" | \"extended\" | \"multiple\";\n\n/** @internal */\nexport type SelectionChangeType = \"add\" | \"replace\" | \"remove\";\n\n/** Props for `useSelectionHandler` hook. */\ntype UseSelectionHandlerProps = Pick<ReturnType<typeof useTree>, \"rootNodes\" | \"selectNodes\"> & {\n  /** Selection mode that the component is working in. */\n  selectionMode: SelectionMode;\n};\n\n/** Result of `useSelectionHandler` hook. */\ninterface UseSelectionHandlerResult {\n  /** Should be called by node renderer when a node component is clicked. */\n  onNodeClick: (nodeId: string, isSelected: boolean, event: React.MouseEvent<HTMLElement, MouseEvent>) => void;\n  /** Should be called by node renderer when a keyboard event happens on a node. */\n  onNodeKeyDown: (nodeId: string, isSelected: boolean, event: React.KeyboardEvent<HTMLElement>) => void;\n}\n\ninterface FlatTreeState {\n  flatNodeList: Array<string>;\n  nodeIdToIndexMap: Map<string, number>;\n}\n\n/**\n * A react hook that helps implement different selection modes in a tree component created using `useTree` hook.\n * @beta\n */\nexport function useSelectionHandler(props: UseSelectionHandlerProps): UseSelectionHandlerResult {\n  const { rootNodes, selectionMode, selectNodes } = props;\n  const previousSelectionRef = useRef<string | undefined>(undefined);\n  const state = useRef<FlatTreeState>({\n    flatNodeList: [],\n    nodeIdToIndexMap: new Map(),\n  });\n\n  useEffect(() => {\n    if (!rootNodes) {\n      return;\n    }\n    state.current = computeFlatNodeList(rootNodes);\n  }, [rootNodes]);\n\n  const getNodeRange = (firstId?: string, secondId?: string) => {\n    const getIndex = (nodeId?: string) => {\n      return nodeId ? state.current.nodeIdToIndexMap.get(nodeId) : 0;\n    };\n    const firstIndex = getIndex(firstId);\n    const secondIndex = getIndex(secondId);\n    if (firstIndex === undefined || secondIndex === undefined) {\n      return [];\n    }\n\n    const startingIndex = Math.min(firstIndex, secondIndex);\n    const endIndex = Math.max(firstIndex, secondIndex);\n    return state.current.flatNodeList.slice(startingIndex, endIndex + 1);\n  };\n\n  const onNodeSelect = (nodeId: string, isSelected: boolean, shiftDown: boolean, ctrlDown: boolean) => {\n    const selection = getSelectionAction(selectionMode, isSelected, shiftDown, ctrlDown);\n    if (selection.type === \"disabled\") {\n      return;\n    }\n\n    const nodes = selection.select === \"range\" ? getNodeRange(previousSelectionRef.current, nodeId) : [nodeId];\n    if (!nodes.length) {\n      return;\n    }\n\n    selection.select !== \"range\" && (previousSelectionRef.current = nodeId);\n    selectNodes(nodes, selection.type);\n  };\n\n  const onNodeClick = (nodeId: string, isSelected: boolean, event: React.MouseEvent<HTMLElement, MouseEvent>) => {\n    return onNodeSelect(nodeId, isSelected, event.shiftKey, event.ctrlKey);\n  };\n\n  const onNodeKeyDown = (nodeId: string, isSelected: boolean, event: React.KeyboardEvent<HTMLElement>) => {\n    if (event.key === \" \" || event.key === \"Spacebar\" || event.key === \"Enter\") {\n      return onNodeSelect(nodeId, isSelected, event.shiftKey, event.ctrlKey);\n    }\n  };\n\n  return { onNodeClick, onNodeKeyDown };\n}\n\ninterface SelectionAction {\n  select: \"node\" | \"range\";\n  type: SelectionChangeType | \"disabled\";\n}\n\nfunction getSelectionAction(selectionMode: SelectionMode, isSelected: boolean, shiftDown: boolean, ctrlDown: boolean): SelectionAction {\n  switch (selectionMode) {\n    case \"none\":\n      return { select: \"node\", type: \"disabled\" };\n    case \"single\":\n      return { select: \"node\", type: isSelected ? \"replace\" : \"remove\" };\n    case \"multiple\":\n      return { select: \"node\", type: isSelected ? \"add\" : \"remove\" };\n    case \"extended\":\n      return getExtendedSelectionAction(isSelected, shiftDown, ctrlDown);\n  }\n}\n\nfunction getExtendedSelectionAction(isSelected: boolean, shiftDown: boolean, ctrlDown: boolean): SelectionAction {\n  if (shiftDown) {\n    return { select: \"range\", type: \"replace\" };\n  }\n  if (!isSelected) {\n    return { select: \"node\", type: \"remove\" };\n  }\n\n  return { select: \"node\", type: ctrlDown ? \"add\" : \"replace\" };\n}\n\nfunction computeFlatNodeList(rootNodes: Array<PresentationTreeNode>): FlatTreeState {\n  const flatNodeList: Array<string> = [];\n  const nodeIdToIndexMap: Map<string, number> = new Map();\n\n  const flattenNodeRecursively = (nodes: Array<PresentationTreeNode>) => {\n    nodes.forEach((node) => {\n      if (!isPresentationHierarchyNode(node)) {\n        return;\n      }\n      nodeIdToIndexMap.set(node.id, flatNodeList.length);\n      flatNodeList.push(node.id);\n      if (node.isExpanded && typeof node.children !== \"boolean\") {\n        flattenNodeRecursively(node.children);\n      }\n    });\n  };\n\n  flattenNodeRecursively(rootNodes);\n  return { flatNodeList, nodeIdToIndexMap };\n}\n"]}

package/lib/cjs/presentation-hierarchies-react/UseTree.d.ts

@@ -0,0 +1,85 @@
+import { createHierarchyProvider, GenericInstanceFilter, HierarchyDefinition, HierarchyNode, HierarchyNodeIdentifiersPath, LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import { ECClassHierarchyInspector, ECSchemaProvider, InstanceKey, IPrimitiveValueFormatter } from "@itwin/presentation-shared";
+import { UseUnifiedTreeSelectionProps } from "./internal/UseUnifiedSelection";
+import { PresentationTreeNode } from "./TreeNode";
+import { SelectionChangeType } from "./UseSelectionHandler";
+/**
+ * A data structure that contains information about a single hierarchy level.
+ * @beta
+ */
+export interface HierarchyLevelDetails {
+    /** The parent node whose hierarchy level's information is contained in this data structure */
+    hierarchyNode: HierarchyNode | undefined;
+    /** A function to get instance keys of the hierarchy level. */
+    getInstanceKeysIterator: (props?: {
+        instanceFilter?: GenericInstanceFilter;
+        hierarchyLevelSizeLimit?: number | "unbounded";
+    }) => AsyncIterableIterator<InstanceKey>;
+    /** Get the limit of how many nodes can be loaded in this hierarchy level. */
+    sizeLimit?: number | "unbounded";
+    /** Set the limit of how many nodes can be loaded in this hierarchy level. */
+    setSizeLimit: (value: undefined | number | "unbounded") => void;
+    /** Get the active instance filter applied to this hierarchy level. */
+    instanceFilter?: GenericInstanceFilter;
+    /** Set the instance filter to apply to this hierarchy level */
+    setInstanceFilter: (filter: GenericInstanceFilter | undefined) => void;
+}
+/**
+ * A React hook that creates state for a tree component.
+ *
+ * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a
+ * component-agnostic result which may be used to render the hierarchy using any UI framework.
+ *
+ * See `README.md` for an example
+ *
+ * @see `useUnifiedSelectionTree`
+ * @beta
+ */
+export declare function useTree(props: UseTreeProps): UseTreeResult;
+/**
+ * A React hook that creates state for a tree component, that is integrated with unified selection
+ * through context provided by `UnifiedSelectionProvider`.
+ *
+ * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a
+ * component-agnostic result which may be used to render the hierarchy using any UI framework.
+ *
+ * See `README.md` for an example
+ *
+ * @see `useTree`
+ * @see `UnifiedSelectionProvider`
+ * @beta
+ */
+export declare function useUnifiedSelectionTree({ imodelKey, sourceName, ...props }: UseTreeProps & Omit<UseUnifiedTreeSelectionProps, "getNode">): UseTreeResult;
+type IModelAccess = ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector;
+interface GetFilteredPathsProps {
+    imodelAccess: IModelAccess;
+}
+interface UseTreeProps {
+    imodelAccess: ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector;
+    getHierarchyDefinition: (props: {
+        imodelAccess: IModelAccess;
+    }) => HierarchyDefinition;
+    getFilteredPaths?: (props: GetFilteredPathsProps) => Promise<HierarchyNodeIdentifiersPath[] | undefined>;
+    localizedStrings?: Parameters<typeof createHierarchyProvider>[0]["localizedStrings"];
+}
+interface UseTreeResult {
+    /**
+     * Array containing root tree nodes. It is `undefined` on initial render until any nodes are loaded.
+     */
+    rootNodes: PresentationTreeNode[] | undefined;
+    /**
+     * Specifies whether tree is loading or not. It is set to `true` when initial tree load is in progress
+     * or tree is reloading.
+     */
+    isLoading: boolean;
+    reloadTree: (options?: {
+        discardState?: boolean;
+    }) => void;
+    expandNode: (nodeId: string, isExpanded: boolean) => void;
+    selectNodes: (nodeIds: Array<string>, changeType: SelectionChangeType) => void;
+    isNodeSelected: (nodeId: string) => boolean;
+    getHierarchyLevelDetails: (nodeId: string | undefined) => HierarchyLevelDetails | undefined;
+    setFormatter: (formatter: IPrimitiveValueFormatter | undefined) => void;
+}
+export {};
+//# sourceMappingURL=UseTree.d.ts.map

package/lib/cjs/presentation-hierarchies-react/UseTree.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseTree.d.ts","sourceRoot":"","sources":["../../../src/presentation-hierarchies-react/UseTree.ts"],"names":[],"mappings":"AAMA,OAAO,EACL,uBAAuB,EACvB,qBAAqB,EACrB,mBAAmB,EACnB,aAAa,EACb,4BAA4B,EAE5B,0BAA0B,EAC3B,MAAM,iCAAiC,CAAC;AACzC,OAAO,EAAE,yBAAyB,EAAE,gBAAgB,EAAE,WAAW,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AAGhI,OAAO,EAA2B,4BAA4B,EAAE,MAAM,gCAAgC,CAAC;AACvG,OAAO,EAA6B,oBAAoB,EAAE,MAAM,YAAY,CAAC;AAC7E,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAE5D;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,8FAA8F;IAC9F,aAAa,EAAE,aAAa,GAAG,SAAS,CAAC;IAEzC,8DAA8D;IAC9D,uBAAuB,EAAE,CAAC,KAAK,CAAC,EAAE;QAChC,cAAc,CAAC,EAAE,qBAAqB,CAAC;QACvC,uBAAuB,CAAC,EAAE,MAAM,GAAG,WAAW,CAAC;KAChD,KAAK,qBAAqB,CAAC,WAAW,CAAC,CAAC;IAEzC,6EAA6E;IAC7E,SAAS,CAAC,EAAE,MAAM,GAAG,WAAW,CAAC;IACjC,6EAA6E;IAC7E,YAAY,EAAE,CAAC,KAAK,EAAE,SAAS,GAAG,MAAM,GAAG,WAAW,KAAK,IAAI,CAAC;IAEhE,sEAAsE;IACtE,cAAc,CAAC,EAAE,qBAAqB,CAAC;IACvC,+DAA+D;IAC/D,iBAAiB,EAAE,CAAC,MAAM,EAAE,qBAAqB,GAAG,SAAS,KAAK,IAAI,CAAC;CACxE;AAED;;;;;;;;;;GAUG;AACH,wBAAgB,OAAO,CAAC,KAAK,EAAE,YAAY,GAAG,aAAa,CAG1D;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,uBAAuB,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,KAAK,EAAE,EAAE,YAAY,GAAG,IAAI,CAAC,4BAA4B,EAAE,SAAS,CAAC,GAAG,aAAa,CAGxJ;AAED,KAAK,YAAY,GAAG,gBAAgB,GAAG,0BAA0B,GAAG,yBAAyB,CAAC;AAE9F,UAAU,qBAAqB;IAC7B,YAAY,EAAE,YAAY,CAAC;CAC5B;AAED,UAAU,YAAY;IACpB,YAAY,EAAE,gBAAgB,GAAG,0BAA0B,GAAG,yBAAyB,CAAC;IACxF,sBAAsB,EAAE,CAAC,KAAK,EAAE;QAAE,YAAY,EAAE,YAAY,CAAA;KAAE,KAAK,mBAAmB,CAAC;IACvF,gBAAgB,CAAC,EAAE,CAAC,KAAK,EAAE,qBAAqB,KAAK,OAAO,CAAC,4BAA4B,EAAE,GAAG,SAAS,CAAC,CAAC;IACzG,gBAAgB,CAAC,EAAE,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC;CACtF;AAED,UAAU,aAAa;IACrB;;OAEG;IACH,SAAS,EAAE,oBAAoB,EAAE,GAAG,SAAS,CAAC;IAC9C;;;OAGG;IACH,SAAS,EAAE,OAAO,CAAC;IACnB,UAAU,EAAE,CAAC,OAAO,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE,KAAK,IAAI,CAAC;IAC3D,UAAU,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO,KAAK,IAAI,CAAC;IAC1D,WAAW,EAAE,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,mBAAmB,KAAK,IAAI,CAAC;IAC/E,cAAc,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC;IAC5C,wBAAwB,EAAE,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,KAAK,qBAAqB,GAAG,SAAS,CAAC;IAC5F,YAAY,EAAE,CAAC,SAAS,EAAE,wBAAwB,GAAG,SAAS,KAAK,IAAI,CAAC;CACzE"}

package/lib/cjs/presentation-hierarchies-react/UseTree.js

@@ -0,0 +1,214 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useUnifiedSelectionTree = exports.useTree = void 0;
+const react_1 = require("react");
+const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
+const TreeActions_1 = require("./internal/TreeActions");
+const TreeModel_1 = require("./internal/TreeModel");
+const UseUnifiedSelection_1 = require("./internal/UseUnifiedSelection");
+/**
+ * A React hook that creates state for a tree component.
+ *
+ * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a
+ * component-agnostic result which may be used to render the hierarchy using any UI framework.
+ *
+ * See `README.md` for an example
+ *
+ * @see `useUnifiedSelectionTree`
+ * @beta
+ */
+function useTree(props) {
+    const { getNode: _, ...rest } = useTreeInternal(props);
+    return rest;
+}
+exports.useTree = useTree;
+/**
+ * A React hook that creates state for a tree component, that is integrated with unified selection
+ * through context provided by `UnifiedSelectionProvider`.
+ *
+ * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a
+ * component-agnostic result which may be used to render the hierarchy using any UI framework.
+ *
+ * See `README.md` for an example
+ *
+ * @see `useTree`
+ * @see `UnifiedSelectionProvider`
+ * @beta
+ */
+function useUnifiedSelectionTree({ imodelKey, sourceName, ...props }) {
+    const { getNode, ...rest } = useTreeInternal(props);
+    return { ...rest, ...(0, UseUnifiedSelection_1.useUnifiedTreeSelection)({ imodelKey, sourceName, getNode }) };
+}
+exports.useUnifiedSelectionTree = useUnifiedSelectionTree;
+function useTreeInternal({ imodelAccess, getHierarchyDefinition, getFilteredPaths, localizedStrings, }) {
+    const [state, setState] = (0, react_1.useState)({
+        model: { idToNode: new Map(), parentChildMap: new Map(), rootNode: { id: undefined, nodeData: undefined } },
+        rootNodes: undefined,
+    });
+    const [hierarchySource, setHierarchySource] = (0, react_1.useState)({ isFiltering: false });
+    const [actions] = (0, react_1.useState)(() => new TreeActions_1.TreeActions((model) => {
+        const rootNodes = model.parentChildMap.get(undefined) !== undefined ? generateTreeStructure(undefined, model) : undefined;
+        setState({
+            model,
+            rootNodes,
+        });
+    }));
+    const currentFormatter = (0, react_1.useRef)();
+    (0, react_1.useEffect)(() => {
+        const updateHierarchyProvider = (provider) => {
+            actions.setHierarchyProvider(provider);
+            actions.reloadTree(undefined);
+            setHierarchySource({ hierarchyProvider: provider, isFiltering: false });
+        };
+        const createProvider = (paths) => {
+            return (0, presentation_hierarchies_1.createHierarchyProvider)({
+                imodelAccess,
+                hierarchyDefinition: getHierarchyDefinition({ imodelAccess }),
+                localizedStrings,
+                formatter: currentFormatter.current,
+                filtering: paths !== undefined
+                    ? {
+                        paths,
+                    }
+                    : undefined,
+            });
+        };
+        const loadHierarchyProvider = async () => {
+            if (!getFilteredPaths) {
+                return createProvider(undefined);
+            }
+            setHierarchySource((prev) => ({ ...prev, isFiltering: true }));
+            try {
+                const filteredPaths = await getFilteredPaths({ imodelAccess });
+                return createProvider(filteredPaths);
+            }
+            catch {
+                return createProvider(undefined);
+            }
+        };
+        let disposed = false;
+        void (async () => {
+            const provider = await loadHierarchyProvider();
+            if (disposed) {
+                return;
+            }
+            updateHierarchyProvider(provider);
+        })();
+        return () => {
+            disposed = true;
+            actions.dispose();
+        };
+    }, [actions, imodelAccess, localizedStrings, getHierarchyDefinition, getFilteredPaths]);
+    const getNode = (0, react_1.useRef)((nodeId) => {
+        return actions.getNode(nodeId);
+    }).current;
+    const expandNode = (0, react_1.useRef)((nodeId, isExpanded) => {
+        actions.expandNode(nodeId, isExpanded);
+    }).current;
+    const reloadTree = (0, react_1.useRef)((options) => {
+        actions.reloadTree(options);
+    }).current;
+    const selectNodes = (0, react_1.useRef)((nodeIds, changeType) => {
+        actions.selectNodes(nodeIds, changeType);
+    }).current;
+    const isNodeSelected = (0, react_1.useCallback)((nodeId) => TreeModel_1.TreeModel.isNodeSelected(state.model, nodeId), [state]);
+    const setFormatter = (0, react_1.useCallback)((formatter) => {
+        currentFormatter.current = formatter;
+        // istanbul ignore if
+        if (!hierarchySource.hierarchyProvider) {
+            return;
+        }
+        hierarchySource.hierarchyProvider.setFormatter(formatter);
+        actions.reloadTree();
+    }, [hierarchySource.hierarchyProvider, actions]);
+    const getHierarchyLevelDetails = (0, react_1.useCallback)((nodeId) => {
+        const node = actions.getNode(nodeId);
+        const hierarchyProvider = hierarchySource.hierarchyProvider;
+        if (!hierarchyProvider || !node || (0, TreeModel_1.isTreeModelInfoNode)(node)) {
+            return undefined;
+        }
+        const hierarchyNode = node.nodeData;
+        if (hierarchyNode && presentation_hierarchies_1.HierarchyNode.isGroupingNode(hierarchyNode)) {
+            return undefined;
+        }
+        return {
+            hierarchyNode,
+            getInstanceKeysIterator: (props) => hierarchyProvider.getNodeInstanceKeys({
+                parentNode: hierarchyNode,
+                instanceFilter: props?.instanceFilter,
+                hierarchyLevelSizeLimit: props?.hierarchyLevelSizeLimit,
+            }),
+            instanceFilter: node.instanceFilter,
+            setInstanceFilter: (filter) => actions.setInstanceFilter(nodeId, filter),
+            sizeLimit: node.hierarchyLimit,
+            setSizeLimit: (value) => actions.setHierarchyLimit(nodeId, value),
+        };
+    }, [actions, hierarchySource.hierarchyProvider]);
+    return {
+        rootNodes: state.rootNodes,
+        isLoading: !!state.model.rootNode.isLoading || hierarchySource.isFiltering,
+        expandNode,
+        reloadTree,
+        selectNodes,
+        isNodeSelected,
+        getNode,
+        getHierarchyLevelDetails,
+        setFormatter,
+    };
+}
+function generateTreeStructure(parentNodeId, model) {
+    const currentChildren = model.parentChildMap.get(parentNodeId);
+    if (!currentChildren) {
+        return undefined;
+    }
+    return currentChildren
+        .map((childId) => model.idToNode.get(childId))
+        .filter((node) => !!node)
+        .map((node) => {
+        if ((0, TreeModel_1.isTreeModelHierarchyNode)(node)) {
+            const children = generateTreeStructure(node.id, model);
+            return {
+                ...toPresentationHierarchyNodeBase(node),
+                children: children ? children : node.children === true ? true : [],
+            };
+        }
+        if (node.type === "ResultSetTooLarge") {
+            return {
+                id: node.id,
+                parentNodeId,
+                type: node.type,
+                resultSetSizeLimit: node.resultSetSizeLimit,
+            };
+        }
+        if (node.type === "NoFilterMatches") {
+            return {
+                id: node.id,
+                parentNodeId,
+                type: node.type,
+            };
+        }
+        return {
+            id: node.id,
+            parentNodeId,
+            type: node.type,
+            message: node.message,
+        };
+    });
+}
+function toPresentationHierarchyNodeBase(node) {
+    return {
+        id: node.id,
+        label: node.label,
+        nodeData: node.nodeData,
+        isLoading: !!node.isLoading,
+        isExpanded: !!node.isExpanded,
+        isFilterable: !presentation_hierarchies_1.HierarchyNode.isGroupingNode(node.nodeData) && !!node.nodeData.supportsFiltering && node.children,
+        isFiltered: !!node.instanceFilter,
+        extendedData: node.nodeData.extendedData,
+    };
+}
+//# sourceMappingURL=UseTree.js.map

package/lib/cjs/presentation-hierarchies-react/UseTree.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseTree.js","sourceRoot":"","sources":["../../../src/presentation-hierarchies-react/UseTree.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAiE;AACjE,8EAQyC;AAEzC,wDAAqD;AACrD,oDAA0J;AAC1J,wEAAuG;AA6BvG;;;;;;;;;;GAUG;AACH,SAAgB,OAAO,CAAC,KAAmB;IACzC,MAAM,EAAE,OAAO,EAAE,CAAC,EAAE,GAAG,IAAI,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACvD,OAAO,IAAI,CAAC;AACd,CAAC;AAHD,0BAGC;AAED;;;;;;;;;;;;GAYG;AACH,SAAgB,uBAAuB,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,GAAG,KAAK,EAAgE;IACvI,MAAM,EAAE,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,eAAe,CAAC,KAAK,CAAC,CAAC;IACpD,OAAO,EAAE,GAAG,IAAI,EAAE,GAAG,IAAA,6CAAuB,EAAC,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;AACrF,CAAC;AAHD,0DAGC;AAsCD,SAAS,eAAe,CAAC,EACvB,YAAY,EACZ,sBAAsB,EACtB,gBAAgB,EAChB,gBAAgB,GACH;IACb,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAY;QAC5C,KAAK,EAAE,EAAE,QAAQ,EAAE,IAAI,GAAG,EAAE,EAAE,cAAc,EAAE,IAAI,GAAG,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,EAAE,SAAS,EAAE,QAAQ,EAAE,SAAS,EAAE,EAAE;QAC3G,SAAS,EAAE,SAAS;KACrB,CAAC,CAAC;IACH,MAAM,CAAC,eAAe,EAAE,kBAAkB,CAAC,GAAG,IAAA,gBAAQ,EAAkE,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;IAChJ,MAAM,CAAC,OAAO,CAAC,GAAG,IAAA,gBAAQ,EACxB,GAAG,EAAE,CACH,IAAI,yBAAW,CAAC,CAAC,KAAK,EAAE,EAAE;QACxB,MAAM,SAAS,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,KAAK,SAAS,CAAC,CAAC,CAAC,qBAAqB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAC1H,QAAQ,CAAC;YACP,KAAK;YACL,SAAS;SACV,CAAC,CAAC;IACL,CAAC,CAAC,CACL,CAAC;IACF,MAAM,gBAAgB,GAAG,IAAA,cAAM,GAA4B,CAAC;IAE5D,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,MAAM,uBAAuB,GAAG,CAAC,QAA2B,EAAE,EAAE;YAC9D,OAAO,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;YACvC,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,CAAC;YAC9B,kBAAkB,CAAC,EAAE,iBAAiB,EAAE,QAAQ,EAAE,WAAW,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC,CAAC;QAEF,MAAM,cAAc,GAAG,CAAC,KAAiD,EAAE,EAAE;YAC3E,OAAO,IAAA,kDAAuB,EAAC;gBAC7B,YAAY;gBACZ,mBAAmB,EAAE,sBAAsB,CAAC,EAAE,YAAY,EAAE,CAAC;gBAC7D,gBAAgB;gBAChB,SAAS,EAAE,gBAAgB,CAAC,OAAO;gBACnC,SAAS,EACP,KAAK,KAAK,SAAS;oBACjB,CAAC,CAAC;wBACE,KAAK;qBACN;oBACH,CAAC,CAAC,SAAS;aAChB,CAAC,CAAC;QACL,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,KAAK,IAAI,EAAE;YACvC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO,cAAc,CAAC,SAAS,CAAC,CAAC;YACnC,CAAC;YAED,kBAAkB,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC/D,IAAI,CAAC;gBACH,MAAM,aAAa,GAAG,MAAM,gBAAgB,CAAC,EAAE,YAAY,EAAE,CAAC,CAAC;gBAC/D,OAAO,cAAc,CAAC,aAAa,CAAC,CAAC;YACvC,CAAC;YAAC,MAAM,CAAC;gBACP,OAAO,cAAc,CAAC,SAAS,CAAC,CAAC;YACnC,CAAC;QACH,CAAC,CAAC;QAEF,IAAI,QAAQ,GAAG,KAAK,CAAC;QACrB,KAAK,CAAC,KAAK,IAAI,EAAE;YACf,MAAM,QAAQ,GAAG,MAAM,qBAAqB,EAAE,CAAC;YAC/C,IAAI,QAAQ,EAAE,CAAC;gBACb,OAAO;YACT,CAAC;YACD,uBAAuB,CAAC,QAAQ,CAAC,CAAC;QACpC,CAAC,CAAC,EAAE,CAAC;QAEL,OAAO,GAAG,EAAE;YACV,QAAQ,GAAG,IAAI,CAAC;YAChB,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,OAAO,EAAE,YAAY,EAAE,gBAAgB,EAAE,sBAAsB,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAExF,MAAM,OAAO,GAAG,IAAA,cAAM,EAAC,CAAC,MAAc,EAAE,EAAE;QACxC,OAAO,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;IACjC,CAAC,CAAC,CAAC,OAAO,CAAC;IAEX,MAAM,UAAU,GAAG,IAAA,cAAM,EAAC,CAAC,MAAc,EAAE,UAAmB,EAAE,EAAE;QAChE,OAAO,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;IACzC,CAAC,CAAC,CAAC,OAAO,CAAC;IAEX,MAAM,UAAU,GAAG,IAAA,cAAM,EAAC,CAAC,OAAoC,EAAE,EAAE;QACjE,OAAO,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;IAC9B,CAAC,CAAC,CAAC,OAAO,CAAC;IAEX,MAAM,WAAW,GAAG,IAAA,cAAM,EAAC,CAAC,OAAsB,EAAE,UAA+B,EAAE,EAAE;QACrF,OAAO,CAAC,WAAW,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;IAC3C,CAAC,CAAC,CAAC,OAAO,CAAC;IAEX,MAAM,cAAc,GAAG,IAAA,mBAAW,EAAC,CAAC,MAAc,EAAE,EAAE,CAAC,qBAAS,CAAC,cAAc,CAAC,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC;IAE/G,MAAM,YAAY,GAAG,IAAA,mBAAW,EAC9B,CAAC,SAA+C,EAAE,EAAE;QAClD,gBAAgB,CAAC,OAAO,GAAG,SAAS,CAAC;QACrC,qBAAqB;QACrB,IAAI,CAAC,eAAe,CAAC,iBAAiB,EAAE,CAAC;YACvC,OAAO;QACT,CAAC;QAED,eAAe,CAAC,iBAAiB,CAAC,YAAY,CAAC,SAAS,CAAC,CAAC;QAC1D,OAAO,CAAC,UAAU,EAAE,CAAC;IACvB,CAAC,EACD,CAAC,eAAe,CAAC,iBAAiB,EAAE,OAAO,CAAC,CAC7C,CAAC;IAEF,MAAM,wBAAwB,GAAG,IAAA,mBAAW,EAC1C,CAAC,MAAM,EAAE,EAAE;QACT,MAAM,IAAI,GAAG,OAAO,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;QACrC,MAAM,iBAAiB,GAAG,eAAe,CAAC,iBAAiB,CAAC;QAC5D,IAAI,CAAC,iBAAiB,IAAI,CAAC,IAAI,IAAI,IAAA,+BAAmB,EAAC,IAAI,CAAC,EAAE,CAAC;YAC7D,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,MAAM,aAAa,GAAG,IAAI,CAAC,QAAQ,CAAC;QACpC,IAAI,aAAa,IAAI,wCAAa,CAAC,cAAc,CAAC,aAAa,CAAC,EAAE,CAAC;YACjE,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO;YACL,aAAa;YACb,uBAAuB,EAAE,CAAC,KAAK,EAAE,EAAE,CACjC,iBAAiB,CAAC,mBAAmB,CAAC;gBACpC,UAAU,EAAE,aAAa;gBACzB,cAAc,EAAE,KAAK,EAAE,cAAc;gBACrC,uBAAuB,EAAE,KAAK,EAAE,uBAAuB;aACxD,CAAC;YACJ,cAAc,EAAE,IAAI,CAAC,cAAc;YACnC,iBAAiB,EAAE,CAAC,MAAM,EAAE,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC;YACxE,SAAS,EAAE,IAAI,CAAC,cAAc;YAC9B,YAAY,EAAE,CAAC,KAAK,EAAE,EAAE,CAAC,OAAO,CAAC,iBAAiB,CAAC,MAAM,EAAE,KAAK,CAAC;SAClE,CAAC;IACJ,CAAC,EACD,CAAC,OAAO,EAAE,eAAe,CAAC,iBAAiB,CAAC,CAC7C,CAAC;IAEF,OAAO;QACL,SAAS,EAAE,KAAK,CAAC,SAAS;QAC1B,SAAS,EAAE,CAAC,CAAC,KAAK,CAAC,KAAK,CAAC,QAAQ,CAAC,SAAS,IAAI,eAAe,CAAC,WAAW;QAC1E,UAAU;QACV,UAAU;QACV,WAAW;QACX,cAAc;QACd,OAAO;QACP,wBAAwB;QACxB,YAAY;KACb,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,YAAgC,EAAE,KAAgB;IAC/E,MAAM,eAAe,GAAG,KAAK,CAAC,cAAc,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC;IAC/D,IAAI,CAAC,eAAe,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,OAAO,eAAe;SACnB,GAAG,CAAC,CAAC,OAAO,EAAE,EAAE,CAAC,KAAK,CAAC,QAAQ,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;SAC7C,MAAM,CAAC,CAAC,IAAI,EAAyB,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC;SAC/C,GAAG,CAAuB,CAAC,IAAI,EAAE,EAAE;QAClC,IAAI,IAAA,oCAAwB,EAAC,IAAI,CAAC,EAAE,CAAC;YACnC,MAAM,QAAQ,GAAG,qBAAqB,CAAC,IAAI,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;YACvD,OAAO;gBACL,GAAG,+BAA+B,CAAC,IAAI,CAAC;gBACxC,QAAQ,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,KAAK,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE;aACnE,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACtC,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,YAAY;gBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;gBACf,kBAAkB,EAAE,IAAI,CAAC,kBAAkB;aAC5C,CAAC;QACJ,CAAC;QAED,IAAI,IAAI,CAAC,IAAI,KAAK,iBAAiB,EAAE,CAAC;YACpC,OAAO;gBACL,EAAE,EAAE,IAAI,CAAC,EAAE;gBACX,YAAY;gBACZ,IAAI,EAAE,IAAI,CAAC,IAAI;aAChB,CAAC;QACJ,CAAC;QAED,OAAO;YACL,EAAE,EAAE,IAAI,CAAC,EAAE;YACX,YAAY;YACZ,IAAI,EAAE,IAAI,CAAC,IAAI;YACf,OAAO,EAAE,IAAI,CAAC,OAAO;SACtB,CAAC;IACJ,CAAC,CAAC,CAAC;AACP,CAAC;AAED,SAAS,+BAA+B,CAAC,IAA4B;IACnE,OAAO;QACL,EAAE,EAAE,IAAI,CAAC,EAAE;QACX,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,QAAQ,EAAE,IAAI,CAAC,QAAQ;QACvB,SAAS,EAAE,CAAC,CAAC,IAAI,CAAC,SAAS;QAC3B,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,UAAU;QAC7B,YAAY,EAAE,CAAC,wCAAa,CAAC,cAAc,CAAC,IAAI,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,iBAAiB,IAAI,IAAI,CAAC,QAAQ;QAChH,UAAU,EAAE,CAAC,CAAC,IAAI,CAAC,cAAc;QACjC,YAAY,EAAE,IAAI,CAAC,QAAQ,CAAC,YAAY;KACzC,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useRef, useState } from \"react\";\nimport {\n  createHierarchyProvider,\n  GenericInstanceFilter,\n  HierarchyDefinition,\n  HierarchyNode,\n  HierarchyNodeIdentifiersPath,\n  HierarchyProvider,\n  LimitingECSqlQueryExecutor,\n} from \"@itwin/presentation-hierarchies\";\nimport { ECClassHierarchyInspector, ECSchemaProvider, InstanceKey, IPrimitiveValueFormatter } from \"@itwin/presentation-shared\";\nimport { TreeActions } from \"./internal/TreeActions\";\nimport { isTreeModelHierarchyNode, isTreeModelInfoNode, TreeModel, TreeModelHierarchyNode, TreeModelNode, TreeModelRootNode } from \"./internal/TreeModel\";\nimport { useUnifiedTreeSelection, UseUnifiedTreeSelectionProps } from \"./internal/UseUnifiedSelection\";\nimport { PresentationHierarchyNode, PresentationTreeNode } from \"./TreeNode\";\nimport { SelectionChangeType } from \"./UseSelectionHandler\";\n\n/**\n * A data structure that contains information about a single hierarchy level.\n * @beta\n */\nexport interface HierarchyLevelDetails {\n  /** The parent node whose hierarchy level's information is contained in this data structure */\n  hierarchyNode: HierarchyNode | undefined;\n\n  /** A function to get instance keys of the hierarchy level. */\n  getInstanceKeysIterator: (props?: {\n    instanceFilter?: GenericInstanceFilter;\n    hierarchyLevelSizeLimit?: number | \"unbounded\";\n  }) => AsyncIterableIterator<InstanceKey>;\n\n  /** Get the limit of how many nodes can be loaded in this hierarchy level. */\n  sizeLimit?: number | \"unbounded\";\n  /** Set the limit of how many nodes can be loaded in this hierarchy level. */\n  setSizeLimit: (value: undefined | number | \"unbounded\") => void;\n\n  /** Get the active instance filter applied to this hierarchy level. */\n  instanceFilter?: GenericInstanceFilter;\n  /** Set the instance filter to apply to this hierarchy level */\n  setInstanceFilter: (filter: GenericInstanceFilter | undefined) => void;\n}\n\n/**\n * A React hook that creates state for a tree component.\n *\n * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a\n * component-agnostic result which may be used to render the hierarchy using any UI framework.\n *\n * See `README.md` for an example\n *\n * @see `useUnifiedSelectionTree`\n * @beta\n */\nexport function useTree(props: UseTreeProps): UseTreeResult {\n  const { getNode: _, ...rest } = useTreeInternal(props);\n  return rest;\n}\n\n/**\n * A React hook that creates state for a tree component, that is integrated with unified selection\n * through context provided by `UnifiedSelectionProvider`.\n *\n * The hook uses `@itwin/presentation-hierarchies` package to load the hierarchy data and returns a\n * component-agnostic result which may be used to render the hierarchy using any UI framework.\n *\n * See `README.md` for an example\n *\n * @see `useTree`\n * @see `UnifiedSelectionProvider`\n * @beta\n */\nexport function useUnifiedSelectionTree({ imodelKey, sourceName, ...props }: UseTreeProps & Omit<UseUnifiedTreeSelectionProps, \"getNode\">): UseTreeResult {\n  const { getNode, ...rest } = useTreeInternal(props);\n  return { ...rest, ...useUnifiedTreeSelection({ imodelKey, sourceName, getNode }) };\n}\n\ntype IModelAccess = ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector;\n\ninterface GetFilteredPathsProps {\n  imodelAccess: IModelAccess;\n}\n\ninterface UseTreeProps {\n  imodelAccess: ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector;\n  getHierarchyDefinition: (props: { imodelAccess: IModelAccess }) => HierarchyDefinition;\n  getFilteredPaths?: (props: GetFilteredPathsProps) => Promise<HierarchyNodeIdentifiersPath[] | undefined>;\n  localizedStrings?: Parameters<typeof createHierarchyProvider>[0][\"localizedStrings\"];\n}\n\ninterface UseTreeResult {\n  /**\n   * Array containing root tree nodes. It is `undefined` on initial render until any nodes are loaded.\n   */\n  rootNodes: PresentationTreeNode[] | undefined;\n  /**\n   * Specifies whether tree is loading or not. It is set to `true` when initial tree load is in progress\n   * or tree is reloading.\n   */\n  isLoading: boolean;\n  reloadTree: (options?: { discardState?: boolean }) => void;\n  expandNode: (nodeId: string, isExpanded: boolean) => void;\n  selectNodes: (nodeIds: Array<string>, changeType: SelectionChangeType) => void;\n  isNodeSelected: (nodeId: string) => boolean;\n  getHierarchyLevelDetails: (nodeId: string | undefined) => HierarchyLevelDetails | undefined;\n  setFormatter: (formatter: IPrimitiveValueFormatter | undefined) => void;\n}\n\ninterface TreeState {\n  model: TreeModel;\n  rootNodes: Array<PresentationTreeNode> | undefined;\n}\n\nfunction useTreeInternal({\n  imodelAccess,\n  getHierarchyDefinition,\n  getFilteredPaths,\n  localizedStrings,\n}: UseTreeProps): UseTreeResult & { getNode: (nodeId: string) => TreeModelRootNode | TreeModelNode | undefined } {\n  const [state, setState] = useState<TreeState>({\n    model: { idToNode: new Map(), parentChildMap: new Map(), rootNode: { id: undefined, nodeData: undefined } },\n    rootNodes: undefined,\n  });\n  const [hierarchySource, setHierarchySource] = useState<{ hierarchyProvider?: HierarchyProvider; isFiltering: boolean }>({ isFiltering: false });\n  const [actions] = useState<TreeActions>(\n    () =>\n      new TreeActions((model) => {\n        const rootNodes = model.parentChildMap.get(undefined) !== undefined ? generateTreeStructure(undefined, model) : undefined;\n        setState({\n          model,\n          rootNodes,\n        });\n      }),\n  );\n  const currentFormatter = useRef<IPrimitiveValueFormatter>();\n\n  useEffect(() => {\n    const updateHierarchyProvider = (provider: HierarchyProvider) => {\n      actions.setHierarchyProvider(provider);\n      actions.reloadTree(undefined);\n      setHierarchySource({ hierarchyProvider: provider, isFiltering: false });\n    };\n\n    const createProvider = (paths: HierarchyNodeIdentifiersPath[] | undefined) => {\n      return createHierarchyProvider({\n        imodelAccess,\n        hierarchyDefinition: getHierarchyDefinition({ imodelAccess }),\n        localizedStrings,\n        formatter: currentFormatter.current,\n        filtering:\n          paths !== undefined\n            ? {\n                paths,\n              }\n            : undefined,\n      });\n    };\n\n    const loadHierarchyProvider = async () => {\n      if (!getFilteredPaths) {\n        return createProvider(undefined);\n      }\n\n      setHierarchySource((prev) => ({ ...prev, isFiltering: true }));\n      try {\n        const filteredPaths = await getFilteredPaths({ imodelAccess });\n        return createProvider(filteredPaths);\n      } catch {\n        return createProvider(undefined);\n      }\n    };\n\n    let disposed = false;\n    void (async () => {\n      const provider = await loadHierarchyProvider();\n      if (disposed) {\n        return;\n      }\n      updateHierarchyProvider(provider);\n    })();\n\n    return () => {\n      disposed = true;\n      actions.dispose();\n    };\n  }, [actions, imodelAccess, localizedStrings, getHierarchyDefinition, getFilteredPaths]);\n\n  const getNode = useRef((nodeId: string) => {\n    return actions.getNode(nodeId);\n  }).current;\n\n  const expandNode = useRef((nodeId: string, isExpanded: boolean) => {\n    actions.expandNode(nodeId, isExpanded);\n  }).current;\n\n  const reloadTree = useRef((options?: { discardState?: boolean }) => {\n    actions.reloadTree(options);\n  }).current;\n\n  const selectNodes = useRef((nodeIds: Array<string>, changeType: SelectionChangeType) => {\n    actions.selectNodes(nodeIds, changeType);\n  }).current;\n\n  const isNodeSelected = useCallback((nodeId: string) => TreeModel.isNodeSelected(state.model, nodeId), [state]);\n\n  const setFormatter = useCallback(\n    (formatter: IPrimitiveValueFormatter | undefined) => {\n      currentFormatter.current = formatter;\n      // istanbul ignore if\n      if (!hierarchySource.hierarchyProvider) {\n        return;\n      }\n\n      hierarchySource.hierarchyProvider.setFormatter(formatter);\n      actions.reloadTree();\n    },\n    [hierarchySource.hierarchyProvider, actions],\n  );\n\n  const getHierarchyLevelDetails = useCallback<UseTreeResult[\"getHierarchyLevelDetails\"]>(\n    (nodeId) => {\n      const node = actions.getNode(nodeId);\n      const hierarchyProvider = hierarchySource.hierarchyProvider;\n      if (!hierarchyProvider || !node || isTreeModelInfoNode(node)) {\n        return undefined;\n      }\n      const hierarchyNode = node.nodeData;\n      if (hierarchyNode && HierarchyNode.isGroupingNode(hierarchyNode)) {\n        return undefined;\n      }\n\n      return {\n        hierarchyNode,\n        getInstanceKeysIterator: (props) =>\n          hierarchyProvider.getNodeInstanceKeys({\n            parentNode: hierarchyNode,\n            instanceFilter: props?.instanceFilter,\n            hierarchyLevelSizeLimit: props?.hierarchyLevelSizeLimit,\n          }),\n        instanceFilter: node.instanceFilter,\n        setInstanceFilter: (filter) => actions.setInstanceFilter(nodeId, filter),\n        sizeLimit: node.hierarchyLimit,\n        setSizeLimit: (value) => actions.setHierarchyLimit(nodeId, value),\n      };\n    },\n    [actions, hierarchySource.hierarchyProvider],\n  );\n\n  return {\n    rootNodes: state.rootNodes,\n    isLoading: !!state.model.rootNode.isLoading || hierarchySource.isFiltering,\n    expandNode,\n    reloadTree,\n    selectNodes,\n    isNodeSelected,\n    getNode,\n    getHierarchyLevelDetails,\n    setFormatter,\n  };\n}\n\nfunction generateTreeStructure(parentNodeId: string | undefined, model: TreeModel): Array<PresentationTreeNode> | undefined {\n  const currentChildren = model.parentChildMap.get(parentNodeId);\n  if (!currentChildren) {\n    return undefined;\n  }\n\n  return currentChildren\n    .map((childId) => model.idToNode.get(childId))\n    .filter((node): node is TreeModelNode => !!node)\n    .map<PresentationTreeNode>((node) => {\n      if (isTreeModelHierarchyNode(node)) {\n        const children = generateTreeStructure(node.id, model);\n        return {\n          ...toPresentationHierarchyNodeBase(node),\n          children: children ? children : node.children === true ? true : [],\n        };\n      }\n\n      if (node.type === \"ResultSetTooLarge\") {\n        return {\n          id: node.id,\n          parentNodeId,\n          type: node.type,\n          resultSetSizeLimit: node.resultSetSizeLimit,\n        };\n      }\n\n      if (node.type === \"NoFilterMatches\") {\n        return {\n          id: node.id,\n          parentNodeId,\n          type: node.type,\n        };\n      }\n\n      return {\n        id: node.id,\n        parentNodeId,\n        type: node.type,\n        message: node.message,\n      };\n    });\n}\n\nfunction toPresentationHierarchyNodeBase(node: TreeModelHierarchyNode): Omit<PresentationHierarchyNode, \"children\"> {\n  return {\n    id: node.id,\n    label: node.label,\n    nodeData: node.nodeData,\n    isLoading: !!node.isLoading,\n    isExpanded: !!node.isExpanded,\n    isFilterable: !HierarchyNode.isGroupingNode(node.nodeData) && !!node.nodeData.supportsFiltering && node.children,\n    isFiltered: !!node.instanceFilter,\n    extendedData: node.nodeData.extendedData,\n  };\n}\n"]}

package/lib/cjs/presentation-hierarchies-react/internal/TreeActions.d.ts

@@ -0,0 +1,27 @@
+import { GenericInstanceFilter, HierarchyProvider } from "@itwin/presentation-hierarchies";
+import { SelectionChangeType } from "../UseSelectionHandler";
+import { TreeModel, TreeModelNode, TreeModelRootNode } from "./TreeModel";
+/** @internal */
+export declare class TreeActions {
+    private _onModelChanged;
+    private _loader;
+    private _currentModel;
+    private _disposed;
+    constructor(_onModelChanged: (model: TreeModel) => void, seed?: TreeModel);
+    private updateTreeModel;
+    private handleLoadedHierarchy;
+    private onLoadingComplete;
+    private loadNodes;
+    private reloadSubTree;
+    dispose(): void;
+    setHierarchyProvider(provider?: HierarchyProvider): void;
+    getNode(nodeId: string | undefined): TreeModelNode | TreeModelRootNode | undefined;
+    selectNodes(nodeIds: Array<string>, changeType: SelectionChangeType): void;
+    expandNode(nodeId: string, isExpanded: boolean): void;
+    setHierarchyLimit(nodeId: string | undefined, limit?: number | "unbounded"): void;
+    setInstanceFilter(nodeId: string | undefined, filter?: GenericInstanceFilter): void;
+    reloadTree(options?: {
+        discardState?: boolean;
+    }): void;
+}
+//# sourceMappingURL=TreeActions.d.ts.map

package/lib/cjs/presentation-hierarchies-react/internal/TreeActions.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"TreeActions.d.ts","sourceRoot":"","sources":["../../../../src/presentation-hierarchies-react/internal/TreeActions.ts"],"names":[],"mappings":"AAOA,OAAO,EAAE,qBAAqB,EAAiB,iBAAiB,EAAE,MAAM,iCAAiC,CAAC;AAC1G,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAE7D,OAAO,EAAiD,SAAS,EAA0B,aAAa,EAAE,iBAAiB,EAAE,MAAM,aAAa,CAAC;AAKjJ,gBAAgB;AAChB,qBAAa,WAAW;IAMpB,OAAO,CAAC,eAAe;IALzB,OAAO,CAAC,OAAO,CAAc;IAC7B,OAAO,CAAC,aAAa,CAAY;IACjC,OAAO,CAAC,SAAS,CAAuB;gBAG9B,eAAe,EAAE,CAAC,KAAK,EAAE,SAAS,KAAK,IAAI,EACnD,IAAI,CAAC,EAAE,SAAS;IAUlB,OAAO,CAAC,eAAe;IAUvB,OAAO,CAAC,qBAAqB;IAM7B,OAAO,CAAC,iBAAiB;IAMzB,OAAO,CAAC,SAAS;IAyBjB,OAAO,CAAC,aAAa;IA8Cd,OAAO;IAIP,oBAAoB,CAAC,QAAQ,CAAC,EAAE,iBAAiB;IAIjD,OAAO,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,GAAG,aAAa,GAAG,iBAAiB,GAAG,SAAS;IAIlF,WAAW,CAAC,OAAO,EAAE,KAAK,CAAC,MAAM,CAAC,EAAE,UAAU,EAAE,mBAAmB;IAMnE,UAAU,CAAC,MAAM,EAAE,MAAM,EAAE,UAAU,EAAE,OAAO;IAa9C,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,EAAE,KAAK,CAAC,EAAE,MAAM,GAAG,WAAW;IAc1E,iBAAiB,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,EAAE,MAAM,CAAC,EAAE,qBAAqB;IAc5E,UAAU,CAAC,OAAO,CAAC,EAAE;QAAE,YAAY,CAAC,EAAE,OAAO,CAAA;KAAE;CAQvD"}

package/lib/cjs/presentation-hierarchies-react/internal/TreeActions.js

@@ -0,0 +1,236 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.TreeActions = void 0;
+const immer_1 = require("immer");
+const rxjs_1 = require("rxjs");
+const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
+const TreeLoader_1 = require("./TreeLoader");
+const TreeModel_1 = require("./TreeModel");
+const Utils_1 = require("./Utils");
+(0, immer_1.enableMapSet)();
+/** @internal */
+class TreeActions {
+    _onModelChanged;
+    _loader;
+    _currentModel;
+    _disposed = new rxjs_1.Subject();
+    constructor(_onModelChanged, seed) {
+        this._onModelChanged = _onModelChanged;
+        this._loader = new NoopTreeLoader();
+        this._currentModel = seed ?? /* istanbul ignore next */ {
+            idToNode: new Map(),
+            parentChildMap: new Map(),
+            rootNode: { id: undefined, nodeData: undefined },
+        };
+    }
+    updateTreeModel(updater) {
+        const newModel = (0, immer_1.produce)(this._currentModel, updater);
+        if (this._currentModel === newModel) {
+            return;
+        }
+        this._currentModel = newModel;
+        this._onModelChanged(this._currentModel);
+    }
+    handleLoadedHierarchy(parentId, loadedHierarchy) {
+        this.updateTreeModel((model) => {
+            TreeModel_1.TreeModel.addHierarchyPart(model, parentId, loadedHierarchy);
+        });
+    }
+    onLoadingComplete(parentId) {
+        this.updateTreeModel((model) => {
+            TreeModel_1.TreeModel.setIsLoading(model, parentId, false);
+        });
+    }
+    loadNodes(parentId, ignoreCache) {
+        const parentNode = this._currentModel.idToNode.get(parentId);
+        // istanbul ignore if
+        if (!parentNode || !(0, TreeModel_1.isTreeModelHierarchyNode)(parentNode)) {
+            return;
+        }
+        this._loader
+            .loadNodes({
+            parent: parentNode,
+            getHierarchyLevelOptions: (node) => createHierarchyLevelOptions(this._currentModel, getNonGroupedParentId(node)),
+            shouldLoadChildren: (node) => !!node.nodeData.autoExpand,
+            ignoreCache,
+        })
+            .pipe(collectTreePartsUntil(this._disposed))
+            .subscribe({
+            next: (loadedHierarchy) => {
+                this.handleLoadedHierarchy(parentId, loadedHierarchy);
+            },
+            complete: () => {
+                this.onLoadingComplete(parentId);
+            },
+        });
+    }
+    reloadSubTree(parentId, oldModel, options) {
+        const currModel = this._currentModel;
+        const expandedNodes = !!options?.discardState ? [] : collectNodes(parentId, oldModel, (node) => node.isExpanded === true);
+        const collapsedNodes = !!options?.discardState ? [] : collectNodes(parentId, oldModel, (node) => node.isExpanded === false);
+        const getHierarchyLevelOptions = (node) => {
+            if (!!options?.discardState) {
+                return { instanceFilter: undefined, hierarchyLevelSizeLimit: undefined };
+            }
+            const filteredNodeId = getNonGroupedParentId(node);
+            return createHierarchyLevelOptions(filteredNodeId === parentId ? currModel : oldModel, filteredNodeId);
+        };
+        const shouldLoadChildren = (node) => {
+            if (expandedNodes.findIndex((expandedNode) => (0, Utils_1.sameNodes)(expandedNode.nodeData, node.nodeData)) !== -1) {
+                return true;
+            }
+            if (collapsedNodes.findIndex((collapsedNode) => (0, Utils_1.sameNodes)(collapsedNode.nodeData, node.nodeData)) !== -1) {
+                return false;
+            }
+            return !!node.nodeData.autoExpand;
+        };
+        const buildNode = (node) => (!!options?.discardState || node.id === parentId ? node : addAttributes(node, oldModel));
+        const rootNode = parentId !== undefined ? this.getNode(parentId) : currModel.rootNode;
+        // istanbul ignore if
+        if (!rootNode || (0, TreeModel_1.isTreeModelInfoNode)(rootNode)) {
+            return;
+        }
+        if (parentId === undefined) {
+            // cancel all ongoing requests
+            this._disposed.next();
+        }
+        this._loader
+            .loadNodes({ parent: rootNode, getHierarchyLevelOptions, shouldLoadChildren, buildNode })
+            .pipe(collectTreePartsUntil(this._disposed, !!options?.discardState ? undefined : { ...currModel.rootNode }))
+            .subscribe({
+            next: (newModel) => {
+                this.handleLoadedHierarchy(parentId, newModel);
+            },
+            complete: () => {
+                this.onLoadingComplete(parentId);
+            },
+        });
+    }
+    dispose() {
+        this._disposed.next();
+    }
+    setHierarchyProvider(provider) {
+        this._loader = provider ? new TreeLoader_1.TreeLoader(provider) : /* istanbul ignore next */ new NoopTreeLoader();
+    }
+    getNode(nodeId) {
+        return TreeModel_1.TreeModel.getNode(this._currentModel, nodeId);
+    }
+    selectNodes(nodeIds, changeType) {
+        this.updateTreeModel((model) => {
+            TreeModel_1.TreeModel.selectNodes(model, nodeIds, changeType);
+        });
+    }
+    expandNode(nodeId, isExpanded) {
+        let childrenAction = "none";
+        this.updateTreeModel((model) => {
+            childrenAction = TreeModel_1.TreeModel.expandNode(model, nodeId, isExpanded);
+        });
+        if (childrenAction === "none") {
+            return;
+        }
+        this.loadNodes(nodeId, childrenAction === "reloadChildren");
+    }
+    setHierarchyLimit(nodeId, limit) {
+        const oldModel = this._currentModel;
+        let loadChildren = false;
+        this.updateTreeModel((model) => {
+            loadChildren = TreeModel_1.TreeModel.setHierarchyLimit(model, nodeId, limit);
+        });
+        if (!loadChildren) {
+            return;
+        }
+        this.reloadSubTree(nodeId, oldModel);
+    }
+    setInstanceFilter(nodeId, filter) {
+        const oldModel = this._currentModel;
+        let loadChildren = false;
+        this.updateTreeModel((model) => {
+            loadChildren = TreeModel_1.TreeModel.setInstanceFilter(model, nodeId, filter);
+        });
+        if (!loadChildren) {
+            return;
+        }
+        this.reloadSubTree(nodeId, oldModel);
+    }
+    reloadTree(options) {
+        const oldModel = this._currentModel;
+        this.updateTreeModel((model) => {
+            model.rootNode.isLoading = true;
+        });
+        this.reloadSubTree(undefined, oldModel, { ...options });
+    }
+}
+exports.TreeActions = TreeActions;
+function collectTreePartsUntil(untilNotifier, rootNode) {
+    return (source) => source.pipe((0, rxjs_1.reduce)((treeModel, loadedPart) => {
+        addNodesToModel(treeModel, loadedPart);
+        return treeModel;
+    }, {
+        idToNode: new Map(),
+        parentChildMap: new Map(),
+        rootNode: rootNode ?? { id: undefined, nodeData: undefined },
+    }), (0, rxjs_1.takeUntil)(untilNotifier));
+}
+function addNodesToModel(model, hierarchyPart) {
+    model.parentChildMap.set(hierarchyPart.parentId, hierarchyPart.loadedNodes.map((node) => node.id));
+    for (const node of hierarchyPart.loadedNodes) {
+        model.idToNode.set(node.id, node);
+    }
+    const parentNode = hierarchyPart.parentId ? model.idToNode.get(hierarchyPart.parentId) : undefined;
+    if (parentNode && (0, TreeModel_1.isTreeModelHierarchyNode)(parentNode)) {
+        parentNode.isExpanded = true;
+    }
+}
+function addAttributes(node, oldModel) {
+    const oldNode = oldModel.idToNode.get(node.id);
+    if (oldNode && (0, TreeModel_1.isTreeModelHierarchyNode)(oldNode)) {
+        node.hierarchyLimit = oldNode.hierarchyLimit;
+        node.instanceFilter = oldNode.instanceFilter;
+        node.isSelected = oldNode.isSelected;
+    }
+    return node;
+}
+function collectNodes(parentId, model, pred) {
+    const currentChildren = model.parentChildMap.get(parentId);
+    if (!currentChildren) {
+        return [];
+    }
+    if (parentId === undefined) {
+        return currentChildren.flatMap((child) => collectNodes(child, model, pred));
+    }
+    const currNode = model.idToNode.get(parentId);
+    if (!currNode || !(0, TreeModel_1.isTreeModelHierarchyNode)(currNode) || !pred(currNode)) {
+        return [];
+    }
+    return [currNode, ...currentChildren.flatMap((child) => collectNodes(child, model, pred))];
+}
+function getNonGroupedParentId(node) {
+    if (!node.nodeData || !presentation_hierarchies_1.HierarchyNode.isGroupingNode(node.nodeData)) {
+        return node.id;
+    }
+    if (!node.nodeData.nonGroupingAncestor) {
+        return undefined;
+    }
+    return (0, Utils_1.createNodeId)(node.nodeData.nonGroupingAncestor);
+}
+function createHierarchyLevelOptions(model, nodeId) {
+    if (nodeId === undefined) {
+        return { instanceFilter: model.rootNode.instanceFilter, hierarchyLevelSizeLimit: model.rootNode.hierarchyLimit };
+    }
+    const modelNode = model.idToNode.get(nodeId);
+    if (!modelNode || (0, TreeModel_1.isTreeModelInfoNode)(modelNode)) {
+        return { instanceFilter: undefined, hierarchyLevelSizeLimit: undefined };
+    }
+    return { instanceFilter: modelNode.instanceFilter, hierarchyLevelSizeLimit: modelNode.hierarchyLimit };
+}
+// istanbul ignore next
+class NoopTreeLoader {
+    loadNodes() {
+        return rxjs_1.EMPTY;
+    }
+}
+//# sourceMappingURL=TreeActions.js.map