@itwin/tree-widget-react 4.0.0-alpha.25 → 4.0.0-alpha.26

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

@@ -0,0 +1,23 @@
+import type { Observable } from "rxjs";
+import type { GuidString, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { CategoryId, SubCategoryId } from "../Types.js";
+interface SubCategoriesCacheProps {
+    queryExecutor: LimitingECSqlQueryExecutor;
+    componentId?: GuidString;
+}
+/** @internal */
+export declare class SubCategoriesCache {
+    #private;
+    constructor(props: SubCategoriesCacheProps);
+    private querySubCategories;
+    getSubCategoriesInfo(): Observable<{
+        subCategoryCategories: Map<SubCategoryId, CategoryId>;
+        categorySubCategories: Map<CategoryId, Array<SubCategoryId>>;
+    }>;
+    getSubCategories(props: {
+        categoryId: Id64String;
+    }): Observable<Array<SubCategoryId>>;
+}
+export {};
+//# sourceMappingURL=SubCategoriesCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/SubCategoriesCache.js

@@ -0,0 +1,56 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { defer, map, reduce, shareReplay } from "rxjs";
+import { Guid } from "@itwin/core-bentley";
+import { CLASS_NAME_SubCategory } from "../ClassNameDefinitions.js";
+import { catchBeSQLiteInterrupts } from "../UseErrorState.js";
+/** @internal */
+export class SubCategoriesCache {
+    #queryExecutor;
+    #componentId;
+    #componentName;
+    #subCategoriesInfo;
+    constructor(props) {
+        this.#queryExecutor = props.queryExecutor;
+        this.#componentId = props.componentId ?? Guid.createValue();
+        this.#componentName = "SubCategoriesCache";
+    }
+    querySubCategories() {
+        return defer(() => {
+            const ecsql = `
+        SELECT
+          sc.ECInstanceId id,
+          sc.Parent.Id categoryId
+        FROM
+          ${CLASS_NAME_SubCategory} sc
+        WHERE
+          NOT sc.IsPrivate
+      `;
+            return this.#queryExecutor.createQueryReader({ ecsql }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded", restartToken: `${this.#componentName}/${this.#componentId}/sub-categories` });
+        }).pipe(catchBeSQLiteInterrupts, map((row) => {
+            return { id: row.id, parentId: row.categoryId };
+        }));
+    }
+    getSubCategoriesInfo() {
+        this.#subCategoriesInfo ??= this.querySubCategories()
+            .pipe(reduce((acc, queriedSubCategory) => {
+            acc.subCategoryCategories.set(queriedSubCategory.id, queriedSubCategory.parentId);
+            const entry = acc.categorySubCategories.get(queriedSubCategory.parentId);
+            if (entry) {
+                entry.push(queriedSubCategory.id);
+            }
+            else {
+                acc.categorySubCategories.set(queriedSubCategory.parentId, [queriedSubCategory.id]);
+            }
+            return acc;
+        }, { subCategoryCategories: new Map(), categorySubCategories: new Map() }))
+            .pipe(shareReplay());
+        return this.#subCategoriesInfo;
+    }
+    getSubCategories(props) {
+        return this.getSubCategoriesInfo().pipe(map(({ categorySubCategories }) => categorySubCategories.get(props.categoryId) ?? []));
+    }
+}
+//# sourceMappingURL=SubCategoriesCache.js.map

package/lib/esm/tree-widget-react/components/trees/common/internal/caches/SubCategoriesCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"SubCategoriesCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/caches/SubCategoriesCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,MAAM,CAAC;AACvD,OAAO,EAAE,IAAI,EAAE,MAAM,qBAAqB,CAAC;AAC3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,4BAA4B,CAAC;AACpE,OAAO,EAAE,uBAAuB,EAAE,MAAM,qBAAqB,CAAC;AAY9D,gBAAgB;AAChB,MAAM,OAAO,kBAAkB;IAC7B,cAAc,CAA6B;IAC3C,YAAY,CAAa;IACzB,cAAc,CAAS;IACvB,kBAAkB,CAEJ;IAEd,YAAY,KAA8B;QACxC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,WAAW,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;QAC5D,IAAI,CAAC,cAAc,GAAG,oBAAoB,CAAC;IAC7C,CAAC;IAEO,kBAAkB;QACxB,OAAO,KAAK,CAAC,GAAG,EAAE;YAChB,MAAM,KAAK,GAAG;;;;;YAKR,sBAAsB;;;OAG3B,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAC1C,EAAE,KAAK,EAAE,EACT,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,YAAY,EAAE,GAAG,IAAI,CAAC,cAAc,IAAI,IAAI,CAAC,YAAY,iBAAiB,EAAE,CACpI,CAAC;QACJ,CAAC,CAAC,CAAC,IAAI,CACL,uBAAuB,EACvB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE;YACV,OAAO,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;QAClD,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEM,oBAAoB;QACzB,IAAI,CAAC,kBAAkB,KAAK,IAAI,CAAC,kBAAkB,EAAE;aAClD,IAAI,CACH,MAAM,CACJ,CAAC,GAAG,EAAE,kBAAkB,EAAE,EAAE;YAC1B,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YAClF,MAAM,KAAK,GAAG,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,CAAC,CAAC;YACzE,IAAI,KAAK,EAAE,CAAC;gBACV,KAAK,CAAC,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YACpC,CAAC;iBAAM,CAAC;gBACN,GAAG,CAAC,qBAAqB,CAAC,GAAG,CAAC,kBAAkB,CAAC,QAAQ,EAAE,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC,CAAC;YACtF,CAAC;YACD,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,qBAAqB,EAAE,IAAI,GAAG,EAA6B,EAAE,qBAAqB,EAAE,IAAI,GAAG,EAAoC,EAAE,CACpI,CACF;aACA,IAAI,CAAC,WAAW,EAAE,CAAC,CAAC;QACvB,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEM,gBAAgB,CAAC,KAAiC;QACvD,OAAO,IAAI,CAAC,oBAAoB,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,qBAAqB,EAAE,EAAE,EAAE,CAAC,qBAAqB,CAAC,GAAG,CAAC,KAAK,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC;IACjI,CAAC;CACF","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 { defer, map, reduce, shareReplay } from \"rxjs\";\nimport { Guid } from \"@itwin/core-bentley\";\nimport { CLASS_NAME_SubCategory } from \"../ClassNameDefinitions.js\";\nimport { catchBeSQLiteInterrupts } from \"../UseErrorState.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { CategoryId, SubCategoryId } from \"../Types.js\";\n\ninterface SubCategoriesCacheProps {\n  queryExecutor: LimitingECSqlQueryExecutor;\n  componentId?: GuidString;\n}\n\n/** @internal */\nexport class SubCategoriesCache {\n  #queryExecutor: LimitingECSqlQueryExecutor;\n  #componentId: GuidString;\n  #componentName: string;\n  #subCategoriesInfo:\n    | Observable<{ subCategoryCategories: Map<SubCategoryId, CategoryId>; categorySubCategories: Map<CategoryId, Array<SubCategoryId>> }>\n    | undefined;\n\n  constructor(props: SubCategoriesCacheProps) {\n    this.#queryExecutor = props.queryExecutor;\n    this.#componentId = props.componentId ?? Guid.createValue();\n    this.#componentName = \"SubCategoriesCache\";\n  }\n\n  private querySubCategories(): Observable<{ id: SubCategoryId; parentId: CategoryId }> {\n    return defer(() => {\n      const ecsql = `\n        SELECT\n          sc.ECInstanceId id,\n          sc.Parent.Id categoryId\n        FROM\n          ${CLASS_NAME_SubCategory} sc\n        WHERE\n          NOT sc.IsPrivate\n      `;\n      return this.#queryExecutor.createQueryReader(\n        { ecsql },\n        { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\", restartToken: `${this.#componentName}/${this.#componentId}/sub-categories` },\n      );\n    }).pipe(\n      catchBeSQLiteInterrupts,\n      map((row) => {\n        return { id: row.id, parentId: row.categoryId };\n      }),\n    );\n  }\n\n  public getSubCategoriesInfo() {\n    this.#subCategoriesInfo ??= this.querySubCategories()\n      .pipe(\n        reduce(\n          (acc, queriedSubCategory) => {\n            acc.subCategoryCategories.set(queriedSubCategory.id, queriedSubCategory.parentId);\n            const entry = acc.categorySubCategories.get(queriedSubCategory.parentId);\n            if (entry) {\n              entry.push(queriedSubCategory.id);\n            } else {\n              acc.categorySubCategories.set(queriedSubCategory.parentId, [queriedSubCategory.id]);\n            }\n            return acc;\n          },\n          { subCategoryCategories: new Map<SubCategoryId, CategoryId>(), categorySubCategories: new Map<CategoryId, Array<SubCategoryId>>() },\n        ),\n      )\n      .pipe(shareReplay());\n    return this.#subCategoriesInfo;\n  }\n\n  public getSubCategories(props: { categoryId: Id64String }): Observable<Array<SubCategoryId>> {\n    return this.getSubCategoriesInfo().pipe(map(({ categorySubCategories }) => categorySubCategories.get(props.categoryId) ?? []));\n  }\n}\n"]}

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

@@ -1,6 +1,6 @@
 import { HierarchyNode } from "@itwin/presentation-hierarchies";
 import { HierarchyVisibilityOverrideHandler } from "../../UseHierarchyVisibility.js";
