@itwin/tree-widget-react 4.0.0-alpha.2 → 4.0.0-alpha.20

package/lib/esm/tree-widget-react/components/trees/common/components/TreeNodeVisibilityButton.js

@@ -1,33 +1,49 @@
+import { jsx as _jsx } from "react/jsx-runtime";
 /*---------------------------------------------------------------------------------------------
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import "./TreeNodeVisibilityButton.css";
-const visibilityShowIcon = new URL("@itwin/itwinui-icons/visibility-show.svg", import.meta.url).href;
-const visibilityHideIcon = new URL("@itwin/itwinui-icons/visibility-hide.svg", import.meta.url).href;
-const visibilityPartialIcon = new URL("@itwin/itwinui-icons/visibility-partial.svg", import.meta.url).href;
-/** @internal */
-export function createVisibilityAction({ getVisibilityButtonState, onVisibilityButtonClick, }) {
-    return (node) => {
-        const state = getVisibilityButtonState(node);
-        const getIcon = () => {
-            switch (state.state) {
-                case "visible":
-                    return visibilityShowIcon;
-                case "hidden":
-                    return visibilityHideIcon;
-                case "partial":
-                    return visibilityPartialIcon;
-            }
-        };
-        return {
-            label: state.tooltip ?? "Determining visibility...",
-            action: () => {
-                onVisibilityButtonClick(node, state.state);
-            },
-            show: state.state !== "visible" ? true : undefined,
-            icon: getIcon(),
-        };
+import { createContext, memo, useContext, useMemo } from "react";
+import { TreeActionBase } from "@itwin/presentation-hierarchies-react";
+import visibilityHideSvg from "@stratakit/icons/visibility-hide.svg";
+import visibilityPartialSvg from "@stratakit/icons/visibility-partial.svg";
+import visibilityShowSvg from "@stratakit/icons/visibility-show.svg";
+import { TreeWidget } from "../../../../TreeWidget.js";
+import { createTooltip } from "../internal/Tooltip.js";
+/**
+ * React component that renders a visibility action for a tree item.
+ * Should be used with `VisibilityTreeRenderer`.
+ * @beta
+ */
+export const VisibilityAction = memo(function VisibilityAction({ node, ...actionAttributes }) {
+    const context = useVisibilityContext();
+    const state = context?.getVisibilityButtonState(node);
+    if (!context || !state || ("isDisabled" in state && state.isDisabled) || "isLoading" in state) {
+        return (_jsx(TreeActionBase, { ...actionAttributes, label: TreeWidget.translate(`visibilityTooltips.status.disabled`), visible: false, icon: visibilityShowSvg, hide: true }));
+    }
+    const getIcon = () => {
+        switch (state.state) {
+            case "visible":
+                return visibilityShowSvg;
+            case "hidden":
+                return visibilityHideSvg;
+            case "partial":
+                return visibilityPartialSvg;
+        }
     };
+    return (_jsx(TreeActionBase, { ...actionAttributes, label: state.tooltip ?? createTooltip(state.state), onClick: () => context.onVisibilityButtonClick(node, state.state), visible: state.state !== "visible" ? true : undefined, icon: getIcon() }));
+});
+const visibilityContext = createContext(undefined);
+/** @internal */
+export const useVisibilityContext = () => {
+    return useContext(visibilityContext);
+};
+/** @internal */
+export function VisibilityContextProvider({ onVisibilityButtonClick, getVisibilityButtonState, children }) {
+    return (_jsx(visibilityContext.Provider, { value: useMemo(() => ({
+            onVisibilityButtonClick,
+            getVisibilityButtonState,
+        }), [getVisibilityButtonState, onVisibilityButtonClick]), children: children }));
 }
 //# sourceMappingURL=TreeNodeVisibilityButton.js.map

package/lib/esm/tree-widget-react/components/trees/common/components/TreeNodeVisibilityButton.js.map

@@ -1 +1 @@
-{"version":3,"file":"TreeNodeVisibilityButton.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeNodeVisibilityButton.tsx"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,gCAAgC,CAAC;AAIxC,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,0CAA0C,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACrG,MAAM,kBAAkB,GAAG,IAAI,GAAG,CAAC,0CAA0C,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AACrG,MAAM,qBAAqB,GAAG,IAAI,GAAG,CAAC,6CAA6C,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAoB3G,gBAAgB;AAChB,MAAM,UAAU,sBAAsB,CAAC,EACrC,wBAAwB,EACxB,uBAAuB,GACO;IAC9B,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,wBAAwB,CAAC,IAAI,CAAC,CAAC;QAE7C,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,QAAQ,KAAK,CAAC,KAAK,EAAE;gBACnB,KAAK,SAAS;oBACZ,OAAO,kBAAkB,CAAC;gBAC5B,KAAK,QAAQ;oBACX,OAAO,kBAAkB,CAAC;gBAC5B,KAAK,SAAS;oBACZ,OAAO,qBAAqB,CAAC;aAChC;QACH,CAAC,CAAC;QACF,OAAO;YACL,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,2BAA2B;YACnD,MAAM,EAAE,GAAG,EAAE;gBACX,uBAAuB,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YAC7C,CAAC;YACD,IAAI,EAAE,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS;YAClD,IAAI,EAAE,OAAO,EAAE;SAChB,CAAC;IACJ,CAAC,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 \"./TreeNodeVisibilityButton.css\";\n\nimport type { PresentationHierarchyNode, TreeItemAction } from \"@itwin/presentation-hierarchies-react\";\n\nconst visibilityShowIcon = new URL(\"@itwin/itwinui-icons/visibility-show.svg\", import.meta.url).href;\nconst visibilityHideIcon = new URL(\"@itwin/itwinui-icons/visibility-hide.svg\", import.meta.url).href;\nconst visibilityPartialIcon = new URL(\"@itwin/itwinui-icons/visibility-partial.svg\", import.meta.url).href;\n\n/**\n * Data structure that describes tree node checkbox state.\n * @beta\n */\ninterface TreeItemVisibilityButtonState {\n  state: \"visible\" | \"partial\" | \"hidden\";\n  isDisabled?: boolean;\n  tooltip?: string;\n}\n\n/** @beta */\nexport interface TreeItemVisibilityButtonProps {\n  /** Callback that should be invoked when checkbox is clicked. */\n  onVisibilityButtonClick: (node: PresentationHierarchyNode, state: TreeItemVisibilityButtonState[\"state\"]) => void;\n  /** Callback that should be used to determine current checkbox state. */\n  getVisibilityButtonState: (node: PresentationHierarchyNode) => TreeItemVisibilityButtonState;\n}\n\n/** @internal */\nexport function createVisibilityAction({\n  getVisibilityButtonState,\n  onVisibilityButtonClick,\n}: TreeItemVisibilityButtonProps): (node: PresentationHierarchyNode) => TreeItemAction {\n  return (node) => {\n    const state = getVisibilityButtonState(node);\n\n    const getIcon = () => {\n      switch (state.state) {\n        case \"visible\":\n          return visibilityShowIcon;\n        case \"hidden\":\n          return visibilityHideIcon;\n        case \"partial\":\n          return visibilityPartialIcon;\n      }\n    };\n    return {\n      label: state.tooltip ?? \"Determining visibility...\",\n      action: () => {\n        onVisibilityButtonClick(node, state.state);\n      },\n      show: state.state !== \"visible\" ? true : undefined,\n      icon: getIcon(),\n    };\n  };\n}\n"]}
+{"version":3,"file":"TreeNodeVisibilityButton.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeNodeVisibilityButton.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,gCAAgC,CAAC;AAExC,OAAO,EAAE,aAAa,EAAE,IAAI,EAAE,UAAU,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AACjE,OAAO,EAAE,cAAc,EAAE,MAAM,uCAAuC,CAAC;AACvE,OAAO,iBAAiB,MAAM,sCAAsC,CAAC;AACrE,OAAO,oBAAoB,MAAM,yCAAyC,CAAC;AAC3E,OAAO,iBAAiB,MAAM,sCAAsC,CAAC;AACrE,OAAO,EAAE,UAAU,EAAE,MAAM,2BAA2B,CAAC;AACvD,OAAO,EAAE,aAAa,EAAE,MAAM,wBAAwB,CAAC;AAuBvD;;;;GAIG;AACH,MAAM,CAAC,MAAM,gBAAgB,GAAG,IAAI,CAAC,SAAS,gBAAgB,CAAC,EAAE,IAAI,EAAE,GAAG,gBAAgB,EAAkE;IAC1J,MAAM,OAAO,GAAG,oBAAoB,EAAE,CAAC;IACvC,MAAM,KAAK,GAAG,OAAO,EAAE,wBAAwB,CAAC,IAAI,CAAC,CAAC;IAEtD,IAAI,CAAC,OAAO,IAAI,CAAC,KAAK,IAAI,CAAC,YAAY,IAAI,KAAK,IAAI,KAAK,CAAC,UAAU,CAAC,IAAI,WAAW,IAAI,KAAK,EAAE,CAAC;QAC9F,OAAO,CACL,KAAC,cAAc,OACT,gBAAgB,EACpB,KAAK,EAAE,UAAU,CAAC,SAAS,CAAC,oCAAoC,CAAC,EACjE,OAAO,EAAE,KAAK,EACd,IAAI,EAAE,iBAAiB,EACvB,IAAI,EAAE,IAAI,GACV,CACH,CAAC;IACJ,CAAC;IAED,MAAM,OAAO,GAAG,GAAG,EAAE;QACnB,QAAQ,KAAK,CAAC,KAAK,EAAE,CAAC;YACpB,KAAK,SAAS;gBACZ,OAAO,iBAAiB,CAAC;YAC3B,KAAK,QAAQ;gBACX,OAAO,iBAAiB,CAAC;YAC3B,KAAK,SAAS;gBACZ,OAAO,oBAAoB,CAAC;QAChC,CAAC;IACH,CAAC,CAAC;IAEF,OAAO,CACL,KAAC,cAAc,OACT,gBAAgB,EACpB,KAAK,EAAE,KAAK,CAAC,OAAO,IAAI,aAAa,CAAC,KAAK,CAAC,KAAK,CAAC,EAClD,OAAO,EAAE,GAAG,EAAE,CAAC,OAAO,CAAC,uBAAuB,CAAC,IAAI,EAAE,KAAK,CAAC,KAAK,CAAC,EACjE,OAAO,EAAE,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,EACrD,IAAI,EAAE,OAAO,EAAE,GACf,CACH,CAAC;AACJ,CAAC,CAAC,CAAC;AAUH,MAAM,iBAAiB,GAAG,aAAa,CAAgC,SAAS,CAAC,CAAC;AAElF,gBAAgB;AAChB,MAAM,CAAC,MAAM,oBAAoB,GAAG,GAAG,EAAE;IACvC,OAAO,UAAU,CAAC,iBAAiB,CAAC,CAAC;AACvC,CAAC,CAAC;AAEF,gBAAgB;AAChB,MAAM,UAAU,yBAAyB,CAAC,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,QAAQ,EAAwC;IAC7I,OAAO,CACL,KAAC,iBAAiB,CAAC,QAAQ,IACzB,KAAK,EAAE,OAAO,CACZ,GAAG,EAAE,CAAC,CAAC;YACL,uBAAuB;YACvB,wBAAwB;SACzB,CAAC,EACF,CAAC,wBAAwB,EAAE,uBAAuB,CAAC,CACpD,YAEA,QAAQ,GACkB,CAC9B,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 \"./TreeNodeVisibilityButton.css\";\n\nimport { createContext, memo, useContext, useMemo } from \"react\";\nimport { TreeActionBase } from \"@itwin/presentation-hierarchies-react\";\nimport visibilityHideSvg from \"@stratakit/icons/visibility-hide.svg\";\nimport visibilityPartialSvg from \"@stratakit/icons/visibility-partial.svg\";\nimport visibilityShowSvg from \"@stratakit/icons/visibility-show.svg\";\nimport { TreeWidget } from \"../../../../TreeWidget.js\";\nimport { createTooltip } from \"../internal/Tooltip.js\";\n\nimport type { PropsWithChildren } from \"react\";\nimport type { PresentationHierarchyNode, TreeActionBaseAttributes } from \"@itwin/presentation-hierarchies-react\";\n\n/** @beta */\ninterface LoadedTreeItemVisibilityButtonState {\n  state: \"visible\" | \"hidden\" | \"partial\";\n  isDisabled?: boolean;\n}\n/**\n * Data structure that describes tree node checkbox state.\n * @beta\n */\nexport type TreeItemVisibilityButtonState = (\n  | LoadedTreeItemVisibilityButtonState\n  | {\n      isLoading: true;\n    }\n) & {\n  tooltip?: string;\n};\n\n/**\n * React component that renders a visibility action for a tree item.\n * Should be used with `VisibilityTreeRenderer`.\n * @beta\n */\nexport const VisibilityAction = memo(function VisibilityAction({ node, ...actionAttributes }: { node: PresentationHierarchyNode } & TreeActionBaseAttributes) {\n  const context = useVisibilityContext();\n  const state = context?.getVisibilityButtonState(node);\n\n  if (!context || !state || (\"isDisabled\" in state && state.isDisabled) || \"isLoading\" in state) {\n    return (\n      <TreeActionBase\n        {...actionAttributes}\n        label={TreeWidget.translate(`visibilityTooltips.status.disabled`)}\n        visible={false}\n        icon={visibilityShowSvg}\n        hide={true}\n      />\n    );\n  }\n\n  const getIcon = () => {\n    switch (state.state) {\n      case \"visible\":\n        return visibilityShowSvg;\n      case \"hidden\":\n        return visibilityHideSvg;\n      case \"partial\":\n        return visibilityPartialSvg;\n    }\n  };\n\n  return (\n    <TreeActionBase\n      {...actionAttributes}\n      label={state.tooltip ?? createTooltip(state.state)}\n      onClick={() => context.onVisibilityButtonClick(node, state.state)}\n      visible={state.state !== \"visible\" ? true : undefined}\n      icon={getIcon()}\n    />\n  );\n});\n\n/** @beta */\nexport interface VisibilityContext {\n  /** Callback that should be invoked when checkbox is clicked. */\n  onVisibilityButtonClick: (node: PresentationHierarchyNode, state: LoadedTreeItemVisibilityButtonState[\"state\"]) => void;\n  /** Callback that should be used to determine current checkbox state. */\n  getVisibilityButtonState: (node: PresentationHierarchyNode) => TreeItemVisibilityButtonState;\n}\n\nconst visibilityContext = createContext<VisibilityContext | undefined>(undefined);\n\n/** @internal */\nexport const useVisibilityContext = () => {\n  return useContext(visibilityContext);\n};\n\n/** @internal */\nexport function VisibilityContextProvider({ onVisibilityButtonClick, getVisibilityButtonState, children }: PropsWithChildren<VisibilityContext>) {\n  return (\n    <visibilityContext.Provider\n      value={useMemo(\n        () => ({\n          onVisibilityButtonClick,\n          getVisibilityButtonState,\n        }),\n        [getVisibilityButtonState, onVisibilityButtonClick],\n      )}\n    >\n      {children}\n    </visibilityContext.Provider>\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/TreeRenderer.d.ts

@@ -1,7 +1,12 @@
 import type { BaseTreeRendererProps } from "./BaseTreeRenderer.js";
+/** @beta */
+type TreeRendererProps = Omit<BaseTreeRendererProps, "getInlineActions" | "getMenuActions" | "getContextMenuActions" | "getDecorations"> & {
+    [Property in keyof Pick<BaseTreeRendererProps, "getInlineActions" | "getMenuActions" | "getContextMenuActions" | "getDecorations">]?: (args: Parameters<NonNullable<BaseTreeRendererProps[Property]>>[0], treeRendererProps: TreeRendererProps) => ReturnType<NonNullable<BaseTreeRendererProps[Property]>>;
+};
 /**
  * Default renderer for rendering tree data.
  * @beta
  */
-export declare function TreeRenderer({ actions, ...props }: BaseTreeRendererProps): import("react/jsx-runtime").JSX.Element;
+export declare function TreeRenderer(props: TreeRendererProps): import("react/jsx-runtime").JSX.Element;
+export {};
 //# sourceMappingURL=TreeRenderer.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/components/TreeRenderer.js

@@ -3,14 +3,23 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-/* eslint-disable @itwin/no-internal */
-import { createFilterAction } from "@itwin/presentation-hierarchies-react";
+import { useCallback, useMemo } from "react";
+import { FilterAction } from "@itwin/presentation-hierarchies-react";
 import { BaseTreeRenderer } from "./BaseTreeRenderer.js";
 /**
  * Default renderer for rendering tree data.
  * @beta
  */
-export function TreeRenderer({ actions, ...props }) {
-    return (_jsx(BaseTreeRenderer, { ...props, actions: [createFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails }), ...(actions ? actions : [])] }));
+export function TreeRenderer(props) {
+    const { getInlineActions, getMenuActions, getContextMenuActions, getDecorations, getHierarchyLevelDetails, onFilterClick, ...restProps } = props;
+    const nodeInlineActions = useCallback((actionsProps) => {
+        return getInlineActions
+            ? getInlineActions(actionsProps, props)
+            : [_jsx(FilterAction, { node: actionsProps.targetNode, onFilter: onFilterClick, getHierarchyLevelDetails: getHierarchyLevelDetails }, "Filter")];
+    }, [getInlineActions, onFilterClick, getHierarchyLevelDetails, props]);
+    const nodeMenuActions = useMemo(() => (getMenuActions ? (actionsProps) => getMenuActions(actionsProps, props) : undefined), [getMenuActions, props]);
+    const nodeContextMenuActions = useMemo(() => (getContextMenuActions ? (actionsProps) => getContextMenuActions(actionsProps, props) : undefined), [getContextMenuActions, props]);
+    const nodeDecorations = useMemo(() => (getDecorations ? (node) => getDecorations(node, props) : undefined), [getDecorations, props]);
+    return (_jsx(BaseTreeRenderer, { ...restProps, onFilterClick: onFilterClick, getHierarchyLevelDetails: getHierarchyLevelDetails, getInlineActions: nodeInlineActions, getMenuActions: nodeMenuActions, getContextMenuActions: nodeContextMenuActions, getDecorations: nodeDecorations }));
 }
 //# sourceMappingURL=TreeRenderer.js.map

package/lib/esm/tree-widget-react/components/trees/common/components/TreeRenderer.js.map

@@ -1 +1 @@
-{"version":3,"file":"TreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,uCAAuC;AAEvC,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAC;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAIzD;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,EAAE,OAAO,EAAE,GAAG,KAAK,EAAyB;IACvE,OAAO,CACL,KAAC,gBAAgB,OACX,KAAK,EACT,OAAO,EAAE,CAAC,kBAAkB,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,aAAa,EAAE,wBAAwB,EAAE,KAAK,CAAC,wBAAwB,EAAE,CAAC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,GACvJ,CACH,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/* eslint-disable @itwin/no-internal */\n\nimport { createFilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\n\n/**\n * Default renderer for rendering tree data.\n * @beta\n */\nexport function TreeRenderer({ actions, ...props }: BaseTreeRendererProps) {\n  return (\n    <BaseTreeRenderer\n      {...props}\n      actions={[createFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails }), ...(actions ? actions : [])]}\n    />\n  );\n}\n"]}
+{"version":3,"file":"TreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/TreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AAYzD;;;GAGG;AACH,MAAM,UAAU,YAAY,CAAC,KAAwB;IACnD,MAAM,EAAE,gBAAgB,EAAE,cAAc,EAAE,qBAAqB,EAAE,cAAc,EAAE,wBAAwB,EAAE,aAAa,EAAE,GAAG,SAAS,EAAE,GAAG,KAAK,CAAC;IAEjJ,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,YAAY,EAAE,EAAE;QACf,OAAO,gBAAgB;YACrB,CAAC,CAAC,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC;YACvC,CAAC,CAAC,CAAC,KAAC,YAAY,IAAgB,IAAI,EAAE,YAAY,CAAC,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,wBAAwB,EAAE,wBAAwB,IAApH,QAAQ,CAAgH,CAAC,CAAC;IACpJ,CAAC,EACD,CAAC,gBAAgB,EAAE,aAAa,EAAE,wBAAwB,EAAE,KAAK,CAAC,CACnE,CAAC;IACF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,cAAc,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EAC1F,CAAC,cAAc,EAAE,KAAK,CAAC,CACxB,CAAC;IACF,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EACxG,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAC/B,CAAC;IACF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EAC1E,CAAC,cAAc,EAAE,KAAK,CAAC,CACxB,CAAC;IAEF,OAAO,CACL,KAAC,gBAAgB,OACX,SAAS,EACb,aAAa,EAAE,aAAa,EAC5B,wBAAwB,EAAE,wBAAwB,EAClD,gBAAgB,EAAE,iBAAiB,EACnC,cAAc,EAAE,eAAe,EAC/B,qBAAqB,EAAE,sBAAsB,EAC7C,cAAc,EAAE,eAAe,GAC/B,CACH,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, useMemo } from \"react\";\nimport { FilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\n\n/** @beta */\ntype TreeRendererProps = Omit<BaseTreeRendererProps, \"getInlineActions\" | \"getMenuActions\" | \"getContextMenuActions\" | \"getDecorations\"> & {\n  [Property in keyof Pick<BaseTreeRendererProps, \"getInlineActions\" | \"getMenuActions\" | \"getContextMenuActions\" | \"getDecorations\">]?: (\n    args: Parameters<NonNullable<BaseTreeRendererProps[Property]>>[0],\n    treeRendererProps: TreeRendererProps,\n  ) => ReturnType<NonNullable<BaseTreeRendererProps[Property]>>;\n};\n\n/**\n * Default renderer for rendering tree data.\n * @beta\n */\nexport function TreeRenderer(props: TreeRendererProps) {\n  const { getInlineActions, getMenuActions, getContextMenuActions, getDecorations, getHierarchyLevelDetails, onFilterClick, ...restProps } = props;\n\n  const nodeInlineActions = useCallback<Required<BaseTreeRendererProps>[\"getInlineActions\"]>(\n    (actionsProps) => {\n      return getInlineActions\n        ? getInlineActions(actionsProps, props)\n        : [<FilterAction key={\"Filter\"} node={actionsProps.targetNode} onFilter={onFilterClick} getHierarchyLevelDetails={getHierarchyLevelDetails} />];\n    },\n    [getInlineActions, onFilterClick, getHierarchyLevelDetails, props],\n  );\n  const nodeMenuActions = useMemo<BaseTreeRendererProps[\"getMenuActions\"]>(\n    () => (getMenuActions ? (actionsProps) => getMenuActions(actionsProps, props) : undefined),\n    [getMenuActions, props],\n  );\n  const nodeContextMenuActions = useMemo<BaseTreeRendererProps[\"getContextMenuActions\"]>(\n    () => (getContextMenuActions ? (actionsProps) => getContextMenuActions(actionsProps, props) : undefined),\n    [getContextMenuActions, props],\n  );\n  const nodeDecorations = useMemo<BaseTreeRendererProps[\"getDecorations\"]>(\n    () => (getDecorations ? (node) => getDecorations(node, props) : undefined),\n    [getDecorations, props],\n  );\n\n  return (\n    <BaseTreeRenderer\n      {...restProps}\n      onFilterClick={onFilterClick}\n      getHierarchyLevelDetails={getHierarchyLevelDetails}\n      getInlineActions={nodeInlineActions}\n      getMenuActions={nodeMenuActions}\n      getContextMenuActions={nodeContextMenuActions}\n      getDecorations={nodeDecorations}\n    />\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.d.ts

@@ -1,13 +1,14 @@
-import type { PresentationTreeNode } from "@itwin/presentation-hierarchies-react";
-import type { TreeItemVisibilityButtonProps } from "./TreeNodeVisibilityButton.js";
+import type { PresentationHierarchyNode } from "@itwin/presentation-hierarchies-react";
+import type { VisibilityContext } from "./TreeNodeVisibilityButton.js";
 interface UseVisibilityButtonHandlerProps {
-    rootNodes: PresentationTreeNode[] | undefined;
+    rootNodes: PresentationHierarchyNode[] | undefined;
     isNodeSelected?: (nodeId: string) => boolean;
-    onClick: TreeItemVisibilityButtonProps["onVisibilityButtonClick"];
+    onClick: VisibilityContext["onVisibilityButtonClick"];
 }
 interface UseVisibilityButtonHandlerResult {
-    onVisibilityButtonClick: TreeItemVisibilityButtonProps["onVisibilityButtonClick"];
+    onVisibilityButtonClick: VisibilityContext["onVisibilityButtonClick"];
 }
+/** @internal */
 export declare function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult;
 export {};
 //# sourceMappingURL=UseVisibilityButtonHandler.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.js

@@ -3,22 +3,22 @@
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import { useCallback } from "react";
-import { isPresentationHierarchyNode } from "@itwin/presentation-hierarchies-react";
+import { useLatest } from "../internal/Utils.js";
+/** @internal */
 export function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }) {
+    const isNodeSelectedRef = useLatest(isNodeSelected);
+    const rootNodesRef = useLatest(rootNodes);
     const onVisibilityButtonClick = useCallback((clickedNode, state) => {
-        if (!isNodeSelected?.(clickedNode.id)) {
+        if (!isNodeSelectedRef.current?.(clickedNode.id)) {
             onClick(clickedNode, state);
             return;
         }
-        rootNodes && forEachSelectedNode(rootNodes, isNodeSelected, (node) => onClick(node, state));
-    }, [rootNodes, isNodeSelected, onClick]);
+        rootNodesRef.current && forEachSelectedNode(rootNodesRef.current, isNodeSelectedRef.current, (node) => onClick(node, state));
+    }, [onClick, isNodeSelectedRef, rootNodesRef]);
     return { onVisibilityButtonClick };
 }
 function forEachSelectedNode(nodes, isNodeSelected, callback) {
     nodes.forEach((node) => {
-        if (!isPresentationHierarchyNode(node)) {
-            return;
-        }
         if (isNodeSelected(node.id)) {
             callback(node);
         }

package/lib/esm/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseVisibilityButtonHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,2BAA2B,EAAE,MAAM,uCAAuC,CAAC;AAepF,MAAM,UAAU,0BAA0B,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAmC;IAChH,MAAM,uBAAuB,GAAG,WAAW,CACzC,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;QACrB,IAAI,CAAC,cAAc,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE;YACrC,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5B,OAAO;SACR;QACD,SAAS,IAAI,mBAAmB,CAAC,SAAS,EAAE,cAAc,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IAC9F,CAAC,EACD,CAAC,SAAS,EAAE,cAAc,EAAE,OAAO,CAAC,CACrC,CAAC;IAEF,OAAO,EAAE,uBAAuB,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAkC,EAClC,cAA2C,EAC3C,QAAmD;IAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACrB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,EAAE;YACtC,OAAO;SACR;QACD,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE;YAC3B,QAAQ,CAAC,IAAI,CAAC,CAAC;SAChB;QACD,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE;YACzD,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;SAC9D;IACH,CAAC,CAAC,CAAC;AACL,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 } from \"react\";\nimport { isPresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\n\nimport type { PresentationHierarchyNode, PresentationTreeNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n\ninterface UseVisibilityButtonHandlerProps {\n  rootNodes: PresentationTreeNode[] | undefined;\n  isNodeSelected?: (nodeId: string) => boolean;\n  onClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\ninterface UseVisibilityButtonHandlerResult {\n  onVisibilityButtonClick: TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"];\n}\n\nexport function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult {\n  const onVisibilityButtonClick = useCallback<TreeItemVisibilityButtonProps[\"onVisibilityButtonClick\"]>(\n    (clickedNode, state) => {\n      if (!isNodeSelected?.(clickedNode.id)) {\n        onClick(clickedNode, state);\n        return;\n      }\n      rootNodes && forEachSelectedNode(rootNodes, isNodeSelected, (node) => onClick(node, state));\n    },\n    [rootNodes, isNodeSelected, onClick],\n  );\n\n  return { onVisibilityButtonClick };\n}\n\nfunction forEachSelectedNode(\n  nodes: Array<PresentationTreeNode>,\n  isNodeSelected: (nodeId: string) => boolean,\n  callback: (node: PresentationHierarchyNode) => void,\n) {\n  nodes.forEach((node) => {\n    if (!isPresentationHierarchyNode(node)) {\n      return;\n    }\n    if (isNodeSelected(node.id)) {\n      callback(node);\n    }\n    if (node.isExpanded && typeof node.children !== \"boolean\") {\n      forEachSelectedNode(node.children, isNodeSelected, callback);\n    }\n  });\n}\n"]}
+{"version":3,"file":"UseVisibilityButtonHandler.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/UseVisibilityButtonHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAejD,gBAAgB;AAChB,MAAM,UAAU,0BAA0B,CAAC,EAAE,SAAS,EAAE,cAAc,EAAE,OAAO,EAAmC;IAChH,MAAM,iBAAiB,GAAG,SAAS,CAAC,cAAc,CAAC,CAAC;IACpD,MAAM,YAAY,GAAG,SAAS,CAAC,SAAS,CAAC,CAAC;IAE1C,MAAM,uBAAuB,GAAG,WAAW,CACzC,CAAC,WAAW,EAAE,KAAK,EAAE,EAAE;QACrB,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YACjD,OAAO,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;YAC5B,OAAO;QACT,CAAC;QACD,YAAY,CAAC,OAAO,IAAI,mBAAmB,CAAC,YAAY,CAAC,OAAO,EAAE,iBAAiB,CAAC,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE,CAAC,OAAO,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC;IAC/H,CAAC,EACD,CAAC,OAAO,EAAE,iBAAiB,EAAE,YAAY,CAAC,CAC3C,CAAC;IAEF,OAAO,EAAE,uBAAuB,EAAE,CAAC;AACrC,CAAC;AAED,SAAS,mBAAmB,CAC1B,KAAuC,EACvC,cAA2C,EAC3C,QAAmD;IAEnD,KAAK,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,EAAE;QACrB,IAAI,cAAc,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,CAAC;YAC5B,QAAQ,CAAC,IAAI,CAAC,CAAC;QACjB,CAAC;QACD,IAAI,IAAI,CAAC,UAAU,IAAI,OAAO,IAAI,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;YAC1D,mBAAmB,CAAC,IAAI,CAAC,QAAQ,EAAE,cAAc,EAAE,QAAQ,CAAC,CAAC;QAC/D,CAAC;IACH,CAAC,CAAC,CAAC;AACL,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 } from \"react\";\nimport { useLatest } from \"../internal/Utils.js\";\n\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityContext } from \"./TreeNodeVisibilityButton.js\";\n\ninterface UseVisibilityButtonHandlerProps {\n  rootNodes: PresentationHierarchyNode[] | undefined;\n  isNodeSelected?: (nodeId: string) => boolean;\n  onClick: VisibilityContext[\"onVisibilityButtonClick\"];\n}\n\ninterface UseVisibilityButtonHandlerResult {\n  onVisibilityButtonClick: VisibilityContext[\"onVisibilityButtonClick\"];\n}\n\n/** @internal */\nexport function useVisibilityButtonHandler({ rootNodes, isNodeSelected, onClick }: UseVisibilityButtonHandlerProps): UseVisibilityButtonHandlerResult {\n  const isNodeSelectedRef = useLatest(isNodeSelected);\n  const rootNodesRef = useLatest(rootNodes);\n\n  const onVisibilityButtonClick = useCallback<VisibilityContext[\"onVisibilityButtonClick\"]>(\n    (clickedNode, state) => {\n      if (!isNodeSelectedRef.current?.(clickedNode.id)) {\n        onClick(clickedNode, state);\n        return;\n      }\n      rootNodesRef.current && forEachSelectedNode(rootNodesRef.current, isNodeSelectedRef.current, (node) => onClick(node, state));\n    },\n    [onClick, isNodeSelectedRef, rootNodesRef],\n  );\n\n  return { onVisibilityButtonClick };\n}\n\nfunction forEachSelectedNode(\n  nodes: Array<PresentationHierarchyNode>,\n  isNodeSelected: (nodeId: string) => boolean,\n  callback: (node: PresentationHierarchyNode) => void,\n) {\n  nodes.forEach((node) => {\n    if (isNodeSelected(node.id)) {\n      callback(node);\n    }\n    if (node.isExpanded && typeof node.children !== \"boolean\") {\n      forEachSelectedNode(node.children, isNodeSelected, callback);\n    }\n  });\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTree.d.ts

@@ -1,9 +1,9 @@
-import type { FunctionProps } from "../Utils.js";
-import type { TreeProps } from "./Tree.js";
 import type { ReactNode } from "react";
-import type { VisibilityTreeRendererProps } from "./VisibilityTreeRenderer.js";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
 import type { HierarchyVisibilityHandler } from "../UseHierarchyVisibility.js";
+import type { FunctionProps } from "../Utils.js";
+import type { TreeProps } from "./Tree.js";
+import type { VisibilityContext } from "./TreeNodeVisibilityButton.js";
 /** @beta */
 export type VisibilityTreeProps = Omit<TreeProps, "treeRenderer" | "imodelAccess"> & {
     /** Callback for creating visibility handler used to control visibility of instances represented by tree nodes. */
@@ -11,11 +11,11 @@ export type VisibilityTreeProps = Omit<TreeProps, "treeRenderer" | "imodelAccess
         imodelAccess: ECClassHierarchyInspector;
     }) => HierarchyVisibilityHandler;
     /** Tree renderer that should be used to render tree data. */
