@itwin/tree-widget-react 3.4.1 → 4.0.0-alpha.1

package/lib/cjs/tree-widget-react/components/trees/common/Tooltip.js

@@ -1,29 +0,0 @@
-"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.createTooltip = exports.createVisibilityStatus = void 0;
-const TreeWidget_js_1 = require("../../../TreeWidget.js");
-function createVisibilityStatus(status, { useTooltip } = {}) {
-    return {
-        state: status === "disabled" ? "hidden" : status,
-        isDisabled: status === "disabled",
-        tooltip: useTooltip === false ? undefined : createTooltip(status, useTooltip),
-    };
-}
-exports.createVisibilityStatus = createVisibilityStatus;
-/** @internal */
-function createTooltip(status, tooltipStringId) {
-    const statusStringId = `visibilityTooltips.status.${status}`;
-    const statusString = TreeWidget_js_1.TreeWidget.translate(statusStringId);
-    if (!tooltipStringId) {
-        return statusString;
-    }
-    tooltipStringId = `visibilityTooltips.${tooltipStringId}`;
-    const tooltipString = TreeWidget_js_1.TreeWidget.translate(tooltipStringId);
-    return `${statusString}: ${tooltipString}`;
-}
-exports.createTooltip = createTooltip;
-//# sourceMappingURL=Tooltip.js.map

package/lib/cjs/tree-widget-react/components/trees/common/Tooltip.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"Tooltip.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/Tooltip.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,0DAAoD;AAmBpD,SAAgB,sBAAsB,CAAC,MAA+B,EAAE,EAAE,UAAU,KAA8B,EAAE;IAClH,OAAO;QACL,KAAK,EAAE,MAAM,KAAK,UAAU,CAAC,CAAC,CAAC,QAAQ,CAAC,CAAC,CAAC,MAAM;QAChD,UAAU,EAAE,MAAM,KAAK,UAAU;QACjC,OAAO,EAAE,UAAU,KAAK,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,aAAa,CAAC,MAAM,EAAE,UAAU,CAAC;KAC9E,CAAC;AACJ,CAAC;AAND,wDAMC;AAED,gBAAgB;AAChB,SAAgB,aAAa,CAAC,MAA+B,EAAE,eAAmC;IAChG,MAAM,cAAc,GAAG,6BAA6B,MAAM,EAAE,CAAC;IAC7D,MAAM,YAAY,GAAG,0BAAU,CAAC,SAAS,CAAC,cAAc,CAAC,CAAC;IAC1D,IAAI,CAAC,eAAe,EAAE;QACpB,OAAO,YAAY,CAAC;KACrB;IAED,eAAe,GAAG,sBAAsB,eAAe,EAAE,CAAC;IAC1D,MAAM,aAAa,GAAG,0BAAU,CAAC,SAAS,CAAC,eAAe,CAAC,CAAC;IAC5D,OAAO,GAAG,YAAY,KAAK,aAAa,EAAE,CAAC;AAC7C,CAAC;AAVD,sCAUC","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 { TreeWidget } from \"../../../TreeWidget.js\";\n\nimport type { VisibilityStatus } from \"./UseHierarchyVisibility.js\";\n\n/** @internal */\nexport type Visibility = \"visible\" | \"hidden\" | \"partial\";\n\n/** @internal */\nexport type NonPartialVisibilityStatus = Omit<VisibilityStatus, \"state\"> & { state: \"visible\" | \"hidden\" };\n\ninterface VisibilityStatusOptions {\n  // id of localized string to use as an additional tooltip or false to not use tooltip at all\n  // if undefined tooltip will be created based only on the visibility status\n  useTooltip?: string | false;\n}\n\n/** @internal */\nexport function createVisibilityStatus(status: \"visible\" | \"hidden\", options?: VisibilityStatusOptions): NonPartialVisibilityStatus;\nexport function createVisibilityStatus(status: \"visible\" | \"hidden\" | \"partial\" | \"disabled\", options?: VisibilityStatusOptions): VisibilityStatus;\nexport function createVisibilityStatus(status: Visibility | \"disabled\", { useTooltip }: VisibilityStatusOptions = {}): VisibilityStatus {\n  return {\n    state: status === \"disabled\" ? \"hidden\" : status,\n    isDisabled: status === \"disabled\",\n    tooltip: useTooltip === false ? undefined : createTooltip(status, useTooltip),\n  };\n}\n\n/** @internal */\nexport function createTooltip(status: Visibility | \"disabled\", tooltipStringId: string | undefined): string {\n  const statusStringId = `visibilityTooltips.status.${status}`;\n  const statusString = TreeWidget.translate(statusStringId);\n  if (!tooltipStringId) {\n    return statusString;\n  }\n\n  tooltipStringId = `visibilityTooltips.${tooltipStringId}`;\n  const tooltipString = TreeWidget.translate(tooltipStringId);\n  return `${statusString}: ${tooltipString}`;\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/common/TreeErrors.d.ts

@@ -1,9 +0,0 @@
-/**
- * Error that is thrown when too many matches are found while filtering the tree.
- * @beta
- */
-export declare class FilterLimitExceededError extends Error {
-    readonly limit: number;
-    constructor(limit: number);
-}
-//# sourceMappingURL=TreeErrors.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/common/TreeErrors.js

@@ -1,19 +0,0 @@
-"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.FilterLimitExceededError = void 0;
-/**
- * Error that is thrown when too many matches are found while filtering the tree.
- * @beta
- */
-class FilterLimitExceededError extends Error {
-    constructor(limit) {
-        super("Too many filter matches");
-        this.limit = limit;
-    }
-}
-exports.FilterLimitExceededError = FilterLimitExceededError;
-//# sourceMappingURL=TreeErrors.js.map

package/lib/cjs/tree-widget-react/components/trees/common/TreeErrors.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"TreeErrors.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/TreeErrors.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG;;;GAGG;AACH,MAAa,wBAAyB,SAAQ,KAAK;IACjD,YAAmC,KAAa;QAC9C,KAAK,CAAC,yBAAyB,CAAC,CAAC;QADA,UAAK,GAAL,KAAK,CAAQ;IAEhD,CAAC;CACF;AAJD,4DAIC","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\n/**\n * Error that is thrown when too many matches are found while filtering the tree.\n * @beta\n */\nexport class FilterLimitExceededError extends Error {\n  public constructor(public readonly limit: number) {\n    super(\"Too many filter matches\");\n  }\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/common/UseActiveViewport.d.ts

@@ -1,7 +0,0 @@
-import type { ScreenViewport } from "@itwin/core-frontend";
-/**
- * A copy of `useActiveViewport` from `@itwin/appui-react` package until X is fixed.
- * @internal
- */
-export declare function useActiveViewport(): ScreenViewport | undefined;
-//# sourceMappingURL=UseActiveViewport.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/common/UseActiveViewport.js

