@itwin/tree-widget-react 2.3.1 → 3.0.0-dev.0

package/lib/cjs/components/trees/stateless/categories-tree/CategoriesVisibilityHandler.js

@@ -0,0 +1,94 @@
+"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.StatelessCategoriesVisibilityHandler = void 0;
+const core_bentley_1 = require("@itwin/core-bentley");
+const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
+const CategoriesVisibilityUtils_1 = require("../../CategoriesVisibilityUtils");
+/** @internal */
+class StatelessCategoriesVisibilityHandler {
+    constructor(props) {
+        this.onVisibilityChange = new core_bentley_1.BeEvent();
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onDisplayStyleChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        // eslint-disable-next-line @typescript-eslint/naming-convention
+        this.onViewedCategoriesChanged = () => {
+            this.onVisibilityChangeInternal();
+        };
+        this._viewManager = props.viewManager;
+        this._imodel = props.imodel;
+        this._viewport = props.viewport;
+        // istanbul ignore next
+        this._useAllViewports = props.allViewports ?? false;
+        this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);
+        this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);
+    }
+    dispose() {
+        this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);
+        this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);
+        clearTimeout(this._pendingVisibilityChange);
+    }
+    /** Returns visibility status of the tree node. */
+    getVisibilityStatus(node) {
+        if (!presentation_hierarchies_1.HierarchyNode.isInstancesNode(node)) {
+            return { state: "hidden", isDisabled: true };
+        }
+        return { state: node.parentKeys.length ? this.getSubCategoryVisibility(node) : this.getCategoryVisibility(node) };
+    }
+    async changeVisibility(node, on) {
+        if (!presentation_hierarchies_1.HierarchyNode.isInstancesNode(node)) {
+            return;
+        }
+        // handle subcategory visibility change
+        if (node.parentKeys.length) {
+            const childId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);
+            const parentCategoryId = node.extendedData?.categoryId;
+            // make sure parent category is enabled
+            if (on && parentCategoryId) {
+                await this.enableCategory([parentCategoryId], true, false);
+            }
+            this.enableSubCategory(childId, on);
+            return;
+        }
+        const instanceId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);
+        await this.enableCategory([instanceId], on, true);
+    }
+    getSubCategoryVisibility(node) {
+        const parentCategoryId = node.extendedData?.categoryId;
+        if (!parentCategoryId) {
+            return "hidden";
+        }
+        const subcategoryId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);
+        const isVisible = this._viewport.view.viewsCategory(parentCategoryId) && this._viewport.isSubCategoryVisible(subcategoryId);
+        return isVisible ? "visible" : "hidden";
+    }
+    getCategoryVisibility(node) {
+        const instanceId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);
+        return this._viewport.view.viewsCategory(instanceId) ? "visible" : "hidden";
+    }
+    onVisibilityChangeInternal() {
+        if (this._pendingVisibilityChange) {
+            return;
+        }
+        this._pendingVisibilityChange = setTimeout(() => {
+            this.onVisibilityChange.raiseEvent();
+            this._pendingVisibilityChange = undefined;
+        }, 0);
+    }
+    static getInstanceIdFromHierarchyNode(node) {
+        return presentation_hierarchies_1.HierarchyNode.isInstancesNode(node) && node.key.instanceKeys.length > 0 ? node.key.instanceKeys[0].id : /* istanbul ignore next */ "";
+    }
+    async enableCategory(ids, enabled, enableAllSubCategories = true) {
+        await (0, CategoriesVisibilityUtils_1.enableCategory)(this._viewManager, this._imodel, ids, enabled, this._useAllViewports, enableAllSubCategories);
+    }
+    enableSubCategory(key, enabled) {
+        (0, CategoriesVisibilityUtils_1.enableSubCategory)(this._viewManager, key, enabled, this._useAllViewports);
+    }
+}
+exports.StatelessCategoriesVisibilityHandler = StatelessCategoriesVisibilityHandler;
+//# sourceMappingURL=CategoriesVisibilityHandler.js.map