-    treeRenderer: (treeProps: FunctionProps<TreeProps["treeRenderer"]> & Pick<VisibilityTreeRendererProps, "getVisibilityButtonState" | "onVisibilityButtonClick">) => ReactNode;
+    treeRenderer: (treeProps: FunctionProps<TreeProps["treeRenderer"]> & Pick<VisibilityContext, "getVisibilityButtonState" | "onVisibilityButtonClick">) => ReactNode;
 };
 /**
  * Tree component that can control visibility of instances represented by tree nodes.
  * @beta
  */
-export declare function VisibilityTree({ visibilityHandlerFactory, treeRenderer, ...props }: VisibilityTreeProps): import("react/jsx-runtime").JSX.Element;
+export declare function VisibilityTree({ visibilityHandlerFactory, treeRenderer, hierarchyLevelSizeLimit, ...props }: VisibilityTreeProps): import("react/jsx-runtime").JSX.Element;
 //# sourceMappingURL=VisibilityTree.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTree.js

@@ -3,20 +3,24 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { useCallback, useMemo } from "react";
+import { useCallback } from "react";
+import { useIModelAccess } from "../internal/UseIModelAccess.js";
 import { useHierarchyVisibility } from "../UseHierarchyVisibility.js";
-import { createIModelAccess } from "../Utils.js";
 import { Tree } from "./Tree.js";
 /**
  * Tree component that can control visibility of instances represented by tree nodes.
  * @beta
  */