@@ -1,25 +0,0 @@
-"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.useActiveViewport = void 0;
-const react_1 = require("react");
-const core_frontend_1 = require("@itwin/core-frontend");
-/**
- * A copy of `useActiveViewport` from `@itwin/appui-react` package until X is fixed.
- * @internal
- */
-function useActiveViewport() {
-    const [activeViewport, setActiveViewport] = (0, react_1.useState)(core_frontend_1.IModelApp.viewManager.selectedView);
-    (0, react_1.useEffect)(() => {
-        setActiveViewport(core_frontend_1.IModelApp.viewManager.selectedView);
-        return core_frontend_1.IModelApp.viewManager.onSelectedViewportChanged.addListener((args) => {
-            setActiveViewport(args.current);
-        });
-    }, []);
-    return activeViewport;
-}
-exports.useActiveViewport = useActiveViewport;
-//# sourceMappingURL=UseActiveViewport.js.map

package/lib/cjs/tree-widget-react/components/trees/common/UseActiveViewport.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"UseActiveViewport.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseActiveViewport.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAA4C;AAE5C,wDAAiD;AAEjD;;;GAGG;AACH,SAAgB,iBAAiB;IAC/B,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,IAAA,gBAAQ,EAAC,yBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;IACzF,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,iBAAiB,CAAC,yBAAS,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC;QACtD,OAAO,yBAAS,CAAC,WAAW,CAAC,yBAAyB,CAAC,WAAW,CAAC,CAAC,IAAI,EAAE,EAAE;YAC1E,iBAAiB,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAClC,CAAC,CAAC,CAAC;IACL,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,cAAc,CAAC;AACxB,CAAC;AAVD,8CAUC","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 type { ScreenViewport } from \"@itwin/core-frontend\";\nimport { IModelApp } from \"@itwin/core-frontend\";\n\n/**\n * A copy of `useActiveViewport` from `@itwin/appui-react` package until X is fixed.\n * @internal\n */\nexport function useActiveViewport(): ScreenViewport | undefined {\n  const [activeViewport, setActiveViewport] = useState(IModelApp.viewManager.selectedView);\n  useEffect(() => {\n    setActiveViewport(IModelApp.viewManager.selectedView);\n    return IModelApp.viewManager.onSelectedViewportChanged.addListener((args) => {\n      setActiveViewport(args.current);\n    });\n  }, []);\n\n  return activeViewport;\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/common/UseFiltering.d.ts

@@ -1,11 +0,0 @@
-interface UseFilteringResult {
-    filter: string;
-    activeMatchIndex: number;
-    totalMatches: number;
-    applyFilter: (filter: string) => void;
-    clearFilter: () => void;
-    onHighlightChanged: (index: number, matches?: number) => void;
-}
-export declare function useFiltering(): UseFilteringResult;
-export {};
-//# sourceMappingURL=UseFiltering.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/common/UseFiltering.js

@@ -1,28 +0,0 @@
-"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.useFiltering = void 0;
-const react_1 = require("react");
-function useFiltering() {
-    const [filter, setFilter] = (0, react_1.useState)("");
-    const [activeMatchIndex, setActiveMatchIndex] = (0, react_1.useState)(0);
-    const [totalMatches, setTotalMatches] = (0, react_1.useState)(0);
-    const applyFilter = (0, react_1.useCallback)((newFilter) => {
-        setActiveMatchIndex(0);
-        setFilter(newFilter);
-    }, []);
-    const clearFilter = (0, react_1.useCallback)(() => {
-        setActiveMatchIndex(0);
-        setFilter("");
-    }, []);
-    const onHighlightChanged = (0, react_1.useCallback)((index, matches) => {
-        setActiveMatchIndex(index);
-        matches && setTotalMatches(matches);
-    }, []);
-    return { filter, activeMatchIndex, totalMatches, applyFilter, clearFilter, onHighlightChanged };
-}
-exports.useFiltering = useFiltering;
-//# sourceMappingURL=UseFiltering.js.map

package/lib/cjs/tree-widget-react/components/trees/common/UseFiltering.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"UseFiltering.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseFiltering.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAA8C;AAW9C,SAAgB,YAAY;IAC1B,MAAM,CAAC,MAAM,EAAE,SAAS,CAAC,GAAG,IAAA,gBAAQ,EAAC,EAAE,CAAC,CAAC;IACzC,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,EAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,IAAA,gBAAQ,EAAC,CAAC,CAAC,CAAC;IAEpD,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,CAAC,SAAiB,EAAE,EAAE;QACpD,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACvB,SAAS,CAAC,SAAS,CAAC,CAAC;IACvB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,WAAW,GAAG,IAAA,mBAAW,EAAC,GAAG,EAAE;QACnC,mBAAmB,CAAC,CAAC,CAAC,CAAC;QACvB,SAAS,CAAC,EAAE,CAAC,CAAC;IAChB,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,MAAM,kBAAkB,GAAG,IAAA,mBAAW,EAAC,CAAC,KAAa,EAAE,OAAgB,EAAE,EAAE;QACzE,mBAAmB,CAAC,KAAK,CAAC,CAAC;QAC3B,OAAO,IAAI,eAAe,CAAC,OAAO,CAAC,CAAC;IACtC,CAAC,EAAE,EAAE,CAAC,CAAC;IAEP,OAAO,EAAE,MAAM,EAAE,gBAAgB,EAAE,YAAY,EAAE,WAAW,EAAE,WAAW,EAAE,kBAAkB,EAAE,CAAC;AAClG,CAAC;AArBD,oCAqBC","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, useState } from \"react\";\n\ninterface UseFilteringResult {\n  filter: string;\n  activeMatchIndex: number;\n  totalMatches: number;\n  applyFilter: (filter: string) => void;\n  clearFilter: () => void;\n  onHighlightChanged: (index: number, matches?: number) => void;\n}\n\nexport function useFiltering(): UseFilteringResult {\n  const [filter, setFilter] = useState(\"\");\n  const [activeMatchIndex, setActiveMatchIndex] = useState(0);\n  const [totalMatches, setTotalMatches] = useState(0);\n\n  const applyFilter = useCallback((newFilter: string) => {\n    setActiveMatchIndex(0);\n    setFilter(newFilter);\n  }, []);\n\n  const clearFilter = useCallback(() => {\n    setActiveMatchIndex(0);\n    setFilter(\"\");\n  }, []);\n\n  const onHighlightChanged = useCallback((index: number, matches?: number) => {\n    setActiveMatchIndex(index);\n    matches && setTotalMatches(matches);\n  }, []);\n\n  return { filter, activeMatchIndex, totalMatches, applyFilter, clearFilter, onHighlightChanged };\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchiesLocalization.d.ts

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

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchiesLocalization.js

