@itwin/presentation-hierarchies 1.0.0 → 1.2.0

package/lib/esm/hierarchies/imodel/IModelHierarchyProvider.js

@@ -5,26 +5,26 @@
 import { catchError, concat, defaultIfEmpty, defer, EMPTY, filter, finalize, from, map, merge, mergeAll, mergeMap, of, Subject, take, takeUntil, tap, } from "rxjs";
 import { assert, BeEvent, Guid, StopWatch } from "@itwin/core-bentley";
 import { createDefaultValueFormatter, formatConcatenatedValue, normalizeFullClassName, } from "@itwin/presentation-shared";
-import { RowsLimitExceededError } from "../HierarchyErrors";
-import { HierarchyNode } from "../HierarchyNode";
-import { HierarchyNodeKey } from "../HierarchyNodeKey";
-import { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_INTERNAL as BASE_LOGGING_NAMESPACE_INTERNAL, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE, createNodeIdentifierForLogging, hasChildren, } from "../internal/Common";
-import { eachValueFrom } from "../internal/EachValueFrom";
-import { doLog, log } from "../internal/LoggingUtils";
-import { partition } from "../internal/operators/Partition";
-import { reduceToMergeMapList } from "../internal/operators/ReduceToMergeMap";
-import { shareReplayWithErrors } from "../internal/operators/ShareReplayWithErrors";
-import { sortNodesByLabelOperator } from "../internal/operators/Sorting";
-import { SubscriptionScheduler } from "../internal/SubscriptionScheduler";
-import { FilteringHierarchyDefinition } from "./FilteringHierarchyDefinition";
-import { HierarchyCache } from "./HierarchyCache";
-import { HierarchyNodesDefinition } from "./IModelHierarchyDefinition";
-import { NodeSelectClauseColumnNames } from "./NodeSelectQueryFactory";
-import { createDetermineChildrenOperator } from "./operators/DetermineChildren";
-import { createGroupingOperator } from "./operators/Grouping";
-import { createHideIfNoChildrenOperator } from "./operators/HideIfNoChildren";
-import { createHideNodesInHierarchyOperator } from "./operators/HideNodesInHierarchy";
-import { readNodes } from "./TreeNodesReader";
+import { RowsLimitExceededError } from "../HierarchyErrors.js";
+import { HierarchyNode } from "../HierarchyNode.js";
+import { HierarchyNodeKey } from "../HierarchyNodeKey.js";
+import { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_INTERNAL as BASE_LOGGING_NAMESPACE_INTERNAL, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE, createNodeIdentifierForLogging, hasChildren, } from "../internal/Common.js";
+import { eachValueFrom } from "../internal/EachValueFrom.js";
+import { doLog, log } from "../internal/LoggingUtils.js";
+import { partition } from "../internal/operators/Partition.js";
+import { reduceToMergeMapList } from "../internal/operators/ReduceToMergeMap.js";
+import { shareReplayWithErrors } from "../internal/operators/ShareReplayWithErrors.js";
+import { sortNodesByLabelOperator } from "../internal/operators/Sorting.js";
+import { SubscriptionScheduler } from "../internal/SubscriptionScheduler.js";
+import { FilteringHierarchyDefinition } from "./FilteringHierarchyDefinition.js";
+import { HierarchyCache } from "./HierarchyCache.js";
+import { HierarchyNodesDefinition } from "./IModelHierarchyDefinition.js";
+import { NodeSelectClauseColumnNames } from "./NodeSelectQueryFactory.js";
+import { createDetermineChildrenOperator } from "./operators/DetermineChildren.js";
+import { createGroupingOperator } from "./operators/Grouping.js";
+import { createHideIfNoChildrenOperator } from "./operators/HideIfNoChildren.js";
+import { createHideNodesInHierarchyOperator } from "./operators/HideNodesInHierarchy.js";
+import { readNodes } from "./TreeNodesReader.js";
 const LOGGING_NAMESPACE = `${BASE_LOGGING_NAMESPACE}.IModelHierarchyProvider`;
 const LOGGING_NAMESPACE_INTERNAL = `${BASE_LOGGING_NAMESPACE_INTERNAL}.IModelHierarchyProvider`;
 const LOGGING_NAMESPACE_PERFORMANCE = `${BASE_LOGGING_NAMESPACE_PERFORMANCE}.IModelHierarchyProvider`;
@@ -34,7 +34,7 @@ const DEFAULT_QUERY_CACHE_SIZE = 1;
  * Creates an instance of `HierarchyProvider` that creates a hierarchy based on given iModel and
  * a hierarchy definition, which defines each hierarchy level through ECSQL queries.
  *