package/lib/cjs/components/trees/stateless/categories-tree/CategoriesVisibilityHandler.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CategoriesVisibilityHandler.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/categories-tree/CategoriesVisibilityHandler.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,sDAA8C;AAC9C,8EAAgE;AAChE,+EAAoF;AAcpF,gBAAgB;AAChB,MAAa,oCAAoC;IAO/C,YAAY,KAAgD;QAgBrD,uBAAkB,GAAG,IAAI,sBAAO,EAA4B,CAAC;QAiDpE,gEAAgE;QACxD,0BAAqB,GAAG,GAAG,EAAE;YACnC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAEF,gEAAgE;QACxD,8BAAyB,GAAG,GAAG,EAAE;YACvC,IAAI,CAAC,0BAA0B,EAAE,CAAC;QACpC,CAAC,CAAC;QAxEA,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,CAAC;QACtC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,uBAAuB;QACvB,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,YAAY,IAAI,KAAK,CAAC;QACpD,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,WAAW,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAC7E,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,WAAW,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;IACvF,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,SAAS,CAAC,qBAAqB,CAAC,cAAc,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;QAChF,IAAI,CAAC,SAAS,CAAC,yBAAyB,CAAC,cAAc,CAAC,IAAI,CAAC,yBAAyB,CAAC,CAAC;QACxF,YAAY,CAAC,IAAI,CAAC,wBAAwB,CAAC,CAAC;IAC9C,CAAC;IAID,kDAAkD;IAC3C,mBAAmB,CAAC,IAAmB;QAC5C,IAAI,CAAC,wCAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC9C;QACD,OAAO,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC,CAAC,IAAI,CAAC,wBAAwB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,qBAAqB,CAAC,IAAI,CAAC,EAAE,CAAC;IACpH,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,EAAW;QAC5D,IAAI,CAAC,wCAAa,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE;YACxC,OAAO;SACR;QAED,uCAAuC;QACvC,IAAI,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;YAC1B,MAAM,OAAO,GAAG,oCAAoC,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC;YAC1F,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;YAEvD,uCAAuC;YACvC,IAAI,EAAE,IAAI,gBAAgB,EAAE;gBAC1B,MAAM,IAAI,CAAC,cAAc,CAAC,CAAC,gBAAgB,CAAC,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;aAC5D;YAED,IAAI,CAAC,iBAAiB,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YACpC,OAAO;SACR;QAED,MAAM,UAAU,GAAG,oCAAoC,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC;QAC7F,MAAM,IAAI,CAAC,cAAc,CAAC,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,IAAI,CAAC,CAAC;IACpD,CAAC;IAEM,wBAAwB,CAAC,IAAmB;QACjD,MAAM,gBAAgB,GAAG,IAAI,CAAC,YAAY,EAAE,UAAU,CAAC;QACvD,IAAI,CAAC,gBAAgB,EAAE;YACrB,OAAO,QAAQ,CAAC;SACjB;QAED,MAAM,aAAa,GAAG,oCAAoC,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC;QAChG,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,gBAAgB,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,oBAAoB,CAAC,aAAa,CAAC,CAAC;QAC5H,OAAO,SAAS,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC1C,CAAC;IAEM,qBAAqB,CAAC,IAAmB;QAC9C,MAAM,UAAU,GAAG,oCAAoC,CAAC,8BAA8B,CAAC,IAAI,CAAC,CAAC;QAC7F,OAAO,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;IAC9E,CAAC;IAYO,0BAA0B;QAChC,IAAI,IAAI,CAAC,wBAAwB,EAAE;YACjC,OAAO;SACR;QAED,IAAI,CAAC,wBAAwB,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9C,IAAI,CAAC,kBAAkB,CAAC,UAAU,EAAE,CAAC;YACrC,IAAI,CAAC,wBAAwB,GAAG,SAAS,CAAC;QAC5C,CAAC,EAAE,CAAC,CAAC,CAAC;IACR,CAAC;IAEM,MAAM,CAAC,8BAA8B,CAAC,IAAmB;QAC9D,OAAO,wCAAa,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,0BAA0B,CAAC,EAAE,CAAC;IAC/I,CAAC;IAEM,KAAK,CAAC,cAAc,CAAC,GAAa,EAAE,OAAgB,EAAE,sBAAsB,GAAG,IAAI;QACxF,MAAM,IAAA,0CAAc,EAAC,IAAI,CAAC,YAAY,EAAE,IAAI,CAAC,OAAO,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,gBAAgB,EAAE,sBAAsB,CAAC,CAAC;IACrH,CAAC;IAEM,iBAAiB,CAAC,GAAW,EAAE,OAAgB;QACpD,IAAA,6CAAiB,EAAC,IAAI,CAAC,YAAY,EAAE,GAAG,EAAE,OAAO,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;IAC5E,CAAC;CACF;AAxGD,oFAwGC","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 { BeEvent } from \"@itwin/core-bentley\";\nimport { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport { enableCategory, enableSubCategory } from \"../../CategoriesVisibilityUtils\";\n\nimport type { CategoryInfo } from \"../../category-tree/CategoriesTreeButtons\";\nimport type { VisibilityChangeListener, VisibilityStatus } from \"../../VisibilityTreeEventHandler\";\nimport type { IModelConnection, ViewManager, Viewport } from \"@itwin/core-frontend\";\n\ninterface StatelessCategoriesVisibilityHandlerProps {\n  viewManager: ViewManager;\n  imodel: IModelConnection;\n  categories: CategoryInfo[];\n  viewport: Viewport;\n  allViewports?: boolean;\n}\n\n/** @internal */\nexport class StatelessCategoriesVisibilityHandler {\n  private _viewManager: ViewManager;\n  private _imodel: IModelConnection;\n  private _pendingVisibilityChange: any | undefined;\n  private _viewport: Viewport;\n  private _useAllViewports: boolean;\n\n  constructor(props: StatelessCategoriesVisibilityHandlerProps) {\n    this._viewManager = props.viewManager;\n    this._imodel = props.imodel;\n    this._viewport = props.viewport;\n    // istanbul ignore next\n    this._useAllViewports = props.allViewports ?? false;\n    this._viewport.onDisplayStyleChanged.addListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.addListener(this.onViewedCategoriesChanged);\n  }\n\n  public dispose() {\n    this._viewport.onDisplayStyleChanged.removeListener(this.onDisplayStyleChanged);\n    this._viewport.onViewedCategoriesChanged.removeListener(this.onViewedCategoriesChanged);\n    clearTimeout(this._pendingVisibilityChange);\n  }\n\n  public onVisibilityChange = new BeEvent<VisibilityChangeListener>();\n\n  /** Returns visibility status of the tree node. */\n  public getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return { state: \"hidden\", isDisabled: true };\n    }\n    return { state: node.parentKeys.length ? this.getSubCategoryVisibility(node) : this.getCategoryVisibility(node) };\n  }\n\n  public async changeVisibility(node: HierarchyNode, on: boolean) {\n    if (!HierarchyNode.isInstancesNode(node)) {\n      return;\n    }\n\n    // handle subcategory visibility change\n    if (node.parentKeys.length) {\n      const childId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);\n      const parentCategoryId = node.extendedData?.categoryId;\n\n      // make sure parent category is enabled\n      if (on && parentCategoryId) {\n        await this.enableCategory([parentCategoryId], true, false);\n      }\n\n      this.enableSubCategory(childId, on);\n      return;\n    }\n\n    const instanceId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);\n    await this.enableCategory([instanceId], on, true);\n  }\n\n  public getSubCategoryVisibility(node: HierarchyNode) {\n    const parentCategoryId = node.extendedData?.categoryId;\n    if (!parentCategoryId) {\n      return \"hidden\";\n    }\n\n    const subcategoryId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);\n    const isVisible = this._viewport.view.viewsCategory(parentCategoryId) && this._viewport.isSubCategoryVisible(subcategoryId);\n    return isVisible ? \"visible\" : \"hidden\";\n  }\n\n  public getCategoryVisibility(node: HierarchyNode) {\n    const instanceId = StatelessCategoriesVisibilityHandler.getInstanceIdFromHierarchyNode(node);\n    return this._viewport.view.viewsCategory(instanceId) ? \"visible\" : \"hidden\";\n  }\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onDisplayStyleChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  private onViewedCategoriesChanged = () => {\n    this.onVisibilityChangeInternal();\n  };\n\n  private onVisibilityChangeInternal() {\n    if (this._pendingVisibilityChange) {\n      return;\n    }\n\n    this._pendingVisibilityChange = setTimeout(() => {\n      this.onVisibilityChange.raiseEvent();\n      this._pendingVisibilityChange = undefined;\n    }, 0);\n  }\n\n  public static getInstanceIdFromHierarchyNode(node: HierarchyNode) {\n    return HierarchyNode.isInstancesNode(node) && node.key.instanceKeys.length > 0 ? node.key.instanceKeys[0].id : /* istanbul ignore next */ \"\";\n  }\n\n  public async enableCategory(ids: string[], enabled: boolean, enableAllSubCategories = true) {\n    await enableCategory(this._viewManager, this._imodel, ids, enabled, this._useAllViewports, enableAllSubCategories);\n  }\n\n  public enableSubCategory(key: string, enabled: boolean) {\n    enableSubCategory(this._viewManager, key, enabled, this._useAllViewports);\n  }\n}\n"]}

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContext.d.ts

@@ -0,0 +1,13 @@
+/// <reference types="react" />
+import type { InstanceKey } from "@itwin/presentation-shared";
+/** @internal */
+export interface FocusedInstancesContext {
+    instanceKeys?: InstanceKey[];
+    enabled: boolean;
+    toggle: () => void;
+}
+/** @internal */
+export declare const focusedInstancesContext: import("react").Context<FocusedInstancesContext>;
+/** @internal */
+export declare function useFocusedInstancesContext(): FocusedInstancesContext;
+//# sourceMappingURL=FocusedInstancesContext.d.ts.map

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContext.js

@@ -0,0 +1,16 @@
+"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.useFocusedInstancesContext = exports.focusedInstancesContext = void 0;
+const react_1 = require("react");
+/** @internal */
+exports.focusedInstancesContext = (0, react_1.createContext)({ enabled: false, toggle: () => { } });
+/** @internal */
+function useFocusedInstancesContext() {
+    return (0, react_1.useContext)(exports.focusedInstancesContext);
+}
+exports.useFocusedInstancesContext = useFocusedInstancesContext;
+//# sourceMappingURL=FocusedInstancesContext.js.map

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContext.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FocusedInstancesContext.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/common/FocusedInstancesContext.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAkD;AAWlD,gBAAgB;AACH,QAAA,uBAAuB,GAAG,IAAA,qBAAa,EAA0B,EAAE,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,GAAG,EAAE,GAAE,CAAC,EAAE,CAAC,CAAC;AAEpH,gBAAgB;AAChB,SAAgB,0BAA0B;IACxC,OAAO,IAAA,kBAAU,EAAC,+BAAuB,CAAC,CAAC;AAC7C,CAAC;AAFD,gEAEC","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 { createContext, useContext } from \"react\";\n\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\n/** @internal */\nexport interface FocusedInstancesContext {\n  instanceKeys?: InstanceKey[];\n  enabled: boolean;\n  toggle: () => void;\n}\n\n/** @internal */\nexport const focusedInstancesContext = createContext<FocusedInstancesContext>({ enabled: false, toggle: () => {} });\n\n/** @internal */\nexport function useFocusedInstancesContext() {\n  return useContext(focusedInstancesContext);\n}\n"]}

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContextProvider.d.ts

@@ -0,0 +1,8 @@
+import type { PropsWithChildren } from "react";
+import type { SelectionStorage } from "@itwin/presentation-hierarchies-react";
+/** @internal */
+export declare function FocusedInstancesContextProvider({ selectionStorage, imodelKey, children, }: PropsWithChildren<{
+    selectionStorage: SelectionStorage;
+    imodelKey: string;
+}>): JSX.Element;
+//# sourceMappingURL=FocusedInstancesContextProvider.d.ts.map

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContextProvider.js