@@ -1,33 +0,0 @@
-"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_js_1 = require("../../../TreeWidget.js");
-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_js_1.TreeWidget.translate("presentation-hierarchies.unspecified"),
-        other: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.other"),
-        // strings for `TreeRenderer` and `TreeNodeRenderer`
-        loading: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.loading"),
-        filterHierarchyLevel: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.filterHierarchyLevel"),
-        clearHierarchyLevelFilter: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.clearHierarchyLevelFilter"),
-        noFilteredChildren: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.noFilteredChildren"),
-        resultLimitExceeded: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.resultLimitExceeded"),
-        resultLimitExceededWithFiltering: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.resultLimitExceededWithFiltering"),
-        increaseHierarchyLimit: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.increaseHierarchyLimit"),
-        increaseHierarchyLimitWithFiltering: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.increaseHierarchyLimitWithFiltering"),
-        retry: TreeWidget_js_1.TreeWidget.translate("presentation-hierarchies.retry"),
-    };
-}
-//# sourceMappingURL=UseHierarchiesLocalization.js.map

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchiesLocalization.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"UseHierarchiesLocalization.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchiesLocalization.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAgC;AAChC,0DAAoD;AASpD,SAAgB,0BAA0B;IACxC,MAAM,YAAY,GAAG,MAAM,CAAC,MAAM,CAAC,mBAAmB,EAAE,CAAC,CAAC;IAC1D,MAAM,gBAAgB,GAAG,IAAA,eAAO,EAAC,mBAAmB,EAAE,YAAY,CAAC,CAAC;IACpE,OAAO,gBAAgB,CAAC;AAC1B,CAAC;AAJD,gEAIC;AAED,SAAS,mBAAmB;IAC1B,OAAO;QACL,iDAAiD;QACjD,WAAW,EAAE,0BAAU,CAAC,SAAS,CAAC,sCAAsC,CAAC;QACzE,KAAK,EAAE,0BAAU,CAAC,SAAS,CAAC,gCAAgC,CAAC;QAE7D,oDAAoD;QACpD,OAAO,EAAE,0BAAU,CAAC,SAAS,CAAC,kCAAkC,CAAC;QACjE,oBAAoB,EAAE,0BAAU,CAAC,SAAS,CAAC,+CAA+C,CAAC;QAC3F,yBAAyB,EAAE,0BAAU,CAAC,SAAS,CAAC,oDAAoD,CAAC;QACrG,kBAAkB,EAAE,0BAAU,CAAC,SAAS,CAAC,6CAA6C,CAAC;QACvF,mBAAmB,EAAE,0BAAU,CAAC,SAAS,CAAC,8CAA8C,CAAC;QACzF,gCAAgC,EAAE,0BAAU,CAAC,SAAS,CAAC,2DAA2D,CAAC;QACnH,sBAAsB,EAAE,0BAAU,CAAC,SAAS,CAAC,iDAAiD,CAAC;QAC/F,mCAAmC,EAAE,0BAAU,CAAC,SAAS,CAAC,8DAA8D,CAAC;QACzH,KAAK,EAAE,0BAAU,CAAC,SAAS,CAAC,gCAAgC,CAAC;KAC9D,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.js\";\n\nimport type { TreeRenderer, useIModelTree } from \"@itwin/presentation-hierarchies-react\";\n\ntype UseTreeLocalizedStrings = NonNullable<Parameters<typeof useIModelTree>[0][\"localizedStrings\"]>;\ntype TreeRendererLocalizedStrings = NonNullable<Parameters<typeof TreeRenderer>[0][\"localizedStrings\"]>;\n\ntype UseHierarchiesLocalizationResult = UseTreeLocalizedStrings & TreeRendererLocalizedStrings;\n\nexport function useHierarchiesLocalization(): UseHierarchiesLocalizationResult {\n  const stringValues = Object.values(getLocalizedStrings());\n  const localizedStrings = useMemo(getLocalizedStrings, stringValues);\n  return localizedStrings;\n}\n\nfunction getLocalizedStrings(): UseHierarchiesLocalizationResult {\n  return {\n    // strings for the `useUnifiedSelectionTree` hook\n    unspecified: TreeWidget.translate(\"presentation-hierarchies.unspecified\"),\n    other: TreeWidget.translate(\"presentation-hierarchies.other\"),\n\n    // strings for `TreeRenderer` and `TreeNodeRenderer`\n    loading: TreeWidget.translate(\"presentation-hierarchies.loading\"),\n    filterHierarchyLevel: TreeWidget.translate(\"presentation-hierarchies.filterHierarchyLevel\"),\n    clearHierarchyLevelFilter: TreeWidget.translate(\"presentation-hierarchies.clearHierarchyLevelFilter\"),\n    noFilteredChildren: TreeWidget.translate(\"presentation-hierarchies.noFilteredChildren\"),\n    resultLimitExceeded: TreeWidget.translate(\"presentation-hierarchies.resultLimitExceeded\"),\n    resultLimitExceededWithFiltering: TreeWidget.translate(\"presentation-hierarchies.resultLimitExceededWithFiltering\"),\n    increaseHierarchyLimit: TreeWidget.translate(\"presentation-hierarchies.increaseHierarchyLimit\"),\n    increaseHierarchyLimitWithFiltering: TreeWidget.translate(\"presentation-hierarchies.increaseHierarchyLimitWithFiltering\"),\n    retry: TreeWidget.translate(\"presentation-hierarchies.retry\"),\n  };\n}\n"]}

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchyFiltering.d.ts

@@ -1,13 +0,0 @@
-/// <reference types="react" />
-import type { IModelConnection } from "@itwin/core-frontend";
-import type { HierarchyLevelDetails } from "@itwin/presentation-hierarchies-react";
-interface UseHierarchyLevelFilteringProps {
-    imodel: IModelConnection;
-    defaultHierarchyLevelSizeLimit: number;
-}
-export declare function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit }: UseHierarchyLevelFilteringProps): {
-    onFilterClick: import("react").Dispatch<import("react").SetStateAction<HierarchyLevelDetails | undefined>>;
-    filteringDialog: JSX.Element;
-};
-export {};
-//# sourceMappingURL=UseHierarchyFiltering.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchyFiltering.js

