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

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/UseSearchPaths.js

@@ -12,7 +12,7 @@ import { useTelemetryContext } from "../../common/UseTelemetryContext.js";
 import { joinHierarchySearchPaths } from "../../common/Utils.js";
 import { ModelsTreeDefinition } from "../ModelsTreeDefinition.js";
 /** @internal */
-export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPaths, getSubTreePaths, getModelsTreeIdsCache, onModelsFiltered, onSearchPathsChanged, componentId, }) {
+export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPaths, getSubTreePaths, idsCache, onModelsFiltered, onSearchPathsChanged, componentId, }) {
     const [searchError, setSearchError] = useState(undefined);
     const [subTreeError, setSubTreeError] = useState(undefined);
     const { onFeatureUsed } = useTelemetryContext();
@@ -36,7 +36,7 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
                     createInstanceKeyPaths: async ({ targetItems }) => ModelsTreeDefinition.createInstanceKeyPaths({
                         imodelAccess,
                         targetItems,
-                        idsCache: getModelsTreeIdsCache(),
+                        idsCache,
                         hierarchyConfig: hierarchyConfiguration,
                         limit: "unbounded",
                         abortSignal,
@@ -51,14 +51,14 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
                 return [];
             }
         };
-    }, [getModelsTreeIdsCache, hierarchyConfiguration, getSubTreePaths, componentId]);
+    }, [idsCache, hierarchyConfiguration, getSubTreePaths, componentId]);
     const getPaths = useMemo(() => {
         const handlePaths = async (searchPaths, classInspector) => {
             onSearchPathsChanged(searchPaths);
             if (!onModelsFiltered) {
                 return;
             }
-            const modelIds = searchPaths ? await getModels(searchPaths, getModelsTreeIdsCache(), classInspector) : undefined;
+            const modelIds = searchPaths ? await getModels(searchPaths, idsCache, classInspector) : undefined;
             onModelsFiltered(modelIds);
         };
         if (loadFocusedItems) {
@@ -69,7 +69,7 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
                         getSearchPaths: async () => {
                             const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
                                 imodelAccess,
-                                idsCache: getModelsTreeIdsCache(),
+                                idsCache,
                                 targetItems: focusedItems,
                                 hierarchyConfig: hierarchyConfiguration,
                                 abortSignal,
@@ -101,7 +101,7 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
                                 createInstanceKeyPaths: async (props) => ModelsTreeDefinition.createInstanceKeyPaths({
                                     ...props,
                                     imodelAccess,
-                                    idsCache: getModelsTreeIdsCache(),
+                                    idsCache,
                                     hierarchyConfig: hierarchyConfiguration,
                                     limit: "unbounded",
                                     abortSignal,
@@ -135,7 +135,7 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
                             const paths = await ModelsTreeDefinition.createInstanceKeyPaths({
                                 imodelAccess,
                                 label: searchText,
-                                idsCache: getModelsTreeIdsCache(),
+                                idsCache,
                                 hierarchyConfig: hierarchyConfiguration,
                                 abortSignal,
                                 componentId,
@@ -161,7 +161,7 @@ export function useSearchPaths({ hierarchyConfiguration, searchText, getSearchPa
     }, [
         searchText,
         loadFocusedItems,
-        getModelsTreeIdsCache,
+        idsCache,
         onFeatureUsed,
         getSearchPaths,
         hierarchyConfiguration,

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/UseSearchPaths.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseSearchPaths.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/UseSearchPaths.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AACjH,OAAO,EAAE,0BAA0B,EAAE,MAAM,yCAAyC,CAAC;AACrF,OAAO,EAAE,2BAA2B,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AAChH,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAelE,gBAAgB;AAChB,MAAM,UAAU,cAAc,CAAC,EAC7B,sBAAsB,EACtB,UAAU,EACV,cAAc,EACd,eAAe,EACf,qBAAqB,EACrB,gBAAgB,EAChB,oBAAoB,EACpB,WAAW,GAkBZ;IAKC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAoC,SAAS,CAAC,CAAC;IAC7F,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAqC,SAAS,CAAC,CAAC;IAEhG,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAChD,MAAM,EAAE,gBAAgB,EAAE,GAAG,0BAA0B,EAAE,CAAC;IAE1D,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,CAAC,SAAS,CAAC,CAAC;QAC1B,eAAe,CAAC,SAAS,CAAC,CAAC;QAC3B,gBAAgB,EAAE,CAAC,SAAS,CAAC,CAAC;QAE9B,yHAAyH;QACzH,IAAI,CAAC,gBAAgB,IAAI,CAAC,cAAc,IAAI,CAAC,UAAU,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5E,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,gBAAgB,EAAE,cAAc,EAAE,eAAe,EAAE,UAAU,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAE5G,MAAM,uBAAuB,GAAG,OAAO,CAErC,GAAG,EAAE;QACL,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;YAC7C,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;oBAClC,sBAAsB,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAChD,oBAAoB,CAAC,sBAAsB,CAAC;wBAC1C,YAAY;wBACZ,WAAW;wBACX,QAAQ,EAAE,qBAAqB,EAAE;wBACjC,eAAe,EAAE,sBAAsB;wBACvC,KAAK,EAAE,WAAW;wBAClB,WAAW;wBACX,WAAW,EAAE,GAAG,WAAW,UAAU;qBACtC,CAAC;iBACL,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,qBAAqB,CAAC;gBACvC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAC1B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,qBAAqB,EAAE,sBAAsB,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC;IAElF,MAAM,QAAQ,GAAG,OAAO,CAAoD,GAAG,EAAE;QAC/E,MAAM,WAAW,GAAG,KAAK,EAAE,WAA8C,EAAE,cAAyC,EAAE,EAAE;YACtH,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,SAAS,CAAC,WAAW,EAAE,qBAAqB,EAAE,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACjH,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;oBACjE,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,QAAQ,EAAE,qBAAqB,EAAE;gCACjC,WAAW,EAAE,YAAY;gCACzB,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,2BAA2B,CAAC;oBACjH,IAAI,QAAQ,KAAK,yBAAyB,EAAE,CAAC;wBAC3C,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC;gCACjC,sBAAsB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CACtC,oBAAoB,CAAC,sBAAsB,CAAC;oCAC1C,GAAG,KAAK;oCACR,YAAY;oCACZ,QAAQ,EAAE,qBAAqB,EAAE;oCACjC,eAAe,EAAE,sBAAsB;oCACvC,KAAK,EAAE,WAAW;oCAClB,WAAW;oCACX,WAAW;iCACZ,CAAC;gCACJ,UAAU;6BACX,CAAC,CAAC;4BACH,OAAO,KAAK,EAAE,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;wBACnD,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,aAAa,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;gBAChE,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,KAAK,EAAE,UAAU;gCACjB,QAAQ,EAAE,qBAAqB,EAAE;gCACjC,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,uBAAuB,CAAC;IACjC,CAAC,EAAE;QACD,UAAU;QACV,gBAAgB;QAChB,qBAAqB;QACrB,aAAa;QACb,cAAc;QACd,sBAAsB;QACtB,gBAAgB;QAChB,oBAAoB;QACpB,uBAAuB;QACvB,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;QACR,WAAW;QACX,YAAY;KACb,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAA4B,EAAE,QAA4B,EAAE,cAAyC;IAC5H,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,IAAI,GAAG,EAAc,CAAC;IAC7C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChE,MAAM;YACR,CAAC;YAED,sEAAsE;YACtE,IAAI,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,SAAS,KAAK,kBAAkB,EAAE,CAAC;gBAC3E,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,8CAA8C;YAC9C,IAAI,MAAM,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,2BAA2B,CAAC,EAAE,CAAC;gBAC3F,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC3F,OAAO,CAAC,GAAG,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,gBAAkF;IACnH,MAAM,YAAY,GAA2C,EAAE,CAAC;IAChE,MAAM,iBAAiB,GAKlB,EAAE,CAAC;IACR,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,gBAAgB,EAAE,EAAE,CAAC;QAC3C,IAAI,IAAI,IAAI,GAAG,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,SAAS;QACX,CAAC;QAED,MAAM,YAAY,GAAG,GAAiC,CAAC;QACvD,IAAI,CAAC,YAAY,CAAC,mBAAmB,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7G,SAAS;QACX,CAAC;QAED,MAAM,SAAS,GAAG,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC;QACvD,MAAM,IAAI,GAAG,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAChG,MAAM,QAAQ,GAAG,CAAE,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAA2B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAC3H,iBAAiB,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,YAAY,CAAC,IAAI,CACf,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/E,MAAM,EACJ,UAAU,KAAK,SAAS;YACtB,CAAC,CAAC,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAChF,CAAC,CAAC,EAAE,IAAI,EAAE,UAAmB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE;QAC/F,YAAY;KACb,CAAC,CAAC,CACJ,CAAC;IACF,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,EACrC,eAAe,EACf,cAAc,EACd,WAAW,GAKZ;IACC,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IAC7F,IAAI,WAA8C,CAAC;IACnD,IAAI,CAAC;QACH,IAAI,YAAY,IAAI,WAAW,EAAE,CAAC;YAChC,OAAO,CAAC,WAAW,GAAG,wBAAwB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,WAAW,GAAG,YAAY,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;YAAS,CAAC;QACT,KAAK,WAAW,CAAC,WAAW,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useEffect, useMemo, useState } from \"react\";\nimport { firstValueFrom } from \"rxjs\";\nimport { HierarchyNodeIdentifier, HierarchyNodeKey, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport { useFocusedInstancesContext } from \"../../common/FocusedInstancesContext.js\";\nimport { CLASS_NAME_GeometricModel3d, CLASS_NAME_Subject } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { SearchLimitExceededError } from \"../../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../../common/UseTelemetryContext.js\";\nimport { joinHierarchySearchPaths } from \"../../common/Utils.js\";\nimport { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\n\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchyNodeIdentifiersPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../common/components/VisibilityTree.js\";\nimport type { NormalizedHierarchySearchPath } from \"../../common/Utils.js\";\nimport type { ClassGroupingHierarchyNode, ElementsGroupInfo, ModelsTreeHierarchyConfiguration } from \"../ModelsTreeDefinition.js\";\nimport type { ModelsTreeIdsCache } from \"./ModelsTreeIdsCache.js\";\n\n/** @internal */\nexport type ModelsTreeSearchError = \"tooManySearchMatches\" | \"tooManyInstancesFocused\" | \"unknownSearchError\" | \"unknownInstanceFocusError\";\n/** @internal */\nexport type ModelsTreeSubTreeError = \"unknownSubTreeError\";\n\n/** @internal */\nexport function useSearchPaths({\n  hierarchyConfiguration,\n  searchText,\n  getSearchPaths,\n  getSubTreePaths,\n  getModelsTreeIdsCache,\n  onModelsFiltered,\n  onSearchPathsChanged,\n  componentId,\n}: {\n  hierarchyConfiguration: ModelsTreeHierarchyConfiguration;\n  searchText?: string;\n  getSearchPaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys or node label. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> } | { label: string }) => Promise<NormalizedHierarchySearchPath[]>;\n    /** Search text which would be used to create search paths if `getSearchPaths` wouldn't be provided. */\n    searchText?: string;\n  }) => Promise<HierarchySearchPath[] | undefined>;\n  getSubTreePaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> }) => Promise<NormalizedHierarchySearchPath[]>;\n  }) => Promise<HierarchySearchPath[]>;\n  getModelsTreeIdsCache: () => ModelsTreeIdsCache;\n  onModelsFiltered?: (modelIds: Id64String[] | undefined) => void;\n  onSearchPathsChanged: (paths: HierarchySearchPath[] | undefined) => void;\n  componentId: GuidString;\n}): {\n  getPaths: VisibilityTreeProps[\"getSearchPaths\"] | undefined;\n  searchError: ModelsTreeSearchError | undefined;\n  subTreeError: ModelsTreeSubTreeError | undefined;\n} {\n  const [searchError, setSearchError] = useState<ModelsTreeSearchError | undefined>(undefined);\n  const [subTreeError, setSubTreeError] = useState<ModelsTreeSubTreeError | undefined>(undefined);\n\n  const { onFeatureUsed } = useTelemetryContext();\n  const { loadFocusedItems } = useFocusedInstancesContext();\n\n  useEffect(() => {\n    setSearchError(undefined);\n    setSubTreeError(undefined);\n    onModelsFiltered?.(undefined);\n\n    // reset search paths if there is no search applied. This allows to keep current search paths until new paths are loaded.\n    if (!loadFocusedItems && !getSearchPaths && !searchText && !getSubTreePaths) {\n      onSearchPathsChanged(undefined);\n    }\n  }, [loadFocusedItems, getSearchPaths, getSubTreePaths, searchText, onModelsFiltered, onSearchPathsChanged]);\n\n  const getSubTreePathsInternal = useMemo<\n    ((...props: Parameters<Required<VisibilityTreeProps>[\"getSearchPaths\"]>) => Promise<HierarchyNodeIdentifiersPath[]>) | undefined\n  >(() => {\n    if (!getSubTreePaths) {\n      return undefined;\n    }\n    return async ({ imodelAccess, abortSignal }) => {\n      try {\n        const paths = await getSubTreePaths({\n          createInstanceKeyPaths: async ({ targetItems }) =>\n            ModelsTreeDefinition.createInstanceKeyPaths({\n              imodelAccess,\n              targetItems,\n              idsCache: getModelsTreeIdsCache(),\n              hierarchyConfig: hierarchyConfiguration,\n              limit: \"unbounded\",\n              abortSignal,\n              componentId: `${componentId}/subTree`,\n            }),\n        });\n        return paths.map(HierarchySearchPath.normalize).map(({ path }) => path);\n      } catch {\n        const newError = \"unknownSubTreeError\";\n        setSubTreeError(newError);\n        return [];\n      }\n    };\n  }, [getModelsTreeIdsCache, hierarchyConfiguration, getSubTreePaths, componentId]);\n\n  const getPaths = useMemo<VisibilityTreeProps[\"getSearchPaths\"] | undefined>(() => {\n    const handlePaths = async (searchPaths: HierarchySearchPath[] | undefined, classInspector: ECClassHierarchyInspector) => {\n      onSearchPathsChanged(searchPaths);\n      if (!onModelsFiltered) {\n        return;\n      }\n\n      const modelIds = searchPaths ? await getModels(searchPaths, getModelsTreeIdsCache(), classInspector) : undefined;\n      onModelsFiltered(modelIds);\n    };\n\n    if (loadFocusedItems) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          const focusedItems = await collectFocusedItems(loadFocusedItems);\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                idsCache: getModelsTreeIdsCache(),\n                targetItems: focusedItems,\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManyInstancesFocused\" : \"unknownInstanceFocusError\";\n          if (newError !== \"tooManyInstancesFocused\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (getSearchPaths) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await getSearchPaths({\n                createInstanceKeyPaths: async (props) =>\n                  ModelsTreeDefinition.createInstanceKeyPaths({\n                    ...props,\n                    imodelAccess,\n                    idsCache: getModelsTreeIdsCache(),\n                    hierarchyConfig: hierarchyConfiguration,\n                    limit: \"unbounded\",\n                    abortSignal,\n                    componentId,\n                  }),\n                searchText,\n              });\n              return paths?.map(HierarchySearchPath.normalize);\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (searchText) {\n      return async ({ imodelAccess, abortSignal }) => {\n        onFeatureUsed({ featureId: \"search\", reportInteraction: true });\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                label: searchText,\n                idsCache: getModelsTreeIdsCache(),\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n    return getSubTreePathsInternal;\n  }, [\n    searchText,\n    loadFocusedItems,\n    getModelsTreeIdsCache,\n    onFeatureUsed,\n    getSearchPaths,\n    hierarchyConfiguration,\n    onModelsFiltered,\n    onSearchPathsChanged,\n    getSubTreePathsInternal,\n    componentId,\n  ]);\n\n  return {\n    getPaths,\n    searchError,\n    subTreeError,\n  };\n}\n\nasync function getModels(paths: HierarchySearchPath[], idsCache: ModelsTreeIdsCache, classInspector: ECClassHierarchyInspector) {\n  if (!paths) {\n    return undefined;\n  }\n\n  const targetModelIds = new Set<Id64String>();\n  const targetSubjectIds = new Set<Id64String>();\n  for (const path of paths) {\n    const currPath = Array.isArray(path) ? path : path.path;\n    for (let i = 0; i < currPath.length; i++) {\n      const currStep = currPath[i];\n      if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currStep)) {\n        break;\n      }\n\n      // if paths end with subject need to get all models under that subject\n      if (i === currPath.length - 1 && currStep.className === CLASS_NAME_Subject) {\n        targetSubjectIds.add(currStep.id);\n        break;\n      }\n\n      // collect all the models from the search path\n      if (await classInspector.classDerivesFrom(currStep.className, CLASS_NAME_GeometricModel3d)) {\n        targetModelIds.add(currStep.id);\n      }\n    }\n  }\n\n  const matchingModels = await firstValueFrom(idsCache.getSubjectModelIds(targetSubjectIds));\n  return [...targetModelIds, ...matchingModels];\n}\n\nasync function collectFocusedItems(loadFocusedItems: () => AsyncIterableIterator<InstanceKey | GroupingHierarchyNode>) {\n  const focusedItems: Array<InstanceKey | ElementsGroupInfo> = [];\n  const groupingNodeInfos: Array<{\n    parentKey: InstancesNodeKey;\n    parentType: \"element\" | \"category\";\n    groupingNode: ClassGroupingHierarchyNode;\n    modelIds: Id64Array;\n  }> = [];\n  for await (const key of loadFocusedItems()) {\n    if (\"id\" in key) {\n      focusedItems.push(key);\n      continue;\n    }\n\n    if (!HierarchyNodeKey.isClassGrouping(key.key)) {\n      continue;\n    }\n\n    const groupingNode = key as ClassGroupingHierarchyNode;\n    if (!groupingNode.nonGroupingAncestor || !HierarchyNodeKey.isInstances(groupingNode.nonGroupingAncestor.key)) {\n      continue;\n    }\n\n    const parentKey = groupingNode.nonGroupingAncestor.key;\n    const type = groupingNode.nonGroupingAncestor.extendedData?.isCategory ? \"category\" : \"element\";\n    const modelIds = ((groupingNode.nonGroupingAncestor.extendedData?.modelIds as Id64String[][]) ?? []).flatMap((ids) => ids);\n    groupingNodeInfos.push({ groupingNode, parentType: type, parentKey, modelIds });\n  }\n  focusedItems.push(\n    ...groupingNodeInfos.map(({ parentKey, parentType, groupingNode, modelIds }) => ({\n      parent:\n        parentType === \"element\"\n          ? { type: \"element\" as const, ids: parentKey.instanceKeys.map((key) => key.id) }\n          : { type: \"category\" as const, ids: parentKey.instanceKeys.map((key) => key.id), modelIds },\n      groupingNode,\n    })),\n  );\n  return focusedItems;\n}\n\nasync function createSearchPathsResult({\n  getSubTreePaths,\n  getSearchPaths,\n  handlePaths,\n}: {\n  getSubTreePaths: () => Promise<HierarchyNodeIdentifiersPath[] | undefined>;\n  getSearchPaths: () => Promise<NormalizedHierarchySearchPath[] | undefined>;\n  handlePaths: (searchPaths: HierarchySearchPath[] | undefined) => Promise<void>;\n}): Promise<HierarchySearchPath[] | undefined> {\n  const [subTreePaths, searchPaths] = await Promise.all([getSubTreePaths(), getSearchPaths()]);\n  let joinedPaths: HierarchySearchPath[] | undefined;\n  try {\n    if (subTreePaths && searchPaths) {\n      return (joinedPaths = joinHierarchySearchPaths(subTreePaths, searchPaths));\n    }\n    if (subTreePaths) {\n      return (joinedPaths = subTreePaths);\n    }\n    if (searchPaths) {\n      return (joinedPaths = searchPaths);\n    }\n  } finally {\n    void handlePaths(joinedPaths);\n  }\n  return joinedPaths;\n}\n"]}
+{"version":3,"file":"UseSearchPaths.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/models-tree/internal/UseSearchPaths.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,SAAS,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,MAAM,CAAC;AACtC,OAAO,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,mBAAmB,EAAE,MAAM,iCAAiC,CAAC;AACjH,OAAO,EAAE,0BAA0B,EAAE,MAAM,yCAAyC,CAAC;AACrF,OAAO,EAAE,2BAA2B,EAAE,kBAAkB,EAAE,MAAM,+CAA+C,CAAC;AAChH,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,mBAAmB,EAAE,MAAM,qCAAqC,CAAC;AAC1E,OAAO,EAAE,wBAAwB,EAAE,MAAM,uBAAuB,CAAC;AACjE,OAAO,EAAE,oBAAoB,EAAE,MAAM,4BAA4B,CAAC;AAelE,gBAAgB;AAChB,MAAM,UAAU,cAAc,CAAC,EAC7B,sBAAsB,EACtB,UAAU,EACV,cAAc,EACd,eAAe,EACf,QAAQ,EACR,gBAAgB,EAChB,oBAAoB,EACpB,WAAW,GAkBZ;IAKC,MAAM,CAAC,WAAW,EAAE,cAAc,CAAC,GAAG,QAAQ,CAAoC,SAAS,CAAC,CAAC;IAC7F,MAAM,CAAC,YAAY,EAAE,eAAe,CAAC,GAAG,QAAQ,CAAqC,SAAS,CAAC,CAAC;IAEhG,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAChD,MAAM,EAAE,gBAAgB,EAAE,GAAG,0BAA0B,EAAE,CAAC;IAE1D,SAAS,CAAC,GAAG,EAAE;QACb,cAAc,CAAC,SAAS,CAAC,CAAC;QAC1B,eAAe,CAAC,SAAS,CAAC,CAAC;QAC3B,gBAAgB,EAAE,CAAC,SAAS,CAAC,CAAC;QAE9B,yHAAyH;QACzH,IAAI,CAAC,gBAAgB,IAAI,CAAC,cAAc,IAAI,CAAC,UAAU,IAAI,CAAC,eAAe,EAAE,CAAC;YAC5E,oBAAoB,CAAC,SAAS,CAAC,CAAC;QAClC,CAAC;IACH,CAAC,EAAE,CAAC,gBAAgB,EAAE,cAAc,EAAE,eAAe,EAAE,UAAU,EAAE,gBAAgB,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAE5G,MAAM,uBAAuB,GAAG,OAAO,CAErC,GAAG,EAAE;QACL,IAAI,CAAC,eAAe,EAAE,CAAC;YACrB,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;YAC7C,IAAI,CAAC;gBACH,MAAM,KAAK,GAAG,MAAM,eAAe,CAAC;oBAClC,sBAAsB,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,EAAE,EAAE,CAChD,oBAAoB,CAAC,sBAAsB,CAAC;wBAC1C,YAAY;wBACZ,WAAW;wBACX,QAAQ;wBACR,eAAe,EAAE,sBAAsB;wBACvC,KAAK,EAAE,WAAW;wBAClB,WAAW;wBACX,WAAW,EAAE,GAAG,WAAW,UAAU;qBACtC,CAAC;iBACL,CAAC,CAAC;gBACH,OAAO,KAAK,CAAC,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;YAC1E,CAAC;YAAC,MAAM,CAAC;gBACP,MAAM,QAAQ,GAAG,qBAAqB,CAAC;gBACvC,eAAe,CAAC,QAAQ,CAAC,CAAC;gBAC1B,OAAO,EAAE,CAAC;YACZ,CAAC;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,QAAQ,EAAE,sBAAsB,EAAE,eAAe,EAAE,WAAW,CAAC,CAAC,CAAC;IAErE,MAAM,QAAQ,GAAG,OAAO,CAAoD,GAAG,EAAE;QAC/E,MAAM,WAAW,GAAG,KAAK,EAAE,WAA8C,EAAE,cAAyC,EAAE,EAAE;YACtH,oBAAoB,CAAC,WAAW,CAAC,CAAC;YAClC,IAAI,CAAC,gBAAgB,EAAE,CAAC;gBACtB,OAAO;YACT,CAAC;YAED,MAAM,QAAQ,GAAG,WAAW,CAAC,CAAC,CAAC,MAAM,SAAS,CAAC,WAAW,EAAE,QAAQ,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAClG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC,CAAC;QAEF,IAAI,gBAAgB,EAAE,CAAC;YACrB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,MAAM,YAAY,GAAG,MAAM,mBAAmB,CAAC,gBAAgB,CAAC,CAAC;oBACjE,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,QAAQ;gCACR,WAAW,EAAE,YAAY;gCACzB,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,yBAAyB,CAAC,CAAC,CAAC,2BAA2B,CAAC;oBACjH,IAAI,QAAQ,KAAK,yBAAyB,EAAE,CAAC;wBAC3C,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,cAAc,EAAE,CAAC;YACnB,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,cAAc,CAAC;gCACjC,sBAAsB,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CACtC,oBAAoB,CAAC,sBAAsB,CAAC;oCAC1C,GAAG,KAAK;oCACR,YAAY;oCACZ,QAAQ;oCACR,eAAe,EAAE,sBAAsB;oCACvC,KAAK,EAAE,WAAW;oCAClB,WAAW;oCACX,WAAW;iCACZ,CAAC;gCACJ,UAAU;6BACX,CAAC,CAAC;4BACH,OAAO,KAAK,EAAE,GAAG,CAAC,mBAAmB,CAAC,SAAS,CAAC,CAAC;wBACnD,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QAED,IAAI,UAAU,EAAE,CAAC;YACf,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,WAAW,EAAE,EAAE,EAAE;gBAC7C,aAAa,CAAC,EAAE,SAAS,EAAE,QAAQ,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;gBAChE,IAAI,CAAC;oBACH,OAAO,MAAM,uBAAuB,CAAC;wBACnC,cAAc,EAAE,KAAK,IAAI,EAAE;4BACzB,MAAM,KAAK,GAAG,MAAM,oBAAoB,CAAC,sBAAsB,CAAC;gCAC9D,YAAY;gCACZ,KAAK,EAAE,UAAU;gCACjB,QAAQ;gCACR,eAAe,EAAE,sBAAsB;gCACvC,WAAW;gCACX,WAAW;6BACZ,CAAC,CAAC;4BACH,OAAO,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,GAAG,OAAO,EAAE,MAAM,EAAE,IAAI,EAAE,EAAE,CAAC,CAAC,CAAC;wBAC7F,CAAC;wBACD,eAAe,EAAE,KAAK,IAAI,EAAE,CAAC,CAAC,uBAAuB,CAAC,CAAC,CAAC,uBAAuB,CAAC,EAAE,YAAY,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;wBAC3H,WAAW,EAAE,KAAK,EAAE,KAAK,EAAE,EAAE,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC;qBAC/D,CAAC,CAAC;gBACL,CAAC;gBAAC,OAAO,CAAC,EAAE,CAAC;oBACX,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;oBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE,CAAC;wBACxC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;wBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;oBAClE,CAAC;oBACD,cAAc,CAAC,QAAQ,CAAC,CAAC;oBACzB,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC,CAAC;QACJ,CAAC;QACD,OAAO,uBAAuB,CAAC;IACjC,CAAC,EAAE;QACD,UAAU;QACV,gBAAgB;QAChB,QAAQ;QACR,aAAa;QACb,cAAc;QACd,sBAAsB;QACtB,gBAAgB;QAChB,oBAAoB;QACpB,uBAAuB;QACvB,WAAW;KACZ,CAAC,CAAC;IAEH,OAAO;QACL,QAAQ;QACR,WAAW;QACX,YAAY;KACb,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,SAAS,CAAC,KAA4B,EAAE,QAA4B,EAAE,cAAyC;IAC5H,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,MAAM,cAAc,GAAG,IAAI,GAAG,EAAc,CAAC;IAC7C,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAAc,CAAC;IAC/C,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;QACzB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACxD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,QAAQ,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YACzC,MAAM,QAAQ,GAAG,QAAQ,CAAC,CAAC,CAAC,CAAC;YAC7B,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE,CAAC;gBAChE,MAAM;YACR,CAAC;YAED,sEAAsE;YACtE,IAAI,CAAC,KAAK,QAAQ,CAAC,MAAM,GAAG,CAAC,IAAI,QAAQ,CAAC,SAAS,KAAK,kBAAkB,EAAE,CAAC;gBAC3E,gBAAgB,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAClC,MAAM;YACR,CAAC;YAED,8CAA8C;YAC9C,IAAI,MAAM,cAAc,CAAC,gBAAgB,CAAC,QAAQ,CAAC,SAAS,EAAE,2BAA2B,CAAC,EAAE,CAAC;gBAC3F,cAAc,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,cAAc,CAAC,QAAQ,CAAC,kBAAkB,CAAC,gBAAgB,CAAC,CAAC,CAAC;IAC3F,OAAO,CAAC,GAAG,cAAc,EAAE,GAAG,cAAc,CAAC,CAAC;AAChD,CAAC;AAED,KAAK,UAAU,mBAAmB,CAAC,gBAAkF;IACnH,MAAM,YAAY,GAA2C,EAAE,CAAC;IAChE,MAAM,iBAAiB,GAKlB,EAAE,CAAC;IACR,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,gBAAgB,EAAE,EAAE,CAAC;QAC3C,IAAI,IAAI,IAAI,GAAG,EAAE,CAAC;YAChB,YAAY,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;YACvB,SAAS;QACX,CAAC;QAED,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE,CAAC;YAC/C,SAAS;QACX,CAAC;QAED,MAAM,YAAY,GAAG,GAAiC,CAAC;QACvD,IAAI,CAAC,YAAY,CAAC,mBAAmB,IAAI,CAAC,gBAAgB,CAAC,WAAW,CAAC,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC;YAC7G,SAAS;QACX,CAAC;QAED,MAAM,SAAS,GAAG,YAAY,CAAC,mBAAmB,CAAC,GAAG,CAAC;QACvD,MAAM,IAAI,GAAG,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,UAAU,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC;QAChG,MAAM,QAAQ,GAAG,CAAE,YAAY,CAAC,mBAAmB,CAAC,YAAY,EAAE,QAA2B,IAAI,EAAE,CAAC,CAAC,OAAO,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC;QAC3H,iBAAiB,CAAC,IAAI,CAAC,EAAE,YAAY,EAAE,UAAU,EAAE,IAAI,EAAE,SAAS,EAAE,QAAQ,EAAE,CAAC,CAAC;IAClF,CAAC;IACD,YAAY,CAAC,IAAI,CACf,GAAG,iBAAiB,CAAC,GAAG,CAAC,CAAC,EAAE,SAAS,EAAE,UAAU,EAAE,YAAY,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC,CAAC;QAC/E,MAAM,EACJ,UAAU,KAAK,SAAS;YACtB,CAAC,CAAC,EAAE,IAAI,EAAE,SAAkB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE;YAChF,CAAC,CAAC,EAAE,IAAI,EAAE,UAAmB,EAAE,GAAG,EAAE,SAAS,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CAAC,EAAE,QAAQ,EAAE;QAC/F,YAAY;KACb,CAAC,CAAC,CACJ,CAAC;IACF,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,KAAK,UAAU,uBAAuB,CAAC,EACrC,eAAe,EACf,cAAc,EACd,WAAW,GAKZ;IACC,MAAM,CAAC,YAAY,EAAE,WAAW,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,eAAe,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC;IAC7F,IAAI,WAA8C,CAAC;IACnD,IAAI,CAAC;QACH,IAAI,YAAY,IAAI,WAAW,EAAE,CAAC;YAChC,OAAO,CAAC,WAAW,GAAG,wBAAwB,CAAC,YAAY,EAAE,WAAW,CAAC,CAAC,CAAC;QAC7E,CAAC;QACD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,CAAC,WAAW,GAAG,YAAY,CAAC,CAAC;QACtC,CAAC;QACD,IAAI,WAAW,EAAE,CAAC;YAChB,OAAO,CAAC,WAAW,GAAG,WAAW,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;YAAS,CAAC;QACT,KAAK,WAAW,CAAC,WAAW,CAAC,CAAC;IAChC,CAAC;IACD,OAAO,WAAW,CAAC;AACrB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useEffect, useMemo, useState } from \"react\";\nimport { firstValueFrom } from \"rxjs\";\nimport { HierarchyNodeIdentifier, HierarchyNodeKey, HierarchySearchPath } from \"@itwin/presentation-hierarchies\";\nimport { useFocusedInstancesContext } from \"../../common/FocusedInstancesContext.js\";\nimport { CLASS_NAME_GeometricModel3d, CLASS_NAME_Subject } from \"../../common/internal/ClassNameDefinitions.js\";\nimport { SearchLimitExceededError } from \"../../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../../common/UseTelemetryContext.js\";\nimport { joinHierarchySearchPaths } from \"../../common/Utils.js\";\nimport { ModelsTreeDefinition } from \"../ModelsTreeDefinition.js\";\n\nimport type { GuidString, Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { GroupingHierarchyNode, HierarchyNodeIdentifiersPath, InstancesNodeKey } from \"@itwin/presentation-hierarchies\";\nimport type { ECClassHierarchyInspector, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { VisibilityTreeProps } from \"../../common/components/VisibilityTree.js\";\nimport type { NormalizedHierarchySearchPath } from \"../../common/Utils.js\";\nimport type { ClassGroupingHierarchyNode, ElementsGroupInfo, ModelsTreeHierarchyConfiguration } from \"../ModelsTreeDefinition.js\";\nimport type { ModelsTreeIdsCache } from \"./ModelsTreeIdsCache.js\";\n\n/** @internal */\nexport type ModelsTreeSearchError = \"tooManySearchMatches\" | \"tooManyInstancesFocused\" | \"unknownSearchError\" | \"unknownInstanceFocusError\";\n/** @internal */\nexport type ModelsTreeSubTreeError = \"unknownSubTreeError\";\n\n/** @internal */\nexport function useSearchPaths({\n  hierarchyConfiguration,\n  searchText,\n  getSearchPaths,\n  getSubTreePaths,\n  idsCache,\n  onModelsFiltered,\n  onSearchPathsChanged,\n  componentId,\n}: {\n  hierarchyConfiguration: ModelsTreeHierarchyConfiguration;\n  searchText?: string;\n  getSearchPaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys or node label. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> } | { label: string }) => Promise<NormalizedHierarchySearchPath[]>;\n    /** Search text which would be used to create search paths if `getSearchPaths` wouldn't be provided. */\n    searchText?: string;\n  }) => Promise<HierarchySearchPath[] | undefined>;\n  getSubTreePaths?: (props: {\n    /** A function that creates search paths based on provided target instance keys. */\n    createInstanceKeyPaths: (props: { targetItems: Array<InstanceKey | ElementsGroupInfo> }) => Promise<NormalizedHierarchySearchPath[]>;\n  }) => Promise<HierarchySearchPath[]>;\n  idsCache: ModelsTreeIdsCache;\n  onModelsFiltered?: (modelIds: Id64String[] | undefined) => void;\n  onSearchPathsChanged: (paths: HierarchySearchPath[] | undefined) => void;\n  componentId: GuidString;\n}): {\n  getPaths: VisibilityTreeProps[\"getSearchPaths\"] | undefined;\n  searchError: ModelsTreeSearchError | undefined;\n  subTreeError: ModelsTreeSubTreeError | undefined;\n} {\n  const [searchError, setSearchError] = useState<ModelsTreeSearchError | undefined>(undefined);\n  const [subTreeError, setSubTreeError] = useState<ModelsTreeSubTreeError | undefined>(undefined);\n\n  const { onFeatureUsed } = useTelemetryContext();\n  const { loadFocusedItems } = useFocusedInstancesContext();\n\n  useEffect(() => {\n    setSearchError(undefined);\n    setSubTreeError(undefined);\n    onModelsFiltered?.(undefined);\n\n    // reset search paths if there is no search applied. This allows to keep current search paths until new paths are loaded.\n    if (!loadFocusedItems && !getSearchPaths && !searchText && !getSubTreePaths) {\n      onSearchPathsChanged(undefined);\n    }\n  }, [loadFocusedItems, getSearchPaths, getSubTreePaths, searchText, onModelsFiltered, onSearchPathsChanged]);\n\n  const getSubTreePathsInternal = useMemo<\n    ((...props: Parameters<Required<VisibilityTreeProps>[\"getSearchPaths\"]>) => Promise<HierarchyNodeIdentifiersPath[]>) | undefined\n  >(() => {\n    if (!getSubTreePaths) {\n      return undefined;\n    }\n    return async ({ imodelAccess, abortSignal }) => {\n      try {\n        const paths = await getSubTreePaths({\n          createInstanceKeyPaths: async ({ targetItems }) =>\n            ModelsTreeDefinition.createInstanceKeyPaths({\n              imodelAccess,\n              targetItems,\n              idsCache,\n              hierarchyConfig: hierarchyConfiguration,\n              limit: \"unbounded\",\n              abortSignal,\n              componentId: `${componentId}/subTree`,\n            }),\n        });\n        return paths.map(HierarchySearchPath.normalize).map(({ path }) => path);\n      } catch {\n        const newError = \"unknownSubTreeError\";\n        setSubTreeError(newError);\n        return [];\n      }\n    };\n  }, [idsCache, hierarchyConfiguration, getSubTreePaths, componentId]);\n\n  const getPaths = useMemo<VisibilityTreeProps[\"getSearchPaths\"] | undefined>(() => {\n    const handlePaths = async (searchPaths: HierarchySearchPath[] | undefined, classInspector: ECClassHierarchyInspector) => {\n      onSearchPathsChanged(searchPaths);\n      if (!onModelsFiltered) {\n        return;\n      }\n\n      const modelIds = searchPaths ? await getModels(searchPaths, idsCache, classInspector) : undefined;\n      onModelsFiltered(modelIds);\n    };\n\n    if (loadFocusedItems) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          const focusedItems = await collectFocusedItems(loadFocusedItems);\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                idsCache,\n                targetItems: focusedItems,\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManyInstancesFocused\" : \"unknownInstanceFocusError\";\n          if (newError !== \"tooManyInstancesFocused\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (getSearchPaths) {\n      return async ({ imodelAccess, abortSignal }) => {\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await getSearchPaths({\n                createInstanceKeyPaths: async (props) =>\n                  ModelsTreeDefinition.createInstanceKeyPaths({\n                    ...props,\n                    imodelAccess,\n                    idsCache,\n                    hierarchyConfig: hierarchyConfiguration,\n                    limit: \"unbounded\",\n                    abortSignal,\n                    componentId,\n                  }),\n                searchText,\n              });\n              return paths?.map(HierarchySearchPath.normalize);\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n\n    if (searchText) {\n      return async ({ imodelAccess, abortSignal }) => {\n        onFeatureUsed({ featureId: \"search\", reportInteraction: true });\n        try {\n          return await createSearchPathsResult({\n            getSearchPaths: async () => {\n              const paths = await ModelsTreeDefinition.createInstanceKeyPaths({\n                imodelAccess,\n                label: searchText,\n                idsCache,\n                hierarchyConfig: hierarchyConfiguration,\n                abortSignal,\n                componentId,\n              });\n              return paths.map(({ path, options }) => ({ path, options: { ...options, reveal: true } }));\n            },\n            getSubTreePaths: async () => (getSubTreePathsInternal ? getSubTreePathsInternal({ imodelAccess, abortSignal }) : undefined),\n            handlePaths: async (paths) => handlePaths(paths, imodelAccess),\n          });\n        } catch (e) {\n          const newError = e instanceof SearchLimitExceededError ? \"tooManySearchMatches\" : \"unknownSearchError\";\n          if (newError !== \"tooManySearchMatches\") {\n            const feature = e instanceof Error && e.message.includes(\"query too long to execute or server is too busy\") ? \"error-timeout\" : \"error-unknown\";\n            onFeatureUsed({ featureId: feature, reportInteraction: false });\n          }\n          setSearchError(newError);\n          return [];\n        }\n      };\n    }\n    return getSubTreePathsInternal;\n  }, [\n    searchText,\n    loadFocusedItems,\n    idsCache,\n    onFeatureUsed,\n    getSearchPaths,\n    hierarchyConfiguration,\n    onModelsFiltered,\n    onSearchPathsChanged,\n    getSubTreePathsInternal,\n    componentId,\n  ]);\n\n  return {\n    getPaths,\n    searchError,\n    subTreeError,\n  };\n}\n\nasync function getModels(paths: HierarchySearchPath[], idsCache: ModelsTreeIdsCache, classInspector: ECClassHierarchyInspector) {\n  if (!paths) {\n    return undefined;\n  }\n\n  const targetModelIds = new Set<Id64String>();\n  const targetSubjectIds = new Set<Id64String>();\n  for (const path of paths) {\n    const currPath = Array.isArray(path) ? path : path.path;\n    for (let i = 0; i < currPath.length; i++) {\n      const currStep = currPath[i];\n      if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(currStep)) {\n        break;\n      }\n\n      // if paths end with subject need to get all models under that subject\n      if (i === currPath.length - 1 && currStep.className === CLASS_NAME_Subject) {\n        targetSubjectIds.add(currStep.id);\n        break;\n      }\n\n      // collect all the models from the search path\n      if (await classInspector.classDerivesFrom(currStep.className, CLASS_NAME_GeometricModel3d)) {\n        targetModelIds.add(currStep.id);\n      }\n    }\n  }\n\n  const matchingModels = await firstValueFrom(idsCache.getSubjectModelIds(targetSubjectIds));\n  return [...targetModelIds, ...matchingModels];\n}\n\nasync function collectFocusedItems(loadFocusedItems: () => AsyncIterableIterator<InstanceKey | GroupingHierarchyNode>) {\n  const focusedItems: Array<InstanceKey | ElementsGroupInfo> = [];\n  const groupingNodeInfos: Array<{\n    parentKey: InstancesNodeKey;\n    parentType: \"element\" | \"category\";\n    groupingNode: ClassGroupingHierarchyNode;\n    modelIds: Id64Array;\n  }> = [];\n  for await (const key of loadFocusedItems()) {\n    if (\"id\" in key) {\n      focusedItems.push(key);\n      continue;\n    }\n\n    if (!HierarchyNodeKey.isClassGrouping(key.key)) {\n      continue;\n    }\n\n    const groupingNode = key as ClassGroupingHierarchyNode;\n    if (!groupingNode.nonGroupingAncestor || !HierarchyNodeKey.isInstances(groupingNode.nonGroupingAncestor.key)) {\n      continue;\n    }\n\n    const parentKey = groupingNode.nonGroupingAncestor.key;\n    const type = groupingNode.nonGroupingAncestor.extendedData?.isCategory ? \"category\" : \"element\";\n    const modelIds = ((groupingNode.nonGroupingAncestor.extendedData?.modelIds as Id64String[][]) ?? []).flatMap((ids) => ids);\n    groupingNodeInfos.push({ groupingNode, parentType: type, parentKey, modelIds });\n  }\n  focusedItems.push(\n    ...groupingNodeInfos.map(({ parentKey, parentType, groupingNode, modelIds }) => ({\n      parent:\n        parentType === \"element\"\n          ? { type: \"element\" as const, ids: parentKey.instanceKeys.map((key) => key.id) }\n          : { type: \"category\" as const, ids: parentKey.instanceKeys.map((key) => key.id), modelIds },\n      groupingNode,\n    })),\n  );\n  return focusedItems;\n}\n\nasync function createSearchPathsResult({\n  getSubTreePaths,\n  getSearchPaths,\n  handlePaths,\n}: {\n  getSubTreePaths: () => Promise<HierarchyNodeIdentifiersPath[] | undefined>;\n  getSearchPaths: () => Promise<NormalizedHierarchySearchPath[] | undefined>;\n  handlePaths: (searchPaths: HierarchySearchPath[] | undefined) => Promise<void>;\n}): Promise<HierarchySearchPath[] | undefined> {\n  const [subTreePaths, searchPaths] = await Promise.all([getSubTreePaths(), getSearchPaths()]);\n  let joinedPaths: HierarchySearchPath[] | undefined;\n  try {\n    if (subTreePaths && searchPaths) {\n      return (joinedPaths = joinHierarchySearchPaths(subTreePaths, searchPaths));\n    }\n    if (subTreePaths) {\n      return (joinedPaths = subTreePaths);\n    }\n    if (searchPaths) {\n      return (joinedPaths = searchPaths);\n    }\n  } finally {\n    void handlePaths(joinedPaths);\n  }\n  return joinedPaths;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/visibility/ModelsTreeVisibilityHandler.d.ts

@@ -1,9 +1,9 @@
 import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
 import type { Observable } from "rxjs";
 import type { Id64Arg } from "@itwin/core-bentley";
-import type { GroupingHierarchyNode, HierarchyNode, HierarchySearchPath } from "@itwin/presentation-hierarchies";
+import type { ClassGroupingNodeKey, GroupingHierarchyNode, HierarchyNode, HierarchySearchPath, InstancesNodeKey } from "@itwin/presentation-hierarchies";
 import type { ECClassHierarchyInspector } from "@itwin/presentation-shared";
-import type { AlwaysAndNeverDrawnElementInfo } from "../../../common/internal/AlwaysAndNeverDrawnElementInfo.js";
+import type { AlwaysAndNeverDrawnElementInfoCache } from "../../../common/internal/caches/AlwaysAndNeverDrawnElementInfoCache.js";
 import type { BaseTreeVisibilityHandlerOverrides, TreeSpecificVisibilityHandler } from "../../../common/internal/visibility/BaseVisibilityHelper.js";
 import type { TreeWidgetViewport } from "../../../common/TreeWidgetViewport.js";
 import type { HierarchyVisibilityHandlerOverridableMethod, HierarchyVisibilityOverrideHandler, VisibilityStatus } from "../../../common/UseHierarchyVisibility.js";
@@ -34,7 +34,7 @@ export interface ModelsTreeVisibilityHandlerOverrides extends BaseTreeVisibility
 export interface ModelsTreeVisibilityHandlerProps {
     idsCache: ModelsTreeIdsCache;
     viewport: TreeWidgetViewport;
-    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfo;
+    alwaysAndNeverDrawnElementInfo: AlwaysAndNeverDrawnElementInfoCache;
     overrideHandler: HierarchyVisibilityOverrideHandler;
     overrides?: ModelsTreeVisibilityHandlerOverrides;
 }
@@ -52,13 +52,9 @@ export declare class ModelsTreeVisibilityHandler implements Disposable, TreeSpec
     getVisibilityStatus(node: HierarchyNode): Observable<VisibilityStatus>;
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node: HierarchyNode, on: boolean): Observable<void>;
-    getSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets): Observable<VisibilityStatus>;
-    private getCategories;
-    private getAllCategories;
-    private getElementsCount;
-    private getModels;
-    private getSubCategories;
-    private getSubModels;
+    getSearchTargetsVisibilityStatus(targets: ModelsTreeSearchTargets, node: HierarchyNode & {
+        key: ClassGroupingNodeKey | InstancesNodeKey;
+    }): Observable<VisibilityStatus>;
 }
 /**
  * Creates models tree visibility handler. Is used by integration and performance tests.

package/lib/esm/tree-widget-react/components/trees/models-tree/internal/visibility/ModelsTreeVisibilityHandler.js

@@ -2,12 +2,14 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
-import { concat, defer, from, map, merge, mergeMap, of } from "rxjs";
-import { assert, Guid, Id64 } from "@itwin/core-bentley";
+import { concat, defer, EMPTY, from, map, merge, mergeAll, mergeMap, of } from "rxjs";
+import { assert, Guid } from "@itwin/core-bentley";
+import { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+import { createVisibilityStatus } from "../../../common/internal/Tooltip.js";
 import { HierarchyVisibilityHandlerImpl } from "../../../common/internal/useTreeHooks/UseCachedVisibility.js";
-import { fromWithRelease } from "../../../common/internal/Utils.js";
+import { fromWithRelease, getIdsFromChildrenTree, getParentElementsIdsPath, setDifference, setIntersection } from "../../../common/internal/Utils.js";
 import { mergeVisibilityStatuses } from "../../../common/internal/VisibilityUtils.js";
-import { ModelsTreeNode } from "../../ModelsTreeNode.js";
+import { ModelsTreeNodeInternal } from "../ModelsTreeNodeInternal.js";
 import { ModelsTreeVisibilityHelper } from "./ModelsTreeVisibilityHelper.js";
 import { createModelsSearchResultsTree } from "./SearchResultsTree.js";
 /**
@@ -21,23 +23,12 @@ export class ModelsTreeVisibilityHandler {
     #props;
     constructor(constructorProps) {
         this.#props = constructorProps;
-        // Remove after https://github.com/iTwin/viewer-components-react/issues/1421.
-        // We won't need to create a custom base ids cache.
-        const baseIdsCache = {
-            getCategories: (props) => this.getCategories(props),
-            getElementsCount: (props) => this.getElementsCount(props),
-            getModels: (props) => this.getModels(props),
-            getSubCategories: (props) => this.getSubCategories(props),
-            getSubModels: (props) => this.getSubModels(props),
-            hasSubModel: (props) => this.#props.idsCache.hasSubModel(props),
-            getAllCategories: () => this.getAllCategories(),
-        };
         this.#visibilityHelper = new ModelsTreeVisibilityHelper({
             viewport: this.#props.viewport,
             idsCache: this.#props.idsCache,
             alwaysAndNeverDrawnElementInfo: this.#props.alwaysAndNeverDrawnElementInfo,
             overrideHandler: this.#props.overrideHandler,
-            baseIdsCache,
+            baseIdsCache: this.#props.idsCache,
             overrides: this.#props.overrides,
         });
     }
@@ -48,6 +39,9 @@ export class ModelsTreeVisibilityHandler {
         return defer(() => {
             const { subjectIds, modelIds, categories, elements } = targets;
             const observables = new Array();
+            if (this.#props.viewport.viewType !== "3d") {
+                return EMPTY;
+            }
             if (subjectIds?.size) {
                 observables.push(this.#visibilityHelper.changeSubjectsVisibilityStatus({ subjectIds, on }));
             }
@@ -62,17 +56,76 @@ export class ModelsTreeVisibilityHandler {
                 }))));
             }
             if (elements?.length) {
-                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.changeElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], on }))));
+                const searchTargetElements = new Array();
+                const elementIdsSet = new Set();
+                const modelCategoryElementMap = new Map();
+                // elements is an array that stores elements grouped by:
+                // 1. Their path
+                // 2. Their modelId and categoryId
+                // When changing visibility of elements, visibility handler does not care about the path.
+                // So we can first get all elements and group them only by modelId and categoryId.
+                for (const { elements: elementsMap, categoryId, modelId } of elements) {
+                    const key = `${modelId}-${categoryId}`;
+                    let mapEntry = modelCategoryElementMap.get(key);
+                    if (!mapEntry) {
+                        mapEntry = [];
+                        modelCategoryElementMap.set(key, mapEntry);
+                    }
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        mapEntry.push(elementId);
+                        elementIdsSet.add(elementId);
+                        if (isSearchTarget) {
+                            searchTargetElements.push(elementId);
+                        }
+                    }
+                }
+                observables.push(
+                // Get children for search targets, since non search targets don't have all the children present in the hierarchy.
+                this.#props.idsCache.getChildElementsTree({ elementIds: searchTargetElements }).pipe(
+                // Need to filter out and keep only those children ids that are not part of elements that are present in search paths.
+                // Elements in search paths will have their visibility changed directly: they will be provided as elementIds to changeElementsVisibilityStatus.
+                map((childrenTree) => ({
+                    childrenNotInSearchPaths: setDifference(getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 }), elementIdsSet),
+                    childrenTree,
+                })), mergeMap(({ childrenNotInSearchPaths, childrenTree }) => fromWithRelease({ source: modelCategoryElementMap.entries(), size: modelCategoryElementMap.size, releaseOnCount: 50 }).pipe(mergeMap(([key, elementsInSearchPathsGroupedByModelAndCategory]) => {
+                    const [modelId, categoryId] = key.split("-");
+                    const childrenIds = new Set();
+                    // A shared children tree was created, need to get the children for each element in the group.
+                    for (const elementId of elementsInSearchPathsGroupedByModelAndCategory) {
+                        const elementChildrenTree = childrenTree.get(elementId)?.children;
+                        if (!elementChildrenTree) {
+                            continue;
+                        }
+                        for (const childId of getIdsFromChildrenTree({ tree: elementChildrenTree })) {
+                            childrenIds.add(childId);
+                        }
+                    }
+                    return this.#visibilityHelper.changeElementsVisibilityStatus({
+                        modelId,
+                        categoryId,
+                        elementIds: elementsInSearchPathsGroupedByModelAndCategory,
+                        // Pass only those children that are not part of search paths.
+                        children: setIntersection(childrenIds, childrenNotInSearchPaths),
+                        on,
+                    });
+                })))));
             }
             return merge(...observables);
         });
     }
     getVisibilityStatus(node) {
-        if (ModelsTreeNode.isElementClassGroupingNode(node)) {
+        if (this.#props.viewport.viewType !== "3d") {
+            return of(createVisibilityStatus("disabled"));
+        }
+        if (ModelsTreeNodeInternal.isElementClassGroupingNode(node)) {
             const result = this.#visibilityHelper.getGroupedElementsVisibilityStatus({
                 categoryId: node.extendedData.categoryId,
                 modelId: node.extendedData.modelId,
                 elementIds: node.groupedInstanceKeys.map((key) => key.id),
+                parentKeys: node.parentKeys,
+                childrenCount: node.extendedData.childrenCount,
+                categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,
+                topMostParentElementId: node.extendedData.topMostParentElementId,
             });
             return this.#props.overrideHandler.createVisibilityHandlerResult({
                 overrideProps: { node },
@@ -80,32 +133,40 @@ export class ModelsTreeVisibilityHandler {
                 override: this.#props.overrides?.getElementGroupingNodeVisibilityStatus,
             });
         }
-        if (ModelsTreeNode.isSubjectNode(node)) {
+        if (ModelsTreeNodeInternal.isSubjectNode(node)) {
             // note: subject nodes may be merged to represent multiple subject instances
             return this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds: node.key.instanceKeys.map((key) => key.id) });
         }
-        if (ModelsTreeNode.isModelNode(node)) {
-            return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), type: "GeometricModel3d" });
+        if (ModelsTreeNodeInternal.isModelNode(node)) {
+            return this.#visibilityHelper.getModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id) });
         }
-        if (ModelsTreeNode.isCategoryNode(node)) {
+        if (ModelsTreeNodeInternal.isCategoryNode(node)) {
             return this.#visibilityHelper.getCategoriesVisibilityStatus({
                 categoryIds: node.key.instanceKeys.map(({ id }) => id),
                 modelId: node.extendedData.modelIds[0],
-                type: "SpatialCategory",
             });
         }
-        assert(ModelsTreeNode.isElementNode(node));
+        assert(ModelsTreeNodeInternal.isElementNode(node));
+        const parentElementsIdsPath = getParentElementsIdsPath({
+            parentInstanceKeys: node.parentKeys.filter((key) => HierarchyNodeKey.isInstances(key)).map((key) => key.instanceKeys),
+            topMostParentElementId: node.extendedData.topMostParentElementId,
+        });
         return this.#visibilityHelper.getElementsVisibilityStatus({
             elementIds: node.key.instanceKeys.map(({ id }) => id),
             modelId: node.extendedData.modelId,
             categoryId: node.extendedData.categoryId,
-            type: "GeometricElement3d",
+            parentElementsIdsPath,
+            childrenCount: node.extendedData?.childrenCount,
+            categoryOfTopMostParentElement: node.extendedData.categoryOfTopMostParentElement,
         });
     }
     /** Changes visibility of the items represented by the tree node. */
     changeVisibilityStatus(node, on) {
         const changeObs = defer(() => {
-            if (ModelsTreeNode.isElementClassGroupingNode(node)) {
+            if (this.#props.viewport.viewType !== "3d") {
+                return EMPTY;
+            }
+            if (ModelsTreeNodeInternal.isElementClassGroupingNode(node)) {
                 const result = this.#visibilityHelper.changeGroupedElementsVisibilityStatus({
                     categoryId: node.extendedData.categoryId,
                     modelId: node.extendedData.modelId,
@@ -118,36 +179,45 @@ export class ModelsTreeVisibilityHandler {
                     override: this.#props.overrides?.changeElementGroupingNodeVisibilityStatus,
                 });
             }
-            if (ModelsTreeNode.isSubjectNode(node)) {
+            if (ModelsTreeNodeInternal.isSubjectNode(node)) {
                 return this.#visibilityHelper.changeSubjectsVisibilityStatus({
                     subjectIds: node.key.instanceKeys.map((key) => key.id),
                     on,
                 });
             }
-            if (ModelsTreeNode.isModelNode(node)) {
+            if (ModelsTreeNodeInternal.isModelNode(node)) {
                 return this.#visibilityHelper.changeModelsVisibilityStatus({ modelIds: node.key.instanceKeys.map(({ id }) => id), on });
             }
-            if (ModelsTreeNode.isCategoryNode(node)) {
+            if (ModelsTreeNodeInternal.isCategoryNode(node)) {
                 return this.#visibilityHelper.changeCategoriesVisibilityStatus({
                     categoryIds: node.key.instanceKeys.map(({ id }) => id),
                     modelId: node.extendedData.modelIds[0],
                     on,
                 });
             }
-            assert(ModelsTreeNode.isElementNode(node));
-            return this.#visibilityHelper.changeElementsVisibilityStatus({
-                elementIds: node.key.instanceKeys.map(({ id }) => id),
+            assert(ModelsTreeNodeInternal.isElementNode(node));
+            const elementIds = node.key.instanceKeys.map(({ id }) => id);
+            return this.#props.idsCache.getChildElementsTree({ elementIds }).pipe(map((childrenTree) => {
+                // Children tree contains provided elementIds, they are at the root of this tree.
+                // We want to skip them and only get ids of children.
+                return getIdsFromChildrenTree({ tree: childrenTree, predicate: ({ depth }) => depth > 0 });
+            }), mergeMap((children) => this.#visibilityHelper.changeElementsVisibilityStatus({
+                elementIds,
                 modelId: node.extendedData.modelId,
+                children: children.size > 0 ? children : undefined,
                 categoryId: node.extendedData.categoryId,
                 on,
-            });
+            })));
         });
         if (this.#props.viewport.isAlwaysDrawnExclusive) {
             return concat(this.#visibilityHelper.removeAlwaysDrawnExclusive(), changeObs);
         }
         return changeObs;
     }