@@ -0,0 +1,47 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.FocusedInstancesContextProvider = void 0;
+const jsx_runtime_1 = require("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.
+ *--------------------------------------------------------------------------------------------*/
+const react_1 = require("react");
+const unified_selection_1 = require("@itwin/unified-selection");
+const FocusedInstancesContext_1 = require("./FocusedInstancesContext");
+/** @internal */
+function FocusedInstancesContextProvider({ selectionStorage, imodelKey, children, }) {
+    const [state, setState] = (0, react_1.useState)({
+        enabled: false,
+        toggle: () => {
+            setState((prev) => ({ ...prev, enabled: !prev.enabled }));
+        },
+    });
+    const enabled = state.enabled;
+    (0, react_1.useEffect)(() => {
+        if (!enabled) {
+            setState((prev) => ({ ...prev, instanceKeys: undefined }));
+            return;
+        }
+        const onSelectionChanged = () => {
+            const selection = selectionStorage.getSelection({ imodelKey, level: 0 });
+            const selectedInstanceKeys = [];
+            unified_selection_1.Selectables.forEach(selection, (key) => {
+                if (unified_selection_1.Selectable.isInstanceKey(key)) {
+                    selectedInstanceKeys.push(key);
+                }
+            });
+            setState((prev) => ({ ...prev, instanceKeys: selectedInstanceKeys.length === 0 ? undefined : selectedInstanceKeys }));
+        };
+        onSelectionChanged();
+        return selectionStorage.selectionChangeEvent.addListener(({ imodelKey: changeImodelKey, level }) => {
+            if (changeImodelKey !== imodelKey || level !== 0) {
+                return;
+            }
+            onSelectionChanged();
+        });
+    }, [enabled, imodelKey, selectionStorage]);
+    return (0, jsx_runtime_1.jsx)(FocusedInstancesContext_1.focusedInstancesContext.Provider, { value: state, children: children });
+}
+exports.FocusedInstancesContextProvider = FocusedInstancesContextProvider;
+//# sourceMappingURL=FocusedInstancesContextProvider.js.map

package/lib/cjs/components/trees/stateless/common/FocusedInstancesContextProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"FocusedInstancesContextProvider.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/common/FocusedInstancesContextProvider.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAEhG,iCAA4C;AAC5C,gEAAmE;AACnE,uEAAoE;AAOpE,gBAAgB;AAChB,SAAgB,+BAA+B,CAAC,EAC9C,gBAAgB,EAChB,SAAS,EACT,QAAQ,GACqE;IAC7E,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAA0B;QAC1D,OAAO,EAAE,KAAK;QACd,MAAM,EAAE,GAAG,EAAE;YACX,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC,CAAC,CAAC;QAC5D,CAAC;KACF,CAAC,CAAC;IACH,MAAM,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;IAE9B,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,IAAI,CAAC,OAAO,EAAE;YACZ,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,SAAS,EAAE,CAAC,CAAC,CAAC;YAC3D,OAAO;SACR;QAED,MAAM,kBAAkB,GAAG,GAAG,EAAE;YAC9B,MAAM,SAAS,GAAG,gBAAgB,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC;YACzE,MAAM,oBAAoB,GAAkB,EAAE,CAAC;YAC/C,+BAAW,CAAC,OAAO,CAAC,SAAS,EAAE,CAAC,GAAG,EAAE,EAAE;gBACrC,IAAI,8BAAU,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE;oBACjC,oBAAoB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;iBAChC;YACH,CAAC,CAAC,CAAC;YACH,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,YAAY,EAAE,oBAAoB,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,oBAAoB,EAAE,CAAC,CAAC,CAAC;QACxH,CAAC,CAAC;QAEF,kBAAkB,EAAE,CAAC;QACrB,OAAO,gBAAgB,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,EAAE,SAAS,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,EAAE;YACjG,IAAI,eAAe,KAAK,SAAS,IAAI,KAAK,KAAK,CAAC,EAAE;gBAChD,OAAO;aACR;YACD,kBAAkB,EAAE,CAAC;QACvB,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,CAAC,OAAO,EAAE,SAAS,EAAE,gBAAgB,CAAC,CAAC,CAAC;IAE3C,OAAO,uBAAC,iDAAuB,CAAC,QAAQ,IAAC,KAAK,EAAE,KAAK,YAAG,QAAQ,GAAoC,CAAC;AACvG,CAAC;AAxCD,0EAwCC","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, useState } from \"react\";\nimport { Selectable, Selectables } from \"@itwin/unified-selection\";\nimport { focusedInstancesContext } from \"./FocusedInstancesContext\";\n\nimport type { PropsWithChildren } from \"react\";\nimport type { InstanceKey } from \"@itwin/presentation-common\";\nimport type { SelectionStorage } from \"@itwin/presentation-hierarchies-react\";\nimport type { FocusedInstancesContext } from \"./FocusedInstancesContext\";\n\n/** @internal */\nexport function FocusedInstancesContextProvider({\n  selectionStorage,\n  imodelKey,\n  children,\n}: PropsWithChildren<{ selectionStorage: SelectionStorage; imodelKey: string }>) {\n  const [state, setState] = useState<FocusedInstancesContext>({\n    enabled: false,\n    toggle: () => {\n      setState((prev) => ({ ...prev, enabled: !prev.enabled }));\n    },\n  });\n  const enabled = state.enabled;\n\n  useEffect(() => {\n    if (!enabled) {\n      setState((prev) => ({ ...prev, instanceKeys: undefined }));\n      return;\n    }\n\n    const onSelectionChanged = () => {\n      const selection = selectionStorage.getSelection({ imodelKey, level: 0 });\n      const selectedInstanceKeys: InstanceKey[] = [];\n      Selectables.forEach(selection, (key) => {\n        if (Selectable.isInstanceKey(key)) {\n          selectedInstanceKeys.push(key);\n        }\n      });\n      setState((prev) => ({ ...prev, instanceKeys: selectedInstanceKeys.length === 0 ? undefined : selectedInstanceKeys }));\n    };\n\n    onSelectionChanged();\n    return selectionStorage.selectionChangeEvent.addListener(({ imodelKey: changeImodelKey, level }) => {\n      if (changeImodelKey !== imodelKey || level !== 0) {\n        return;\n      }\n      onSelectionChanged();\n    });\n  }, [enabled, imodelKey, selectionStorage]);\n\n  return <focusedInstancesContext.Provider value={state}>{children}</focusedInstancesContext.Provider>;\n}\n"]}

package/lib/cjs/components/trees/stateless/common/UseHierarchiesLocalization.d.ts

@@ -0,0 +1,8 @@
+import type { TreeRenderer, useTree } from "@itwin/presentation-hierarchies-react";
+type UseTreeLocalizedStrings = Parameters<typeof useTree>[0]["localizedStrings"];
+type TreeRendererLocalizedStrings = Parameters<typeof TreeRenderer>[0]["localizedStrings"];
+type useHierarchiesLocalizationResult = UseTreeLocalizedStrings & TreeRendererLocalizedStrings;
+/** @internal */
+export declare function useHierarchiesLocalization(): useHierarchiesLocalizationResult;
+export {};
+//# sourceMappingURL=UseHierarchiesLocalization.d.ts.map

package/lib/cjs/components/trees/stateless/common/UseHierarchiesLocalization.js

