@itwin/tree-widget-react 4.0.0-alpha.3 → 4.0.0-alpha.4

package/lib/esm/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.js

@@ -4,23 +4,36 @@ import { jsx as _jsx } from "react/jsx-runtime";
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
 import { useCallback, useMemo, useState } from "react";
+import { assert } from "@itwin/core-bentley";
+import categorySvg from "@itwin/itwinui-icons/bis-category-3d.svg";
+import subcategorySvg from "@itwin/itwinui-icons/bis-category-subcategory.svg";
+import definitionContainerSvg from "@itwin/itwinui-icons/bis-definitions-container.svg";
 import { Text } from "@itwin/itwinui-react/bricks";
-import { HierarchyNodeIdentifier } from "@itwin/presentation-hierarchies";
+import { createECSqlQueryExecutor } from "@itwin/presentation-core-interop";
+import { HierarchyFilteringPath, HierarchyNodeIdentifier } from "@itwin/presentation-hierarchies";
 import { TreeWidget } from "../../../TreeWidget.js";
 import { EmptyTreeContent } from "../common/components/EmptyTreeContent.js";
 import { FilterLimitExceededError } from "../common/TreeErrors.js";
 import { useTelemetryContext } from "../common/UseTelemetryContext.js";
 import { CategoriesTreeDefinition } from "./CategoriesTreeDefinition.js";