-    getSearchTargetsVisibilityStatus(targets) {
+    getSearchTargetsVisibilityStatus(targets, node) {
+        if (this.#props.viewport.viewType !== "3d") {
+            return of(createVisibilityStatus("disabled"));
+        }
         return defer(() => {
             const { subjectIds, modelIds, categories, elements } = targets;
             const observables = new Array();
@@ -155,54 +225,91 @@ export class ModelsTreeVisibilityHandler {
                 observables.push(this.#visibilityHelper.getSubjectsVisibilityStatus({ subjectIds }));
             }
             if (modelIds?.size) {
-                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds, type: "GeometricModel3d" }));
+                observables.push(this.#visibilityHelper.getModelsVisibilityStatus({ modelIds }));
             }
             if (categories?.length) {
                 observables.push(from(categories).pipe(mergeMap(({ modelId, categoryIds }) => this.#visibilityHelper.getCategoriesVisibilityStatus({
                     categoryIds,
                     modelId,
-                    type: "SpatialCategory",
                 }))));
             }
             if (elements?.length) {
-                observables.push(fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId }) => this.#visibilityHelper.getElementsVisibilityStatus({ modelId, categoryId, elementIds: [...elementsMap.keys()], type: "GeometricElement3d" }))));
-            }
-            return merge(...observables);
-        }).pipe(mergeVisibilityStatuses);
-    }
-    getCategories(props) {
-        return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => this.#props.idsCache.getModelCategoryIds(modelId).pipe(map((categoryIds) => ({ id: modelId, spatialCategories: categoryIds })))));
-    }
-    getAllCategories() {
-        return this.#props.idsCache.getAllCategories().pipe(map((categories) => ({ spatialCategories: categories })));
-    }
-    getElementsCount(props) {
-        return this.#props.idsCache.getCategoryElementsCount(props.modelId, props.categoryId);
-    }
-    getModels(props) {
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds);
-    }
-    getSubCategories(props) {
-        return this.#props.idsCache.getSubCategories(props.categoryId);
-    }
-    getSubModels(props) {
-        if ("modelIds" in props) {
-            return from(Id64.iterable(props.modelIds)).pipe(mergeMap((modelId) => {
-                if (props.categoryId) {
-                    return this.#props.idsCache.getCategoriesModeledElements(modelId, props.categoryId).pipe(map((subModels) => ({ id: modelId, subModels })));
+                const searchTargetElements = new Array();
+                for (const { elements: elementsMap } of elements) {
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        if (isSearchTarget) {
+                            searchTargetElements.push(elementId);
+                        }
+                    }
                 }
-                return this.#props.idsCache.getModelCategoryIds(modelId).pipe(mergeMap((categoryIds) => this.#props.idsCache.getCategoriesModeledElements(modelId, categoryIds)), map((subModels) => ({ id: modelId, subModels })));
-            }));
-        }
-        if (props.modelId) {
-            return from(Id64.iterable(props.categoryIds)).pipe(mergeMap((categoryId) => this.#props.idsCache.getCategoriesModeledElements(props.modelId, categoryId).pipe(map((subModels) => ({ id: categoryId, subModels })))));
-        }
-        return this.#props.idsCache.getCategoriesElementModels(props.categoryIds).pipe(mergeMap(({ id, models }) => {
-            if (!models) {
-                return of({ id, subModels: undefined });
+                let childrenCountMapObs;
+                if (ModelsTreeNodeInternal.isElementClassGroupingNode(node)) {
+                    const groupingNodesSearchTargets = node.extendedData.searchTargets;
+                    const nestedSearchTargetElements = searchTargetElements.filter((searchTarget) => !groupingNodesSearchTargets?.has(searchTarget));
+                    // Only need to request children count for indirect children search targets.
+                    childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: nestedSearchTargetElements }).pipe(map((elementCountMap) => {
+                        // Direct children search targets already have children count stored in grouping nodes extended data.
+                        for (const [key, value] of node.extendedData.searchTargets ?? []) {
+                            elementCountMap.set(key, value.childrenCount);
+                        }
+                        return elementCountMap;
+                    }));
+                }
+                else {
+                    childrenCountMapObs = this.#props.idsCache.getAllChildElementsCount({ elementIds: searchTargetElements });
+                }
+                observables.push(childrenCountMapObs.pipe(mergeMap((elementsChildrenCountMap) => fromWithRelease({ source: elements, releaseOnCount: 50 }).pipe(mergeMap(({ modelId, elements: elementsMap, categoryId, pathToElements, topMostParentElementId }) => {
+                    const parentElementsIdsPath = topMostParentElementId
+                        ? getParentElementsIdsPath({
+                            parentInstanceKeys: pathToElements.map((instanceKey) => [instanceKey]),
+                            topMostParentElementId,
+                        })
+                        : [];
+                    let totalSearchTargetsChildrenCount = 0;
+                    const nonSearchTargetIds = new Array();
+                    const searchTargetIds = new Array();
+                    for (const [elementId, { isSearchTarget }] of elementsMap) {
+                        if (!isSearchTarget) {
+                            nonSearchTargetIds.push(elementId);
+                            continue;
+                        }
+                        searchTargetIds.push(elementId);
+                        const childCount = elementsChildrenCountMap.get(elementId);
+                        if (childCount) {
+                            totalSearchTargetsChildrenCount += childCount;
+                        }
+                    }
+                    return merge(searchTargetIds.length > 0
+                        ? this.#visibilityHelper.getElementsVisibilityStatus({
+                            modelId,
+                            categoryId,
+                            elementIds: searchTargetIds,
+                            parentElementsIdsPath,
+                            childrenCount: totalSearchTargetsChildrenCount,
+                            // Search results tree is created on search paths. Since search paths contain only categories that are directly under models
+                            // or at the root, categoryId can be used here here.
+                            categoryOfTopMostParentElement: categoryId,
+                        })
+                        : EMPTY, 
+                    // Set childrenCount to undefined for non search targets, as some of their child elements might be filtered out.
+                    // Since childrenCount is set to undefined, these elements won't check child always/never drawn child elements status.
+                    // Child always/never drawn elements will be in search paths, and their visibility status will be handled separately.
+                    nonSearchTargetIds.length > 0
+                        ? this.#visibilityHelper.getElementsVisibilityStatus({
+                            modelId,
+                            categoryId,
+                            elementIds: nonSearchTargetIds,
+                            parentElementsIdsPath,
+                            childrenCount: undefined,
+                            // Search results tree is created on search paths. Since search paths contain only categories that are directly under models
+                            // or at the root, categoryId can be used here here.
+                            categoryOfTopMostParentElement: categoryId,
+                        })
+                        : EMPTY).pipe(mergeVisibilityStatuses());
+                })))));
             }
-            return from(models).pipe(mergeMap((modelId) => this.#props.idsCache.getCategoriesModeledElements(modelId, id)), map((subModels) => ({ id, subModels })));
-        }));
+            return from(observables).pipe(mergeAll(), mergeVisibilityStatuses());
+        });
     }
 }
 /**