@@ -1,158 +0,0 @@
-"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 core_bentley_1 = require("@itwin/core-bentley");
-const itwinui_react_1 = require("@itwin/itwinui-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_js_1 = require("../../../TreeWidget.js");
-const Delayed_js_1 = require("./components/Delayed.js");
-const UseTelemetryContext_js_1 = require("./UseTelemetryContext.js");
-function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit }) {
-    const [filteringOptions, setFilteringOptions] = (0, react_1.useState)();
-    const { onFeatureUsed } = (0, UseTelemetryContext_js_1.useTelemetryContext)();
-    const propertiesSource = (0, react_1.useMemo)(() => {
-        if (!filteringOptions) {
-            return undefined;
-        }
-        return async () => {
-            const keySet = await collectInstanceKeys(filteringOptions.getInstanceKeysIterator());
-            if (keySet.isEmpty) {
-                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: keySet,
-            });
-            if (!descriptor) {
-                throw new Error("Failed to create content descriptor");
-            }
-            return { descriptor, inputKeys: keySet };
-        };
-    }, [filteringOptions, imodel]);
-    const getInitialFilter = (0, react_1.useMemo)(() => {
-        const currentFilter = filteringOptions?.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;
-            }
-            onFeatureUsed({ featureId: info ? "hierarchy-level-filtering" : undefined, reportInteraction: true });
-            filteringOptions.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, defaultHierarchyLevelSizeLimit: defaultHierarchyLevelSizeLimit }));
-        } }));
-    return {
-        onFilterClick: setFilteringOptions,
-        filteringDialog,
-    };
-}
-exports.useHierarchyLevelFiltering = useHierarchyLevelFiltering;
-function MatchingInstancesCount({ filter, defaultHierarchyLevelSizeLimit, hierarchyLevelDetails }) {
-    const { value, inProgress } = (0, components_react_1.useDebouncedAsyncValue)((0, react_1.useCallback)(async () => {
-        const instanceFilter = toGenericFilter(filter);
-        try {
-            const instanceCount = await countInstanceKeys(hierarchyLevelDetails.getInstanceKeysIterator({
-                instanceFilter,
-                hierarchyLevelSizeLimit: hierarchyLevelDetails.sizeLimit ?? defaultHierarchyLevelSizeLimit,
-            }));
-            return TreeWidget_js_1.TreeWidget.translate("filteringDialog.matchingInstancesCount", {
-                instanceCount: instanceCount.toLocaleString(undefined, { useGrouping: true }),
-            });
-        }
-        catch (e) {
-            if (e instanceof presentation_hierarchies_1.RowsLimitExceededError) {
-                return TreeWidget_js_1.TreeWidget.translate("filteringDialog.filterExceedsLimit", { limit: e.limit.toLocaleString(undefined, { useGrouping: true }) });
-            }
-            return TreeWidget_js_1.TreeWidget.translate("filteringDialog.failedToCalculateMatchingInstancesCount");
-        }
-    }, [filter, hierarchyLevelDetails, defaultHierarchyLevelSizeLimit]));
-    if (inProgress) {
-        return ((0, jsx_runtime_1.jsxs)(Delayed_js_1.Delayed, { show: true, children: [TreeWidget_js_1.TreeWidget.translate("filteringDialog.matchingInstancesCount", { instanceCount: "" }), (0, jsx_runtime_1.jsx)(itwinui_react_1.ProgressRadial, { size: "x-small" })] }));
-    }
-    return value ? (0, jsx_runtime_1.jsx)(jsx_runtime_1.Fragment, { children: value }) : null;
-}
-async function countInstanceKeys(iterator) {
-    let result = 0;
-    for await (const _ of iterator) {
-        result++;
-    }
-    return result;
-}
-async function collectInstanceKeys(iterator) {
-    const idsByClassName = new Map();
-    for await (const { className, id } of iterator) {
-        let idSet = idsByClassName.get(className);
-        if (!idSet) {
-            idSet = [];
-            idsByClassName.set(className, idSet);
-        }
-        idSet.push(id);
-    }
-    const instanceKeys = new Array();
-    for (const [className, ids] of idsByClassName) {
-        instanceKeys.push([className.toLowerCase(), core_bentley_1.CompressedId64Set.sortAndCompress(ids)]);
-    }
-    return presentation_common_1.KeySet.fromJSON({ instanceKeys, nodeKeys: [] });
-}
-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/tree-widget-react/components/trees/common/UseHierarchyFiltering.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"UseHierarchyFiltering.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyFiltering.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAEhG,iCAAuD;AACvD,8DAAiE;AACjE,sDAAwD;AACxD,wDAAsD;AACtD,oEAAgF;AAChF,4EAA8G;AAC9G,wEAA4D;AAC5D,8EAAgG;AAChG,0DAAoD;AACpD,wDAAkD;AAClD,qEAA+D;AAc/D,SAAgB,0BAA0B,CAAC,EAAE,MAAM,EAAE,8BAA8B,EAAmC;IACpH,MAAM,CAAC,gBAAgB,EAAE,mBAAmB,CAAC,GAAG,IAAA,gBAAQ,GAAyB,CAAC;IAClF,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,4CAAmB,GAAE,CAAC;IAEhD,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,MAAM,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,uBAAuB,EAAE,CAAC,CAAC;YACrF,IAAI,MAAM,CAAC,OAAO,EAAE;gBAClB,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,MAAM;aACb,CAAC,CAAC;YACH,IAAI,CAAC,UAAU,EAAE;gBACf,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;aACxD;YAED,OAAO,EAAE,UAAU,EAAE,SAAS,EAAE,MAAM,EAAE,CAAC;QAC3C,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,cAAc,CAAC;QACvD,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,aAAa,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,CAAC,CAAC,2BAA2B,CAAC,CAAC,CAAC,SAAS,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACtG,gBAAgB,CAAC,iBAAiB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,CAAC;YAC1D,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,IAAC,MAAM,EAAE,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,CACpJ,CAAC;QACJ,CAAC,GACD,CACH,CAAC;IAEF,OAAO;QACL,aAAa,EAAE,mBAAmB;QAClC,eAAe;KAChB,CAAC;AACJ,CAAC;AAnFD,gEAmFC;AAQD,SAAS,sBAAsB,CAAC,EAAE,MAAM,EAAE,8BAA8B,EAAE,qBAAqB,EAA+B;IAC5H,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,GAAG,IAAA,yCAAsB,EAClD,IAAA,mBAAW,EAAC,KAAK,IAAI,EAAE;QACrB,MAAM,cAAc,GAAG,eAAe,CAAC,MAAM,CAAC,CAAC;QAC/C,IAAI;YACF,MAAM,aAAa,GAAG,MAAM,iBAAiB,CAC3C,qBAAqB,CAAC,uBAAuB,CAAC;gBAC5C,cAAc;gBACd,uBAAuB,EAAE,qBAAqB,CAAC,SAAS,IAAI,8BAA8B;aAC3F,CAAC,CACH,CAAC;YACF,OAAO,0BAAU,CAAC,SAAS,CAAC,wCAAwC,EAAE;gBACpE,aAAa,EAAE,aAAa,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC;aAC9E,CAAC,CAAC;SACJ;QAAC,OAAO,CAAC,EAAE;YACV,IAAI,CAAC,YAAY,iDAAsB,EAAE;gBACvC,OAAO,0BAAU,CAAC,SAAS,CAAC,oCAAoC,EAAE,EAAE,KAAK,EAAE,CAAC,CAAC,KAAK,CAAC,cAAc,CAAC,SAAS,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;aACxI;YACD,OAAO,0BAAU,CAAC,SAAS,CAAC,yDAAyD,CAAC,CAAC;SACxF;IACH,CAAC,EAAE,CAAC,MAAM,EAAE,qBAAqB,EAAE,8BAA8B,CAAC,CAAC,CACpE,CAAC;IAEF,IAAI,UAAU,EAAE;QACd,OAAO,CACL,wBAAC,oBAAO,IAAC,IAAI,EAAE,IAAI,aAChB,0BAAU,CAAC,SAAS,CAAC,wCAAwC,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,CAAC,EACtF,uBAAC,8BAAc,IAAC,IAAI,EAAC,SAAS,GAAG,IACzB,CACX,CAAC;KACH;IAED,OAAO,KAAK,CAAC,CAAC,CAAC,2DAAG,KAAK,GAAI,CAAC,CAAC,CAAC,IAAI,CAAC;AACrC,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,QAA4C;IAC3E,IAAI,MAAM,GAAG,CAAC,CAAC;IACf,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,QAAQ,EAAE;QAC9B,MAAM,EAAE,CAAC;KACV;IACD,OAAO,MAAM,CAAC;AAChB,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,QAA4C;IAC7E,MAAM,cAAc,GAAG,IAAI,GAAG,EAAyB,CAAC;IACxD,IAAI,KAAK,EAAE,MAAM,EAAE,SAAS,EAAE,EAAE,EAAE,IAAI,QAAQ,EAAE;QAC9C,IAAI,KAAK,GAAG,cAAc,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAC1C,IAAI,CAAC,KAAK,EAAE;YACV,KAAK,GAAG,EAAE,CAAC;YACX,cAAc,CAAC,GAAG,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;SACtC;QACD,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;KAChB;IAED,MAAM,YAAY,GAAG,IAAI,KAAK,EAAyD,CAAC;IACxF,KAAK,MAAM,CAAC,SAAS,EAAE,GAAG,CAAC,IAAI,cAAc,EAAE;QAC7C,YAAY,CAAC,IAAI,CAAC,CAAC,SAAS,CAAC,WAAW,EAAE,EAAE,gCAAiB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;KACtF;IACD,OAAO,4BAAM,CAAC,QAAQ,CAAC,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;AACzD,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 { CompressedId64Set } from \"@itwin/core-bentley\";\nimport { ProgressRadial } from \"@itwin/itwinui-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.js\";\nimport { Delayed } from \"./components/Delayed.js\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\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 } from \"@itwin/presentation-hierarchies-react\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface UseHierarchyLevelFilteringProps {\n  imodel: IModelConnection;\n  defaultHierarchyLevelSizeLimit: number;\n}\n\nexport function useHierarchyLevelFiltering({ imodel, defaultHierarchyLevelSizeLimit }: UseHierarchyLevelFilteringProps) {\n  const [filteringOptions, setFilteringOptions] = useState<HierarchyLevelDetails>();\n  const { onFeatureUsed } = useTelemetryContext();\n\n  const propertiesSource = useMemo<(() => Promise<PresentationInstanceFilterPropertiesSource>) | undefined>(() => {\n    if (!filteringOptions) {\n      return undefined;\n    }\n\n    return async () => {\n      const keySet = await collectInstanceKeys(filteringOptions.getInstanceKeysIterator());\n      if (keySet.isEmpty) {\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: keySet,\n      });\n      if (!descriptor) {\n        throw new Error(\"Failed to create content descriptor\");\n      }\n\n      return { descriptor, inputKeys: keySet };\n    };\n  }, [filteringOptions, imodel]);\n\n  const getInitialFilter = useMemo(() => {\n    const currentFilter = filteringOptions?.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        onFeatureUsed({ featureId: info ? \"hierarchy-level-filtering\" : undefined, reportInteraction: true });\n        filteringOptions.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 filter={filter} hierarchyLevelDetails={filteringOptions} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n        );\n      }}\n    />\n  );\n\n  return {\n    onFilterClick: setFilteringOptions,\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, inProgress } = useDebouncedAsyncValue(\n    useCallback(async () => {\n      const instanceFilter = toGenericFilter(filter);\n      try {\n        const instanceCount = await countInstanceKeys(\n          hierarchyLevelDetails.getInstanceKeysIterator({\n            instanceFilter,\n            hierarchyLevelSizeLimit: hierarchyLevelDetails.sizeLimit ?? defaultHierarchyLevelSizeLimit,\n          }),\n        );\n        return TreeWidget.translate(\"filteringDialog.matchingInstancesCount\", {\n          instanceCount: instanceCount.toLocaleString(undefined, { useGrouping: true }),\n        });\n      } catch (e) {\n        if (e instanceof RowsLimitExceededError) {\n          return TreeWidget.translate(\"filteringDialog.filterExceedsLimit\", { limit: e.limit.toLocaleString(undefined, { useGrouping: true }) });\n        }\n        return TreeWidget.translate(\"filteringDialog.failedToCalculateMatchingInstancesCount\");\n      }\n    }, [filter, hierarchyLevelDetails, defaultHierarchyLevelSizeLimit]),\n  );\n\n  if (inProgress) {\n    return (\n      <Delayed show={true}>\n        {TreeWidget.translate(\"filteringDialog.matchingInstancesCount\", { instanceCount: \"\" })}\n        <ProgressRadial size=\"x-small\" />\n      </Delayed>\n    );\n  }\n\n  return value ? <>{value}</> : null;\n}\n\nasync function countInstanceKeys(iterator: AsyncIterableIterator<InstanceKey>) {\n  let result = 0;\n  for await (const _ of iterator) {\n    result++;\n  }\n  return result;\n}\n\nasync function collectInstanceKeys(iterator: AsyncIterableIterator<InstanceKey>) {\n  const idsByClassName = new Map<Id64String, Id64Array>();\n  for await (const { className, id } of iterator) {\n    let idSet = idsByClassName.get(className);\n    if (!idSet) {\n      idSet = [];\n      idsByClassName.set(className, idSet);\n    }\n    idSet.push(id);\n  }\n\n  const instanceKeys = new Array<[className: string, compressedIds: CompressedId64Set]>();\n  for (const [className, ids] of idsByClassName) {\n    instanceKeys.push([className.toLowerCase(), CompressedId64Set.sortAndCompress(ids)]);\n  }\n  return KeySet.fromJSON({ instanceKeys, nodeKeys: [] });\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/tree-widget-react/components/trees/common/UseHierarchyVisibility.d.ts