-import { AlwaysAndNeverDrawnElementInfo } from "../AlwaysAndNeverDrawnElementInfo.js";
+import { AlwaysAndNeverDrawnElementInfoCache } from "../caches/AlwaysAndNeverDrawnElementInfoCache.js";
 import type { GuidString } from "@itwin/core-bentley";
 import type { HierarchySearchPath } from "@itwin/presentation-hierarchies";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
@@ -10,14 +10,14 @@ import type { SearchResultsTree } from "../visibility/BaseSearchResultsTree.js";
 import type { TreeSpecificVisibilityHandler } from "../visibility/BaseVisibilityHelper.js";
 /** @internal */
 export interface CreateSearchResultsTreeProps<TCache> {
-    getCache: () => TCache;
+    idsCache: TCache;
     imodelAccess: ECClassHierarchyInspector;
     searchPaths: HierarchySearchPath[];
 }
 /** @internal */
 export interface CreateTreeSpecificVisibilityHandlerProps<TCache> {
-    info: AlwaysAndNeverDrawnElementInfo;
-    getCache: () => TCache;
+    info: AlwaysAndNeverDrawnElementInfoCache;
+    idsCache: TCache;
     viewport: TreeWidgetViewport;
     overrideHandler: HierarchyVisibilityOverrideHandler;
 }
@@ -25,7 +25,7 @@ export interface CreateTreeSpecificVisibilityHandlerProps<TCache> {
 export interface UseCachedVisibilityProps<TCache, TSearchTargets> {
     componentId: GuidString;
     activeView: TreeWidgetViewport;
-    getCache: () => TCache;
+    idsCache: TCache;
     createSearchResultsTree: (props: CreateSearchResultsTreeProps<TCache>) => Promise<SearchResultsTree<TSearchTargets>>;
     createTreeSpecificVisibilityHandler: (props: CreateTreeSpecificVisibilityHandlerProps<TCache>) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;
 }
@@ -40,7 +40,7 @@ export declare function useCachedVisibility<TCache, TSearchTargets>(props: UseCa
 /** @internal */
 export interface HierarchyVisibilityHandlerImplProps<TSearchTargets> {
     viewport: TreeWidgetViewport;
-    getTreeSpecificVisibilityHandler: (info: AlwaysAndNeverDrawnElementInfo, overrideHandler: HierarchyVisibilityOverrideHandler) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;
+    getTreeSpecificVisibilityHandler: (info: AlwaysAndNeverDrawnElementInfoCache, overrideHandler: HierarchyVisibilityOverrideHandler) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;
     getSearchResultsTree: () => Promise<SearchResultsTree<TSearchTargets>> | undefined;
     componentId?: GuidString;
 }

package/lib/esm/tree-widget-react/components/trees/common/internal/useTreeHooks/UseCachedVisibility.js

@@ -4,20 +4,19 @@
  *--------------------------------------------------------------------------------------------*/
 import { useCallback, useEffect, useState } from "react";
 import { defaultIfEmpty, EMPTY, filter, firstValueFrom, from, fromEventPattern, map, mergeMap, Subject, takeUntil, tap } from "rxjs";
-import { assert } from "@itwin/core-bentley";
 import { HierarchyNode, HierarchyNodeKey } from "@itwin/presentation-hierarchies";
 import { HierarchyVisibilityOverrideHandler } from "../../UseHierarchyVisibility.js";
-import { AlwaysAndNeverDrawnElementInfo } from "../AlwaysAndNeverDrawnElementInfo.js";
+import { AlwaysAndNeverDrawnElementInfoCache } from "../caches/AlwaysAndNeverDrawnElementInfoCache.js";
 import { toVoidPromise } from "../Rxjs.js";
 import { createVisibilityStatus } from "../Tooltip.js";
 import { createVisibilityChangeEventListener } from "../VisibilityChangeEventListener.js";
 /** @internal */
 export function useCachedVisibility(props) {
     const [searchPaths, setSearchPaths] = useState(undefined);
-    const { activeView, getCache, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId } = props;
+    const { activeView, idsCache, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId } = props;
     const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState(() => createVisibilityHandlerFactory({
         activeView,
-        getCache,
+        idsCache,
         createSearchResultsTree,
         createTreeSpecificVisibilityHandler,
         searchPaths,
@@ -26,13 +25,13 @@ export function useCachedVisibility(props) {
     useEffect(() => {
         setVisibilityHandlerFactory(() => createVisibilityHandlerFactory({
             activeView,
-            getCache,
+            idsCache,
             createSearchResultsTree,
             createTreeSpecificVisibilityHandler,
             searchPaths,
             componentId,
         }));
-    }, [activeView, getCache, searchPaths, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId]);
+    }, [activeView, idsCache, searchPaths, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId]);
     return {
         visibilityHandlerFactory,
         onSearchPathsChanged: useCallback((paths) => setSearchPaths(paths), []),
@@ -40,19 +39,19 @@ export function useCachedVisibility(props) {
     };
 }
 function createVisibilityHandlerFactory(props) {
-    const { activeView, createSearchResultsTree, createTreeSpecificVisibilityHandler, getCache, searchPaths, componentId } = props;
+    const { activeView, createSearchResultsTree, createTreeSpecificVisibilityHandler, idsCache, searchPaths, componentId } = props;
     return ({ imodelAccess }) => new HierarchyVisibilityHandlerImpl({
         componentId,
         viewport: activeView,
         getSearchResultsTree: () => {
             if (searchPaths) {
-                return createSearchResultsTree({ imodelAccess, searchPaths, getCache });
+                return createSearchResultsTree({ imodelAccess, searchPaths, idsCache });
             }
             return undefined;
         },
         getTreeSpecificVisibilityHandler: (info, overrideHandler) => createTreeSpecificVisibilityHandler({
             info,
-            getCache,
+            idsCache,
             viewport: activeView,
             overrideHandler,
         }),
@@ -84,9 +83,12 @@ export class HierarchyVisibilityHandlerImpl {
                 displayStyle: true,
             },
         });
-        this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo({ viewport: this.#props.viewport, componentId: props.componentId });
-        this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(this.#alwaysAndNeverDrawnElements, new HierarchyVisibilityOverrideHandler(this));
         this.#searchResultsTree = this.#props.getSearchResultsTree();
+        this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfoCache({
+            viewport: this.#props.viewport,
+            componentId: props.componentId,
+        });
+        this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(this.#alwaysAndNeverDrawnElements, new HierarchyVisibilityOverrideHandler(this));
     }
     get onVisibilityChange() {
         return this.#eventListener.onVisibilityChange;
@@ -100,7 +102,7 @@ export class HierarchyVisibilityHandlerImpl {
             this.#eventListener.onVisibilityChange.addListener(handler);
         }, (handler) => {
             this.#eventListener.onVisibilityChange.removeListener(handler);
-        })), defaultIfEmpty(createVisibilityStatus("hidden"))));
+        })), defaultIfEmpty(createVisibilityStatus("disabled"))));
     }
     async changeVisibility(node, shouldDisplay) {
         // notify about new change request
@@ -130,7 +132,10 @@ export class HierarchyVisibilityHandlerImpl {
                 return this.getSearchResultsNodeVisibility({ node });
             }
         }
-        if (HierarchyNode.isInstancesNode(node) && node.search?.childrenTargetPaths?.length && !node.search.isSearchTarget) {
+        if (HierarchyNode.isInstancesNode(node) &&
+            node.search?.childrenTargetPaths?.length &&
+            !node.search.isSearchTarget &&
+            !node.search.hasSearchTargetAncestor) {
             return this.getSearchResultsNodeVisibility({ node });
         }
         return this.#treeSpecificVisibilityHandler.getVisibilityStatus(node);
@@ -141,7 +146,10 @@ export class HierarchyVisibilityHandlerImpl {
                 return this.changeSearchResultsNodeVisibility({ node, on });
             }
         }
