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

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

@@ -2,81 +2,154 @@
  * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
  * See LICENSE.md in the project root for license terms and full copyright notice.
  *--------------------------------------------------------------------------------------------*/
+import { defer, EMPTY, from, lastValueFrom, map, mergeMap, toArray } from "rxjs";
 import { createNodesQueryClauseFactory, createPredicateBasedHierarchyDefinition } from "@itwin/presentation-hierarchies";
 import { createBisInstanceLabelSelectClauseFactory, ECSql } from "@itwin/presentation-shared";
 import { FilterLimitExceededError } from "../common/TreeErrors.js";
+import { getClassesByView } from "./internal/CategoriesTreeIdsCache.js";
+import { DEFINITION_CONTAINER_CLASS, DEFINITION_ELEMENT_CLASS, SUB_CATEGORY_CLASS } from "./internal/ClassNameDefinitions.js";
 const MAX_FILTERING_INSTANCE_KEY_COUNT = 100;
 export class CategoriesTreeDefinition {
     constructor(props) {
-        this._impl = createPredicateBasedHierarchyDefinition({
-            classHierarchyInspector: props.imodelAccess,
-            hierarchy: {
-                rootNodes: async (requestProps) => this.createRootHierarchyLevelDefinition({ ...requestProps, viewType: props.viewType }),
-                childNodes: [
-                    {
-                        parentInstancesNodePredicate: "BisCore.Category",
-                        definitions: async (requestProps) => this.createSubcategoryQuery(requestProps),
-                    },
-                ],
-            },
-        });
+        this._iModelAccess = props.imodelAccess;
+        this._viewType = props.viewType;
+        this._idsCache = props.idsCache;
         this._nodeLabelSelectClauseFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });
         this._selectQueryFactory = createNodesQueryClauseFactory({
             imodelAccess: props.imodelAccess,
             instanceLabelSelectClauseFactory: this._nodeLabelSelectClauseFactory,
         });
     }
+    async getHierarchyDefinition() {
+        this._impl ??= (async () => {
+            const isDefinitionContainerSupported = await this._idsCache.getIsDefinitionContainerSupported();
+            return createPredicateBasedHierarchyDefinition({
+                classHierarchyInspector: this._iModelAccess,
+                hierarchy: {
+                    rootNodes: async (requestProps) => this.createDefinitionContainersAndCategoriesQuery({ ...requestProps, viewType: this._viewType }),
+                    childNodes: [
+                        {
+                            parentInstancesNodePredicate: "BisCore.Category",
+                            definitions: async (requestProps) => this.createSubcategoryQuery(requestProps),
+                        },
+                        ...(isDefinitionContainerSupported
+                            ? [
+                                {
+                                    parentInstancesNodePredicate: DEFINITION_CONTAINER_CLASS,
+                                    definitions: async (requestProps) => this.createDefinitionContainersAndCategoriesQuery({
+                                        ...requestProps,
+                                        viewType: this._viewType,
+                                    }),
+                                },
+                            ]
+                            : []),
+                    ],
+                },
+            });
+        })();
+        return this._impl;
+    }
     async defineHierarchyLevel(props) {
-        return this._impl.defineHierarchyLevel(props);
+        return (await this.getHierarchyDefinition()).defineHierarchyLevel(props);
     }