-export function VisibilityTree({ visibilityHandlerFactory, treeRenderer, ...props }) {
-    const { imodel, getSchemaContext } = props;
-    const imodelAccess = useMemo(() => createIModelAccess({ imodel, getSchemaContext }), [imodel, getSchemaContext]);
+export function VisibilityTree({ visibilityHandlerFactory, treeRenderer, hierarchyLevelSizeLimit, ...props }) {
+    const { imodel, treeName } = props;
+    const { imodelAccess, currentHierarchyLevelSizeLimit } = useIModelAccess({
+        imodel,
+        treeName,
+        hierarchyLevelSizeLimit,
+    });
     const { getVisibilityButtonState, onVisibilityButtonClick, triggerRefresh } = useHierarchyVisibility({
         visibilityHandlerFactory: useCallback(() => visibilityHandlerFactory({ imodelAccess }), [visibilityHandlerFactory, imodelAccess]),
     });
-    return (_jsx(Tree, { ...props, onReload: triggerRefresh, imodelAccess: imodelAccess, treeRenderer: (treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick }) }));
+    return (_jsx(Tree, { ...props, onReload: triggerRefresh, imodelAccess: imodelAccess, hierarchyLevelSizeLimit: currentHierarchyLevelSizeLimit, treeRenderer: (treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick }) }));
 }
 //# sourceMappingURL=VisibilityTree.js.map

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"VisibilityTree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,kBAAkB,EAAE,MAAM,aAAa,CAAC;AACjD,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAmBjC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,GAAG,KAAK,EAAuB;IACtG,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,KAAK,CAAC;IAC3C,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE,CAAC,kBAAkB,CAAC,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,EAAE,CAAC,MAAM,EAAE,gBAAgB,CAAC,CAAC,CAAC;IACjH,MAAM,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,cAAc,EAAE,GAAG,sBAAsB,CAAC;QACnG,wBAAwB,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;KAClI,CAAC,CAAC;IAEH,OAAO,CACL,KAAC,IAAI,OACC,KAAK,EACT,QAAQ,EAAE,cAAc,EACxB,YAAY,EAAE,YAAY,EAC1B,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,GAAG,SAAS,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,CAAC,GAC9G,CACH,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, useMemo } from \"react\";\nimport { useHierarchyVisibility } from \"../UseHierarchyVisibility.js\";\nimport { createIModelAccess } from \"../Utils.js\";\nimport { Tree } from \"./Tree.js\";\n\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { TreeProps } from \"./Tree.js\";\nimport type { ReactNode } from \"react\";\nimport type { VisibilityTreeRendererProps } from \"./VisibilityTreeRenderer.js\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { HierarchyVisibilityHandler } from \"../UseHierarchyVisibility.js\";\n\n/** @beta */\nexport type VisibilityTreeProps = Omit<TreeProps, \"treeRenderer\" | \"imodelAccess\"> & {\n  /** Callback for creating visibility handler used to control visibility of instances represented by tree nodes. */\n  visibilityHandlerFactory: (props: { imodelAccess: ECClassHierarchyInspector }) => HierarchyVisibilityHandler;\n  /** Tree renderer that should be used to render tree data. */\n  treeRenderer: (\n    treeProps: FunctionProps<TreeProps[\"treeRenderer\"]> & Pick<VisibilityTreeRendererProps, \"getVisibilityButtonState\" | \"onVisibilityButtonClick\">,\n  ) => ReactNode;\n};\n\n/**\n * Tree component that can control visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTree({ visibilityHandlerFactory, treeRenderer, ...props }: VisibilityTreeProps) {\n  const { imodel, getSchemaContext } = props;\n  const imodelAccess = useMemo(() => createIModelAccess({ imodel, getSchemaContext }), [imodel, getSchemaContext]);\n  const { getVisibilityButtonState, onVisibilityButtonClick, triggerRefresh } = useHierarchyVisibility({\n    visibilityHandlerFactory: useCallback(() => visibilityHandlerFactory({ imodelAccess }), [visibilityHandlerFactory, imodelAccess]),\n  });\n\n  return (\n    <Tree\n      {...props}\n      onReload={triggerRefresh}\n      imodelAccess={imodelAccess}\n      treeRenderer={(treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick })}\n    />\n  );\n}\n"]}
+{"version":3,"file":"VisibilityTree.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,OAAO,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AACjE,OAAO,EAAE,sBAAsB,EAAE,MAAM,8BAA8B,CAAC;AACtE,OAAO,EAAE,IAAI,EAAE,MAAM,WAAW,CAAC;AAmBjC;;;GAGG;AACH,MAAM,UAAU,cAAc,CAAC,EAAE,wBAAwB,EAAE,YAAY,EAAE,uBAAuB,EAAE,GAAG,KAAK,EAAuB;IAC/H,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACnC,MAAM,EAAE,YAAY,EAAE,8BAA8B,EAAE,GAAG,eAAe,CAAC;QACvE,MAAM;QACN,QAAQ;QACR,uBAAuB;KACxB,CAAC,CAAC;IAEH,MAAM,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,cAAc,EAAE,GAAG,sBAAsB,CAAC;QACnG,wBAAwB,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,wBAAwB,CAAC,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC,wBAAwB,EAAE,YAAY,CAAC,CAAC;KAClI,CAAC,CAAC;IAEH,OAAO,CACL,KAAC,IAAI,OACC,KAAK,EACT,QAAQ,EAAE,cAAc,EACxB,YAAY,EAAE,YAAY,EAC1B,uBAAuB,EAAE,8BAA8B,EACvD,YAAY,EAAE,CAAC,SAAS,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,GAAG,SAAS,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,CAAC,GAC9G,CACH,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 } from \"react\";\nimport { useIModelAccess } from \"../internal/UseIModelAccess.js\";\nimport { useHierarchyVisibility } from \"../UseHierarchyVisibility.js\";\nimport { Tree } from \"./Tree.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { HierarchyVisibilityHandler } from \"../UseHierarchyVisibility.js\";\nimport type { FunctionProps } from \"../Utils.js\";\nimport type { TreeProps } from \"./Tree.js\";\nimport type { VisibilityContext } from \"./TreeNodeVisibilityButton.js\";\n\n/** @beta */\nexport type VisibilityTreeProps = Omit<TreeProps, \"treeRenderer\" | \"imodelAccess\"> & {\n  /** Callback for creating visibility handler used to control visibility of instances represented by tree nodes. */\n  visibilityHandlerFactory: (props: { imodelAccess: ECClassHierarchyInspector }) => HierarchyVisibilityHandler;\n  /** Tree renderer that should be used to render tree data. */\n  treeRenderer: (\n    treeProps: FunctionProps<TreeProps[\"treeRenderer\"]> & Pick<VisibilityContext, \"getVisibilityButtonState\" | \"onVisibilityButtonClick\">,\n  ) => ReactNode;\n};\n\n/**\n * Tree component that can control visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTree({ visibilityHandlerFactory, treeRenderer, hierarchyLevelSizeLimit, ...props }: VisibilityTreeProps) {\n  const { imodel, treeName } = props;\n  const { imodelAccess, currentHierarchyLevelSizeLimit } = useIModelAccess({\n    imodel,\n    treeName,\n    hierarchyLevelSizeLimit,\n  });\n\n  const { getVisibilityButtonState, onVisibilityButtonClick, triggerRefresh } = useHierarchyVisibility({\n    visibilityHandlerFactory: useCallback(() => visibilityHandlerFactory({ imodelAccess }), [visibilityHandlerFactory, imodelAccess]),\n  });\n\n  return (\n    <Tree\n      {...props}\n      onReload={triggerRefresh}\n      imodelAccess={imodelAccess}\n      hierarchyLevelSizeLimit={currentHierarchyLevelSizeLimit}\n      treeRenderer={(treeProps) => treeRenderer({ ...treeProps, getVisibilityButtonState, onVisibilityButtonClick })}\n    />\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.d.ts

@@ -1,10 +1,12 @@
 import type { BaseTreeRendererProps } from "./BaseTreeRenderer.js";
-import type { TreeItemVisibilityButtonProps } from "./TreeNodeVisibilityButton.js";
+import type { VisibilityContext } from "./TreeNodeVisibilityButton.js";
 /** @beta */
-export type VisibilityTreeRendererProps = BaseTreeRendererProps & TreeItemVisibilityButtonProps;
+export type VisibilityTreeRendererProps = Omit<BaseTreeRendererProps, "getInlineActions" | "getMenuActions" | "getContextMenuActions" | "getDecorations"> & {
+    [Property in keyof Pick<BaseTreeRendererProps, "getInlineActions" | "getMenuActions" | "getContextMenuActions" | "getDecorations">]?: (args: Parameters<NonNullable<BaseTreeRendererProps[Property]>>[0], treeRendererProps: VisibilityTreeRendererProps) => ReturnType<NonNullable<BaseTreeRendererProps[Property]>>;
+} & VisibilityContext;
 /**
  * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.
  * @beta
  */
-export declare function VisibilityTreeRenderer({ getVisibilityButtonState, onVisibilityButtonClick: onClick, actions, ...props }: VisibilityTreeRendererProps): import("react/jsx-runtime").JSX.Element;
+export declare function VisibilityTreeRenderer(props: VisibilityTreeRendererProps): import("react/jsx-runtime").JSX.Element;
 //# sourceMappingURL=VisibilityTreeRenderer.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.js

@@ -3,27 +3,29 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-/* eslint-disable @itwin/no-internal */
-import { useMemo } from "react";
-import { createFilterAction } from "@itwin/presentation-hierarchies-react";
+import { useCallback, useMemo } from "react";
+import { FilterAction } from "@itwin/presentation-hierarchies-react";
 import { BaseTreeRenderer } from "./BaseTreeRenderer.js";
-import { createVisibilityAction } from "./TreeNodeVisibilityButton.js";
+import { VisibilityAction, VisibilityContextProvider } from "./TreeNodeVisibilityButton.js";
 import { useVisibilityButtonHandler } from "./UseVisibilityButtonHandler.js";
 /**
  * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.
  * @beta
  */
-export function VisibilityTreeRenderer({ getVisibilityButtonState, onVisibilityButtonClick: onClick, actions, ...props }) {
+export function VisibilityTreeRenderer(props) {
+    const { getVisibilityButtonState, onVisibilityButtonClick: onClick, getInlineActions, getMenuActions, getContextMenuActions, getDecorations, onFilterClick, getHierarchyLevelDetails, ...restProps } = props;
     const { onVisibilityButtonClick } = useVisibilityButtonHandler({ rootNodes: props.rootNodes, isNodeSelected: props.isNodeSelected, onClick });
-    const visibilityButtonProps = useMemo(() => ({
-        variant: "eyeball",
-        getVisibilityButtonState,
-        onVisibilityButtonClick,
-    }), [getVisibilityButtonState, onVisibilityButtonClick]);
-    return (_jsx(BaseTreeRenderer, { ...props, actions: [
-            ...(visibilityButtonProps ? [createVisibilityAction(visibilityButtonProps)] : []),
-            createFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails }),
-            ...(actions ? actions : []),
-        ] }));
+    const nodeInlineActions = useCallback((actionsProps) => {
+        return getInlineActions
+            ? getInlineActions(actionsProps, props)
+            : [
+                _jsx(VisibilityAction, { node: actionsProps.targetNode }, "Visibility"),
+                _jsx(FilterAction, { node: actionsProps.targetNode, onFilter: onFilterClick, getHierarchyLevelDetails: getHierarchyLevelDetails }, "Filter"),
+            ];
+    }, [onFilterClick, getHierarchyLevelDetails, getInlineActions, props]);
+    const nodeMenuActions = useMemo(() => (getMenuActions ? (actionsProps) => getMenuActions(actionsProps, props) : undefined), [getMenuActions, props]);
+    const nodeContextMenuActions = useMemo(() => (getContextMenuActions ? (actionsProps) => getContextMenuActions(actionsProps, props) : undefined), [getContextMenuActions, props]);
+    const nodeDecorations = useMemo(() => (getDecorations ? (node) => getDecorations(node, props) : undefined), [getDecorations, props]);
+    return (_jsx(VisibilityContextProvider, { onVisibilityButtonClick: onVisibilityButtonClick, getVisibilityButtonState: getVisibilityButtonState, children: _jsx(BaseTreeRenderer, { ...restProps, onFilterClick: onFilterClick, getHierarchyLevelDetails: getHierarchyLevelDetails, getInlineActions: nodeInlineActions, getMenuActions: nodeMenuActions, getContextMenuActions: nodeContextMenuActions, getDecorations: nodeDecorations }) }));
 }
 //# sourceMappingURL=VisibilityTreeRenderer.js.map