-        if (HierarchyNode.isInstancesNode(node) && node.search?.childrenTargetPaths?.length && !node.search.isSearchTarget) {
+        if (HierarchyNode.isInstancesNode(node) &&
+            node.search?.childrenTargetPaths?.length &&
+            !node.search.isSearchTarget &&
+            !node.search.hasSearchTargetAncestor) {
             return this.changeSearchResultsNodeVisibility({ node, on });
         }
         return this.#treeSpecificVisibilityHandler.changeVisibilityStatus(node, on);
@@ -151,12 +159,14 @@ export class HierarchyVisibilityHandlerImpl {
             if (!targets) {
                 return EMPTY;
             }
-            return this.#treeSpecificVisibilityHandler.getSearchTargetsVisibilityStatus(targets);
+            return this.#treeSpecificVisibilityHandler.getSearchTargetsVisibilityStatus(targets, props.node);
         }));
     }
     getSearchResultsTreeTargets({ node, }) {
-        assert(this.#searchResultsTree !== undefined);
-        return from(this.#searchResultsTree).pipe(map((searchResultsTree) => searchResultsTree.getSearchTargets(node)));
+        // There can be cases where search paths used to exist and are removed, search results tree becomes undefined,
+        // but visibility is re-requested for old nodes (have search paths).
+        // In such cases return EMPTY.
+        return this.#searchResultsTree ? from(this.#searchResultsTree).pipe(map((searchResultsTree) => searchResultsTree.getSearchTargets(node))) : EMPTY;
     }
     changeSearchResultsNodeVisibility({ on, node, }) {
         return this.getSearchResultsTreeTargets({ node }).pipe(mergeMap((targets) => {

package/lib/esm/tree-widget-react/components/trees/common/internal/useTreeHooks/UseCachedVisibility.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseCachedVisibility.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/useTreeHooks/UseCachedVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACrI,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,kCAAkC,EAAE,MAAM,iCAAiC,CAAC;AACrF,OAAO,EAAE,8BAA8B,EAAE,MAAM,sCAAsC,CAAC;AACtF,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AACvD,OAAO,EAAE,mCAAmC,EAAE,MAAM,qCAAqC,CAAC;AAqC1F,gBAAgB;AAChB,MAAM,UAAU,mBAAmB,CAAyB,KAAuD;IACjH,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAoC,SAAS,CAAC,CAAC;IAC7F,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAElH,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAkD,GAAG,EAAE,CAC7H,8BAA8B,CAAC;QAC7B,UAAU;QACV,QAAQ;QACR,uBAAuB;QACvB,mCAAmC;QACnC,WAAW;QACX,WAAW;KACZ,CAAC,CACH,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,GAAG,EAAE,CAC/B,8BAA8B,CAAC;YAC7B,UAAU;YACV,QAAQ;YACR,uBAAuB;YACvB,mCAAmC;YACnC,WAAW;YACX,WAAW;SACZ,CAAC,CACH,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,WAAW,CAAC,CAAC,CAAC;IAEnH,OAAO;QACL,wBAAwB;QACxB,oBAAoB,EAAE,WAAW,CAAC,CAAC,KAAwC,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QAC1G,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,KAEC;IAED,MAAM,EAAE,UAAU,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC/H,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAC1B,IAAI,8BAA8B,CAAiB;QACjD,WAAW;QACX,QAAQ,EAAE,UAAU;QACpB,oBAAoB,EAAE,GAA2D,EAAE;YACjF,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC1E,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE,CAC1D,mCAAmC,CAAC;YAClC,IAAI;YACJ,QAAQ;YACR,QAAQ,EAAE,UAAU;YACpB,eAAe;SAChB,CAAC;KACL,CAAC,CAAC;AACP,CAAC;AAaD;;;;;;;GAOG;AACH,MAAM,OAAO,8BAA8B;IAChC,MAAM,CAAsD;IAC5D,cAAc,CAAiC;IAC/C,4BAA4B,CAAiC;IACtE,8BAA8B,CAA6D;IAC3F,cAAc,GAAG,IAAI,OAAO,EAA4C,CAAC;IACzE,kBAAkB,CAAyD;IAE3E,YAAY,KAA0D;QACpE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,mCAAmC,CAAC;YACxD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,SAAS,EAAE;gBACT,MAAM,EAAE,IAAI;gBACZ,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,4BAA4B,GAAG,IAAI,8BAA8B,CAAC,EAAE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,EAAE,WAAW,EAAE,KAAK,CAAC,WAAW,EAAE,CAAC,CAAC;QAC3I,IAAI,CAAC,8BAA8B,GAAG,IAAI,CAAC,MAAM,CAAC,gCAAgC,CAChF,IAAI,CAAC,4BAA4B,EACjC,IAAI,kCAAkC,CAAC,IAAI,CAAC,CAC7C,CAAC;QACF,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;IAC/D,CAAC;IAED,IAAW,kBAAkB;QAC3B,OAAO,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,OAAO,cAAc,CACnB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,IAAI;QACzC,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3I,oCAAoC;QACpC,SAAS,CACP,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC,EACD,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CACF,CACF,EACD,cAAc,CAAC,sBAAsB,CAAC,QAAQ,CAAC,CAAC,CACjD,CACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,aAAsB;QACvE,kCAAkC;QAClC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3E,MAAM,gBAAgB,GAAG,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC,IAAI;QACpF,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAC3I,GAAG,CAAC;YACF,SAAS,EAAE,GAAG,EAAE;gBACd,IAAI,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC;gBAC3C,IAAI,CAAC,4BAA4B,CAAC,oBAAoB,EAAE,CAAC;YAC3D,CAAC;YACD,QAAQ,EAAE,GAAG,EAAE;gBACb,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;gBACzC,IAAI,CAAC,4BAA4B,CAAC,kBAAkB,EAAE,CAAC;YACzD,CAAC;SACF,CAAC,CACH,CAAC;QAEF,OAAO,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACzC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,IAAI,CAAC,4BAA4B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,8BAA8B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACxD,CAAC;IAEO,2BAA2B,CAAC,IAAmB;QACrD,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,yBAAyB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAChG,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,IAAI,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YACnH,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC;IAEO,8BAA8B,CAAC,IAAmB,EAAE,EAAW;QACrE,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,yBAAyB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAChG,OAAO,IAAI,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,IAAI,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC,IAAI,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE,MAAM,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc,EAAE,CAAC;YACnH,OAAO,IAAI,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,sBAAsB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAEO,8BAA8B,CAAC,KAItC;QACC,OAAO,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,IAAI,CACjD,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,gCAAgC,CAAC,OAAO,CAAC,CAAC;QACvF,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,2BAA2B,CAAC,EAClC,IAAI,GAKL;QACC,MAAM,CAAC,IAAI,CAAC,kBAAkB,KAAK,SAAS,CAAC,CAAC;QAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC;IAClH,CAAC;IAEO,iCAAiC,CAAC,EACxC,EAAE,EACF,IAAI,GAML;QACC,OAAO,IAAI,CAAC,2BAA2B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CACpD,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mCAAmC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9F,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useState } from \"react\";\nimport { defaultIfEmpty, EMPTY, filter, firstValueFrom, from, fromEventPattern, map, mergeMap, Subject, takeUntil, tap } from \"rxjs\";\nimport { assert } from \"@itwin/core-bentley\";\nimport { HierarchyNode, HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { HierarchyVisibilityOverrideHandler } from \"../../UseHierarchyVisibility.js\";\nimport { AlwaysAndNeverDrawnElementInfo } from \"../AlwaysAndNeverDrawnElementInfo.js\";\nimport { toVoidPromise } from \"../Rxjs.js\";\nimport { createVisibilityStatus } from \"../Tooltip.js\";\nimport { createVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString } from \"@itwin/core-bentley\";\nimport type { ClassGroupingNodeKey, HierarchySearchPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../components/VisibilityTree.js\";\nimport type { TreeWidgetViewport } from \"../../TreeWidgetViewport.js\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../UseHierarchyVisibility.js\";\nimport type { SearchResultsTree } from \"../visibility/BaseSearchResultsTree.js\";\nimport type { TreeSpecificVisibilityHandler } from \"../visibility/BaseVisibilityHelper.js\";\nimport type { IVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\n/** @internal */\nexport interface CreateSearchResultsTreeProps<TCache> {\n  getCache: () => TCache;\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n}\n\n/** @internal */\nexport interface CreateTreeSpecificVisibilityHandlerProps<TCache> {\n  info: AlwaysAndNeverDrawnElementInfo;\n  getCache: () => TCache;\n  viewport: TreeWidgetViewport;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n}\n\n/** @internal */\nexport interface UseCachedVisibilityProps<TCache, TSearchTargets> {\n  componentId: GuidString;\n  activeView: TreeWidgetViewport;\n  getCache: () => TCache;\n  createSearchResultsTree: (props: CreateSearchResultsTreeProps<TCache>) => Promise<SearchResultsTree<TSearchTargets>>;\n  createTreeSpecificVisibilityHandler: (props: CreateTreeSpecificVisibilityHandlerProps<TCache>) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n}\n\n/** @internal */\nexport function useCachedVisibility<TCache, TSearchTargets>(props: UseCachedVisibilityProps<TCache, TSearchTargets>) {\n  const [searchPaths, setSearchPaths] = useState<HierarchySearchPath[] | undefined>(undefined);\n  const { activeView, getCache, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId } = props;\n\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState<VisibilityTreeProps[\"visibilityHandlerFactory\"]>(() =>\n    createVisibilityHandlerFactory({\n      activeView,\n      getCache,\n      createSearchResultsTree,\n      createTreeSpecificVisibilityHandler,\n      searchPaths,\n      componentId,\n    }),\n  );\n\n  useEffect(() => {\n    setVisibilityHandlerFactory(() =>\n      createVisibilityHandlerFactory({\n        activeView,\n        getCache,\n        createSearchResultsTree,\n        createTreeSpecificVisibilityHandler,\n        searchPaths,\n        componentId,\n      }),\n    );\n  }, [activeView, getCache, searchPaths, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId]);\n\n  return {\n    visibilityHandlerFactory,\n    onSearchPathsChanged: useCallback((paths: HierarchySearchPath[] | undefined) => setSearchPaths(paths), []),\n    searchPaths,\n  };\n}\n\nfunction createVisibilityHandlerFactory<TCache, TSearchTargets>(\n  props: UseCachedVisibilityProps<TCache, TSearchTargets> & {\n    searchPaths: HierarchySearchPath[] | undefined;\n  },\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  const { activeView, createSearchResultsTree, createTreeSpecificVisibilityHandler, getCache, searchPaths, componentId } = props;\n  return ({ imodelAccess }) =>\n    new HierarchyVisibilityHandlerImpl<TSearchTargets>({\n      componentId,\n      viewport: activeView,\n      getSearchResultsTree: (): Promise<SearchResultsTree<TSearchTargets>> | undefined => {\n        if (searchPaths) {\n          return createSearchResultsTree({ imodelAccess, searchPaths, getCache });\n        }\n        return undefined;\n      },\n      getTreeSpecificVisibilityHandler: (info, overrideHandler) =>\n        createTreeSpecificVisibilityHandler({\n          info,\n          getCache,\n          viewport: activeView,\n          overrideHandler,\n        }),\n    });\n}\n\n/** @internal */\nexport interface HierarchyVisibilityHandlerImplProps<TSearchTargets> {\n  viewport: TreeWidgetViewport;\n  getTreeSpecificVisibilityHandler: (\n    info: AlwaysAndNeverDrawnElementInfo,\n    overrideHandler: HierarchyVisibilityOverrideHandler,\n  ) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n  getSearchResultsTree: () => Promise<SearchResultsTree<TSearchTargets>> | undefined;\n  componentId?: GuidString;\n}\n\n/**\n * Class that handles visibility of hierarchy nodes in a tree.\n *\n * - It provides methods to get and change visibility status of nodes.\n * - Also handles filtered tree nodes visibility.\n * - Listens to visibility change events and updates visibility status accordingly.\n * @internal\n */\nexport class HierarchyVisibilityHandlerImpl<TSearchTargets> implements HierarchyVisibilityHandler, Disposable {\n  readonly #props: HierarchyVisibilityHandlerImplProps<TSearchTargets>;\n  readonly #eventListener: IVisibilityChangeEventListener;\n  readonly #alwaysAndNeverDrawnElements: AlwaysAndNeverDrawnElementInfo;\n  #treeSpecificVisibilityHandler: TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n  #changeRequest = new Subject<{ key: HierarchyNodeKey; depth: number }>();\n  #searchResultsTree: Promise<SearchResultsTree<TSearchTargets>> | undefined;\n\n  constructor(props: HierarchyVisibilityHandlerImplProps<TSearchTargets>) {\n    this.#props = props;\n    this.#eventListener = createVisibilityChangeEventListener({\n      viewport: this.#props.viewport,\n      listeners: {\n        models: true,\n        categories: true,\n        elements: true,\n        displayStyle: true,\n      },\n    });\n    this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfo({ viewport: this.#props.viewport, componentId: props.componentId });\n    this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(\n      this.#alwaysAndNeverDrawnElements,\n      new HierarchyVisibilityOverrideHandler(this),\n    );\n    this.#searchResultsTree = this.#props.getSearchResultsTree();\n  }\n\n  public get onVisibilityChange() {\n    return this.#eventListener.onVisibilityChange;\n  }\n\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    return firstValueFrom(\n      this.getVisibilityStatusInternal(node).pipe(\n        // unsubscribe from the observable if the change request for this node is received\n        takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n        // unsubscribe if visibility changes\n        takeUntil(\n          fromEventPattern(\n            (handler) => {\n              this.#eventListener.onVisibilityChange.addListener(handler);\n            },\n            (handler) => {\n              this.#eventListener.onVisibilityChange.removeListener(handler);\n            },\n          ),\n        ),\n        defaultIfEmpty(createVisibilityStatus(\"hidden\")),\n      ),\n    );\n  }\n\n  public async changeVisibility(node: HierarchyNode, shouldDisplay: boolean): Promise<void> {\n    // notify about new change request\n    this.#changeRequest.next({ key: node.key, depth: node.parentKeys.length });\n\n    const changeObservable = this.changeVisibilityStatusInternal(node, shouldDisplay).pipe(\n      // unsubscribe from the observable if the change request for this node is received\n      takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n      tap({\n        subscribe: () => {\n          this.#eventListener.suppressChangeEvents();\n          this.#alwaysAndNeverDrawnElements.suppressChangeEvents();\n        },\n        finalize: () => {\n          this.#eventListener.resumeChangeEvents();\n          this.#alwaysAndNeverDrawnElements.resumeChangeEvents();\n        },\n      }),\n    );\n\n    return toVoidPromise(changeObservable);\n  }\n\n  public [Symbol.dispose]() {\n    this.#eventListener[Symbol.dispose]();\n    this.#alwaysAndNeverDrawnElements[Symbol.dispose]();\n    this.#treeSpecificVisibilityHandler[Symbol.dispose]();\n  }\n\n  private getVisibilityStatusInternal(node: HierarchyNode) {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonSearchTargets && !node.extendedData?.hasSearchTargetAncestor) {\n        return this.getSearchResultsNodeVisibility({ node });\n      }\n    }\n\n    if (HierarchyNode.isInstancesNode(node) && node.search?.childrenTargetPaths?.length && !node.search.isSearchTarget) {\n      return this.getSearchResultsNodeVisibility({ node });\n    }\n    return this.#treeSpecificVisibilityHandler.getVisibilityStatus(node);\n  }\n\n  private changeVisibilityStatusInternal(node: HierarchyNode, on: boolean): Observable<void> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonSearchTargets && !node.extendedData?.hasSearchTargetAncestor) {\n        return this.changeSearchResultsNodeVisibility({ node, on });\n      }\n    }\n    if (HierarchyNode.isInstancesNode(node) && node.search?.childrenTargetPaths?.length && !node.search.isSearchTarget) {\n      return this.changeSearchResultsNodeVisibility({ node, on });\n    }\n    return this.#treeSpecificVisibilityHandler.changeVisibilityStatus(node, on);\n  }\n\n  private getSearchResultsNodeVisibility(props: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getSearchResultsTreeTargets(props).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.getSearchTargetsVisibilityStatus(targets);\n      }),\n    );\n  }\n\n  private getSearchResultsTreeTargets({\n    node,\n  }: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }): Observable<TSearchTargets | undefined> {\n    assert(this.#searchResultsTree !== undefined);\n    return from(this.#searchResultsTree).pipe(map((searchResultsTree) => searchResultsTree.getSearchTargets(node)));\n  }\n\n  private changeSearchResultsNodeVisibility({\n    on,\n    node,\n  }: {\n    on: boolean;\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getSearchResultsTreeTargets({ node }).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.changeSearchTargetsVisibilityStatus(targets, on);\n      }),\n    );\n  }\n}\n"]}
+{"version":3,"file":"UseCachedVisibility.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/useTreeHooks/UseCachedVisibility.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACzD,OAAO,EAAE,cAAc,EAAE,KAAK,EAAE,MAAM,EAAE,cAAc,EAAE,IAAI,EAAE,gBAAgB,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,SAAS,EAAE,GAAG,EAAE,MAAM,MAAM,CAAC;AACrI,OAAO,EAAE,aAAa,EAAE,gBAAgB,EAAE,MAAM,iCAAiC,CAAC;AAClF,OAAO,EAAE,kCAAkC,EAAE,MAAM,iCAAiC,CAAC;AACrF,OAAO,EAAE,mCAAmC,EAAE,MAAM,kDAAkD,CAAC;AACvG,OAAO,EAAE,aAAa,EAAE,MAAM,YAAY,CAAC;AAC3C,OAAO,EAAE,sBAAsB,EAAE,MAAM,eAAe,CAAC;AACvD,OAAO,EAAE,mCAAmC,EAAE,MAAM,qCAAqC,CAAC;AAqC1F,gBAAgB;AAChB,MAAM,UAAU,mBAAmB,CAAyB,KAAuD;IACjH,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAoC,SAAS,CAAC,CAAC;IAC7F,MAAM,EAAE,UAAU,EAAE,QAAQ,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAElH,MAAM,CAAC,wBAAwB,EAAE,2BAA2B,CAAC,GAAG,QAAQ,CAAkD,GAAG,EAAE,CAC7H,8BAA8B,CAAC;QAC7B,UAAU;QACV,QAAQ;QACR,uBAAuB;QACvB,mCAAmC;QACnC,WAAW;QACX,WAAW;KACZ,CAAC,CACH,CAAC;IAEF,SAAS,CAAC,GAAG,EAAE;QACb,2BAA2B,CAAC,GAAG,EAAE,CAC/B,8BAA8B,CAAC;YAC7B,UAAU;YACV,QAAQ;YACR,uBAAuB;YACvB,mCAAmC;YACnC,WAAW;YACX,WAAW;SACZ,CAAC,CACH,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,EAAE,QAAQ,EAAE,WAAW,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,WAAW,CAAC,CAAC,CAAC;IAEnH,OAAO;QACL,wBAAwB;QACxB,oBAAoB,EAAE,WAAW,CAAC,CAAC,KAAwC,EAAE,EAAE,CAAC,cAAc,CAAC,KAAK,CAAC,EAAE,EAAE,CAAC;QAC1G,WAAW;KACZ,CAAC;AACJ,CAAC;AAED,SAAS,8BAA8B,CACrC,KAEC;IAED,MAAM,EAAE,UAAU,EAAE,uBAAuB,EAAE,mCAAmC,EAAE,QAAQ,EAAE,WAAW,EAAE,WAAW,EAAE,GAAG,KAAK,CAAC;IAC/H,OAAO,CAAC,EAAE,YAAY,EAAE,EAAE,EAAE,CAC1B,IAAI,8BAA8B,CAAiB;QACjD,WAAW;QACX,QAAQ,EAAE,UAAU;QACpB,oBAAoB,EAAE,GAA2D,EAAE;YACjF,IAAI,WAAW,EAAE,CAAC;gBAChB,OAAO,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,QAAQ,EAAE,CAAC,CAAC;YAC1E,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,gCAAgC,EAAE,CAAC,IAAI,EAAE,eAAe,EAAE,EAAE,CAC1D,mCAAmC,CAAC;YAClC,IAAI;YACJ,QAAQ;YACR,QAAQ,EAAE,UAAU;YACpB,eAAe;SAChB,CAAC;KACL,CAAC,CAAC;AACP,CAAC;AAaD;;;;;;;GAOG;AACH,MAAM,OAAO,8BAA8B;IAChC,MAAM,CAAsD;IAC5D,cAAc,CAAiC;IAC/C,4BAA4B,CAAsC;IAC3E,8BAA8B,CAA6D;IAC3F,cAAc,GAAG,IAAI,OAAO,EAA4C,CAAC;IACzE,kBAAkB,CAAyD;IAE3E,YAAY,KAA0D;QACpE,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC;QACpB,IAAI,CAAC,cAAc,GAAG,mCAAmC,CAAC;YACxD,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,SAAS,EAAE;gBACT,MAAM,EAAE,IAAI;gBACZ,UAAU,EAAE,IAAI;gBAChB,QAAQ,EAAE,IAAI;gBACd,YAAY,EAAE,IAAI;aACnB;SACF,CAAC,CAAC;QACH,IAAI,CAAC,kBAAkB,GAAG,IAAI,CAAC,MAAM,CAAC,oBAAoB,EAAE,CAAC;QAC7D,IAAI,CAAC,4BAA4B,GAAG,IAAI,mCAAmC,CAAC;YAC1E,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YAC9B,WAAW,EAAE,KAAK,CAAC,WAAW;SAC/B,CAAC,CAAC;QACH,IAAI,CAAC,8BAA8B,GAAG,IAAI,CAAC,MAAM,CAAC,gCAAgC,CAChF,IAAI,CAAC,4BAA4B,EACjC,IAAI,kCAAkC,CAAC,IAAI,CAAC,CAC7C,CAAC;IACJ,CAAC;IAED,IAAW,kBAAkB;QAC3B,OAAO,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC;IAChD,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAAC,IAAmB;QAClD,OAAO,cAAc,CACnB,IAAI,CAAC,2BAA2B,CAAC,IAAI,CAAC,CAAC,IAAI;QACzC,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC;QAC3I,oCAAoC;QACpC,SAAS,CACP,gBAAgB,CACd,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,WAAW,CAAC,OAAO,CAAC,CAAC;QAC9D,CAAC,EACD,CAAC,OAAO,EAAE,EAAE;YACV,IAAI,CAAC,cAAc,CAAC,kBAAkB,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACjE,CAAC,CACF,CACF,EACD,cAAc,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC,CACnD,CACF,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,IAAmB,EAAE,aAAsB;QACvE,kCAAkC;QAClC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,GAAG,EAAE,KAAK,EAAE,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,CAAC;QAE3E,MAAM,gBAAgB,GAAG,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,aAAa,CAAC,CAAC,IAAI;QACpF,kFAAkF;QAClF,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,KAAK,KAAK,IAAI,CAAC,UAAU,CAAC,MAAM,IAAI,gBAAgB,CAAC,MAAM,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC,CAAC,EAC3I,GAAG,CAAC;YACF,SAAS,EAAE,GAAG,EAAE;gBACd,IAAI,CAAC,cAAc,CAAC,oBAAoB,EAAE,CAAC;gBAC3C,IAAI,CAAC,4BAA4B,CAAC,oBAAoB,EAAE,CAAC;YAC3D,CAAC;YACD,QAAQ,EAAE,GAAG,EAAE;gBACb,IAAI,CAAC,cAAc,CAAC,kBAAkB,EAAE,CAAC;gBACzC,IAAI,CAAC,4BAA4B,CAAC,kBAAkB,EAAE,CAAC;YACzD,CAAC;SACF,CAAC,CACH,CAAC;QAEF,OAAO,aAAa,CAAC,gBAAgB,CAAC,CAAC;IACzC,CAAC;IAEM,CAAC,MAAM,CAAC,OAAO,CAAC;QACrB,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACtC,IAAI,CAAC,4BAA4B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;QACpD,IAAI,CAAC,8BAA8B,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;IACxD,CAAC;IAEO,2BAA2B,CAAC,IAAmB;QACrD,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,yBAAyB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAChG,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;YACvD,CAAC;QACH,CAAC;QAED,IACE,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC;YACnC,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE,MAAM;YACxC,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;YAC3B,CAAC,IAAI,CAAC,MAAM,CAAC,uBAAuB,EACpC,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC;QACvD,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mBAAmB,CAAC,IAAI,CAAC,CAAC;IACvE,CAAC;IAEO,8BAA8B,CAAC,IAAmB,EAAE,EAAW;QACrE,IAAI,aAAa,CAAC,mBAAmB,CAAC,IAAI,CAAC,EAAE,CAAC;YAC5C,IAAI,IAAI,CAAC,YAAY,EAAE,yBAAyB,IAAI,CAAC,IAAI,CAAC,YAAY,EAAE,uBAAuB,EAAE,CAAC;gBAChG,OAAO,IAAI,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;YAC9D,CAAC;QACH,CAAC;QACD,IACE,aAAa,CAAC,eAAe,CAAC,IAAI,CAAC;YACnC,IAAI,CAAC,MAAM,EAAE,mBAAmB,EAAE,MAAM;YACxC,CAAC,IAAI,CAAC,MAAM,CAAC,cAAc;YAC3B,CAAC,IAAI,CAAC,MAAM,CAAC,uBAAuB,EACpC,CAAC;YACD,OAAO,IAAI,CAAC,iCAAiC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC;QAC9D,CAAC;QACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,sBAAsB,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;IAC9E,CAAC;IAEO,8BAA8B,CAAC,KAItC;QACC,OAAO,IAAI,CAAC,2BAA2B,CAAC,KAAK,CAAC,CAAC,IAAI,CACjD,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,gCAAgC,CAAC,OAAO,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QACnG,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;IAEO,2BAA2B,CAAC,EAClC,IAAI,GAKL;QACC,8GAA8G;QAC9G,oEAAoE;QACpE,8BAA8B;QAC9B,OAAO,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,iBAAiB,EAAE,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IACpJ,CAAC;IAEO,iCAAiC,CAAC,EACxC,EAAE,EACF,IAAI,GAML;QACC,OAAO,IAAI,CAAC,2BAA2B,CAAC,EAAE,IAAI,EAAE,CAAC,CAAC,IAAI,CACpD,QAAQ,CAAC,CAAC,OAAO,EAAE,EAAE;YACnB,IAAI,CAAC,OAAO,EAAE,CAAC;gBACb,OAAO,KAAK,CAAC;YACf,CAAC;YACD,OAAO,IAAI,CAAC,8BAA8B,CAAC,mCAAmC,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;QAC9F,CAAC,CAAC,CACH,CAAC;IACJ,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useState } from \"react\";\nimport { defaultIfEmpty, EMPTY, filter, firstValueFrom, from, fromEventPattern, map, mergeMap, Subject, takeUntil, tap } from \"rxjs\";\nimport { HierarchyNode, HierarchyNodeKey } from \"@itwin/presentation-hierarchies\";\nimport { HierarchyVisibilityOverrideHandler } from \"../../UseHierarchyVisibility.js\";\nimport { AlwaysAndNeverDrawnElementInfoCache } from \"../caches/AlwaysAndNeverDrawnElementInfoCache.js\";\nimport { toVoidPromise } from \"../Rxjs.js\";\nimport { createVisibilityStatus } from \"../Tooltip.js\";\nimport { createVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { GuidString } from \"@itwin/core-bentley\";\nimport type { ClassGroupingNodeKey, HierarchySearchPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../components/VisibilityTree.js\";\nimport type { TreeWidgetViewport } from \"../../TreeWidgetViewport.js\";\nimport type { HierarchyVisibilityHandler, VisibilityStatus } from \"../../UseHierarchyVisibility.js\";\nimport type { SearchResultsTree } from \"../visibility/BaseSearchResultsTree.js\";\nimport type { TreeSpecificVisibilityHandler } from \"../visibility/BaseVisibilityHelper.js\";\nimport type { IVisibilityChangeEventListener } from \"../VisibilityChangeEventListener.js\";\n\n/** @internal */\nexport interface CreateSearchResultsTreeProps<TCache> {\n  idsCache: TCache;\n  imodelAccess: ECClassHierarchyInspector;\n  searchPaths: HierarchySearchPath[];\n}\n\n/** @internal */\nexport interface CreateTreeSpecificVisibilityHandlerProps<TCache> {\n  info: AlwaysAndNeverDrawnElementInfoCache;\n  idsCache: TCache;\n  viewport: TreeWidgetViewport;\n  overrideHandler: HierarchyVisibilityOverrideHandler;\n}\n\n/** @internal */\nexport interface UseCachedVisibilityProps<TCache, TSearchTargets> {\n  componentId: GuidString;\n  activeView: TreeWidgetViewport;\n  idsCache: TCache;\n  createSearchResultsTree: (props: CreateSearchResultsTreeProps<TCache>) => Promise<SearchResultsTree<TSearchTargets>>;\n  createTreeSpecificVisibilityHandler: (props: CreateTreeSpecificVisibilityHandlerProps<TCache>) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n}\n\n/** @internal */\nexport function useCachedVisibility<TCache, TSearchTargets>(props: UseCachedVisibilityProps<TCache, TSearchTargets>) {\n  const [searchPaths, setSearchPaths] = useState<HierarchySearchPath[] | undefined>(undefined);\n  const { activeView, idsCache, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId } = props;\n\n  const [visibilityHandlerFactory, setVisibilityHandlerFactory] = useState<VisibilityTreeProps[\"visibilityHandlerFactory\"]>(() =>\n    createVisibilityHandlerFactory({\n      activeView,\n      idsCache,\n      createSearchResultsTree,\n      createTreeSpecificVisibilityHandler,\n      searchPaths,\n      componentId,\n    }),\n  );\n\n  useEffect(() => {\n    setVisibilityHandlerFactory(() =>\n      createVisibilityHandlerFactory({\n        activeView,\n        idsCache,\n        createSearchResultsTree,\n        createTreeSpecificVisibilityHandler,\n        searchPaths,\n        componentId,\n      }),\n    );\n  }, [activeView, idsCache, searchPaths, createSearchResultsTree, createTreeSpecificVisibilityHandler, componentId]);\n\n  return {\n    visibilityHandlerFactory,\n    onSearchPathsChanged: useCallback((paths: HierarchySearchPath[] | undefined) => setSearchPaths(paths), []),\n    searchPaths,\n  };\n}\n\nfunction createVisibilityHandlerFactory<TCache, TSearchTargets>(\n  props: UseCachedVisibilityProps<TCache, TSearchTargets> & {\n    searchPaths: HierarchySearchPath[] | undefined;\n  },\n): VisibilityTreeProps[\"visibilityHandlerFactory\"] {\n  const { activeView, createSearchResultsTree, createTreeSpecificVisibilityHandler, idsCache, searchPaths, componentId } = props;\n  return ({ imodelAccess }) =>\n    new HierarchyVisibilityHandlerImpl<TSearchTargets>({\n      componentId,\n      viewport: activeView,\n      getSearchResultsTree: (): Promise<SearchResultsTree<TSearchTargets>> | undefined => {\n        if (searchPaths) {\n          return createSearchResultsTree({ imodelAccess, searchPaths, idsCache });\n        }\n        return undefined;\n      },\n      getTreeSpecificVisibilityHandler: (info, overrideHandler) =>\n        createTreeSpecificVisibilityHandler({\n          info,\n          idsCache,\n          viewport: activeView,\n          overrideHandler,\n        }),\n    });\n}\n\n/** @internal */\nexport interface HierarchyVisibilityHandlerImplProps<TSearchTargets> {\n  viewport: TreeWidgetViewport;\n  getTreeSpecificVisibilityHandler: (\n    info: AlwaysAndNeverDrawnElementInfoCache,\n    overrideHandler: HierarchyVisibilityOverrideHandler,\n  ) => TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n  getSearchResultsTree: () => Promise<SearchResultsTree<TSearchTargets>> | undefined;\n  componentId?: GuidString;\n}\n\n/**\n * Class that handles visibility of hierarchy nodes in a tree.\n *\n * - It provides methods to get and change visibility status of nodes.\n * - Also handles filtered tree nodes visibility.\n * - Listens to visibility change events and updates visibility status accordingly.\n * @internal\n */\nexport class HierarchyVisibilityHandlerImpl<TSearchTargets> implements HierarchyVisibilityHandler, Disposable {\n  readonly #props: HierarchyVisibilityHandlerImplProps<TSearchTargets>;\n  readonly #eventListener: IVisibilityChangeEventListener;\n  readonly #alwaysAndNeverDrawnElements: AlwaysAndNeverDrawnElementInfoCache;\n  #treeSpecificVisibilityHandler: TreeSpecificVisibilityHandler<TSearchTargets> & Disposable;\n  #changeRequest = new Subject<{ key: HierarchyNodeKey; depth: number }>();\n  #searchResultsTree: Promise<SearchResultsTree<TSearchTargets>> | undefined;\n\n  constructor(props: HierarchyVisibilityHandlerImplProps<TSearchTargets>) {\n    this.#props = props;\n    this.#eventListener = createVisibilityChangeEventListener({\n      viewport: this.#props.viewport,\n      listeners: {\n        models: true,\n        categories: true,\n        elements: true,\n        displayStyle: true,\n      },\n    });\n    this.#searchResultsTree = this.#props.getSearchResultsTree();\n    this.#alwaysAndNeverDrawnElements = new AlwaysAndNeverDrawnElementInfoCache({\n      viewport: this.#props.viewport,\n      componentId: props.componentId,\n    });\n    this.#treeSpecificVisibilityHandler = this.#props.getTreeSpecificVisibilityHandler(\n      this.#alwaysAndNeverDrawnElements,\n      new HierarchyVisibilityOverrideHandler(this),\n    );\n  }\n\n  public get onVisibilityChange() {\n    return this.#eventListener.onVisibilityChange;\n  }\n\n  public async getVisibilityStatus(node: HierarchyNode): Promise<VisibilityStatus> {\n    return firstValueFrom(\n      this.getVisibilityStatusInternal(node).pipe(\n        // unsubscribe from the observable if the change request for this node is received\n        takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n        // unsubscribe if visibility changes\n        takeUntil(\n          fromEventPattern(\n            (handler) => {\n              this.#eventListener.onVisibilityChange.addListener(handler);\n            },\n            (handler) => {\n              this.#eventListener.onVisibilityChange.removeListener(handler);\n            },\n          ),\n        ),\n        defaultIfEmpty(createVisibilityStatus(\"disabled\")),\n      ),\n    );\n  }\n\n  public async changeVisibility(node: HierarchyNode, shouldDisplay: boolean): Promise<void> {\n    // notify about new change request\n    this.#changeRequest.next({ key: node.key, depth: node.parentKeys.length });\n\n    const changeObservable = this.changeVisibilityStatusInternal(node, shouldDisplay).pipe(\n      // unsubscribe from the observable if the change request for this node is received\n      takeUntil(this.#changeRequest.pipe(filter(({ key, depth }) => depth === node.parentKeys.length && HierarchyNodeKey.equals(node.key, key)))),\n      tap({\n        subscribe: () => {\n          this.#eventListener.suppressChangeEvents();\n          this.#alwaysAndNeverDrawnElements.suppressChangeEvents();\n        },\n        finalize: () => {\n          this.#eventListener.resumeChangeEvents();\n          this.#alwaysAndNeverDrawnElements.resumeChangeEvents();\n        },\n      }),\n    );\n\n    return toVoidPromise(changeObservable);\n  }\n\n  public [Symbol.dispose]() {\n    this.#eventListener[Symbol.dispose]();\n    this.#alwaysAndNeverDrawnElements[Symbol.dispose]();\n    this.#treeSpecificVisibilityHandler[Symbol.dispose]();\n  }\n\n  private getVisibilityStatusInternal(node: HierarchyNode) {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonSearchTargets && !node.extendedData?.hasSearchTargetAncestor) {\n        return this.getSearchResultsNodeVisibility({ node });\n      }\n    }\n\n    if (\n      HierarchyNode.isInstancesNode(node) &&\n      node.search?.childrenTargetPaths?.length &&\n      !node.search.isSearchTarget &&\n      !node.search.hasSearchTargetAncestor\n    ) {\n      return this.getSearchResultsNodeVisibility({ node });\n    }\n    return this.#treeSpecificVisibilityHandler.getVisibilityStatus(node);\n  }\n\n  private changeVisibilityStatusInternal(node: HierarchyNode, on: boolean): Observable<void> {\n    if (HierarchyNode.isClassGroupingNode(node)) {\n      if (node.extendedData?.hasDirectNonSearchTargets && !node.extendedData?.hasSearchTargetAncestor) {\n        return this.changeSearchResultsNodeVisibility({ node, on });\n      }\n    }\n    if (\n      HierarchyNode.isInstancesNode(node) &&\n      node.search?.childrenTargetPaths?.length &&\n      !node.search.isSearchTarget &&\n      !node.search.hasSearchTargetAncestor\n    ) {\n      return this.changeSearchResultsNodeVisibility({ node, on });\n    }\n    return this.#treeSpecificVisibilityHandler.changeVisibilityStatus(node, on);\n  }\n\n  private getSearchResultsNodeVisibility(props: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getSearchResultsTreeTargets(props).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.getSearchTargetsVisibilityStatus(targets, props.node);\n      }),\n    );\n  }\n\n  private getSearchResultsTreeTargets({\n    node,\n  }: {\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }): Observable<TSearchTargets | undefined> {\n    // There can be cases where search paths used to exist and are removed, search results tree becomes undefined,\n    // but visibility is re-requested for old nodes (have search paths).\n    // In such cases return EMPTY.\n    return this.#searchResultsTree ? from(this.#searchResultsTree).pipe(map((searchResultsTree) => searchResultsTree.getSearchTargets(node))) : EMPTY;\n  }\n\n  private changeSearchResultsNodeVisibility({\n    on,\n    node,\n  }: {\n    on: boolean;\n    node: HierarchyNode & {\n      key: ClassGroupingNodeKey | InstancesNodeKey;\n    };\n  }) {\n    return this.getSearchResultsTreeTargets({ node }).pipe(\n      mergeMap((targets) => {\n        if (!targets) {\n          return EMPTY;\n        }\n        return this.#treeSpecificVisibilityHandler.changeSearchTargetsVisibilityStatus(targets, on);\n      }),\n    );\n  }\n}\n"]}

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

@@ -1,20 +1,14 @@
-import type { GuidString } from "@itwin/core-bentley";
 import type { IModelConnection } from "@itwin/core-frontend";
+type CacheKey = string;
 /** @internal */
-export interface CreateCacheProps<TCacheSpecificProps> {
+export interface GetCacheProps<TCache> {
+    createCache: () => TCache;
+    cacheKey: CacheKey;
     imodel: IModelConnection;
-    specificProps: TCacheSpecificProps;
-    componentId: GuidString;
 }
 /** @internal */
-export interface UseIdsCacheProps<TCache, TCacheSpecificProps> {
-    imodel: IModelConnection;
-    createCache: (props: CreateCacheProps<TCacheSpecificProps>) => TCache;
-    cacheSpecificProps: TCacheSpecificProps;
-    componentId: GuidString;
-}
-/** @internal */
-export declare function useIdsCache<TCache extends Disposable, TCacheSpecificProps extends object>(props: UseIdsCacheProps<TCache, TCacheSpecificProps>): {
-    getCache: () => TCache;
+export declare function useIdsCache(): {
+    getCache: <TCache extends Disposable>(createCacheProps: GetCacheProps<TCache>) => TCache;
 };
+export {};
 //# sourceMappingURL=UseIdsCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/useTreeHooks/UseIdsCache.js

@@ -2,43 +2,44 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { useCallback, useEffect, useRef, useState } from "react";
-import { useIModelChangeListener } from "../UseIModelChangeListener.js";
+import { useCallback, useRef, useState } from "react";
+import { registerTxnListeners } from "@itwin/presentation-core-interop";
 /** @internal */
-export function useIdsCache(props) {
-    const cacheRef = useRef(undefined);
-    const clearCacheRef = useRef(() => {
-        cacheRef.current?.[Symbol.dispose]?.();
-        cacheRef.current = undefined;
-    });
-    const { imodel, createCache, cacheSpecificProps } = props;
-    const createCacheGetterRef = useRef((currImodel, specificProps, componentId) => {
-        return () => {
-            if (cacheRef.current === undefined) {
-                cacheRef.current = createCache({ imodel: currImodel, specificProps, componentId });
+export function useIdsCache() {
+    const state = useRef({});
+    const [forceRerender, setForceRerender] = useState({});
+    const getCache = useCallback(({ createCache, cacheKey, imodel }) => {
+        const imodelCaches = state.current[imodel.key];
+        if (imodelCaches && imodelCaches[cacheKey]) {
+            return imodelCaches[cacheKey];
+        }
+        // Create the cache before adding event listeners, to avoid race conditions
+        const cache = createCache();
+        state.current = { ...state.current, [imodel.key]: { ...state.current[imodel.key], [cacheKey]: cache } };
+        if (!imodelCaches) {
+            let listener;
+            if (imodel.isBriefcaseConnection()) {
+                listener = registerTxnListeners(imodel.txns, () => {
+                    for (const cacheToDispose of Object.values(state.current[imodel.key])) {
+                        cacheToDispose[Symbol.dispose]();
+                    }
+                    state.current[imodel.key] = {};
+                    setForceRerender({});
+                });
             }
-            return cacheRef.current;
-        };
-    });
-    const [getCache, setCacheGetter] = useState(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));
-    useEffect(() => {
-        // clear cache in case it was created before `useEffect` was run first time
-        clearCacheRef.current();
-        // make sure all cache users rerender
-        setCacheGetter(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));
-        return () => {
-            // eslint-disable-next-line react-hooks/exhaustive-deps
-            clearCacheRef.current();
-        };
-    }, [imodel, cacheSpecificProps, props.componentId]);
-    useIModelChangeListener({
-        imodel,
-        action: useCallback(() => {
-            clearCacheRef.current();
-            // make sure all cache users rerender
-            setCacheGetter(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));
-        }, [imodel, cacheSpecificProps, props.componentId]),
-    });
+            imodel.onClose.addOnce(() => {
+                for (const cacheToDispose of Object.values(state.current[imodel.key])) {
+                    cacheToDispose[Symbol.dispose]();
+                }
+                delete state.current[imodel.key];
+                listener?.();
+                setForceRerender({});
+            });
+        }
+        return cache;
+    }, 
+    // eslint-disable-next-line react-hooks/exhaustive-deps
+    [forceRerender]);
     return {
         getCache,
     };

package/lib/esm/tree-widget-react/components/trees/common/internal/useTreeHooks/UseIdsCache.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseIdsCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/useTreeHooks/UseIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACjE,OAAO,EAAE,uBAAuB,EAAE,MAAM,+BAA+B,CAAC;AAoBxE,gBAAgB;AAChB,MAAM,UAAU,WAAW,CACzB,KAAoD;IAEpD,MAAM,QAAQ,GAAG,MAAM,CAAqB,SAAS,CAAC,CAAC;IACvD,MAAM,aAAa,GAAG,MAAM,CAAC,GAAG,EAAE;QAChC,QAAQ,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,EAAE,CAAC;QACvC,QAAQ,CAAC,OAAO,GAAG,SAAS,CAAC;IAC/B,CAAC,CAAC,CAAC;IACH,MAAM,EAAE,MAAM,EAAE,WAAW,EAAE,kBAAkB,EAAE,GAAG,KAAK,CAAC;IAE1D,MAAM,oBAAoB,GAAG,MAAM,CAAC,CAAC,UAA4B,EAAE,aAAkC,EAAE,WAAuB,EAAE,EAAE;QAChI,OAAO,GAAG,EAAE;YACV,IAAI,QAAQ,CAAC,OAAO,KAAK,SAAS,EAAE,CAAC;gBACnC,QAAQ,CAAC,OAAO,GAAG,WAAW,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,WAAW,EAAE,CAAC,CAAC;YACrF,CAAC;YACD,OAAO,QAAQ,CAAC,OAAO,CAAC;QAC1B,CAAC,CAAC;IACJ,CAAC,CAAC,CAAC;IAEH,MAAM,CAAC,QAAQ,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAe,GAAG,EAAE,CAAC,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;IAE7I,SAAS,CAAC,GAAG,EAAE;QACb,2EAA2E;QAC3E,aAAa,CAAC,OAAO,EAAE,CAAC;QAExB,qCAAqC;QACrC,cAAc,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QAClG,OAAO,GAAG,EAAE;YACV,uDAAuD;YACvD,aAAa,CAAC,OAAO,EAAE,CAAC;QAC1B,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;IAEpD,uBAAuB,CAAC;QACtB,MAAM;QACN,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE;YACvB,aAAa,CAAC,OAAO,EAAE,CAAC;YACxB,qCAAqC;YACrC,cAAc,CAAC,GAAG,EAAE,CAAC,oBAAoB,CAAC,OAAO,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC,CAAC;QACpG,CAAC,EAAE,CAAC,MAAM,EAAE,kBAAkB,EAAE,KAAK,CAAC,WAAW,CAAC,CAAC;KACpD,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;KACT,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useEffect, useRef, useState } from \"react\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener.js\";\n\nimport type { GuidString } from \"@itwin/core-bentley\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\n\n/** @internal */\nexport interface CreateCacheProps<TCacheSpecificProps> {\n  imodel: IModelConnection;\n  specificProps: TCacheSpecificProps;\n  componentId: GuidString;\n}\n\n/** @internal */\nexport interface UseIdsCacheProps<TCache, TCacheSpecificProps> {\n  imodel: IModelConnection;\n  createCache: (props: CreateCacheProps<TCacheSpecificProps>) => TCache;\n  cacheSpecificProps: TCacheSpecificProps;\n  componentId: GuidString;\n}\n\n/** @internal */\nexport function useIdsCache<TCache extends Disposable, TCacheSpecificProps extends object>(\n  props: UseIdsCacheProps<TCache, TCacheSpecificProps>,\n): { getCache: () => TCache } {\n  const cacheRef = useRef<TCache | undefined>(undefined);\n  const clearCacheRef = useRef(() => {\n    cacheRef.current?.[Symbol.dispose]?.();\n    cacheRef.current = undefined;\n  });\n  const { imodel, createCache, cacheSpecificProps } = props;\n\n  const createCacheGetterRef = useRef((currImodel: IModelConnection, specificProps: TCacheSpecificProps, componentId: GuidString) => {\n    return () => {\n      if (cacheRef.current === undefined) {\n        cacheRef.current = createCache({ imodel: currImodel, specificProps, componentId });\n      }\n      return cacheRef.current;\n    };\n  });\n\n  const [getCache, setCacheGetter] = useState<() => TCache>(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));\n\n  useEffect(() => {\n    // clear cache in case it was created before `useEffect` was run first time\n    clearCacheRef.current();\n\n    // make sure all cache users rerender\n    setCacheGetter(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));\n    return () => {\n      // eslint-disable-next-line react-hooks/exhaustive-deps\n      clearCacheRef.current();\n    };\n  }, [imodel, cacheSpecificProps, props.componentId]);\n\n  useIModelChangeListener({\n    imodel,\n    action: useCallback(() => {\n      clearCacheRef.current();\n      // make sure all cache users rerender\n      setCacheGetter(() => createCacheGetterRef.current(imodel, cacheSpecificProps, props.componentId));\n    }, [imodel, cacheSpecificProps, props.componentId]),\n  });\n\n  return {\n    getCache,\n  };\n}\n"]}
+{"version":3,"file":"UseIdsCache.js","sourceRoot":"","sources":["../../../../../../../../src/tree-widget-react/components/trees/common/internal/useTreeHooks/UseIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,MAAM,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACtD,OAAO,EAAE,oBAAoB,EAAE,MAAM,kCAAkC,CAAC;AAcxE,gBAAgB;AAChB,MAAM,UAAU,WAAW;IAGzB,MAAM,KAAK,GAAG,MAAM,CAAkD,EAAE,CAAC,CAAC;IAC1E,MAAM,CAAC,aAAa,EAAE,gBAAgB,CAAC,GAAG,QAAQ,CAAC,EAAE,CAAC,CAAC;IACvD,MAAM,QAAQ,GAAG,WAAW,CAC1B,CAA4B,EAAE,WAAW,EAAE,QAAQ,EAAE,MAAM,EAAyB,EAAE,EAAE;QACtF,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;QAC/C,IAAI,YAAY,IAAI,YAAY,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3C,OAAO,YAAY,CAAC,QAAQ,CAAW,CAAC;QAC1C,CAAC;QAED,2EAA2E;QAC3E,MAAM,KAAK,GAAG,WAAW,EAAE,CAAC;QAC5B,KAAK,CAAC,OAAO,GAAG,EAAE,GAAG,KAAK,CAAC,OAAO,EAAE,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,EAAE,CAAC,QAAQ,CAAC,EAAE,KAAK,EAAE,EAAE,CAAC;QAExG,IAAI,CAAC,YAAY,EAAE,CAAC;YAClB,IAAI,QAAkC,CAAC;YACvC,IAAI,MAAM,CAAC,qBAAqB,EAAE,EAAE,CAAC;gBACnC,QAAQ,GAAG,oBAAoB,CAAC,MAAM,CAAC,IAAI,EAAE,GAAG,EAAE;oBAChD,KAAK,MAAM,cAAc,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;wBACtE,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;oBACnC,CAAC;oBACD,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC;oBAC/B,gBAAgB,CAAC,EAAE,CAAC,CAAC;gBACvB,CAAC,CAAC,CAAC;YACL,CAAC;YACD,MAAM,CAAC,OAAO,CAAC,OAAO,CAAC,GAAG,EAAE;gBAC1B,KAAK,MAAM,cAAc,IAAI,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC;oBACtE,cAAc,CAAC,MAAM,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,CAAC;gBACD,OAAO,KAAK,CAAC,OAAO,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;gBACjC,QAAQ,EAAE,EAAE,CAAC;gBACb,gBAAgB,CAAC,EAAE,CAAC,CAAC;YACvB,CAAC,CAAC,CAAC;QACL,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IACD,uDAAuD;IACvD,CAAC,aAAa,CAAC,CAChB,CAAC;IAEF,OAAO;QACL,QAAQ;KACT,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useRef, useState } from \"react\";\nimport { registerTxnListeners } from \"@itwin/presentation-core-interop\";\n\nimport type { IModelConnection } from \"@itwin/core-frontend\";\n\ntype IModelKey = string;\ntype CacheKey = string;\n\n/** @internal */\nexport interface GetCacheProps<TCache> {\n  createCache: () => TCache;\n  cacheKey: CacheKey;\n  imodel: IModelConnection;\n}\n\n/** @internal */\nexport function useIdsCache(): {\n  getCache: <TCache extends Disposable>(createCacheProps: GetCacheProps<TCache>) => TCache;\n} {\n  const state = useRef<Record<IModelKey, Record<CacheKey, Disposable>>>({});\n  const [forceRerender, setForceRerender] = useState({});\n  const getCache = useCallback(\n    <TCache extends Disposable>({ createCache, cacheKey, imodel }: GetCacheProps<TCache>) => {\n      const imodelCaches = state.current[imodel.key];\n      if (imodelCaches && imodelCaches[cacheKey]) {\n        return imodelCaches[cacheKey] as TCache;\n      }\n\n      // Create the cache before adding event listeners, to avoid race conditions\n      const cache = createCache();\n      state.current = { ...state.current, [imodel.key]: { ...state.current[imodel.key], [cacheKey]: cache } };\n\n      if (!imodelCaches) {\n        let listener: undefined | (() => void);\n        if (imodel.isBriefcaseConnection()) {\n          listener = registerTxnListeners(imodel.txns, () => {\n            for (const cacheToDispose of Object.values(state.current[imodel.key])) {\n              cacheToDispose[Symbol.dispose]();\n            }\n            state.current[imodel.key] = {};\n            setForceRerender({});\n          });\n        }\n        imodel.onClose.addOnce(() => {\n          for (const cacheToDispose of Object.values(state.current[imodel.key])) {\n            cacheToDispose[Symbol.dispose]();\n          }\n          delete state.current[imodel.key];\n          listener?.();\n          setForceRerender({});\n        });\n      }\n      return cache;\n    },\n    // eslint-disable-next-line react-hooks/exhaustive-deps\n    [forceRerender],\n  );\n\n  return {\n    getCache,\n  };\n}\n"]}

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

@@ -13,6 +13,15 @@ export type SearchResultsTreeNodeChildren<TSearchResultsTreeNode> = Map<Id64Stri
 export interface SearchResultsTreeRootNode<TSearchResultsTreeNode extends BaseSearchResultsTreeNode<TSearchResultsTreeNode>> {
     children: SearchResultsTreeNodeChildren<TSearchResultsTreeNode>;
 }
+type SearchResultsNodeType = string;
+type SearchResultsNodeId = Id64String;
+/** @internal */
+export interface SearchResultsNodeIdentifier {
+    type: SearchResultsNodeType;
+    id: SearchResultsNodeId;
+}
+/** @internal */
+export type SearchResultsNodeIdentifierAsString = `${SearchResultsNodeType}-${SearchResultsNodeId}`;
 /**
  * A generic interface for a search results tree node.
  *
@@ -24,6 +33,8 @@ export interface BaseSearchResultsTreeNode<TSearchResultsTreeNode extends BaseSe
     id: Id64String;
     children?: SearchResultsTreeNodeChildren<TSearchResultsTreeNode>;
     isSearchTarget: boolean;
+    /** Represents the path from the root to this node. It depends on the hierarchy structure. */
+    pathToNode: Array<SearchResultsNodeIdentifier>;
 }
 /**
  * Class that provides methods to handle search results nodes in a tree structure.
@@ -38,6 +49,8 @@ export declare abstract class SearchResultsNodesHandler<TProcessedSearchResultsN
     readonly searchResultsNodesArr: TSearchResultsTreeNode[];
     /** Returns search results tree node type based on its' className */
     abstract getType(className: string): Promise<TSearchResultsTreeNode["type"]>;