@@ -0,0 +1,33 @@
+"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.useHierarchiesLocalization = void 0;
+const react_1 = require("react");
+const TreeWidget_1 = require("../../../../TreeWidget");
+/** @internal */
+function useHierarchiesLocalization() {
+    const stringValues = Object.values(getLocalizedStrings());
+    const localizedStrings = (0, react_1.useMemo)(getLocalizedStrings, stringValues);
+    return localizedStrings;
+}
+exports.useHierarchiesLocalization = useHierarchiesLocalization;
+function getLocalizedStrings() {
+    return {
+        // strings for the `useUnifiedSelectionTree` hook
+        unspecified: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.unspecified"),
+        other: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.other"),
+        // strings for `TreeRenderer` and `TreeNodeRenderer`
+        loading: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.loading"),
+        filterHierarchyLevel: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.filterHierarchyLevel"),
+        clearHierarchyLevelFilter: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.clearHierarchyLevelFilter"),
+        noFilteredChildren: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.noFilteredChildren"),
+        resultLimitExceeded: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.resultLimitExceeded"),
+        resultLimitExceededWithFiltering: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.resultLimitExceededWithFiltering"),
+        increaseHierarchyLimit: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.increaseHierarchyLimit"),
+        increaseHierarchyLimitWithFiltering: TreeWidget_1.TreeWidget.translate("stateless.hierarchies.increaseHierarchyLimitWithFiltering"),
+    };
+}
+//# sourceMappingURL=UseHierarchiesLocalization.js.map

package/lib/cjs/components/trees/stateless/common/UseHierarchiesLocalization.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseHierarchiesLocalization.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/common/UseHierarchiesLocalization.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAgC;AAChC,uDAAoD;AASpD,gBAAgB;AAChB,SAAgB,0BAA0B;IACxC,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAG,CAAC,CAAC;IAC3D,MAAM,gBAAgB,GAAG,IAAA,eAAO,EAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;IAEpE,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AALD,gEAKC;AAED,SAAS,mBAAmB;IAC1B,OAAO;QACL,iDAAiD;QACjD,WAAW,EAAE,uBAAU,CAAC,SAAS,CAAC,mCAAmC,CAAC;QACtE,KAAK,EAAE,uBAAU,CAAC,SAAS,CAAC,6BAA6B,CAAC;QAE1D,oDAAoD;QACpD,OAAO,EAAE,uBAAU,CAAC,SAAS,CAAC,+BAA+B,CAAC;QAC9D,oBAAoB,EAAE,uBAAU,CAAC,SAAS,CAAC,4CAA4C,CAAC;QACxF,yBAAyB,EAAE,uBAAU,CAAC,SAAS,CAAC,iDAAiD,CAAC;QAClG,kBAAkB,EAAE,uBAAU,CAAC,SAAS,CAAC,0CAA0C,CAAC;QACpF,mBAAmB,EAAE,uBAAU,CAAC,SAAS,CAAC,2CAA2C,CAAC;QACtF,gCAAgC,EAAE,uBAAU,CAAC,SAAS,CAAC,wDAAwD,CAAC;QAChH,sBAAsB,EAAE,uBAAU,CAAC,SAAS,CAAC,8CAA8C,CAAC;QAC5F,mCAAmC,EAAE,uBAAU,CAAC,SAAS,CAAC,2DAA2D,CAAC;KACvH,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 { useMemo } from \"react\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\n\nimport type { TreeRenderer, useTree } from \"@itwin/presentation-hierarchies-react\";\n\ntype UseTreeLocalizedStrings = Parameters<typeof useTree>[0][\"localizedStrings\"];\ntype TreeRendererLocalizedStrings = Parameters<typeof TreeRenderer>[0][\"localizedStrings\"];\n\ntype useHierarchiesLocalizationResult = UseTreeLocalizedStrings & TreeRendererLocalizedStrings;\n\n/** @internal */\nexport function useHierarchiesLocalization(): useHierarchiesLocalizationResult {\n  const stringValues = Object.values(getLocalizedStrings()!);\n  const localizedStrings = useMemo(getLocalizedStrings, stringValues);\n\n  return localizedStrings;\n}\n\nfunction getLocalizedStrings(): useHierarchiesLocalizationResult {\n  return {\n    // strings for the `useUnifiedSelectionTree` hook\n    unspecified: TreeWidget.translate(\"stateless.hierarchies.unspecified\"),\n    other: TreeWidget.translate(\"stateless.hierarchies.other\"),\n\n    // strings for `TreeRenderer` and `TreeNodeRenderer`\n    loading: TreeWidget.translate(\"stateless.hierarchies.loading\"),\n    filterHierarchyLevel: TreeWidget.translate(\"stateless.hierarchies.filterHierarchyLevel\"),\n    clearHierarchyLevelFilter: TreeWidget.translate(\"stateless.hierarchies.clearHierarchyLevelFilter\"),\n    noFilteredChildren: TreeWidget.translate(\"stateless.hierarchies.noFilteredChildren\"),\n    resultLimitExceeded: TreeWidget.translate(\"stateless.hierarchies.resultLimitExceeded\"),\n    resultLimitExceededWithFiltering: TreeWidget.translate(\"stateless.hierarchies.resultLimitExceededWithFiltering\"),\n    increaseHierarchyLimit: TreeWidget.translate(\"stateless.hierarchies.increaseHierarchyLimit\"),\n    increaseHierarchyLimitWithFiltering: TreeWidget.translate(\"stateless.hierarchies.increaseHierarchyLimitWithFiltering\"),\n  };\n}\n"]}

package/lib/cjs/components/trees/stateless/common/UseHierarchyFiltering.d.ts

@@ -0,0 +1,16 @@
+/// <reference types="react" />
+import type { IModelConnection } from "@itwin/core-frontend";
+import type { useTree } from "@itwin/presentation-hierarchies-react";
+type UseTreeResult = ReturnType<typeof useTree>;
+interface UseHierarchyLevelFilteringOwnProps {
+    imodel: IModelConnection;
+    defaultHierarchyLevelSizeLimit: number;
+}
+type UseHierarchyLevelFilteringProps = UseHierarchyLevelFilteringOwnProps & Pick<UseTreeResult, "getHierarchyLevelDetails">;
+/** @internal */
+export declare function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit, getHierarchyLevelDetails }: UseHierarchyLevelFilteringProps): {
+    onFilterClick: (nodeId: string | undefined) => void;
+    filteringDialog: JSX.Element;
+};
+export {};
+//# sourceMappingURL=UseHierarchyFiltering.d.ts.map

package/lib/cjs/components/trees/stateless/common/UseHierarchyFiltering.js