package/lib/esm/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.js.map

@@ -1 +1 @@
-{"version":3,"file":"VisibilityTreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,uCAAuC;AAEvC,OAAO,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAChC,OAAO,EAAE,kBAAkB,EAAE,MAAM,uCAAuC,CAAC;AAC3E,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,sBAAsB,EAAE,MAAM,+BAA+B,CAAC;AACvE,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAO7E;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,EAAE,wBAAwB,EAAE,uBAAuB,EAAE,OAAO,EAAE,OAAO,EAAE,GAAG,KAAK,EAA+B;IACnJ,MAAM,EAAE,uBAAuB,EAAE,GAAG,0BAA0B,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;IAC9I,MAAM,qBAAqB,GAAkC,OAAO,CAClE,GAAG,EAAE,CAAC,CAAC;QACL,OAAO,EAAE,SAAS;QAClB,wBAAwB;QACxB,uBAAuB;KACxB,CAAC,EACF,CAAC,wBAAwB,EAAE,uBAAuB,CAAC,CACpD,CAAC;IAEF,OAAO,CACL,KAAC,gBAAgB,OACX,KAAK,EACT,OAAO,EAAE;YACP,GAAG,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,sBAAsB,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;YACjF,kBAAkB,CAAC,EAAE,QAAQ,EAAE,KAAK,CAAC,aAAa,EAAE,wBAAwB,EAAE,KAAK,CAAC,wBAAwB,EAAE,CAAC;YAC/G,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;SAC5B,GACD,CACH,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/* eslint-disable @itwin/no-internal */\n\nimport { useMemo } from \"react\";\nimport { createFilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\nimport { createVisibilityAction } from \"./TreeNodeVisibilityButton.js\";\nimport { useVisibilityButtonHandler } from \"./UseVisibilityButtonHandler.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { TreeItemVisibilityButtonProps } from \"./TreeNodeVisibilityButton.js\";\n/** @beta */\nexport type VisibilityTreeRendererProps = BaseTreeRendererProps & TreeItemVisibilityButtonProps;\n\n/**\n * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTreeRenderer({ getVisibilityButtonState, onVisibilityButtonClick: onClick, actions, ...props }: VisibilityTreeRendererProps) {\n  const { onVisibilityButtonClick } = useVisibilityButtonHandler({ rootNodes: props.rootNodes, isNodeSelected: props.isNodeSelected, onClick });\n  const visibilityButtonProps: TreeItemVisibilityButtonProps = useMemo(\n    () => ({\n      variant: \"eyeball\",\n      getVisibilityButtonState,\n      onVisibilityButtonClick,\n    }),\n    [getVisibilityButtonState, onVisibilityButtonClick],\n  );\n\n  return (\n    <BaseTreeRenderer\n      {...props}\n      actions={[\n        ...(visibilityButtonProps ? [createVisibilityAction(visibilityButtonProps)] : []),\n        createFilterAction({ onFilter: props.onFilterClick, getHierarchyLevelDetails: props.getHierarchyLevelDetails }),\n        ...(actions ? actions : []),\n      ]}\n    />\n  );\n}\n"]}
+{"version":3,"file":"VisibilityTreeRenderer.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/common/components/VisibilityTreeRenderer.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,MAAM,OAAO,CAAC;AAC7C,OAAO,EAAE,YAAY,EAAE,MAAM,uCAAuC,CAAC;AACrE,OAAO,EAAE,gBAAgB,EAAE,MAAM,uBAAuB,CAAC;AACzD,OAAO,EAAE,gBAAgB,EAAE,yBAAyB,EAAE,MAAM,+BAA+B,CAAC;AAC5F,OAAO,EAAE,0BAA0B,EAAE,MAAM,iCAAiC,CAAC;AAa7E;;;GAGG;AACH,MAAM,UAAU,sBAAsB,CAAC,KAAkC;IACvE,MAAM,EACJ,wBAAwB,EACxB,uBAAuB,EAAE,OAAO,EAChC,gBAAgB,EAChB,cAAc,EACd,qBAAqB,EACrB,cAAc,EACd,aAAa,EACb,wBAAwB,EACxB,GAAG,SAAS,EACb,GAAG,KAAK,CAAC;IACV,MAAM,EAAE,uBAAuB,EAAE,GAAG,0BAA0B,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC;IAE9I,MAAM,iBAAiB,GAAG,WAAW,CACnC,CAAC,YAAY,EAAE,EAAE;QACf,OAAO,gBAAgB;YACrB,CAAC,CAAC,gBAAgB,CAAC,YAAY,EAAE,KAAK,CAAC;YACvC,CAAC,CAAC;gBACE,KAAC,gBAAgB,IAAoB,IAAI,EAAE,YAAY,CAAC,UAAU,IAA3C,YAAY,CAAmC;gBACtE,KAAC,YAAY,IAAgB,IAAI,EAAE,YAAY,CAAC,UAAU,EAAE,QAAQ,EAAE,aAAa,EAAE,wBAAwB,EAAE,wBAAwB,IAApH,QAAQ,CAAgH;aAC5I,CAAC;IACR,CAAC,EACD,CAAC,aAAa,EAAE,wBAAwB,EAAE,gBAAgB,EAAE,KAAK,CAAC,CACnE,CAAC;IACF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,cAAc,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EAC1F,CAAC,cAAc,EAAE,KAAK,CAAC,CACxB,CAAC;IACF,MAAM,sBAAsB,GAAG,OAAO,CACpC,GAAG,EAAE,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,qBAAqB,CAAC,YAAY,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EACxG,CAAC,qBAAqB,EAAE,KAAK,CAAC,CAC/B,CAAC;IACF,MAAM,eAAe,GAAG,OAAO,CAC7B,GAAG,EAAE,CAAC,CAAC,cAAc,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,cAAc,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,EAC1E,CAAC,cAAc,EAAE,KAAK,CAAC,CACxB,CAAC;IAEF,OAAO,CACL,KAAC,yBAAyB,IAAC,uBAAuB,EAAE,uBAAuB,EAAE,wBAAwB,EAAE,wBAAwB,YAC7H,KAAC,gBAAgB,OACX,SAAS,EACb,aAAa,EAAE,aAAa,EAC5B,wBAAwB,EAAE,wBAAwB,EAClD,gBAAgB,EAAE,iBAAiB,EACnC,cAAc,EAAE,eAAe,EAC/B,qBAAqB,EAAE,sBAAsB,EAC7C,cAAc,EAAE,eAAe,GAC/B,GACwB,CAC7B,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, useMemo } from \"react\";\nimport { FilterAction } from \"@itwin/presentation-hierarchies-react\";\nimport { BaseTreeRenderer } from \"./BaseTreeRenderer.js\";\nimport { VisibilityAction, VisibilityContextProvider } from \"./TreeNodeVisibilityButton.js\";\nimport { useVisibilityButtonHandler } from \"./UseVisibilityButtonHandler.js\";\n\nimport type { BaseTreeRendererProps } from \"./BaseTreeRenderer.js\";\nimport type { VisibilityContext } from \"./TreeNodeVisibilityButton.js\";\n\n/** @beta */\nexport type VisibilityTreeRendererProps = Omit<BaseTreeRendererProps, \"getInlineActions\" | \"getMenuActions\" | \"getContextMenuActions\" | \"getDecorations\"> & {\n  [Property in keyof Pick<BaseTreeRendererProps, \"getInlineActions\" | \"getMenuActions\" | \"getContextMenuActions\" | \"getDecorations\">]?: (\n    args: Parameters<NonNullable<BaseTreeRendererProps[Property]>>[0],\n    treeRendererProps: VisibilityTreeRendererProps,\n  ) => ReturnType<NonNullable<BaseTreeRendererProps[Property]>>;\n} & VisibilityContext;\n\n/**\n * Tree renderer that renders tree nodes with eye checkboxes for controlling visibility of instances represented by tree nodes.\n * @beta\n */\nexport function VisibilityTreeRenderer(props: VisibilityTreeRendererProps) {\n  const {\n    getVisibilityButtonState,\n    onVisibilityButtonClick: onClick,\n    getInlineActions,\n    getMenuActions,\n    getContextMenuActions,\n    getDecorations,\n    onFilterClick,\n    getHierarchyLevelDetails,\n    ...restProps\n  } = props;\n  const { onVisibilityButtonClick } = useVisibilityButtonHandler({ rootNodes: props.rootNodes, isNodeSelected: props.isNodeSelected, onClick });\n\n  const nodeInlineActions = useCallback<Required<BaseTreeRendererProps>[\"getInlineActions\"]>(\n    (actionsProps) => {\n      return getInlineActions\n        ? getInlineActions(actionsProps, props)\n        : [\n            <VisibilityAction key={\"Visibility\"} node={actionsProps.targetNode} />,\n            <FilterAction key={\"Filter\"} node={actionsProps.targetNode} onFilter={onFilterClick} getHierarchyLevelDetails={getHierarchyLevelDetails} />,\n          ];\n    },\n    [onFilterClick, getHierarchyLevelDetails, getInlineActions, props],\n  );\n  const nodeMenuActions = useMemo<BaseTreeRendererProps[\"getMenuActions\"]>(\n    () => (getMenuActions ? (actionsProps) => getMenuActions(actionsProps, props) : undefined),\n    [getMenuActions, props],\n  );\n  const nodeContextMenuActions = useMemo<BaseTreeRendererProps[\"getContextMenuActions\"]>(\n    () => (getContextMenuActions ? (actionsProps) => getContextMenuActions(actionsProps, props) : undefined),\n    [getContextMenuActions, props],\n  );\n  const nodeDecorations = useMemo<BaseTreeRendererProps[\"getDecorations\"]>(\n    () => (getDecorations ? (node) => getDecorations(node, props) : undefined),\n    [getDecorations, props],\n  );\n\n  return (\n    <VisibilityContextProvider onVisibilityButtonClick={onVisibilityButtonClick} getVisibilityButtonState={getVisibilityButtonState}>\n      <BaseTreeRenderer\n        {...restProps}\n        onFilterClick={onFilterClick}\n        getHierarchyLevelDetails={getHierarchyLevelDetails}\n        getInlineActions={nodeInlineActions}\n        getMenuActions={nodeMenuActions}\n        getContextMenuActions={nodeContextMenuActions}\n        getDecorations={nodeDecorations}\n      />\n    </VisibilityContextProvider>\n  );\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/common/internal/AlwaysAndNeverDrawnElementInfo.d.ts

@@ -0,0 +1,74 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64Arg, Id64String } from "@itwin/core-bentley";
+import type { TreeWidgetViewport } from "../TreeWidgetViewport.js";
+import type { ChildrenTree } from "./Utils.js";
+/** @internal */
+export declare const SET_CHANGE_DEBOUNCE_TIME = 20;
+type SetType = "always" | "never";
+interface GetElementsTreeByModelProps {
+    /** Only always/never drawn elements that have the specified models will be returned. */
+    modelIds: Id64Arg;
+    /**
+     * The type of set from which tree should be retrieved.
+     * `always` - ChildrenTree will be created from `alwaysDrawn` set.
+     * `never` - ChildrenTree will be created from `neverDrawn` set.
+     */
+    setType: SetType;
+}
+interface GetElementsTreeByCategoryProps {
+    modelIds: Id64Arg | undefined;
+    /**
+     * Categories of root elements.
+     *
+     * Elements are filtered by given categories. Children of those elements are also included, no matter their category.
+     */
+    categoryIds: Id64Arg;
+    /**
+     * The type of set from which tree should be retrieved.
+     * `always` - ChildrenTree will be created from `alwaysDrawn` set.
+     * `never` - ChildrenTree will be created from `neverDrawn` set.
+     */
+    setType: SetType;
+}
+interface GetElementsTreeByElementProps extends GetElementsTreeByCategoryProps {
+    /** Path to element for which to get its' child always/never drawn elements. When undefined, models and categories will be used to get the always/never drawn elements. */
+    parentElementIdsPath: Array<Id64Arg>;
+}
+/** @internal */
+export type GetElementsTreeProps = GetElementsTreeByModelProps | GetElementsTreeByCategoryProps | GetElementsTreeByElementProps;
+/**
+ * - `categoryId` is assigned only to the elements in always/never drawn set.
+ * - `isInAlwaysOrNeverDrawnSet` flag determines if key (ECInstanceId) is in always/never set.
+ * @internal
+ */
+export type MapEntry = {
+    isInAlwaysOrNeverDrawnSet: true;
+    categoryId: Id64String;
+} | {
+    isInAlwaysOrNeverDrawnSet: false;
+};
+type CachedNodesMap = ChildrenTree<MapEntry>;
+/** @internal */
+export declare class AlwaysAndNeverDrawnElementInfo implements Disposable {
+    #private;
+    constructor(props: {
+        viewport: TreeWidgetViewport;
+        elementClassName?: string;
+        componentId?: GuidString;
+    });
+    suppressChangeEvents(): void;
+    resumeChangeEvents(): void;
+    getElementsTree({ setType, modelIds, ...props }: GetElementsTreeProps): Observable<CachedNodesMap>;
+    private getChildrenTree;
+    private createCacheEntryObservable;
+    [Symbol.dispose](): void;
+    private queryAlwaysOrNeverDrawnElementInfo;
+    private queryElementInfo;
+    getAlwaysOrNeverDrawnElements(props: GetElementsTreeProps): Observable<Set<string>>;
+    clearAlwaysAndNeverDrawnElements(props: {
+        categoryIds: Id64Arg;
+        modelId: Id64String | undefined;
+    }): Observable<void>;
+}
+export {};
+//# sourceMappingURL=AlwaysAndNeverDrawnElementInfo.d.ts.map