-    async createRootHierarchyLevelDefinition(props) {
-        const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);
-        const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({
-            filter: props.instanceFilter,
-            contentClass: { fullName: categoryClass, alias: "this" },
+    async createDefinitionContainersAndCategoriesQuery(props) {
+        const { parentNodeInstanceIds, instanceFilter, viewType } = props;
+        const { definitionContainers, categories } = parentNodeInstanceIds === undefined
+            ? await this._idsCache.getRootDefinitionContainersAndCategories()
+            : await this._idsCache.getDirectChildDefinitionContainersAndCategories(parentNodeInstanceIds);
+        if (categories.length === 0 && definitionContainers.length === 0) {
+            return [];
+        }
+        const categoriesWithSingleChild = new Array();
+        const categoriesWithMultipleChildren = new Array();
+        categories.forEach((category) => {
+            if (category.childCount > 1) {
+                categoriesWithMultipleChildren.push(category.id);
+            }
+            else {
+                categoriesWithSingleChild.push(category.id);
+            }
         });
-        return [
-            {
-                fullClassName: categoryClass,
-                query: {
-                    ecsql: `
-            SELECT
+        const dataToDetermineHasChildren = categoriesWithSingleChild.length > categoriesWithMultipleChildren.length
+            ? { ids: categoriesWithMultipleChildren, ifTrue: 1, ifFalse: 0 }
+            : { ids: categoriesWithSingleChild, ifTrue: 0, ifFalse: 1 };
+        const { categoryClass } = getClassesByView(viewType);
+        const [categoriesInstanceFilterClauses, definitionContainersInstanceFilterClauses] = await Promise.all([categoryClass, ...(definitionContainers.length > 0 ? [DEFINITION_CONTAINER_CLASS] : [])].map(async (className) => this._selectQueryFactory.createFilterClauses({
+            filter: instanceFilter,
+            contentClass: { fullName: className, alias: "this" },
+        })));
+        const definitionContainersQuery = definitionContainers.length > 0
+            ? `
+          SELECT
+            ${await this._selectQueryFactory.createSelectClause({
+                ecClassId: { selector: ECSql.createRawPropertyValueSelector("this", "ECClassId") },
+                ecInstanceId: { selector: "this.ECInstanceId" },
+                nodeLabel: {
+                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                        classAlias: "this",
+                        className: DEFINITION_CONTAINER_CLASS,
+                    }),
+                },
+                extendedData: {
+                    isDefinitionContainer: true,
+                    imageId: "icon-definition-container",
+                },
+                hasChildren: true,
+                supportsFiltering: true,
+            })}
+          FROM
+            ${definitionContainersInstanceFilterClauses.from} this
+            ${definitionContainersInstanceFilterClauses.joins}
+          WHERE
+            this.ECInstanceId IN (${definitionContainers.join(", ")})
+            ${definitionContainersInstanceFilterClauses.where ? `AND ${definitionContainersInstanceFilterClauses.where}` : ""}
+        `
+            : undefined;
+        const categoriesQuery = categories.length > 0
+            ? `
+          SELECT
               ${await this._selectQueryFactory.createSelectClause({
-                        ecClassId: { selector: ECSql.createRawPropertyValueSelector("this", "ECClassId") },
-                        ecInstanceId: { selector: "this.ECInstanceId" },
-                        nodeLabel: {
-                            selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
-                                classAlias: "this",
-                                className: categoryClass,
-                            }),
-                        },
+                ecClassId: { selector: ECSql.createRawPropertyValueSelector("this", "ECClassId") },
+                ecInstanceId: { selector: "this.ECInstanceId" },
+                nodeLabel: {
+                    selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
+                        classAlias: "this",
+                        className: categoryClass,
+                    }),
+                },
+                ...(dataToDetermineHasChildren.ids.length > 0
+                    ? {
                         hasChildren: {
                             selector: `
-                    IFNULL((
-                      SELECT 1
-                      FROM (
-                        SELECT COUNT(1) AS ChildCount
-                        FROM BisCore.SubCategory sc
-                        WHERE sc.Parent.Id = this.ECInstanceId
-                      )
-                      WHERE ChildCount > 1
-                    ), 0)
-                  `,
+                            IIF(this.ECInstanceId IN (${dataToDetermineHasChildren.ids.join(",")}),
+                              ${dataToDetermineHasChildren.ifTrue},
+                              ${dataToDetermineHasChildren.ifFalse}
+                            )
+                          `,
                         },
-                        extendedData: {
-                            description: { selector: "this.Description" },
-                        },
-                        supportsFiltering: true,
-                    })}
-            FROM ${instanceFilterClauses.from} this
-            ${instanceFilterClauses.joins}
-            JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id
+                    }
+                    : { hasChildren: !!dataToDetermineHasChildren.ifFalse }),
+                extendedData: {
+                    description: { selector: "this.Description" },
+                    isCategory: true,
+                    imageId: "icon-layers",
+                },
+                supportsFiltering: true,
+            })}
+            FROM
+              ${categoriesInstanceFilterClauses.from} this
+              ${categoriesInstanceFilterClauses.joins}
             WHERE
-              NOT this.IsPrivate
-              AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))
-              AND EXISTS (SELECT 1 FROM ${categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)
-              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : ""}
-          `,
+              this.ECInstanceId IN (${categories.map((category) => category.id).join(", ")})
+              ${categoriesInstanceFilterClauses.where ? `AND ${categoriesInstanceFilterClauses.where}` : ""}
+        `
+            : undefined;
+        const queries = [categoriesQuery, definitionContainersQuery].filter((query) => query !== undefined);
+        return [
+            {
+                fullClassName: DEFINITION_ELEMENT_CLASS,
+                query: {
+                    ecsql: queries.join(" UNION ALL "),
                 },
             },
         ];
@@ -84,11 +157,11 @@ export class CategoriesTreeDefinition {
     async createSubcategoryQuery({ parentNodeInstanceIds: elementIds, instanceFilter, }) {
         const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({
             filter: instanceFilter,
-            contentClass: { fullName: "BisCore.SubCategory", alias: "this" },
+            contentClass: { fullName: SUB_CATEGORY_CLASS, alias: "this" },
         });
         return [
             {
-                fullClassName: "BisCore.SubCategory",
+                fullClassName: SUB_CATEGORY_CLASS,
                 query: {
                     ecsql: `
             SELECT
@@ -98,11 +171,13 @@ export class CategoriesTreeDefinition {
                         nodeLabel: {
                             selector: await this._nodeLabelSelectClauseFactory.createSelectClause({
                                 classAlias: "this",
-                                className: "BisCore.SubCategory",
+                                className: SUB_CATEGORY_CLASS,
                             }),
                         },
                         extendedData: {
                             categoryId: { selector: "printf('0x%x', this.Parent.Id)" },
+                            isSubCategory: true,
+                            imageId: "icon-layers-isolate",
                         },
                         supportsFiltering: false,
                     })}
@@ -119,85 +194,127 @@ export class CategoriesTreeDefinition {
     }
     static async createInstanceKeyPaths(props) {
         const labelsFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });
-        return createInstanceKeyPathsFromInstanceLabel({ ...props, labelsFactory });
+        return createInstanceKeyPathsFromInstanceLabel({ ...props, labelsFactory, cache: props.idsCache });
     }
 }
-function getClassesByView(viewType) {
-    return viewType === "2d"
-        ? { categoryClass: "BisCore.DrawingCategory", categoryElementClass: "BisCore:GeometricElement2d" }
-        : { categoryClass: "BisCore.SpatialCategory", categoryElementClass: "BisCore:GeometricElement3d" };
-}
 async function createInstanceKeyPathsFromInstanceLabel(props) {
-    const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);
+    const { definitionContainers, categories } = await props.cache.getAllDefinitionContainersAndCategories();
+    if (categories.length === 0) {
+        return [];
+    }
+    const { categoryClass } = getClassesByView(props.viewType);
     const adjustedLabel = props.label.replace(/[%_\\]/g, "\\$&");
-    const reader = props.imodelAccess.createQueryReader({
-        ctes: [
-            `RootCategoriesWithLabels(ClassName, ECInstanceId, ChildCount, DisplayLabel) as (
-          SELECT
-            ec_classname(this.ECClassId, 's.c'),
-            this.ECInstanceId,
-            COUNT(sc.ECInstanceId),
-            ${await props.labelsFactory.createSelectClause({
-                classAlias: "this",
-                className: categoryClass,
-            })}
-          FROM ${categoryClass} this
-          JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id
-          JOIN BisCore.SubCategory sc ON sc.Parent.Id = this.ECInstanceId
-          WHERE
-            NOT this.IsPrivate
-            AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))
-            AND EXISTS (SELECT 1 FROM ${categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)
-          GROUP BY this.ECInstanceId
-        )`,
-            `SubCategoriesWithLabels(ClassName, ECInstanceId, ParentId, DisplayLabel) as (
-          SELECT
-            ec_classname(this.ECClassId, 's.c'),
-            this.ECInstanceId,
-            this.Parent.Id,
-            ${await props.labelsFactory.createSelectClause({
-                classAlias: "this",
-                className: "BisCore.SubCategory",
-            })}
-          FROM BisCore.SubCategory this
-          WHERE NOT this.IsPrivate
-        )`,
-        ],
-        ecsql: `
-      SELECT * FROM (
-        SELECT
-          c.ClassName AS CategoryClass,
-          c.ECInstanceId AS CategoryId,
-          sc.ClassName AS SubcategoryClass,
-          sc.ECInstanceId AS SubcategoryId
-        FROM RootCategoriesWithLabels c
-        JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId
-        WHERE c.ChildCount > 1 AND sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
-        UNION ALL
-        SELECT
-          c.ClassName AS CategoryClass,
-          c.ECInstanceId AS CategoryId,
-          CAST(NULL AS TEXT) AS SubcategoryClass,
-          CAST(NULL AS TEXT) AS SubcategoryId
-        FROM RootCategoriesWithLabels c
-        WHERE c.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+    const CATEGORIES_WITH_LABELS_CTE = "CategoriesWithLabels";
+    const SUBCATEGORIES_WITH_LABELS_CTE = "SubCategoriesWithLabels";
+    const DEFINITION_CONTAINERS_WITH_LABELS_CTE = "DefinitionContainersWithLabels";
+    const [categoryLabelSelectClause, subCategoryLabelSelectClause, definitionContainerLabelSelectClause] = await Promise.all([categoryClass, SUB_CATEGORY_CLASS, ...(definitionContainers.length > 0 ? [DEFINITION_CONTAINER_CLASS] : [])].map(async (className) => props.labelsFactory.createSelectClause({ classAlias: "this", className })));
+    return lastValueFrom(defer(() => {
+        const ctes = [
+            `${CATEGORIES_WITH_LABELS_CTE}(ClassName, ECInstanceId, ChildCount, DisplayLabel) AS (
+            SELECT
+              'c',
+              this.ECInstanceId,
+              COUNT(sc.ECInstanceId),
+              ${categoryLabelSelectClause}
+            FROM
+              ${categoryClass} this
+              JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId
+            WHERE
+              this.ECInstanceId IN (${categories.join(", ")})
+              GROUP BY this.ECInstanceId
+          )`,
+            `${SUBCATEGORIES_WITH_LABELS_CTE}(ClassName, ECInstanceId, ParentId, DisplayLabel) AS (
+            SELECT
+              'sc',
+              this.ECInstanceId,
+              this.Parent.Id,
+              ${subCategoryLabelSelectClause}
+            FROM
+              ${SUB_CATEGORY_CLASS} this
+            WHERE
+              NOT this.IsPrivate
+              AND this.Parent.Id IN (${categories.join(", ")})
+          )`,
+            ...(definitionContainers.length > 0
+                ? [
+                    `${DEFINITION_CONTAINERS_WITH_LABELS_CTE}(ClassName, ECInstanceId, DisplayLabel) AS (
+                SELECT
+                  'dc',
+                  this.ECInstanceId,
+                  ${definitionContainerLabelSelectClause}
+                FROM
+                  ${DEFINITION_CONTAINER_CLASS} this
+                WHERE
+                  this.ECInstanceId IN (${definitionContainers.join(", ")})
+              )`,
+                ]
+                : []),
+        ];
+        const ecsql = `
+        SELECT * FROM (
+            SELECT
+              sc.ClassName AS ClassName,
+              sc.ECInstanceId AS ECInstanceId
+            FROM
+              ${CATEGORIES_WITH_LABELS_CTE} c
+              JOIN ${SUBCATEGORIES_WITH_LABELS_CTE} sc ON sc.ParentId = c.ECInstanceId
+            WHERE
+              c.ChildCount > 1
+              AND sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+
+            UNION ALL
+
+            SELECT
+              c.ClassName AS ClassName,
+              c.ECInstanceId AS ECInstanceId
+            FROM
+              ${CATEGORIES_WITH_LABELS_CTE} c
+            WHERE
+              c.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+
+            ${definitionContainers.length > 0
+            ? `
+                  UNION ALL
+                  SELECT
+                    dc.ClassName AS ClassName,
+                    dc.ECInstanceId AS ECInstanceId
+                  FROM
+                    ${DEFINITION_CONTAINERS_WITH_LABELS_CTE} dc
+                  WHERE
+                    dc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+                `
+            : ""}
         )
-        LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}
-      `,
-        bindings: [
+        ${props.limit === undefined ? `LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}` : props.limit !== "unbounded" ? `LIMIT ${props.limit}` : ""}
+      `;
+        const bindings = [
             { type: "string", value: adjustedLabel },
             { type: "string", value: adjustedLabel },
-        ],
-    }, { restartToken: "tree-widget/categories-tree/filter-by-label-query" });
-    const paths = [];
-    for await (const row of reader) {
-        const path = { path: [{ className: row.CategoryClass, id: row.CategoryId }], options: { autoExpand: true } };
-        row.SubcategoryId && path.path.push({ className: row.SubcategoryClass, id: row.SubcategoryId });
-        paths.push(path);
+            ...(definitionContainers.length > 0 ? [{ type: "string", value: adjustedLabel }] : []),
+        ];
+        return props.imodelAccess.createQueryReader({ ctes, ecsql, bindings }, { restartToken: "tree-widget/categories-tree/filter-by-label-query", limit: props.limit });
+    }).pipe(map((row) => ({
+        className: row.ClassName === "c" ? categoryClass : row.ClassName === "sc" ? SUB_CATEGORY_CLASS : DEFINITION_CONTAINER_CLASS,
+        id: row.ECInstanceId,
+    })), toArray(), mergeMap((targetItems) => createInstanceKeyPathsFromTargetItems({ ...props, targetItems })), toArray()));
+}
+function createInstanceKeyPathsFromTargetItems(props) {
+    const { limit, targetItems, viewType, idsCache } = props;
+    if (limit !== "unbounded" && targetItems.length > (limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT)) {
+        throw new FilterLimitExceededError(limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT);
     }
-    if (paths.length > MAX_FILTERING_INSTANCE_KEY_COUNT) {
-        throw new FilterLimitExceededError(MAX_FILTERING_INSTANCE_KEY_COUNT);
+    if (targetItems.length === 0) {
+        return EMPTY;
     }
-    return paths;
+    const { categoryClass } = getClassesByView(viewType);
+    return from(targetItems).pipe(mergeMap(async (targetItem) => {
+        if (targetItem.className === SUB_CATEGORY_CLASS) {
+            return { path: await idsCache.getInstanceKeyPaths({ subCategoryId: targetItem.id }), options: { autoExpand: true } };
+        }
+        if (targetItem.className === categoryClass) {
+            return { path: await idsCache.getInstanceKeyPaths({ categoryId: targetItem.id }), options: { autoExpand: true } };
+        }
+        return { path: await idsCache.getInstanceKeyPaths({ definitionContainerId: targetItem.id }), options: { autoExpand: true } };
+    }));
 }
 //# sourceMappingURL=CategoriesTreeDefinition.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeDefinition.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/CategoriesTreeDefinition.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,6BAA6B,EAAE,uCAAuC,EAAE,MAAM,iCAAiC,CAAC;AACzH,OAAO,EAAE,yCAAyC,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AAcnE,MAAM,gCAAgC,GAAG,GAAG,CAAC;AAa7C,MAAM,OAAO,wBAAwB;IAKnC,YAAmB,KAAoC;QACrD,IAAI,CAAC,KAAK,GAAG,uCAAuC,CAAC;YACnD,uBAAuB,EAAE,KAAK,CAAC,YAAY;YAC3C,SAAS,EAAE;gBACT,SAAS,EAAE,KAAK,EAAE,YAAY,EAAE,EAAE,CAAC,IAAI,CAAC,kCAAkC,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC;gBACzH,UAAU,EAAE;oBACV;wBACE,4BAA4B,EAAE,kBAAkB;wBAChD,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC;qBAC3H;iBACF;aACF;SACF,CAAC,CAAC;QACH,IAAI,CAAC,6BAA6B,GAAG,yCAAyC,CAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QAChI,IAAI,CAAC,mBAAmB,GAAG,6BAA6B,CAAC;YACvD,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,gCAAgC,EAAE,IAAI,CAAC,6BAA6B;SACrE,CAAC,CAAC;IACL,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,KAAgC;QAChE,OAAO,IAAI,CAAC,KAAK,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAChD,CAAC;IAEO,KAAK,CAAC,kCAAkC,CAAC,KAAgE;QAC/G,MAAM,EAAE,aAAa,EAAE,oBAAoB,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;QACjF,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC/E,MAAM,EAAE,KAAK,CAAC,cAAc;YAC5B,YAAY,EAAE,EAAE,QAAQ,EAAE,aAAa,EAAE,KAAK,EAAE,MAAM,EAAE;SACzD,CAAC,CAAC;QACH,OAAO;YACL;gBACE,aAAa,EAAE,aAAa;gBAC5B,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE;wBAClF,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,aAAa;6BACzB,CAAC;yBACH;wBACD,WAAW,EAAE;4BACX,QAAQ,EAAE;;;;;;;;;;mBAUT;yBACF;wBACD,YAAY,EAAE;4BACZ,WAAW,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE;yBAC9C;wBACD,iBAAiB,EAAE,IAAI;qBACxB,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;;;;0CAKC,oBAAoB;gBAC9C,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;iBACF;aACF;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAAC,EACnC,qBAAqB,EAAE,UAAU,EACjC,cAAc,GAC6B;QAC3C,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC/E,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,qBAAqB,EAAE,KAAK,EAAE,MAAM,EAAE;SACjE,CAAC,CAAC;QACH,OAAO;YACL;gBACE,aAAa,EAAE,qBAAqB;gBACpC,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,qBAAqB;6BACjC,CAAC;yBACH;wBACD,YAAY,EAAE;4BACZ,UAAU,EAAE,EAAE,QAAQ,EAAE,gCAAgC,EAAE;yBAC3D;wBACD,iBAAiB,EAAE,KAAK;qBACzB,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;0DAEiB,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC7E,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;oBACD,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC9D;aACF;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,KAA2D;QACpG,MAAM,aAAa,GAAG,yCAAyC,CAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACjH,OAAO,uCAAuC,CAAC,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IAC9E,CAAC;CACF;AAED,SAAS,gBAAgB,CAAC,QAAqB;IAC7C,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;AAED,KAAK,UAAU,uCAAuC,CACpD,KAAkH;IAElH,MAAM,EAAE,aAAa,EAAE,oBAAoB,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IACjF,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAC7D,MAAM,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,iBAAiB,CACjD;QACE,IAAI,EAAE;YACJ;;;;;cAKM,MAAM,KAAK,CAAC,aAAa,CAAC,kBAAkB,CAAC;gBAC7C,UAAU,EAAE,MAAM;gBAClB,SAAS,EAAE,aAAa;aACzB,CAAC;iBACG,aAAa;;;;;;wCAMU,oBAAoB;;UAElD;YACF;;;;;cAKM,MAAM,KAAK,CAAC,aAAa,CAAC,kBAAkB,CAAC;gBAC7C,UAAU,EAAE,MAAM;gBAClB,SAAS,EAAE,qBAAqB;aACjC,CAAC;;;UAGJ;SACH;QACD,KAAK,EAAE;;;;;;;;;;;;;;;;;;;gBAmBG,gCAAgC,GAAG,CAAC;OAC7C;QACD,QAAQ,EAAE;YACR,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;YACxC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,aAAa,EAAE;SACzC;KACF,EACD,EAAE,YAAY,EAAE,mDAAmD,EAAE,CACtE,CAAC;IACF,MAAM,KAAK,GAA6B,EAAE,CAAC;IAC3C,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE;QAC9B,MAAM,IAAI,GAAG,EAAE,IAAI,EAAE,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,aAAa,EAAE,EAAE,EAAE,GAAG,CAAC,UAAU,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC;QAC7G,GAAG,CAAC,aAAa,IAAI,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,GAAG,CAAC,gBAAgB,EAAE,EAAE,EAAE,GAAG,CAAC,aAAa,EAAE,CAAC,CAAC;QAChG,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;KAClB;IACD,IAAI,KAAK,CAAC,MAAM,GAAG,gCAAgC,EAAE;QACnD,MAAM,IAAI,wBAAwB,CAAC,gCAAgC,CAAC,CAAC;KACtE;IACD,OAAO,KAAK,CAAC;AACf,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 { createNodesQueryClauseFactory, createPredicateBasedHierarchyDefinition } from \"@itwin/presentation-hierarchies\";\nimport { createBisInstanceLabelSelectClauseFactory, ECSql } from \"@itwin/presentation-shared\";\nimport { FilterLimitExceededError } from \"../common/TreeErrors.js\";\n\nimport type { ECClassHierarchyInspector, ECSchemaProvider, IInstanceLabelSelectClauseFactory } from \"@itwin/presentation-shared\";\nimport type {\n  DefineHierarchyLevelProps,\n  DefineInstanceNodeChildHierarchyLevelProps,\n  DefineRootHierarchyLevelProps,\n  HierarchyDefinition,\n  HierarchyFilteringPath,\n  HierarchyLevelDefinition,\n  LimitingECSqlQueryExecutor,\n  NodesQueryClauseFactory,\n} from \"@itwin/presentation-hierarchies\";\n\nconst MAX_FILTERING_INSTANCE_KEY_COUNT = 100;\n\ninterface CategoriesTreeDefinitionProps {\n  imodelAccess: ECSchemaProvider & ECClassHierarchyInspector;\n  viewType: \"2d\" | \"3d\";\n}\n\ninterface CategoriesTreeInstanceKeyPathsFromInstanceLabelProps {\n  imodelAccess: ECClassHierarchyInspector & LimitingECSqlQueryExecutor;\n  label: string;\n  viewType: \"2d\" | \"3d\";\n}\n\nexport class CategoriesTreeDefinition implements HierarchyDefinition {\n  private _impl: HierarchyDefinition;\n  private _selectQueryFactory: NodesQueryClauseFactory;\n  private _nodeLabelSelectClauseFactory: IInstanceLabelSelectClauseFactory;\n\n  public constructor(props: CategoriesTreeDefinitionProps) {\n    this._impl = createPredicateBasedHierarchyDefinition({\n      classHierarchyInspector: props.imodelAccess,\n      hierarchy: {\n        rootNodes: async (requestProps) => this.createRootHierarchyLevelDefinition({ ...requestProps, viewType: props.viewType }),\n        childNodes: [\n          {\n            parentInstancesNodePredicate: \"BisCore.Category\",\n            definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createSubcategoryQuery(requestProps),\n          },\n        ],\n      },\n    });\n    this._nodeLabelSelectClauseFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });\n    this._selectQueryFactory = createNodesQueryClauseFactory({\n      imodelAccess: props.imodelAccess,\n      instanceLabelSelectClauseFactory: this._nodeLabelSelectClauseFactory,\n    });\n  }\n\n  public async defineHierarchyLevel(props: DefineHierarchyLevelProps) {\n    return this._impl.defineHierarchyLevel(props);\n  }\n\n  private async createRootHierarchyLevelDefinition(props: DefineRootHierarchyLevelProps & { viewType: \"2d\" | \"3d\" }): Promise<HierarchyLevelDefinition> {\n    const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);\n    const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: props.instanceFilter,\n      contentClass: { fullName: categoryClass, alias: \"this\" },\n    });\n    return [\n      {\n        fullClassName: categoryClass,\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: ECSql.createRawPropertyValueSelector(\"this\", \"ECClassId\") },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: categoryClass,\n                  }),\n                },\n                hasChildren: {\n                  selector: `\n                    IFNULL((\n                      SELECT 1\n                      FROM (\n                        SELECT COUNT(1) AS ChildCount\n                        FROM BisCore.SubCategory sc\n                        WHERE sc.Parent.Id = this.ECInstanceId\n                      )\n                      WHERE ChildCount > 1\n                    ), 0)\n                  `,\n                },\n                extendedData: {\n                  description: { selector: \"this.Description\" },\n                },\n                supportsFiltering: true,\n              })}\n            FROM ${instanceFilterClauses.from} this\n            ${instanceFilterClauses.joins}\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 ${categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n          `,\n        },\n      },\n    ];\n  }\n\n  private async createSubcategoryQuery({\n    parentNodeInstanceIds: elementIds,\n    instanceFilter,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: instanceFilter,\n      contentClass: { fullName: \"BisCore.SubCategory\", alias: \"this\" },\n    });\n    return [\n      {\n        fullClassName: \"BisCore.SubCategory\",\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: \"BisCore.SubCategory\",\n                  }),\n                },\n                extendedData: {\n                  categoryId: { selector: \"printf('0x%x', this.Parent.Id)\" },\n                },\n                supportsFiltering: false,\n              })}\n            FROM ${instanceFilterClauses.from} this\n            ${instanceFilterClauses.joins}\n            WHERE\n              NOT this.IsPrivate AND this.Parent.Id IN (${elementIds.map(() => \"?\").join(\",\")})\n              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n          `,\n          bindings: elementIds.map((id) => ({ type: \"id\", value: id })),\n        },\n      },\n    ];\n  }\n\n  public static async createInstanceKeyPaths(props: CategoriesTreeInstanceKeyPathsFromInstanceLabelProps) {\n    const labelsFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });\n    return createInstanceKeyPathsFromInstanceLabel({ ...props, labelsFactory });\n  }\n}\n\nfunction getClassesByView(viewType: \"2d\" | \"3d\") {\n  return viewType === \"2d\"\n    ? { categoryClass: \"BisCore.DrawingCategory\", categoryElementClass: \"BisCore:GeometricElement2d\" }\n    : { categoryClass: \"BisCore.SpatialCategory\", categoryElementClass: \"BisCore:GeometricElement3d\" };\n}\n\nasync function createInstanceKeyPathsFromInstanceLabel(\n  props: CategoriesTreeInstanceKeyPathsFromInstanceLabelProps & { labelsFactory: IInstanceLabelSelectClauseFactory },\n) {\n  const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);\n  const adjustedLabel = props.label.replace(/[%_\\\\]/g, \"\\\\$&\");\n  const reader = props.imodelAccess.createQueryReader(\n    {\n      ctes: [\n        `RootCategoriesWithLabels(ClassName, ECInstanceId, ChildCount, DisplayLabel) as (\n          SELECT\n            ec_classname(this.ECClassId, 's.c'),\n            this.ECInstanceId,\n            COUNT(sc.ECInstanceId),\n            ${await props.labelsFactory.createSelectClause({\n              classAlias: \"this\",\n              className: categoryClass,\n            })}\n          FROM ${categoryClass} this\n          JOIN BisCore.Model m ON m.ECInstanceId = this.Model.Id\n          JOIN BisCore.SubCategory sc ON sc.Parent.Id = this.ECInstanceId\n          WHERE\n            NOT this.IsPrivate\n            AND (NOT m.IsPrivate OR m.ECClassId IS (BisCore.DictionaryModel))\n            AND EXISTS (SELECT 1 FROM ${categoryElementClass} e WHERE e.Category.Id = this.ECInstanceId)\n          GROUP BY this.ECInstanceId\n        )`,\n        `SubCategoriesWithLabels(ClassName, ECInstanceId, ParentId, DisplayLabel) as (\n          SELECT\n            ec_classname(this.ECClassId, 's.c'),\n            this.ECInstanceId,\n            this.Parent.Id,\n            ${await props.labelsFactory.createSelectClause({\n              classAlias: \"this\",\n              className: \"BisCore.SubCategory\",\n            })}\n          FROM BisCore.SubCategory this\n          WHERE NOT this.IsPrivate\n        )`,\n      ],\n      ecsql: `\n      SELECT * FROM (\n        SELECT\n          c.ClassName AS CategoryClass,\n          c.ECInstanceId AS CategoryId,\n          sc.ClassName AS SubcategoryClass,\n          sc.ECInstanceId AS SubcategoryId\n        FROM RootCategoriesWithLabels c\n        JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId\n        WHERE c.ChildCount > 1 AND sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n        UNION ALL\n        SELECT\n          c.ClassName AS CategoryClass,\n          c.ECInstanceId AS CategoryId,\n          CAST(NULL AS TEXT) AS SubcategoryClass,\n          CAST(NULL AS TEXT) AS SubcategoryId\n        FROM RootCategoriesWithLabels c\n        WHERE c.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n        )\n        LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}\n      `,\n      bindings: [\n        { type: \"string\", value: adjustedLabel },\n        { type: \"string\", value: adjustedLabel },\n      ],\n    },\n    { restartToken: \"tree-widget/categories-tree/filter-by-label-query\" },\n  );\n  const paths: HierarchyFilteringPath[] = [];\n  for await (const row of reader) {\n    const path = { path: [{ className: row.CategoryClass, id: row.CategoryId }], options: { autoExpand: true } };\n    row.SubcategoryId && path.path.push({ className: row.SubcategoryClass, id: row.SubcategoryId });\n    paths.push(path);\n  }\n  if (paths.length > MAX_FILTERING_INSTANCE_KEY_COUNT) {\n    throw new FilterLimitExceededError(MAX_FILTERING_INSTANCE_KEY_COUNT);\n  }\n  return paths;\n}\n"]}
+{"version":3,"file":"CategoriesTreeDefinition.js","sourceRoot":"","sources":["../../../../../../src/tree-widget-react/components/trees/categories-tree/CategoriesTreeDefinition.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,aAAa,EAAE,GAAG,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AACjF,OAAO,EAAE,6BAA6B,EAAE,uCAAuC,EAAE,MAAM,iCAAiC,CAAC;AACzH,OAAO,EAAE,yCAAyC,EAAE,KAAK,EAAE,MAAM,4BAA4B,CAAC;AAC9F,OAAO,EAAE,wBAAwB,EAAE,MAAM,yBAAyB,CAAC;AACnE,OAAO,EAAE,gBAAgB,EAAE,MAAM,sCAAsC,CAAC;AACxE,OAAO,EAAE,0BAA0B,EAAE,wBAAwB,EAAE,kBAAkB,EAAE,MAAM,oCAAoC,CAAC;AAkB9H,MAAM,gCAAgC,GAAG,GAAG,CAAC;AAgB7C,MAAM,OAAO,wBAAwB;IAQnC,YAAmB,KAAoC;QACrD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,QAAQ,CAAC;QAChC,IAAI,CAAC,6BAA6B,GAAG,yCAAyC,CAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QAChI,IAAI,CAAC,mBAAmB,GAAG,6BAA6B,CAAC;YACvD,YAAY,EAAE,KAAK,CAAC,YAAY;YAChC,gCAAgC,EAAE,IAAI,CAAC,6BAA6B;SACrE,CAAC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,sBAAsB;QAClC,IAAI,CAAC,KAAK,KAAK,CAAC,KAAK,IAAI,EAAE;YACzB,MAAM,8BAA8B,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,iCAAiC,EAAE,CAAC;YAChG,OAAO,uCAAuC,CAAC;gBAC7C,uBAAuB,EAAE,IAAI,CAAC,aAAa;gBAC3C,SAAS,EAAE;oBACT,SAAS,EAAE,KAAK,EAAE,YAA2C,EAAE,EAAE,CAC/D,IAAI,CAAC,4CAA4C,CAAC,EAAE,GAAG,YAAY,EAAE,QAAQ,EAAE,IAAI,CAAC,SAAS,EAAE,CAAC;oBAClG,UAAU,EAAE;wBACV;4BACE,4BAA4B,EAAE,kBAAkB;4BAChD,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,YAAY,CAAC;yBAC3H;wBACD,GAAG,CAAC,8BAA8B;4BAChC,CAAC,CAAC;gCACE;oCACE,4BAA4B,EAAE,0BAA0B;oCACxD,WAAW,EAAE,KAAK,EAAE,YAAwD,EAAE,EAAE,CAC9E,IAAI,CAAC,4CAA4C,CAAC;wCAChD,GAAG,YAAY;wCACf,QAAQ,EAAE,IAAI,CAAC,SAAS;qCACzB,CAAC;iCACL;6BACF;4BACH,CAAC,CAAC,EAAE,CAAC;qBACR;iBACF;aACF,CAAC,CAAC;QACL,CAAC,CAAC,EAAE,CAAC;QACL,OAAO,IAAI,CAAC,KAAK,CAAC;IACpB,CAAC;IAEM,KAAK,CAAC,oBAAoB,CAAC,KAAgC;QAChE,OAAO,CAAC,MAAM,IAAI,CAAC,sBAAsB,EAAE,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAC3E,CAAC;IAEO,KAAK,CAAC,4CAA4C,CAAC,KAI1D;QACC,MAAM,EAAE,qBAAqB,EAAE,cAAc,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;QAClE,MAAM,EAAE,oBAAoB,EAAE,UAAU,EAAE,GACxC,qBAAqB,KAAK,SAAS;YACjC,CAAC,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,wCAAwC,EAAE;YACjE,CAAC,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,+CAA+C,CAAC,qBAAqB,CAAC,CAAC;QAClG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,IAAI,oBAAoB,CAAC,MAAM,KAAK,CAAC,EAAE;YAChE,OAAO,EAAE,CAAC;SACX;QAED,MAAM,yBAAyB,GAAG,IAAI,KAAK,EAAc,CAAC;QAC1D,MAAM,8BAA8B,GAAG,IAAI,KAAK,EAAc,CAAC;QAC/D,UAAU,CAAC,OAAO,CAAC,CAAC,QAAQ,EAAE,EAAE;YAC9B,IAAI,QAAQ,CAAC,UAAU,GAAG,CAAC,EAAE;gBAC3B,8BAA8B,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;aAClD;iBAAM;gBACL,yBAAyB,CAAC,IAAI,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC;aAC7C;QACH,CAAC,CAAC,CAAC;QACH,MAAM,0BAA0B,GAC9B,yBAAyB,CAAC,MAAM,GAAG,8BAA8B,CAAC,MAAM;YACtE,CAAC,CAAC,EAAE,GAAG,EAAE,8BAA8B,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE;YAChE,CAAC,CAAC,EAAE,GAAG,EAAE,yBAAyB,EAAE,MAAM,EAAE,CAAC,EAAE,OAAO,EAAE,CAAC,EAAE,CAAC;QAEhE,MAAM,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;QAErD,MAAM,CAAC,+BAA+B,EAAE,yCAAyC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CACpG,CAAC,aAAa,EAAE,GAAG,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CAChH,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC3C,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,MAAM,EAAE;SACrD,CAAC,CACH,CACF,CAAC;QAEF,MAAM,yBAAyB,GAC7B,oBAAoB,CAAC,MAAM,GAAG,CAAC;YAC7B,CAAC,CAAC;;cAEI,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;gBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE;gBAClF,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;gBAC/C,SAAS,EAAE;oBACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;wBACpE,UAAU,EAAE,MAAM;wBAClB,SAAS,EAAE,0BAA0B;qBACtC,CAAC;iBACH;gBACD,YAAY,EAAE;oBACZ,qBAAqB,EAAE,IAAI;oBAC3B,OAAO,EAAE,2BAA2B;iBACrC;gBACD,WAAW,EAAE,IAAI;gBACjB,iBAAiB,EAAE,IAAI;aACxB,CAAC;;cAEA,yCAAyC,CAAC,IAAI;cAC9C,yCAAyC,CAAC,KAAK;;oCAEzB,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC;cACrD,yCAAyC,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,yCAAyC,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;SACpH;YACD,CAAC,CAAC,SAAS,CAAC;QAEhB,MAAM,eAAe,GACnB,UAAU,CAAC,MAAM,GAAG,CAAC;YACnB,CAAC,CAAC;;gBAEM,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;gBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,8BAA8B,CAAC,MAAM,EAAE,WAAW,CAAC,EAAE;gBAClF,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;gBAC/C,SAAS,EAAE;oBACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;wBACpE,UAAU,EAAE,MAAM;wBAClB,SAAS,EAAE,aAAa;qBACzB,CAAC;iBACH;gBACD,GAAG,CAAC,0BAA0B,CAAC,GAAG,CAAC,MAAM,GAAG,CAAC;oBAC3C,CAAC,CAAC;wBACE,WAAW,EAAE;4BACX,QAAQ,EAAE;wDACsB,0BAA0B,CAAC,GAAG,CAAC,IAAI,CAAC,GAAG,CAAC;gCAChE,0BAA0B,CAAC,MAAM;gCACjC,0BAA0B,CAAC,OAAO;;2BAEvC;yBACJ;qBACF;oBACH,CAAC,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,0BAA0B,CAAC,OAAO,EAAE,CAAC;gBAC1D,YAAY,EAAE;oBACZ,WAAW,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE;oBAC7C,UAAU,EAAE,IAAI;oBAChB,OAAO,EAAE,aAAa;iBACvB;gBACD,iBAAiB,EAAE,IAAI;aACxB,CAAC;;gBAEA,+BAA+B,CAAC,IAAI;gBACpC,+BAA+B,CAAC,KAAK;;sCAEf,UAAU,CAAC,GAAG,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC;gBAC1E,+BAA+B,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,+BAA+B,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;SAClG;YACD,CAAC,CAAC,SAAS,CAAC;QAChB,MAAM,OAAO,GAAG,CAAC,eAAe,EAAE,yBAAyB,CAAC,CAAC,MAAM,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC;QACpG,OAAO;YACL;gBACE,aAAa,EAAE,wBAAwB;gBACvC,KAAK,EAAE;oBACL,KAAK,EAAE,OAAO,CAAC,IAAI,CAAC,aAAa,CAAC;iBACnC;aACF;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAAC,EACnC,qBAAqB,EAAE,UAAU,EACjC,cAAc,GAC6B;QAC3C,MAAM,qBAAqB,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,mBAAmB,CAAC;YAC/E,MAAM,EAAE,cAAc;YACtB,YAAY,EAAE,EAAE,QAAQ,EAAE,kBAAkB,EAAE,KAAK,EAAE,MAAM,EAAE;SAC9D,CAAC,CAAC;QACH,OAAO;YACL;gBACE,aAAa,EAAE,kBAAkB;gBACjC,KAAK,EAAE;oBACL,KAAK,EAAE;;gBAED,MAAM,IAAI,CAAC,mBAAmB,CAAC,kBAAkB,CAAC;wBAClD,SAAS,EAAE,EAAE,QAAQ,EAAE,gBAAgB,EAAE;wBACzC,YAAY,EAAE,EAAE,QAAQ,EAAE,mBAAmB,EAAE;wBAC/C,SAAS,EAAE;4BACT,QAAQ,EAAE,MAAM,IAAI,CAAC,6BAA6B,CAAC,kBAAkB,CAAC;gCACpE,UAAU,EAAE,MAAM;gCAClB,SAAS,EAAE,kBAAkB;6BAC9B,CAAC;yBACH;wBACD,YAAY,EAAE;4BACZ,UAAU,EAAE,EAAE,QAAQ,EAAE,gCAAgC,EAAE;4BAC1D,aAAa,EAAE,IAAI;4BACnB,OAAO,EAAE,qBAAqB;yBAC/B;wBACD,iBAAiB,EAAE,KAAK;qBACzB,CAAC;mBACG,qBAAqB,CAAC,IAAI;cAC/B,qBAAqB,CAAC,KAAK;;0DAEiB,UAAU,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;gBAC7E,qBAAqB,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,qBAAqB,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;WAC5E;oBACD,QAAQ,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC;iBAC9D;aACF;SACF,CAAC;IACJ,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,sBAAsB,CAAC,KAA2D;QACpG,MAAM,aAAa,GAAG,yCAAyC,CAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACjH,OAAO,uCAAuC,CAAC,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,KAAK,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;IACrG,CAAC;CACF;AAED,KAAK,UAAU,uCAAuC,CACpD,KAAiJ;IAEjJ,MAAM,EAAE,oBAAoB,EAAE,UAAU,EAAE,GAAG,MAAM,KAAK,CAAC,KAAK,CAAC,uCAAuC,EAAE,CAAC;IACzG,IAAI,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE;QAC3B,OAAO,EAAE,CAAC;KACX;IAED,MAAM,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC,KAAK,CAAC,QAAQ,CAAC,CAAC;IAC3D,MAAM,aAAa,GAAG,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,CAAC;IAE7D,MAAM,0BAA0B,GAAG,sBAAsB,CAAC;IAC1D,MAAM,6BAA6B,GAAG,yBAAyB,CAAC;IAChE,MAAM,qCAAqC,GAAG,gCAAgC,CAAC;IAC/E,MAAM,CAAC,yBAAyB,EAAE,4BAA4B,EAAE,oCAAoC,CAAC,GAAG,MAAM,OAAO,CAAC,GAAG,CACvH,CAAC,aAAa,EAAE,kBAAkB,EAAE,GAAG,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,0BAA0B,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,KAAK,EAAE,SAAS,EAAE,EAAE,CACpI,KAAK,CAAC,aAAa,CAAC,kBAAkB,CAAC,EAAE,UAAU,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC,CAC1E,CACF,CAAC;IACF,OAAO,aAAa,CAClB,KAAK,CAAC,GAAG,EAAE;QACT,MAAM,IAAI,GAAG;YACX,GAAG,0BAA0B;;;;;gBAKrB,yBAAyB;;gBAEzB,aAAa;qBACR,kBAAkB;;sCAED,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;;YAE/C;YACJ,GAAG,6BAA6B;;;;;gBAKxB,4BAA4B;;gBAE5B,kBAAkB;;;uCAGK,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC;YAChD;YACJ,GAAG,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC;gBACjC,CAAC,CAAC;oBACE,GAAG,qCAAqC;;;;oBAIlC,oCAAoC;;oBAEpC,0BAA0B;;0CAEJ,oBAAoB,CAAC,IAAI,CAAC,IAAI,CAAC;gBACzD;iBACH;gBACH,CAAC,CAAC,EAAE,CAAC;SACR,CAAC;QACF,MAAM,KAAK,GAAG;;;;;;gBAMJ,0BAA0B;qBACrB,6BAA6B;;;;;;;;;;;gBAWlC,0BAA0B;;;;cAK5B,oBAAoB,CAAC,MAAM,GAAG,CAAC;YAC7B,CAAC,CAAC;;;;;;sBAMI,qCAAqC;;;iBAG1C;YACD,CAAC,CAAC,EACN;;UAEF,KAAK,CAAC,KAAK,KAAK,SAAS,CAAC,CAAC,CAAC,SAAS,gCAAgC,GAAG,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC,KAAK,KAAK,WAAW,CAAC,CAAC,CAAC,SAAS,KAAK,CAAC,KAAK,EAAE,CAAC,CAAC,CAAC,EAAE;OAC1I,CAAC;QACF,MAAM,QAAQ,GAAG;YACf,EAAE,IAAI,EAAE,QAAiB,EAAE,KAAK,EAAE,aAAa,EAAE;YACjD,EAAE,IAAI,EAAE,QAAiB,EAAE,KAAK,EAAE,aAAa,EAAE;YACjD,GAAG,CAAC,oBAAoB,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,IAAI,EAAE,QAAiB,EAAE,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC;SAChG,CAAC;QACF,OAAO,KAAK,CAAC,YAAY,CAAC,iBAAiB,CACzC,EAAE,IAAI,EAAE,KAAK,EAAE,QAAQ,EAAE,EACzB,EAAE,YAAY,EAAE,mDAAmD,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,EAAE,CAC1F,CAAC;IACJ,CAAC,CAAC,CAAC,IAAI,CACL,GAAG,CACD,CAAC,GAAG,EAAe,EAAE,CAAC,CAAC;QACrB,SAAS,EAAE,GAAG,CAAC,SAAS,KAAK,GAAG,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,GAAG,CAAC,SAAS,KAAK,IAAI,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC,CAAC,0BAA0B;QAC3H,EAAE,EAAE,GAAG,CAAC,YAAY;KACrB,CAAC,CACH,EACD,OAAO,EAAE,EACT,QAAQ,CAAC,CAAC,WAAW,EAAsC,EAAE,CAAC,qCAAqC,CAAC,EAAE,GAAG,KAAK,EAAE,WAAW,EAAE,CAAC,CAAC,EAC/H,OAAO,EAAE,CACV,CACF,CAAC;AACJ,CAAC;AAED,SAAS,qCAAqC,CAC5C,KAEC;IAED,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,QAAQ,EAAE,QAAQ,EAAE,GAAG,KAAK,CAAC;IACzD,IAAI,KAAK,KAAK,WAAW,IAAI,WAAW,CAAC,MAAM,GAAG,CAAC,KAAK,IAAI,gCAAgC,CAAC,EAAE;QAC7F,MAAM,IAAI,wBAAwB,CAAC,KAAK,IAAI,gCAAgC,CAAC,CAAC;KAC/E;IAED,IAAI,WAAW,CAAC,MAAM,KAAK,CAAC,EAAE;QAC5B,OAAO,KAAK,CAAC;KACd;IAED,MAAM,EAAE,aAAa,EAAE,GAAG,gBAAgB,CAAC,QAAQ,CAAC,CAAC;IACrD,OAAO,IAAI,CAAC,WAAW,CAAC,CAAC,IAAI,CAC3B,QAAQ,CAAC,KAAK,EAAE,UAAU,EAAE,EAAE;QAC5B,IAAI,UAAU,CAAC,SAAS,KAAK,kBAAkB,EAAE;YAC/C,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC,mBAAmB,CAAC,EAAE,aAAa,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC;SACtH;QACD,IAAI,UAAU,CAAC,SAAS,KAAK,aAAa,EAAE;YAC1C,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC,mBAAmB,CAAC,EAAE,UAAU,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC;SACnH;QACD,OAAO,EAAE,IAAI,EAAE,MAAM,QAAQ,CAAC,mBAAmB,CAAC,EAAE,qBAAqB,EAAE,UAAU,CAAC,EAAE,EAAE,CAAC,EAAE,OAAO,EAAE,EAAE,UAAU,EAAE,IAAI,EAAE,EAAE,CAAC;IAC/H,CAAC,CAAC,CACH,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { defer, EMPTY, from, lastValueFrom, map, mergeMap, toArray } from \"rxjs\";\nimport { createNodesQueryClauseFactory, createPredicateBasedHierarchyDefinition } from \"@itwin/presentation-hierarchies\";\nimport { createBisInstanceLabelSelectClauseFactory, ECSql } from \"@itwin/presentation-shared\";\nimport { FilterLimitExceededError } from \"../common/TreeErrors.js\";\nimport { getClassesByView } from \"./internal/CategoriesTreeIdsCache.js\";\nimport { DEFINITION_CONTAINER_CLASS, DEFINITION_ELEMENT_CLASS, SUB_CATEGORY_CLASS } from \"./internal/ClassNameDefinitions.js\";\n\nimport type { Observable } from \"rxjs\";\nimport type { Id64Array, Id64String } from \"@itwin/core-bentley\";\nimport type { ECClassHierarchyInspector, ECSchemaProvider, IInstanceLabelSelectClauseFactory, InstanceKey } from \"@itwin/presentation-shared\";\nimport type { CategoriesTreeIdsCache } from \"./internal/CategoriesTreeIdsCache.js\";\nimport type {\n  DefineHierarchyLevelProps,\n  DefineInstanceNodeChildHierarchyLevelProps,\n  DefineRootHierarchyLevelProps,\n  GenericInstanceFilter,\n  HierarchyDefinition,\n  HierarchyFilteringPath,\n  HierarchyLevelDefinition,\n  LimitingECSqlQueryExecutor,\n  NodesQueryClauseFactory,\n} from \"@itwin/presentation-hierarchies\";\n\nconst MAX_FILTERING_INSTANCE_KEY_COUNT = 100;\n\ninterface CategoriesTreeDefinitionProps {\n  imodelAccess: ECSchemaProvider & ECClassHierarchyInspector & LimitingECSqlQueryExecutor;\n  viewType: \"2d\" | \"3d\";\n  idsCache: CategoriesTreeIdsCache;\n}\n\ninterface CategoriesTreeInstanceKeyPathsFromInstanceLabelProps {\n  imodelAccess: ECClassHierarchyInspector & LimitingECSqlQueryExecutor;\n  label: string;\n  viewType: \"2d\" | \"3d\";\n  limit?: number | \"unbounded\";\n  idsCache: CategoriesTreeIdsCache;\n}\n\nexport class CategoriesTreeDefinition implements HierarchyDefinition {\n  private _impl: Promise<HierarchyDefinition> | undefined;\n  private _selectQueryFactory: NodesQueryClauseFactory;\n  private _nodeLabelSelectClauseFactory: IInstanceLabelSelectClauseFactory;\n  private _idsCache: CategoriesTreeIdsCache;\n  private _viewType: \"2d\" | \"3d\";\n  private _iModelAccess: ECSchemaProvider & ECClassHierarchyInspector & LimitingECSqlQueryExecutor;\n\n  public constructor(props: CategoriesTreeDefinitionProps) {\n    this._iModelAccess = props.imodelAccess;\n    this._viewType = props.viewType;\n    this._idsCache = props.idsCache;\n    this._nodeLabelSelectClauseFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });\n    this._selectQueryFactory = createNodesQueryClauseFactory({\n      imodelAccess: props.imodelAccess,\n      instanceLabelSelectClauseFactory: this._nodeLabelSelectClauseFactory,\n    });\n  }\n\n  private async getHierarchyDefinition(): Promise<HierarchyDefinition> {\n    this._impl ??= (async () => {\n      const isDefinitionContainerSupported = await this._idsCache.getIsDefinitionContainerSupported();\n      return createPredicateBasedHierarchyDefinition({\n        classHierarchyInspector: this._iModelAccess,\n        hierarchy: {\n          rootNodes: async (requestProps: DefineRootHierarchyLevelProps) =>\n            this.createDefinitionContainersAndCategoriesQuery({ ...requestProps, viewType: this._viewType }),\n          childNodes: [\n            {\n              parentInstancesNodePredicate: \"BisCore.Category\",\n              definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) => this.createSubcategoryQuery(requestProps),\n            },\n            ...(isDefinitionContainerSupported\n              ? [\n                  {\n                    parentInstancesNodePredicate: DEFINITION_CONTAINER_CLASS,\n                    definitions: async (requestProps: DefineInstanceNodeChildHierarchyLevelProps) =>\n                      this.createDefinitionContainersAndCategoriesQuery({\n                        ...requestProps,\n                        viewType: this._viewType,\n                      }),\n                  },\n                ]\n              : []),\n          ],\n        },\n      });\n    })();\n    return this._impl;\n  }\n\n  public async defineHierarchyLevel(props: DefineHierarchyLevelProps) {\n    return (await this.getHierarchyDefinition()).defineHierarchyLevel(props);\n  }\n\n  private async createDefinitionContainersAndCategoriesQuery(props: {\n    parentNodeInstanceIds?: Id64Array;\n    instanceFilter?: GenericInstanceFilter;\n    viewType: \"2d\" | \"3d\";\n  }): Promise<HierarchyLevelDefinition> {\n    const { parentNodeInstanceIds, instanceFilter, viewType } = props;\n    const { definitionContainers, categories } =\n      parentNodeInstanceIds === undefined\n        ? await this._idsCache.getRootDefinitionContainersAndCategories()\n        : await this._idsCache.getDirectChildDefinitionContainersAndCategories(parentNodeInstanceIds);\n    if (categories.length === 0 && definitionContainers.length === 0) {\n      return [];\n    }\n\n    const categoriesWithSingleChild = new Array<Id64String>();\n    const categoriesWithMultipleChildren = new Array<Id64String>();\n    categories.forEach((category) => {\n      if (category.childCount > 1) {\n        categoriesWithMultipleChildren.push(category.id);\n      } else {\n        categoriesWithSingleChild.push(category.id);\n      }\n    });\n    const dataToDetermineHasChildren =\n      categoriesWithSingleChild.length > categoriesWithMultipleChildren.length\n        ? { ids: categoriesWithMultipleChildren, ifTrue: 1, ifFalse: 0 }\n        : { ids: categoriesWithSingleChild, ifTrue: 0, ifFalse: 1 };\n\n    const { categoryClass } = getClassesByView(viewType);\n\n    const [categoriesInstanceFilterClauses, definitionContainersInstanceFilterClauses] = await Promise.all(\n      [categoryClass, ...(definitionContainers.length > 0 ? [DEFINITION_CONTAINER_CLASS] : [])].map(async (className) =>\n        this._selectQueryFactory.createFilterClauses({\n          filter: instanceFilter,\n          contentClass: { fullName: className, alias: \"this\" },\n        }),\n      ),\n    );\n\n    const definitionContainersQuery =\n      definitionContainers.length > 0\n        ? `\n          SELECT\n            ${await this._selectQueryFactory.createSelectClause({\n              ecClassId: { selector: ECSql.createRawPropertyValueSelector(\"this\", \"ECClassId\") },\n              ecInstanceId: { selector: \"this.ECInstanceId\" },\n              nodeLabel: {\n                selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                  classAlias: \"this\",\n                  className: DEFINITION_CONTAINER_CLASS,\n                }),\n              },\n              extendedData: {\n                isDefinitionContainer: true,\n                imageId: \"icon-definition-container\",\n              },\n              hasChildren: true,\n              supportsFiltering: true,\n            })}\n          FROM\n            ${definitionContainersInstanceFilterClauses.from} this\n            ${definitionContainersInstanceFilterClauses.joins}\n          WHERE\n            this.ECInstanceId IN (${definitionContainers.join(\", \")})\n            ${definitionContainersInstanceFilterClauses.where ? `AND ${definitionContainersInstanceFilterClauses.where}` : \"\"}\n        `\n        : undefined;\n\n    const categoriesQuery =\n      categories.length > 0\n        ? `\n          SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: ECSql.createRawPropertyValueSelector(\"this\", \"ECClassId\") },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: categoryClass,\n                  }),\n                },\n                ...(dataToDetermineHasChildren.ids.length > 0\n                  ? {\n                      hasChildren: {\n                        selector: `\n                            IIF(this.ECInstanceId IN (${dataToDetermineHasChildren.ids.join(\",\")}),\n                              ${dataToDetermineHasChildren.ifTrue},\n                              ${dataToDetermineHasChildren.ifFalse}\n                            )\n                          `,\n                      },\n                    }\n                  : { hasChildren: !!dataToDetermineHasChildren.ifFalse }),\n                extendedData: {\n                  description: { selector: \"this.Description\" },\n                  isCategory: true,\n                  imageId: \"icon-layers\",\n                },\n                supportsFiltering: true,\n              })}\n            FROM\n              ${categoriesInstanceFilterClauses.from} this\n              ${categoriesInstanceFilterClauses.joins}\n            WHERE\n              this.ECInstanceId IN (${categories.map((category) => category.id).join(\", \")})\n              ${categoriesInstanceFilterClauses.where ? `AND ${categoriesInstanceFilterClauses.where}` : \"\"}\n        `\n        : undefined;\n    const queries = [categoriesQuery, definitionContainersQuery].filter((query) => query !== undefined);\n    return [\n      {\n        fullClassName: DEFINITION_ELEMENT_CLASS,\n        query: {\n          ecsql: queries.join(\" UNION ALL \"),\n        },\n      },\n    ];\n  }\n\n  private async createSubcategoryQuery({\n    parentNodeInstanceIds: elementIds,\n    instanceFilter,\n  }: DefineInstanceNodeChildHierarchyLevelProps): Promise<HierarchyLevelDefinition> {\n    const instanceFilterClauses = await this._selectQueryFactory.createFilterClauses({\n      filter: instanceFilter,\n      contentClass: { fullName: SUB_CATEGORY_CLASS, alias: \"this\" },\n    });\n    return [\n      {\n        fullClassName: SUB_CATEGORY_CLASS,\n        query: {\n          ecsql: `\n            SELECT\n              ${await this._selectQueryFactory.createSelectClause({\n                ecClassId: { selector: \"this.ECClassId\" },\n                ecInstanceId: { selector: \"this.ECInstanceId\" },\n                nodeLabel: {\n                  selector: await this._nodeLabelSelectClauseFactory.createSelectClause({\n                    classAlias: \"this\",\n                    className: SUB_CATEGORY_CLASS,\n                  }),\n                },\n                extendedData: {\n                  categoryId: { selector: \"printf('0x%x', this.Parent.Id)\" },\n                  isSubCategory: true,\n                  imageId: \"icon-layers-isolate\",\n                },\n                supportsFiltering: false,\n              })}\n            FROM ${instanceFilterClauses.from} this\n            ${instanceFilterClauses.joins}\n            WHERE\n              NOT this.IsPrivate AND this.Parent.Id IN (${elementIds.map(() => \"?\").join(\",\")})\n              ${instanceFilterClauses.where ? `AND ${instanceFilterClauses.where}` : \"\"}\n          `,\n          bindings: elementIds.map((id) => ({ type: \"id\", value: id })),\n        },\n      },\n    ];\n  }\n\n  public static async createInstanceKeyPaths(props: CategoriesTreeInstanceKeyPathsFromInstanceLabelProps): Promise<HierarchyFilteringPath[]> {\n    const labelsFactory = createBisInstanceLabelSelectClauseFactory({ classHierarchyInspector: props.imodelAccess });\n    return createInstanceKeyPathsFromInstanceLabel({ ...props, labelsFactory, cache: props.idsCache });\n  }\n}\n\nasync function createInstanceKeyPathsFromInstanceLabel(\n  props: CategoriesTreeInstanceKeyPathsFromInstanceLabelProps & { labelsFactory: IInstanceLabelSelectClauseFactory; cache: CategoriesTreeIdsCache },\n): Promise<HierarchyFilteringPath[]> {\n  const { definitionContainers, categories } = await props.cache.getAllDefinitionContainersAndCategories();\n  if (categories.length === 0) {\n    return [];\n  }\n\n  const { categoryClass } = getClassesByView(props.viewType);\n  const adjustedLabel = props.label.replace(/[%_\\\\]/g, \"\\\\$&\");\n\n  const CATEGORIES_WITH_LABELS_CTE = \"CategoriesWithLabels\";\n  const SUBCATEGORIES_WITH_LABELS_CTE = \"SubCategoriesWithLabels\";\n  const DEFINITION_CONTAINERS_WITH_LABELS_CTE = \"DefinitionContainersWithLabels\";\n  const [categoryLabelSelectClause, subCategoryLabelSelectClause, definitionContainerLabelSelectClause] = await Promise.all(\n    [categoryClass, SUB_CATEGORY_CLASS, ...(definitionContainers.length > 0 ? [DEFINITION_CONTAINER_CLASS] : [])].map(async (className) =>\n      props.labelsFactory.createSelectClause({ classAlias: \"this\", className }),\n    ),\n  );\n  return lastValueFrom(\n    defer(() => {\n      const ctes = [\n        `${CATEGORIES_WITH_LABELS_CTE}(ClassName, ECInstanceId, ChildCount, DisplayLabel) AS (\n            SELECT\n              'c',\n              this.ECInstanceId,\n              COUNT(sc.ECInstanceId),\n              ${categoryLabelSelectClause}\n            FROM\n              ${categoryClass} this\n              JOIN ${SUB_CATEGORY_CLASS} sc ON sc.Parent.Id = this.ECInstanceId\n            WHERE\n              this.ECInstanceId IN (${categories.join(\", \")})\n              GROUP BY this.ECInstanceId\n          )`,\n        `${SUBCATEGORIES_WITH_LABELS_CTE}(ClassName, ECInstanceId, ParentId, DisplayLabel) AS (\n            SELECT\n              'sc',\n              this.ECInstanceId,\n              this.Parent.Id,\n              ${subCategoryLabelSelectClause}\n            FROM\n              ${SUB_CATEGORY_CLASS} this\n            WHERE\n              NOT this.IsPrivate\n              AND this.Parent.Id IN (${categories.join(\", \")})\n          )`,\n        ...(definitionContainers.length > 0\n          ? [\n              `${DEFINITION_CONTAINERS_WITH_LABELS_CTE}(ClassName, ECInstanceId, DisplayLabel) AS (\n                SELECT\n                  'dc',\n                  this.ECInstanceId,\n                  ${definitionContainerLabelSelectClause}\n                FROM\n                  ${DEFINITION_CONTAINER_CLASS} this\n                WHERE\n                  this.ECInstanceId IN (${definitionContainers.join(\", \")})\n              )`,\n            ]\n          : []),\n      ];\n      const ecsql = `\n        SELECT * FROM (\n            SELECT\n              sc.ClassName AS ClassName,\n              sc.ECInstanceId AS ECInstanceId\n            FROM\n              ${CATEGORIES_WITH_LABELS_CTE} c\n              JOIN ${SUBCATEGORIES_WITH_LABELS_CTE} sc ON sc.ParentId = c.ECInstanceId\n            WHERE\n              c.ChildCount > 1\n              AND sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n\n            UNION ALL\n\n            SELECT\n              c.ClassName AS ClassName,\n              c.ECInstanceId AS ECInstanceId\n            FROM\n              ${CATEGORIES_WITH_LABELS_CTE} c\n            WHERE\n              c.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n\n            ${\n              definitionContainers.length > 0\n                ? `\n                  UNION ALL\n                  SELECT\n                    dc.ClassName AS ClassName,\n                    dc.ECInstanceId AS ECInstanceId\n                  FROM\n                    ${DEFINITION_CONTAINERS_WITH_LABELS_CTE} dc\n                  WHERE\n                    dc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n                `\n                : \"\"\n            }\n        )\n        ${props.limit === undefined ? `LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}` : props.limit !== \"unbounded\" ? `LIMIT ${props.limit}` : \"\"}\n      `;\n      const bindings = [\n        { type: \"string\" as const, value: adjustedLabel },\n        { type: \"string\" as const, value: adjustedLabel },\n        ...(definitionContainers.length > 0 ? [{ type: \"string\" as const, value: adjustedLabel }] : []),\n      ];\n      return props.imodelAccess.createQueryReader(\n        { ctes, ecsql, bindings },\n        { restartToken: \"tree-widget/categories-tree/filter-by-label-query\", limit: props.limit },\n      );\n    }).pipe(\n      map(\n        (row): InstanceKey => ({\n          className: row.ClassName === \"c\" ? categoryClass : row.ClassName === \"sc\" ? SUB_CATEGORY_CLASS : DEFINITION_CONTAINER_CLASS,\n          id: row.ECInstanceId,\n        }),\n      ),\n      toArray(),\n      mergeMap((targetItems): Observable<HierarchyFilteringPath> => createInstanceKeyPathsFromTargetItems({ ...props, targetItems })),\n      toArray(),\n    ),\n  );\n}\n\nfunction createInstanceKeyPathsFromTargetItems(\n  props: Pick<CategoriesTreeInstanceKeyPathsFromInstanceLabelProps, \"idsCache\" | \"limit\" | \"viewType\"> & {\n    targetItems: InstanceKey[];\n  },\n): Observable<HierarchyFilteringPath> {\n  const { limit, targetItems, viewType, idsCache } = props;\n  if (limit !== \"unbounded\" && targetItems.length > (limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT)) {\n    throw new FilterLimitExceededError(limit ?? MAX_FILTERING_INSTANCE_KEY_COUNT);\n  }\n\n  if (targetItems.length === 0) {\n    return EMPTY;\n  }\n\n  const { categoryClass } = getClassesByView(viewType);\n  return from(targetItems).pipe(\n    mergeMap(async (targetItem) => {\n      if (targetItem.className === SUB_CATEGORY_CLASS) {\n        return { path: await idsCache.getInstanceKeyPaths({ subCategoryId: targetItem.id }), options: { autoExpand: true } };\n      }\n      if (targetItem.className === categoryClass) {\n        return { path: await idsCache.getInstanceKeyPaths({ categoryId: targetItem.id }), options: { autoExpand: true } };\n      }\n      return { path: await idsCache.getInstanceKeyPaths({ definitionContainerId: targetItem.id }), options: { autoExpand: true } };\n    }),\n  );\n}\n"]}