@@ -1,55 +0,0 @@
-import type { Observable } from "rxjs";
-import type { BeEvent, IDisposable } from "@itwin/core-bentley";
-import type { HierarchyNode } from "@itwin/presentation-hierarchies-react";
-import type { TreeCheckboxProps } from "./components/TreeNodeCheckbox.js";
-/**
- * Data structure that describes instance visibility status.
- * @beta
- */
-export interface VisibilityStatus {
-    /** Instance visibility state. */
-    state: "visible" | "partial" | "hidden";
-    /** Specifies whether visibility changing is disabled or not. */
-    isDisabled?: boolean;
-    /** Tooltip that should be displayed when hovering over the visibility checkbox. */
-    tooltip?: string;
-}
-/**
- * Handler that can be used to determine and change visibility of instances represented by tree nodes.
- * @beta
- */
-export interface HierarchyVisibilityHandler extends IDisposable {
-    /** Event used to notify tree about visibility changes from outside. */
-    readonly onVisibilityChange: BeEvent<() => void>;
-    /** Returns current visibility status for tree node. */
-    getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;
-    /** Changes visibility of the instance represented by tree node. */
-    changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;
-}
-interface UseHierarchyVisibilityProps {
-    visibilityHandlerFactory: () => HierarchyVisibilityHandler;
-}
-export declare function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): TreeCheckboxProps & {
-    triggerRefresh: () => void;
-};
-/**
- * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.
- * @beta
- */
-export type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult ? TProps & {
-    /** A callback that produces the value from the original implementation. */
-    readonly originalImplementation: () => TResult;
-    /**
-     * Reference to the hierarchy based handler.
-     * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.
-     */
-    readonly handler: HierarchyVisibilityHandler;
-} : never;
-/**
- * Function type for an overridden method of `HierarchyVisibilityHandler`.
- * @beta
- */
-export type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult : never;
-export declare function createVisibilityHandlerResult<TResult, TOverrideProps>(handler: HierarchyVisibilityHandler, props: TOverrideProps, obs: Observable<TResult>, override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined): Observable<TResult>;
-export {};
-//# sourceMappingURL=UseHierarchyVisibility.d.ts.map

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchyVisibility.js