@@ -0,0 +1,139 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.useHierarchyLevelFiltering = void 0;
+const jsx_runtime_1 = require("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.
+ *--------------------------------------------------------------------------------------------*/
+const react_1 = require("react");
+const components_react_1 = require("@itwin/components-react");
+const presentation_common_1 = require("@itwin/presentation-common");
+const presentation_components_1 = require("@itwin/presentation-components");
+const presentation_frontend_1 = require("@itwin/presentation-frontend");
+const presentation_hierarchies_1 = require("@itwin/presentation-hierarchies");
+const TreeWidget_1 = require("../../../../TreeWidget");
+/** @internal */
+function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit, getHierarchyLevelDetails }) {
+    const [filteringOptions, setFilteringOptions] = (0, react_1.useState)();
+    const onFilterClick = (0, react_1.useCallback)((nodeId) => {
+        const hierarchyDetails = getHierarchyLevelDetails(nodeId);
+        setFilteringOptions(hierarchyDetails ? { nodeId, hierarchyDetails } : undefined);
+    }, [getHierarchyLevelDetails]);
+    const propertiesSource = (0, react_1.useMemo)(() => {
+        if (!filteringOptions) {
+            return undefined;
+        }
+        return async () => {
+            const inputKeys = await collectInstanceKeys(filteringOptions.hierarchyDetails.getInstanceKeysIterator());
+            if (inputKeys.length === 0) {
+                throw new Error("Hierarchy level is empty - unable to create content descriptor.");
+            }
+            const descriptor = await presentation_frontend_1.Presentation.presentation.getContentDescriptor({
+                imodel,
+                rulesetOrId: {
+                    id: `Hierarchy level descriptor ruleset`,
+                    rules: [
+                        {
+                            ruleType: "Content",
+                            specifications: [
+                                {
+                                    specType: "SelectedNodeInstances",
+                                },
+                            ],
+                        },
+                    ],
+                },
+                displayType: presentation_common_1.DefaultContentDisplayTypes.PropertyPane,
+                keys: new presentation_common_1.KeySet(inputKeys),
+            });
+            if (!descriptor) {
+                throw new Error("Failed to create content descriptor");
+            }
+            return { descriptor, inputKeys };
+        };
+    }, [filteringOptions, imodel]);
+    const getInitialFilter = (0, react_1.useMemo)(() => {
+        const currentFilter = filteringOptions?.hierarchyDetails.instanceFilter;
+        if (!currentFilter) {
+            return undefined;
+        }
+        return (descriptor) => fromGenericFilter(descriptor, currentFilter);
+    }, [filteringOptions]);
+    const filteringDialog = ((0, jsx_runtime_1.jsx)(presentation_components_1.PresentationInstanceFilterDialog, { imodel: imodel, isOpen: !!filteringOptions, onApply: (info) => {
+            if (!filteringOptions) {
+                return;
+            }
+            filteringOptions.hierarchyDetails.setInstanceFilter(toGenericFilter(info));
+            setFilteringOptions(undefined);
+        }, onClose: () => {
+            setFilteringOptions(undefined);
+        }, propertiesSource: propertiesSource, initialFilter: getInitialFilter, filterResultsCountRenderer: (filter) => {
+            if (!filteringOptions) {
+                return null;
+            }
+            return ((0, jsx_runtime_1.jsx)(MatchingInstancesCount, { filter: filter, hierarchyLevelDetails: filteringOptions.hierarchyDetails, defaultHierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit }));
+        } }));
+    return {
+        onFilterClick,
+        filteringDialog,
+    };
+}
+exports.useHierarchyLevelFiltering = useHierarchyLevelFiltering;
+function MatchingInstancesCount({ filter, defaultHierarchyLevelSizeLimit, hierarchyLevelDetails }) {
+    const { value } = (0, components_react_1.useDebouncedAsyncValue)((0, react_1.useCallback)(async () => {
+        const instanceFilter = toGenericFilter(filter);
+        try {
+            const instanceKeys = await collectInstanceKeys(hierarchyLevelDetails.getInstanceKeysIterator({
+                instanceFilter,
+                hierarchyLevelSizeLimit: hierarchyLevelDetails.sizeLimit ?? defaultHierarchyLevelSizeLimit,
+            }));
+            return TreeWidget_1.TreeWidget.translate("stateless.matchingInstancesCount", { count: instanceKeys.length });
+        }
+        catch (e) {
+            if (e instanceof presentation_hierarchies_1.RowsLimitExceededError) {
+                return TreeWidget_1.TreeWidget.translate("stateless.filterExceedsLimit", { limit: e.limit });
+            }
+            return TreeWidget_1.TreeWidget.translate("stateless.failedToCalculateMatchingInstances");
+        }
+    }, [filter, hierarchyLevelDetails, defaultHierarchyLevelSizeLimit]));
+    if (!value) {
+        return null;
+    }
+    return (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, { children: value });
+}
+async function collectInstanceKeys(iterator) {
+    const inputKeys = [];
+    for await (const inputKey of iterator) {
+        inputKeys.push(inputKey);
+    }
+    return inputKeys;
+}
+function fromGenericFilter(descriptor, filter) {
+    const presentationFilter = presentation_hierarchies_1.GenericInstanceFilter.isFilterRuleGroup(filter.rules) && filter.rules.rules.length === 0
+        ? undefined
+        : presentation_components_1.PresentationInstanceFilter.fromGenericInstanceFilter(descriptor, filter);
+    return {
+        filter: presentationFilter,
+        usedClasses: (filter.filteredClassNames ?? [])
+            .map((name) => descriptor.selectClasses.find((selectClass) => selectClass.selectClassInfo.name === name)?.selectClassInfo)
+            .filter((classInfo) => classInfo !== undefined),
+    };
+}
+function toGenericFilter(filterInfo) {
+    if (!filterInfo) {
+        return undefined;
+    }
+    if (!filterInfo.filter) {
+        return filterInfo.usedClasses.length > 0
+            ? {
+                propertyClassNames: [],
+                relatedInstances: [],
+                filteredClassNames: filterInfo.usedClasses.map((info) => info.name),
+                rules: { operator: "and", rules: [] },
+            }
+            : undefined;
+    }
+    return presentation_components_1.PresentationInstanceFilter.toGenericInstanceFilter(filterInfo.filter, filterInfo.usedClasses);
+}
+//# sourceMappingURL=UseHierarchyFiltering.js.map

