@itwin/tree-widget-react 3.2.2 → 3.2.4

package/CHANGELOG.md

@@ -1,12 +1,28 @@
 # Change Log - @itwin/tree-widget-react
 
-This log was last generated on Tue, 07 Jan 2025 16:30:45 GMT and should not be manually modified.
+This log was last generated on Tue, 14 Jan 2025 02:50:44 GMT and should not be manually modified.
 
 <!-- Start content -->
 
+## 3.2.4
+
+Tue, 14 Jan 2025 02:50:44 GMT
+
+### Patches
+
+- Fixed a bug in `CategoriesTreeDefinition.createInstanceKeyPaths`, where sometimes incorrect key paths would be created. Because of this, `CategoriesTree` would show incorrect categories with applied label filter. ([#1139](https://github.com/iTwin/viewer-components-react/pull/1139))
+
+## 3.2.3
+
+Wed, 08 Jan 2025 13:28:10 GMT
+
+### Patches
+
+- Changed `onHierarchyLoad` to log errors to console ([#1135](https://github.com/iTwin/viewer-components-react/pull/1135))
+
 ## 3.2.2
 
-Tue, 07 Jan 2025 16:30:45 GMT
+Tue, 07 Jan 2025 16:30:55 GMT
 
 ### Patches
 

package/lib/cjs/components/trees/categories-tree/CategoriesTreeDefinition.js

@@ -133,13 +133,18 @@ function getClassesByView(viewType) {
 }
 async function createInstanceKeyPathsFromInstanceLabel(props) {
     const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);
+    const adjustedLabel = props.label.replace(/[%_\\]/g, "\\$&");
     const reader = props.imodelAccess.createQueryReader({
         ctes: [
-            `RootCategories(ClassName, ECInstanceId, ChildCount) as (
+            `RootCategoriesWithLabels(ClassName, ECInstanceId, ChildCount, DisplayLabel) as (
           SELECT
             ec_classname(this.ECClassId, 's.c'),
             this.ECInstanceId,
-            COUNT(sc.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
@@ -163,17 +168,30 @@ async function createInstanceKeyPathsFromInstanceLabel(props) {
         )`,
         ],
         ecsql: `
+      SELECT * FROM (
         SELECT
           c.ClassName AS CategoryClass,
           c.ECInstanceId AS CategoryId,
-          IIF(c.ChildCount > 1, sc.ClassName, NULL) AS SubcategoryClass,
-          IIF(c.ChildCount > 1, sc.ECInstanceId, NULL) AS SubcategoryId
-        FROM RootCategories c
+          sc.ClassName AS SubcategoryClass,
+          sc.ECInstanceId AS SubcategoryId
+        FROM RootCategoriesWithLabels c
         JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId
-        WHERE sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+        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 '\\'
+        )
         LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}
       `,
-        bindings: [{ type: "string", value: props.label.replace(/[%_\\]/g, "\\$&") }],
+        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) {

package/lib/cjs/components/trees/categories-tree/CategoriesTreeDefinition.js.map

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeDefinition.js","sourceRoot":"","sources":["../../../../../src/components/trees/categories-tree/CategoriesTreeDefinition.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,8EAAyH;AACzH,oEAA8F;AAC9F,qDAAgE;AAchE,MAAM,gCAAgC,GAAG,GAAG,CAAC;AAa7C,MAAa,wBAAwB;IAKnC,YAAmB,KAAoC;QACrD,IAAI,CAAC,KAAK,GAAG,IAAA,kEAAuC,EAAC;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,IAAA,+DAAyC,EAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QAChI,IAAI,CAAC,mBAAmB,GAAG,IAAA,wDAA6B,EAAC;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,2BAAK,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,IAAA,+DAAyC,EAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACjH,OAAO,uCAAuC,CAAC,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IAC9E,CAAC;CACF;AA9HD,4DA8HC;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,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,iBAAiB,CACjD;QACE,IAAI,EAAE;YACJ;;;;;iBAKS,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;;;;;;;;;gBASG,gCAAgC,GAAG,CAAC;OAC7C;QACD,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;KAC9E,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,qCAAwB,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\";\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 reader = props.imodelAccess.createQueryReader(\n    {\n      ctes: [\n        `RootCategories(ClassName, ECInstanceId, ChildCount) as (\n          SELECT\n            ec_classname(this.ECClassId, 's.c'),\n            this.ECInstanceId,\n            COUNT(sc.ECInstanceId)\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\n          c.ClassName AS CategoryClass,\n          c.ECInstanceId AS CategoryId,\n          IIF(c.ChildCount > 1, sc.ClassName, NULL) AS SubcategoryClass,\n          IIF(c.ChildCount > 1, sc.ECInstanceId, NULL) AS SubcategoryId\n        FROM RootCategories c\n        JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId\n        WHERE sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n        LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}\n      `,\n      bindings: [{ type: \"string\", value: props.label.replace(/[%_\\\\]/g, \"\\\\$&\") }],\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/components/trees/categories-tree/CategoriesTreeDefinition.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAEhG,8EAAyH;AACzH,oEAA8F;AAC9F,qDAAgE;AAchE,MAAM,gCAAgC,GAAG,GAAG,CAAC;AAa7C,MAAa,wBAAwB;IAKnC,YAAmB,KAAoC;QACrD,IAAI,CAAC,KAAK,GAAG,IAAA,kEAAuC,EAAC;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,IAAA,+DAAyC,EAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QAChI,IAAI,CAAC,mBAAmB,GAAG,IAAA,wDAA6B,EAAC;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,2BAAK,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,IAAA,+DAAyC,EAAC,EAAE,uBAAuB,EAAE,KAAK,CAAC,YAAY,EAAE,CAAC,CAAC;QACjH,OAAO,uCAAuC,CAAC,EAAE,GAAG,KAAK,EAAE,aAAa,EAAE,CAAC,CAAC;IAC9E,CAAC;CACF;AA9HD,4DA8HC;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,qCAAwB,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\";\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"]}

package/lib/cjs/components/trees/common/components/Tree.js

@@ -50,7 +50,11 @@ function TreeImpl({ imodel, imodelAccess, treeName, noDataMessage, getFilteredPa
             onPerformanceMeasured(action, duration);
         },
         onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: "hierarchy-level-size-limit-hit", reportInteraction: false }),
-        onHierarchyLoadError: ({ type }) => onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false }),
+        onHierarchyLoadError: ({ type, error }) => {
+            // eslint-disable-next-line no-console
+            console.error(error);
+            onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });
+        },
     });
     (0, UseIModelChangeListener_1.useIModelChangeListener)({ imodel, action: (0, react_1.useCallback)(() => imodelChanged.raiseEvent(), [imodelChanged]) });
     const selectNodes = useSelectionPredicate({

package/lib/cjs/components/trees/common/components/Tree.js.map

@@ -1 +1 @@
-{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../src/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAEhG,iCAAuD;AACvD,sDAA8C;AAC9C,wDAAkE;AAClE,4EAA+E;AAC/E,0FAAqI;AACrI,uDAAoD;AACpD,8EAA2E;AAC3E,oEAAsE;AACtE,wEAAqE;AACrE,gEAA6D;AAC7D,gEAAiF;AACjF,oCAA8C;AAC9C,uCAAoC;AACpC,uDAAoD;AAkDpD;;;GAGG;AACH,SAAgB,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,IAAA,0BAAkB,EAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,uBAAC,uDAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,uBAAC,yDAAwB,IAAC,OAAO,EAAE,gBAAgB,YACjD,uBAAC,QAAQ,OAAK,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GAC1F,GACG,CACjC,CAAC;AACJ,CAAC;AAdD,oBAcC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,OAAO,EACP,SAAS,GAC2H;IACpI,MAAM,gBAAgB,GAAG,IAAA,uDAA0B,GAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,IAAA,yCAAmB,GAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,gBAAQ,EAAC,IAAI,sBAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,IAAA,8DAA6B,EAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;KAC5G,CAAC,CAAC;IACH,IAAA,iDAAuB,EAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAA,mBAAW,EAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,IAAA,oDAAmB,EAAC,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,IAAI,QAAQ,EAAE,CAAC,CAAC;IACjI,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,kDAA0B,EAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,yCAAmB,EAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,CACL,uBAAC,oBAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC/G,uBAAC,iBAAO,IAAC,IAAI,EAAE,IAAI,YACjB,uBAAC,8BAAc,IAAC,IAAI,EAAC,OAAO,GAAG,GACvB,GACL,CACR,CAAC;KACH;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,CACL,uBAAC,oBAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC9G,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,uBAAC,oBAAI,cAAE,uBAAU,CAAC,SAAS,CAAC,6BAA6B,CAAC,GAAQ,GAC9F,CACR,CAAC;KACH;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,WAAW;QACX,aAAa;QACb,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;QACR,IAAI,EAAE,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;KACnD,CAAC;IAEF,OAAO,CACL,iCAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,iCAAK,EAAE,EAAC,4BAA4B,EAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAC7E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,uBAAC,iBAAO,IAAC,IAAI,EAAE,SAAS,YACtB,uBAAC,iCAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,IAAA,mBAAW,EAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { Flex, ProgressRadial, Text } from \"@itwin/itwinui-react\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { UnifiedSelectionProvider, useIModelUnifiedSelectionTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\nimport { useHierarchiesLocalization } from \"../UseHierarchiesLocalization\";\nimport { useHierarchyLevelFiltering } from \"../UseHierarchyFiltering\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext\";\nimport { createIModelAccess } from \"../Utils\";\nimport { Delayed } from \"./Delayed\";\nimport { ProgressOverlay } from \"./ProgressOverlay\";\n\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting\";\nimport type { TreeRendererProps } from \"./TreeRenderer\";\n\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          TreeRendererProps,\n          \"rootNodes\" | \"expandNode\" | \"onNodeClick\" | \"onNodeKeyDown\" | \"onFilterClick\" | \"isNodeSelected\" | \"getHierarchyLevelDetails\" | \"size\" | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Modifies the density of tree nodes. `enlarged` tree nodes have bigger button hit boxes. */\n    density?: \"default\" | \"enlarged\";\n    /** Message that should be renderer if there are no tree nodes. */\n    noDataMessage?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @Beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <UnifiedSelectionProvider storage={selectionStorage}>\n        <TreeImpl {...props} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n      </UnifiedSelectionProvider>\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  noDataMessage,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  density,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\" | \"selectionStorage\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type }) => onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false }),\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { onNodeClick, onNodeKeyDown } = useSelectionHandler({ rootNodes, selectNodes, selectionMode: selectionMode ?? \"single\" });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        <Delayed show={true}>\n          <ProgressRadial size=\"large\" />\n        </Delayed>\n      </Flex>\n    );\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        {noDataMessage ? noDataMessage : <Text>{TreeWidget.translate(\"baseTree.dataIsNotAvailable\")}</Text>}\n      </Flex>\n    );\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    onNodeClick,\n    onNodeKeyDown,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n    size: density === \"enlarged\" ? \"default\" : \"small\",\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div id=\"tw-tree-renderer-container\" style={{ overflow: \"auto\", height: \"100%\" }}>\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}
+{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../src/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";;;;AAAA;;;gGAGgG;AAEhG,iCAAuD;AACvD,sDAA8C;AAC9C,wDAAkE;AAClE,4EAA+E;AAC/E,0FAAqI;AACrI,uDAAoD;AACpD,8EAA2E;AAC3E,oEAAsE;AACtE,wEAAqE;AACrE,gEAA6D;AAC7D,gEAAiF;AACjF,oCAA8C;AAC9C,uCAAoC;AACpC,uDAAoD;AAkDpD;;;GAGG;AACH,SAAgB,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,IAAA,eAAO,EAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,IAAA,0BAAkB,EAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,uBAAC,uDAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,uBAAC,yDAAwB,IAAC,OAAO,EAAE,gBAAgB,YACjD,uBAAC,QAAQ,OAAK,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GAC1F,GACG,CACjC,CAAC;AACJ,CAAC;AAdD,oBAcC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,OAAO,EACP,SAAS,GAC2H;IACpI,MAAM,gBAAgB,GAAG,IAAA,uDAA0B,GAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,IAAA,yCAAmB,GAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,IAAA,gBAAQ,EAAC,IAAI,sBAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,IAAA,8DAA6B,EAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;KACF,CAAC,CAAC;IACH,IAAA,iDAAuB,EAAC,EAAE,MAAM,EAAE,MAAM,EAAE,IAAA,mBAAW,EAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,IAAA,oDAAmB,EAAC,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,IAAI,QAAQ,EAAE,CAAC,CAAC;IACjI,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,IAAA,kDAA0B,EAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,IAAA,wCAAkB,EAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,IAAA,yCAAmB,EAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,CACL,uBAAC,oBAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC/G,uBAAC,iBAAO,IAAC,IAAI,EAAE,IAAI,YACjB,uBAAC,8BAAc,IAAC,IAAI,EAAC,OAAO,GAAG,GACvB,GACL,CACR,CAAC;KACH;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,CACL,uBAAC,oBAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC9G,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,uBAAC,oBAAI,cAAE,uBAAU,CAAC,SAAS,CAAC,6BAA6B,CAAC,GAAQ,GAC9F,CACR,CAAC;KACH;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,WAAW;QACX,aAAa;QACb,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;QACR,IAAI,EAAE,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;KACnD,CAAC;IAEF,OAAO,CACL,iCAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,iCAAK,EAAE,EAAC,4BAA4B,EAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAC7E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,uBAAC,iBAAO,IAAC,IAAI,EAAE,SAAS,YACtB,uBAAC,iCAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,IAAA,mBAAW,EAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { Flex, ProgressRadial, Text } from \"@itwin/itwinui-react\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { UnifiedSelectionProvider, useIModelUnifiedSelectionTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\nimport { useHierarchiesLocalization } from \"../UseHierarchiesLocalization\";\nimport { useHierarchyLevelFiltering } from \"../UseHierarchyFiltering\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext\";\nimport { createIModelAccess } from \"../Utils\";\nimport { Delayed } from \"./Delayed\";\nimport { ProgressOverlay } from \"./ProgressOverlay\";\n\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting\";\nimport type { TreeRendererProps } from \"./TreeRenderer\";\n\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          TreeRendererProps,\n          \"rootNodes\" | \"expandNode\" | \"onNodeClick\" | \"onNodeKeyDown\" | \"onFilterClick\" | \"isNodeSelected\" | \"getHierarchyLevelDetails\" | \"size\" | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Modifies the density of tree nodes. `enlarged` tree nodes have bigger button hit boxes. */\n    density?: \"default\" | \"enlarged\";\n    /** Message that should be renderer if there are no tree nodes. */\n    noDataMessage?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @Beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <UnifiedSelectionProvider storage={selectionStorage}>\n        <TreeImpl {...props} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n      </UnifiedSelectionProvider>\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  noDataMessage,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  density,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\" | \"selectionStorage\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type, error }) => {\n      // eslint-disable-next-line no-console\n      console.error(error);\n      onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });\n    },\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { onNodeClick, onNodeKeyDown } = useSelectionHandler({ rootNodes, selectNodes, selectionMode: selectionMode ?? \"single\" });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        <Delayed show={true}>\n          <ProgressRadial size=\"large\" />\n        </Delayed>\n      </Flex>\n    );\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        {noDataMessage ? noDataMessage : <Text>{TreeWidget.translate(\"baseTree.dataIsNotAvailable\")}</Text>}\n      </Flex>\n    );\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    onNodeClick,\n    onNodeKeyDown,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n    size: density === \"enlarged\" ? \"default\" : \"small\",\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div id=\"tw-tree-renderer-container\" style={{ overflow: \"auto\", height: \"100%\" }}>\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}

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

@@ -129,13 +129,18 @@ function getClassesByView(viewType) {
 }
 async function createInstanceKeyPathsFromInstanceLabel(props) {
     const { categoryClass, categoryElementClass } = getClassesByView(props.viewType);
+    const adjustedLabel = props.label.replace(/[%_\\]/g, "\\$&");
     const reader = props.imodelAccess.createQueryReader({
         ctes: [
-            `RootCategories(ClassName, ECInstanceId, ChildCount) as (
+            `RootCategoriesWithLabels(ClassName, ECInstanceId, ChildCount, DisplayLabel) as (
           SELECT
             ec_classname(this.ECClassId, 's.c'),
             this.ECInstanceId,
-            COUNT(sc.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
@@ -159,17 +164,30 @@ async function createInstanceKeyPathsFromInstanceLabel(props) {
         )`,
         ],
         ecsql: `
+      SELECT * FROM (
         SELECT
           c.ClassName AS CategoryClass,
           c.ECInstanceId AS CategoryId,
-          IIF(c.ChildCount > 1, sc.ClassName, NULL) AS SubcategoryClass,
-          IIF(c.ChildCount > 1, sc.ECInstanceId, NULL) AS SubcategoryId
-        FROM RootCategories c
+          sc.ClassName AS SubcategoryClass,
+          sc.ECInstanceId AS SubcategoryId
+        FROM RootCategoriesWithLabels c
         JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId
-        WHERE sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\'
+        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 '\\'
+        )
         LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}
       `,
-        bindings: [{ type: "string", value: props.label.replace(/[%_\\]/g, "\\$&") }],
+        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) {

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

@@ -1 +1 @@
-{"version":3,"file":"CategoriesTreeDefinition.js","sourceRoot":"","sources":["../../../../../src/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,sBAAsB,CAAC;AAchE,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,MAAM,GAAG,KAAK,CAAC,YAAY,CAAC,iBAAiB,CACjD;QACE,IAAI,EAAE;YACJ;;;;;iBAKS,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;;;;;;;;;gBASG,gCAAgC,GAAG,CAAC;OAC7C;QACD,QAAQ,EAAE,CAAC,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,KAAK,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,EAAE,MAAM,CAAC,EAAE,CAAC;KAC9E,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\";\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 reader = props.imodelAccess.createQueryReader(\n    {\n      ctes: [\n        `RootCategories(ClassName, ECInstanceId, ChildCount) as (\n          SELECT\n            ec_classname(this.ECClassId, 's.c'),\n            this.ECInstanceId,\n            COUNT(sc.ECInstanceId)\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\n          c.ClassName AS CategoryClass,\n          c.ECInstanceId AS CategoryId,\n          IIF(c.ChildCount > 1, sc.ClassName, NULL) AS SubcategoryClass,\n          IIF(c.ChildCount > 1, sc.ECInstanceId, NULL) AS SubcategoryId\n        FROM RootCategories c\n        JOIN SubCategoriesWithLabels sc ON sc.ParentId = c.ECInstanceId\n        WHERE sc.DisplayLabel LIKE '%' || ? || '%' ESCAPE '\\\\'\n        LIMIT ${MAX_FILTERING_INSTANCE_KEY_COUNT + 1}\n      `,\n      bindings: [{ type: \"string\", value: props.label.replace(/[%_\\\\]/g, \"\\\\$&\") }],\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/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,sBAAsB,CAAC;AAchE,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\";\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"]}

package/lib/esm/components/trees/common/components/Tree.js

@@ -46,7 +46,11 @@ function TreeImpl({ imodel, imodelAccess, treeName, noDataMessage, getFilteredPa
             onPerformanceMeasured(action, duration);
         },
         onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: "hierarchy-level-size-limit-hit", reportInteraction: false }),
-        onHierarchyLoadError: ({ type }) => onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false }),
+        onHierarchyLoadError: ({ type, error }) => {
+            // eslint-disable-next-line no-console
+            console.error(error);
+            onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });
+        },
     });
     useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });
     const selectNodes = useSelectionPredicate({

package/lib/esm/components/trees/common/components/Tree.js.map

@@ -1 +1 @@
-{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../src/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AACrI,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAkDpD;;;GAGG;AACH,MAAM,UAAU,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,KAAC,6BAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,KAAC,wBAAwB,IAAC,OAAO,EAAE,gBAAgB,YACjD,KAAC,QAAQ,OAAK,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GAC1F,GACG,CACjC,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,OAAO,EACP,SAAS,GAC2H;IACpI,MAAM,gBAAgB,GAAG,0BAA0B,EAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,mBAAmB,EAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC,IAAI,OAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,6BAA6B,CAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;KAC5G,CAAC,CAAC;IACH,uBAAuB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,IAAI,QAAQ,EAAE,CAAC,CAAC;IACjI,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,0BAA0B,CAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,CACL,KAAC,IAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC/G,KAAC,OAAO,IAAC,IAAI,EAAE,IAAI,YACjB,KAAC,cAAc,IAAC,IAAI,EAAC,OAAO,GAAG,GACvB,GACL,CACR,CAAC;KACH;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,CACL,KAAC,IAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC9G,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,6BAA6B,CAAC,GAAQ,GAC9F,CACR,CAAC;KACH;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,WAAW;QACX,aAAa;QACb,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;QACR,IAAI,EAAE,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;KACnD,CAAC;IAEF,OAAO,CACL,eAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,eAAK,EAAE,EAAC,4BAA4B,EAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAC7E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,KAAC,OAAO,IAAC,IAAI,EAAE,SAAS,YACtB,KAAC,eAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,WAAW,CAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { Flex, ProgressRadial, Text } from \"@itwin/itwinui-react\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { UnifiedSelectionProvider, useIModelUnifiedSelectionTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\nimport { useHierarchiesLocalization } from \"../UseHierarchiesLocalization\";\nimport { useHierarchyLevelFiltering } from \"../UseHierarchyFiltering\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext\";\nimport { createIModelAccess } from \"../Utils\";\nimport { Delayed } from \"./Delayed\";\nimport { ProgressOverlay } from \"./ProgressOverlay\";\n\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting\";\nimport type { TreeRendererProps } from \"./TreeRenderer\";\n\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          TreeRendererProps,\n          \"rootNodes\" | \"expandNode\" | \"onNodeClick\" | \"onNodeKeyDown\" | \"onFilterClick\" | \"isNodeSelected\" | \"getHierarchyLevelDetails\" | \"size\" | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Modifies the density of tree nodes. `enlarged` tree nodes have bigger button hit boxes. */\n    density?: \"default\" | \"enlarged\";\n    /** Message that should be renderer if there are no tree nodes. */\n    noDataMessage?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @Beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <UnifiedSelectionProvider storage={selectionStorage}>\n        <TreeImpl {...props} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n      </UnifiedSelectionProvider>\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  noDataMessage,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  density,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\" | \"selectionStorage\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type }) => onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false }),\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { onNodeClick, onNodeKeyDown } = useSelectionHandler({ rootNodes, selectNodes, selectionMode: selectionMode ?? \"single\" });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        <Delayed show={true}>\n          <ProgressRadial size=\"large\" />\n        </Delayed>\n      </Flex>\n    );\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        {noDataMessage ? noDataMessage : <Text>{TreeWidget.translate(\"baseTree.dataIsNotAvailable\")}</Text>}\n      </Flex>\n    );\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    onNodeClick,\n    onNodeKeyDown,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n    size: density === \"enlarged\" ? \"default\" : \"small\",\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div id=\"tw-tree-renderer-container\" style={{ overflow: \"auto\", height: \"100%\" }}>\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}
+{"version":3,"file":"Tree.js","sourceRoot":"","sources":["../../../../../../src/components/trees/common/components/Tree.tsx"],"names":[],"mappings":";AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,QAAQ,EAAE,MAAM,OAAO,CAAC;AACvD,OAAO,EAAE,OAAO,EAAE,MAAM,qBAAqB,CAAC;AAC9C,OAAO,EAAE,IAAI,EAAE,cAAc,EAAE,IAAI,EAAE,MAAM,sBAAsB,CAAC;AAClE,OAAO,EAAE,6BAA6B,EAAE,MAAM,gCAAgC,CAAC;AAC/E,OAAO,EAAE,wBAAwB,EAAE,6BAA6B,EAAE,mBAAmB,EAAE,MAAM,uCAAuC,CAAC;AACrI,OAAO,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AACpD,OAAO,EAAE,0BAA0B,EAAE,MAAM,+BAA+B,CAAC;AAC3E,OAAO,EAAE,0BAA0B,EAAE,MAAM,0BAA0B,CAAC;AACtE,OAAO,EAAE,uBAAuB,EAAE,MAAM,4BAA4B,CAAC;AACrE,OAAO,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAC7D,OAAO,EAAE,kBAAkB,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AACjF,OAAO,EAAE,kBAAkB,EAAE,MAAM,UAAU,CAAC;AAC9C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AACpC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAkDpD;;;GAGG;AACH,MAAM,UAAU,IAAI,CAAC,EAAE,gBAAgB,EAAE,uBAAuB,EAAE,gBAAgB,EAAE,YAAY,EAAE,oBAAoB,EAAE,GAAG,KAAK,EAAa;IAC3I,MAAM,8BAA8B,GAAG,uBAAuB,IAAI,IAAI,CAAC;IAEvE,MAAM,YAAY,GAAG,OAAO,CAAC,GAAG,EAAE;QAChC,OAAO,oBAAoB,IAAI,kBAAkB,CAAC,EAAE,gBAAgB,EAAE,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC;IAChG,CAAC,EAAE,CAAC,oBAAoB,EAAE,gBAAgB,EAAE,KAAK,CAAC,MAAM,CAAC,CAAC,CAAC;IAE3D,OAAO,CACL,KAAC,6BAA6B,IAAC,MAAM,EAAE,KAAK,CAAC,MAAM,EAAE,qBAAqB,EAAE,gBAAgB,YAC1F,KAAC,wBAAwB,IAAC,OAAO,EAAE,gBAAgB,YACjD,KAAC,QAAQ,OAAK,KAAK,EAAE,YAAY,EAAE,YAAY,EAAE,8BAA8B,EAAE,8BAA8B,GAAI,GAC1F,GACG,CACjC,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,EAChB,MAAM,EACN,YAAY,EACZ,QAAQ,EACR,aAAa,EACb,gBAAgB,EAChB,8BAA8B,EAC9B,sBAAsB,EACtB,kBAAkB,EAClB,aAAa,EACb,QAAQ,EACR,YAAY,EACZ,OAAO,EACP,SAAS,GAC2H;IACpI,MAAM,gBAAgB,GAAG,0BAA0B,EAAE,CAAC;IACtD,MAAM,EAAE,aAAa,EAAE,qBAAqB,EAAE,GAAG,mBAAmB,EAAE,CAAC;IACvE,MAAM,CAAC,aAAa,CAAC,GAAG,QAAQ,CAAC,IAAI,OAAO,EAAc,CAAC,CAAC;IAC5D,MAAM,EACJ,SAAS,EACT,OAAO,EACP,SAAS,EACT,WAAW,EAAE,iBAAiB,EAC9B,YAAY,EAAE,aAAa,EAC3B,UAAU,EACV,GAAG,SAAS,EACb,GAAG,6BAA6B,CAAC;QAChC,YAAY;QACZ,aAAa;QACb,sBAAsB;QACtB,gBAAgB;QAChB,UAAU,EAAE,QAAQ;QACpB,gBAAgB;QAChB,qBAAqB,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,EAAE;YAC1C,IAAI,MAAM,KAAK,QAAQ,EAAE;gBACvB,QAAQ,EAAE,EAAE,CAAC;aACd;YACD,qBAAqB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QACD,wBAAwB,EAAE,GAAG,EAAE,CAAC,aAAa,CAAC,EAAE,SAAS,EAAE,gCAAgC,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC;QACxH,oBAAoB,EAAE,CAAC,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE;YACxC,sCAAsC;YACtC,OAAO,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;YACrB,aAAa,CAAC,EAAE,SAAS,EAAE,SAAS,IAAI,EAAE,EAAE,iBAAiB,EAAE,KAAK,EAAE,CAAC,CAAC;QAC1E,CAAC;KACF,CAAC,CAAC;IACH,uBAAuB,CAAC,EAAE,MAAM,EAAE,MAAM,EAAE,WAAW,CAAC,GAAG,EAAE,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,aAAa,CAAC,CAAC,EAAE,CAAC,CAAC;IAE5G,MAAM,WAAW,GAAG,qBAAqB,CAAC;QACxC,MAAM,EAAE,kBAAkB,CAAC,EAAE,MAAM,EAAE,iBAAiB,EAAE,CAAC;QACzD,SAAS,EAAE,kBAAkB;QAC7B,OAAO;KACR,CAAC,CAAC;IACH,MAAM,EAAE,WAAW,EAAE,aAAa,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,aAAa,EAAE,aAAa,IAAI,QAAQ,EAAE,CAAC,CAAC;IACjI,MAAM,EAAE,eAAe,EAAE,aAAa,EAAE,GAAG,0BAA0B,CAAC;QACpE,MAAM;QACN,8BAA8B;KAC/B,CAAC,CAAC;IACH,MAAM,mBAAmB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;IACvE,MAAM,wBAAwB,GAAG,kBAAkB,CAAC,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;IAC/E,MAAM,EAAE,QAAQ,EAAE,GAAG,mBAAmB,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC,CAAC;IAEnE,IAAI,SAAS,KAAK,SAAS,EAAE;QAC3B,OAAO,CACL,KAAC,IAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC/G,KAAC,OAAO,IAAC,IAAI,EAAE,IAAI,YACjB,KAAC,cAAc,IAAC,IAAI,EAAC,OAAO,GAAG,GACvB,GACL,CACR,CAAC;KACH;IAED,IAAI,SAAS,CAAC,MAAM,KAAK,CAAC,IAAI,CAAC,SAAS,EAAE;QACxC,OAAO,CACL,KAAC,IAAI,IAAC,UAAU,EAAC,QAAQ,EAAC,cAAc,EAAC,QAAQ,EAAC,aAAa,EAAC,QAAQ,EAAC,KAAK,EAAE,EAAE,KAAK,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,YAC9G,aAAa,CAAC,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC,KAAC,IAAI,cAAE,UAAU,CAAC,SAAS,CAAC,6BAA6B,CAAC,GAAQ,GAC9F,CACR,CAAC;KACH;IAED,MAAM,iBAAiB,GAA6C;QAClE,GAAG,SAAS;QACZ,SAAS;QACT,WAAW;QACX,aAAa;QACb,UAAU,EAAE,mBAAmB;QAC/B,aAAa,EAAE,wBAAwB;QACvC,QAAQ;QACR,IAAI,EAAE,OAAO,KAAK,UAAU,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;KACnD,CAAC;IAEF,OAAO,CACL,eAAK,KAAK,EAAE,EAAE,QAAQ,EAAE,UAAU,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,EAAE,QAAQ,EAAE,aACtE,eAAK,EAAE,EAAC,4BAA4B,EAAC,KAAK,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,aAC7E,YAAY,CAAC,iBAAiB,CAAC,EAC/B,eAAe,IACZ,EACN,KAAC,OAAO,IAAC,IAAI,EAAE,SAAS,YACtB,KAAC,eAAe,KAAG,GACX,IACN,CACP,CAAC;AACJ,CAAC;AAED,SAAS,qBAAqB,CAAC,EAC7B,MAAM,EACN,SAAS,EACT,OAAO,GAKR;IACC,OAAO,WAAW,CAChB,CAAC,OAAO,EAAE,UAAU,EAAE,EAAE,CACtB,MAAM,CACJ,OAAO,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,EAAE;QACxB,MAAM,IAAI,GAAG,OAAO,CAAC,MAAM,CAAC,CAAC;QAC7B,OAAO,IAAI,IAAI,CAAC,CAAC,SAAS,IAAI,SAAS,CAAC,IAAI,CAAC,CAAC,CAAC;IACjD,CAAC,CAAC,EACF,UAAU,CACX,EACH,CAAC,MAAM,EAAE,OAAO,EAAE,SAAS,CAAC,CAC7B,CAAC;AACJ,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport { useCallback, useMemo, useState } from \"react\";\nimport { BeEvent } from \"@itwin/core-bentley\";\nimport { Flex, ProgressRadial, Text } from \"@itwin/itwinui-react\";\nimport { SchemaMetadataContextProvider } from \"@itwin/presentation-components\";\nimport { UnifiedSelectionProvider, useIModelUnifiedSelectionTree, useSelectionHandler } from \"@itwin/presentation-hierarchies-react\";\nimport { TreeWidget } from \"../../../../TreeWidget\";\nimport { useHierarchiesLocalization } from \"../UseHierarchiesLocalization\";\nimport { useHierarchyLevelFiltering } from \"../UseHierarchyFiltering\";\nimport { useIModelChangeListener } from \"../UseIModelChangeListener\";\nimport { useNodeHighlighting } from \"../UseNodeHighlighting\";\nimport { useReportingAction, useTelemetryContext } from \"../UseTelemetryContext\";\nimport { createIModelAccess } from \"../Utils\";\nimport { Delayed } from \"./Delayed\";\nimport { ProgressOverlay } from \"./ProgressOverlay\";\n\nimport type { MarkRequired } from \"@itwin/core-bentley\";\nimport type { FunctionProps } from \"../Utils\";\nimport type { ReactNode } from \"react\";\nimport type { IModelConnection } from \"@itwin/core-frontend\";\nimport type { SchemaContext } from \"@itwin/ecschema-metadata\";\nimport type { PresentationHierarchyNode, SelectionStorage, useIModelTree } from \"@itwin/presentation-hierarchies-react\";\nimport type { HighlightInfo } from \"../UseNodeHighlighting\";\nimport type { TreeRendererProps } from \"./TreeRenderer\";\n\n/** @beta */\nexport type TreeProps = Pick<FunctionProps<typeof useIModelTree>, \"getFilteredPaths\" | \"getHierarchyDefinition\"> &\n  Partial<Pick<FunctionProps<typeof useSelectionHandler>, \"selectionMode\">> & {\n    /** iModel connection that should be used to pull data from. */\n    imodel: IModelConnection;\n    /** Callback for getting `SchemaContext` for specific iModel. */\n    getSchemaContext: (imodel: IModelConnection) => SchemaContext;\n    /** Unique tree component name that will be used as unified selection change event source when selecting node. */\n    treeName: string;\n    /** Unified selection storage that should be used by tree to handle tree selection changes. */\n    selectionStorage: SelectionStorage;\n    /**\n     * An optional predicate to allow or prohibit selection of a node.\n     * When not supplied, all nodes are selectable.\n     */\n    selectionPredicate?: (node: PresentationHierarchyNode) => boolean;\n    /** Tree renderer that should be used to render tree data. */\n    treeRenderer: (\n      treeProps: Required<\n        Pick<\n          TreeRendererProps,\n          \"rootNodes\" | \"expandNode\" | \"onNodeClick\" | \"onNodeKeyDown\" | \"onFilterClick\" | \"isNodeSelected\" | \"getHierarchyLevelDetails\" | \"size\" | \"getLabel\"\n        >\n      >,\n    ) => ReactNode;\n    /** Custom iModel access that is stored outside tree component. If not provided it new iModel access will be created using `imodel` prop. */\n    imodelAccess?: FunctionProps<typeof useIModelTree>[\"imodelAccess\"];\n    /** Size limit that should be applied on each hierarchy level. Default to `1000`. */\n    hierarchyLevelSizeLimit?: number;\n    /** Modifies the density of tree nodes. `enlarged` tree nodes have bigger button hit boxes. */\n    density?: \"default\" | \"enlarged\";\n    /** Message that should be renderer if there are no tree nodes. */\n    noDataMessage?: ReactNode;\n    /** Callback that this invoked when tree reloads. */\n    onReload?: () => void;\n    /** Options for highlighting node labels. */\n    highlight?: HighlightInfo;\n  };\n\n/**\n * Default tree component that manages tree state and renders using supplied `treeRenderer`.\n * @Beta\n */\nexport function Tree({ getSchemaContext, hierarchyLevelSizeLimit, selectionStorage, imodelAccess: providedIModelAccess, ...props }: TreeProps) {\n  const defaultHierarchyLevelSizeLimit = hierarchyLevelSizeLimit ?? 1000;\n\n  const imodelAccess = useMemo(() => {\n    return providedIModelAccess ?? createIModelAccess({ getSchemaContext, imodel: props.imodel });\n  }, [providedIModelAccess, getSchemaContext, props.imodel]);\n\n  return (\n    <SchemaMetadataContextProvider imodel={props.imodel} schemaContextProvider={getSchemaContext}>\n      <UnifiedSelectionProvider storage={selectionStorage}>\n        <TreeImpl {...props} imodelAccess={imodelAccess} defaultHierarchyLevelSizeLimit={defaultHierarchyLevelSizeLimit} />\n      </UnifiedSelectionProvider>\n    </SchemaMetadataContextProvider>\n  );\n}\n\nfunction TreeImpl({\n  imodel,\n  imodelAccess,\n  treeName,\n  noDataMessage,\n  getFilteredPaths,\n  defaultHierarchyLevelSizeLimit,\n  getHierarchyDefinition,\n  selectionPredicate,\n  selectionMode,\n  onReload,\n  treeRenderer,\n  density,\n  highlight,\n}: MarkRequired<Omit<TreeProps, \"getSchemaContext\" | \"selectionStorage\">, \"imodelAccess\"> & { defaultHierarchyLevelSizeLimit: number }) {\n  const localizedStrings = useHierarchiesLocalization();\n  const { onFeatureUsed, onPerformanceMeasured } = useTelemetryContext();\n  const [imodelChanged] = useState(new BeEvent<() => void>());\n  const {\n    rootNodes,\n    getNode,\n    isLoading,\n    selectNodes: selectNodesAction,\n    setFormatter: _setFormatter,\n    expandNode,\n    ...treeProps\n  } = useIModelUnifiedSelectionTree({\n    imodelAccess,\n    imodelChanged,\n    getHierarchyDefinition,\n    getFilteredPaths,\n    sourceName: treeName,\n    localizedStrings,\n    onPerformanceMeasured: (action, duration) => {\n      if (action === \"reload\") {\n        onReload?.();\n      }\n      onPerformanceMeasured(action, duration);\n    },\n    onHierarchyLimitExceeded: () => onFeatureUsed({ featureId: \"hierarchy-level-size-limit-hit\", reportInteraction: false }),\n    onHierarchyLoadError: ({ type, error }) => {\n      // eslint-disable-next-line no-console\n      console.error(error);\n      onFeatureUsed({ featureId: `error-${type}`, reportInteraction: false });\n    },\n  });\n  useIModelChangeListener({ imodel, action: useCallback(() => imodelChanged.raiseEvent(), [imodelChanged]) });\n\n  const selectNodes = useSelectionPredicate({\n    action: useReportingAction({ action: selectNodesAction }),\n    predicate: selectionPredicate,\n    getNode,\n  });\n  const { onNodeClick, onNodeKeyDown } = useSelectionHandler({ rootNodes, selectNodes, selectionMode: selectionMode ?? \"single\" });\n  const { filteringDialog, onFilterClick } = useHierarchyLevelFiltering({\n    imodel,\n    defaultHierarchyLevelSizeLimit,\n  });\n  const reportingExpandNode = useReportingAction({ action: expandNode });\n  const reportingOnFilterClicked = useReportingAction({ action: onFilterClick });\n  const { getLabel } = useNodeHighlighting({ rootNodes, highlight });\n\n  if (rootNodes === undefined) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        <Delayed show={true}>\n          <ProgressRadial size=\"large\" />\n        </Delayed>\n      </Flex>\n    );\n  }\n\n  if (rootNodes.length === 0 && !isLoading) {\n    return (\n      <Flex alignItems=\"center\" justifyContent=\"center\" flexDirection=\"column\" style={{ width: \"100%\", height: \"100%\" }}>\n        {noDataMessage ? noDataMessage : <Text>{TreeWidget.translate(\"baseTree.dataIsNotAvailable\")}</Text>}\n      </Flex>\n    );\n  }\n\n  const treeRendererProps: FunctionProps<TreeProps[\"treeRenderer\"]> = {\n    ...treeProps,\n    rootNodes,\n    onNodeClick,\n    onNodeKeyDown,\n    expandNode: reportingExpandNode,\n    onFilterClick: reportingOnFilterClicked,\n    getLabel,\n    size: density === \"enlarged\" ? \"default\" : \"small\",\n  };\n\n  return (\n    <div style={{ position: \"relative\", height: \"100%\", overflow: \"hidden\" }}>\n      <div id=\"tw-tree-renderer-container\" style={{ overflow: \"auto\", height: \"100%\" }}>\n        {treeRenderer(treeRendererProps)}\n        {filteringDialog}\n      </div>\n      <Delayed show={isLoading}>\n        <ProgressOverlay />\n      </Delayed>\n    </div>\n  );\n}\n\nfunction useSelectionPredicate({\n  action,\n  predicate,\n  getNode,\n}: {\n  action: (...args: any[]) => void;\n  predicate?: (node: PresentationHierarchyNode) => boolean;\n  getNode: (nodeId: string) => PresentationHierarchyNode | undefined;\n}): ReturnType<typeof useIModelUnifiedSelectionTree>[\"selectNodes\"] {\n  return useCallback(\n    (nodeIds, changeType) =>\n      action(\n        nodeIds.filter((nodeId) => {\n          const node = getNode(nodeId);\n          return node && (!predicate || predicate(node));\n        }),\n        changeType,\n      ),\n    [action, getNode, predicate],\n  );\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@itwin/tree-widget-react",
-  "version": "3.2.2",
+  "version": "3.2.4",
   "description": "Tree Widget React",
   "keywords": [
     "Bentley",
@@ -64,7 +64,7 @@
     "@itwin/itwinui-illustrations-react": "^2.1.0",
     "@itwin/presentation-core-interop": "^1.1.2",
     "@itwin/presentation-hierarchies": "^1.4.0",
-    "@itwin/presentation-hierarchies-react": "^1.3.0",
+    "@itwin/presentation-hierarchies-react": "^1.4.0",
     "@itwin/presentation-shared": "^1.2.0",
     "@itwin/unified-selection": "^1.2.0",
     "classnames": "^2.5.1",