@@ -1,114 +0,0 @@
-"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.createVisibilityHandlerResult = exports.useHierarchyVisibility = void 0;
-const react_1 = require("react");
-const rxjs_1 = require("rxjs");
-const UseTelemetryContext_js_1 = require("./UseTelemetryContext.js");
-function useHierarchyVisibility({ visibilityHandlerFactory }) {
-    const visibilityStatusMap = (0, react_1.useRef)(new Map());
-    const [state, setState] = (0, react_1.useState)({
-        getCheckboxState: () => ({ state: "off", isDisabled: true }),
-        onCheckboxClicked: () => { },
-        triggerRefresh: () => { },
-    });
-    const { onFeatureUsed } = (0, UseTelemetryContext_js_1.useTelemetryContext)();
-    (0, react_1.useEffect)(() => {
-        visibilityStatusMap.current.clear();
-        const handler = visibilityHandlerFactory();
-        const visibilityChanged = new rxjs_1.Subject();
-        const calculate = new rxjs_1.Subject();
-        const calculateNodeStatus = (node) => {
-            calculate.next(node);
-        };
-        const resetCache = () => {
-            visibilityStatusMap.current.forEach((value) => {
-                value.needsRefresh = true;
-            });
-            visibilityChanged.next();
-        };
-        const triggerCheckboxUpdate = () => {
-            setState((prev) => ({
-                ...prev,
-                getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),
-            }));
-        };
-        const subscription = calculate
-            .pipe((0, rxjs_1.distinct)(undefined, visibilityChanged), (0, rxjs_1.observeOn)(rxjs_1.asyncScheduler), (0, rxjs_1.mergeMap)((node) => (0, rxjs_1.defer)(async () => handler.getVisibilityStatus(node.nodeData)).pipe((0, rxjs_1.tap)({
-            next: (status) => {
-                visibilityStatusMap.current.set(node.id, {
-                    node,
-                    status,
-                    needsRefresh: false,
-                });
-            },
-        }), (0, rxjs_1.takeUntil)(visibilityChanged), (0, rxjs_1.onErrorResumeNextWith)(rxjs_1.EMPTY))), (0, rxjs_1.throttleTime)(100, undefined, { leading: false, trailing: true }))
-            .subscribe({
-            next: () => {
-                triggerCheckboxUpdate();
-            },
-        });
-        const changeVisibility = (node, checked) => {
-            onFeatureUsed({ featureId: "visibility-change", reportInteraction: true });
-            void handler.changeVisibility(node.nodeData, checked);
-            const entry = visibilityStatusMap.current.get(node.id);
-            if (!entry) {
-                return;
-            }
-            entry.status.state = checked ? "visible" : "hidden";
-            entry.status.tooltip = undefined;
-            triggerCheckboxUpdate();
-        };
-        setState({
-            onCheckboxClicked: changeVisibility,
-            getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),
-            triggerRefresh: () => {
-                resetCache();
-                triggerCheckboxUpdate();
-            },
-        });
-        const removeListener = handler.onVisibilityChange.addListener(() => {
-            resetCache();
-            triggerCheckboxUpdate();
-        });
-        return () => {
-            subscription.unsubscribe();
-            removeListener();
-            handler.dispose();
-        };
-    }, [visibilityHandlerFactory, onFeatureUsed]);
-    return state;
-}
-exports.useHierarchyVisibility = useHierarchyVisibility;
-function createStateGetter(map, calculateVisibility) {
-    return (node) => {
-        const entry = map.current.get(node.id);
-        if (entry === undefined) {
-            calculateVisibility(node);
-            return { state: "off", isDisabled: true };
-        }
-        if (entry.needsRefresh) {
-            calculateVisibility(node);
-        }
-        const status = entry.status;
-        return {
-            state: status.state === "visible" ? "on" : status.state === "hidden" ? "off" : "partial",
-            tooltip: status.tooltip,
-            isDisabled: status.isDisabled,
-        };
-    };
-}
-function createVisibilityHandlerResult(handler, props, obs, override) {
-    return override
-        ? (0, rxjs_1.from)(override({
-            ...props,
-            originalImplementation: async () => (0, rxjs_1.lastValueFrom)(obs, { defaultValue: undefined }),
-            handler,
-        }))
-        : obs;
-}
-exports.createVisibilityHandlerResult = createVisibilityHandlerResult;
-//# sourceMappingURL=UseHierarchyVisibility.js.map