package/lib/cjs/components/trees/stateless/common/UseHierarchyFiltering.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseHierarchyFiltering.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/common/UseHierarchyFiltering.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAEhG,iCAAuD;AACvD,8DAAiE;AACjE,oEAAgF;AAChF,4EAA8G;AAC9G,wEAA4D;AAC5D,8EAAgG;AAChG,uDAAoD;AAiBpD,gBAAgB;AAChB,SAAgB,0BAA0B,CAAC,EAAE,MAAM,EAAE,8BAA8B,EAAE,wBAAwB,EAAmC;IAC9I,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,GAA2E,CAAC;IACpI,MAAM,aAAa,GAAG,IAAA,mBAAW,EAC/B,CAAC,MAA0B,EAAE,EAAE;QAC7B,MAAM,gBAAgB,GAAG,wBAAwB,CAAC,MAAM,CAAC,CAAC;QAC1D,mBAAmB,CAAC,gBAAgB,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,gBAAgB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;IACnF,CAAC,EACD,CAAC,wBAAwB,CAAC,CAC3B,CAAC;IAEF,MAAM,gBAAgB,GAAG,IAAA,eAAO,EAA0E,GAAG,EAAE;QAC7G,IAAI,CAAC,gBAAgB,EAAE;YACrB,OAAO,SAAS,CAAC;SAClB;QAED,OAAO,KAAK,IAAI,EAAE;YAChB,MAAM,SAAS,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,gBAAgB,CAAC,uBAAuB,EAAE,CAAC,CAAC;YACzG,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,EAAE;gBAC1B,MAAM,IAAI,KAAK,CAAC,iEAAiE,CAAC,CAAC;aACpF;YAED,MAAM,UAAU,GAAG,MAAM,oCAAY,CAAC,YAAY,CAAC,oBAAoB,CAAC;gBACtE,MAAM;gBACN,WAAW,EAAE;oBACX,EAAE,EAAE,oCAAoC;oBACxC,KAAK,EAAE;wBACL;4BACE,QAAQ,EAAE,SAAS;4BACnB,cAAc,EAAE;gCACd;oCACE,QAAQ,EAAE,uBAAuB;iCAClC;6BACF;yBACF;qBACF;iBACF;gBACD,WAAW,EAAE,gDAA0B,CAAC,YAAY;gBACpD,IAAI,EAAE,IAAI,4BAAM,CAAC,SAAS,CAAC;aAC5B,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,EAAE;gBACf,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;aACxD;YAED,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,CAAC;QACnC,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,gBAAgB,EAAE,MAAM,CAAC,CAAC,CAAC;IAE/B,MAAM,gBAAgB,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QACpC,MAAM,aAAa,GAAG,gBAAgB,EAAE,gBAAgB,CAAC,cAAc,CAAC;QACxE,IAAI,CAAC,aAAa,EAAE;YAClB,OAAO,SAAS,CAAC;SAClB;QAED,OAAO,CAAC,UAAsB,EAAE,EAAE,CAAC,iBAAiB,CAAC,UAAU,EAAE,aAAa,CAAC,CAAC;IAClF,CAAC,EAAE,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAEvB,MAAM,eAAe,GAAG,CACtB,uBAAC,0DAAgC,IAC/B,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,CAAC,CAAC,gBAAgB,EAC1B,OAAO,EAAE,CAAC,IAAI,EAAE,EAAE;YAChB,IAAI,CAAC,gBAAgB,EAAE;gBACrB,OAAO;aACR;YACD,gBAAgB,CAAC,gBAAgB,CAAC,iBAAiB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;YAC3E,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC,EACD,OAAO,EAAE,GAAG,EAAE;YACZ,mBAAmB,CAAC,SAAS,CAAC,CAAC;QACjC,CAAC,EACD,gBAAgB,EAAE,gBAAgB,EAClC,aAAa,EAAE,gBAAgB,EAC/B,0BAA0B,EAAE,CAAC,MAAM,EAAE,EAAE;YACrC,IAAI,CAAC,gBAAgB,EAAE;gBACrB,OAAO,IAAI,CAAC;aACb;YAED,OAAO,CACL,uBAAC,sBAAsB,IACrB,MAAM,EAAE,MAAM,EACd,qBAAqB,EAAE,gBAAgB,CAAC,gBAAgB,EACxD,8BAA8B,EAAE,8BAA8B,GAC9D,CACH,CAAC;QACJ,CAAC,GACD,CACH,CAAC;IAEF,OAAO;QACL,aAAa;QACb,eAAe;KAChB,CAAC;AACJ,CAAC;AA5FD,gEA4FC;AAQD,SAAS,sBAAsB,CAAC,EAAE,MAAM,EAAE,8BAA8B,EAAE,qBAAqB,EAA+B;IAC5H,MAAM,EAAE,KAAK,EAAE,GAAG,IAAA,yCAAsB,EACtC,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QACrB,MAAM,cAAc,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QAC/C,IAAI;YACF,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAC5C,qBAAqB,CAAC,uBAAuB,CAAC;gBAC5C,cAAc;gBACd,uBAAuB,EAAE,qBAAqB,CAAC,SAAS,IAAI,8BAA8B;aAC3F,CAAC,CACH,CAAC;YACF,OAAO,uBAAU,CAAC,SAAS,CAAC,kCAAkC,EAAE,EAAE,KAAK,EAAE,YAAY,CAAC,MAAM,EAAE,CAAC,CAAC;SACjG;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,YAAY,iDAAsB,EAAE;gBACvC,OAAO,uBAAU,CAAC,SAAS,CAAC,8BAA8B,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,EAAE,CAAC,CAAC;aACjF;YACD,OAAO,uBAAU,CAAC,SAAS,CAAC,8CAA8C,CAAC,CAAC;SAC7E;IACH,CAAC,EAAE,CAAC,MAAM,EAAE,qBAAqB,EAAE,8BAA8B,CAAC,CAAC,CACpE,CAAC;IAEF,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,IAAI,CAAC;KACb;IAED,OAAO,2DAAG,KAAK,GAAI,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,QAA4C;IAC7E,MAAM,SAAS,GAAG,EAAE,CAAC;IACrB,IAAI,KAAK,EAAE,MAAM,QAAQ,IAAI,QAAQ,EAAE;QACrC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;KAC1B;IACD,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,iBAAiB,CAAC,UAAsB,EAAE,MAA6B;IAC9E,MAAM,kBAAkB,GACtB,gDAAqB,CAAC,iBAAiB,CAAC,MAAM,CAAC,KAAK,CAAC,IAAI,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,KAAK,CAAC;QACtF,CAAC,CAAC,SAAS;QACX,CAAC,CAAC,oDAA0B,CAAC,yBAAyB,CAAC,UAAU,EAAE,MAAM,CAAC,CAAC;IAC/E,OAAO;QACL,MAAM,EAAE,kBAAkB;QAC1B,WAAW,EAAE,CAAC,MAAM,CAAC,kBAAkB,IAAI,EAAE,CAAC;aAC3C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,UAAU,CAAC,aAAa,CAAC,IAAI,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,WAAW,CAAC,eAAe,CAAC,IAAI,KAAK,IAAI,CAAC,EAAE,eAAe,CAAC;aACzH,MAAM,CAAC,CAAC,SAAS,EAA0B,EAAE,CAAC,SAAS,KAAK,SAAS,CAAC;KAC1E,CAAC;AACJ,CAAC;AAED,SAAS,eAAe,CAAC,UAA2C;IAClE,IAAI,CAAC,UAAU,EAAE;QACf,OAAO,SAAS,CAAC;KAClB;IAED,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE;QACtB,OAAO,UAAU,CAAC,WAAW,CAAC,MAAM,GAAG,CAAC;YACtC,CAAC,CAAC;gBACE,kBAAkB,EAAE,EAAE;gBACtB,gBAAgB,EAAE,EAAE;gBACpB,kBAAkB,EAAE,UAAU,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC;gBACnE,KAAK,EAAE,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE;aACtC;YACH,CAAC,CAAC,SAAS,CAAC;KACf;IAED,OAAO,oDAA0B,CAAC,uBAAuB,CAAC,UAAU,CAAC,MAAM,EAAE,UAAU,CAAC,WAAW,CAAC,CAAC;AACvG,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, useState } from \"react\";\nimport { useDebouncedAsyncValue } from \"@itwin/components-react\";\nimport { DefaultContentDisplayTypes, KeySet } from \"@itwin/presentation-common\";\nimport { PresentationInstanceFilter, PresentationInstanceFilterDialog } from \"@itwin/presentation-components\";\nimport { Presentation } from \"@itwin/presentation-frontend\";\nimport { GenericInstanceFilter, RowsLimitExceededError } from \"@itwin/presentation-hierarchies\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\n\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { ClassInfo, Descriptor } from \"@itwin/presentation-common\";\nimport type { PresentationInstanceFilterInfo, PresentationInstanceFilterPropertiesSource } from \"@itwin/presentation-components\";\nimport type { HierarchyLevelDetails, useTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ntype UseTreeResult = ReturnType<typeof useTree>;\n\ninterface UseHierarchyLevelFilteringOwnProps {\n  imodel: IModelConnection;\n  defaultHierarchyLevelSizeLimit: number;\n}\n\ntype UseHierarchyLevelFilteringProps = UseHierarchyLevelFilteringOwnProps & Pick<UseTreeResult, \"getHierarchyLevelDetails\">;\n\n/** @internal */\nexport function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit, getHierarchyLevelDetails }: UseHierarchyLevelFilteringProps) {\n  const [filteringOptions, setFilteringOptions] = useState<{ nodeId: string | undefined; hierarchyDetails: HierarchyLevelDetails }>();\n  const onFilterClick = useCallback(\n    (nodeId: string | undefined) => {\n      const hierarchyDetails = getHierarchyLevelDetails(nodeId);\n      setFilteringOptions(hierarchyDetails ? { nodeId, hierarchyDetails } : undefined);\n    },\n    [getHierarchyLevelDetails],\n  );\n\n  const propertiesSource = useMemo<(() => Promise<PresentationInstanceFilterPropertiesSource>) | undefined>(() => {\n    if (!filteringOptions) {\n      return undefined;\n    }\n\n    return async () => {\n      const inputKeys = await collectInstanceKeys(filteringOptions.hierarchyDetails.getInstanceKeysIterator());\n      if (inputKeys.length === 0) {\n        throw new Error(\"Hierarchy level is empty - unable to create content descriptor.\");\n      }\n\n      const descriptor = await Presentation.presentation.getContentDescriptor({\n        imodel,\n        rulesetOrId: {\n          id: `Hierarchy level descriptor ruleset`,\n          rules: [\n            {\n              ruleType: \"Content\",\n              specifications: [\n                {\n                  specType: \"SelectedNodeInstances\",\n                },\n              ],\n            },\n          ],\n        },\n        displayType: DefaultContentDisplayTypes.PropertyPane,\n        keys: new KeySet(inputKeys),\n      });\n      if (!descriptor) {\n        throw new Error(\"Failed to create content descriptor\");\n      }\n\n      return { descriptor, inputKeys };\n    };\n  }, [filteringOptions, imodel]);\n\n  const getInitialFilter = useMemo(() => {\n    const currentFilter = filteringOptions?.hierarchyDetails.instanceFilter;\n    if (!currentFilter) {\n      return undefined;\n    }\n\n    return (descriptor: Descriptor) => fromGenericFilter(descriptor, currentFilter);\n  }, [filteringOptions]);\n\n  const filteringDialog = (\n    <PresentationInstanceFilterDialog\n      imodel={imodel}\n      isOpen={!!filteringOptions}\n      onApply={(info) => {\n        if (!filteringOptions) {\n          return;\n        }\n        filteringOptions.hierarchyDetails.setInstanceFilter(toGenericFilter(info));\n        setFilteringOptions(undefined);\n      }}\n      onClose={() => {\n        setFilteringOptions(undefined);\n      }}\n      propertiesSource={propertiesSource}\n      initialFilter={getInitialFilter}\n      filterResultsCountRenderer={(filter) => {\n        if (!filteringOptions) {\n          return null;\n        }\n\n        return (\n          <MatchingInstancesCount\n            filter={filter}\n            hierarchyLevelDetails={filteringOptions.hierarchyDetails}\n            defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit}\n          />\n        );\n      }}\n    />\n  );\n\n  return {\n    onFilterClick,\n    filteringDialog,\n  };\n}\n\ninterface MatchingInstancesCountProps {\n  filter: PresentationInstanceFilterInfo;\n  hierarchyLevelDetails: HierarchyLevelDetails;\n  defaultHierarchyLevelSizeLimit: number;\n}\n\nfunction MatchingInstancesCount({ filter, defaultHierarchyLevelSizeLimit, hierarchyLevelDetails }: MatchingInstancesCountProps) {\n  const { value } = useDebouncedAsyncValue(\n    useCallback(async () => {\n      const instanceFilter = toGenericFilter(filter);\n      try {\n        const instanceKeys = await collectInstanceKeys(\n          hierarchyLevelDetails.getInstanceKeysIterator({\n            instanceFilter,\n            hierarchyLevelSizeLimit: hierarchyLevelDetails.sizeLimit ?? defaultHierarchyLevelSizeLimit,\n          }),\n        );\n        return TreeWidget.translate(\"stateless.matchingInstancesCount\", { count: instanceKeys.length });\n      } catch (e) {\n        if (e instanceof RowsLimitExceededError) {\n          return TreeWidget.translate(\"stateless.filterExceedsLimit\", { limit: e.limit });\n        }\n        return TreeWidget.translate(\"stateless.failedToCalculateMatchingInstances\");\n      }\n    }, [filter, hierarchyLevelDetails, defaultHierarchyLevelSizeLimit]),\n  );\n\n  if (!value) {\n    return null;\n  }\n\n  return <>{value}</>;\n}\n\nasync function collectInstanceKeys(iterator: AsyncIterableIterator<InstanceKey>) {\n  const inputKeys = [];\n  for await (const inputKey of iterator) {\n    inputKeys.push(inputKey);\n  }\n  return inputKeys;\n}\n\nfunction fromGenericFilter(descriptor: Descriptor, filter: GenericInstanceFilter): PresentationInstanceFilterInfo {\n  const presentationFilter =\n    GenericInstanceFilter.isFilterRuleGroup(filter.rules) && filter.rules.rules.length === 0\n      ? undefined\n      : PresentationInstanceFilter.fromGenericInstanceFilter(descriptor, filter);\n  return {\n    filter: presentationFilter,\n    usedClasses: (filter.filteredClassNames ?? [])\n      .map((name) => descriptor.selectClasses.find((selectClass) => selectClass.selectClassInfo.name === name)?.selectClassInfo)\n      .filter((classInfo): classInfo is ClassInfo => classInfo !== undefined),\n  };\n}\n\nfunction toGenericFilter(filterInfo?: PresentationInstanceFilterInfo): GenericInstanceFilter | undefined {\n  if (!filterInfo) {\n    return undefined;\n  }\n\n  if (!filterInfo.filter) {\n    return filterInfo.usedClasses.length > 0\n      ? {\n          propertyClassNames: [],\n          relatedInstances: [],\n          filteredClassNames: filterInfo.usedClasses.map((info) => info.name),\n          rules: { operator: \"and\", rules: [] },\n        }\n      : undefined;\n  }\n\n  return PresentationInstanceFilter.toGenericInstanceFilter(filterInfo.filter, filterInfo.usedClasses);\n}\n"]}