- * @beta
+ * @public
  */
 export function createIModelHierarchyProvider(props) {
     return new IModelHierarchyProviderImpl(props);
@@ -81,7 +81,7 @@ class IModelHierarchyProviderImpl {
     invalidateHierarchyCache(reason) {
         doLog({
             category: `${LOGGING_NAMESPACE}.Events`,
-            message: /* istanbul ignore next */ () => (reason ? `${reason}: clear hierarchy cache` : `Clear hierarchy cache`),
+            message: /* c8 ignore next */ () => (reason ? `${reason}: clear hierarchy cache` : `Clear hierarchy cache`),
         });
         this._nodesCache?.clear();
     }
@@ -115,7 +115,7 @@ class IModelHierarchyProviderImpl {
         const { requestContext, ...defineHierarchyLevelProps } = props;
         doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Requesting hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${requestContext.requestId}] Requesting hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,
         });
         return from(defer(async () => {
             let parentNode = props.parentNode;
@@ -138,7 +138,7 @@ class IModelHierarchyProviderImpl {
             return this._activeHierarchyDefinition.defineHierarchyLevel({ ...defineHierarchyLevelProps, parentNode });
         })).pipe(mergeAll(), finalize(() => doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Received all hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${requestContext.requestId}] Received all hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,
         })));
     }
     createSourceNodesObservable(props) {
@@ -147,13 +147,20 @@ class IModelHierarchyProviderImpl {
             if (HierarchyNodesDefinition.isGenericNode(def)) {
                 return of({ ...def.node, key: { type: "generic", id: def.node.key, source: this._imodelAccess.imodelKey } });
             }
-            return this._queryScheduler.scheduleSubscription(of(def.query).pipe(map((query) => filterQueryByInstanceKeys(query, props.filteredInstanceKeys)), mergeMap((query) => readNodes({ queryExecutor: this._imodelAccess, query, limit: props.hierarchyLevelSizeLimit, parser: this._activeHierarchyDefinition.parseNode })), map((node) => ({
+            return this._queryScheduler.scheduleSubscription(of(def.query).pipe(map((query) => filterQueryByInstanceKeys(query, props.filteredInstanceKeys)), mergeMap((query) => readNodes({
+                queryExecutor: this._imodelAccess,
+                query,
+                limit: props.hierarchyLevelSizeLimit,
+                parser: this._activeHierarchyDefinition.parseNode
+                    ? async (row) => this._activeHierarchyDefinition.parseNode(row, props.parentNode)
+                    : undefined,
+            })), map((node) => ({
                 ...node,
                 key: { ...node.key, instanceKeys: node.key.instanceKeys.map((key) => ({ ...key, imodelKey: this._imodelAccess.imodelKey })) },
             }))));
         }), finalize(() => doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Read all child nodes ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Read all child nodes ${createNodeIdentifierForLogging(props.parentNode)}`,
         })), shareReplayWithErrors());
     }
     createPreProcessedNodesObservable(queryNodesObservable, props) {
@@ -171,13 +178,13 @@ class IModelHierarchyProviderImpl {
         // note: for child nodes created because of hidden parent, we want to use parent's request props (instance filter, limit)
         (n) => this.getChildNodesObservables({ ...props, parentNode: n }).processedNodes, false), finalize(() => doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Finished pre-processing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Finished pre-processing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
         })));
     }
     createProcessedNodesObservable(preprocessedNodesObservable, props) {
         return preprocessedNodesObservable.pipe(createGroupingOperator(this._imodelAccess, props.parentNode, this._valuesFormatter, this._localizedStrings, (gn) => this.onGroupingNodeCreated(gn, props)), finalize(() => doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Finished grouping child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Finished grouping child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
         })));
     }
     createFinalizedNodesObservable(processedNodesObservable, props) {
@@ -190,34 +197,33 @@ class IModelHierarchyProviderImpl {
             });
         }), takeUntil(this._dispose), finalize(() => doLog({
             category: LOGGING_NAMESPACE_PERFORMANCE,
-            message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Finished finalizing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Finished finalizing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
         })));
     }
     createHasNodesObservable(preprocessedNodesObservable, possiblyKnownChildrenObservable, props) {
         const loggingCategory = `${LOGGING_NAMESPACE_INTERNAL}.HasNodes`;
         return concat((possiblyKnownChildrenObservable ?? EMPTY).pipe(filter((n) => hasChildren(n))), preprocessedNodesObservable).pipe(log({
             category: loggingCategory,
-            message: /* istanbul ignore next */ (n) => `[${props.requestContext.requestId}] Node before mapping to 'true': ${createNodeIdentifierForLogging(n)}`,
+            message: /* c8 ignore next */ (n) => `[${props.requestContext.requestId}] Node before mapping to 'true': ${createNodeIdentifierForLogging(n)}`,
         }), take(1), map(() => true), defaultIfEmpty(false), catchError((e) => {
             doLog({
                 category: loggingCategory,
-                message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Error while determining children: ${e.message}`,
+                message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Error while determining children: ${e.message}`,
             });
             if (e instanceof RowsLimitExceededError) {
                 return of(true);
             }
             throw e;
-        }), log({ category: loggingCategory, message: /* istanbul ignore next */ (r) => `[${props.requestContext.requestId}] Result: ${r}` }));
+        }), log({ category: loggingCategory, message: /* c8 ignore next */ (r) => `[${props.requestContext.requestId}] Result: ${r}` }));
     }
     getCachedObservableEntry(props) {
         const loggingCategory = `${LOGGING_NAMESPACE}.QueryResultsCache`;
         const { parentNode, ...restProps } = props;
         const cached = props.ignoreCache || !this._nodesCache ? undefined : this._nodesCache.get(props);
         if (cached) {
-            // istanbul ignore next
             doLog({
                 category: loggingCategory,
-                message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Found query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,
+                message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Found query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,
             });
             return cached;
         }
@@ -245,7 +251,7 @@ class IModelHierarchyProviderImpl {
         this._nodesCache?.set(nonGroupingNodeChildrenRequestProps, value);
         doLog({
             category: loggingCategory,
-            message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Saved query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,
+            message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Saved query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,
         });
         return value;
     }
@@ -282,7 +288,7 @@ class IModelHierarchyProviderImpl {
         let nodesCount = 0;
         doLog({
             category: loggingCategory,
-            message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Requesting child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${requestContext.requestId}] Requesting child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,
         });
         return eachValueFrom(this.getChildNodesObservables({ ...props, requestContext }).finalizedNodes.pipe(tap(() => ++nodesCount), catchError((e) => {
             error = e;
@@ -290,7 +296,8 @@ class IModelHierarchyProviderImpl {
         }), finalize(() => {
             doLog({
                 category: loggingCategory,
-                message: /* istanbul ignore next */ () => error
+                message: /* c8 ignore next */ () => 
+                /* c8 ignore next 3 */ error
                     ? `[${requestContext.requestId}] Error creating child nodes for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`
                     : `[${requestContext.requestId}] Returned ${nodesCount} child nodes for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,
             });
@@ -360,7 +367,7 @@ class IModelHierarchyProviderImpl {
         const timer = new StopWatch(undefined, true);
         doLog({
             category: loggingCategory,
-            message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Requesting keys for ${createNodeIdentifierForLogging(props.parentNode)}`,
+            message: /* c8 ignore next */ () => `[${requestContext.requestId}] Requesting keys for ${createNodeIdentifierForLogging(props.parentNode)}`,
         });
         let error;
         let keysCount = 0;
@@ -370,7 +377,8 @@ class IModelHierarchyProviderImpl {
         }), finalize(() => {
             doLog({
                 category: loggingCategory,
-                message: /* istanbul ignore next */ () => error
+                message: /* c8 ignore next */ () => 
+                /* c8 ignore next 3 */ error
                     ? `[${requestContext.requestId}] Error creating node instance keys for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`
                     : `[${requestContext.requestId}] Returned ${keysCount} instance keys for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,
             });
@@ -410,7 +418,6 @@ function filterQueryByInstanceKeys(query, filteredInstanceKeys) {
         return query;
     }
     const MAX_ALLOWED_BINDINGS = 1000;
-    // istanbul ignore else
     if (filteredInstanceKeys.length < MAX_ALLOWED_BINDINGS) {
         return {
             ...query,
@@ -422,7 +429,7 @@ function filterQueryByInstanceKeys(query, filteredInstanceKeys) {
             bindings: [...(query.bindings ?? []), ...filteredInstanceKeys.map((k) => ({ type: "id", value: k.id }))],
         };
     }
-    // istanbul ignore next
+    /* c8 ignore start */
     return {
         ...query,
         ecsql: `
@@ -432,5 +439,6 @@ function filterQueryByInstanceKeys(query, filteredInstanceKeys) {
     `,
         bindings: [...(query.bindings ?? []), { type: "idset", value: filteredInstanceKeys.map((k) => k.id) }],
     };
+    /* c8 ignore end */
 }
 //# sourceMappingURL=IModelHierarchyProvider.js.map

package/lib/esm/hierarchies/imodel/IModelHierarchyProvider.js.map

@@ -1 +1 @@
-{"version":3,"file":"IModelHierarchyProvider.js","sourceRoot":"","sources":["../../../../src/hierarchies/imodel/IModelHierarchyProvider.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,UAAU,EACV,MAAM,EACN,cAAc,EACd,KAAK,EACL,KAAK,EACL,MAAM,EACN,QAAQ,EACR,IAAI,EACJ,GAAG,EACH,KAAK,EACL,QAAQ,EACR,QAAQ,EAGR,EAAE,EACF,OAAO,EACP,IAAI,EACJ,SAAS,EACT,GAAG,GACJ,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAEL,2BAA2B,EAM3B,uBAAuB,EAGvB,sBAAsB,GACvB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAE5D,OAAO,EAAE,aAAa,EAAiD,MAAM,kBAAkB,CAAC;AAChG,OAAO,EAAkB,gBAAgB,EAAuC,MAAM,qBAAqB,CAAC;AAE5G,OAAO,EACL,iBAAiB,IAAI,sBAAsB,EAC3C,0BAA0B,IAAI,+BAA+B,EAC7D,6BAA6B,IAAI,kCAAkC,EACnE,8BAA8B,EAC9B,WAAW,GACZ,MAAM,oBAAoB,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,2BAA2B,CAAC;AAC1D,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,0BAA0B,CAAC;AACtD,OAAO,EAAE,SAAS,EAAE,MAAM,iCAAiC,CAAC;AAC5D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wCAAwC,CAAC;AAC9E,OAAO,EAAE,qBAAqB,EAAE,MAAM,6CAA6C,CAAC;AACpF,OAAO,EAAE,wBAAwB,EAAE,MAAM,+BAA+B,CAAC;AACzE,OAAO,EAAE,qBAAqB,EAAE,MAAM,mCAAmC,CAAC;AAC1E,OAAO,EAAE,4BAA4B,EAAE,MAAM,gCAAgC,CAAC;AAC9E,OAAO,EAAE,cAAc,EAAE,MAAM,kBAAkB,CAAC;AAClD,OAAO,EAAkD,wBAAwB,EAAE,MAAM,6BAA6B,CAAC;AAUvH,OAAO,EAAE,2BAA2B,EAAE,MAAM,0BAA0B,CAAC;AACvE,OAAO,EAAE,+BAA+B,EAAE,MAAM,+BAA+B,CAAC;AAChF,OAAO,EAAE,sBAAsB,EAAE,MAAM,sBAAsB,CAAC;AAC9D,OAAO,EAAE,8BAA8B,EAAE,MAAM,8BAA8B,CAAC;AAC9E,OAAO,EAAE,kCAAkC,EAAE,MAAM,kCAAkC,CAAC;AACtF,OAAO,EAAE,SAAS,EAAE,MAAM,mBAAmB,CAAC;AAE9C,MAAM,iBAAiB,GAAG,GAAG,sBAAsB,0BAA0B,CAAC;AAC9E,MAAM,0BAA0B,GAAG,GAAG,+BAA+B,0BAA0B,CAAC;AAChG,MAAM,6BAA6B,GAAG,GAAG,kCAAkC,0BAA0B,CAAC;AAEtG,MAAM,yBAAyB,GAAG,EAAE,CAAC;AACrC,MAAM,wBAAwB,GAAG,CAAC,CAAC;AAsEnC;;;;;GAKG;AACH,MAAM,UAAU,6BAA6B,CAAC,KAAmC;IAC/E,OAAO,IAAI,2BAA2B,CAAC,KAAK,CAAC,CAAC;AAChD,CAAC;AAQD,MAAM,2BAA2B;IACvB,aAAa,CAAe;IAC5B,cAAc,CAAoB;IAClC,gBAAgB,CAA2B;IAC3C,0BAA0B,CAAsB;IAChD,0BAA0B,CAAsB;IAChD,iBAAiB,CAA0C;IAC3D,eAAe,CAAwB;IACvC,WAAW,CAAuC;IAClD,YAAY,CAAc;IAC1B,QAAQ,GAAG,IAAI,OAAO,EAAQ,CAAC;IAEvC,YAAmB,KAAmC;QACpD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC,mBAAmB,CAAC;QAC9F,IAAI,CAAC,gBAAgB,GAAG,KAAK,EAAE,SAAS,IAAI,2BAA2B,EAAE,CAAC;QAC1E,IAAI,CAAC,iBAAiB,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,gBAAgB,EAAE,CAAC;QACtG,IAAI,CAAC,eAAe,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,gBAAgB,IAAI,yBAAyB,CAAC,CAAC;QACtG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEzC,MAAM,cAAc,GAAG,KAAK,CAAC,cAAc,IAAI,wBAAwB,CAAC;QACxE,IAAI,cAAc,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,WAAW,GAAG,IAAI,cAAc,CAAC;gBACpC,uFAAuF;gBACvF,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;gBACnC,eAAe,EAAE,CAAC;aACnB,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,aAAa,EAAE,WAAW,CAAC,GAAG,EAAE;gBACxD,IAAI,CAAC,wBAAwB,CAAC,qBAAqB,CAAC,CAAC;gBACrD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrB,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;IACxB,CAAC;IAED,IAAW,gBAAgB;QACzB,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAEO,wBAAwB,CAAC,MAAe;QAC9C,KAAK,CAAC;YACJ,QAAQ,EAAE,GAAG,iBAAiB,SAAS;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,yBAAyB,CAAC,CAAC,CAAC,uBAAuB,CAAC;SAClH,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACI,YAAY,CAAC,SAA+C;QACjE,IAAI,CAAC,gBAAgB,GAAG,SAAS,IAAI,2BAA2B,EAAE,CAAC;IACrE,CAAC;IAEM,kBAAkB,CAAC,KAAgD;QACxE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,IAAI,CAAC,0BAA0B,KAAK,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACxE,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,0BAA0B,CAAC;gBAClE,IAAI,CAAC,wBAAwB,CAAC,wBAAwB,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,0BAA0B,GAAG,IAAI,4BAA4B,CAAC;YACjE,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,MAAM,EAAE,IAAI,CAAC,0BAA0B;YACvC,mBAAmB,EAAE,KAAK,CAAC,KAAK;SACjC,CAAC,CAAC;QACH,IAAI,CAAC,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAEO,qBAAqB,CAAC,YAA4C,EAAE,KAA6B;QACvG,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,gBAAgB,EAAE,eAAe,EAAE,CAAC,CAAC;IAChJ,CAAC;IAEO,yCAAyC,CAAC,KAAqD;QACrG,MAAM,EAAE,cAAc,EAAE,GAAG,yBAAyB,EAAE,GAAG,KAAK,CAAC;QAC/D,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,cAAc,CAAC,SAAS,gDAAgD,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACjI,CAAC,CAAC;QACH,OAAO,IAAI,CACT,KAAK,CAAC,KAAK,IAAI,EAAE;YACf,IAAI,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YAClC,IAAI,UAAU,IAAI,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;gBACzI,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,IAAI,UAAU,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5D,UAAU,GAAG;oBACX,GAAG,UAAU;oBACb,GAAG,EAAE;wBACH,GAAG,UAAU,CAAC,GAAG;wBACjB,YAAY,EAAE,UAAU,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;qBACzH;iBACF,CAAC;gBACF,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;gBACrD,IAAI,UAAU,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7C,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC,0BAA0B,CAAC,oBAAoB,CAAC,EAAE,GAAG,yBAAyB,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5G,CAAC,CAAC,CACH,CAAC,IAAI,CACJ,QAAQ,EAAE,EACV,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,cAAc,CAAC,SAAS,kDAAkD,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACnI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,2BAA2B,CACjC,KAAgJ;QAEhJ,yDAAyD;QACzD,OAAO,IAAI,CAAC,yCAAyC,CAAC,KAAK,CAAC,CAAC,IAAI,CAC/D,QAAQ,CAAC,CAAC,GAAG,EAAwC,EAAE;YACrD,IAAI,wBAAwB,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChD,OAAO,EAAE,CAAC,EAAE,GAAG,GAAG,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAC/G,CAAC;YACD,OAAO,IAAI,CAAC,eAAe,CAAC,oBAAoB,CAC9C,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAChB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,yBAAyB,CAAC,KAAK,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC,EAC5E,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACjB,SAAS,CAAC,EAAE,aAAa,EAAE,IAAI,CAAC,aAAa,EAAE,KAAK,EAAE,KAAK,EAAE,KAAK,CAAC,uBAAuB,EAAE,MAAM,EAAE,IAAI,CAAC,0BAA0B,CAAC,SAAS,EAAE,CAAC,CACjJ,EACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACb,GAAG,IAAI;gBACP,GAAG,EAAE,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE;aAC9H,CAAC,CAAC,CACJ,CACF,CAAC;QACJ,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,0BAA0B,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACjH,CAAC,CACH,EACD,qBAAqB,EAAE,CACxB,CAAC;IACJ,CAAC;IAEO,iCAAiC,CACvC,oBAA2C,EAC3C,KAAkD;QAElD,OAAO,oBAAoB,CAAC,IAAI;QAC9B,8GAA8G;QAC9G,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QAC5B,0CAA0C;QAC1C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,wBAAwB,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC9F,uDAAuD;QACvD,QAAQ,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC5E,6CAA6C;QAC7C,eAAe,CAAC,IAAI,CAAC,0BAA0B,CAAC;QAChD,iBAAiB;QACjB,8BAA8B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,EACtI,kCAAkC;QAChC,yHAAyH;QACzH,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,cAAc,EAChF,KAAK,CACN,EACD,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,6CAA6C,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACpI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,8BAA8B,CACpC,2BAA+D,EAC/D,KAAkD;QAElD,OAAO,2BAA2B,CAAC,IAAI,CACrC,sBAAsB,CAAC,IAAI,CAAC,aAAa,EAAE,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE,CACjH,IAAI,CAAC,qBAAqB,CAAC,EAAE,EAAE,KAAK,CAAC,CACtC,EACD,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,uCAAuC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAC9H,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,8BAA8B,CACpC,wBAA4D,EAC5D,KAAkD;QAElD,OAAO,wBAAwB,CAAC,IAAI,CAClC,+BAA+B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,EACvI,gBAAgB,CAAC,IAAI,CAAC,0BAA0B,CAAC,EACjD,wBAAwB,EACxB,GAAG,CAAC,CAAC,CAAC,EAAiB,EAAE;YACvB,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,CAAC,gBAAgB,CAAC;YAC5B,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE;gBACtB,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,EACF,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EACxB,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,yCAAyC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAChI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAC9B,2BAA+D,EAC/D,+BAAkE,EAClE,KAAyB;QAEzB,MAAM,eAAe,GAAG,GAAG,0BAA0B,WAAW,CAAC;QACjE,OAAO,MAAM,CAAC,CAAC,+BAA+B,IAAI,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,2BAA2B,CAAC,CAAC,IAAI,CAC7H,GAAG,CAAC;YACF,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,0BAA0B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,oCAAoC,8BAA8B,CAAC,CAAC,CAAC,EAAE;SACrJ,CAAC,EACF,IAAI,CAAC,CAAC,CAAC,EACP,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EACf,cAAc,CAAC,KAAK,CAAC,EACrB,UAAU,CAAC,CAAC,CAAQ,EAAE,EAAE;YACtB,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,uCAAuC,CAAC,CAAC,OAAO,EAAE;aAC/H,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,sBAAsB,EAAE,CAAC;gBACxC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,GAAG,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,0BAA0B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,aAAa,CAAC,EAAE,EAAE,CAAC,CAClI,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAAC,KAAkD;QACjF,MAAM,eAAe,GAAG,GAAG,iBAAiB,oBAAoB,CAAC;QACjE,MAAM,EAAE,UAAU,EAAE,GAAG,SAAS,EAAE,GAAG,KAAK,CAAC;QAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChG,IAAI,MAAM,EAAE,CAAC;YACX,uBAAuB;YACvB,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,sCAAsC,8BAA8B,CAAC,UAAU,CAAC,EAAE;aACvH,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,uGAAuG;QACvG,0CAA0C;QAC1C,IAAI,oBAA+C,CAAC;QACpD,IAAI,qBAAgF,CAAC;QACrF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,aAAa,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7C,qBAAqB,GAAG,UAAU,CAAC,mBAAmB,CAAC;gBACvD,oBAAoB,GAAG,UAAU,CAAC,mBAAmB,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,iDAAiD;gBACjD,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzF,qBAAqB,GAAG,UAAU,CAAC;YACrC,CAAC;QACH,CAAC;QAED,MAAM,mCAAmC,GAAG;YAC1C,GAAG,SAAS;YACZ,UAAU,EAAE,qBAAqB;YACjC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;SACjE,CAAC;QACF,MAAM,KAAK,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,2BAA2B,CAAC,mCAAmC,CAAC,EAAE,gBAAgB,EAAE,MAAe,EAAE,CAAC;QACvI,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;QAClE,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,sCAAsC,8BAA8B,CAAC,UAAU,CAAC,EAAE;SACvH,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,wBAAwB,CAAC,KAAuG;QACtI,MAAM,KAAK,GAAG,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACnD,QAAQ,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC/B,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,MAAM,GAAG,GAAG,IAAI,CAAC,iCAAiC,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5E,MAAM,IAAI,GAAG,IAAI,CAAC,8BAA8B,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC7D,OAAO;oBACL,cAAc,EAAE,IAAI;oBACpB,QAAQ,EAAE,IAAI,CAAC,wBAAwB,CAAC,GAAG,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC;oBACrE,cAAc,EAAE,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC;iBACjE,CAAC;YACJ,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,IAAI,GAAG,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC1E,OAAO;oBACL,cAAc,EAAE,IAAI;oBACpB,QAAQ,EAAE,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC;oBAC3E,cAAc,EAAE,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC;iBACjE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACI,QAAQ,CAAC,KAA6B;QAC3C,MAAM,cAAc,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACzD,MAAM,eAAe,GAAG,GAAG,iBAAiB,WAAW,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7C,IAAI,KAAU,CAAC;QACf,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CAAC,IAAI,cAAc,CAAC,SAAS,gCAAgC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACzJ,CAAC,CAAC;QACH,OAAO,aAAa,CAClB,IAAI,CAAC,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,cAAc,CAAC,IAAI,CAC7E,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,UAAU,CAAC,EACvB,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE;YACf,KAAK,GAAG,CAAC,CAAC;YACV,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE;YACZ,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,KAAK;oBACH,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,oCAAoC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE;oBAClL,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,cAAc,UAAU,oBAAoB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK;aAC1K,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAAC,KAAuE;QACpG,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,uBAAuB,GAAG,WAAW,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC;QACpG,IAAI,UAAU,IAAI,aAAa,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3D,OAAO,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,CAAC,CAAC,UAAU,IAAI,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;QAExG,MAAM,yBAAyB,GAAG,IAAI,CAAC,yCAAyC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC,CAAC;QAEjI,yFAAyF;QACzF,MAAM,CAAC,WAAW,EAAE,YAAY,CAAC,GAAG,SAAS,CAAC,yBAAyB,EAAE,wBAAwB,CAAC,aAAa,CAAC,CAAC;QAEjH,+DAA+D;QAC/D,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CACpC,QAAQ,CAAC,CAAC,GAAG,EAAE,EAAE,CACf,IAAI,CAAC,eAAe,CAAC,oBAAoB,CACvC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAChB,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,KAAK,GAAG;;oBAER,2BAA2B,CAAC,aAAa;oBACzC,2BAA2B,CAAC,YAAY;oBACxC,2BAA2B,CAAC,mBAAmB;;oBAE/C,KAAK,CAAC,KAAK;;eAEhB,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YACnI,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACZ,GAAG,EAAE;oBACH,SAAS,EAAE,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;oBACV,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS;iBACZ;gBAC7B,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;aACf,CAAC,CAAC,CACJ,CAAC;QACJ,CAAC,CAAC,EACF,QAAQ,EAAE,CACX,CACF,CACF,CACF,CAAC;QACF,mFAAmF;QACnF,MAAM,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,GAAG,SAAS,CAAC,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QAEvG,0BAA0B;QAC1B,yHAAyH;QACzH,gIAAgI;QAChI,MAAM,iBAAiB,GAAG,KAAK,CAC7B,WAAW,CAAC,IAAI,CACd,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,CAAC,EAC7D,GAAG,CACD,CAAC,GAAG,EAAkB,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS;SACrC,CAAC,CACH,CACF,EACD,sBAAsB,CAAC,IAAI;QACzB,gCAAgC;QAChC,oBAAoB,CAClB,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,EAC1B,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CACpB;QACD,+CAA+C;QAC/C,QAAQ,CAAC,CAAC,SAAS,EAAsB,EAAE,CACzC,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YAClD,IAAI,EAAE,WAAW;YACjB,YAAY,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC;SAC5F,CAAC,CAAC,CACJ,CACF,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACZ,GAAG;YACH,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,EAAE;SACV,CAAC,CAAC,CACJ,CAAC;QAEF,0GAA0G;QAC1G,OAAO,KAAK,CACV,uBAAuB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EACnD,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAC1H,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACI,mBAAmB,CAAC,KAAkD;QAC3E,MAAM,cAAc,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACzD,MAAM,eAAe,GAAG,GAAG,iBAAiB,sBAAsB,CAAC;QACnE,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7C,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CAAC,IAAI,cAAc,CAAC,SAAS,yBAAyB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAClJ,CAAC,CAAC;QACH,IAAI,KAAU,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,OAAO,aAAa,CAClB,IAAI,CAAC,sBAAsB,CAAC,EAAE,GAAG,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,SAAS,CAAC,EACtB,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE;YACf,KAAK,GAAG,CAAC,CAAC;YACV,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE;YACZ,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CACvC,KAAK;oBACH,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,2CAA2C,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE;oBACzL,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,cAAc,SAAS,sBAAsB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK;aAC3K,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CACF,CAAC;IACJ,CAAC;CACF;AASD,SAAS,eAAe,CAAC,gBAAqC;IAC5D,OAAO,gBAAgB,CAAC,cAAc;QACpC,CAAC,CAAC,YAAY,CAAC,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC,CAA6E,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3F,CAAC;AAED,SAAS,gBAAgB,CAAyB,gBAAqC;IACrF,OAAO,gBAAgB,CAAC,eAAe;QACrC,CAAC,CAAC,YAAY,CAAC,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC,CAAqC,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,YAAY,CAAQ,SAAsD;IACjF,OAAO,CAAC,KAAwB,EAAE,EAAE,CAClC,KAAK,CAAC,IAAI,CACR,QAAQ,CAAC,SAAS,CAAC,EACnB,MAAM,CAAC,CAAC,CAAC,EAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC/B,CAAC;AACN,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,IAAW,EACX,cAAwC;IAExC,OAAO;QACL,GAAG,IAAI;QACP,KAAK,EAAE,MAAM,uBAAuB,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,cAAc;SACf,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,UAA2C;IAC3E,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,CAAC,GAAG,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAoB,EAAE,oBAA+C;IACtG,IAAI,CAAC,oBAAoB,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAC1D,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,oBAAoB,GAAG,IAAI,CAAC;IAClC,uBAAuB;IACvB,IAAI,oBAAoB,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;QACvD,OAAO;YACL,GAAG,KAAK;YACR,KAAK,EAAE;;gBAEG,KAAK,CAAC,KAAK;mCACQ,oBAAoB,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;OACzE;YACD,QAAQ,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;SACvH,CAAC;IACJ,CAAC;IACD,uBAAuB;IACvB,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE;;cAEG,KAAK,CAAC,KAAK;;KAEpB;QACD,QAAQ,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;KACvG,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 {\n  catchError,\n  concat,\n  defaultIfEmpty,\n  defer,\n  EMPTY,\n  filter,\n  finalize,\n  from,\n  map,\n  merge,\n  mergeAll,\n  mergeMap,\n  Observable,\n  ObservableInput,\n  of,\n  Subject,\n  take,\n  takeUntil,\n  tap,\n} from \"rxjs\";\nimport { assert, BeEvent, Guid, StopWatch } from \"@itwin/core-bentley\";\nimport {\n  ConcatenatedValue,\n  createDefaultValueFormatter,\n  ECClassHierarchyInspector,\n  ECSchemaProvider,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  Event,\n  formatConcatenatedValue,\n  InstanceKey,\n  IPrimitiveValueFormatter,\n  normalizeFullClassName,\n} from \"@itwin/presentation-shared\";\nimport { RowsLimitExceededError } from \"../HierarchyErrors\";\nimport { HierarchyFilteringPath } from \"../HierarchyFiltering\";\nimport { HierarchyNode, NonGroupingHierarchyNode, ParentHierarchyNode } from \"../HierarchyNode\";\nimport { GenericNodeKey, HierarchyNodeKey, IModelInstanceKey, InstancesNodeKey } from \"../HierarchyNodeKey\";\nimport { GetHierarchyNodesProps, HierarchyProvider } from \"../HierarchyProvider\";\nimport {\n  LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE,\n  LOGGING_NAMESPACE_INTERNAL as BASE_LOGGING_NAMESPACE_INTERNAL,\n  LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE,\n  createNodeIdentifierForLogging,\n  hasChildren,\n} from \"../internal/Common\";\nimport { eachValueFrom } from \"../internal/EachValueFrom\";\nimport { doLog, log } from \"../internal/LoggingUtils\";\nimport { partition } from \"../internal/operators/Partition\";\nimport { reduceToMergeMapList } from \"../internal/operators/ReduceToMergeMap\";\nimport { shareReplayWithErrors } from \"../internal/operators/ShareReplayWithErrors\";\nimport { sortNodesByLabelOperator } from \"../internal/operators/Sorting\";\nimport { SubscriptionScheduler } from \"../internal/SubscriptionScheduler\";\nimport { FilteringHierarchyDefinition } from \"./FilteringHierarchyDefinition\";\nimport { HierarchyCache } from \"./HierarchyCache\";\nimport { DefineHierarchyLevelProps, HierarchyDefinition, HierarchyNodesDefinition } from \"./IModelHierarchyDefinition\";\nimport {\n  ProcessedGenericHierarchyNode,\n  ProcessedGroupingHierarchyNode,\n  ProcessedHierarchyNode,\n  ProcessedInstanceHierarchyNode,\n  SourceGenericHierarchyNode,\n  SourceInstanceHierarchyNode,\n} from \"./IModelHierarchyNode\";\nimport { LimitingECSqlQueryExecutor } from \"./LimitingECSqlQueryExecutor\";\nimport { NodeSelectClauseColumnNames } from \"./NodeSelectQueryFactory\";\nimport { createDetermineChildrenOperator } from \"./operators/DetermineChildren\";\nimport { createGroupingOperator } from \"./operators/Grouping\";\nimport { createHideIfNoChildrenOperator } from \"./operators/HideIfNoChildren\";\nimport { createHideNodesInHierarchyOperator } from \"./operators/HideNodesInHierarchy\";\nimport { readNodes } from \"./TreeNodesReader\";\n\nconst LOGGING_NAMESPACE = `${BASE_LOGGING_NAMESPACE}.IModelHierarchyProvider`;\nconst LOGGING_NAMESPACE_INTERNAL = `${BASE_LOGGING_NAMESPACE_INTERNAL}.IModelHierarchyProvider`;\nconst LOGGING_NAMESPACE_PERFORMANCE = `${BASE_LOGGING_NAMESPACE_PERFORMANCE}.IModelHierarchyProvider`;\n\nconst DEFAULT_QUERY_CONCURRENCY = 10;\nconst DEFAULT_QUERY_CACHE_SIZE = 1;\n\n/**\n * Defines the strings used by hierarchy provider.\n * @beta\n */\ninterface IModelHierarchyProviderLocalizedStrings {\n  /**\n   * A string for \"Unspecified\". Used for labels of property grouping nodes\n   * that group by an empty value.\n   */\n  unspecified: string;\n\n  /**\n   * A string for \"Other\". Used for label of a range property grouping node that\n   * groups values which don't fit into any other range.\n   */\n  other: string;\n}\n\n/** @beta */\ntype IModelAccess = ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector & { imodelKey: string };\n\n/**\n * Props for `createIModelHierarchyProvider`.\n * @beta\n */\ninterface IModelHierarchyProviderProps {\n  /**\n   * An object that provides access to iModel's data and metadata.\n   *\n   * @see `ECSchemaProvider`\n   * @see `LimitingECSqlQueryExecutor`\n   * @see `ECClassHierarchyInspector`\n   */\n  imodelAccess: IModelAccess;\n\n  /** An event that is raised when iModel data, that may affect the hierarchy, changes. */\n  imodelChanged?: Event<() => void>;\n\n  /**\n   * A function that returns a hierarchy definition, describing how the hierarchy that the provider should be create. The\n   * function is called once during the provider's construction.\n   */\n  hierarchyDefinition: HierarchyDefinition;\n\n  /** Maximum number of queries that the provider attempts to execute in parallel. Defaults to `10`. */\n  queryConcurrency?: number;\n  /**\n   * The amount of queries whose results are stored in-memory for quick retrieval. Defaults to `1`,\n   * which means only results of the last run query are cached.\n   */\n  queryCacheSize?: number;\n\n  /** A set of localized strings to use. Defaults to English strings. */\n  localizedStrings?: Partial<IModelHierarchyProviderLocalizedStrings>;\n\n  /**\n   * A values formatter for formatting node labels. Defaults to the\n   * result of `createDefaultValueFormatter` called with default parameters. May be overridden\n   * by calling `setFormatter` on the provider instance.\n   */\n  formatter?: IPrimitiveValueFormatter;\n  /** Props for filtering the hierarchy. */\n  filtering?: {\n    /** A list of node identifiers from root to target node. */\n    paths: HierarchyFilteringPath[];\n  };\n}\n\n/**\n * Creates an instance of `HierarchyProvider` that creates a hierarchy based on given iModel and\n * a hierarchy definition, which defines each hierarchy level through ECSQL queries.\n *\n * @beta\n */\nexport function createIModelHierarchyProvider(props: IModelHierarchyProviderProps): HierarchyProvider & { dispose: () => void } {\n  return new IModelHierarchyProviderImpl(props);\n}\n\ninterface RequestContextProp {\n  requestContext: {\n    requestId: string;\n  };\n}\n\nclass IModelHierarchyProviderImpl implements HierarchyProvider {\n  private _imodelAccess: IModelAccess;\n  private _imodelChanged: Event<() => void>;\n  private _valuesFormatter: IPrimitiveValueFormatter;\n  private _sourceHierarchyDefinition: HierarchyDefinition;\n  private _activeHierarchyDefinition: HierarchyDefinition;\n  private _localizedStrings: IModelHierarchyProviderLocalizedStrings;\n  private _queryScheduler: SubscriptionScheduler;\n  private _nodesCache?: HierarchyCache<HierarchyCacheEntry>;\n  private _unsubscribe?: () => void;\n  private _dispose = new Subject<void>();\n\n  public constructor(props: IModelHierarchyProviderProps) {\n    this._imodelAccess = props.imodelAccess;\n    this._imodelChanged = props.imodelChanged ?? new BeEvent();\n    this._activeHierarchyDefinition = this._sourceHierarchyDefinition = props.hierarchyDefinition;\n    this._valuesFormatter = props?.formatter ?? createDefaultValueFormatter();\n    this._localizedStrings = { other: \"Other\", unspecified: \"Not specified\", ...props?.localizedStrings };\n    this._queryScheduler = new SubscriptionScheduler(props.queryConcurrency ?? DEFAULT_QUERY_CONCURRENCY);\n    this.setHierarchyFilter(props.filtering);\n\n    const queryCacheSize = props.queryCacheSize ?? DEFAULT_QUERY_CACHE_SIZE;\n    if (queryCacheSize !== 0) {\n      this._nodesCache = new HierarchyCache({\n        // we divide the size by 2, because each variation also counts as a query that we cache\n        size: Math.ceil(queryCacheSize / 2),\n        variationsCount: 1,\n      });\n      this._unsubscribe = props.imodelChanged?.addListener(() => {\n        this.invalidateHierarchyCache(\"Data source changed\");\n        this._dispose.next();\n      });\n    }\n  }\n\n  public dispose() {\n    this._dispose.next();\n    this._unsubscribe?.();\n  }\n\n  public get hierarchyChanged() {\n    return this._imodelChanged;\n  }\n\n  private invalidateHierarchyCache(reason?: string) {\n    doLog({\n      category: `${LOGGING_NAMESPACE}.Events`,\n      message: /* istanbul ignore next */ () => (reason ? `${reason}: clear hierarchy cache` : `Clear hierarchy cache`),\n    });\n    this._nodesCache?.clear();\n  }\n\n  /**\n   * Sets `HierarchyProvider` values formatter that formats nodes' labels. If provided `undefined`, then defaults to the\n   * result of `createDefaultValueFormatter` called with default parameters.\n   */\n  public setFormatter(formatter: IPrimitiveValueFormatter | undefined) {\n    this._valuesFormatter = formatter ?? createDefaultValueFormatter();\n  }\n\n  public setHierarchyFilter(props: IModelHierarchyProviderProps[\"filtering\"]) {\n    if (!props) {\n      if (this._sourceHierarchyDefinition !== this._activeHierarchyDefinition) {\n        this._activeHierarchyDefinition = this._sourceHierarchyDefinition;\n        this.invalidateHierarchyCache(\"Hierarchy filter reset\");\n      }\n      return;\n    }\n    this._activeHierarchyDefinition = new FilteringHierarchyDefinition({\n      imodelAccess: this._imodelAccess,\n      source: this._sourceHierarchyDefinition,\n      nodeIdentifierPaths: props.paths,\n    });\n    this.invalidateHierarchyCache(\"Hierarchy filter set\");\n    this._dispose.next();\n  }\n\n  private onGroupingNodeCreated(groupingNode: ProcessedGroupingHierarchyNode, props: GetHierarchyNodesProps) {\n    this._nodesCache?.set({ ...props, parentNode: groupingNode }, { observable: from(groupingNode.children), processingStatus: \"pre-processed\" });\n  }\n\n  private createHierarchyLevelDefinitionsObservable(props: DefineHierarchyLevelProps & RequestContextProp): Observable<HierarchyNodesDefinition> {\n    const { requestContext, ...defineHierarchyLevelProps } = props;\n    doLog({\n      category: LOGGING_NAMESPACE_PERFORMANCE,\n      message: /* istanbul ignore next */ () =>\n        `[${requestContext.requestId}] Requesting hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    return from(\n      defer(async () => {\n        let parentNode = props.parentNode;\n        if (parentNode && HierarchyNode.isGeneric(parentNode) && parentNode.key.source && parentNode.key.source !== this._imodelAccess.imodelKey) {\n          return [];\n        }\n        if (parentNode && HierarchyNode.isInstancesNode(parentNode)) {\n          parentNode = {\n            ...parentNode,\n            key: {\n              ...parentNode.key,\n              instanceKeys: parentNode.key.instanceKeys.filter((ik) => !ik.imodelKey || ik.imodelKey === this._imodelAccess.imodelKey),\n            },\n          };\n          assert(HierarchyNodeKey.isInstances(parentNode.key));\n          if (parentNode.key.instanceKeys.length === 0) {\n            return [];\n          }\n        }\n        return this._activeHierarchyDefinition.defineHierarchyLevel({ ...defineHierarchyLevelProps, parentNode });\n      }),\n    ).pipe(\n      mergeAll(),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* istanbul ignore next */ () =>\n            `[${requestContext.requestId}] Received all hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createSourceNodesObservable(\n    props: DefineHierarchyLevelProps & { hierarchyLevelSizeLimit?: number | \"unbounded\"; filteredInstanceKeys?: InstanceKey[] } & RequestContextProp,\n  ): SourceNodesObservable {\n    // pipe definitions to nodes and put \"share replay\" on it\n    return this.createHierarchyLevelDefinitionsObservable(props).pipe(\n      mergeMap((def): ObservableInput<SourceHierarchyNode> => {\n        if (HierarchyNodesDefinition.isGenericNode(def)) {\n          return of({ ...def.node, key: { type: \"generic\", id: def.node.key, source: this._imodelAccess.imodelKey } });\n        }\n        return this._queryScheduler.scheduleSubscription(\n          of(def.query).pipe(\n            map((query) => filterQueryByInstanceKeys(query, props.filteredInstanceKeys)),\n            mergeMap((query) =>\n              readNodes({ queryExecutor: this._imodelAccess, query, limit: props.hierarchyLevelSizeLimit, parser: this._activeHierarchyDefinition.parseNode }),\n            ),\n            map((node) => ({\n              ...node,\n              key: { ...node.key, instanceKeys: node.key.instanceKeys.map((key) => ({ ...key, imodelKey: this._imodelAccess.imodelKey })) },\n            })),\n          ),\n        );\n      }),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* istanbul ignore next */ () =>\n            `[${props.requestContext.requestId}] Read all child nodes ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n      shareReplayWithErrors(),\n    );\n  }\n\n  private createPreProcessedNodesObservable(\n    queryNodesObservable: SourceNodesObservable,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<ProcessedHierarchyNode> {\n    return queryNodesObservable.pipe(\n      // we're going to be mutating the nodes, but don't want to mutate the original one, so just clone it here once\n      map((node) => ({ ...node })),\n      // set parent node keys on the source node\n      map((node) => Object.assign(node, { parentKeys: createParentNodeKeysList(props.parentNode) })),\n      // format `ConcatenatedValue` labels into string labels\n      mergeMap(async (node) => applyLabelsFormatting(node, this._valuesFormatter)),\n      // we have `ProcessedHierarchyNode` from here\n      preProcessNodes(this._activeHierarchyDefinition),\n      // process hiding\n      createHideIfNoChildrenOperator((n) => this.getChildNodesObservables({ parentNode: n, requestContext: props.requestContext }).hasNodes),\n      createHideNodesInHierarchyOperator(\n        // note: for child nodes created because of hidden parent, we want to use parent's request props (instance filter, limit)\n        (n) => this.getChildNodesObservables({ ...props, parentNode: n }).processedNodes,\n        false,\n      ),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* istanbul ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished pre-processing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createProcessedNodesObservable(\n    preprocessedNodesObservable: Observable<ProcessedHierarchyNode>,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<ProcessedHierarchyNode> {\n    return preprocessedNodesObservable.pipe(\n      createGroupingOperator(this._imodelAccess, props.parentNode, this._valuesFormatter, this._localizedStrings, (gn) =>\n        this.onGroupingNodeCreated(gn, props),\n      ),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* istanbul ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished grouping child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createFinalizedNodesObservable(\n    processedNodesObservable: Observable<ProcessedHierarchyNode>,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<HierarchyNode> {\n    return processedNodesObservable.pipe(\n      createDetermineChildrenOperator((n) => this.getChildNodesObservables({ parentNode: n, requestContext: props.requestContext }).hasNodes),\n      postProcessNodes(this._activeHierarchyDefinition),\n      sortNodesByLabelOperator,\n      map((n): HierarchyNode => {\n        if (\"processingParams\" in n) {\n          delete n.processingParams;\n        }\n        return Object.assign(n, {\n          children: hasChildren(n),\n        });\n      }),\n      takeUntil(this._dispose),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* istanbul ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished finalizing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createHasNodesObservable(\n    preprocessedNodesObservable: Observable<ProcessedHierarchyNode>,\n    possiblyKnownChildrenObservable: SourceNodesObservable | undefined,\n    props: RequestContextProp,\n  ): Observable<boolean> {\n    const loggingCategory = `${LOGGING_NAMESPACE_INTERNAL}.HasNodes`;\n    return concat((possiblyKnownChildrenObservable ?? EMPTY).pipe(filter((n) => hasChildren(n))), preprocessedNodesObservable).pipe(\n      log({\n        category: loggingCategory,\n        message: /* istanbul ignore next */ (n) => `[${props.requestContext.requestId}] Node before mapping to 'true': ${createNodeIdentifierForLogging(n)}`,\n      }),\n      take(1),\n      map(() => true),\n      defaultIfEmpty(false),\n      catchError((e: Error) => {\n        doLog({\n          category: loggingCategory,\n          message: /* istanbul ignore next */ () => `[${props.requestContext.requestId}] Error while determining children: ${e.message}`,\n        });\n        if (e instanceof RowsLimitExceededError) {\n          return of(true);\n        }\n        throw e;\n      }),\n      log({ category: loggingCategory, message: /* istanbul ignore next */ (r) => `[${props.requestContext.requestId}] Result: ${r}` }),\n    );\n  }\n\n  private getCachedObservableEntry(props: GetHierarchyNodesProps & RequestContextProp): HierarchyCacheEntry {\n    const loggingCategory = `${LOGGING_NAMESPACE}.QueryResultsCache`;\n    const { parentNode, ...restProps } = props;\n    const cached = props.ignoreCache || !this._nodesCache ? undefined : this._nodesCache.get(props);\n    if (cached) {\n      // istanbul ignore next\n      doLog({\n        category: loggingCategory,\n        message: /* istanbul ignore next */ () =>\n          `[${props.requestContext.requestId}] Found query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,\n      });\n      return cached;\n    }\n\n    // if we don't find an entry for a grouping node, we load its instances by getting a query and applying\n    // a filter based on grouped instance keys\n    let filteredInstanceKeys: InstanceKey[] | undefined;\n    let parentNonGroupingNode: ParentHierarchyNode<NonGroupingHierarchyNode> | undefined;\n    if (parentNode) {\n      if (HierarchyNode.isGroupingNode(parentNode)) {\n        parentNonGroupingNode = parentNode.nonGroupingAncestor;\n        filteredInstanceKeys = parentNode.groupedInstanceKeys;\n      } else {\n        // not sure why type checker doesn't pick this up\n        assert(HierarchyNode.isGeneric(parentNode) || HierarchyNode.isInstancesNode(parentNode));\n        parentNonGroupingNode = parentNode;\n      }\n    }\n\n    const nonGroupingNodeChildrenRequestProps = {\n      ...restProps,\n      parentNode: parentNonGroupingNode,\n      ...(filteredInstanceKeys ? { filteredInstanceKeys } : undefined),\n    };\n    const value = { observable: this.createSourceNodesObservable(nonGroupingNodeChildrenRequestProps), processingStatus: \"none\" as const };\n    this._nodesCache?.set(nonGroupingNodeChildrenRequestProps, value);\n    doLog({\n      category: loggingCategory,\n      message: /* istanbul ignore next */ () =>\n        `[${props.requestContext.requestId}] Saved query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,\n    });\n    return value;\n  }\n\n  private getChildNodesObservables(props: GetHierarchyNodesProps & { hierarchyLevelSizeLimit?: number | \"unbounded\" } & RequestContextProp) {\n    const entry = this.getCachedObservableEntry(props);\n    switch (entry.processingStatus) {\n      case \"none\": {\n        const pre = this.createPreProcessedNodesObservable(entry.observable, props);\n        const post = this.createProcessedNodesObservable(pre, props);\n        return {\n          processedNodes: post,\n          hasNodes: this.createHasNodesObservable(pre, entry.observable, props),\n          finalizedNodes: this.createFinalizedNodesObservable(post, props),\n        };\n      }\n      case \"pre-processed\": {\n        const post = this.createProcessedNodesObservable(entry.observable, props);\n        return {\n          processedNodes: post,\n          hasNodes: this.createHasNodesObservable(entry.observable, undefined, props),\n          finalizedNodes: this.createFinalizedNodesObservable(post, props),\n        };\n      }\n    }\n  }\n\n  /**\n   * Creates and runs a query based on provided props, then processes retrieved nodes and returns them.\n   */\n  public getNodes(props: GetHierarchyNodesProps): AsyncIterableIterator<HierarchyNode> {\n    const requestContext = { requestId: Guid.createValue() };\n    const loggingCategory = `${LOGGING_NAMESPACE}.GetNodes`;\n    const timer = new StopWatch(undefined, true);\n    let error: any;\n    let nodesCount = 0;\n    doLog({\n      category: loggingCategory,\n      message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Requesting child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    return eachValueFrom(\n      this.getChildNodesObservables({ ...props, requestContext }).finalizedNodes.pipe(\n        tap(() => ++nodesCount),\n        catchError((e) => {\n          error = e;\n          throw e;\n        }),\n        finalize(() => {\n          doLog({\n            category: loggingCategory,\n            message: /* istanbul ignore next */ () =>\n              error\n                ? `[${requestContext.requestId}] Error creating child nodes for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`\n                : `[${requestContext.requestId}] Returned ${nodesCount} child nodes for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,\n          });\n        }),\n      ),\n    );\n  }\n\n  private getNodeInstanceKeysObs(props: Omit<GetHierarchyNodesProps, \"ignoreCache\"> & RequestContextProp): Observable<InstanceKey> {\n    const { parentNode, instanceFilter, hierarchyLevelSizeLimit = \"unbounded\", requestContext } = props;\n    if (parentNode && HierarchyNode.isGroupingNode(parentNode)) {\n      return from(parentNode.groupedInstanceKeys);\n    }\n    assert(!parentNode || HierarchyNode.isGeneric(parentNode) || HierarchyNode.isInstancesNode(parentNode));\n\n    const hierarchyLevelDefinitions = this.createHierarchyLevelDefinitionsObservable({ parentNode, instanceFilter, requestContext });\n\n    // split the definitions based on whether they're for generic nodes or for instance nodes\n    const [genericDefs, instanceDefs] = partition(hierarchyLevelDefinitions, HierarchyNodesDefinition.isGenericNode);\n\n    // query instance keys and a flag whether they should be hidden\n    const instanceKeys = instanceDefs.pipe(\n      mergeMap((def) =>\n        this._queryScheduler.scheduleSubscription(\n          of(def.query).pipe(\n            mergeMap(async (query) => {\n              const ecsql = `\n                SELECT\n                  ${NodeSelectClauseColumnNames.FullClassName},\n                  ${NodeSelectClauseColumnNames.ECInstanceId},\n                  ${NodeSelectClauseColumnNames.HideNodeInHierarchy}\n                FROM (\n                  ${query.ecsql}\n                )\n              `;\n              const reader = this._imodelAccess.createQueryReader({ ...query, ecsql }, { rowFormat: \"Indexes\", limit: hierarchyLevelSizeLimit });\n              return from(reader).pipe(\n                map((row) => ({\n                  key: {\n                    className: normalizeFullClassName(row[0]),\n                    id: row[1],\n                    imodelKey: this._imodelAccess.imodelKey,\n                  } satisfies IModelInstanceKey,\n                  hide: !!row[2],\n                })),\n              );\n            }),\n            mergeAll(),\n          ),\n        ),\n      ),\n    );\n    // split the instance keys observable based on whether they should be hidden or not\n    const [visibleNodeInstanceKeys, hiddenNodeInstanceKeys] = partition(instanceKeys, ({ hide }) => !hide);\n\n    // hidden items' handling:\n    // - if a generic node is hidden, we'll want to retrieve instance keys of its children, otherwise we don't care about it,\n    // - if an instance key is hidden, we want to create a merged node similar to what we do in `createHideNodesInHierarchyOperator`\n    const hiddenParentNodes = merge(\n      genericDefs.pipe(\n        filter((def) => !!def.node.processingParams?.hideInHierarchy),\n        map(\n          (def): GenericNodeKey => ({\n            type: \"generic\",\n            id: def.node.key,\n            source: this._imodelAccess.imodelKey,\n          }),\n        ),\n      ),\n      hiddenNodeInstanceKeys.pipe(\n        // first merge all keys by class\n        reduceToMergeMapList(\n          ({ key }) => key.className,\n          ({ key }) => key.id,\n        ),\n        // then, for each class, create an instance key\n        mergeMap((mergedMap): InstancesNodeKey[] =>\n          [...mergedMap.entries()].map(([className, ids]) => ({\n            type: \"instances\",\n            instanceKeys: ids.map((id) => ({ className, id, imodelKey: this._imodelAccess.imodelKey })),\n          })),\n        ),\n      ),\n    ).pipe(\n      map((key) => ({\n        key,\n        parentKeys: [],\n        label: \"\",\n      })),\n    );\n\n    // merge visible instance keys from this level & the ones we get recursively requesting from deeper levels\n    return merge(\n      visibleNodeInstanceKeys.pipe(map(({ key }) => key)),\n      hiddenParentNodes.pipe(mergeMap((hiddenNode) => this.getNodeInstanceKeysObs({ parentNode: hiddenNode, requestContext }))),\n    ).pipe(takeUntil(this._dispose));\n  }\n\n  /**\n   * Creates an iterator for all child hierarchy level instance keys, taking into account any hidden hierarchy levels\n   * that there may be under the given parent node.\n   */\n  public getNodeInstanceKeys(props: Omit<GetHierarchyNodesProps, \"ignoreCache\">): AsyncIterableIterator<InstanceKey> {\n    const requestContext = { requestId: Guid.createValue() };\n    const loggingCategory = `${LOGGING_NAMESPACE}.GetNodeInstanceKeys`;\n    const timer = new StopWatch(undefined, true);\n    doLog({\n      category: loggingCategory,\n      message: /* istanbul ignore next */ () => `[${requestContext.requestId}] Requesting keys for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    let error: any;\n    let keysCount = 0;\n    return eachValueFrom(\n      this.getNodeInstanceKeysObs({ ...props, requestContext }).pipe(\n        tap(() => ++keysCount),\n        catchError((e) => {\n          error = e;\n          throw e;\n        }),\n        finalize(() => {\n          doLog({\n            category: loggingCategory,\n            message: /* istanbul ignore next */ () =>\n              error\n                ? `[${requestContext.requestId}] Error creating node instance keys for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`\n                : `[${requestContext.requestId}] Returned ${keysCount} instance keys for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,\n          });\n        }),\n      ),\n    );\n  }\n}\n\ntype SourceHierarchyNode = SourceInstanceHierarchyNode | (Omit<SourceGenericHierarchyNode, \"key\"> & { key: GenericNodeKey });\ntype SourceNodesObservable = Observable<SourceHierarchyNode>;\ntype ProcessedNodesObservable = Observable<ProcessedHierarchyNode>;\ntype HierarchyCacheEntry =\n  | { observable: SourceNodesObservable; processingStatus: \"none\" }\n  | { observable: ProcessedNodesObservable; processingStatus: \"pre-processed\" };\n\nfunction preProcessNodes(hierarchyFactory: HierarchyDefinition) {\n  return hierarchyFactory.preProcessNode\n    ? processNodes(hierarchyFactory.preProcessNode.bind(hierarchyFactory))\n    : (o: Observable<ProcessedGenericHierarchyNode | ProcessedInstanceHierarchyNode>) => o;\n}\n\nfunction postProcessNodes<ProcessedHierarchyNode>(hierarchyFactory: HierarchyDefinition) {\n  return hierarchyFactory.postProcessNode\n    ? processNodes(hierarchyFactory.postProcessNode.bind(hierarchyFactory))\n    : (o: Observable<ProcessedHierarchyNode>) => o;\n}\n\nfunction processNodes<TNode>(processor: (node: TNode) => Promise<TNode | undefined>) {\n  return (nodes: Observable<TNode>) =>\n    nodes.pipe(\n      mergeMap(processor),\n      filter((n): n is TNode => !!n),\n    );\n}\n\nasync function applyLabelsFormatting<TNode extends { label: string | ConcatenatedValue }>(\n  node: TNode,\n  valueFormatter: IPrimitiveValueFormatter,\n): Promise<TNode & { label: string }> {\n  return {\n    ...node,\n    label: await formatConcatenatedValue({\n      value: node.label,\n      valueFormatter,\n    }),\n  };\n}\n\nfunction createParentNodeKeysList(parentNode: ParentHierarchyNode | undefined) {\n  if (!parentNode) {\n    return [];\n  }\n  return [...parentNode.parentKeys, parentNode.key];\n}\n\nfunction filterQueryByInstanceKeys(query: ECSqlQueryDef, filteredInstanceKeys: InstanceKey[] | undefined): ECSqlQueryDef {\n  if (!filteredInstanceKeys || !filteredInstanceKeys.length) {\n    return query;\n  }\n  const MAX_ALLOWED_BINDINGS = 1000;\n  // istanbul ignore else\n  if (filteredInstanceKeys.length < MAX_ALLOWED_BINDINGS) {\n    return {\n      ...query,\n      ecsql: `\n        SELECT *\n        FROM (${query.ecsql}) q\n        WHERE q.ECInstanceId IN (${filteredInstanceKeys.map(() => \"?\").join(\",\")})\n      `,\n      bindings: [...(query.bindings ?? []), ...filteredInstanceKeys.map((k): ECSqlBinding => ({ type: \"id\", value: k.id }))],\n    };\n  }\n  // istanbul ignore next\n  return {\n    ...query,\n    ecsql: `\n      SELECT *\n      FROM (${query.ecsql}) q\n      WHERE InVirtualSet(?, q.ECInstanceId)\n    `,\n    bindings: [...(query.bindings ?? []), { type: \"idset\", value: filteredInstanceKeys.map((k) => k.id) }],\n  };\n}\n"]}
+{"version":3,"file":"IModelHierarchyProvider.js","sourceRoot":"","sources":["../../../../src/hierarchies/imodel/IModelHierarchyProvider.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EACL,UAAU,EACV,MAAM,EACN,cAAc,EACd,KAAK,EACL,KAAK,EACL,MAAM,EACN,QAAQ,EACR,IAAI,EACJ,GAAG,EACH,KAAK,EACL,QAAQ,EACR,QAAQ,EAGR,EAAE,EACF,OAAO,EACP,IAAI,EACJ,SAAS,EACT,GAAG,GACJ,MAAM,MAAM,CAAC;AACd,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AACvE,OAAO,EAEL,2BAA2B,EAM3B,uBAAuB,EAGvB,sBAAsB,GACvB,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAE/D,OAAO,EAAE,aAAa,EAAiD,MAAM,qBAAqB,CAAC;AACnG,OAAO,EAAkB,gBAAgB,EAAuC,MAAM,wBAAwB,CAAC;AAE/G,OAAO,EACL,iBAAiB,IAAI,sBAAsB,EAC3C,0BAA0B,IAAI,+BAA+B,EAC7D,6BAA6B,IAAI,kCAAkC,EACnE,8BAA8B,EAC9B,WAAW,GACZ,MAAM,uBAAuB,CAAC;AAC/B,OAAO,EAAE,aAAa,EAAE,MAAM,8BAA8B,CAAC;AAC7D,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,MAAM,6BAA6B,CAAC;AACzD,OAAO,EAAE,SAAS,EAAE,MAAM,oCAAoC,CAAC;AAC/D,OAAO,EAAE,oBAAoB,EAAE,MAAM,2CAA2C,CAAC;AACjF,OAAO,EAAE,qBAAqB,EAAE,MAAM,gDAAgD,CAAC;AACvF,OAAO,EAAE,wBAAwB,EAAE,MAAM,kCAAkC,CAAC;AAC5E,OAAO,EAAE,qBAAqB,EAAE,MAAM,sCAAsC,CAAC;AAC7E,OAAO,EAAE,4BAA4B,EAAE,MAAM,mCAAmC,CAAC;AACjF,OAAO,EAAE,cAAc,EAAE,MAAM,qBAAqB,CAAC;AACrD,OAAO,EAAkD,wBAAwB,EAAE,MAAM,gCAAgC,CAAC;AAU1H,OAAO,EAAE,2BAA2B,EAAE,MAAM,6BAA6B,CAAC;AAC1E,OAAO,EAAE,+BAA+B,EAAE,MAAM,kCAAkC,CAAC;AACnF,OAAO,EAAE,sBAAsB,EAAE,MAAM,yBAAyB,CAAC;AACjE,OAAO,EAAE,8BAA8B,EAAE,MAAM,iCAAiC,CAAC;AACjF,OAAO,EAAE,kCAAkC,EAAE,MAAM,qCAAqC,CAAC;AACzF,OAAO,EAAE,SAAS,EAAE,MAAM,sBAAsB,CAAC;AAEjD,MAAM,iBAAiB,GAAG,GAAG,sBAAsB,0BAA0B,CAAC;AAC9E,MAAM,0BAA0B,GAAG,GAAG,+BAA+B,0BAA0B,CAAC;AAChG,MAAM,6BAA6B,GAAG,GAAG,kCAAkC,0BAA0B,CAAC;AAEtG,MAAM,yBAAyB,GAAG,EAAE,CAAC;AACrC,MAAM,wBAAwB,GAAG,CAAC,CAAC;AAsEnC;;;;;GAKG;AACH,MAAM,UAAU,6BAA6B,CAAC,KAAmC;IAC/E,OAAO,IAAI,2BAA2B,CAAC,KAAK,CAAC,CAAC;AAChD,CAAC;AAQD,MAAM,2BAA2B;IACvB,aAAa,CAAe;IAC5B,cAAc,CAAoB;IAClC,gBAAgB,CAA2B;IAC3C,0BAA0B,CAAsB;IAChD,0BAA0B,CAAsB;IAChD,iBAAiB,CAA0C;IAC3D,eAAe,CAAwB;IACvC,WAAW,CAAuC;IAClD,YAAY,CAAc;IAC1B,QAAQ,GAAG,IAAI,OAAO,EAAQ,CAAC;IAEvC,YAAmB,KAAmC;QACpD,IAAI,CAAC,aAAa,GAAG,KAAK,CAAC,YAAY,CAAC;QACxC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,IAAI,IAAI,OAAO,EAAE,CAAC;QAC3D,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,0BAA0B,GAAG,KAAK,CAAC,mBAAmB,CAAC;QAC9F,IAAI,CAAC,gBAAgB,GAAG,KAAK,EAAE,SAAS,IAAI,2BAA2B,EAAE,CAAC;QAC1E,IAAI,CAAC,iBAAiB,GAAG,EAAE,KAAK,EAAE,OAAO,EAAE,WAAW,EAAE,eAAe,EAAE,GAAG,KAAK,EAAE,gBAAgB,EAAE,CAAC;QACtG,IAAI,CAAC,eAAe,GAAG,IAAI,qBAAqB,CAAC,KAAK,CAAC,gBAAgB,IAAI,yBAAyB,CAAC,CAAC;QACtG,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,SAAS,CAAC,CAAC;QAEzC,MAAM,cAAc,GAAG,KAAK,CAAC,cAAc,IAAI,wBAAwB,CAAC;QACxE,IAAI,cAAc,KAAK,CAAC,EAAE,CAAC;YACzB,IAAI,CAAC,WAAW,GAAG,IAAI,cAAc,CAAC;gBACpC,uFAAuF;gBACvF,IAAI,EAAE,IAAI,CAAC,IAAI,CAAC,cAAc,GAAG,CAAC,CAAC;gBACnC,eAAe,EAAE,CAAC;aACnB,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,GAAG,KAAK,CAAC,aAAa,EAAE,WAAW,CAAC,GAAG,EAAE;gBACxD,IAAI,CAAC,wBAAwB,CAAC,qBAAqB,CAAC,CAAC;gBACrD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;YACvB,CAAC,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;QACrB,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC;IACxB,CAAC;IAED,IAAW,gBAAgB;QACzB,OAAO,IAAI,CAAC,cAAc,CAAC;IAC7B,CAAC;IAEO,wBAAwB,CAAC,MAAe;QAC9C,KAAK,CAAC;YACJ,QAAQ,EAAE,GAAG,iBAAiB,SAAS;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,GAAG,MAAM,yBAAyB,CAAC,CAAC,CAAC,uBAAuB,CAAC;SAC5G,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,EAAE,KAAK,EAAE,CAAC;IAC5B,CAAC;IAED;;;OAGG;IACI,YAAY,CAAC,SAA+C;QACjE,IAAI,CAAC,gBAAgB,GAAG,SAAS,IAAI,2BAA2B,EAAE,CAAC;IACrE,CAAC;IAEM,kBAAkB,CAAC,KAAgD;QACxE,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,IAAI,IAAI,CAAC,0BAA0B,KAAK,IAAI,CAAC,0BAA0B,EAAE,CAAC;gBACxE,IAAI,CAAC,0BAA0B,GAAG,IAAI,CAAC,0BAA0B,CAAC;gBAClE,IAAI,CAAC,wBAAwB,CAAC,wBAAwB,CAAC,CAAC;YAC1D,CAAC;YACD,OAAO;QACT,CAAC;QACD,IAAI,CAAC,0BAA0B,GAAG,IAAI,4BAA4B,CAAC;YACjE,YAAY,EAAE,IAAI,CAAC,aAAa;YAChC,MAAM,EAAE,IAAI,CAAC,0BAA0B;YACvC,mBAAmB,EAAE,KAAK,CAAC,KAAK;SACjC,CAAC,CAAC;QACH,IAAI,CAAC,wBAAwB,CAAC,sBAAsB,CAAC,CAAC;QACtD,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,CAAC;IACvB,CAAC;IAEO,qBAAqB,CAAC,YAA4C,EAAE,KAA6B;QACvG,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,YAAY,EAAE,EAAE,EAAE,UAAU,EAAE,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,EAAE,gBAAgB,EAAE,eAAe,EAAE,CAAC,CAAC;IAChJ,CAAC;IAEO,yCAAyC,CAAC,KAAqD;QACrG,MAAM,EAAE,cAAc,EAAE,GAAG,yBAAyB,EAAE,GAAG,KAAK,CAAC;QAC/D,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,cAAc,CAAC,SAAS,gDAAgD,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACjI,CAAC,CAAC;QACH,OAAO,IAAI,CACT,KAAK,CAAC,KAAK,IAAI,EAAE;YACf,IAAI,UAAU,GAAG,KAAK,CAAC,UAAU,CAAC;YAClC,IAAI,UAAU,IAAI,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,IAAI,UAAU,CAAC,GAAG,CAAC,MAAM,KAAK,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC;gBACzI,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,IAAI,UAAU,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC5D,UAAU,GAAG;oBACX,GAAG,UAAU;oBACb,GAAG,EAAE;wBACH,GAAG,UAAU,CAAC,GAAG;wBACjB,YAAY,EAAE,UAAU,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,CAAC,SAAS,IAAI,EAAE,CAAC,SAAS,KAAK,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC;qBACzH;iBACF,CAAC;gBACF,MAAM,CAAC,gBAAgB,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,CAAC;gBACrD,IAAI,UAAU,CAAC,GAAG,CAAC,YAAY,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7C,OAAO,EAAE,CAAC;gBACZ,CAAC;YACH,CAAC;YACD,OAAO,IAAI,CAAC,0BAA0B,CAAC,oBAAoB,CAAC,EAAE,GAAG,yBAAyB,EAAE,UAAU,EAAE,CAAC,CAAC;QAC5G,CAAC,CAAC,CACH,CAAC,IAAI,CACJ,QAAQ,EAAE,EACV,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,cAAc,CAAC,SAAS,kDAAkD,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACnI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,2BAA2B,CACjC,KAAgJ;QAEhJ,yDAAyD;QACzD,OAAO,IAAI,CAAC,yCAAyC,CAAC,KAAK,CAAC,CAAC,IAAI,CAC/D,QAAQ,CAAC,CAAC,GAAG,EAAwC,EAAE;YACrD,IAAI,wBAAwB,CAAC,aAAa,CAAC,GAAG,CAAC,EAAE,CAAC;gBAChD,OAAO,EAAE,CAAC,EAAE,GAAG,GAAG,CAAC,IAAI,EAAE,GAAG,EAAE,EAAE,IAAI,EAAE,SAAS,EAAE,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,EAAE,CAAC,CAAC;YAC/G,CAAC;YACD,OAAO,IAAI,CAAC,eAAe,CAAC,oBAAoB,CAC9C,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAChB,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,yBAAyB,CAAC,KAAK,EAAE,KAAK,CAAC,oBAAoB,CAAC,CAAC,EAC5E,QAAQ,CAAC,CAAC,KAAK,EAAE,EAAE,CACjB,SAAS,CAAC;gBACR,aAAa,EAAE,IAAI,CAAC,aAAa;gBACjC,KAAK;gBACL,KAAK,EAAE,KAAK,CAAC,uBAAuB;gBACpC,MAAM,EAAE,IAAI,CAAC,0BAA0B,CAAC,SAAS;oBAC/C,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE,CAAC,IAAI,CAAC,0BAA0B,CAAC,SAAU,CAAC,GAAG,EAAE,KAAK,CAAC,UAAU,CAAC;oBAClF,CAAC,CAAC,SAAS;aACd,CAAC,CACH,EACD,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;gBACb,GAAG,IAAI;gBACP,GAAG,EAAE,EAAE,GAAG,IAAI,CAAC,GAAG,EAAE,YAAY,EAAE,IAAI,CAAC,GAAG,CAAC,YAAY,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC,EAAE;aAC9H,CAAC,CAAC,CACJ,CACF,CAAC;QACJ,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,0BAA0B,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACnJ,CAAC,CACH,EACD,qBAAqB,EAAE,CACxB,CAAC;IACJ,CAAC;IAEO,iCAAiC,CACvC,oBAA2C,EAC3C,KAAkD;QAElD,OAAO,oBAAoB,CAAC,IAAI;QAC9B,8GAA8G;QAC9G,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,IAAI,EAAE,CAAC,CAAC;QAC5B,0CAA0C;QAC1C,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,EAAE,UAAU,EAAE,wBAAwB,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC9F,uDAAuD;QACvD,QAAQ,CAAC,KAAK,EAAE,IAAI,EAAE,EAAE,CAAC,qBAAqB,CAAC,IAAI,EAAE,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAC5E,6CAA6C;QAC7C,eAAe,CAAC,IAAI,CAAC,0BAA0B,CAAC;QAChD,iBAAiB;QACjB,8BAA8B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,EACtI,kCAAkC;QAChC,yHAAyH;QACzH,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,UAAU,EAAE,CAAC,EAAE,CAAC,CAAC,cAAc,EAChF,KAAK,CACN,EACD,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,6CAA6C,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACpI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,8BAA8B,CACpC,2BAA+D,EAC/D,KAAkD;QAElD,OAAO,2BAA2B,CAAC,IAAI,CACrC,sBAAsB,CAAC,IAAI,CAAC,aAAa,EAAE,KAAK,CAAC,UAAU,EAAE,IAAI,CAAC,gBAAgB,EAAE,IAAI,CAAC,iBAAiB,EAAE,CAAC,EAAE,EAAE,EAAE,CACjH,IAAI,CAAC,qBAAqB,CAAC,EAAE,EAAE,KAAK,CAAC,CACtC,EACD,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,uCAAuC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAC9H,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,8BAA8B,CACpC,wBAA4D,EAC5D,KAAkD;QAElD,OAAO,wBAAwB,CAAC,IAAI,CAClC,+BAA+B,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,wBAAwB,CAAC,EAAE,UAAU,EAAE,CAAC,EAAE,cAAc,EAAE,KAAK,CAAC,cAAc,EAAE,CAAC,CAAC,QAAQ,CAAC,EACvI,gBAAgB,CAAC,IAAI,CAAC,0BAA0B,CAAC,EACjD,wBAAwB,EACxB,GAAG,CAAC,CAAC,CAAC,EAAiB,EAAE;YACvB,IAAI,kBAAkB,IAAI,CAAC,EAAE,CAAC;gBAC5B,OAAO,CAAC,CAAC,gBAAgB,CAAC;YAC5B,CAAC;YACD,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,EAAE;gBACtB,QAAQ,EAAE,WAAW,CAAC,CAAC,CAAC;aACzB,CAAC,CAAC;QACL,CAAC,CAAC,EACF,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,EACxB,QAAQ,CAAC,GAAG,EAAE,CACZ,KAAK,CAAC;YACJ,QAAQ,EAAE,6BAA6B;YACvC,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,yCAAyC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAChI,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAC9B,2BAA+D,EAC/D,+BAAkE,EAClE,KAAyB;QAEzB,MAAM,eAAe,GAAG,GAAG,0BAA0B,WAAW,CAAC;QACjE,OAAO,MAAM,CAAC,CAAC,+BAA+B,IAAI,KAAK,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,WAAW,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,2BAA2B,CAAC,CAAC,IAAI,CAC7H,GAAG,CAAC;YACF,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,oCAAoC,8BAA8B,CAAC,CAAC,CAAC,EAAE;SAC/I,CAAC,EACF,IAAI,CAAC,CAAC,CAAC,EACP,GAAG,CAAC,GAAG,EAAE,CAAC,IAAI,CAAC,EACf,cAAc,CAAC,KAAK,CAAC,EACrB,UAAU,CAAC,CAAC,CAAQ,EAAE,EAAE;YACtB,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,uCAAuC,CAAC,CAAC,OAAO,EAAE;aACzH,CAAC,CAAC;YACH,IAAI,CAAC,YAAY,sBAAsB,EAAE,CAAC;gBACxC,OAAO,EAAE,CAAC,IAAI,CAAC,CAAC;YAClB,CAAC;YACD,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,GAAG,CAAC,EAAE,QAAQ,EAAE,eAAe,EAAE,OAAO,EAAE,oBAAoB,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,aAAa,CAAC,EAAE,EAAE,CAAC,CAC5H,CAAC;IACJ,CAAC;IAEO,wBAAwB,CAAC,KAAkD;QACjF,MAAM,eAAe,GAAG,GAAG,iBAAiB,oBAAoB,CAAC;QACjE,MAAM,EAAE,UAAU,EAAE,GAAG,SAAS,EAAE,GAAG,KAAK,CAAC;QAC3C,MAAM,MAAM,GAAG,KAAK,CAAC,WAAW,IAAI,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QAChG,IAAI,MAAM,EAAE,CAAC;YACX,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CACjC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,sCAAsC,8BAA8B,CAAC,UAAU,CAAC,EAAE;aACvH,CAAC,CAAC;YACH,OAAO,MAAM,CAAC;QAChB,CAAC;QAED,uGAAuG;QACvG,0CAA0C;QAC1C,IAAI,oBAA+C,CAAC;QACpD,IAAI,qBAAgF,CAAC;QACrF,IAAI,UAAU,EAAE,CAAC;YACf,IAAI,aAAa,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;gBAC7C,qBAAqB,GAAG,UAAU,CAAC,mBAAmB,CAAC;gBACvD,oBAAoB,GAAG,UAAU,CAAC,mBAAmB,CAAC;YACxD,CAAC;iBAAM,CAAC;gBACN,iDAAiD;gBACjD,MAAM,CAAC,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;gBACzF,qBAAqB,GAAG,UAAU,CAAC;YACrC,CAAC;QACH,CAAC;QAED,MAAM,mCAAmC,GAAG;YAC1C,GAAG,SAAS;YACZ,UAAU,EAAE,qBAAqB;YACjC,GAAG,CAAC,oBAAoB,CAAC,CAAC,CAAC,EAAE,oBAAoB,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC;SACjE,CAAC;QACF,MAAM,KAAK,GAAG,EAAE,UAAU,EAAE,IAAI,CAAC,2BAA2B,CAAC,mCAAmC,CAAC,EAAE,gBAAgB,EAAE,MAAe,EAAE,CAAC;QACvI,IAAI,CAAC,WAAW,EAAE,GAAG,CAAC,mCAAmC,EAAE,KAAK,CAAC,CAAC;QAClE,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,IAAI,KAAK,CAAC,cAAc,CAAC,SAAS,sCAAsC,8BAA8B,CAAC,UAAU,CAAC,EAAE;SACzJ,CAAC,CAAC;QACH,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,wBAAwB,CAAC,KAAuG;QACtI,MAAM,KAAK,GAAG,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,CAAC;QACnD,QAAQ,KAAK,CAAC,gBAAgB,EAAE,CAAC;YAC/B,KAAK,MAAM,CAAC,CAAC,CAAC;gBACZ,MAAM,GAAG,GAAG,IAAI,CAAC,iCAAiC,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC5E,MAAM,IAAI,GAAG,IAAI,CAAC,8BAA8B,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;gBAC7D,OAAO;oBACL,cAAc,EAAE,IAAI;oBACpB,QAAQ,EAAE,IAAI,CAAC,wBAAwB,CAAC,GAAG,EAAE,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC;oBACrE,cAAc,EAAE,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC;iBACjE,CAAC;YACJ,CAAC;YACD,KAAK,eAAe,CAAC,CAAC,CAAC;gBACrB,MAAM,IAAI,GAAG,IAAI,CAAC,8BAA8B,CAAC,KAAK,CAAC,UAAU,EAAE,KAAK,CAAC,CAAC;gBAC1E,OAAO;oBACL,cAAc,EAAE,IAAI;oBACpB,QAAQ,EAAE,IAAI,CAAC,wBAAwB,CAAC,KAAK,CAAC,UAAU,EAAE,SAAS,EAAE,KAAK,CAAC;oBAC3E,cAAc,EAAE,IAAI,CAAC,8BAA8B,CAAC,IAAI,EAAE,KAAK,CAAC;iBACjE,CAAC;YACJ,CAAC;QACH,CAAC;IACH,CAAC;IAED;;OAEG;IACI,QAAQ,CAAC,KAA6B;QAC3C,MAAM,cAAc,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACzD,MAAM,eAAe,GAAG,GAAG,iBAAiB,WAAW,CAAC;QACxD,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7C,IAAI,KAAU,CAAC;QACf,IAAI,UAAU,GAAG,CAAC,CAAC;QACnB,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,IAAI,cAAc,CAAC,SAAS,gCAAgC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SACnJ,CAAC,CAAC;QACH,OAAO,aAAa,CAClB,IAAI,CAAC,wBAAwB,CAAC,EAAE,GAAG,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,cAAc,CAAC,IAAI,CAC7E,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,UAAU,CAAC,EACvB,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE;YACf,KAAK,GAAG,CAAC,CAAC;YACV,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE;YACZ,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE;gBACjC,sBAAsB,CAAC,KAAK;oBAC1B,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,oCAAoC,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE;oBAClL,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,cAAc,UAAU,oBAAoB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK;aAC1K,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CACF,CAAC;IACJ,CAAC;IAEO,sBAAsB,CAAC,KAAuE;QACpG,MAAM,EAAE,UAAU,EAAE,cAAc,EAAE,uBAAuB,GAAG,WAAW,EAAE,cAAc,EAAE,GAAG,KAAK,CAAC;QACpG,IAAI,UAAU,IAAI,aAAa,CAAC,cAAc,CAAC,UAAU,CAAC,EAAE,CAAC;YAC3D,OAAO,IAAI,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;QAC9C,CAAC;QACD,MAAM,CAAC,CAAC,UAAU,IAAI,aAAa,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,aAAa,CAAC,eAAe,CAAC,UAAU,CAAC,CAAC,CAAC;QAExG,MAAM,yBAAyB,GAAG,IAAI,CAAC,yCAAyC,CAAC,EAAE,UAAU,EAAE,cAAc,EAAE,cAAc,EAAE,CAAC,CAAC;QAEjI,yFAAyF;QACzF,MAAM,CAAC,WAAW,EAAE,YAAY,CAAC,GAAG,SAAS,CAAC,yBAAyB,EAAE,wBAAwB,CAAC,aAAa,CAAC,CAAC;QAEjH,+DAA+D;QAC/D,MAAM,YAAY,GAAG,YAAY,CAAC,IAAI,CACpC,QAAQ,CAAC,CAAC,GAAG,EAAE,EAAE,CACf,IAAI,CAAC,eAAe,CAAC,oBAAoB,CACvC,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC,IAAI,CAChB,QAAQ,CAAC,KAAK,EAAE,KAAK,EAAE,EAAE;YACvB,MAAM,KAAK,GAAG;;oBAER,2BAA2B,CAAC,aAAa;oBACzC,2BAA2B,CAAC,YAAY;oBACxC,2BAA2B,CAAC,mBAAmB;;oBAE/C,KAAK,CAAC,KAAK;;eAEhB,CAAC;YACF,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,iBAAiB,CAAC,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,EAAE,EAAE,SAAS,EAAE,SAAS,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;YACnI,OAAO,IAAI,CAAC,MAAM,CAAC,CAAC,IAAI,CACtB,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;gBACZ,GAAG,EAAE;oBACH,SAAS,EAAE,sBAAsB,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;oBACzC,EAAE,EAAE,GAAG,CAAC,CAAC,CAAC;oBACV,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS;iBACZ;gBAC7B,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;aACf,CAAC,CAAC,CACJ,CAAC;QACJ,CAAC,CAAC,EACF,QAAQ,EAAE,CACX,CACF,CACF,CACF,CAAC;QACF,mFAAmF;QACnF,MAAM,CAAC,uBAAuB,EAAE,sBAAsB,CAAC,GAAG,SAAS,CAAC,YAAY,EAAE,CAAC,EAAE,IAAI,EAAE,EAAE,EAAE,CAAC,CAAC,IAAI,CAAC,CAAC;QAEvG,0BAA0B;QAC1B,yHAAyH;QACzH,gIAAgI;QAChI,MAAM,iBAAiB,GAAG,KAAK,CAC7B,WAAW,CAAC,IAAI,CACd,MAAM,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,IAAI,CAAC,gBAAgB,EAAE,eAAe,CAAC,EAC7D,GAAG,CACD,CAAC,GAAG,EAAkB,EAAE,CAAC,CAAC;YACxB,IAAI,EAAE,SAAS;YACf,EAAE,EAAE,GAAG,CAAC,IAAI,CAAC,GAAG;YAChB,MAAM,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS;SACrC,CAAC,CACH,CACF,EACD,sBAAsB,CAAC,IAAI;QACzB,gCAAgC;QAChC,oBAAoB,CAClB,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,SAAS,EAC1B,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,EAAE,CACpB;QACD,+CAA+C;QAC/C,QAAQ,CAAC,CAAC,SAAS,EAAsB,EAAE,CACzC,CAAC,GAAG,SAAS,CAAC,OAAO,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,SAAS,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YAClD,IAAI,EAAE,WAAW;YACjB,YAAY,EAAE,GAAG,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,SAAS,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS,EAAE,CAAC,CAAC;SAC5F,CAAC,CAAC,CACJ,CACF,CACF,CAAC,IAAI,CACJ,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,CAAC;YACZ,GAAG;YACH,UAAU,EAAE,EAAE;YACd,KAAK,EAAE,EAAE;SACV,CAAC,CAAC,CACJ,CAAC;QAEF,0GAA0G;QAC1G,OAAO,KAAK,CACV,uBAAuB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,CAAC,EACnD,iBAAiB,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,IAAI,CAAC,sBAAsB,CAAC,EAAE,UAAU,EAAE,UAAU,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAC1H,CAAC,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;IACnC,CAAC;IAED;;;OAGG;IACI,mBAAmB,CAAC,KAAkD;QAC3E,MAAM,cAAc,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,WAAW,EAAE,EAAE,CAAC;QACzD,MAAM,eAAe,GAAG,GAAG,iBAAiB,sBAAsB,CAAC;QACnE,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;QAC7C,KAAK,CAAC;YACJ,QAAQ,EAAE,eAAe;YACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,IAAI,cAAc,CAAC,SAAS,yBAAyB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,EAAE;SAC5I,CAAC,CAAC;QACH,IAAI,KAAU,CAAC;QACf,IAAI,SAAS,GAAG,CAAC,CAAC;QAClB,OAAO,aAAa,CAClB,IAAI,CAAC,sBAAsB,CAAC,EAAE,GAAG,KAAK,EAAE,cAAc,EAAE,CAAC,CAAC,IAAI,CAC5D,GAAG,CAAC,GAAG,EAAE,CAAC,EAAE,SAAS,CAAC,EACtB,UAAU,CAAC,CAAC,CAAC,EAAE,EAAE;YACf,KAAK,GAAG,CAAC,CAAC;YACV,MAAM,CAAC,CAAC;QACV,CAAC,CAAC,EACF,QAAQ,CAAC,GAAG,EAAE;YACZ,KAAK,CAAC;gBACJ,QAAQ,EAAE,eAAe;gBACzB,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE;gBACjC,sBAAsB,CAAC,KAAK;oBAC1B,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,2CAA2C,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,KAAK,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE;oBACzL,CAAC,CAAC,IAAI,cAAc,CAAC,SAAS,cAAc,SAAS,sBAAsB,8BAA8B,CAAC,KAAK,CAAC,UAAU,CAAC,OAAO,KAAK,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC,CAAC,KAAK;aAC3K,CAAC,CAAC;QACL,CAAC,CAAC,CACH,CACF,CAAC;IACJ,CAAC;CACF;AASD,SAAS,eAAe,CAAC,gBAAqC;IAC5D,OAAO,gBAAgB,CAAC,cAAc;QACpC,CAAC,CAAC,YAAY,CAAC,gBAAgB,CAAC,cAAc,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACtE,CAAC,CAAC,CAAC,CAA6E,EAAE,EAAE,CAAC,CAAC,CAAC;AAC3F,CAAC;AAED,SAAS,gBAAgB,CAAyB,gBAAqC;IACrF,OAAO,gBAAgB,CAAC,eAAe;QACrC,CAAC,CAAC,YAAY,CAAC,gBAAgB,CAAC,eAAe,CAAC,IAAI,CAAC,gBAAgB,CAAC,CAAC;QACvE,CAAC,CAAC,CAAC,CAAqC,EAAE,EAAE,CAAC,CAAC,CAAC;AACnD,CAAC;AAED,SAAS,YAAY,CAAQ,SAAsD;IACjF,OAAO,CAAC,KAAwB,EAAE,EAAE,CAClC,KAAK,CAAC,IAAI,CACR,QAAQ,CAAC,SAAS,CAAC,EACnB,MAAM,CAAC,CAAC,CAAC,EAAc,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC/B,CAAC;AACN,CAAC;AAED,KAAK,UAAU,qBAAqB,CAClC,IAAW,EACX,cAAwC;IAExC,OAAO;QACL,GAAG,IAAI;QACP,KAAK,EAAE,MAAM,uBAAuB,CAAC;YACnC,KAAK,EAAE,IAAI,CAAC,KAAK;YACjB,cAAc;SACf,CAAC;KACH,CAAC;AACJ,CAAC;AAED,SAAS,wBAAwB,CAAC,UAA2C;IAC3E,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,OAAO,EAAE,CAAC;IACZ,CAAC;IACD,OAAO,CAAC,GAAG,UAAU,CAAC,UAAU,EAAE,UAAU,CAAC,GAAG,CAAC,CAAC;AACpD,CAAC;AAED,SAAS,yBAAyB,CAAC,KAAoB,EAAE,oBAA+C;IACtG,IAAI,CAAC,oBAAoB,IAAI,CAAC,oBAAoB,CAAC,MAAM,EAAE,CAAC;QAC1D,OAAO,KAAK,CAAC;IACf,CAAC;IACD,MAAM,oBAAoB,GAAG,IAAI,CAAC;IAClC,IAAI,oBAAoB,CAAC,MAAM,GAAG,oBAAoB,EAAE,CAAC;QACvD,OAAO;YACL,GAAG,KAAK;YACR,KAAK,EAAE;;gBAEG,KAAK,CAAC,KAAK;mCACQ,oBAAoB,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC;OACzE;YACD,QAAQ,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,GAAG,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAgB,EAAE,CAAC,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;SACvH,CAAC;IACJ,CAAC;IACD,qBAAqB;IACrB,OAAO;QACL,GAAG,KAAK;QACR,KAAK,EAAE;;cAEG,KAAK,CAAC,KAAK;;KAEpB;QACD,QAAQ,EAAE,CAAC,GAAG,CAAC,KAAK,CAAC,QAAQ,IAAI,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,oBAAoB,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,CAAC;KACvG,CAAC;IACF,mBAAmB;AACrB,CAAC","sourcesContent":["/*---------------------------------------------------------------------------------------------\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\n * See LICENSE.md in the project root for license terms and full copyright notice.\n *--------------------------------------------------------------------------------------------*/\n\nimport {\n  catchError,\n  concat,\n  defaultIfEmpty,\n  defer,\n  EMPTY,\n  filter,\n  finalize,\n  from,\n  map,\n  merge,\n  mergeAll,\n  mergeMap,\n  Observable,\n  ObservableInput,\n  of,\n  Subject,\n  take,\n  takeUntil,\n  tap,\n} from \"rxjs\";\nimport { assert, BeEvent, Guid, StopWatch } from \"@itwin/core-bentley\";\nimport {\n  ConcatenatedValue,\n  createDefaultValueFormatter,\n  ECClassHierarchyInspector,\n  ECSchemaProvider,\n  ECSqlBinding,\n  ECSqlQueryDef,\n  Event,\n  formatConcatenatedValue,\n  InstanceKey,\n  IPrimitiveValueFormatter,\n  normalizeFullClassName,\n} from \"@itwin/presentation-shared\";\nimport { RowsLimitExceededError } from \"../HierarchyErrors.js\";\nimport { HierarchyFilteringPath } from \"../HierarchyFiltering.js\";\nimport { HierarchyNode, NonGroupingHierarchyNode, ParentHierarchyNode } from \"../HierarchyNode.js\";\nimport { GenericNodeKey, HierarchyNodeKey, IModelInstanceKey, InstancesNodeKey } from \"../HierarchyNodeKey.js\";\nimport { GetHierarchyNodesProps, HierarchyProvider } from \"../HierarchyProvider.js\";\nimport {\n  LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE,\n  LOGGING_NAMESPACE_INTERNAL as BASE_LOGGING_NAMESPACE_INTERNAL,\n  LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE,\n  createNodeIdentifierForLogging,\n  hasChildren,\n} from \"../internal/Common.js\";\nimport { eachValueFrom } from \"../internal/EachValueFrom.js\";\nimport { doLog, log } from \"../internal/LoggingUtils.js\";\nimport { partition } from \"../internal/operators/Partition.js\";\nimport { reduceToMergeMapList } from \"../internal/operators/ReduceToMergeMap.js\";\nimport { shareReplayWithErrors } from \"../internal/operators/ShareReplayWithErrors.js\";\nimport { sortNodesByLabelOperator } from \"../internal/operators/Sorting.js\";\nimport { SubscriptionScheduler } from \"../internal/SubscriptionScheduler.js\";\nimport { FilteringHierarchyDefinition } from \"./FilteringHierarchyDefinition.js\";\nimport { HierarchyCache } from \"./HierarchyCache.js\";\nimport { DefineHierarchyLevelProps, HierarchyDefinition, HierarchyNodesDefinition } from \"./IModelHierarchyDefinition.js\";\nimport {\n  ProcessedGenericHierarchyNode,\n  ProcessedGroupingHierarchyNode,\n  ProcessedHierarchyNode,\n  ProcessedInstanceHierarchyNode,\n  SourceGenericHierarchyNode,\n  SourceInstanceHierarchyNode,\n} from \"./IModelHierarchyNode.js\";\nimport { LimitingECSqlQueryExecutor } from \"./LimitingECSqlQueryExecutor.js\";\nimport { NodeSelectClauseColumnNames } from \"./NodeSelectQueryFactory.js\";\nimport { createDetermineChildrenOperator } from \"./operators/DetermineChildren.js\";\nimport { createGroupingOperator } from \"./operators/Grouping.js\";\nimport { createHideIfNoChildrenOperator } from \"./operators/HideIfNoChildren.js\";\nimport { createHideNodesInHierarchyOperator } from \"./operators/HideNodesInHierarchy.js\";\nimport { readNodes } from \"./TreeNodesReader.js\";\n\nconst LOGGING_NAMESPACE = `${BASE_LOGGING_NAMESPACE}.IModelHierarchyProvider`;\nconst LOGGING_NAMESPACE_INTERNAL = `${BASE_LOGGING_NAMESPACE_INTERNAL}.IModelHierarchyProvider`;\nconst LOGGING_NAMESPACE_PERFORMANCE = `${BASE_LOGGING_NAMESPACE_PERFORMANCE}.IModelHierarchyProvider`;\n\nconst DEFAULT_QUERY_CONCURRENCY = 10;\nconst DEFAULT_QUERY_CACHE_SIZE = 1;\n\n/**\n * Defines the strings used by hierarchy provider.\n * @public\n */\ninterface IModelHierarchyProviderLocalizedStrings {\n  /**\n   * A string for \"Unspecified\". Used for labels of property grouping nodes\n   * that group by an empty value.\n   */\n  unspecified: string;\n\n  /**\n   * A string for \"Other\". Used for label of a range property grouping node that\n   * groups values which don't fit into any other range.\n   */\n  other: string;\n}\n\n/** @public */\ntype IModelAccess = ECSchemaProvider & LimitingECSqlQueryExecutor & ECClassHierarchyInspector & { imodelKey: string };\n\n/**\n * Props for `createIModelHierarchyProvider`.\n * @public\n */\ninterface IModelHierarchyProviderProps {\n  /**\n   * An object that provides access to iModel's data and metadata.\n   *\n   * @see `ECSchemaProvider`\n   * @see `LimitingECSqlQueryExecutor`\n   * @see `ECClassHierarchyInspector`\n   */\n  imodelAccess: IModelAccess;\n\n  /** An event that is raised when iModel data, that may affect the hierarchy, changes. */\n  imodelChanged?: Event<() => void>;\n\n  /**\n   * A function that returns a hierarchy definition, describing how the hierarchy that the provider should be create. The\n   * function is called once during the provider's construction.\n   */\n  hierarchyDefinition: HierarchyDefinition;\n\n  /** Maximum number of queries that the provider attempts to execute in parallel. Defaults to `10`. */\n  queryConcurrency?: number;\n  /**\n   * The amount of queries whose results are stored in-memory for quick retrieval. Defaults to `1`,\n   * which means only results of the last run query are cached.\n   */\n  queryCacheSize?: number;\n\n  /** A set of localized strings to use. Defaults to English strings. */\n  localizedStrings?: Partial<IModelHierarchyProviderLocalizedStrings>;\n\n  /**\n   * A values formatter for formatting node labels. Defaults to the\n   * result of `createDefaultValueFormatter` called with default parameters. May be overridden\n   * by calling `setFormatter` on the provider instance.\n   */\n  formatter?: IPrimitiveValueFormatter;\n  /** Props for filtering the hierarchy. */\n  filtering?: {\n    /** A list of node identifiers from root to target node. */\n    paths: HierarchyFilteringPath[];\n  };\n}\n\n/**\n * Creates an instance of `HierarchyProvider` that creates a hierarchy based on given iModel and\n * a hierarchy definition, which defines each hierarchy level through ECSQL queries.\n *\n * @public\n */\nexport function createIModelHierarchyProvider(props: IModelHierarchyProviderProps): HierarchyProvider & { dispose: () => void } {\n  return new IModelHierarchyProviderImpl(props);\n}\n\ninterface RequestContextProp {\n  requestContext: {\n    requestId: string;\n  };\n}\n\nclass IModelHierarchyProviderImpl implements HierarchyProvider {\n  private _imodelAccess: IModelAccess;\n  private _imodelChanged: Event<() => void>;\n  private _valuesFormatter: IPrimitiveValueFormatter;\n  private _sourceHierarchyDefinition: HierarchyDefinition;\n  private _activeHierarchyDefinition: HierarchyDefinition;\n  private _localizedStrings: IModelHierarchyProviderLocalizedStrings;\n  private _queryScheduler: SubscriptionScheduler;\n  private _nodesCache?: HierarchyCache<HierarchyCacheEntry>;\n  private _unsubscribe?: () => void;\n  private _dispose = new Subject<void>();\n\n  public constructor(props: IModelHierarchyProviderProps) {\n    this._imodelAccess = props.imodelAccess;\n    this._imodelChanged = props.imodelChanged ?? new BeEvent();\n    this._activeHierarchyDefinition = this._sourceHierarchyDefinition = props.hierarchyDefinition;\n    this._valuesFormatter = props?.formatter ?? createDefaultValueFormatter();\n    this._localizedStrings = { other: \"Other\", unspecified: \"Not specified\", ...props?.localizedStrings };\n    this._queryScheduler = new SubscriptionScheduler(props.queryConcurrency ?? DEFAULT_QUERY_CONCURRENCY);\n    this.setHierarchyFilter(props.filtering);\n\n    const queryCacheSize = props.queryCacheSize ?? DEFAULT_QUERY_CACHE_SIZE;\n    if (queryCacheSize !== 0) {\n      this._nodesCache = new HierarchyCache({\n        // we divide the size by 2, because each variation also counts as a query that we cache\n        size: Math.ceil(queryCacheSize / 2),\n        variationsCount: 1,\n      });\n      this._unsubscribe = props.imodelChanged?.addListener(() => {\n        this.invalidateHierarchyCache(\"Data source changed\");\n        this._dispose.next();\n      });\n    }\n  }\n\n  public dispose() {\n    this._dispose.next();\n    this._unsubscribe?.();\n  }\n\n  public get hierarchyChanged() {\n    return this._imodelChanged;\n  }\n\n  private invalidateHierarchyCache(reason?: string) {\n    doLog({\n      category: `${LOGGING_NAMESPACE}.Events`,\n      message: /* c8 ignore next */ () => (reason ? `${reason}: clear hierarchy cache` : `Clear hierarchy cache`),\n    });\n    this._nodesCache?.clear();\n  }\n\n  /**\n   * Sets `HierarchyProvider` values formatter that formats nodes' labels. If provided `undefined`, then defaults to the\n   * result of `createDefaultValueFormatter` called with default parameters.\n   */\n  public setFormatter(formatter: IPrimitiveValueFormatter | undefined) {\n    this._valuesFormatter = formatter ?? createDefaultValueFormatter();\n  }\n\n  public setHierarchyFilter(props: IModelHierarchyProviderProps[\"filtering\"]) {\n    if (!props) {\n      if (this._sourceHierarchyDefinition !== this._activeHierarchyDefinition) {\n        this._activeHierarchyDefinition = this._sourceHierarchyDefinition;\n        this.invalidateHierarchyCache(\"Hierarchy filter reset\");\n      }\n      return;\n    }\n    this._activeHierarchyDefinition = new FilteringHierarchyDefinition({\n      imodelAccess: this._imodelAccess,\n      source: this._sourceHierarchyDefinition,\n      nodeIdentifierPaths: props.paths,\n    });\n    this.invalidateHierarchyCache(\"Hierarchy filter set\");\n    this._dispose.next();\n  }\n\n  private onGroupingNodeCreated(groupingNode: ProcessedGroupingHierarchyNode, props: GetHierarchyNodesProps) {\n    this._nodesCache?.set({ ...props, parentNode: groupingNode }, { observable: from(groupingNode.children), processingStatus: \"pre-processed\" });\n  }\n\n  private createHierarchyLevelDefinitionsObservable(props: DefineHierarchyLevelProps & RequestContextProp): Observable<HierarchyNodesDefinition> {\n    const { requestContext, ...defineHierarchyLevelProps } = props;\n    doLog({\n      category: LOGGING_NAMESPACE_PERFORMANCE,\n      message: /* c8 ignore next */ () =>\n        `[${requestContext.requestId}] Requesting hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    return from(\n      defer(async () => {\n        let parentNode = props.parentNode;\n        if (parentNode && HierarchyNode.isGeneric(parentNode) && parentNode.key.source && parentNode.key.source !== this._imodelAccess.imodelKey) {\n          return [];\n        }\n        if (parentNode && HierarchyNode.isInstancesNode(parentNode)) {\n          parentNode = {\n            ...parentNode,\n            key: {\n              ...parentNode.key,\n              instanceKeys: parentNode.key.instanceKeys.filter((ik) => !ik.imodelKey || ik.imodelKey === this._imodelAccess.imodelKey),\n            },\n          };\n          assert(HierarchyNodeKey.isInstances(parentNode.key));\n          if (parentNode.key.instanceKeys.length === 0) {\n            return [];\n          }\n        }\n        return this._activeHierarchyDefinition.defineHierarchyLevel({ ...defineHierarchyLevelProps, parentNode });\n      }),\n    ).pipe(\n      mergeAll(),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* c8 ignore next */ () =>\n            `[${requestContext.requestId}] Received all hierarchy level definitions for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createSourceNodesObservable(\n    props: DefineHierarchyLevelProps & { hierarchyLevelSizeLimit?: number | \"unbounded\"; filteredInstanceKeys?: InstanceKey[] } & RequestContextProp,\n  ): SourceNodesObservable {\n    // pipe definitions to nodes and put \"share replay\" on it\n    return this.createHierarchyLevelDefinitionsObservable(props).pipe(\n      mergeMap((def): ObservableInput<SourceHierarchyNode> => {\n        if (HierarchyNodesDefinition.isGenericNode(def)) {\n          return of({ ...def.node, key: { type: \"generic\", id: def.node.key, source: this._imodelAccess.imodelKey } });\n        }\n        return this._queryScheduler.scheduleSubscription(\n          of(def.query).pipe(\n            map((query) => filterQueryByInstanceKeys(query, props.filteredInstanceKeys)),\n            mergeMap((query) =>\n              readNodes({\n                queryExecutor: this._imodelAccess,\n                query,\n                limit: props.hierarchyLevelSizeLimit,\n                parser: this._activeHierarchyDefinition.parseNode\n                  ? async (row) => this._activeHierarchyDefinition.parseNode!(row, props.parentNode)\n                  : undefined,\n              }),\n            ),\n            map((node) => ({\n              ...node,\n              key: { ...node.key, instanceKeys: node.key.instanceKeys.map((key) => ({ ...key, imodelKey: this._imodelAccess.imodelKey })) },\n            })),\n          ),\n        );\n      }),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Read all child nodes ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n      shareReplayWithErrors(),\n    );\n  }\n\n  private createPreProcessedNodesObservable(\n    queryNodesObservable: SourceNodesObservable,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<ProcessedHierarchyNode> {\n    return queryNodesObservable.pipe(\n      // we're going to be mutating the nodes, but don't want to mutate the original one, so just clone it here once\n      map((node) => ({ ...node })),\n      // set parent node keys on the source node\n      map((node) => Object.assign(node, { parentKeys: createParentNodeKeysList(props.parentNode) })),\n      // format `ConcatenatedValue` labels into string labels\n      mergeMap(async (node) => applyLabelsFormatting(node, this._valuesFormatter)),\n      // we have `ProcessedHierarchyNode` from here\n      preProcessNodes(this._activeHierarchyDefinition),\n      // process hiding\n      createHideIfNoChildrenOperator((n) => this.getChildNodesObservables({ parentNode: n, requestContext: props.requestContext }).hasNodes),\n      createHideNodesInHierarchyOperator(\n        // note: for child nodes created because of hidden parent, we want to use parent's request props (instance filter, limit)\n        (n) => this.getChildNodesObservables({ ...props, parentNode: n }).processedNodes,\n        false,\n      ),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* c8 ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished pre-processing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createProcessedNodesObservable(\n    preprocessedNodesObservable: Observable<ProcessedHierarchyNode>,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<ProcessedHierarchyNode> {\n    return preprocessedNodesObservable.pipe(\n      createGroupingOperator(this._imodelAccess, props.parentNode, this._valuesFormatter, this._localizedStrings, (gn) =>\n        this.onGroupingNodeCreated(gn, props),\n      ),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* c8 ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished grouping child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createFinalizedNodesObservable(\n    processedNodesObservable: Observable<ProcessedHierarchyNode>,\n    props: GetHierarchyNodesProps & RequestContextProp,\n  ): Observable<HierarchyNode> {\n    return processedNodesObservable.pipe(\n      createDetermineChildrenOperator((n) => this.getChildNodesObservables({ parentNode: n, requestContext: props.requestContext }).hasNodes),\n      postProcessNodes(this._activeHierarchyDefinition),\n      sortNodesByLabelOperator,\n      map((n): HierarchyNode => {\n        if (\"processingParams\" in n) {\n          delete n.processingParams;\n        }\n        return Object.assign(n, {\n          children: hasChildren(n),\n        });\n      }),\n      takeUntil(this._dispose),\n      finalize(() =>\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          message: /* c8 ignore next */ () =>\n            `[${props.requestContext.requestId}] Finished finalizing child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n        }),\n      ),\n    );\n  }\n\n  private createHasNodesObservable(\n    preprocessedNodesObservable: Observable<ProcessedHierarchyNode>,\n    possiblyKnownChildrenObservable: SourceNodesObservable | undefined,\n    props: RequestContextProp,\n  ): Observable<boolean> {\n    const loggingCategory = `${LOGGING_NAMESPACE_INTERNAL}.HasNodes`;\n    return concat((possiblyKnownChildrenObservable ?? EMPTY).pipe(filter((n) => hasChildren(n))), preprocessedNodesObservable).pipe(\n      log({\n        category: loggingCategory,\n        message: /* c8 ignore next */ (n) => `[${props.requestContext.requestId}] Node before mapping to 'true': ${createNodeIdentifierForLogging(n)}`,\n      }),\n      take(1),\n      map(() => true),\n      defaultIfEmpty(false),\n      catchError((e: Error) => {\n        doLog({\n          category: loggingCategory,\n          message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Error while determining children: ${e.message}`,\n        });\n        if (e instanceof RowsLimitExceededError) {\n          return of(true);\n        }\n        throw e;\n      }),\n      log({ category: loggingCategory, message: /* c8 ignore next */ (r) => `[${props.requestContext.requestId}] Result: ${r}` }),\n    );\n  }\n\n  private getCachedObservableEntry(props: GetHierarchyNodesProps & RequestContextProp): HierarchyCacheEntry {\n    const loggingCategory = `${LOGGING_NAMESPACE}.QueryResultsCache`;\n    const { parentNode, ...restProps } = props;\n    const cached = props.ignoreCache || !this._nodesCache ? undefined : this._nodesCache.get(props);\n    if (cached) {\n      doLog({\n        category: loggingCategory,\n        message: /* c8 ignore next */ () =>\n          `[${props.requestContext.requestId}] Found query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,\n      });\n      return cached;\n    }\n\n    // if we don't find an entry for a grouping node, we load its instances by getting a query and applying\n    // a filter based on grouped instance keys\n    let filteredInstanceKeys: InstanceKey[] | undefined;\n    let parentNonGroupingNode: ParentHierarchyNode<NonGroupingHierarchyNode> | undefined;\n    if (parentNode) {\n      if (HierarchyNode.isGroupingNode(parentNode)) {\n        parentNonGroupingNode = parentNode.nonGroupingAncestor;\n        filteredInstanceKeys = parentNode.groupedInstanceKeys;\n      } else {\n        // not sure why type checker doesn't pick this up\n        assert(HierarchyNode.isGeneric(parentNode) || HierarchyNode.isInstancesNode(parentNode));\n        parentNonGroupingNode = parentNode;\n      }\n    }\n\n    const nonGroupingNodeChildrenRequestProps = {\n      ...restProps,\n      parentNode: parentNonGroupingNode,\n      ...(filteredInstanceKeys ? { filteredInstanceKeys } : undefined),\n    };\n    const value = { observable: this.createSourceNodesObservable(nonGroupingNodeChildrenRequestProps), processingStatus: \"none\" as const };\n    this._nodesCache?.set(nonGroupingNodeChildrenRequestProps, value);\n    doLog({\n      category: loggingCategory,\n      message: /* c8 ignore next */ () => `[${props.requestContext.requestId}] Saved query nodes observable for ${createNodeIdentifierForLogging(parentNode)}`,\n    });\n    return value;\n  }\n\n  private getChildNodesObservables(props: GetHierarchyNodesProps & { hierarchyLevelSizeLimit?: number | \"unbounded\" } & RequestContextProp) {\n    const entry = this.getCachedObservableEntry(props);\n    switch (entry.processingStatus) {\n      case \"none\": {\n        const pre = this.createPreProcessedNodesObservable(entry.observable, props);\n        const post = this.createProcessedNodesObservable(pre, props);\n        return {\n          processedNodes: post,\n          hasNodes: this.createHasNodesObservable(pre, entry.observable, props),\n          finalizedNodes: this.createFinalizedNodesObservable(post, props),\n        };\n      }\n      case \"pre-processed\": {\n        const post = this.createProcessedNodesObservable(entry.observable, props);\n        return {\n          processedNodes: post,\n          hasNodes: this.createHasNodesObservable(entry.observable, undefined, props),\n          finalizedNodes: this.createFinalizedNodesObservable(post, props),\n        };\n      }\n    }\n  }\n\n  /**\n   * Creates and runs a query based on provided props, then processes retrieved nodes and returns them.\n   */\n  public getNodes(props: GetHierarchyNodesProps): AsyncIterableIterator<HierarchyNode> {\n    const requestContext = { requestId: Guid.createValue() };\n    const loggingCategory = `${LOGGING_NAMESPACE}.GetNodes`;\n    const timer = new StopWatch(undefined, true);\n    let error: any;\n    let nodesCount = 0;\n    doLog({\n      category: loggingCategory,\n      message: /* c8 ignore next */ () => `[${requestContext.requestId}] Requesting child nodes for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    return eachValueFrom(\n      this.getChildNodesObservables({ ...props, requestContext }).finalizedNodes.pipe(\n        tap(() => ++nodesCount),\n        catchError((e) => {\n          error = e;\n          throw e;\n        }),\n        finalize(() => {\n          doLog({\n            category: loggingCategory,\n            message: /* c8 ignore next */ () =>\n              /* c8 ignore next 3 */ error\n                ? `[${requestContext.requestId}] Error creating child nodes for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`\n                : `[${requestContext.requestId}] Returned ${nodesCount} child nodes for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,\n          });\n        }),\n      ),\n    );\n  }\n\n  private getNodeInstanceKeysObs(props: Omit<GetHierarchyNodesProps, \"ignoreCache\"> & RequestContextProp): Observable<InstanceKey> {\n    const { parentNode, instanceFilter, hierarchyLevelSizeLimit = \"unbounded\", requestContext } = props;\n    if (parentNode && HierarchyNode.isGroupingNode(parentNode)) {\n      return from(parentNode.groupedInstanceKeys);\n    }\n    assert(!parentNode || HierarchyNode.isGeneric(parentNode) || HierarchyNode.isInstancesNode(parentNode));\n\n    const hierarchyLevelDefinitions = this.createHierarchyLevelDefinitionsObservable({ parentNode, instanceFilter, requestContext });\n\n    // split the definitions based on whether they're for generic nodes or for instance nodes\n    const [genericDefs, instanceDefs] = partition(hierarchyLevelDefinitions, HierarchyNodesDefinition.isGenericNode);\n\n    // query instance keys and a flag whether they should be hidden\n    const instanceKeys = instanceDefs.pipe(\n      mergeMap((def) =>\n        this._queryScheduler.scheduleSubscription(\n          of(def.query).pipe(\n            mergeMap(async (query) => {\n              const ecsql = `\n                SELECT\n                  ${NodeSelectClauseColumnNames.FullClassName},\n                  ${NodeSelectClauseColumnNames.ECInstanceId},\n                  ${NodeSelectClauseColumnNames.HideNodeInHierarchy}\n                FROM (\n                  ${query.ecsql}\n                )\n              `;\n              const reader = this._imodelAccess.createQueryReader({ ...query, ecsql }, { rowFormat: \"Indexes\", limit: hierarchyLevelSizeLimit });\n              return from(reader).pipe(\n                map((row) => ({\n                  key: {\n                    className: normalizeFullClassName(row[0]),\n                    id: row[1],\n                    imodelKey: this._imodelAccess.imodelKey,\n                  } satisfies IModelInstanceKey,\n                  hide: !!row[2],\n                })),\n              );\n            }),\n            mergeAll(),\n          ),\n        ),\n      ),\n    );\n    // split the instance keys observable based on whether they should be hidden or not\n    const [visibleNodeInstanceKeys, hiddenNodeInstanceKeys] = partition(instanceKeys, ({ hide }) => !hide);\n\n    // hidden items' handling:\n    // - if a generic node is hidden, we'll want to retrieve instance keys of its children, otherwise we don't care about it,\n    // - if an instance key is hidden, we want to create a merged node similar to what we do in `createHideNodesInHierarchyOperator`\n    const hiddenParentNodes = merge(\n      genericDefs.pipe(\n        filter((def) => !!def.node.processingParams?.hideInHierarchy),\n        map(\n          (def): GenericNodeKey => ({\n            type: \"generic\",\n            id: def.node.key,\n            source: this._imodelAccess.imodelKey,\n          }),\n        ),\n      ),\n      hiddenNodeInstanceKeys.pipe(\n        // first merge all keys by class\n        reduceToMergeMapList(\n          ({ key }) => key.className,\n          ({ key }) => key.id,\n        ),\n        // then, for each class, create an instance key\n        mergeMap((mergedMap): InstancesNodeKey[] =>\n          [...mergedMap.entries()].map(([className, ids]) => ({\n            type: \"instances\",\n            instanceKeys: ids.map((id) => ({ className, id, imodelKey: this._imodelAccess.imodelKey })),\n          })),\n        ),\n      ),\n    ).pipe(\n      map((key) => ({\n        key,\n        parentKeys: [],\n        label: \"\",\n      })),\n    );\n\n    // merge visible instance keys from this level & the ones we get recursively requesting from deeper levels\n    return merge(\n      visibleNodeInstanceKeys.pipe(map(({ key }) => key)),\n      hiddenParentNodes.pipe(mergeMap((hiddenNode) => this.getNodeInstanceKeysObs({ parentNode: hiddenNode, requestContext }))),\n    ).pipe(takeUntil(this._dispose));\n  }\n\n  /**\n   * Creates an iterator for all child hierarchy level instance keys, taking into account any hidden hierarchy levels\n   * that there may be under the given parent node.\n   */\n  public getNodeInstanceKeys(props: Omit<GetHierarchyNodesProps, \"ignoreCache\">): AsyncIterableIterator<InstanceKey> {\n    const requestContext = { requestId: Guid.createValue() };\n    const loggingCategory = `${LOGGING_NAMESPACE}.GetNodeInstanceKeys`;\n    const timer = new StopWatch(undefined, true);\n    doLog({\n      category: loggingCategory,\n      message: /* c8 ignore next */ () => `[${requestContext.requestId}] Requesting keys for ${createNodeIdentifierForLogging(props.parentNode)}`,\n    });\n    let error: any;\n    let keysCount = 0;\n    return eachValueFrom(\n      this.getNodeInstanceKeysObs({ ...props, requestContext }).pipe(\n        tap(() => ++keysCount),\n        catchError((e) => {\n          error = e;\n          throw e;\n        }),\n        finalize(() => {\n          doLog({\n            category: loggingCategory,\n            message: /* c8 ignore next */ () =>\n              /* c8 ignore next 3 */ error\n                ? `[${requestContext.requestId}] Error creating node instance keys for ${createNodeIdentifierForLogging(props.parentNode)}: ${error instanceof Error ? error.message : error.toString()}`\n                : `[${requestContext.requestId}] Returned ${keysCount} instance keys for ${createNodeIdentifierForLogging(props.parentNode)} in ${timer.currentSeconds.toFixed(2)} s.`,\n          });\n        }),\n      ),\n    );\n  }\n}\n\ntype SourceHierarchyNode = SourceInstanceHierarchyNode | (Omit<SourceGenericHierarchyNode, \"key\"> & { key: GenericNodeKey });\ntype SourceNodesObservable = Observable<SourceHierarchyNode>;\ntype ProcessedNodesObservable = Observable<ProcessedHierarchyNode>;\ntype HierarchyCacheEntry =\n  | { observable: SourceNodesObservable; processingStatus: \"none\" }\n  | { observable: ProcessedNodesObservable; processingStatus: \"pre-processed\" };\n\nfunction preProcessNodes(hierarchyFactory: HierarchyDefinition) {\n  return hierarchyFactory.preProcessNode\n    ? processNodes(hierarchyFactory.preProcessNode.bind(hierarchyFactory))\n    : (o: Observable<ProcessedGenericHierarchyNode | ProcessedInstanceHierarchyNode>) => o;\n}\n\nfunction postProcessNodes<ProcessedHierarchyNode>(hierarchyFactory: HierarchyDefinition) {\n  return hierarchyFactory.postProcessNode\n    ? processNodes(hierarchyFactory.postProcessNode.bind(hierarchyFactory))\n    : (o: Observable<ProcessedHierarchyNode>) => o;\n}\n\nfunction processNodes<TNode>(processor: (node: TNode) => Promise<TNode | undefined>) {\n  return (nodes: Observable<TNode>) =>\n    nodes.pipe(\n      mergeMap(processor),\n      filter((n): n is TNode => !!n),\n    );\n}\n\nasync function applyLabelsFormatting<TNode extends { label: string | ConcatenatedValue }>(\n  node: TNode,\n  valueFormatter: IPrimitiveValueFormatter,\n): Promise<TNode & { label: string }> {\n  return {\n    ...node,\n    label: await formatConcatenatedValue({\n      value: node.label,\n      valueFormatter,\n    }),\n  };\n}\n\nfunction createParentNodeKeysList(parentNode: ParentHierarchyNode | undefined) {\n  if (!parentNode) {\n    return [];\n  }\n  return [...parentNode.parentKeys, parentNode.key];\n}\n\nfunction filterQueryByInstanceKeys(query: ECSqlQueryDef, filteredInstanceKeys: InstanceKey[] | undefined): ECSqlQueryDef {\n  if (!filteredInstanceKeys || !filteredInstanceKeys.length) {\n    return query;\n  }\n  const MAX_ALLOWED_BINDINGS = 1000;\n  if (filteredInstanceKeys.length < MAX_ALLOWED_BINDINGS) {\n    return {\n      ...query,\n      ecsql: `\n        SELECT *\n        FROM (${query.ecsql}) q\n        WHERE q.ECInstanceId IN (${filteredInstanceKeys.map(() => \"?\").join(\",\")})\n      `,\n      bindings: [...(query.bindings ?? []), ...filteredInstanceKeys.map((k): ECSqlBinding => ({ type: \"id\", value: k.id }))],\n    };\n  }\n  /* c8 ignore start */\n  return {\n    ...query,\n    ecsql: `\n      SELECT *\n      FROM (${query.ecsql}) q\n      WHERE InVirtualSet(?, q.ECInstanceId)\n    `,\n    bindings: [...(query.bindings ?? []), { type: \"idset\", value: filteredInstanceKeys.map((k) => k.id) }],\n  };\n  /* c8 ignore end */\n}\n"]}

package/lib/esm/hierarchies/imodel/LimitingECSqlQueryExecutor.d.ts

@@ -1,7 +1,7 @@
 import { ECSqlQueryDef, ECSqlQueryExecutor, ECSqlQueryReaderOptions } from "@itwin/presentation-shared";
 /**
  * An interface for something that knows how to create a limiting ECSQL query reader.
- * @beta
+ * @public
  */
 export interface LimitingECSqlQueryExecutor {
     /**
@@ -15,7 +15,7 @@ export interface LimitingECSqlQueryExecutor {
 }
 /**
  * Creates an `LimitingECSqlQueryExecutor` that throws `RowsLimitExceededError` if the query exceeds given amount of rows.
- * @beta
+ * @public
  */
 export declare function createLimitingECSqlQueryExecutor(baseExecutor: ECSqlQueryExecutor, defaultLimit: number | "unbounded"): LimitingECSqlQueryExecutor;
 //# sourceMappingURL=LimitingECSqlQueryExecutor.d.ts.map

package/lib/esm/hierarchies/imodel/LimitingECSqlQueryExecutor.js

@@ -4,12 +4,12 @@
  *--------------------------------------------------------------------------------------------*/
 import { Guid, StopWatch } from "@itwin/core-bentley";
 import { createMainThreadReleaseOnTimePassedHandler, trimWhitespace, } from "@itwin/presentation-shared";
-import { RowsLimitExceededError } from "../HierarchyErrors";
-import { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE } from "../internal/Common";
-import { doLog } from "../internal/LoggingUtils";
+import { RowsLimitExceededError } from "../HierarchyErrors.js";
+import { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE } from "../internal/Common.js";
+import { doLog } from "../internal/LoggingUtils.js";
 /**
  * Creates an `LimitingECSqlQueryExecutor` that throws `RowsLimitExceededError` if the query exceeds given amount of rows.
- * @beta
+ * @public
  */
 export function createLimitingECSqlQueryExecutor(baseExecutor, defaultLimit) {
     return {
@@ -71,7 +71,7 @@ function createQueryLogger(query, firstStepWarningThreshold = 3000, allRowsWarni
     const queryId = Guid.createValue();
     doLog({
         category: LOGGING_NAMESPACE,
-        message: /* istanbul ignore next */ () => `Executing query [${queryId}]: ${createQueryLogMessage(query)}`,
+        message: /* c8 ignore next */ () => `Executing query [${queryId}]: ${createQueryLogMessage(query)}`,
     });
     let firstStep = true;
     let rowsCount = 0;
@@ -79,27 +79,29 @@ function createQueryLogger(query, firstStepWarningThreshold = 3000, allRowsWarni
     return {
         onStep() {
             if (firstStep) {
-                // istanbul ignore next
+                /* c8 ignore start */
                 doLog({
                     category: LOGGING_NAMESPACE_PERFORMANCE,
                     severity: timer.current.milliseconds >= firstStepWarningThreshold ? "warning" : "trace",
                     message: () => `[${queryId}] First step took ${timer.currentSeconds} s.`,
                 });
+                /* c8 ignore end */
                 firstStep = false;
             }
             ++rowsCount;
         },
         onComplete() {
-            // istanbul ignore next
+            /* c8 ignore start */
             doLog({
                 category: LOGGING_NAMESPACE_PERFORMANCE,
                 severity: timer.current.milliseconds >= allRowsWarningThreshold ? "warning" : "trace",
                 message: () => `[${queryId}] Query took ${timer.currentSeconds} s. for ${rowsCount} rows.`,
             });
+            /* c8 ignore end */
         },
     };
 }
-// istanbul ignore next
+/* c8 ignore start */
 function createQueryLogMessage(query) {
     const ctes = query.ctes?.map((cte) => `    ${trimWhitespace(cte)}`).join(", \n");
     const bindings = query.bindings?.map((b) => JSON.stringify(b.value)).join(", ");
@@ -114,4 +116,5 @@ function createQueryLogMessage(query) {
     output += "}";
     return output;
 }
+/* c8 ignore end */
 //# sourceMappingURL=LimitingECSqlQueryExecutor.js.map

package/lib/esm/hierarchies/imodel/LimitingECSqlQueryExecutor.js.map

@@ -1 +1 @@
-{"version":3,"file":"LimitingECSqlQueryExecutor.js","sourceRoot":"","sources":["../../../../src/hierarchies/imodel/LimitingECSqlQueryExecutor.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EACL,0CAA0C,EAK1C,cAAc,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,sBAAsB,EAAE,MAAM,oBAAoB,CAAC;AAC5D,OAAO,EAAE,iBAAiB,IAAI,sBAAsB,EAAE,6BAA6B,IAAI,kCAAkC,EAAE,MAAM,oBAAoB,CAAC;AACtJ,OAAO,EAAE,KAAK,EAAE,MAAM,0BAA0B,CAAC;AAkBjD;;;GAGG;AACH,MAAM,UAAU,gCAAgC,CAAC,YAAgC,EAAE,YAAkC;IACnH,OAAO;QACL,KAAK,CAAC,CAAC,iBAAiB,CAAC,KAAoB,EAAE,MAAmE;YAChH,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,IAAI,EAAE,CAAC;YAC3D,MAAM,KAAK,GAAG,WAAW,IAAI,YAAY,CAAC;YAC1C,MAAM,WAAW,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,iBAAiB,GAAG,0CAA0C,EAAE,CAAC;YAEvE,2CAA2C;YAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,iBAAiB,CAAC,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC7G,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;wBAC/B,MAAM,iBAAiB,EAAE,CAAC;wBAC1B,WAAW,CAAC,MAAM,EAAE,CAAC;wBACrB,MAAM,GAAG,CAAC;oBACZ,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,WAAW,CAAC,UAAU,EAAE,CAAC;gBAC3B,CAAC;gBACD,OAAO;YACT,CAAC;YAED,2DAA2D;YAC3D,MAAM,MAAM,GAAoB,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;oBAC/B,MAAM,iBAAiB,EAAE,CAAC;oBAC1B,WAAW,CAAC,MAAM,EAAE,CAAC;oBACrB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACjB,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;wBAC1B,MAAM,IAAI,sBAAsB,CAAC,KAAK,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,WAAW,CAAC,UAAU,EAAE,CAAC;YAC3B,CAAC;YAED,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBACzB,MAAM,iBAAiB,EAAE,CAAC;gBAC1B,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,EAAE,KAA2B;IAC1D,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO;;YAEG,KAAK;YACL,KAAK,GAAG,CAAC;GAClB,CAAC;AACJ,CAAC;AAED,MAAM,iBAAiB,GAAG,GAAG,sBAAsB,UAAU,CAAC;AAC9D,MAAM,6BAA6B,GAAG,GAAG,kCAAkC,UAAU,CAAC;AACtF,SAAS,iBAAiB,CAAC,KAAoB,EAAE,yBAAyB,GAAG,IAAI,EAAE,uBAAuB,GAAG,IAAI;IAC/G,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACnC,KAAK,CAAC;QACJ,QAAQ,EAAE,iBAAiB;QAC3B,OAAO,EAAE,0BAA0B,CAAC,GAAG,EAAE,CAAC,oBAAoB,OAAO,MAAM,qBAAqB,CAAC,KAAK,CAAC,EAAE;KAC1G,CAAC,CAAC;IAEH,IAAI,SAAS,GAAG,IAAI,CAAC;IACrB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC7C,OAAO;QACL,MAAM;YACJ,IAAI,SAAS,EAAE,CAAC;gBACd,uBAAuB;gBACvB,KAAK,CAAC;oBACJ,QAAQ,EAAE,6BAA6B;oBACvC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;oBACvF,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,qBAAqB,KAAK,CAAC,cAAc,KAAK;iBACzE,CAAC,CAAC;gBACH,SAAS,GAAG,KAAK,CAAC;YACpB,CAAC;YACD,EAAE,SAAS,CAAC;QACd,CAAC;QACD,UAAU;YACR,uBAAuB;YACvB,KAAK,CAAC;gBACJ,QAAQ,EAAE,6BAA6B;gBACvC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,uBAAuB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;gBACrF,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,gBAAgB,KAAK,CAAC,cAAc,WAAW,SAAS,QAAQ;aAC3F,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED,uBAAuB;AACvB,SAAS,qBAAqB,CAAC,KAAoB;IACjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjF,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChF,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,eAAe,IAAI,UAAU,CAAC;IAC1C,CAAC;IACD,MAAM,IAAI,YAAY,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;IACxD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,gBAAgB,QAAQ,OAAO,CAAC;IAC5C,CAAC;IACD,MAAM,IAAI,GAAG,CAAC;IACd,OAAO,MAAM,CAAC;AAChB,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 { Guid, StopWatch } from \"@itwin/core-bentley\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n  trimWhitespace,\n} from \"@itwin/presentation-shared\";\nimport { RowsLimitExceededError } from \"../HierarchyErrors\";\nimport { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE } from \"../internal/Common\";\nimport { doLog } from \"../internal/LoggingUtils\";\n\n/**\n * An interface for something that knows how to create a limiting ECSQL query reader.\n * @beta\n */\nexport interface LimitingECSqlQueryExecutor {\n  /**\n   * Creates a query reader for given query, but makes sure it doesn't return more than the configured\n   * limit of rows.\n   * @throws `RowsLimitExceededError` when the query returns more than configured limit of rows.\n   */\n  createQueryReader(\n    query: ECSqlQueryDef,\n    config?: ECSqlQueryReaderOptions & { limit?: number | \"unbounded\" },\n  ): ReturnType<ECSqlQueryExecutor[\"createQueryReader\"]>;\n}\n\n/**\n * Creates an `LimitingECSqlQueryExecutor` that throws `RowsLimitExceededError` if the query exceeds given amount of rows.\n * @beta\n */\nexport function createLimitingECSqlQueryExecutor(baseExecutor: ECSqlQueryExecutor, defaultLimit: number | \"unbounded\"): LimitingECSqlQueryExecutor {\n  return {\n    async *createQueryReader(query: ECSqlQueryDef, config?: ECSqlQueryReaderOptions & { limit?: number | \"unbounded\" }) {\n      const { limit: configLimit, ...restConfig } = config ?? {};\n      const limit = configLimit ?? defaultLimit;\n      const queryLogger = createQueryLogger(query);\n      const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n\n      // handle \"unbounded\" case without a buffer\n      const reader = baseExecutor.createQueryReader({ ...query, ecsql: addLimit(query.ecsql, limit) }, restConfig);\n      if (limit === \"unbounded\") {\n        try {\n          for await (const row of reader) {\n            await releaseMainThread();\n            queryLogger.onStep();\n            yield row;\n          }\n        } finally {\n          queryLogger.onComplete();\n        }\n        return;\n      }\n\n      // avoid streaming until we know the number of rows is okay\n      const buffer: ECSqlQueryRow[] = [];\n      try {\n        for await (const row of reader) {\n          await releaseMainThread();\n          queryLogger.onStep();\n          buffer.push(row);\n          if (buffer.length > limit) {\n            throw new RowsLimitExceededError(limit);\n          }\n        }\n      } finally {\n        queryLogger.onComplete();\n      }\n\n      for (const row of buffer) {\n        await releaseMainThread();\n        yield row;\n      }\n    },\n  };\n}\n\nfunction addLimit(ecsql: string, limit: number | \"unbounded\") {\n  if (limit === \"unbounded\") {\n    return ecsql;\n  }\n  return `\n    SELECT *\n    FROM (${ecsql})\n    LIMIT ${limit + 1}\n  `;\n}\n\nconst LOGGING_NAMESPACE = `${BASE_LOGGING_NAMESPACE}.Queries`;\nconst LOGGING_NAMESPACE_PERFORMANCE = `${BASE_LOGGING_NAMESPACE_PERFORMANCE}.Queries`;\nfunction createQueryLogger(query: ECSqlQueryDef, firstStepWarningThreshold = 3000, allRowsWarningThreshold = 5000) {\n  const queryId = Guid.createValue();\n  doLog({\n    category: LOGGING_NAMESPACE,\n    message: /* istanbul ignore next */ () => `Executing query [${queryId}]: ${createQueryLogMessage(query)}`,\n  });\n\n  let firstStep = true;\n  let rowsCount = 0;\n  const timer = new StopWatch(undefined, true);\n  return {\n    onStep() {\n      if (firstStep) {\n        // istanbul ignore next\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          severity: timer.current.milliseconds >= firstStepWarningThreshold ? \"warning\" : \"trace\",\n          message: () => `[${queryId}] First step took ${timer.currentSeconds} s.`,\n        });\n        firstStep = false;\n      }\n      ++rowsCount;\n    },\n    onComplete() {\n      // istanbul ignore next\n      doLog({\n        category: LOGGING_NAMESPACE_PERFORMANCE,\n        severity: timer.current.milliseconds >= allRowsWarningThreshold ? \"warning\" : \"trace\",\n        message: () => `[${queryId}] Query took ${timer.currentSeconds} s. for ${rowsCount} rows.`,\n      });\n    },\n  };\n}\n\n// istanbul ignore next\nfunction createQueryLogMessage(query: ECSqlQueryDef): string {\n  const ctes = query.ctes?.map((cte) => `    ${trimWhitespace(cte)}`).join(\", \\n\");\n  const bindings = query.bindings?.map((b) => JSON.stringify(b.value)).join(\", \");\n  let output = \"{\\n\";\n  if (ctes) {\n    output += `  ctes: [ \\n${ctes} \\n], \\n`;\n  }\n  output += `  ecsql: ${trimWhitespace(query.ecsql)}, \\n`;\n  if (bindings) {\n    output += `  bindings: [${bindings}], \\n`;\n  }\n  output += \"}\";\n  return output;\n}\n"]}
+{"version":3,"file":"LimitingECSqlQueryExecutor.js","sourceRoot":"","sources":["../../../../src/hierarchies/imodel/LimitingECSqlQueryExecutor.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG,OAAO,EAAE,IAAI,EAAE,SAAS,EAAE,MAAM,qBAAqB,CAAC;AACtD,OAAO,EACL,0CAA0C,EAK1C,cAAc,GACf,MAAM,4BAA4B,CAAC;AACpC,OAAO,EAAE,sBAAsB,EAAE,MAAM,uBAAuB,CAAC;AAC/D,OAAO,EAAE,iBAAiB,IAAI,sBAAsB,EAAE,6BAA6B,IAAI,kCAAkC,EAAE,MAAM,uBAAuB,CAAC;AACzJ,OAAO,EAAE,KAAK,EAAE,MAAM,6BAA6B,CAAC;AAkBpD;;;GAGG;AACH,MAAM,UAAU,gCAAgC,CAAC,YAAgC,EAAE,YAAkC;IACnH,OAAO;QACL,KAAK,CAAC,CAAC,iBAAiB,CAAC,KAAoB,EAAE,MAAmE;YAChH,MAAM,EAAE,KAAK,EAAE,WAAW,EAAE,GAAG,UAAU,EAAE,GAAG,MAAM,IAAI,EAAE,CAAC;YAC3D,MAAM,KAAK,GAAG,WAAW,IAAI,YAAY,CAAC;YAC1C,MAAM,WAAW,GAAG,iBAAiB,CAAC,KAAK,CAAC,CAAC;YAC7C,MAAM,iBAAiB,GAAG,0CAA0C,EAAE,CAAC;YAEvE,2CAA2C;YAC3C,MAAM,MAAM,GAAG,YAAY,CAAC,iBAAiB,CAAC,EAAE,GAAG,KAAK,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,CAAC,KAAK,EAAE,KAAK,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;YAC7G,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;gBAC1B,IAAI,CAAC;oBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;wBAC/B,MAAM,iBAAiB,EAAE,CAAC;wBAC1B,WAAW,CAAC,MAAM,EAAE,CAAC;wBACrB,MAAM,GAAG,CAAC;oBACZ,CAAC;gBACH,CAAC;wBAAS,CAAC;oBACT,WAAW,CAAC,UAAU,EAAE,CAAC;gBAC3B,CAAC;gBACD,OAAO;YACT,CAAC;YAED,2DAA2D;YAC3D,MAAM,MAAM,GAAoB,EAAE,CAAC;YACnC,IAAI,CAAC;gBACH,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;oBAC/B,MAAM,iBAAiB,EAAE,CAAC;oBAC1B,WAAW,CAAC,MAAM,EAAE,CAAC;oBACrB,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;oBACjB,IAAI,MAAM,CAAC,MAAM,GAAG,KAAK,EAAE,CAAC;wBAC1B,MAAM,IAAI,sBAAsB,CAAC,KAAK,CAAC,CAAC;oBAC1C,CAAC;gBACH,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,WAAW,CAAC,UAAU,EAAE,CAAC;YAC3B,CAAC;YAED,KAAK,MAAM,GAAG,IAAI,MAAM,EAAE,CAAC;gBACzB,MAAM,iBAAiB,EAAE,CAAC;gBAC1B,MAAM,GAAG,CAAC;YACZ,CAAC;QACH,CAAC;KACF,CAAC;AACJ,CAAC;AAED,SAAS,QAAQ,CAAC,KAAa,EAAE,KAA2B;IAC1D,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;QAC1B,OAAO,KAAK,CAAC;IACf,CAAC;IACD,OAAO;;YAEG,KAAK;YACL,KAAK,GAAG,CAAC;GAClB,CAAC;AACJ,CAAC;AAED,MAAM,iBAAiB,GAAG,GAAG,sBAAsB,UAAU,CAAC;AAC9D,MAAM,6BAA6B,GAAG,GAAG,kCAAkC,UAAU,CAAC;AACtF,SAAS,iBAAiB,CAAC,KAAoB,EAAE,yBAAyB,GAAG,IAAI,EAAE,uBAAuB,GAAG,IAAI;IAC/G,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;IACnC,KAAK,CAAC;QACJ,QAAQ,EAAE,iBAAiB;QAC3B,OAAO,EAAE,oBAAoB,CAAC,GAAG,EAAE,CAAC,oBAAoB,OAAO,MAAM,qBAAqB,CAAC,KAAK,CAAC,EAAE;KACpG,CAAC,CAAC;IAEH,IAAI,SAAS,GAAG,IAAI,CAAC;IACrB,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,MAAM,KAAK,GAAG,IAAI,SAAS,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;IAC7C,OAAO;QACL,MAAM;YACJ,IAAI,SAAS,EAAE,CAAC;gBACd,qBAAqB;gBACrB,KAAK,CAAC;oBACJ,QAAQ,EAAE,6BAA6B;oBACvC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,yBAAyB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;oBACvF,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,qBAAqB,KAAK,CAAC,cAAc,KAAK;iBACzE,CAAC,CAAC;gBACH,mBAAmB;gBACnB,SAAS,GAAG,KAAK,CAAC;YACpB,CAAC;YACD,EAAE,SAAS,CAAC;QACd,CAAC;QACD,UAAU;YACR,qBAAqB;YACrB,KAAK,CAAC;gBACJ,QAAQ,EAAE,6BAA6B;gBACvC,QAAQ,EAAE,KAAK,CAAC,OAAO,CAAC,YAAY,IAAI,uBAAuB,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,OAAO;gBACrF,OAAO,EAAE,GAAG,EAAE,CAAC,IAAI,OAAO,gBAAgB,KAAK,CAAC,cAAc,WAAW,SAAS,QAAQ;aAC3F,CAAC,CAAC;YACH,mBAAmB;QACrB,CAAC;KACF,CAAC;AACJ,CAAC;AAED,qBAAqB;AACrB,SAAS,qBAAqB,CAAC,KAAoB;IACjD,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,cAAc,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACjF,MAAM,QAAQ,GAAG,KAAK,CAAC,QAAQ,EAAE,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;IAChF,IAAI,MAAM,GAAG,KAAK,CAAC;IACnB,IAAI,IAAI,EAAE,CAAC;QACT,MAAM,IAAI,eAAe,IAAI,UAAU,CAAC;IAC1C,CAAC;IACD,MAAM,IAAI,YAAY,cAAc,CAAC,KAAK,CAAC,KAAK,CAAC,MAAM,CAAC;IACxD,IAAI,QAAQ,EAAE,CAAC;QACb,MAAM,IAAI,gBAAgB,QAAQ,OAAO,CAAC;IAC5C,CAAC;IACD,MAAM,IAAI,GAAG,CAAC;IACd,OAAO,MAAM,CAAC;AAChB,CAAC;AACD,mBAAmB","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 { Guid, StopWatch } from \"@itwin/core-bentley\";\nimport {\n  createMainThreadReleaseOnTimePassedHandler,\n  ECSqlQueryDef,\n  ECSqlQueryExecutor,\n  ECSqlQueryReaderOptions,\n  ECSqlQueryRow,\n  trimWhitespace,\n} from \"@itwin/presentation-shared\";\nimport { RowsLimitExceededError } from \"../HierarchyErrors.js\";\nimport { LOGGING_NAMESPACE as BASE_LOGGING_NAMESPACE, LOGGING_NAMESPACE_PERFORMANCE as BASE_LOGGING_NAMESPACE_PERFORMANCE } from \"../internal/Common.js\";\nimport { doLog } from \"../internal/LoggingUtils.js\";\n\n/**\n * An interface for something that knows how to create a limiting ECSQL query reader.\n * @public\n */\nexport interface LimitingECSqlQueryExecutor {\n  /**\n   * Creates a query reader for given query, but makes sure it doesn't return more than the configured\n   * limit of rows.\n   * @throws `RowsLimitExceededError` when the query returns more than configured limit of rows.\n   */\n  createQueryReader(\n    query: ECSqlQueryDef,\n    config?: ECSqlQueryReaderOptions & { limit?: number | \"unbounded\" },\n  ): ReturnType<ECSqlQueryExecutor[\"createQueryReader\"]>;\n}\n\n/**\n * Creates an `LimitingECSqlQueryExecutor` that throws `RowsLimitExceededError` if the query exceeds given amount of rows.\n * @public\n */\nexport function createLimitingECSqlQueryExecutor(baseExecutor: ECSqlQueryExecutor, defaultLimit: number | \"unbounded\"): LimitingECSqlQueryExecutor {\n  return {\n    async *createQueryReader(query: ECSqlQueryDef, config?: ECSqlQueryReaderOptions & { limit?: number | \"unbounded\" }) {\n      const { limit: configLimit, ...restConfig } = config ?? {};\n      const limit = configLimit ?? defaultLimit;\n      const queryLogger = createQueryLogger(query);\n      const releaseMainThread = createMainThreadReleaseOnTimePassedHandler();\n\n      // handle \"unbounded\" case without a buffer\n      const reader = baseExecutor.createQueryReader({ ...query, ecsql: addLimit(query.ecsql, limit) }, restConfig);\n      if (limit === \"unbounded\") {\n        try {\n          for await (const row of reader) {\n            await releaseMainThread();\n            queryLogger.onStep();\n            yield row;\n          }\n        } finally {\n          queryLogger.onComplete();\n        }\n        return;\n      }\n\n      // avoid streaming until we know the number of rows is okay\n      const buffer: ECSqlQueryRow[] = [];\n      try {\n        for await (const row of reader) {\n          await releaseMainThread();\n          queryLogger.onStep();\n          buffer.push(row);\n          if (buffer.length > limit) {\n            throw new RowsLimitExceededError(limit);\n          }\n        }\n      } finally {\n        queryLogger.onComplete();\n      }\n\n      for (const row of buffer) {\n        await releaseMainThread();\n        yield row;\n      }\n    },\n  };\n}\n\nfunction addLimit(ecsql: string, limit: number | \"unbounded\") {\n  if (limit === \"unbounded\") {\n    return ecsql;\n  }\n  return `\n    SELECT *\n    FROM (${ecsql})\n    LIMIT ${limit + 1}\n  `;\n}\n\nconst LOGGING_NAMESPACE = `${BASE_LOGGING_NAMESPACE}.Queries`;\nconst LOGGING_NAMESPACE_PERFORMANCE = `${BASE_LOGGING_NAMESPACE_PERFORMANCE}.Queries`;\nfunction createQueryLogger(query: ECSqlQueryDef, firstStepWarningThreshold = 3000, allRowsWarningThreshold = 5000) {\n  const queryId = Guid.createValue();\n  doLog({\n    category: LOGGING_NAMESPACE,\n    message: /* c8 ignore next */ () => `Executing query [${queryId}]: ${createQueryLogMessage(query)}`,\n  });\n\n  let firstStep = true;\n  let rowsCount = 0;\n  const timer = new StopWatch(undefined, true);\n  return {\n    onStep() {\n      if (firstStep) {\n        /* c8 ignore start */\n        doLog({\n          category: LOGGING_NAMESPACE_PERFORMANCE,\n          severity: timer.current.milliseconds >= firstStepWarningThreshold ? \"warning\" : \"trace\",\n          message: () => `[${queryId}] First step took ${timer.currentSeconds} s.`,\n        });\n        /* c8 ignore end */\n        firstStep = false;\n      }\n      ++rowsCount;\n    },\n    onComplete() {\n      /* c8 ignore start */\n      doLog({\n        category: LOGGING_NAMESPACE_PERFORMANCE,\n        severity: timer.current.milliseconds >= allRowsWarningThreshold ? \"warning\" : \"trace\",\n        message: () => `[${queryId}] Query took ${timer.currentSeconds} s. for ${rowsCount} rows.`,\n      });\n      /* c8 ignore end */\n    },\n  };\n}\n\n/* c8 ignore start */\nfunction createQueryLogMessage(query: ECSqlQueryDef): string {\n  const ctes = query.ctes?.map((cte) => `    ${trimWhitespace(cte)}`).join(\", \\n\");\n  const bindings = query.bindings?.map((b) => JSON.stringify(b.value)).join(\", \");\n  let output = \"{\\n\";\n  if (ctes) {\n    output += `  ctes: [ \\n${ctes} \\n], \\n`;\n  }\n  output += `  ecsql: ${trimWhitespace(query.ecsql)}, \\n`;\n  if (bindings) {\n    output += `  bindings: [${bindings}], \\n`;\n  }\n  output += \"}\";\n  return output;\n}\n/* c8 ignore end */\n"]}