+    /** Returns search results tree node className based on its' type */
+    abstract getClassName(type: TSearchResultsTreeNode["type"]): string;
     /** Converts nodes to search targets */
     abstract convertNodesToSearchTargets(searchResultsNodes: TSearchResultsTreeNode[], processedSearchResultsNodes: TProcessedSearchResultsNodes): TSearchTargets | undefined;
     /**
@@ -61,6 +74,10 @@ export declare abstract class SearchResultsNodesHandler<TProcessedSearchResultsN
             key: ClassGroupingNodeKey | InstancesNodeKey;
         }) => TSearchTargets | undefined;
     }>;
+    /** Converts a search results node identifier to a string representation. */
+    convertSearchResultsNodeIdentifierToString(identifier: SearchResultsNodeIdentifier): SearchResultsNodeIdentifierAsString;
+    /** Converts a string representation of a search results node identifier back to a hierarchy node identifier. */
+    convertSearchResultsNodeIdentifierStringToHierarchyNodeIdentifier(identifier: SearchResultsNodeIdentifierAsString): InstanceKey;
     /** Takes a new node and adds it to the tree structure. */
     accept(props: {
         instanceKey: InstanceKey;
@@ -88,4 +105,5 @@ export interface CreateSearchResultsTreeProps<TProcessedSearchResultsNodes, TSea
  * @internal
  */
 export declare function createSearchResultsTree<TProcessedSearchResultsNodes, TSearchTargets, TSearchResultsTreeNode extends BaseSearchResultsTreeNode<TSearchResultsTreeNode>>(props: CreateSearchResultsTreeProps<TProcessedSearchResultsNodes, TSearchTargets, TSearchResultsTreeNode>): Promise<SearchResultsTree<TSearchTargets>>;
+export {};
 //# sourceMappingURL=BaseSearchResultsTree.d.ts.map

package/lib/esm/tree-widget-react/components/trees/common/internal/visibility/BaseSearchResultsTree.js

@@ -23,6 +23,15 @@ export class SearchResultsNodesHandler {
             getNodeSearchTargets: (node) => this.getNodeSearchTargets(node, processedSearchResultsNodes),
         };
     }
+    /** Converts a search results node identifier to a string representation. */
+    convertSearchResultsNodeIdentifierToString(identifier) {
+        return `${identifier.type}-${identifier.id}`;
+    }
+    /** Converts a string representation of a search results node identifier back to a hierarchy node identifier. */
+    convertSearchResultsNodeIdentifierStringToHierarchyNodeIdentifier(identifier) {
+        const [type, id] = identifier.split("-");
+        return { className: this.getClassName(type), id };
+    }
     /** Takes a new node and adds it to the tree structure. */
     async accept(props) {
         const { instanceKey, parentNode, isSearchTarget } = props;
@@ -81,7 +90,11 @@ export class SearchResultsNodesHandler {
  */
 export async function createSearchResultsTree(props) {
     const { searchPaths, searchResultsNodesHandler } = props;
-    for (const searchPath of searchPaths) {
+    // Sort search paths by their length, this ensures that nodes which have search target ancestors, and are search targets, won't be included in the
+    // search results tree. Search results tree is only needed to find which nodes might have children missing. Since nodes which have search target ancestors
+    // always have all children, they can be skipped.
+    const sortedSearchPaths = searchPaths.sort((lhs, rhs) => HierarchySearchPath.normalize(lhs).path.length - HierarchySearchPath.normalize(rhs).path.length);
+    for (const searchPath of sortedSearchPaths) {
         const normalizedPath = HierarchySearchPath.normalize(searchPath).path;
         let parentNode = searchResultsNodesHandler.root;
         for (let i = 0; i < normalizedPath.length; ++i) {