package/lib/cjs/components/trees/stateless/common/UseHierarchyVisibility.d.ts

@@ -0,0 +1,21 @@
+import type { BeEvent } from "@itwin/core-bentley";
+import type { HierarchyNode } from "@itwin/presentation-hierarchies";
+import type { PresentationHierarchyNode } from "@itwin/presentation-hierarchies-react";
+import type { VisibilityStatus } from "../../VisibilityTreeEventHandler";
+interface HierarchyVisibilityHandler {
+    getVisibilityStatus: (node: HierarchyNode) => Promise<VisibilityStatus>;
+    changeVisibility: (node: HierarchyNode, on: boolean) => Promise<void>;
+    onVisibilityChange: BeEvent<() => void>;
+    dispose: () => void;
+}
+interface UseHierarchyVisibilityProps {
+    visibilityHandlerFactory: () => HierarchyVisibilityHandler;
+}
+interface UseHierarchyVisibilityResult {
+    onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;
+    getCheckboxStatus: (node: PresentationHierarchyNode) => VisibilityStatus;
+}
+/** @internal */
+export declare function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): UseHierarchyVisibilityResult;
+export {};
+//# sourceMappingURL=UseHierarchyVisibility.d.ts.map

package/lib/cjs/components/trees/stateless/common/UseHierarchyVisibility.js

@@ -0,0 +1,82 @@
+"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.useHierarchyVisibility = void 0;
+const react_1 = require("react");
+const rxjs_1 = require("rxjs");
+/** @internal */
+function useHierarchyVisibility({ visibilityHandlerFactory }) {
+    const statusMap = (0, react_1.useRef)(new Map());
+    const [state, setState] = (0, react_1.useState)({
+        getCheckboxStatus: (_node) => ({ state: "hidden", isDisabled: true }),
+        onCheckboxClicked: (_node, _checked) => { },
+    });
+    (0, react_1.useEffect)(() => {
+        statusMap.current.clear();
+        const handler = visibilityHandlerFactory();
+        const visibilityChanged = new rxjs_1.Subject();
+        const calculate = new rxjs_1.Subject();
+        const calculateNodeStatus = (node) => {
+            calculate.next(node);
+        };
+        const subscription = calculate
+            .pipe((0, rxjs_1.distinct)(undefined, visibilityChanged), (0, rxjs_1.mergeMap)((node) => (0, rxjs_1.defer)(async () => ({ node, status: await handler.getVisibilityStatus(node.nodeData) })).pipe((0, rxjs_1.takeUntil)(visibilityChanged))))
+            .subscribe({
+            next: ({ node, status }) => {
+                statusMap.current.set(node.id, { node, status, needsRefresh: false });
+                setState((prev) => ({
+                    ...prev,
+                    getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),
+                }));
+            },
+        });
+        const changeVisibility = (node, checked) => {
+            void handler.changeVisibility(node.nodeData, checked);
+            const status = statusMap.current.get(node.id);
+            if (!status) {
+                return;
+            }
+            status.status.state = checked ? "visible" : "hidden";
+            status.status.tooltip = undefined;
+            setState((prev) => ({ ...prev, getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus) }));
+        };
+        setState({
+            onCheckboxClicked: changeVisibility,
+            getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),
+        });
+        const removeListener = handler.onVisibilityChange.addListener(() => {
+            statusMap.current.forEach((value) => {
+                value.needsRefresh = true;
+            });
+            visibilityChanged.next();
+            setState((prev) => ({
+                ...prev,
+                getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),
+            }));
+        });
+        return () => {
+            subscription.unsubscribe();
+            removeListener();
+            handler.dispose();
+        };
+    }, [visibilityHandlerFactory]);
+    return state;
+}
+exports.useHierarchyVisibility = useHierarchyVisibility;
+function createStatusGetter(map, calculateVisibility) {
+    return (node) => {
+        const status = map.current.get(node.id);
+        if (status === undefined) {
+            calculateVisibility(node);
+            return { state: "hidden", isDisabled: true };
+        }
+        if (status.needsRefresh) {
+            calculateVisibility(node);
+        }
+        return status.status;
+    };
+}
+//# sourceMappingURL=UseHierarchyVisibility.js.map