-import { CategoriesVisibilityHandler } from "./CategoriesVisibilityHandler.js";
+import { CategoriesTreeIdsCache } from "./internal/CategoriesTreeIdsCache.js";
+import { CategoriesVisibilityHandler } from "./internal/CategoriesVisibilityHandler.js";
+import { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from "./internal/ClassNameDefinitions.js";
 /**
  * Custom hook to create and manage state for the categories tree.
  * @beta
  */
 export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, emptyTreeContent }) {
     const [filteringError, setFilteringError] = useState();
+    const viewType = activeView.view.is2d() ? "2d" : "3d";
+    const iModel = activeView.iModel;
+    const idsCache = useMemo(() => {
+        return new CategoriesTreeIdsCache(createECSqlQueryExecutor(iModel), viewType);
+    }, [viewType, iModel]);
     const visibilityHandlerFactory = useCallback(() => {
         const visibilityHandler = new CategoriesVisibilityHandler({
             viewport: activeView,
+            idsCache,
         });
         return {
             getVisibilityStatus: async (node) => visibilityHandler.getVisibilityStatus(node),
@@ -28,11 +41,11 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
             onVisibilityChange: visibilityHandler.onVisibilityChange,
             dispose: () => visibilityHandler.dispose(),
         };
-    }, [activeView]);
+    }, [activeView, idsCache]);
     const { onFeatureUsed } = useTelemetryContext();
     const getHierarchyDefinition = useCallback((props) => {
-        return new CategoriesTreeDefinition({ ...props, viewType: activeView.view.is2d() ? "2d" : "3d" });
-    }, [activeView]);
+        return new CategoriesTreeDefinition({ ...props, viewType, idsCache });
+    }, [viewType, idsCache]);
     const getFilteredPaths = useMemo(() => {
         setFilteringError(undefined);
         onCategoriesFiltered?.(undefined);
@@ -42,8 +55,8 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
         return async ({ imodelAccess }) => {
             onFeatureUsed({ featureId: "filtering", reportInteraction: true });
             try {
-                const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({ imodelAccess, label: filter, viewType: activeView.view.is2d() ? "2d" : "3d" });
-                onCategoriesFiltered?.(getCategories(paths));
+                const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({ imodelAccess, label: filter, viewType, idsCache });
+                onCategoriesFiltered?.(await getCategoriesFromPaths(paths, idsCache));
                 return paths;
             }
             catch (e) {
@@ -56,7 +69,7 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
                 return [];
             }
         };
-    }, [filter, activeView, onFeatureUsed, onCategoriesFiltered]);
+    }, [filter, viewType, onFeatureUsed, onCategoriesFiltered, idsCache]);
     return {
         categoriesTreeProps: {
             treeName: "categories-tree-v2",
@@ -72,22 +85,48 @@ export function useCategoriesTree({ filter, activeView, onCategoriesFiltered, em
         },
     };
 }
-function getCategories(paths) {
+async function getCategoriesFromPaths(paths, idsCache) {
     if (!paths) {
         return undefined;
     }
     const categories = new Map();
     for (const path of paths) {
-        const currPath = Array.isArray(path) ? path : path.path;
-        const [category, subCategory] = currPath;
-        if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(category)) {
+        const currPath = HierarchyFilteringPath.normalize(path).path;
+        if (currPath.length === 0) {
             continue;
         }
-        if (!categories.has(category.id)) {
-            categories.set(category.id, []);
+        let category;
+        let subCategory;
+        const lastNode = currPath[currPath.length - 1];
+        if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(lastNode)) {
+            continue;
+        }
+        if (lastNode.className === DEFINITION_CONTAINER_CLASS) {
+            const definitionContainerCategories = await idsCache.getAllContainedCategories([lastNode.id]);
+            for (const categoryId of definitionContainerCategories) {
+                const value = categories.get(categoryId);
+                if (value === undefined) {
+                    categories.set(categoryId, []);
+                }
+            }
+            continue;
+        }
+        if (lastNode.className === SUB_CATEGORY_CLASS) {
+            const secondToLastNode = currPath.length > 1 ? currPath[currPath.length - 2] : undefined;
+            assert(secondToLastNode !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(secondToLastNode));
+            subCategory = lastNode;
+            category = secondToLastNode;
         }
-        if (subCategory && HierarchyNodeIdentifier.isInstanceNodeIdentifier(subCategory)) {
-            categories.get(category.id).push(subCategory.id);
+        else {
+            category = lastNode;
+        }
+        let entry = categories.get(category.id);
+        if (entry === undefined) {
+            entry = [];
+            categories.set(category.id, entry);
+        }
+        if (subCategory) {
+            entry.push(subCategory.id);
         }
     }
     return [...categories.entries()].map(([categoryId, subCategoryIds]) => ({
@@ -105,11 +144,21 @@ function getEmptyTreeContentComponent(filter, error, emptyTreeContent) {
     if (emptyTreeContent) {
         return emptyTreeContent;
     }
-    return _jsx(EmptyTreeContent, { icon: categoryIcon });
+    return _jsx(EmptyTreeContent, { icon: categorySvg });
 }
-const categoryIcon = new URL("@itwin/itwinui-icons/bis-category-3d.svg", import.meta.url).href;
-function getIcon() {
-    return categoryIcon;
+function getIcon(node) {
+    if (node.extendedData?.imageId === undefined) {
+        return undefined;
+    }
+    switch (node.extendedData.imageId) {
+        case "icon-layers":
+            return categorySvg;
+        case "icon-layers-isolate":
+            return subcategorySvg;
+        case "icon-definition-container":
+            return definitionContainerSvg;
+    }
+    return undefined;
 }
 function getSublabel(node) {
     return node.extendedData?.description;

package/lib/esm/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.js.map

@@ -1 +1 @@
-{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACnD,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAC1E,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AACzE,OAAO,EAAE,2BAA2B,EAAE,MAAM,kCAAkC,CAAC;AA+B/E;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,oBAAoB,EAAE,gBAAgB,EAA0B;IACtH,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,EAA4C,CAAC;IACjG,MAAM,wBAAwB,GAAG,WAAW,CAAC,GAAG,EAAE;QAChD,MAAM,iBAAiB,GAAG,IAAI,2BAA2B,CAAC;YACxD,QAAQ,EAAE,UAAU;SACrB,CAAC,CAAC;QACH,OAAO;YACL,mBAAmB,EAAE,KAAK,EAAE,IAAmB,EAAE,EAAE,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC/F,gBAAgB,EAAE,KAAK,EAAE,IAAmB,EAAE,EAAW,EAAE,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1G,kBAAkB,EAAE,iBAAiB,CAAC,kBAAkB;YACxD,OAAO,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE;SAC3C,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,CAAC,CAAC,CAAC;IACjB,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;IACpG,CAAC,EACD,CAAC,UAAU,CAAC,CACb,CAAC;IAEF,MAAM,gBAAgB,GAAG,OAAO,CAAsD,GAAG,EAAE;QACzF,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,oBAAoB,EAAE,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,SAAS,CAAC;SAClB;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;YAChC,aAAa,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,IAAI;gBACF,MAAM,KAAK,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,CAAC;gBACrJ,oBAAoB,EAAE,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,CAAC;gBAC7C,OAAO,KAAK,CAAC;aACd;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;gBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE;oBACvC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;oBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;iBACjE;gBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;aACX;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,EAAE,UAAU,EAAE,aAAa,EAAE,oBAAoB,CAAC,CAAC,CAAC;IAE9D,OAAO;QACL,mBAAmB,EAAE;YACnB,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,gBAAgB;YAChB,wBAAwB;YACxB,gBAAgB,EAAE,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC;YACxF,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;SACjD;QACD,aAAa,EAAE;YACb,OAAO;YACP,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,SAAS,aAAa,CAAC,KAA8B;IACnD,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4B,CAAC;IACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;QACxD,MAAM,CAAC,QAAQ,EAAE,WAAW,CAAC,GAAG,QAAQ,CAAC;QAEzC,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE;YAC/D,SAAS;SACV;QAED,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,EAAE;YAChC,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;SACjC;QAED,IAAI,WAAW,IAAI,uBAAuB,CAAC,wBAAwB,CAAC,WAAW,CAAC,EAAE;YAChF,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAE,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;SACnD;KACF;IAED,OAAO,CAAC,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QACtE,UAAU;QACV,cAAc,EAAE,cAAc,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;KACzE,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAoC,EAAE,gBAAkC;IAC7H,IAAI,KAAK,EAAE;QACT,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,4BAA4B,KAAK,EAAE,CAAC,GAAQ,CAAC;KACjF;IACD,IAAI,MAAM,EAAE;QACV,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,CAAC,GAAQ,CAAC;KAC9F;IACD,IAAI,gBAAgB,EAAE;QACpB,OAAO,gBAAgB,CAAC;KACzB;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,YAAY,GAAI,CAAC;AAClD,CAAC;AAED,MAAM,YAAY,GAAG,IAAI,GAAG,CAAC,0CAA0C,EAAE,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC;AAC/F,SAAS,OAAO;IACd,OAAO,YAAY,CAAC;AACtB,CAAC;AAED,SAAS,WAAW,CAAC,IAA+B;IAClD,OAAO,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;AACxC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { Text } from \"@itwin/itwinui-react/bricks\";\nimport { HierarchyNodeIdentifier } from \"@itwin/presentation-hierarchies\";\nimport { TreeWidget } from \"../../../TreeWidget.js\";\nimport { EmptyTreeContent } from \"../common/components/EmptyTreeContent.js\";\nimport { FilterLimitExceededError } from \"../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../common/UseTelemetryContext.js\";\nimport { CategoriesTreeDefinition } from \"./CategoriesTreeDefinition.js\";\nimport { CategoriesVisibilityHandler } from \"./CategoriesVisibilityHandler.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\n\ntype CategoriesTreeFilteringError = \"tooManyFilterMatches\" | \"unknownFilterError\";\ntype HierarchyFilteringPaths = Awaited<ReturnType<Required<VisibilityTreeProps>[\"getFilteredPaths\"]>>;\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: Viewport;\n  onCategoriesFiltered?: (categories: CategoryInfo[] | undefined) => void;\n  filter?: string;\n  emptyTreeContent?: ReactNode;\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  categoriesTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlight\" | \"emptyTreeContent\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getIcon\" | \"getSublabel\">>;\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({ filter, activeView, onCategoriesFiltered, emptyTreeContent }: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const [filteringError, setFilteringError] = useState<CategoriesTreeFilteringError | undefined>();\n  const visibilityHandlerFactory = useCallback(() => {\n    const visibilityHandler = new CategoriesVisibilityHandler({\n      viewport: activeView,\n    });\n    return {\n      getVisibilityStatus: async (node: HierarchyNode) => visibilityHandler.getVisibilityStatus(node),\n      changeVisibility: async (node: HierarchyNode, on: boolean) => visibilityHandler.changeVisibility(node, on),\n      onVisibilityChange: visibilityHandler.onVisibilityChange,\n      dispose: () => visibilityHandler.dispose(),\n    };\n  }, [activeView]);\n  const { onFeatureUsed } = useTelemetryContext();\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType: activeView.view.is2d() ? \"2d\" : \"3d\" });\n    },\n    [activeView],\n  );\n\n  const getFilteredPaths = useMemo<VisibilityTreeProps[\"getFilteredPaths\"] | undefined>(() => {\n    setFilteringError(undefined);\n    onCategoriesFiltered?.(undefined);\n    if (!filter) {\n      return undefined;\n    }\n    return async ({ imodelAccess }) => {\n      onFeatureUsed({ featureId: \"filtering\", reportInteraction: true });\n      try {\n        const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({ imodelAccess, label: filter, viewType: activeView.view.is2d() ? \"2d\" : \"3d\" });\n        onCategoriesFiltered?.(getCategories(paths));\n        return paths;\n      } catch (e) {\n        const newError = e instanceof FilterLimitExceededError ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n        if (newError !== \"tooManyFilterMatches\") {\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        setFilteringError(newError);\n        return [];\n      }\n    };\n  }, [filter, activeView, onFeatureUsed, onCategoriesFiltered]);\n\n  return {\n    categoriesTreeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getFilteredPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent),\n      highlight: filter ? { text: filter } : undefined,\n    },\n    rendererProps: {\n      getIcon,\n      getSublabel,\n    },\n  };\n}\n\nfunction getCategories(paths: HierarchyFilteringPaths): CategoryInfo[] | undefined {\n  if (!paths) {\n    return undefined;\n  }\n\n  const categories = new Map<Id64String, Id64String[]>();\n  for (const path of paths) {\n    const currPath = Array.isArray(path) ? path : path.path;\n    const [category, subCategory] = currPath;\n\n    if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(category)) {\n      continue;\n    }\n\n    if (!categories.has(category.id)) {\n      categories.set(category.id, []);\n    }\n\n    if (subCategory && HierarchyNodeIdentifier.isInstanceNodeIdentifier(subCategory)) {\n      categories.get(category.id)!.push(subCategory.id);\n    }\n  }\n\n  return [...categories.entries()].map(([categoryId, subCategoryIds]) => ({\n    categoryId,\n    subCategoryIds: subCategoryIds.length === 0 ? undefined : subCategoryIds,\n  }));\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: CategoriesTreeFilteringError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    return <Text>{TreeWidget.translate(`categoriesTree.filtering.${error}`)}</Text>;\n  }\n  if (filter) {\n    return <Text>{TreeWidget.translate(\"categoriesTree.filtering.noMatches\", { filter })}</Text>;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categoryIcon} />;\n}\n\nconst categoryIcon = new URL(\"@itwin/itwinui-icons/bis-category-3d.svg\", import.meta.url).href;\nfunction getIcon() {\n  return categoryIcon;\n}\n\nfunction getSublabel(node: PresentationHierarchyNode) {\n  return node.extendedData?.description;\n}\n"]}
+{"version":3,"file":"UseCategoriesTree.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/UseCategoriesTree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,MAAM,EAAE,MAAM,qBAAqB,CAAC;AAC7C,OAAO,WAAW,MAAM,0CAA0C,CAAC;AACnE,OAAO,cAAc,MAAM,mDAAmD,CAAC;AAC/E,OAAO,sBAAsB,MAAM,oDAAoD,CAAC;AACxF,OAAO,EAAE,IAAI,EAAE,MAAM,6BAA6B,CAAC;AACnD,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,sBAAsB,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAC;AAClG,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,gBAAgB,EAAE,MAAM,0CAA0C,CAAC;AAC5E,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,mBAAmB,EAAE,MAAM,kCAAkC,CAAC;AACvE,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AACzE,OAAO,EAAE,sBAAsB,EAAE,MAAM,sCAAsC,CAAC;AAC9E,OAAO,EAAE,2BAA2B,EAAE,MAAM,2CAA2C,CAAC;AACxF,OAAO,EAAE,0BAA0B,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AA+BpG;;;GAGG;AACH,MAAM,UAAU,iBAAiB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,oBAAoB,EAAE,gBAAgB,EAA0B;IACtH,MAAM,CAAC,cAAc,EAAE,iBAAiB,CAAC,GAAG,QAAQ,EAA4C,CAAC;IAEjG,MAAM,QAAQ,GAAG,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC;IACtD,MAAM,MAAM,GAAG,UAAU,CAAC,MAAM,CAAC;IAEjC,MAAM,QAAQ,GAAG,OAAO,CAAC,GAAG,EAAE;QAC5B,OAAO,IAAI,sBAAsB,CAAC,wBAAwB,CAAC,MAAM,CAAC,EAAE,QAAQ,CAAC,CAAC;IAChF,CAAC,EAAE,CAAC,QAAQ,EAAE,MAAM,CAAC,CAAC,CAAC;IAEvB,MAAM,wBAAwB,GAAG,WAAW,CAAC,GAAG,EAAE;QAChD,MAAM,iBAAiB,GAAG,IAAI,2BAA2B,CAAC;YACxD,QAAQ,EAAE,UAAU;YACpB,QAAQ;SACT,CAAC,CAAC;QACH,OAAO;YACL,mBAAmB,EAAE,KAAK,EAAE,IAAmB,EAAE,EAAE,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,IAAI,CAAC;YAC/F,gBAAgB,EAAE,KAAK,EAAE,IAAmB,EAAE,EAAW,EAAE,EAAE,CAAC,iBAAiB,CAAC,gBAAgB,CAAC,IAAI,EAAE,EAAE,CAAC;YAC1G,kBAAkB,EAAE,iBAAiB,CAAC,kBAAkB;YACxD,OAAO,EAAE,GAAG,EAAE,CAAC,iBAAiB,CAAC,OAAO,EAAE;SAC3C,CAAC;IACJ,CAAC,EAAE,CAAC,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC;IAC3B,MAAM,EAAE,aAAa,EAAE,GAAG,mBAAmB,EAAE,CAAC;IAEhD,MAAM,sBAAsB,GAAG,WAAW,CACxC,CAAC,KAAK,EAAE,EAAE;QACR,OAAO,IAAI,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;IACxE,CAAC,EACD,CAAC,QAAQ,EAAE,QAAQ,CAAC,CACrB,CAAC;IAEF,MAAM,gBAAgB,GAAG,OAAO,CAAsD,GAAG,EAAE;QACzF,iBAAiB,CAAC,SAAS,CAAC,CAAC;QAC7B,oBAAoB,EAAE,CAAC,SAAS,CAAC,CAAC;QAClC,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,SAAS,CAAC;SAClB;QACD,OAAO,KAAK,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE;YAChC,aAAa,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,iBAAiB,EAAE,IAAI,EAAE,CAAC,CAAC;YACnE,IAAI;gBACF,MAAM,KAAK,GAAG,MAAM,wBAAwB,CAAC,sBAAsB,CAAC,EAAE,YAAY,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,CAAC,CAAC;gBACzH,oBAAoB,EAAE,CAAC,MAAM,sBAAsB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;gBACtE,OAAO,KAAK,CAAC;aACd;YAAC,OAAO,CAAC,EAAE;gBACV,MAAM,QAAQ,GAAG,CAAC,YAAY,wBAAwB,CAAC,CAAC,CAAC,sBAAsB,CAAC,CAAC,CAAC,oBAAoB,CAAC;gBACvG,IAAI,QAAQ,KAAK,sBAAsB,EAAE;oBACvC,MAAM,OAAO,GAAG,CAAC,YAAY,KAAK,IAAI,CAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,iDAAiD,CAAC,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,eAAe,CAAC;oBAChJ,aAAa,CAAC,EAAE,SAAS,EAAE,OAAO,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;iBACjE;gBACD,iBAAiB,CAAC,QAAQ,CAAC,CAAC;gBAC5B,OAAO,EAAE,CAAC;aACX;QACH,CAAC,CAAC;IACJ,CAAC,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,aAAa,EAAE,oBAAoB,EAAE,QAAQ,CAAC,CAAC,CAAC;IAEtE,OAAO;QACL,mBAAmB,EAAE;YACnB,QAAQ,EAAE,oBAAoB;YAC9B,sBAAsB;YACtB,gBAAgB;YAChB,wBAAwB;YACxB,gBAAgB,EAAE,4BAA4B,CAAC,MAAM,EAAE,cAAc,EAAE,gBAAgB,CAAC;YACxF,SAAS,EAAE,MAAM,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,MAAM,EAAE,CAAC,CAAC,CAAC,SAAS;SACjD;QACD,aAAa,EAAE;YACb,OAAO;YACP,WAAW;SACZ;KACF,CAAC;AACJ,CAAC;AAED,KAAK,UAAU,sBAAsB,CAAC,KAA8B,EAAE,QAAgC;IACpG,IAAI,CAAC,KAAK,EAAE;QACV,OAAO,SAAS,CAAC;KAClB;IAED,MAAM,UAAU,GAAG,IAAI,GAAG,EAA4B,CAAC;IACvD,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE;QACxB,MAAM,QAAQ,GAAG,sBAAsB,CAAC,SAAS,CAAC,IAAI,CAAC,CAAC,IAAI,CAAC;QAC7D,IAAI,QAAQ,CAAC,MAAM,KAAK,CAAC,EAAE;YACzB,SAAS;SACV;QAED,IAAI,QAAiC,CAAC;QACtC,IAAI,WAAgD,CAAC;QACrD,MAAM,QAAQ,GAAG,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;QAE/C,IAAI,CAAC,uBAAuB,CAAC,wBAAwB,CAAC,QAAQ,CAAC,EAAE;YAC/D,SAAS;SACV;QAED,IAAI,QAAQ,CAAC,SAAS,KAAK,0BAA0B,EAAE;YACrD,MAAM,6BAA6B,GAAG,MAAM,QAAQ,CAAC,yBAAyB,CAAC,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YAC9F,KAAK,MAAM,UAAU,IAAI,6BAA6B,EAAE;gBACtD,MAAM,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,UAAU,CAAC,CAAC;gBACzC,IAAI,KAAK,KAAK,SAAS,EAAE;oBACvB,UAAU,CAAC,GAAG,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;iBAChC;aACF;YACD,SAAS;SACV;QAED,IAAI,QAAQ,CAAC,SAAS,KAAK,kBAAkB,EAAE;YAC7C,MAAM,gBAAgB,GAAG,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,CAAC,QAAQ,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACzF,MAAM,CAAC,gBAAgB,KAAK,SAAS,IAAI,uBAAuB,CAAC,wBAAwB,CAAC,gBAAgB,CAAC,CAAC,CAAC;YAE7G,WAAW,GAAG,QAAQ,CAAC;YACvB,QAAQ,GAAG,gBAAgB,CAAC;SAC7B;aAAM;YACL,QAAQ,GAAG,QAAQ,CAAC;SACrB;QAED,IAAI,KAAK,GAAG,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;QACxC,IAAI,KAAK,KAAK,SAAS,EAAE;YACvB,KAAK,GAAG,EAAE,CAAC;YACX,UAAU,CAAC,GAAG,CAAC,QAAQ,CAAC,EAAE,EAAE,KAAK,CAAC,CAAC;SACpC;QAED,IAAI,WAAW,EAAE;YACf,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC;SAC5B;KACF;IAED,OAAO,CAAC,GAAG,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,UAAU,EAAE,cAAc,CAAC,EAAE,EAAE,CAAC,CAAC;QACtE,UAAU;QACV,cAAc,EAAE,cAAc,CAAC,MAAM,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,cAAc;KACzE,CAAC,CAAC,CAAC;AACN,CAAC;AAED,SAAS,4BAA4B,CAAC,MAAe,EAAE,KAAoC,EAAE,gBAAkC;IAC7H,IAAI,KAAK,EAAE;QACT,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,4BAA4B,KAAK,EAAE,CAAC,GAAQ,CAAC;KACjF;IACD,IAAI,MAAM,EAAE;QACV,OAAO,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,CAAC,GAAQ,CAAC;KAC9F;IACD,IAAI,gBAAgB,EAAE;QACpB,OAAO,gBAAgB,CAAC;KACzB;IACD,OAAO,KAAC,gBAAgB,IAAC,IAAI,EAAE,WAAW,GAAI,CAAC;AACjD,CAAC;AAGD,SAAS,OAAO,CAAC,IAA+B;IAC9C,IAAI,IAAI,CAAC,YAAY,EAAE,OAAO,KAAK,SAAS,EAAE;QAC5C,OAAO,SAAS,CAAC;KAClB;IAED,QAAQ,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE;QACjC,KAAK,aAAa;YAChB,OAAO,WAAW,CAAC;QACrB,KAAK,qBAAqB;YACxB,OAAO,cAAc,CAAC;QACxB,KAAK,2BAA2B;YAC9B,OAAO,sBAAsB,CAAC;KACjC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,WAAW,CAAC,IAA+B;IAClD,OAAO,IAAI,CAAC,YAAY,EAAE,WAAW,CAAC;AACxC,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { assert } from \"@itwin/core-bentley\";\nimport categorySvg from \"@itwin/itwinui-icons/bis-category-3d.svg\";\nimport subcategorySvg from \"@itwin/itwinui-icons/bis-category-subcategory.svg\";\nimport definitionContainerSvg from \"@itwin/itwinui-icons/bis-definitions-container.svg\";\nimport { Text } from \"@itwin/itwinui-react/bricks\";\nimport { createECSqlQueryExecutor } from \"@itwin/presentation-core-interop\";\nimport { HierarchyFilteringPath, HierarchyNodeIdentifier } from \"@itwin/presentation-hierarchies\";\nimport { TreeWidget } from \"../../../TreeWidget.js\";\nimport { EmptyTreeContent } from \"../common/components/EmptyTreeContent.js\";\nimport { FilterLimitExceededError } from \"../common/TreeErrors.js\";\nimport { useTelemetryContext } from \"../common/UseTelemetryContext.js\";\nimport { CategoriesTreeDefinition } from \"./CategoriesTreeDefinition.js\";\nimport { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport { CategoriesVisibilityHandler } from \"./internal/CategoriesVisibilityHandler.js\";\nimport { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from \"./internal/ClassNameDefinitions.js\";\n\nimport type { ReactNode } from \"react\";\nimport type { Id64String } from \"@itwin/core-bentley\";\nimport type { HierarchyNode } from \"@itwin/presentation-hierarchies\";\nimport type { VisibilityTreeProps } from \"../common/components/VisibilityTree.js\";\nimport type { Viewport } from \"@itwin/core-frontend\";\nimport type { PresentationHierarchyNode } from \"@itwin/presentation-hierarchies-react\";\nimport type { VisibilityTreeRendererProps } from \"../common/components/VisibilityTreeRenderer.js\";\nimport type { CategoryInfo } from \"../common/CategoriesVisibilityUtils.js\";\n\ntype CategoriesTreeFilteringError = \"tooManyFilterMatches\" | \"unknownFilterError\";\ntype HierarchyFilteringPaths = Awaited<ReturnType<Required<VisibilityTreeProps>[\"getFilteredPaths\"]>>;\n\n/** @beta */\nexport interface UseCategoriesTreeProps {\n  activeView: Viewport;\n  onCategoriesFiltered?: (categories: CategoryInfo[] | undefined) => void;\n  filter?: string;\n  emptyTreeContent?: ReactNode;\n}\n\n/** @beta */\ninterface UseCategoriesTreeResult {\n  categoriesTreeProps: Pick<\n    VisibilityTreeProps,\n    \"treeName\" | \"getHierarchyDefinition\" | \"getFilteredPaths\" | \"visibilityHandlerFactory\" | \"highlight\" | \"emptyTreeContent\"\n  >;\n  rendererProps: Required<Pick<VisibilityTreeRendererProps, \"getIcon\" | \"getSublabel\">>;\n}\n\n/**\n * Custom hook to create and manage state for the categories tree.\n * @beta\n */\nexport function useCategoriesTree({ filter, activeView, onCategoriesFiltered, emptyTreeContent }: UseCategoriesTreeProps): UseCategoriesTreeResult {\n  const [filteringError, setFilteringError] = useState<CategoriesTreeFilteringError | undefined>();\n\n  const viewType = activeView.view.is2d() ? \"2d\" : \"3d\";\n  const iModel = activeView.iModel;\n\n  const idsCache = useMemo(() => {\n    return new CategoriesTreeIdsCache(createECSqlQueryExecutor(iModel), viewType);\n  }, [viewType, iModel]);\n\n  const visibilityHandlerFactory = useCallback(() => {\n    const visibilityHandler = new CategoriesVisibilityHandler({\n      viewport: activeView,\n      idsCache,\n    });\n    return {\n      getVisibilityStatus: async (node: HierarchyNode) => visibilityHandler.getVisibilityStatus(node),\n      changeVisibility: async (node: HierarchyNode, on: boolean) => visibilityHandler.changeVisibility(node, on),\n      onVisibilityChange: visibilityHandler.onVisibilityChange,\n      dispose: () => visibilityHandler.dispose(),\n    };\n  }, [activeView, idsCache]);\n  const { onFeatureUsed } = useTelemetryContext();\n\n  const getHierarchyDefinition = useCallback<VisibilityTreeProps[\"getHierarchyDefinition\"]>(\n    (props) => {\n      return new CategoriesTreeDefinition({ ...props, viewType, idsCache });\n    },\n    [viewType, idsCache],\n  );\n\n  const getFilteredPaths = useMemo<VisibilityTreeProps[\"getFilteredPaths\"] | undefined>(() => {\n    setFilteringError(undefined);\n    onCategoriesFiltered?.(undefined);\n    if (!filter) {\n      return undefined;\n    }\n    return async ({ imodelAccess }) => {\n      onFeatureUsed({ featureId: \"filtering\", reportInteraction: true });\n      try {\n        const paths = await CategoriesTreeDefinition.createInstanceKeyPaths({ imodelAccess, label: filter, viewType, idsCache });\n        onCategoriesFiltered?.(await getCategoriesFromPaths(paths, idsCache));\n        return paths;\n      } catch (e) {\n        const newError = e instanceof FilterLimitExceededError ? \"tooManyFilterMatches\" : \"unknownFilterError\";\n        if (newError !== \"tooManyFilterMatches\") {\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        setFilteringError(newError);\n        return [];\n      }\n    };\n  }, [filter, viewType, onFeatureUsed, onCategoriesFiltered, idsCache]);\n\n  return {\n    categoriesTreeProps: {\n      treeName: \"categories-tree-v2\",\n      getHierarchyDefinition,\n      getFilteredPaths,\n      visibilityHandlerFactory,\n      emptyTreeContent: getEmptyTreeContentComponent(filter, filteringError, emptyTreeContent),\n      highlight: filter ? { text: filter } : undefined,\n    },\n    rendererProps: {\n      getIcon,\n      getSublabel,\n    },\n  };\n}\n\nasync function getCategoriesFromPaths(paths: HierarchyFilteringPaths, idsCache: CategoriesTreeIdsCache): Promise<CategoryInfo[] | undefined> {\n  if (!paths) {\n    return undefined;\n  }\n\n  const categories = new Map<Id64String, Id64String[]>();\n  for (const path of paths) {\n    const currPath = HierarchyFilteringPath.normalize(path).path;\n    if (currPath.length === 0) {\n      continue;\n    }\n\n    let category: HierarchyNodeIdentifier;\n    let subCategory: HierarchyNodeIdentifier | undefined;\n    const lastNode = currPath[currPath.length - 1];\n\n    if (!HierarchyNodeIdentifier.isInstanceNodeIdentifier(lastNode)) {\n      continue;\n    }\n\n    if (lastNode.className === DEFINITION_CONTAINER_CLASS) {\n      const definitionContainerCategories = await idsCache.getAllContainedCategories([lastNode.id]);\n      for (const categoryId of definitionContainerCategories) {\n        const value = categories.get(categoryId);\n        if (value === undefined) {\n          categories.set(categoryId, []);\n        }\n      }\n      continue;\n    }\n\n    if (lastNode.className === SUB_CATEGORY_CLASS) {\n      const secondToLastNode = currPath.length > 1 ? currPath[currPath.length - 2] : undefined;\n      assert(secondToLastNode !== undefined && HierarchyNodeIdentifier.isInstanceNodeIdentifier(secondToLastNode));\n\n      subCategory = lastNode;\n      category = secondToLastNode;\n    } else {\n      category = lastNode;\n    }\n\n    let entry = categories.get(category.id);\n    if (entry === undefined) {\n      entry = [];\n      categories.set(category.id, entry);\n    }\n\n    if (subCategory) {\n      entry.push(subCategory.id);\n    }\n  }\n\n  return [...categories.entries()].map(([categoryId, subCategoryIds]) => ({\n    categoryId,\n    subCategoryIds: subCategoryIds.length === 0 ? undefined : subCategoryIds,\n  }));\n}\n\nfunction getEmptyTreeContentComponent(filter?: string, error?: CategoriesTreeFilteringError, emptyTreeContent?: React.ReactNode) {\n  if (error) {\n    return <Text>{TreeWidget.translate(`categoriesTree.filtering.${error}`)}</Text>;\n  }\n  if (filter) {\n    return <Text>{TreeWidget.translate(\"categoriesTree.filtering.noMatches\", { filter })}</Text>;\n  }\n  if (emptyTreeContent) {\n    return emptyTreeContent;\n  }\n  return <EmptyTreeContent icon={categorySvg} />;\n}\n\n\nfunction getIcon(node: PresentationHierarchyNode): string | undefined {\n  if (node.extendedData?.imageId === undefined) {\n    return undefined;\n  }\n\n  switch (node.extendedData.imageId) {\n    case \"icon-layers\":\n      return categorySvg;\n    case \"icon-layers-isolate\":\n      return subcategorySvg;\n    case \"icon-definition-container\":\n      return definitionContainerSvg;\n  }\n\n  return undefined;\n}\n\nfunction getSublabel(node: PresentationHierarchyNode) {\n  return node.extendedData?.description;\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.d.ts

@@ -0,0 +1,54 @@
+import type { Id64Array, Id64String } from "@itwin/core-bentley";
+import type { LimitingECSqlQueryExecutor } from "@itwin/presentation-hierarchies";
+import type { InstanceKey } from "@itwin/presentation-shared";
+interface CategoryInfo {
+    id: Id64String;
+    childCount: number;
+}
+/** @internal */
+export declare class CategoriesTreeIdsCache {
+    private _queryExecutor;
+    private _definitionContainersInfo;
+    private _modelsCategoriesInfo;
+    private _subCategoriesInfo;
+    private _categoryClass;
+    private _categoryElementClass;
+    private _isDefinitionContainerSupported;
+    constructor(_queryExecutor: LimitingECSqlQueryExecutor, viewType: "3d" | "2d");
+    private queryCategories;
+    private queryIsDefinitionContainersSupported;
+    private queryDefinitionContainers;
+    private queryVisibleSubCategories;
+    private getModelsCategoriesInfo;
+    private getSubCategoriesInfo;
+    private getDefinitionContainersInfo;
+    getDirectChildDefinitionContainersAndCategories(parentDefinitionContainerIds: Id64Array): Promise<{
+        categories: CategoryInfo[];
+        definitionContainers: Id64Array;
+    }>;
+    getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array>;
+    getInstanceKeyPaths(props: {
+        categoryId: Id64String;
+    } | {
+        definitionContainerId: Id64String;
+    } | {
+        subCategoryId: Id64String;
+    }): Promise<InstanceKey[]>;
+    getAllDefinitionContainersAndCategories(): Promise<{
+        categories: Id64Array;
+        definitionContainers: Id64Array;
+    }>;
+    getRootDefinitionContainersAndCategories(): Promise<{
+        categories: CategoryInfo[];
+        definitionContainers: Id64Array;
+    }>;
+    getSubCategories(categoryId: Id64String): Promise<Id64Array>;
+    getIsDefinitionContainerSupported(): Promise<boolean>;
+}
+/** @internal */
+export declare function getClassesByView(viewType: "2d" | "3d"): {
+    categoryClass: string;
+    categoryElementClass: string;
+};
+export {};
+//# sourceMappingURL=CategoriesTreeIdsCache.d.ts.map

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.js

@@ -0,0 +1,280 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+import { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from "./ClassNameDefinitions.js";
+/** @internal */
+export class CategoriesTreeIdsCache {
+    constructor(_queryExecutor, viewType) {
+        this._queryExecutor = _queryExecutor;
+        const { categoryClass, categoryElementClass } = getClassesByView(viewType);
+        this._categoryClass = categoryClass;
+        this._categoryElementClass = categoryElementClass;
+    }
+    async *queryCategories() {
+        const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();
+        const categoriesQuery = `
+      SELECT
+        this.ECInstanceId id,
+        COUNT(sc.ECInstanceId) childCount,
+        this.Model.Id modelId,
+        ${isDefinitionContainerSupported
+            ? `
+            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS} dc),
+              true,
+              false
+            )`
+            : "false"} parentDefinitionContainerExists
+      FROM
+        ${this._categoryClass} this
+        JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId
+        JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id
+      WHERE
+        NOT this.IsPrivate
+        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))
+        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)
+      GROUP BY this.ECInstanceId
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, modelId: row.modelId, parentDefinitionContainerExists: row.parentDefinitionContainerExists, childCount: row.childCount };
+        }
+    }
+    async queryIsDefinitionContainersSupported() {
+        const query = `
+      SELECT
+        1
+      FROM
+        ECDbMeta.ECSchemaDef s
+        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId
+      WHERE
+        s.Name = 'BisCore'
+        AND c.Name = 'DefinitionContainer'
+    `;
+        for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query })) {
+            return true;
+        }
+        return false;
+    }
+    async *queryDefinitionContainers() {
+        // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:
+        //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}
+        //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId
+        //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId
+        const DEFINITION_CONTAINERS_CTE = "DefinitionContainers";
+        const ctes = [
+            `
+        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (
+          SELECT
+            dc.ECInstanceId,
+            dc.Model.Id
+          FROM
+            ${DEFINITION_CONTAINER_CLASS} dc
+          WHERE
+            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate)
+            AND NOT dc.IsPrivate
+
+          UNION ALL
+
+          SELECT
+            pdc.ECInstanceId,
+            pdc.Model.Id
+          FROM
+            ${DEFINITION_CONTAINERS_CTE} cdc
+            JOIN ${DEFINITION_CONTAINER_CLASS} pdc ON pdc.ECInstanceId = cdc.ModelId
+          WHERE
+            NOT pdc.IsPrivate
+        )
+      `,
+        ];
+        const definitionsQuery = `
+      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, modelId: row.modelId };
+        }
+    }
+    async *queryVisibleSubCategories(categoriesInfo) {
+        const definitionsQuery = `
+      SELECT
+        sc.ECInstanceId id,
+        sc.Parent.Id categoryId
+      FROM
+        ${SUB_CATEGORY_CLASS} sc
+      WHERE
+        NOT sc.IsPrivate
+        AND sc.Parent.Id IN (${categoriesInfo.join(",")})
+    `;
+        for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: "ECSqlPropertyNames", limit: "unbounded" })) {
+            yield { id: row.id, parentId: row.categoryId };
+        }
+    }
+    async getModelsCategoriesInfo() {
+        this._modelsCategoriesInfo ??= (async () => {
+            const allModelsCategories = new Map();
+            for await (const queriedCategory of this.queryCategories()) {
+                let modelCategories = allModelsCategories.get(queriedCategory.modelId);
+                if (modelCategories === undefined) {
+                    modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };
+                    allModelsCategories.set(queriedCategory.modelId, modelCategories);
+                }
+                modelCategories.childCategories.push({ id: queriedCategory.id, childCount: queriedCategory.childCount });
+            }
+            return allModelsCategories;
+        })();
+        return this._modelsCategoriesInfo;
+    }
+    async getSubCategoriesInfo() {
+        this._subCategoriesInfo ??= (async () => {
+            const allSubCategories = new Map();
+            const modelsCategoriesInfo = await this.getModelsCategoriesInfo();
+            const categoriesWithMoreThanOneSubCategory = new Array();
+            for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {
+                categoriesWithMoreThanOneSubCategory.push(...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.childCount > 1).map((categoryInfo) => categoryInfo.id));
+            }
+            if (categoriesWithMoreThanOneSubCategory.length === 0) {
+                return allSubCategories;
+            }
+            for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {
+                allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });
+            }
+            return allSubCategories;
+        })();
+        return this._subCategoriesInfo;
+    }
+    async getDefinitionContainersInfo() {
+        this._definitionContainersInfo ??= (async () => {
+            const definitionContainersInfo = new Map();
+            const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([
+                this.getIsDefinitionContainerSupported(),
+                this.getModelsCategoriesInfo(),
+            ]);
+            if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {
+                return definitionContainersInfo;
+            }
+            for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {
+                const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);
+                definitionContainersInfo.set(queriedDefinitionContainer.id, {
+                    childCategories: modelCategoriesInfo?.childCategories ?? [],
+                    modelId: queriedDefinitionContainer.modelId,
+                    childDefinitionContainers: [],
+                    parentDefinitionContainerExists: false,
+                });
+            }
+            for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {
+                const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);
+                if (parentDefinitionContainer !== undefined) {
+                    parentDefinitionContainer.childDefinitionContainers.push(definitionContainerId);
+                    definitionContainerInfo.parentDefinitionContainerExists = true;
+                }
+            }
+            return definitionContainersInfo;
+        })();
+        return this._definitionContainersInfo;
+    }
+    async getDirectChildDefinitionContainersAndCategories(parentDefinitionContainerIds) {
+        const definitionContainersInfo = await this.getDefinitionContainersInfo();
+        const result = { definitionContainers: new Array(), categories: new Array() };
+        parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {
+            const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);
+            if (parentDefinitionContainerInfo !== undefined) {
+                result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainers);
+                result.categories.push(...parentDefinitionContainerInfo.childCategories);
+            }
+        });
+        return result;
+    }
+    async getAllContainedCategories(definitionContainerIds) {
+        const result = new Array();
+        const definitionContainersInfo = await this.getDefinitionContainersInfo();
+        const indirectCategories = await Promise.all(definitionContainerIds.map(async (definitionContainerId) => {
+            const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);
+            if (definitionContainerInfo === undefined) {
+                return [];
+            }
+            result.push(...definitionContainerInfo.childCategories.map((category) => category.id));
+            return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainers);
+        }));
+        for (const categories of indirectCategories) {
+            result.push(...categories);
+        }
+        return result;
+    }
+    async getInstanceKeyPaths(props) {
+        if ("subCategoryId" in props) {
+            const subCategoriesInfo = await this.getSubCategoriesInfo();
+            const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);
+            if (subCategoryInfo === undefined) {
+                return [];
+            }
+            return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS }];
+        }
+        if ("categoryId" in props) {
+            const modelsCategoriesInfo = await this.getModelsCategoriesInfo();
+            for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {
+                if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {
+                    if (!modelCategoriesInfo.parentDefinitionContainerExists) {
+                        return [{ id: props.categoryId, className: this._categoryClass }];
+                    }
+                    return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];
+                }
+            }
+            return [];
+        }
+        const definitionContainersInfo = await this.getDefinitionContainersInfo();
+        const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);
+        if (definitionContainerInfo === undefined) {
+            return [];
+        }
+        if (!definitionContainerInfo.parentDefinitionContainerExists) {
+            return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS }];
+        }
+        return [
+            ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),
+            { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS },
+        ];
+    }
+    async getAllDefinitionContainersAndCategories() {
+        const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);
+        const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array() };
+        for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {
+            result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));
+        }
+        return result;
+    }
+    async getRootDefinitionContainersAndCategories() {
+        const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);
+        const result = { definitionContainers: new Array(), categories: new Array() };
+        for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {
+            if (!modelCategoriesInfo.parentDefinitionContainerExists) {
+                result.categories.push(...modelCategoriesInfo.childCategories);
+            }
+        }
+        for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {
+            if (!definitionContainerInfo.parentDefinitionContainerExists) {
+                result.definitionContainers.push(definitionContainerId);
+            }
+        }
+        return result;
+    }
+    async getSubCategories(categoryId) {
+        const subCategoriesInfo = await this.getSubCategoriesInfo();
+        const result = new Array();
+        for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {
+            if (subCategoryInfo.categoryId === categoryId) {
+                result.push(subCategoryId);
+            }
+        }
+        return result;
+    }
+    async getIsDefinitionContainerSupported() {
+        this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();
+        return this._isDefinitionContainerSupported;
+    }
+}
+/** @internal */
+export function getClassesByView(viewType) {
+    return viewType === "2d"
+        ? { categoryClass: "BisCore.DrawingCategory", categoryElementClass: "BisCore.GeometricElement2d" }
+        : { categoryClass: "BisCore.SpatialCategory", categoryElementClass: "BisCore.GeometricElement3d" };
+}
+//# sourceMappingURL=CategoriesTreeIdsCache.js.map

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CategoriesTreeIdsCache.js","sourceRoot":"","sources":["../../../../../../../src/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeIdsCache.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,0BAA0B,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAC;AA2B3F,gBAAgB;AAChB,MAAM,OAAO,sBAAsB;IAQjC,YACU,cAA0C,EAClD,QAAqB;QADb,mBAAc,GAAd,cAAc,CAA4B;QAGlD,MAAM,EAAE,aAAa,EAAE,oBAAoB,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAC3E,IAAI,CAAC,cAAc,GAAG,aAAa,CAAC;QACpC,IAAI,CAAC,qBAAqB,GAAG,oBAAoB,CAAC;IACpD,CAAC;IAEO,KAAK,CAAC,CAAC,eAAe;QAM5B,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,iCAAiC,EAAE,CAAC;QACtF,MAAM,eAAe,GAAG;;;;;UAMlB,8BAA8B;YAC5B,CAAC,CAAC;gEACkD,0BAA0B;;;cAG5E;YACF,CAAC,CAAC,OACN;;UAEE,IAAI,CAAC,cAAc;eACd,kBAAkB;;;;;oCAKG,IAAI,CAAC,qBAAqB;;KAEzD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,eAAe,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YAClJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,+BAA+B,EAAE,GAAG,CAAC,+BAA+B,EAAE,UAAU,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAC9I;IACH,CAAC;IAEO,KAAK,CAAC,oCAAoC;QAChD,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,IAAI,KAAK,EAAE,MAAM,IAAI,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,EAAE;YAChF,OAAO,IAAI,CAAC;SACb;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB;QACtC,iKAAiK;QACjK,2EAA2E;QAC3E,2GAA2G;QAC3G,kFAAkF;QAClF,MAAM,yBAAyB,GAAG,sBAAsB,CAAC;QACzD,MAAM,IAAI,GAAG;YACX;UACI,yBAAyB;;;;;cAKrB,0BAA0B;;yDAEiB,IAAI,CAAC,cAAc;;;;;;;;;cAS9D,yBAAyB;mBACpB,0BAA0B;;;;OAItC;SACF,CAAC;QACF,MAAM,gBAAgB,GAAG;2DAC8B,yBAAyB;KAC/E,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACzJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,OAAO,EAAE,GAAG,CAAC,OAAO,EAAE,CAAC;SAC5C;IACH,CAAC;IAEO,KAAK,CAAC,CAAC,yBAAyB,CAAC,cAAyB;QAChE,MAAM,gBAAgB,GAAG;;;;;UAKnB,kBAAkB;;;+BAGG,cAAc,CAAC,IAAI,CAAC,GAAG,CAAC;KAClD,CAAC;QACF,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,EAAE,KAAK,EAAE,gBAAgB,EAAE,EAAE,EAAE,SAAS,EAAE,oBAAoB,EAAE,KAAK,EAAE,WAAW,EAAE,CAAC,EAAE;YACnJ,MAAM,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,QAAQ,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC;SAChD;IACH,CAAC;IAEO,KAAK,CAAC,uBAAuB;QACnC,IAAI,CAAC,qBAAqB,KAAK,CAAC,KAAK,IAAI,EAAE;YACzC,MAAM,mBAAmB,GAAG,IAAI,GAAG,EAA8B,CAAC;YAClE,IAAI,KAAK,EAAE,MAAM,eAAe,IAAI,IAAI,CAAC,eAAe,EAAE,EAAE;gBAC1D,IAAI,eAAe,GAAG,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,CAAC;gBACvE,IAAI,eAAe,KAAK,SAAS,EAAE;oBACjC,eAAe,GAAG,EAAE,+BAA+B,EAAE,eAAe,CAAC,+BAA+B,EAAE,eAAe,EAAE,EAAE,EAAE,CAAC;oBAC5H,mBAAmB,CAAC,GAAG,CAAC,eAAe,CAAC,OAAO,EAAE,eAAe,CAAC,CAAC;iBACnE;gBACD,eAAe,CAAC,eAAe,CAAC,IAAI,CAAC,EAAE,EAAE,EAAE,eAAe,CAAC,EAAE,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC;aAC1G;YACD,OAAO,mBAAmB,CAAC;QAC7B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,qBAAqB,CAAC;IACpC,CAAC;IAEO,KAAK,CAAC,oBAAoB;QAChC,IAAI,CAAC,kBAAkB,KAAK,CAAC,KAAK,IAAI,EAAE;YACtC,MAAM,gBAAgB,GAAG,IAAI,GAAG,EAA+B,CAAC;YAChE,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,MAAM,oCAAoC,GAAG,IAAI,KAAK,EAAc,CAAC;YACrE,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;gBAC/D,oCAAoC,CAAC,IAAI,CACvC,GAAG,mBAAmB,CAAC,eAAe,CAAC,MAAM,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,UAAU,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,YAAY,EAAE,EAAE,CAAC,YAAY,CAAC,EAAE,CAAC,CACpI,CAAC;aACH;YAED,IAAI,oCAAoC,CAAC,MAAM,KAAK,CAAC,EAAE;gBACrD,OAAO,gBAAgB,CAAC;aACzB;YAED,IAAI,KAAK,EAAE,MAAM,kBAAkB,IAAI,IAAI,CAAC,yBAAyB,CAAC,oCAAoC,CAAC,EAAE;gBAC3G,gBAAgB,CAAC,GAAG,CAAC,kBAAkB,CAAC,EAAE,EAAE,EAAE,UAAU,EAAE,kBAAkB,CAAC,QAAQ,EAAE,CAAC,CAAC;aAC1F;YACD,OAAO,gBAAgB,CAAC;QAC1B,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,kBAAkB,CAAC;IACjC,CAAC;IAEO,KAAK,CAAC,2BAA2B;QACvC,IAAI,CAAC,yBAAyB,KAAK,CAAC,KAAK,IAAI,EAAE;YAC7C,MAAM,wBAAwB,GAAG,IAAI,GAAG,EAAuC,CAAC;YAChF,MAAM,CAAC,8BAA8B,EAAE,oBAAoB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC;gBAC/E,IAAI,CAAC,iCAAiC,EAAE;gBACxC,IAAI,CAAC,uBAAuB,EAAE;aAC/B,CAAC,CAAC;YACH,IAAI,CAAC,8BAA8B,IAAI,oBAAoB,CAAC,IAAI,KAAK,CAAC,EAAE;gBACtE,OAAO,wBAAwB,CAAC;aACjC;YAED,IAAI,KAAK,EAAE,MAAM,0BAA0B,IAAI,IAAI,CAAC,yBAAyB,EAAE,EAAE;gBAC/E,MAAM,mBAAmB,GAAG,oBAAoB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,CAAC,CAAC;gBAEpF,wBAAwB,CAAC,GAAG,CAAC,0BAA0B,CAAC,EAAE,EAAE;oBAC1D,eAAe,EAAE,mBAAmB,EAAE,eAAe,IAAI,EAAE;oBAC3D,OAAO,EAAE,0BAA0B,CAAC,OAAO;oBAC3C,yBAAyB,EAAE,EAAE;oBAC7B,+BAA+B,EAAE,KAAK;iBACvC,CAAC,CAAC;aACJ;YAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;gBACvF,MAAM,yBAAyB,GAAG,wBAAwB,CAAC,GAAG,CAAC,uBAAuB,CAAC,OAAO,CAAC,CAAC;gBAChG,IAAI,yBAAyB,KAAK,SAAS,EAAE;oBAC3C,yBAAyB,CAAC,yBAAyB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;oBAChF,uBAAuB,CAAC,+BAA+B,GAAG,IAAI,CAAC;iBAChE;aACF;YAED,OAAO,wBAAwB,CAAC;QAClC,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,yBAAyB,CAAC;IACxC,CAAC;IAEM,KAAK,CAAC,+CAA+C,CAC1D,4BAAuC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAE1E,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QAExG,4BAA4B,CAAC,OAAO,CAAC,CAAC,2BAA2B,EAAE,EAAE;YACnE,MAAM,6BAA6B,GAAG,wBAAwB,CAAC,GAAG,CAAC,2BAA2B,CAAC,CAAC;YAChG,IAAI,6BAA6B,KAAK,SAAS,EAAE;gBAC/C,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,yBAAyB,CAAC,CAAC;gBAC7F,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,6BAA6B,CAAC,eAAe,CAAC,CAAC;aAC1E;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,yBAAyB,CAAC,sBAAiC;QACtE,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QAEvC,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,kBAAkB,GAAG,MAAM,OAAO,CAAC,GAAG,CAC1C,sBAAsB,CAAC,GAAG,CAAC,KAAK,EAAE,qBAAqB,EAAE,EAAE;YACzD,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,qBAAqB,CAAC,CAAC;YACpF,IAAI,uBAAuB,KAAK,SAAS,EAAE;gBACzC,OAAO,EAAE,CAAC;aACX;YACD,MAAM,CAAC,IAAI,CAAC,GAAG,uBAAuB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,CAAC;YACvF,OAAO,IAAI,CAAC,yBAAyB,CAAC,uBAAuB,CAAC,yBAAyB,CAAC,CAAC;QAC3F,CAAC,CAAC,CACH,CAAC;QACF,KAAK,MAAM,UAAU,IAAI,kBAAkB,EAAE;YAC3C,MAAM,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,CAAC;SAC5B;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,mBAAmB,CAC9B,KAAyG;QAEzG,IAAI,eAAe,IAAI,KAAK,EAAE;YAC5B,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;YAC5D,MAAM,eAAe,GAAG,iBAAiB,CAAC,GAAG,CAAC,KAAK,CAAC,aAAa,CAAC,CAAC;YACnE,IAAI,eAAe,KAAK,SAAS,EAAE;gBACjC,OAAO,EAAE,CAAC;aACX;YACD,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,eAAe,CAAC,UAAU,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,aAAa,EAAE,SAAS,EAAE,kBAAkB,EAAE,CAAC,CAAC;SACtJ;QAED,IAAI,YAAY,IAAI,KAAK,EAAE;YACzB,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,uBAAuB,EAAE,CAAC;YAClE,KAAK,MAAM,CAAC,OAAO,EAAE,mBAAmB,CAAC,IAAI,oBAAoB,EAAE;gBACjE,IAAI,mBAAmB,CAAC,eAAe,CAAC,IAAI,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,KAAK,KAAK,CAAC,UAAU,CAAC,EAAE;oBACtG,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;wBACxD,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;qBACnE;oBAED,OAAO,CAAC,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,OAAO,EAAE,CAAC,CAAC,EAAE,EAAE,EAAE,EAAE,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,CAAC,CAAC;iBAC5I;aACF;YACD,OAAO,EAAE,CAAC;SACX;QAED,MAAM,wBAAwB,GAAG,MAAM,IAAI,CAAC,2BAA2B,EAAE,CAAC;QAC1E,MAAM,uBAAuB,GAAG,wBAAwB,CAAC,GAAG,CAAC,KAAK,CAAC,qBAAqB,CAAC,CAAC;QAC1F,IAAI,uBAAuB,KAAK,SAAS,EAAE;YACzC,OAAO,EAAE,CAAC;SACX;QAED,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;YAC5D,OAAO,CAAC,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE,CAAC,CAAC;SACrF;QAED,OAAO;YACL,GAAG,CAAC,MAAM,IAAI,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,uBAAuB,CAAC,OAAO,EAAE,CAAC,CAAC;YAC/F,EAAE,EAAE,EAAE,KAAK,CAAC,qBAAqB,EAAE,SAAS,EAAE,0BAA0B,EAAE;SAC3E,CAAC;IACJ,CAAC;IAEM,KAAK,CAAC,uCAAuC;QAClD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,CAAC,GAAG,wBAAwB,CAAC,IAAI,EAAE,CAAC,EAAE,UAAU,EAAE,IAAI,KAAK,EAAc,EAAE,CAAC;QACnH,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,aAAa,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC,CAAC;SACzG;QAED,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,wCAAwC;QACnD,MAAM,CAAC,oBAAoB,EAAE,wBAAwB,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,uBAAuB,EAAE,EAAE,IAAI,CAAC,2BAA2B,EAAE,CAAC,CAAC,CAAC;QACjJ,MAAM,MAAM,GAAG,EAAE,oBAAoB,EAAE,IAAI,KAAK,EAAc,EAAE,UAAU,EAAE,IAAI,KAAK,EAAgB,EAAE,CAAC;QACxG,KAAK,MAAM,mBAAmB,IAAI,oBAAoB,CAAC,MAAM,EAAE,EAAE;YAC/D,IAAI,CAAC,mBAAmB,CAAC,+BAA+B,EAAE;gBACxD,MAAM,CAAC,UAAU,CAAC,IAAI,CAAC,GAAG,mBAAmB,CAAC,eAAe,CAAC,CAAC;aAChE;SACF;QAED,KAAK,MAAM,CAAC,qBAAqB,EAAE,uBAAuB,CAAC,IAAI,wBAAwB,EAAE;YACvF,IAAI,CAAC,uBAAuB,CAAC,+BAA+B,EAAE;gBAC5D,MAAM,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,CAAC;aACzD;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,gBAAgB,CAAC,UAAsB;QAClD,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,oBAAoB,EAAE,CAAC;QAC5D,MAAM,MAAM,GAAG,IAAI,KAAK,EAAc,CAAC;QACvC,KAAK,MAAM,CAAC,aAAa,EAAE,eAAe,CAAC,IAAI,iBAAiB,EAAE;YAChE,IAAI,eAAe,CAAC,UAAU,KAAK,UAAU,EAAE;gBAC7C,MAAM,CAAC,IAAI,CAAC,aAAa,CAAC,CAAC;aAC5B;SACF;QACD,OAAO,MAAM,CAAC;IAChB,CAAC;IAEM,KAAK,CAAC,iCAAiC;QAC5C,IAAI,CAAC,+BAA+B,KAAK,IAAI,CAAC,oCAAoC,EAAE,CAAC;QACrF,OAAO,IAAI,CAAC,+BAA+B,CAAC;IAC9C,CAAC;CACF;AAED,gBAAgB;AAChB,MAAM,UAAU,gBAAgB,CAAC,QAAqB;IACpD,OAAO,QAAQ,KAAK,IAAI;QACtB,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE;QAClG,CAAC,CAAC,EAAE,aAAa,EAAE,yBAAyB,EAAE,oBAAoB,EAAE,4BAA4B,EAAE,CAAC;AACvG,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { DEFINITION_CONTAINER_CLASS, SUB_CATEGORY_CLASS } from \"./ClassNameDefinitions.js\";\n\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { LimitingECSqlQueryExecutor } from \"@itwin/presentation-hierarchies\";\nimport type { InstanceKey } from \"@itwin/presentation-shared\";\n\ninterface DefinitionContainerInfo {\n  modelId: Id64String;\n  parentDefinitionContainerExists: boolean;\n  childCategories: CategoryInfo[];\n  childDefinitionContainers: Id64Array;\n}\n\ninterface CategoriesInfo {\n  childCategories: CategoryInfo[];\n  parentDefinitionContainerExists: boolean;\n}\n\ninterface CategoryInfo {\n  id: Id64String;\n  childCount: number;\n}\n\ninterface SubCategoryInfo {\n  categoryId: Id64String;\n}\n\n/** @internal */\nexport class CategoriesTreeIdsCache {\n  private _definitionContainersInfo: Promise<Map<Id64String, DefinitionContainerInfo>> | undefined;\n  private _modelsCategoriesInfo: Promise<Map<Id64String, CategoriesInfo>> | undefined;\n  private _subCategoriesInfo: Promise<Map<Id64String, SubCategoryInfo>> | undefined;\n  private _categoryClass: string;\n  private _categoryElementClass: string;\n  private _isDefinitionContainerSupported: Promise<boolean> | undefined;\n\n  constructor(\n    private _queryExecutor: LimitingECSqlQueryExecutor,\n    viewType: \"3d\" | \"2d\",\n  ) {\n    const { categoryClass, categoryElementClass } = getClassesByView(viewType);\n    this._categoryClass = categoryClass;\n    this._categoryElementClass = categoryElementClass;\n  }\n\n  private async *queryCategories(): AsyncIterableIterator<{\n    id: Id64String;\n    modelId: Id64String;\n    parentDefinitionContainerExists: boolean;\n    childCount: number;\n  }> {\n    const isDefinitionContainerSupported = await this.getIsDefinitionContainerSupported();\n    const categoriesQuery = `\n      SELECT\n        this.ECInstanceId id,\n        COUNT(sc.ECInstanceId) childCount,\n        this.Model.Id modelId,\n        ${\n          isDefinitionContainerSupported\n            ? `\n            IIF(this.Model.Id IN (SELECT dc.ECInstanceId FROM ${DEFINITION_CONTAINER_CLASS} dc),\n              true,\n              false\n            )`\n            : \"false\"\n        } parentDefinitionContainerExists\n      FROM\n        ${this._categoryClass} this\n        JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId\n        JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id\n      WHERE\n        NOT this.IsPrivate\n        AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n        AND EXISTS (SELECT 1 FROM ${this._categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n      GROUP BY this.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: categoriesQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId, parentDefinitionContainerExists: row.parentDefinitionContainerExists, childCount: row.childCount };\n    }\n  }\n\n  private async queryIsDefinitionContainersSupported(): Promise<boolean> {\n    const query = `\n      SELECT\n        1\n      FROM\n        ECDbMeta.ECSchemaDef s\n        JOIN ECDbMeta.ECClassDef c ON c.Schema.Id = s.ECInstanceId\n      WHERE\n        s.Name = 'BisCore'\n        AND c.Name = 'DefinitionContainer'\n    `;\n\n    for await (const _row of this._queryExecutor.createQueryReader({ ecsql: query })) {\n      return true;\n    }\n    return false;\n  }\n\n  private async *queryDefinitionContainers(): AsyncIterableIterator<{ id: Id64String; modelId: Id64String }> {\n    // DefinitionModel ECInstanceId will always be the same as modeled DefinitionContainer ECInstanceId, if this wasn't the case, we would need to do something like:\n    //  JOIN BisCore.DefinitionModel dm ON dm.ECInstanceId = ${modelIdAccessor}\n    //  JOIN BisCore.DefinitionModelBreaksDownDefinitionContainer dr ON dr.SourceECInstanceId = dm.ECInstanceId\n    //  JOIN BisCore.DefinitionContainer dc ON dc.ECInstanceId = dr.TargetECInstanceId\n    const DEFINITION_CONTAINERS_CTE = \"DefinitionContainers\";\n    const ctes = [\n      `\n        ${DEFINITION_CONTAINERS_CTE}(ECInstanceId, ModelId) AS (\n          SELECT\n            dc.ECInstanceId,\n            dc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINER_CLASS} dc\n          WHERE\n            dc.ECInstanceId IN (SELECT c.Model.Id FROM ${this._categoryClass} c WHERE NOT c.IsPrivate)\n            AND NOT dc.IsPrivate\n\n          UNION ALL\n\n          SELECT\n            pdc.ECInstanceId,\n            pdc.Model.Id\n          FROM\n            ${DEFINITION_CONTAINERS_CTE} cdc\n            JOIN ${DEFINITION_CONTAINER_CLASS} pdc ON pdc.ECInstanceId = cdc.ModelId\n          WHERE\n            NOT pdc.IsPrivate\n        )\n      `,\n    ];\n    const definitionsQuery = `\n      SELECT dc.ECInstanceId id, dc.ModelId modelId FROM ${DEFINITION_CONTAINERS_CTE} dc GROUP BY dc.ECInstanceId\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ctes, ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, modelId: row.modelId };\n    }\n  }\n\n  private async *queryVisibleSubCategories(categoriesInfo: Id64Array): AsyncIterableIterator<{ id: Id64String; parentId: Id64String }> {\n    const definitionsQuery = `\n      SELECT\n        sc.ECInstanceId id,\n        sc.Parent.Id categoryId\n      FROM\n        ${SUB_CATEGORY_CLASS} sc\n      WHERE\n        NOT sc.IsPrivate\n        AND sc.Parent.Id IN (${categoriesInfo.join(\",\")})\n    `;\n    for await (const row of this._queryExecutor.createQueryReader({ ecsql: definitionsQuery }, { rowFormat: \"ECSqlPropertyNames\", limit: \"unbounded\" })) {\n      yield { id: row.id, parentId: row.categoryId };\n    }\n  }\n\n  private async getModelsCategoriesInfo() {\n    this._modelsCategoriesInfo ??= (async () => {\n      const allModelsCategories = new Map<Id64String, CategoriesInfo>();\n      for await (const queriedCategory of this.queryCategories()) {\n        let modelCategories = allModelsCategories.get(queriedCategory.modelId);\n        if (modelCategories === undefined) {\n          modelCategories = { parentDefinitionContainerExists: queriedCategory.parentDefinitionContainerExists, childCategories: [] };\n          allModelsCategories.set(queriedCategory.modelId, modelCategories);\n        }\n        modelCategories.childCategories.push({ id: queriedCategory.id, childCount: queriedCategory.childCount });\n      }\n      return allModelsCategories;\n    })();\n    return this._modelsCategoriesInfo;\n  }\n\n  private async getSubCategoriesInfo() {\n    this._subCategoriesInfo ??= (async () => {\n      const allSubCategories = new Map<Id64String, SubCategoryInfo>();\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      const categoriesWithMoreThanOneSubCategory = new Array<Id64String>();\n      for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n        categoriesWithMoreThanOneSubCategory.push(\n          ...modelCategoriesInfo.childCategories.filter((categoryInfo) => categoryInfo.childCount > 1).map((categoryInfo) => categoryInfo.id),\n        );\n      }\n\n      if (categoriesWithMoreThanOneSubCategory.length === 0) {\n        return allSubCategories;\n      }\n\n      for await (const queriedSubCategory of this.queryVisibleSubCategories(categoriesWithMoreThanOneSubCategory)) {\n        allSubCategories.set(queriedSubCategory.id, { categoryId: queriedSubCategory.parentId });\n      }\n      return allSubCategories;\n    })();\n    return this._subCategoriesInfo;\n  }\n\n  private async getDefinitionContainersInfo() {\n    this._definitionContainersInfo ??= (async () => {\n      const definitionContainersInfo = new Map<Id64String, DefinitionContainerInfo>();\n      const [isDefinitionContainerSupported, modelsCategoriesInfo] = await Promise.all([\n        this.getIsDefinitionContainerSupported(),\n        this.getModelsCategoriesInfo(),\n      ]);\n      if (!isDefinitionContainerSupported || modelsCategoriesInfo.size === 0) {\n        return definitionContainersInfo;\n      }\n\n      for await (const queriedDefinitionContainer of this.queryDefinitionContainers()) {\n        const modelCategoriesInfo = modelsCategoriesInfo.get(queriedDefinitionContainer.id);\n\n        definitionContainersInfo.set(queriedDefinitionContainer.id, {\n          childCategories: modelCategoriesInfo?.childCategories ?? [],\n          modelId: queriedDefinitionContainer.modelId,\n          childDefinitionContainers: [],\n          parentDefinitionContainerExists: false,\n        });\n      }\n\n      for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n        const parentDefinitionContainer = definitionContainersInfo.get(definitionContainerInfo.modelId);\n        if (parentDefinitionContainer !== undefined) {\n          parentDefinitionContainer.childDefinitionContainers.push(definitionContainerId);\n          definitionContainerInfo.parentDefinitionContainerExists = true;\n        }\n      }\n\n      return definitionContainersInfo;\n    })();\n    return this._definitionContainersInfo;\n  }\n\n  public async getDirectChildDefinitionContainersAndCategories(\n    parentDefinitionContainerIds: Id64Array,\n  ): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n\n    parentDefinitionContainerIds.forEach((parentDefinitionContainerId) => {\n      const parentDefinitionContainerInfo = definitionContainersInfo.get(parentDefinitionContainerId);\n      if (parentDefinitionContainerInfo !== undefined) {\n        result.definitionContainers.push(...parentDefinitionContainerInfo.childDefinitionContainers);\n        result.categories.push(...parentDefinitionContainerInfo.childCategories);\n      }\n    });\n    return result;\n  }\n\n  public async getAllContainedCategories(definitionContainerIds: Id64Array): Promise<Id64Array> {\n    const result = new Array<Id64String>();\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const indirectCategories = await Promise.all(\n      definitionContainerIds.map(async (definitionContainerId) => {\n        const definitionContainerInfo = definitionContainersInfo.get(definitionContainerId);\n        if (definitionContainerInfo === undefined) {\n          return [];\n        }\n        result.push(...definitionContainerInfo.childCategories.map((category) => category.id));\n        return this.getAllContainedCategories(definitionContainerInfo.childDefinitionContainers);\n      }),\n    );\n    for (const categories of indirectCategories) {\n      result.push(...categories);\n    }\n\n    return result;\n  }\n\n  public async getInstanceKeyPaths(\n    props: { categoryId: Id64String } | { definitionContainerId: Id64String } | { subCategoryId: Id64String },\n  ): Promise<InstanceKey[]> {\n    if (\"subCategoryId\" in props) {\n      const subCategoriesInfo = await this.getSubCategoriesInfo();\n      const subCategoryInfo = subCategoriesInfo.get(props.subCategoryId);\n      if (subCategoryInfo === undefined) {\n        return [];\n      }\n      return [...(await this.getInstanceKeyPaths({ categoryId: subCategoryInfo.categoryId })), { id: props.subCategoryId, className: SUB_CATEGORY_CLASS }];\n    }\n\n    if (\"categoryId\" in props) {\n      const modelsCategoriesInfo = await this.getModelsCategoriesInfo();\n      for (const [modelId, modelCategoriesInfo] of modelsCategoriesInfo) {\n        if (modelCategoriesInfo.childCategories.find((childCategory) => childCategory.id === props.categoryId)) {\n          if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n            return [{ id: props.categoryId, className: this._categoryClass }];\n          }\n\n          return [...(await this.getInstanceKeyPaths({ definitionContainerId: modelId })), { id: props.categoryId, className: this._categoryClass }];\n        }\n      }\n      return [];\n    }\n\n    const definitionContainersInfo = await this.getDefinitionContainersInfo();\n    const definitionContainerInfo = definitionContainersInfo.get(props.definitionContainerId);\n    if (definitionContainerInfo === undefined) {\n      return [];\n    }\n\n    if (!definitionContainerInfo.parentDefinitionContainerExists) {\n      return [{ id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS }];\n    }\n\n    return [\n      ...(await this.getInstanceKeyPaths({ definitionContainerId: definitionContainerInfo.modelId })),\n      { id: props.definitionContainerId, className: DEFINITION_CONTAINER_CLASS },\n    ];\n  }\n\n  public async getAllDefinitionContainersAndCategories(): Promise<{ categories: Id64Array; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: [...definitionContainersInfo.keys()], categories: new Array<Id64String>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      result.categories.push(...modelCategoriesInfo.childCategories.map((childCategory) => childCategory.id));\n    }\n\n    return result;\n  }\n\n  public async getRootDefinitionContainersAndCategories(): Promise<{ categories: CategoryInfo[]; definitionContainers: Id64Array }> {\n    const [modelsCategoriesInfo, definitionContainersInfo] = await Promise.all([this.getModelsCategoriesInfo(), this.getDefinitionContainersInfo()]);\n    const result = { definitionContainers: new Array<Id64String>(), categories: new Array<CategoryInfo>() };\n    for (const modelCategoriesInfo of modelsCategoriesInfo.values()) {\n      if (!modelCategoriesInfo.parentDefinitionContainerExists) {\n        result.categories.push(...modelCategoriesInfo.childCategories);\n      }\n    }\n\n    for (const [definitionContainerId, definitionContainerInfo] of definitionContainersInfo) {\n      if (!definitionContainerInfo.parentDefinitionContainerExists) {\n        result.definitionContainers.push(definitionContainerId);\n      }\n    }\n    return result;\n  }\n\n  public async getSubCategories(categoryId: Id64String): Promise<Id64Array> {\n    const subCategoriesInfo = await this.getSubCategoriesInfo();\n    const result = new Array<Id64String>();\n    for (const [subCategoryId, subCategoryInfo] of subCategoriesInfo) {\n      if (subCategoryInfo.categoryId === categoryId) {\n        result.push(subCategoryId);\n      }\n    }\n    return result;\n  }\n\n  public async getIsDefinitionContainerSupported(): Promise<boolean> {\n    this._isDefinitionContainerSupported ??= this.queryIsDefinitionContainersSupported();\n    return this._isDefinitionContainerSupported;\n  }\n}\n\n/** @internal */\nexport function getClassesByView(viewType: \"2d\" | \"3d\") {\n  return viewType === \"2d\"\n    ? { categoryClass: \"BisCore.DrawingCategory\", categoryElementClass: \"BisCore.GeometricElement2d\" }\n    : { categoryClass: \"BisCore.SpatialCategory\", categoryElementClass: \"BisCore.GeometricElement3d\" };\n}\n"]}

package/lib/esm/tree-widget-react/components/trees/categories-tree/internal/CategoriesTreeNode.d.ts

@@ -0,0 +1,26 @@
+import type { HierarchyNodeKey } from "@itwin/presentation-hierarchies";
+interface CategoriesTreeNode {
+    key: HierarchyNodeKey;
+    extendedData?: {
+        [id: string]: any;
+    };
+}
+/**
+ * @internal
+ */
+export declare namespace CategoriesTreeNode {
+    /**
+     * Determines if node represents a definition container.
+     */
+    const isDefinitionContainerNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a category.
+     */
+    const isCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+    /**
+     * Determines if node represents a sub-category.
+     */
+    const isSubCategoryNode: (node: Pick<CategoriesTreeNode, "extendedData">) => boolean | undefined;
+}
+export {};
+//# sourceMappingURL=CategoriesTreeNode.d.ts.map