package/lib/cjs/tree-widget-react/components/trees/common/UseHierarchyVisibility.js.map

@@ -1 +0,0 @@
-{"version":3,"file":"UseHierarchyVisibility.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/common/UseHierarchyVisibility.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,iCAAoD;AACpD,+BAEc;AACd,qEAA+D;AAsC/D,SAAgB,sBAAsB,CAAC,EAAE,wBAAwB,EAA+B;IAC9F,MAAM,mBAAmB,GAAG,IAAA,cAAM,EAAC,IAAI,GAAG,EAAgG,CAAC,CAAC;IAC5I,MAAM,CAAC,KAAK,EAAE,QAAQ,CAAC,GAAG,IAAA,gBAAQ,EAAqD;QACrF,gBAAgB,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;QAC5D,iBAAiB,EAAE,GAAG,EAAE,GAAE,CAAC;QAC3B,cAAc,EAAE,GAAG,EAAE,GAAE,CAAC;KACzB,CAAC,CAAC;IACH,MAAM,EAAE,aAAa,EAAE,GAAG,IAAA,4CAAmB,GAAE,CAAC;IAEhD,IAAA,iBAAS,EAAC,GAAG,EAAE;QACb,mBAAmB,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;QACpC,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,UAAU,GAAG,GAAG,EAAE;YACtB,mBAAmB,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;gBAC5C,KAAK,CAAC,YAAY,GAAG,IAAI,CAAC;YAC5B,CAAC,CAAC,CAAC;YACH,iBAAiB,CAAC,IAAI,EAAE,CAAC;QAC3B,CAAC,CAAC;QAEF,MAAM,qBAAqB,GAAG,GAAG,EAAE;YACjC,QAAQ,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBAClB,GAAG,IAAI;gBACP,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;aAC9E,CAAC,CAAC,CAAC;QACN,CAAC,CAAC;QAEF,MAAM,YAAY,GAAG,SAAS;aAC3B,IAAI,CACH,IAAA,eAAQ,EAAC,SAAS,EAAE,iBAAiB,CAAC,EACtC,IAAA,gBAAS,EAAC,qBAAc,CAAC,EACzB,IAAA,eAAQ,EAAC,CAAC,IAAI,EAAE,EAAE,CAChB,IAAA,YAAK,EAAC,KAAK,IAAI,EAAE,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,IAAI,CAChE,IAAA,UAAG,EAAC;YACF,IAAI,EAAE,CAAC,MAAM,EAAE,EAAE;gBACf,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,EAAE;oBACvC,IAAI;oBACJ,MAAM;oBACN,YAAY,EAAE,KAAK;iBACpB,CAAC,CAAC;YACL,CAAC;SACF,CAAC,EACF,IAAA,gBAAS,EAAC,iBAAiB,CAAC,EAC5B,IAAA,4BAAqB,EAAC,YAAK,CAAC,CAC7B,CACF,EACD,IAAA,mBAAY,EAAC,GAAG,EAAE,SAAS,EAAE,EAAE,OAAO,EAAE,KAAK,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CACjE;aACA,SAAS,CAAC;YACT,IAAI,EAAE,GAAG,EAAE;gBACT,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEL,MAAM,gBAAgB,GAAG,CAAC,IAA+B,EAAE,OAAgB,EAAE,EAAE;YAC7E,aAAa,CAAC,EAAE,SAAS,EAAE,mBAAmB,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YAC3E,KAAK,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,QAAQ,EAAE,OAAO,CAAC,CAAC;YACtD,MAAM,KAAK,GAAG,mBAAmB,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;YACvD,IAAI,CAAC,KAAK,EAAE;gBACV,OAAO;aACR;YACD,KAAK,CAAC,MAAM,CAAC,KAAK,GAAG,OAAO,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,QAAQ,CAAC;YACpD,KAAK,CAAC,MAAM,CAAC,OAAO,GAAG,SAAS,CAAC;YACjC,qBAAqB,EAAE,CAAC;QAC1B,CAAC,CAAC;QAEF,QAAQ,CAAC;YACP,iBAAiB,EAAE,gBAAgB;YACnC,gBAAgB,EAAE,iBAAiB,CAAC,mBAAmB,EAAE,mBAAmB,CAAC;YAC7E,cAAc,EAAE,GAAG,EAAE;gBACnB,UAAU,EAAE,CAAC;gBACb,qBAAqB,EAAE,CAAC;YAC1B,CAAC;SACF,CAAC,CAAC;QAEH,MAAM,cAAc,GAAG,OAAO,CAAC,kBAAkB,CAAC,WAAW,CAAC,GAAG,EAAE;YACjE,UAAU,EAAE,CAAC;YACb,qBAAqB,EAAE,CAAC;QAC1B,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,EAAE,aAAa,CAAC,CAAC,CAAC;IAE9C,OAAO,KAAK,CAAC;AACf,CAAC;AA9FD,wDA8FC;AAED,SAAS,iBAAiB,CACxB,GAAwH,EACxH,mBAA8D;IAE9D,OAAO,CAAC,IAAI,EAAE,EAAE;QACd,MAAM,KAAK,GAAG,GAAG,CAAC,OAAO,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACvC,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,mBAAmB,CAAC,IAAI,CAAC,CAAC;YAC1B,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,UAAU,EAAE,IAAI,EAAE,CAAC;SAC3C;QACD,IAAI,KAAK,CAAC,YAAY,EAAE;YACtB,mBAAmB,CAAC,IAAI,CAAC,CAAC;SAC3B;QAED,MAAM,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,OAAO;YACL,KAAK,EAAE,MAAM,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;YACxF,OAAO,EAAE,MAAM,CAAC,OAAO;YACvB,UAAU,EAAE,MAAM,CAAC,UAAU;SAC9B,CAAC;IACJ,CAAC,CAAC;AACJ,CAAC;AA0BD,SAAgB,6BAA6B,CAC3C,OAAmC,EACnC,KAAqB,EACrB,GAAwB,EACxB,QAA8G;IAE9G,OAAO,QAAQ;QACb,CAAC,CAAC,IAAA,WAAI,EACF,QAAQ,CAAC;YACP,GAAG,KAAK;YACR,sBAAsB,EAAE,KAAK,IAAI,EAAE,CAAC,IAAA,oBAAa,EAAC,GAAG,EAAE,EAAE,YAAY,EAAE,SAAoB,EAAE,CAAC;YAC9F,OAAO;SACR,CAAC,CACH;QACH,CAAC,CAAC,GAAG,CAAC;AACV,CAAC;AAfD,sEAeC","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 {\n  asyncScheduler, defer, distinct, EMPTY, from, lastValueFrom, mergeMap, observeOn, onErrorResumeNextWith, Subject, takeUntil, tap, throttleTime,\n} from \"rxjs\";\nimport { useTelemetryContext } from \"./UseTelemetryContext.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { MutableRefObject } from \"react\";\nimport type { BeEvent, IDisposable } from \"@itwin/core-bentley\";\nimport type { HierarchyNode, PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { TreeCheckboxProps } from \"./components/TreeNodeCheckbox.js\";\n\n/**\n * Data structure that describes instance visibility status.\n * @beta\n */\nexport interface VisibilityStatus {\n  /** Instance visibility state. */\n  state: \"visible\" | \"partial\" | \"hidden\";\n  /** Specifies whether visibility changing is disabled or not. */\n  isDisabled?: boolean;\n  /** Tooltip that should be displayed when hovering over the visibility checkbox. */\n  tooltip?: string;\n}\n\n/**\n * Handler that can be used to determine and change visibility of instances represented by tree nodes.\n * @beta\n */\nexport interface HierarchyVisibilityHandler extends IDisposable {\n  /** Event used to notify tree about visibility changes from outside. */\n  readonly onVisibilityChange: BeEvent<() => void>;\n  /** Returns current visibility status for tree node. */\n  getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> | VisibilityStatus;\n  /** Changes visibility of the instance represented by tree node. */\n  changeVisibility(node: HierarchyNode, on: boolean): Promise<void>;\n}\n\ninterface UseHierarchyVisibilityProps {\n  visibilityHandlerFactory: () => HierarchyVisibilityHandler;\n}\n\nexport function useHierarchyVisibility({ visibilityHandlerFactory }: UseHierarchyVisibilityProps): TreeCheckboxProps & { triggerRefresh: () => void } {\n  const visibilityStatusMap = useRef(new Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>());\n  const [state, setState] = useState<TreeCheckboxProps & { triggerRefresh: () => void }>({\n    getCheckboxState: () => ({ state: \"off\", isDisabled: true }),\n    onCheckboxClicked: () => {},\n    triggerRefresh: () => {},\n  });\n  const { onFeatureUsed } = useTelemetryContext();\n\n  useEffect(() => {\n    visibilityStatusMap.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 resetCache = () => {\n      visibilityStatusMap.current.forEach((value) => {\n        value.needsRefresh = true;\n      });\n      visibilityChanged.next();\n    };\n\n    const triggerCheckboxUpdate = () => {\n      setState((prev) => ({\n        ...prev,\n        getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      }));\n    };\n\n    const subscription = calculate\n      .pipe(\n        distinct(undefined, visibilityChanged),\n        observeOn(asyncScheduler),\n        mergeMap((node) =>\n          defer(async () => handler.getVisibilityStatus(node.nodeData)).pipe(\n            tap({\n              next: (status) => {\n                visibilityStatusMap.current.set(node.id, {\n                  node,\n                  status,\n                  needsRefresh: false,\n                });\n              },\n            }),\n            takeUntil(visibilityChanged),\n            onErrorResumeNextWith(EMPTY),\n          ),\n        ),\n        throttleTime(100, undefined, { leading: false, trailing: true }),\n      )\n      .subscribe({\n        next: () => {\n          triggerCheckboxUpdate();\n        },\n      });\n\n    const changeVisibility = (node: PresentationHierarchyNode, checked: boolean) => {\n      onFeatureUsed({ featureId: \"visibility-change\", reportInteraction: true });\n      void handler.changeVisibility(node.nodeData, checked);\n      const entry = visibilityStatusMap.current.get(node.id);\n      if (!entry) {\n        return;\n      }\n      entry.status.state = checked ? \"visible\" : \"hidden\";\n      entry.status.tooltip = undefined;\n      triggerCheckboxUpdate();\n    };\n\n    setState({\n      onCheckboxClicked: changeVisibility,\n      getCheckboxState: createStateGetter(visibilityStatusMap, calculateNodeStatus),\n      triggerRefresh: () => {\n        resetCache();\n        triggerCheckboxUpdate();\n      },\n    });\n\n    const removeListener = handler.onVisibilityChange.addListener(() => {\n      resetCache();\n      triggerCheckboxUpdate();\n    });\n\n    return () => {\n      subscription.unsubscribe();\n      removeListener();\n      handler.dispose();\n    };\n  }, [visibilityHandlerFactory, onFeatureUsed]);\n\n  return state;\n}\n\nfunction createStateGetter(\n  map: MutableRefObject<Map<string, { node: PresentationHierarchyNode; status: VisibilityStatus; needsRefresh: boolean }>>,\n  calculateVisibility: (node: PresentationHierarchyNode) => void,\n): TreeCheckboxProps[\"getCheckboxState\"] {\n  return (node) => {\n    const entry = map.current.get(node.id);\n    if (entry === undefined) {\n      calculateVisibility(node);\n      return { state: \"off\", isDisabled: true };\n    }\n    if (entry.needsRefresh) {\n      calculateVisibility(node);\n    }\n\n    const status = entry.status;\n    return {\n      state: status.state === \"visible\" ? \"on\" : status.state === \"hidden\" ? \"off\" : \"partial\",\n      tooltip: status.tooltip,\n      isDisabled: status.isDisabled,\n    };\n  };\n}\n\n/**\n * Properties for an overridden method of a `HierarchyVisibilityHandler` implementation.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethodProps<TFunc> = TFunc extends (props: infer TProps) => infer TResult\n  ? TProps & {\n      /** A callback that produces the value from the original implementation. */\n      readonly originalImplementation: () => TResult;\n      /**\n       * Reference to the hierarchy based handler.\n       * @note Calling `getVisibility` or `changeVisibility` of this object invokes the overridden implementation as well.\n       */\n      readonly handler: HierarchyVisibilityHandler;\n    }\n  : never;\n\n/**\n * Function type for an overridden method of `HierarchyVisibilityHandler`.\n * @beta\n */\nexport type HierarchyVisibilityHandlerOverridableMethod<TFunc> = TFunc extends (...args: any[]) => infer TResult\n  ? (props: HierarchyVisibilityHandlerOverridableMethodProps<TFunc>) => TResult\n  : never;\n\nexport function createVisibilityHandlerResult<TResult, TOverrideProps>(\n  handler: HierarchyVisibilityHandler,\n  props: TOverrideProps,\n  obs: Observable<TResult>,\n  override: HierarchyVisibilityHandlerOverridableMethod<(props: TOverrideProps) => Promise<TResult>> | undefined,\n): Observable<TResult> {\n  return override\n    ? from(\n        override({\n          ...props,\n          originalImplementation: async () => lastValueFrom(obs, { defaultValue: undefined as TResult }),\n          handler,\n        }),\n      )\n    : obs;\n}\n"]}