package/lib/cjs/components/trees/stateless/common/UseHierarchyVisibility.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/components/trees/stateless/common/UseHierarchyVisibility.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAoD;AACpD,+BAAqE;AAwBrE,gBAAgB;AAChB,SAAgB,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,SAAS,GAAG,IAAA,cAAM,EAAC,IAAI,GAAG,EAAgG,CAAC,CAAC;IAClI,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAC;QACjC,iBAAiB,EAAE,CAAC,KAAgC,EAAoB,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAClH,iBAAiB,EAAE,CAAC,KAAgC,EAAE,QAAiB,EAAQ,EAAE,GAAE,CAAC;KACrF,CAAC,CAAC;IAEH,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QAC1B,MAAM,OAAO,GAAG,wBAAwB,EAAE,CAAC;QAE3C,MAAM,iBAAiB,GAAG,IAAI,cAAO,EAAQ,CAAC;QAC9C,MAAM,SAAS,GAAG,IAAI,cAAO,EAA6B,CAAC;QAC3D,MAAM,mBAAmB,GAAG,CAAC,IAA+B,EAAE,EAAE;YAC9D,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,IAAA,eAAQ,EAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,IAAA,eAAQ,EAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAA,YAAK,EAAC,KAAK,IAAI,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,MAAM,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,IAAA,gBAAS,EAAC,iBAAiB,CAAC,CAAC,CAAC,CAC/I;aACA,SAAS,CAAC;YACT,IAAI,EAAE,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,EAAE,EAAE;gBACzB,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,IAAI,EAAE,MAAM,EAAE,YAAY,EAAE,KAAK,EAAE,CAAC,CAAC;gBACtE,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;oBAClB,GAAG,IAAI;oBACP,iBAAiB,EAAE,kBAAkB,CAAC,SAAS,EAAE,mBAAmB,CAAC;iBACtE,CAAC,CAAC,CAAC;YACN,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAG,CAAC,IAA+B,EAAE,OAAgB,EAAE,EAAE;YAC7E,KAAK,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,MAAM,GAAG,SAAS,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YAC9C,IAAI,CAAC,MAAM,EAAE;gBACX,OAAO;aACR;YACD,MAAM,CAAC,MAAM,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;YACrD,MAAM,CAAC,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;YAClC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,iBAAiB,EAAE,kBAAkB,CAAC,SAAS,EAAE,mBAAmB,CAAC,EAAE,CAAC,CAAC,CAAC;QAC3G,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,iBAAiB,EAAE,gBAAgB;YACnC,iBAAiB,EAAE,kBAAkB,CAAC,SAAS,EAAE,mBAAmB,CAAC;SACtE,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,SAAS,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBAClC,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YAEH,iBAAiB,CAAC,IAAI,EAAE,CAAC;YACzB,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,iBAAiB,EAAE,kBAAkB,CAAC,SAAS,EAAE,mBAAmB,CAAC;aACtE,CAAC,CAAC,CAAC;QACN,CAAC,CAAC,CAAC;QAEH,OAAO,GAAG,EAAE;YACV,YAAY,CAAC,WAAW,EAAE,CAAC;YAC3B,cAAc,EAAE,CAAC;YACjB,OAAO,CAAC,OAAO,EAAE,CAAC;QACpB,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,wBAAwB,CAAC,CAAC,CAAC;IAE/B,OAAO,KAAK,CAAC;AACf,CAAC;AApED,wDAoEC;AAED,SAAS,kBAAkB,CACzB,GAAwH,EACxH,mBAA8D;IAE9D,OAAO,CAAC,IAA+B,EAAoB,EAAE;QAC3D,MAAM,MAAM,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,MAAM,KAAK,SAAS,EAAE;YACxB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC9C;QACD,IAAI,MAAM,CAAC,YAAY,EAAE;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;SAC3B;QAED,OAAO,MAAM,CAAC,MAAM,CAAC;IACvB,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 { useEffect, useRef, useState } from \"react\";\nimport { defer, distinct, mergeMap, Subject, takeUntil } from \"rxjs\";\n\nimport type { MutableRefObject } from \"react\";\nimport type { BeEvent } from \"@itwin/core-bentley\";\nimport type { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityStatus } from \"../../VisibilityTreeEventHandler\";\n\ninterface HierarchyVisibilityHandler {\n  getVisibilityStatus: (node: HierarchyNode) => Promise<VisibilityStatus>;\n  changeVisibility: (node: HierarchyNode, on: boolean) => Promise<void>;\n  onVisibilityChange: BeEvent<() => void>;\n  dispose: () => void;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\ninterface UseHierarchyVisibilityResult {\n  onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;\n  getCheckboxStatus: (node: PresentationHierarchyNode) => VisibilityStatus;\n}\n\n/** @internal */\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): UseHierarchyVisibilityResult {\n  const statusMap = useRef(new Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>());\n  const [state, setState] = useState({\n    getCheckboxStatus: (_node: PresentationHierarchyNode): VisibilityStatus => ({ state: \"hidden\", isDisabled: true }),\n    onCheckboxClicked: (_node: PresentationHierarchyNode, _checked: boolean): void => {},\n  });\n\n  useEffect(() => {\n    statusMap.current.clear();\n    const handler = visibilityHandlerFactory();\n\n    const visibilityChanged = new Subject<void>();\n    const calculate = new Subject<PresentationHierarchyNode>();\n    const calculateNodeStatus = (node: PresentationHierarchyNode) => {\n      calculate.next(node);\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        mergeMap((node) => defer(async () => ({ node, status: await handler.getVisibilityStatus(node.nodeData) })).pipe(takeUntil(visibilityChanged))),\n      )\n      .subscribe({\n        next: ({ node, status }) => {\n          statusMap.current.set(node.id, { node, status, needsRefresh: false });\n          setState((prev) => ({\n            ...prev,\n            getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),\n          }));\n        },\n      });\n\n    const changeVisibility = (node: PresentationHierarchyNode, checked: boolean) => {\n      void handler.changeVisibility(node.nodeData, checked);\n      const status = statusMap.current.get(node.id);\n      if (!status) {\n        return;\n      }\n      status.status.state = checked ? \"visible\" : \"hidden\";\n      status.status.tooltip = undefined;\n      setState((prev) => ({ ...prev, getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus) }));\n    };\n\n    setState({\n      onCheckboxClicked: changeVisibility,\n      getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      statusMap.current.forEach((value) => {\n        value.needsRefresh = true;\n      });\n\n      visibilityChanged.next();\n      setState((prev) => ({\n        ...prev,\n        getCheckboxStatus: createStatusGetter(statusMap, calculateNodeStatus),\n      }));\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler.dispose();\n    };\n  }, [visibilityHandlerFactory]);\n\n  return state;\n}\n\nfunction createStatusGetter(\n  map: MutableRefObject<Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>>,\n  calculateVisibility: (node: PresentationHierarchyNode) => void,\n) {\n  return (node: PresentationHierarchyNode): VisibilityStatus => {\n    const status = map.current.get(node.id);\n    if (status === undefined) {\n      calculateVisibility(node);\n      return { state: \"hidden\", isDisabled: true };\n    }\n    if (status.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    return status.status;\n  };\n}\n"]}

package/lib/cjs/components/trees/stateless/common/UseMultiCheckboxHandler.d.ts

@@ -0,0 +1,13 @@
+import type { PresentationHierarchyNode, PresentationTreeNode } from "@itwin/presentation-hierarchies-react";
+interface UseMultiCheckboxHandlerProps {
+    rootNodes: PresentationTreeNode[] | undefined;
+    isNodeSelected: (nodeId: string) => boolean;
+    onClick: (node: PresentationHierarchyNode, checked: boolean) => void;
+}
+interface UseMultiCheckboxHandlerResult {
+    onCheckboxClicked: (node: PresentationHierarchyNode, checked: boolean) => void;
+}
+/** @internal */
+export declare function useMultiCheckboxHandler({ rootNodes, isNodeSelected, onClick }: UseMultiCheckboxHandlerProps): UseMultiCheckboxHandlerResult;
+export {};
+//# sourceMappingURL=UseMultiCheckboxHandler.d.ts.map