@itwin/unified-selection 0.1.0 → 0.2.0

package/lib/cjs/unified-selection/queries/ECSqlCore.js

@@ -0,0 +1,23 @@
+"use strict";
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formIdBindings = void 0;
+/**
+ * Forms ECSql bindings from given ID's.
+ */
+function formIdBindings(property, ids, bindings) {
+    if (ids.length > 1000) {
+        bindings.push({ type: "idset", value: ids });
+        return `InVirtualSet(?, ${property})`;
+    }
+    if (ids.length === 0) {
+        return `FALSE`;
+    }
+    ids.forEach((id) => bindings.push({ type: "id", value: id }));
+    return `${property} IN (${ids.map(() => "?").join(",")})`;
+}
+exports.formIdBindings = formIdBindings;
+//# sourceMappingURL=ECSqlCore.js.map

package/lib/cjs/unified-selection/queries/ECSqlCore.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ECSqlCore.js","sourceRoot":"","sources":["../../../../src/unified-selection/queries/ECSqlCore.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;AAyEhG;;GAEG;AACH,SAAgB,cAAc,CAAC,QAAgB,EAAE,GAAa,EAAE,QAAwB;IACtF,IAAI,GAAG,CAAC,MAAM,GAAG,IAAI,EAAE;QACrB,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,OAAO,EAAE,KAAK,EAAE,GAAG,EAAE,CAAC,CAAC;QAC7C,OAAO,mBAAmB,QAAQ,GAAG,CAAC;KACvC;IAED,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC,EAAE;QACpB,OAAO,OAAO,CAAC;KAChB;IAED,GAAG,CAAC,OAAO,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,QAAQ,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC;IAC9D,OAAO,GAAG,QAAQ,QAAQ,GAAG,CAAC,GAAG,CAAC,GAAG,EAAE,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,GAAG,CAAC;AAC5D,CAAC;AAZD,wCAYC","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\n/**\n * Types of values that can be bound to an ECSql query.\n */\nexport type ECSqlBindingType = \"boolean\" | \"double\" | \"id\" | \"idset\" | \"int\" | \"long\" | \"string\" | \"point2d\" | \"point3d\";\n\n/**\n * Defines an ECSql binding consisting of a value and its type. Necessary to differentiate between numeric\n * types and `Id64String` vs `string`.\n */\nexport type ECSqlBinding =\n  | {\n      type: \"boolean\";\n      value?: boolean;\n    }\n  | {\n      type: \"double\" | \"int\" | \"long\";\n      value?: number;\n    }\n  | {\n      type: \"id\";\n      value?: string;\n    }\n  | {\n      type: \"idset\";\n      value?: string[];\n    }\n  | {\n      type: \"string\";\n      value?: string;\n    }\n  | {\n      type: \"point2d\";\n      value?: { x: number; y: number };\n    }\n  | {\n      type: \"point3d\";\n      value?: { x: number; y: number; z: number };\n    };\n\n/**\n * Defines requested ECSQL result row format.\n */\nexport type ECSqlQueryRowFormat = \"ECSqlPropertyNames\" | \"Indexes\";\n\n/**\n * Defines options for ECSQL query reader.\n */\nexport interface ECSqlQueryReaderOptions {\n  rowFormat?: ECSqlQueryRowFormat;\n}\n\n/**\n * Represents a single row of an ECSQL query result.\n */\nexport interface ECSqlQueryRow {\n  [propertyName: string]: any;\n  [propertyIndex: number]: any;\n}\n\n/**\n * Represents ECSQL query results reader.\n */\nexport type ECSqlQueryReader = AsyncIterableIterator<ECSqlQueryRow>;\n\n/**\n * An interface for something that knows how to create an ECSQL query reader.\n */\nexport interface IECSqlQueryExecutor {\n  createQueryReader(ecsql: string, bindings?: ECSqlBinding[], config?: ECSqlQueryReaderOptions): ECSqlQueryReader;\n}\n\n/**\n * Forms ECSql bindings from given ID's.\n */\nexport function formIdBindings(property: string, ids: string[], bindings: ECSqlBinding[]): string {\n  if (ids.length > 1000) {\n    bindings.push({ type: \"idset\", value: ids });\n    return `InVirtualSet(?, ${property})`;\n  }\n\n  if (ids.length === 0) {\n    return `FALSE`;\n  }\n\n  ids.forEach((id) => bindings.push({ type: \"id\", value: id }));\n  return `${property} IN (${ids.map(() => \"?\").join(\",\")})`;\n}\n"]}

package/lib/cjs/unified-selection.d.ts

@@ -5,6 +5,9 @@
  * API for working with unified selection.
  */
 export * from "./unified-selection/Selectable";
-export * from "./unified-selection/SelectionStorage";
+export { SelectionStorage, createStorage } from "./unified-selection/SelectionStorage";
+export { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from "./unified-selection/HiliteSetProvider";
+export { createCachingHiliteSetProvider, CachingHiliteSetProvider } from "./unified-selection/CachingHiliteSetProvider";
+export { computeSelection } from "./unified-selection/SelectionScope";
 export { StorageSelectionChangeType, StorageSelectionChangeEventArgs, StorageSelectionChangesListener, SelectionChangeEvent, } from "./unified-selection/SelectionChangeEvent";
 //# sourceMappingURL=unified-selection.d.ts.map

package/lib/cjs/unified-selection.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"unified-selection.d.ts","sourceRoot":"","sources":["../../src/unified-selection.ts"],"names":[],"mappings":"AAKA;;;;;GAKG;AACH,cAAc,gCAAgC,CAAC;AAC/C,cAAc,sCAAsC,CAAC;AACrD,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,+BAA+B,EAC/B,oBAAoB,GACrB,MAAM,0CAA0C,CAAC"}
+{"version":3,"file":"unified-selection.d.ts","sourceRoot":"","sources":["../../src/unified-selection.ts"],"names":[],"mappings":"AAKA;;;;;GAKG;AACH,cAAc,gCAAgC,CAAC;AAC/C,OAAO,EAAE,gBAAgB,EAAE,aAAa,EAAE,MAAM,sCAAsC,CAAC;AACvF,OAAO,EAAE,uBAAuB,EAAE,SAAS,EAAE,iBAAiB,EAAE,MAAM,uCAAuC,CAAC;AAC9G,OAAO,EAAE,8BAA8B,EAAE,wBAAwB,EAAE,MAAM,8CAA8C,CAAC;AACxH,OAAO,EAAE,gBAAgB,EAAE,MAAM,oCAAoC,CAAC;AACtE,OAAO,EACL,0BAA0B,EAC1B,+BAA+B,EAC/B,+BAA+B,EAC/B,oBAAoB,GACrB,MAAM,0CAA0C,CAAC"}

package/lib/cjs/unified-selection.js

@@ -18,6 +18,7 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.computeSelection = exports.createCachingHiliteSetProvider = exports.createHiliteSetProvider = exports.createStorage = void 0;
 /**
  * @module UnifiedSelection
  *
@@ -25,5 +26,12 @@ Object.defineProperty(exports, "__esModule", { value: true });
  * API for working with unified selection.
  */
 __exportStar(require("./unified-selection/Selectable"), exports);
-__exportStar(require("./unified-selection/SelectionStorage"), exports);
+var SelectionStorage_1 = require("./unified-selection/SelectionStorage");
+Object.defineProperty(exports, "createStorage", { enumerable: true, get: function () { return SelectionStorage_1.createStorage; } });
+var HiliteSetProvider_1 = require("./unified-selection/HiliteSetProvider");
+Object.defineProperty(exports, "createHiliteSetProvider", { enumerable: true, get: function () { return HiliteSetProvider_1.createHiliteSetProvider; } });
+var CachingHiliteSetProvider_1 = require("./unified-selection/CachingHiliteSetProvider");
+Object.defineProperty(exports, "createCachingHiliteSetProvider", { enumerable: true, get: function () { return CachingHiliteSetProvider_1.createCachingHiliteSetProvider; } });
+var SelectionScope_1 = require("./unified-selection/SelectionScope");
+Object.defineProperty(exports, "computeSelection", { enumerable: true, get: function () { return SelectionScope_1.computeSelection; } });
 //# sourceMappingURL=unified-selection.js.map

package/lib/cjs/unified-selection.js.map

@@ -1 +1 @@
-{"version":3,"file":"unified-selection.js","sourceRoot":"","sources":["../../src/unified-selection.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;;;;;AAEhG;;;;;GAKG;AACH,iEAA+C;AAC/C,uEAAqD","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\n/**\n * @module UnifiedSelection\n *\n * @docs-group-description UnifiedSelection\n * API for working with unified selection.\n */\nexport * from \"./unified-selection/Selectable\";\nexport * from \"./unified-selection/SelectionStorage\";\nexport {\n  StorageSelectionChangeType,\n  StorageSelectionChangeEventArgs,\n  StorageSelectionChangesListener,\n  SelectionChangeEvent,\n} from \"./unified-selection/SelectionChangeEvent\";\n"]}
+{"version":3,"file":"unified-selection.js","sourceRoot":"","sources":["../../src/unified-selection.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;;;;;;;;;;;;;;;;;AAEhG;;;;;GAKG;AACH,iEAA+C;AAC/C,yEAAuF;AAA5D,iHAAA,aAAa,OAAA;AACxC,2EAA8G;AAArG,4HAAA,uBAAuB,OAAA;AAChC,yFAAwH;AAA/G,0IAAA,8BAA8B,OAAA;AACvC,qEAAsE;AAA7D,kHAAA,gBAAgB,OAAA","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\n/**\n * @module UnifiedSelection\n *\n * @docs-group-description UnifiedSelection\n * API for working with unified selection.\n */\nexport * from \"./unified-selection/Selectable\";\nexport { SelectionStorage, createStorage } from \"./unified-selection/SelectionStorage\";\nexport { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from \"./unified-selection/HiliteSetProvider\";\nexport { createCachingHiliteSetProvider, CachingHiliteSetProvider } from \"./unified-selection/CachingHiliteSetProvider\";\nexport { computeSelection } from \"./unified-selection/SelectionScope\";\nexport {\n  StorageSelectionChangeType,\n  StorageSelectionChangeEventArgs,\n  StorageSelectionChangesListener,\n  SelectionChangeEvent,\n} from \"./unified-selection/SelectionChangeEvent\";\n"]}

package/lib/esm/unified-selection/CachingHiliteSetProvider.d.ts

@@ -0,0 +1,37 @@
+import { HiliteSet } from "./HiliteSetProvider";
+import { IMetadataProvider } from "./queries/ECMetadata";
+import { IECSqlQueryExecutor } from "./queries/ECSqlCore";
+import { SelectionStorage } from "./SelectionStorage";
+/**
+ * Props for creating a `CachingHiliteSetProvider` instance.
+ * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.
+ */
+export interface CachingHiliteSetProviderProps {
+    selectionStorage: SelectionStorage;
+    iModelProvider: (iModelKey: string) => {
+        queryExecutor: IECSqlQueryExecutor;
+        metadataProvider: IMetadataProvider;
+    };
+}
+/**
+ * Defines return value of `createCachingHiliteSetProvider`.
+ *
+ * @beta Used in public API as a return value. Not expected to be created / extended by package
+ * consumers, may be supplemented with required attributes any time.
+ */
+export interface CachingHiliteSetProvider {
+    /** Get the current hilite set iterator for the specified imodel */
+    getHiliteSet(props: {
+        /** iModel to get hilite set for */
+        iModelKey: string;
+    }): AsyncIterableIterator<HiliteSet>;
+    /** Disposes the cache. */
+    dispose(): void;
+}
+/**
+ * Creates a hilite set provider that caches hilite set for current selection for given iModel so any subsequent
+ * hilite set requests for the same iModel don't cost until selection in given selection storage changes.
+ * @beta
+ */
+export declare function createCachingHiliteSetProvider(props: CachingHiliteSetProviderProps): CachingHiliteSetProvider;
+//# sourceMappingURL=CachingHiliteSetProvider.d.ts.map

package/lib/esm/unified-selection/CachingHiliteSetProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"CachingHiliteSetProvider.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/CachingHiliteSetProvider.ts"],"names":[],"mappings":"AAWA,OAAO,EAA2B,SAAS,EAAqB,MAAM,qBAAqB,CAAC;AAC5F,OAAO,EAAE,iBAAiB,EAAE,MAAM,sBAAsB,CAAC;AACzD,OAAO,EAAE,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AAE1D,OAAO,EAAuC,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AAE3F;;;GAGG;AACH,MAAM,WAAW,6BAA6B;IAC5C,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,cAAc,EAAE,CAAC,SAAS,EAAE,MAAM,KAAK;QAAE,aAAa,EAAE,mBAAmB,CAAC;QAAC,gBAAgB,EAAE,iBAAiB,CAAA;KAAE,CAAC;CACpH;AAED;;;;;GAKG;AACH,MAAM,WAAW,wBAAwB;IACvC,mEAAmE;IACnE,YAAY,CAAC,KAAK,EAAE;QAClB,mCAAmC;QACnC,SAAS,EAAE,MAAM,CAAC;KACnB,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;IAErC,0BAA0B;IAC1B,OAAO,IAAI,IAAI,CAAC;CACjB;AAED;;;;GAIG;AACH,wBAAgB,8BAA8B,CAAC,KAAK,EAAE,6BAA6B,GAAG,wBAAwB,CAE7G"}

package/lib/esm/unified-selection/CachingHiliteSetProvider.js

@@ -0,0 +1,61 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module UnifiedSelection
+ */
+import { from, shareReplay } from "rxjs";
+import { eachValueFrom } from "rxjs-for-await";
+import { createHiliteSetProvider } from "./HiliteSetProvider";
+import { IMODEL_CLOSE_SELECTION_CLEAR_SOURCE } from "./SelectionStorage";
+/**
+ * Creates a hilite set provider that caches hilite set for current selection for given iModel so any subsequent
+ * hilite set requests for the same iModel don't cost until selection in given selection storage changes.
+ * @beta
+ */
+export function createCachingHiliteSetProvider(props) {
+    return new CachingHiliteSetProviderImpl(props);
+}
+class CachingHiliteSetProviderImpl {
+    _selectionStorage;
+    _hiliteSetProviders = new Map();
+    _cache = new Map();
+    _removeListener;
+    _iModelProvider;
+    constructor(props) {
+        this._selectionStorage = props.selectionStorage;
+        this._iModelProvider = props.iModelProvider;
+        this._removeListener = this._selectionStorage.selectionChangeEvent.addListener((args) => {
+            this._cache.delete(args.iModelKey);
+            if (args.changeType === "clear" && args.source === IMODEL_CLOSE_SELECTION_CLEAR_SOURCE) {
+                this._hiliteSetProviders.delete(args.iModelKey);
+            }
+        });
+    }
+    getHiliteSet({ iModelKey }) {
+        const { queryExecutor, metadataProvider } = this._iModelProvider(iModelKey);
+        const provider = this.getHiliteSetProvider(iModelKey, queryExecutor, metadataProvider);
+        let hiliteSet = this._cache.get(iModelKey);
+        if (!hiliteSet) {
+            const selectables = this._selectionStorage.getSelection({ iModelKey });
+            hiliteSet = from(provider.getHiliteSet({ selectables })).pipe(shareReplay({ refCount: true }));
+            this._cache.set(iModelKey, hiliteSet);
+        }
+        return eachValueFrom(hiliteSet);
+    }
+    dispose() {
+        this._removeListener();
+        this._hiliteSetProviders = new Map();
+        this._cache = new Map();
+    }
+    getHiliteSetProvider(iModelKey, queryExecutor, metadataProvider) {
+        let provider = this._hiliteSetProviders.get(iModelKey);
+        if (!provider) {
+            provider = createHiliteSetProvider({ queryExecutor, metadataProvider });
+            this._hiliteSetProviders.set(iModelKey, provider);
+        }
+        return provider;
+    }
+}
+//# sourceMappingURL=CachingHiliteSetProvider.js.map

package/lib/esm/unified-selection/CachingHiliteSetProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"CachingHiliteSetProvider.js","sourceRoot":"","sources":["../../../src/unified-selection/CachingHiliteSetProvider.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG;;GAEG;AAEH,OAAO,EAAE,IAAI,EAAc,WAAW,EAAE,MAAM,MAAM,CAAC;AACrD,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EAAE,uBAAuB,EAAgC,MAAM,qBAAqB,CAAC;AAI5F,OAAO,EAAE,mCAAmC,EAAoB,MAAM,oBAAoB,CAAC;AA4B3F;;;;GAIG;AACH,MAAM,UAAU,8BAA8B,CAAC,KAAoC;IACjF,OAAO,IAAI,4BAA4B,CAAC,KAAK,CAAC,CAAC;AACjD,CAAC;AAED,MAAM,4BAA4B;IACxB,iBAAiB,CAAmB;IACpC,mBAAmB,GAAG,IAAI,GAAG,EAA6B,CAAC;IAC3D,MAAM,GAAG,IAAI,GAAG,EAAiC,CAAC;IAClD,eAAe,CAAa;IAC5B,eAAe,CAAqG;IAE5H,YAAY,KAAoC;QAC9C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,eAAe,GAAG,KAAK,CAAC,cAAc,CAAC;QAC5C,IAAI,CAAC,eAAe,GAAG,IAAI,CAAC,iBAAiB,CAAC,oBAAoB,CAAC,WAAW,CAAC,CAAC,IAAqC,EAAE,EAAE;YACvH,IAAI,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;YACnC,IAAI,IAAI,CAAC,UAAU,KAAK,OAAO,IAAI,IAAI,CAAC,MAAM,KAAK,mCAAmC,EAAE;gBACtF,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;aACjD;QACH,CAAC,CAAC,CAAC;IACL,CAAC;IAEM,YAAY,CAAC,EAAE,SAAS,EAAyB;QACtD,MAAM,EAAE,aAAa,EAAE,gBAAgB,EAAE,GAAG,IAAI,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAC5E,MAAM,QAAQ,GAAG,IAAI,CAAC,oBAAoB,CAAC,SAAS,EAAE,aAAa,EAAE,gBAAgB,CAAC,CAAC;QACvF,IAAI,SAAS,GAAG,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QAE3C,IAAI,CAAC,SAAS,EAAE;YACd,MAAM,WAAW,GAAG,IAAI,CAAC,iBAAiB,CAAC,YAAY,CAAC,EAAE,SAAS,EAAE,CAAC,CAAC;YACvE,SAAS,GAAG,IAAI,CAAC,QAAQ,CAAC,YAAY,CAAC,EAAE,WAAW,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,EAAE,QAAQ,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC;YAC/F,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,SAAS,EAAE,SAAS,CAAC,CAAC;SACvC;QAED,OAAO,aAAa,CAAC,SAAS,CAAC,CAAC;IAClC,CAAC;IAEM,OAAO;QACZ,IAAI,CAAC,eAAe,EAAE,CAAC;QACvB,IAAI,CAAC,mBAAmB,GAAG,IAAI,GAAG,EAAE,CAAC;QACrC,IAAI,CAAC,MAAM,GAAG,IAAI,GAAG,EAAE,CAAC;IAC1B,CAAC;IAEO,oBAAoB,CAAC,SAAiB,EAAE,aAAkC,EAAE,gBAAmC;QACrH,IAAI,QAAQ,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACvD,IAAI,CAAC,QAAQ,EAAE;YACb,QAAQ,GAAG,uBAAuB,CAAC,EAAE,aAAa,EAAE,gBAAgB,EAAE,CAAC,CAAC;YACxE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;SACnD;QACD,OAAO,QAAQ,CAAC;IAClB,CAAC;CACF","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\n/** @packageDocumentation\n * @module UnifiedSelection\n */\n\nimport { from, Observable, shareReplay } from \"rxjs\";\nimport { eachValueFrom } from \"rxjs-for-await\";\nimport { createHiliteSetProvider, HiliteSet, HiliteSetProvider } from \"./HiliteSetProvider\";\nimport { IMetadataProvider } from \"./queries/ECMetadata\";\nimport { IECSqlQueryExecutor } from \"./queries/ECSqlCore\";\nimport { StorageSelectionChangeEventArgs } from \"./SelectionChangeEvent\";\nimport { IMODEL_CLOSE_SELECTION_CLEAR_SOURCE, SelectionStorage } from \"./SelectionStorage\";\n\n/**\n * Props for creating a `CachingHiliteSetProvider` instance.\n * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.\n */\nexport interface CachingHiliteSetProviderProps {\n  selectionStorage: SelectionStorage;\n  iModelProvider: (iModelKey: string) => { queryExecutor: IECSqlQueryExecutor; metadataProvider: IMetadataProvider };\n}\n\n/**\n * Defines return value of `createCachingHiliteSetProvider`.\n *\n * @beta Used in public API as a return value. Not expected to be created / extended by package\n * consumers, may be supplemented with required attributes any time.\n */\nexport interface CachingHiliteSetProvider {\n  /** Get the current hilite set iterator for the specified imodel */\n  getHiliteSet(props: {\n    /** iModel to get hilite set for */\n    iModelKey: string;\n  }): AsyncIterableIterator<HiliteSet>;\n\n  /** Disposes the cache. */\n  dispose(): void;\n}\n\n/**\n * Creates a hilite set provider that caches hilite set for current selection for given iModel so any subsequent\n * hilite set requests for the same iModel don't cost until selection in given selection storage changes.\n * @beta\n */\nexport function createCachingHiliteSetProvider(props: CachingHiliteSetProviderProps): CachingHiliteSetProvider {\n  return new CachingHiliteSetProviderImpl(props);\n}\n\nclass CachingHiliteSetProviderImpl implements CachingHiliteSetProvider {\n  private _selectionStorage: SelectionStorage;\n  private _hiliteSetProviders = new Map<string, HiliteSetProvider>();\n  private _cache = new Map<string, Observable<HiliteSet>>();\n  private _removeListener: () => void;\n  private _iModelProvider: (iModelKey: string) => { queryExecutor: IECSqlQueryExecutor; metadataProvider: IMetadataProvider };\n\n  constructor(props: CachingHiliteSetProviderProps) {\n    this._selectionStorage = props.selectionStorage;\n    this._iModelProvider = props.iModelProvider;\n    this._removeListener = this._selectionStorage.selectionChangeEvent.addListener((args: StorageSelectionChangeEventArgs) => {\n      this._cache.delete(args.iModelKey);\n      if (args.changeType === \"clear\" && args.source === IMODEL_CLOSE_SELECTION_CLEAR_SOURCE) {\n        this._hiliteSetProviders.delete(args.iModelKey);\n      }\n    });\n  }\n\n  public getHiliteSet({ iModelKey }: { iModelKey: string }): AsyncIterableIterator<HiliteSet> {\n    const { queryExecutor, metadataProvider } = this._iModelProvider(iModelKey);\n    const provider = this.getHiliteSetProvider(iModelKey, queryExecutor, metadataProvider);\n    let hiliteSet = this._cache.get(iModelKey);\n\n    if (!hiliteSet) {\n      const selectables = this._selectionStorage.getSelection({ iModelKey });\n      hiliteSet = from(provider.getHiliteSet({ selectables })).pipe(shareReplay({ refCount: true }));\n      this._cache.set(iModelKey, hiliteSet);\n    }\n\n    return eachValueFrom(hiliteSet);\n  }\n\n  public dispose(): void {\n    this._removeListener();\n    this._hiliteSetProviders = new Map();\n    this._cache = new Map();\n  }\n\n  private getHiliteSetProvider(iModelKey: string, queryExecutor: IECSqlQueryExecutor, metadataProvider: IMetadataProvider) {\n    let provider = this._hiliteSetProviders.get(iModelKey);\n    if (!provider) {\n      provider = createHiliteSetProvider({ queryExecutor, metadataProvider });\n      this._hiliteSetProviders.set(iModelKey, provider);\n    }\n    return provider;\n  }\n}\n"]}

package/lib/esm/unified-selection/HiliteSetProvider.d.ts

@@ -0,0 +1,39 @@
+import { IMetadataProvider } from "./queries/ECMetadata";
+import { IECSqlQueryExecutor } from "./queries/ECSqlCore";
+import { Selectables } from "./Selectable";
+/**
+ * A set of model, subcategory and element ids that can be used for specifying hilite.
+ * @see https://www.itwinjs.org/reference/core-frontend/selectionset/hiliteset/
+ * @beta
+ */
+export interface HiliteSet {
+    models: string[];
+    subCategories: string[];
+    elements: string[];
+}
+/**
+ * Props for creating a `HiliteSetProvider` instance.
+ * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.
+ */
+export interface HiliteSetProviderProps {
+    queryExecutor: IECSqlQueryExecutor;
+    metadataProvider: IMetadataProvider;
+}
+/**
+ * Defines return value of `createHiliteSetProvider`.
+ *
+ * @beta Used in public API as a return value. Not expected to be created / extended by package
+ * consumers, may be supplemented with required attributes any time.
+ */
+export interface HiliteSetProvider {
+    /** Get the current hilite set iterator for the specified imodel */
+    getHiliteSet(props: {
+        selectables: Selectables;
+    }): AsyncIterableIterator<HiliteSet>;
+}
+/**
+ * Creates a hilite set provider that returns a `HiliteSet` for given selectables.
+ * @beta
+ */
+export declare function createHiliteSetProvider(props: HiliteSetProviderProps): HiliteSetProvider;
+//# sourceMappingURL=HiliteSetProvider.d.ts.map

package/lib/esm/unified-selection/HiliteSetProvider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"HiliteSetProvider.d.ts","sourceRoot":"","sources":["../../../src/unified-selection/HiliteSetProvider.ts"],"names":[],"mappings":"AAWA,OAAO,EAAW,iBAAiB,EAAsB,MAAM,sBAAsB,CAAC;AACtF,OAAO,EAAgC,mBAAmB,EAAE,MAAM,qBAAqB,CAAC;AACxF,OAAO,EAAyB,WAAW,EAAE,MAAM,cAAc,CAAC;AAIlE;;;;GAIG;AACH,MAAM,WAAW,SAAS;IACxB,MAAM,EAAE,MAAM,EAAE,CAAC;IACjB,aAAa,EAAE,MAAM,EAAE,CAAC;IACxB,QAAQ,EAAE,MAAM,EAAE,CAAC;CACpB;AAED;;;GAGG;AACH,MAAM,WAAW,sBAAsB;IACrC,aAAa,EAAE,mBAAmB,CAAC;IACnC,gBAAgB,EAAE,iBAAiB,CAAC;CACrC;AAED;;;;;GAKG;AACH,MAAM,WAAW,iBAAiB;IAChC,mEAAmE;IACnE,YAAY,CAAC,KAAK,EAAE;QAAE,WAAW,EAAE,WAAW,CAAA;KAAE,GAAG,qBAAqB,CAAC,SAAS,CAAC,CAAC;CACrF;AAED;;;GAGG;AACH,wBAAgB,uBAAuB,CAAC,KAAK,EAAE,sBAAsB,GAAG,iBAAiB,CAExF"}

package/lib/esm/unified-selection/HiliteSetProvider.js

@@ -0,0 +1,294 @@
+/*---------------------------------------------------------------------------------------------
+ * Copyright (c) Bentley Systems, Incorporated. All rights reserved.
+ * See LICENSE.md in the project root for license terms and full copyright notice.
+ *--------------------------------------------------------------------------------------------*/
+/** @packageDocumentation
+ * @module UnifiedSelection
+ */
+import { EMPTY, filter, forkJoin, from, map, merge, mergeMap, scan, shareReplay, Subject, toArray } from "rxjs";
+import { eachValueFrom } from "rxjs-for-await";
+import { parseFullClassName } from "./queries/ECMetadata";
+import { formIdBindings } from "./queries/ECSqlCore";
+const HILITE_SET_EMIT_FREQUENCY = 20;
+/**
+ * Creates a hilite set provider that returns a `HiliteSet` for given selectables.
+ * @beta
+ */
+export function createHiliteSetProvider(props) {
+    return new HiliteSetProviderImpl(props);
+}
+class HiliteSetProviderImpl {
+    _queryExecutor;
+    _metadataProvider;
+    // Map between a class name and its type
+    _classRelationCache;
+    constructor(props) {
+        this._queryExecutor = props.queryExecutor;
+        this._metadataProvider = props.metadataProvider;
+        this._classRelationCache = new Map();
+    }
+    /**
+     * Get hilite set iterator for supplied `Selectables`.
+     */
+    getHiliteSet(props) {
+        const obs = this.getHiliteSetObservable(props);
+        return eachValueFrom(obs);
+    }
+    /**
+     * Returns a "hot" observable of hilite sets.
+     */
+    getHiliteSetObservable({ selectables }) {
+        const instancesByType = this.getInstancesByType(selectables);
+        const observables = {
+            models: this.getHilitedModels(instancesByType),
+            subCategories: this.getHilitedSubCategories(instancesByType),
+            elements: this.getHilitedElements(instancesByType),
+        };
+        let hiliteSet = { models: [], subCategories: [], elements: [] };
+        let lastEmitTime = performance.now();
+        const subject = new Subject();
+        const subscriptions = ["models", "subCategories", "elements"].map((key) => observables[key].subscribe({
+            next(val) {
+                hiliteSet[key].push(val);
+                if (performance.now() - lastEmitTime < HILITE_SET_EMIT_FREQUENCY) {
+                    return;
+                }
+                subject.next(hiliteSet);
+                hiliteSet = { models: [], subCategories: [], elements: [] };
+                lastEmitTime = performance.now();
+            },
+            complete() {
+                observables[key] = undefined;
+                if (observables.models || observables.subCategories || observables.elements) {
+                    return;
+                }
+                // Emit last batch before completing the observable.
+                if (hiliteSet.elements.length || hiliteSet.models.length || hiliteSet.subCategories.length) {
+                    subject.next(hiliteSet);
+                }
+                subject.complete();
+            },
+            error(err) {
+                subscriptions.forEach((x) => x.unsubscribe());
+                subscriptions.length = 0;
+                subject.error(err);
+            },
+        }));
+        return subject;
+    }
+    async getType(key) {
+        const cachedType = this._classRelationCache.get(key.className.replace(".", ":"));
+        if (cachedType) {
+            return cachedType;
+        }
+        const promise = this.getTypeImpl(key).then((res) => {
+            // Update the cache with the result of the promise.
+            this._classRelationCache.set(key.className, res);
+            return res;
+        });
+        // Add the promise to cache to prevent `getTypeImpl` being called multiple times.
+        this._classRelationCache.set(key.className, promise);
+        return promise;
+    }
+    async getTypeImpl(key) {
+        const keyClass = await this.getClass(key.className);
+        return ((keyClass &&
+            ((await this.checkType(keyClass, "BisCore", "Subject", "subject")) ??
+                (await this.checkType(keyClass, "BisCore", "Model", "model")) ??
+                (await this.checkType(keyClass, "BisCore", "Category", "category")) ??
+                (await this.checkType(keyClass, "BisCore", "SubCategory", "subCategory")) ??
+                (await this.checkType(keyClass, "Functional", "FunctionalElement", "functionalElement")) ??
+                (await this.checkType(keyClass, "BisCore", "GroupInformationElement", "groupInformationElement")) ??
+                (await this.checkType(keyClass, "BisCore", "GeometricElement", "geometricElement")) ??
+                (await this.checkType(keyClass, "BisCore", "Element", "element")))) ??
+            "unknown");
+    }
+    async checkType(keyClass, schemaName, className, type) {
+        return (await keyClass.is(className, schemaName)) ? type : undefined;
+    }
+    getInstancesByType(selectables) {
+        const keyTypeObs = merge(from(selectables.custom.values()).pipe(mergeMap((selectable) => selectable.loadInstanceKeys())), from(selectables.instanceKeys).pipe(mergeMap(([className, idSet]) => from(idSet).pipe(map((id) => ({ className, id })))))).pipe(
+        // Get types for each instance key
+        mergeMap((instanceKey) => from(this.getType(instanceKey)).pipe(map((instanceIdType) => ({ instanceId: instanceKey.id, instanceIdType })))), 
+        // Cache the results
+        shareReplay());
+        return Object.fromEntries(INSTANCE_TYPES.map((type) => [
+            type,
+            keyTypeObs.pipe(filter(({ instanceIdType }) => instanceIdType === type), map(({ instanceId }) => instanceId), unique()),
+        ]));
+    }
+    getHilitedModels(instancesByType) {
+        return forkJoin({
+            modelKeys: instancesByType.model.pipe(toArray()),
+            subjectKeys: instancesByType.subject.pipe(toArray()),
+        }).pipe(mergeMap(({ modelKeys, subjectKeys }) => {
+            if (!modelKeys.length && !subjectKeys.length) {
+                return EMPTY;
+            }
+            const bindings = [];
+            const query = `WITH RECURSIVE
+                    ChildSubjects (ECInstanceId, JsonProperties) AS (
+                      SELECT ECInstanceId, JsonProperties FROM BisCore.Subject WHERE ${formIdBindings("ECInstanceId", subjectKeys, bindings)}
+                      UNION ALL
+                      SELECT r.ECInstanceId, r.JsonProperties FROM ChildSubjects s
+                        JOIN BisCore.Subject r ON r.Parent.Id = s.ECInstanceId
+                    ),
+                    Models (ECInstanceId) AS (
+                      SELECT s.ECInstanceId as \`ECInstanceId\` FROM BisCore.Model s
+                      WHERE ${formIdBindings("ECInstanceId", modelKeys, bindings)}
+                    )
+                    SELECT r.ECInstanceId as \`ECInstanceId\` FROM ChildSubjects s
+                      JOIN BisCore.PhysicalPartition r
+                        ON r.Parent.Id = s.ECInstanceId
+                        OR json_extract(s.JsonProperties,'$.Subject.Model.TargetPartition') = printf('0x%x', r.ECInstanceId)
+                    UNION
+                    SELECT ECInstanceId FROM Models;`;
+            return from(this._queryExecutor.createQueryReader(query, bindings));
+        }), map((row) => row.ECInstanceId));
+    }
+    getHilitedSubCategories(instancesByType) {
+        return forkJoin({
+            subCategoryKeys: instancesByType.subCategory.pipe(toArray()),
+            categoryKeys: instancesByType.category.pipe(toArray()),
+        }).pipe(mergeMap(({ subCategoryKeys, categoryKeys }) => {
+            if (!subCategoryKeys.length && !categoryKeys.length) {
+                return EMPTY;
+            }
+            const bindings = [];
+            const query = `WITH
+                    CategorySubCategories (ECInstanceId) AS (
+                      SELECT r.ECInstanceId as \`ECInstanceId\` FROM BisCore.Category s
+                        JOIN BisCore.SubCategory r ON r.Parent.Id = s.ECInstanceId
+                      WHERE ${formIdBindings("s.ECInstanceId", categoryKeys, bindings)}
+                    ),
+                    SubCategories (ECInstanceId) AS (
+                      SELECT s.ECInstanceId as \`ECInstanceId\` FROM BisCore.SubCategory s
+                      WHERE ${formIdBindings("s.ECInstanceId", subCategoryKeys, bindings)}
+                    )
+                   SELECT ECInstanceId FROM CategorySubCategories
+                   UNION
+                   SELECT ECInstanceId FROM SubCategories;`;
+            return from(this._queryExecutor.createQueryReader(query, bindings));
+        }), map((row) => row.ECInstanceId));
+    }
+    getHilitedElements(instancesByType) {
+        return forkJoin({
+            groupInformationElementKeys: instancesByType.groupInformationElement.pipe(toArray()),
+            geometricElementKeys: instancesByType.geometricElement.pipe(toArray()),
+            functionalElements: instancesByType.functionalElement.pipe(toArray()),
+            elementKeys: instancesByType.element.pipe(toArray()),
+        }).pipe(mergeMap(({ groupInformationElementKeys, geometricElementKeys, functionalElements, elementKeys }) => {
+            const hasFunctionalElements = !!functionalElements.length;
+            if (!groupInformationElementKeys.length && !geometricElementKeys.length && !elementKeys.length && !hasFunctionalElements) {
+                return EMPTY;
+            }
+            const bindings = [];
+            const query = `WITH RECURSIVE
+                    ${hasFunctionalElements ? this.getHilitedFunctionalElementsQuery(functionalElements, bindings) : ""}
+                    GroupMembers (ECInstanceId, ECClassId) AS (
+                      SELECT TargetECInstanceId, TargetECClassId FROM BisCore.ElementGroupsMembers
+                      WHERE ${formIdBindings("SourceECInstanceId", groupInformationElementKeys, bindings)}
+                    ),
+                    GroupGeometricElements (ECInstanceId, ECClassId) AS (
+                      SELECT ECInstanceId, ECClassId FROM GroupMembers
+                      UNION ALL
+                      SELECT r.ECInstanceId, r.ECClassId FROM GroupGeometricElements s
+                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+                    ),
+                    ElementGeometricElements (ECInstanceId, ECClassId) AS (
+                      SELECT ECInstanceId, ECClassId FROM BisCore.Element WHERE ${formIdBindings("ECInstanceId", elementKeys, bindings)}
+                      UNION ALL
+                      SELECT r.ECInstanceId, r.ECClassId FROM ElementGeometricElements s
+                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+                    ),
+                    GeometricElementGeometricElements (ECInstanceId, ECClassId) AS (
+                      SELECT ECInstanceId, ECClassId FROM BisCore.GeometricElement WHERE ${formIdBindings("ECInstanceId", geometricElementKeys, bindings)}
+                      UNION ALL
+                      SELECT r.ECInstanceId, r.ECClassId FROM GeometricElementGeometricElements s
+                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+                    )
+                   ${hasFunctionalElements
+                ? `
+                   SELECT ECInstanceId FROM FunctionalElementChildGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)
+                   UNION
+                   `
+                : ``}
+                   SELECT ECInstanceId FROM GeometricElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)
+                   UNION
+                   SELECT ECInstanceId FROM GroupGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)
+                   UNION
+                   SELECT ECInstanceId FROM ElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement);`;
+            return from(this._queryExecutor.createQueryReader(query, bindings));
+        }), map((row) => row.ECInstanceId));
+    }
+    getHilitedFunctionalElementsQuery(functionalElements, bindings) {
+        return `ChildFunctionalElements (ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId FROM Functional.FunctionalElement WHERE ${formIdBindings("ECInstanceId", functionalElements, bindings)}
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId FROM ChildFunctionalElements s
+                JOIN Functional.FunctionalElement r ON r.Parent.Id = s.ECInstanceId
+            ),
+            PhysicalElements (ECInstanceId, ECClassId) AS (
+              SELECT r.SourceECInstanceId, r.SourceECClassId FROM ChildFunctionalElements s
+                JOIN Functional.PhysicalElementFulfillsFunction r ON r.TargetECInstanceId = s.ECInstanceId
+            ),
+            DrawingGraphicElements (ECInstanceId, ECClassId) AS (
+              SELECT r.SourceECInstanceId, r.SourceECClassId FROM ChildFunctionalElements s
+                JOIN Functional.DrawingGraphicRepresentsFunctionalElement r ON r.TargetECInstanceId = s.ECInstanceId
+            ),
+            PhysicalElementGeometricElements (ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId FROM PhysicalElements
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId FROM PhysicalElementGeometricElements s
+                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+            ),
+            DrawingGraphicElementGeometricElements (ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId FROM DrawingGraphicElements
+              UNION ALL
+              SELECT r.ECInstanceId, r.ECClassId FROM DrawingGraphicElementGeometricElements s
+                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId
+            ),
+            FunctionalElementChildGeometricElements (ECInstanceId, ECClassId) AS (
+              SELECT ECInstanceId, ECClassId FROM PhysicalElementGeometricElements
+              UNION
+              SELECT ECInstanceId, ECClassId FROM DrawingGraphicElementGeometricElements
+            ),`;
+    }
+    async getClass(fullClassName) {
+        const { schemaName, className } = parseFullClassName(fullClassName);
+        const schema = await this._metadataProvider.getSchema(schemaName);
+        if (!schema) {
+            return undefined;
+        }
+        const schemaClass = await schema.getClass(className);
+        if (!schemaClass) {
+            return undefined;
+        }
+        return schemaClass;
+    }
+}
+const INSTANCE_TYPES = [
+    "subject",
+    "model",
+    "category",
+    "subCategory",
+    "functionalElement",
+    "groupInformationElement",
+    "geometricElement",
+    "element",
+    "unknown",
+];
+function unique() {
+    return function (obs) {
+        return obs.pipe(scan((acc, val) => {
+            if (acc.set.has(val)) {
+                delete acc.val;
+                return acc;
+            }
+            acc.set.add(val);
+            acc.val = val;
+            return acc;
+        }, { set: new Set() }), map(({ val }) => val), filter((x) => !!x));
+    };
+}
+//# sourceMappingURL=HiliteSetProvider.js.map

package/lib/esm/unified-selection/HiliteSetProvider.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"HiliteSetProvider.js","sourceRoot":"","sources":["../../../src/unified-selection/HiliteSetProvider.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAEhG;;GAEG;AAEH,OAAO,EAAE,KAAK,EAAE,MAAM,EAAE,QAAQ,EAAE,IAAI,EAAE,GAAG,EAAE,KAAK,EAAE,QAAQ,EAAc,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,EAAE,MAAM,MAAM,CAAC;AAC5H,OAAO,EAAE,aAAa,EAAE,MAAM,gBAAgB,CAAC;AAC/C,OAAO,EAA8B,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AACtF,OAAO,EAAgB,cAAc,EAAuB,MAAM,qBAAqB,CAAC;AAGxF,MAAM,yBAAyB,GAAG,EAAE,CAAC;AAiCrC;;;GAGG;AACH,MAAM,UAAU,uBAAuB,CAAC,KAA6B;IACnE,OAAO,IAAI,qBAAqB,CAAC,KAAK,CAAC,CAAC;AAC1C,CAAC;AAED,MAAM,qBAAqB;IACjB,cAAc,CAAsB;IACpC,iBAAiB,CAAoB;IAC7C,wCAAwC;IAChC,mBAAmB,CAAwD;IAEnF,YAAY,KAA6B;QACvC,IAAI,CAAC,cAAc,GAAG,KAAK,CAAC,aAAa,CAAC;QAC1C,IAAI,CAAC,iBAAiB,GAAG,KAAK,CAAC,gBAAgB,CAAC;QAChD,IAAI,CAAC,mBAAmB,GAAG,IAAI,GAAG,EAAE,CAAC;IACvC,CAAC;IAED;;OAEG;IACI,YAAY,CAAC,KAAmC;QACrD,MAAM,GAAG,GAAG,IAAI,CAAC,sBAAsB,CAAC,KAAK,CAAC,CAAC;QAC/C,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC;IAC5B,CAAC;IAED;;OAEG;IACK,sBAAsB,CAAC,EAAE,WAAW,EAAgC;QAC1E,MAAM,eAAe,GAAG,IAAI,CAAC,kBAAkB,CAAC,WAAW,CAAC,CAAC;QAC7D,MAAM,WAAW,GAAsD;YACrE,MAAM,EAAE,IAAI,CAAC,gBAAgB,CAAC,eAAe,CAAC;YAC9C,aAAa,EAAE,IAAI,CAAC,uBAAuB,CAAC,eAAe,CAAC;YAC5D,QAAQ,EAAE,IAAI,CAAC,kBAAkB,CAAC,eAAe,CAAC;SACnD,CAAC;QAEF,IAAI,SAAS,GAAc,EAAE,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;QAC3E,IAAI,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACrC,MAAM,OAAO,GAAG,IAAI,OAAO,EAAa,CAAC;QACzC,MAAM,aAAa,GAAI,CAAC,QAAQ,EAAE,eAAe,EAAE,UAAU,CAAW,CAAC,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CACnF,WAAW,CAAC,GAAG,CAAE,CAAC,SAAS,CAAC;YAC1B,IAAI,CAAC,GAAG;gBACN,SAAS,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;gBACzB,IAAI,WAAW,CAAC,GAAG,EAAE,GAAG,YAAY,GAAG,yBAAyB,EAAE;oBAChE,OAAO;iBACR;gBACD,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;gBACxB,SAAS,GAAG,EAAE,MAAM,EAAE,EAAE,EAAE,aAAa,EAAE,EAAE,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;gBAC5D,YAAY,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;YACnC,CAAC;YACD,QAAQ;gBACN,WAAW,CAAC,GAAG,CAAC,GAAG,SAAS,CAAC;gBAC7B,IAAI,WAAW,CAAC,MAAM,IAAI,WAAW,CAAC,aAAa,IAAI,WAAW,CAAC,QAAQ,EAAE;oBAC3E,OAAO;iBACR;gBACD,oDAAoD;gBACpD,IAAI,SAAS,CAAC,QAAQ,CAAC,MAAM,IAAI,SAAS,CAAC,MAAM,CAAC,MAAM,IAAI,SAAS,CAAC,aAAa,CAAC,MAAM,EAAE;oBAC1F,OAAO,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;iBACzB;gBACD,OAAO,CAAC,QAAQ,EAAE,CAAC;YACrB,CAAC;YACD,KAAK,CAAC,GAAG;gBACP,aAAa,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;gBAC9C,aAAa,CAAC,MAAM,GAAG,CAAC,CAAC;gBACzB,OAAO,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACrB,CAAC;SACF,CAAC,CACH,CAAC;QAEF,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,OAAO,CAAC,GAA0B;QAC9C,MAAM,UAAU,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;QACjF,IAAI,UAAU,EAAE;YACd,OAAO,UAAU,CAAC;SACnB;QAED,MAAM,OAAO,GAAG,IAAI,CAAC,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,CAAC,GAAG,EAAE,EAAE;YACjD,mDAAmD;YACnD,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;YACjD,OAAO,GAAG,CAAC;QACb,CAAC,CAAC,CAAC;QAEH,iFAAiF;QACjF,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,GAAG,CAAC,SAAS,EAAE,OAAO,CAAC,CAAC;QACrD,OAAO,OAAO,CAAC;IACjB,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,GAA0B;QAClD,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,GAAG,CAAC,SAAS,CAAC,CAAC;QACpD,OAAO,CACL,CAAC,QAAQ;YACP,CAAC,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;gBAChE,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,OAAO,EAAE,OAAO,CAAC,CAAC;gBAC7D,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,EAAE,UAAU,CAAC,CAAC;gBACnE,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,aAAa,EAAE,aAAa,CAAC,CAAC;gBACzE,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,YAAY,EAAE,mBAAmB,EAAE,mBAAmB,CAAC,CAAC;gBACxF,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,yBAAyB,EAAE,yBAAyB,CAAC,CAAC;gBACjG,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,kBAAkB,EAAE,kBAAkB,CAAC,CAAC;gBACnF,CAAC,MAAM,IAAI,CAAC,SAAS,CAAC,QAAQ,EAAE,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC,CAAC,CAAC;YACvE,SAAS,CACV,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,SAAS,CAAC,QAAiB,EAAE,UAAkB,EAAE,SAAiB,EAAE,IAAoB;QACpG,OAAO,CAAC,MAAM,QAAQ,CAAC,EAAE,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,SAAS,CAAC;IACvE,CAAC;IAEO,kBAAkB,CAAC,WAAwB;QACjD,MAAM,UAAU,GAAG,KAAK,CACtB,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,MAAM,EAAE,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,UAAU,EAAE,EAAE,CAAC,UAAU,CAAC,gBAAgB,EAAE,CAAC,CAAC,EAC/F,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,EAAE,SAAS,EAAE,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC1H,CAAC,IAAI;QACJ,kCAAkC;QAClC,QAAQ,CAAC,CAAC,WAAW,EAAE,EAAE,CAAC,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,EAAE,UAAU,EAAE,WAAW,CAAC,EAAE,EAAE,cAAc,EAAE,CAAC,CAAC,CAAC,CAAC;QAC1I,oBAAoB;QACpB,WAAW,EAAE,CACd,CAAC;QAEF,OAAO,MAAM,CAAC,WAAW,CACvB,cAAc,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3B,IAAI;YACJ,UAAU,CAAC,IAAI,CACb,MAAM,CAAC,CAAC,EAAE,cAAc,EAAE,EAAE,EAAE,CAAC,cAAc,KAAK,IAAI,CAAC,EACvD,GAAG,CAAC,CAAC,EAAE,UAAU,EAAE,EAAE,EAAE,CAAC,UAAU,CAAC,EACnC,MAAM,EAAE,CACT;SACF,CAAC,CACgB,CAAC;IACvB,CAAC;IAEO,gBAAgB,CAAC,eAAgC;QACvD,OAAO,QAAQ,CAAC;YACd,SAAS,EAAE,eAAe,CAAC,KAAK,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAChD,WAAW,EAAE,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;SACrD,CAAC,CAAC,IAAI,CACL,QAAQ,CAAC,CAAC,EAAE,SAAS,EAAE,WAAW,EAAE,EAAE,EAAE;YACtC,IAAI,CAAC,SAAS,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,EAAE;gBAC5C,OAAO,KAAK,CAAC;aACd;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG;;uFAEiE,cAAc,CAAC,cAAc,EAAE,WAAW,EAAE,QAAQ,CAAC;;;;;;;8BAO9G,cAAc,CAAC,cAAc,EAAE,SAAS,EAAE,QAAQ,CAAC;;;;;;;qDAO5B,CAAC;YAC9C,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,EACF,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,CAC/B,CAAC;IACJ,CAAC;IAEO,uBAAuB,CAAC,eAAgC;QAC9D,OAAO,QAAQ,CAAC;YACd,eAAe,EAAE,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YAC5D,YAAY,EAAE,eAAe,CAAC,QAAQ,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;SACvD,CAAC,CAAC,IAAI,CACL,QAAQ,CAAC,CAAC,EAAE,eAAe,EAAE,YAAY,EAAE,EAAE,EAAE;YAC7C,IAAI,CAAC,eAAe,CAAC,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE;gBACnD,OAAO,KAAK,CAAC;aACd;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG;;;;8BAIQ,cAAc,CAAC,gBAAgB,EAAE,YAAY,EAAE,QAAQ,CAAC;;;;8BAIxD,cAAc,CAAC,gBAAgB,EAAE,eAAe,EAAE,QAAQ,CAAC;;;;2DAI9B,CAAC;YACpD,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,EACF,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAY,CAAC,CAC/B,CAAC;IACJ,CAAC;IAEO,kBAAkB,CAAC,eAAgC;QACzD,OAAO,QAAQ,CAAC;YACd,2BAA2B,EAAE,eAAe,CAAC,uBAAuB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACpF,oBAAoB,EAAE,eAAe,CAAC,gBAAgB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACtE,kBAAkB,EAAE,eAAe,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;YACrE,WAAW,EAAE,eAAe,CAAC,OAAO,CAAC,IAAI,CAAC,OAAO,EAAE,CAAC;SACrD,CAAC,CAAC,IAAI,CACL,QAAQ,CAAC,CAAC,EAAE,2BAA2B,EAAE,oBAAoB,EAAE,kBAAkB,EAAE,WAAW,EAAE,EAAE,EAAE;YAClG,MAAM,qBAAqB,GAAG,CAAC,CAAC,kBAAkB,CAAC,MAAM,CAAC;YAC1D,IAAI,CAAC,2BAA2B,CAAC,MAAM,IAAI,CAAC,oBAAoB,CAAC,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,IAAI,CAAC,qBAAqB,EAAE;gBACxH,OAAO,KAAK,CAAC;aACd;YAED,MAAM,QAAQ,GAAmB,EAAE,CAAC;YACpC,MAAM,KAAK,GAAG;sBACA,qBAAqB,CAAC,CAAC,CAAC,IAAI,CAAC,iCAAiC,CAAC,kBAAkB,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,EAAE;;;8BAGzF,cAAc,CAAC,oBAAoB,EAAE,2BAA2B,EAAE,QAAQ,CAAC;;;;;;;;;kFASvB,cAAc,CAAC,cAAc,EAAE,WAAW,EAAE,QAAQ,CAAC;;;;;;2FAM5C,cAAc,CAAC,cAAc,EAAE,oBAAoB,EAAE,QAAQ,CAAC;;;;;qBAMpI,qBAAqB;gBACnB,CAAC,CAAC;;;oBAGL;gBACG,CAAC,CAAC,EACN;;;;;oHAKiG,CAAC;YAC7G,OAAO,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC,CAAC;QACtE,CAAC,CAAC,EACF,GAAG,CAAC,CAAC,GAAG,EAAE,EAAE,CAAC,GAAG,CAAC,YAAsB,CAAC,CACzC,CAAC;IACJ,CAAC;IAEO,iCAAiC,CAAC,kBAA4B,EAAE,QAAwB;QAC9F,OAAO;uFAC4E,cAAc,CAAC,cAAc,EAAE,kBAAkB,EAAE,QAAQ,CAAC;;;;;;;;;;;;;;;;;;;;;;;;;;;;;eA6BpI,CAAC;IACd,CAAC;IAEO,KAAK,CAAC,QAAQ,CAAC,aAAqB;QAC1C,MAAM,EAAE,UAAU,EAAE,SAAS,EAAE,GAAG,kBAAkB,CAAC,aAAa,CAAC,CAAC;QACpE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,iBAAiB,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QAClE,IAAI,CAAC,MAAM,EAAE;YACX,OAAO,SAAS,CAAC;SAClB;QACD,MAAM,WAAW,GAAG,MAAM,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;QACrD,IAAI,CAAC,WAAW,EAAE;YAChB,OAAO,SAAS,CAAC;SAClB;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;CACF;AAED,MAAM,cAAc,GAAG;IACrB,SAAS;IACT,OAAO;IACP,UAAU;IACV,aAAa;IACb,mBAAmB;IACnB,yBAAyB;IACzB,kBAAkB;IAClB,SAAS;IACT,SAAS;CACD,CAAC;AAQX,SAAS,MAAM;IACb,OAAO,UAAU,GAAkB;QACjC,OAAO,GAAG,CAAC,IAAI,CACb,IAAI,CACF,CAAC,GAAG,EAAE,GAAG,EAAE,EAAE;YACX,IAAI,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,EAAE;gBACpB,OAAO,GAAG,CAAC,GAAG,CAAC;gBACf,OAAO,GAAG,CAAC;aACZ;YAED,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YACjB,GAAG,CAAC,GAAG,GAAG,GAAG,CAAC;YACd,OAAO,GAAG,CAAC;QACb,CAAC,EACD,EAAE,GAAG,EAAE,IAAI,GAAG,EAAK,EAA8B,CAClD,EACD,GAAG,CAAC,CAAC,EAAE,GAAG,EAAE,EAAE,EAAE,CAAC,GAAG,CAAC,EACrB,MAAM,CAAC,CAAC,CAAC,EAAU,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,CAC3B,CAAC;IACJ,CAAC,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\n/** @packageDocumentation\n * @module UnifiedSelection\n */\n\nimport { EMPTY, filter, forkJoin, from, map, merge, mergeMap, Observable, scan, shareReplay, Subject, toArray } from \"rxjs\";\nimport { eachValueFrom } from \"rxjs-for-await\";\nimport { ECClass, IMetadataProvider, parseFullClassName } from \"./queries/ECMetadata\";\nimport { ECSqlBinding, formIdBindings, IECSqlQueryExecutor } from \"./queries/ECSqlCore\";\nimport { SelectableInstanceKey, Selectables } from \"./Selectable\";\n\nconst HILITE_SET_EMIT_FREQUENCY = 20;\n\n/**\n * A set of model, subcategory and element ids that can be used for specifying hilite.\n * @see https://www.itwinjs.org/reference/core-frontend/selectionset/hiliteset/\n * @beta\n */\nexport interface HiliteSet {\n  models: string[];\n  subCategories: string[];\n  elements: string[];\n}\n\n/**\n * Props for creating a `HiliteSetProvider` instance.\n * @internal Not exported through barrel, but used in public API as an argument. May be supplemented with optional attributes any time.\n */\nexport interface HiliteSetProviderProps {\n  queryExecutor: IECSqlQueryExecutor;\n  metadataProvider: IMetadataProvider;\n}\n\n/**\n * Defines return value of `createHiliteSetProvider`.\n *\n * @beta Used in public API as a return value. Not expected to be created / extended by package\n * consumers, may be supplemented with required attributes any time.\n */\nexport interface HiliteSetProvider {\n  /** Get the current hilite set iterator for the specified imodel */\n  getHiliteSet(props: { selectables: Selectables }): AsyncIterableIterator<HiliteSet>;\n}\n\n/**\n * Creates a hilite set provider that returns a `HiliteSet` for given selectables.\n * @beta\n */\nexport function createHiliteSetProvider(props: HiliteSetProviderProps): HiliteSetProvider {\n  return new HiliteSetProviderImpl(props);\n}\n\nclass HiliteSetProviderImpl implements HiliteSetProvider {\n  private _queryExecutor: IECSqlQueryExecutor;\n  private _metadataProvider: IMetadataProvider;\n  // Map between a class name and its type\n  private _classRelationCache: Map<string, InstanceIdType | Promise<InstanceIdType>>;\n\n  constructor(props: HiliteSetProviderProps) {\n    this._queryExecutor = props.queryExecutor;\n    this._metadataProvider = props.metadataProvider;\n    this._classRelationCache = new Map();\n  }\n\n  /**\n   * Get hilite set iterator for supplied `Selectables`.\n   */\n  public getHiliteSet(props: { selectables: Selectables }): AsyncIterableIterator<HiliteSet> {\n    const obs = this.getHiliteSetObservable(props);\n    return eachValueFrom(obs);\n  }\n\n  /**\n   * Returns a \"hot\" observable of hilite sets.\n   */\n  private getHiliteSetObservable({ selectables }: { selectables: Selectables }): Subject<HiliteSet> {\n    const instancesByType = this.getInstancesByType(selectables);\n    const observables: { [key in keyof HiliteSet]?: Observable<string> } = {\n      models: this.getHilitedModels(instancesByType),\n      subCategories: this.getHilitedSubCategories(instancesByType),\n      elements: this.getHilitedElements(instancesByType),\n    };\n\n    let hiliteSet: HiliteSet = { models: [], subCategories: [], elements: [] };\n    let lastEmitTime = performance.now();\n    const subject = new Subject<HiliteSet>();\n    const subscriptions = ([\"models\", \"subCategories\", \"elements\"] as const).map((key) =>\n      observables[key]!.subscribe({\n        next(val) {\n          hiliteSet[key].push(val);\n          if (performance.now() - lastEmitTime < HILITE_SET_EMIT_FREQUENCY) {\n            return;\n          }\n          subject.next(hiliteSet);\n          hiliteSet = { models: [], subCategories: [], elements: [] };\n          lastEmitTime = performance.now();\n        },\n        complete() {\n          observables[key] = undefined;\n          if (observables.models || observables.subCategories || observables.elements) {\n            return;\n          }\n          // Emit last batch before completing the observable.\n          if (hiliteSet.elements.length || hiliteSet.models.length || hiliteSet.subCategories.length) {\n            subject.next(hiliteSet);\n          }\n          subject.complete();\n        },\n        error(err) {\n          subscriptions.forEach((x) => x.unsubscribe());\n          subscriptions.length = 0;\n          subject.error(err);\n        },\n      }),\n    );\n\n    return subject;\n  }\n\n  private async getType(key: SelectableInstanceKey): Promise<InstanceIdType> {\n    const cachedType = this._classRelationCache.get(key.className.replace(\".\", \":\"));\n    if (cachedType) {\n      return cachedType;\n    }\n\n    const promise = this.getTypeImpl(key).then((res) => {\n      // Update the cache with the result of the promise.\n      this._classRelationCache.set(key.className, res);\n      return res;\n    });\n\n    // Add the promise to cache to prevent `getTypeImpl` being called multiple times.\n    this._classRelationCache.set(key.className, promise);\n    return promise;\n  }\n\n  private async getTypeImpl(key: SelectableInstanceKey): Promise<InstanceIdType> {\n    const keyClass = await this.getClass(key.className);\n    return (\n      (keyClass &&\n        ((await this.checkType(keyClass, \"BisCore\", \"Subject\", \"subject\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"Model\", \"model\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"Category\", \"category\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"SubCategory\", \"subCategory\")) ??\n          (await this.checkType(keyClass, \"Functional\", \"FunctionalElement\", \"functionalElement\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"GroupInformationElement\", \"groupInformationElement\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"GeometricElement\", \"geometricElement\")) ??\n          (await this.checkType(keyClass, \"BisCore\", \"Element\", \"element\")))) ??\n      \"unknown\"\n    );\n  }\n\n  private async checkType(keyClass: ECClass, schemaName: string, className: string, type: InstanceIdType) {\n    return (await keyClass.is(className, schemaName)) ? type : undefined;\n  }\n\n  private getInstancesByType(selectables: Selectables): InstancesByType {\n    const keyTypeObs = merge(\n      from(selectables.custom.values()).pipe(mergeMap((selectable) => selectable.loadInstanceKeys())),\n      from(selectables.instanceKeys).pipe(mergeMap(([className, idSet]) => from(idSet).pipe(map((id) => ({ className, id }))))),\n    ).pipe(\n      // Get types for each instance key\n      mergeMap((instanceKey) => from(this.getType(instanceKey)).pipe(map((instanceIdType) => ({ instanceId: instanceKey.id, instanceIdType })))),\n      // Cache the results\n      shareReplay(),\n    );\n\n    return Object.fromEntries(\n      INSTANCE_TYPES.map((type) => [\n        type,\n        keyTypeObs.pipe(\n          filter(({ instanceIdType }) => instanceIdType === type),\n          map(({ instanceId }) => instanceId),\n          unique(),\n        ),\n      ]),\n    ) as InstancesByType;\n  }\n\n  private getHilitedModels(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      modelKeys: instancesByType.model.pipe(toArray()),\n      subjectKeys: instancesByType.subject.pipe(toArray()),\n    }).pipe(\n      mergeMap(({ modelKeys, subjectKeys }) => {\n        if (!modelKeys.length && !subjectKeys.length) {\n          return EMPTY;\n        }\n\n        const bindings: ECSqlBinding[] = [];\n        const query = `WITH RECURSIVE\n                    ChildSubjects (ECInstanceId, JsonProperties) AS (\n                      SELECT ECInstanceId, JsonProperties FROM BisCore.Subject WHERE ${formIdBindings(\"ECInstanceId\", subjectKeys, bindings)}\n                      UNION ALL\n                      SELECT r.ECInstanceId, r.JsonProperties FROM ChildSubjects s\n                        JOIN BisCore.Subject r ON r.Parent.Id = s.ECInstanceId\n                    ),\n                    Models (ECInstanceId) AS (\n                      SELECT s.ECInstanceId as \\`ECInstanceId\\` FROM BisCore.Model s\n                      WHERE ${formIdBindings(\"ECInstanceId\", modelKeys, bindings)}\n                    )\n                    SELECT r.ECInstanceId as \\`ECInstanceId\\` FROM ChildSubjects s\n                      JOIN BisCore.PhysicalPartition r\n                        ON r.Parent.Id = s.ECInstanceId\n                        OR json_extract(s.JsonProperties,'$.Subject.Model.TargetPartition') = printf('0x%x', r.ECInstanceId)\n                    UNION\n                    SELECT ECInstanceId FROM Models;`;\n        return from(this._queryExecutor.createQueryReader(query, bindings));\n      }),\n      map((row) => row.ECInstanceId),\n    );\n  }\n\n  private getHilitedSubCategories(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      subCategoryKeys: instancesByType.subCategory.pipe(toArray()),\n      categoryKeys: instancesByType.category.pipe(toArray()),\n    }).pipe(\n      mergeMap(({ subCategoryKeys, categoryKeys }) => {\n        if (!subCategoryKeys.length && !categoryKeys.length) {\n          return EMPTY;\n        }\n\n        const bindings: ECSqlBinding[] = [];\n        const query = `WITH\n                    CategorySubCategories (ECInstanceId) AS (\n                      SELECT r.ECInstanceId as \\`ECInstanceId\\` FROM BisCore.Category s\n                        JOIN BisCore.SubCategory r ON r.Parent.Id = s.ECInstanceId\n                      WHERE ${formIdBindings(\"s.ECInstanceId\", categoryKeys, bindings)}\n                    ),\n                    SubCategories (ECInstanceId) AS (\n                      SELECT s.ECInstanceId as \\`ECInstanceId\\` FROM BisCore.SubCategory s\n                      WHERE ${formIdBindings(\"s.ECInstanceId\", subCategoryKeys, bindings)}\n                    )\n                   SELECT ECInstanceId FROM CategorySubCategories\n                   UNION\n                   SELECT ECInstanceId FROM SubCategories;`;\n        return from(this._queryExecutor.createQueryReader(query, bindings));\n      }),\n      map((row) => row.ECInstanceId),\n    );\n  }\n\n  private getHilitedElements(instancesByType: InstancesByType): Observable<string> {\n    return forkJoin({\n      groupInformationElementKeys: instancesByType.groupInformationElement.pipe(toArray()),\n      geometricElementKeys: instancesByType.geometricElement.pipe(toArray()),\n      functionalElements: instancesByType.functionalElement.pipe(toArray()),\n      elementKeys: instancesByType.element.pipe(toArray()),\n    }).pipe(\n      mergeMap(({ groupInformationElementKeys, geometricElementKeys, functionalElements, elementKeys }) => {\n        const hasFunctionalElements = !!functionalElements.length;\n        if (!groupInformationElementKeys.length && !geometricElementKeys.length && !elementKeys.length && !hasFunctionalElements) {\n          return EMPTY;\n        }\n\n        const bindings: ECSqlBinding[] = [];\n        const query = `WITH RECURSIVE\n                    ${hasFunctionalElements ? this.getHilitedFunctionalElementsQuery(functionalElements, bindings) : \"\"}\n                    GroupMembers (ECInstanceId, ECClassId) AS (\n                      SELECT TargetECInstanceId, TargetECClassId FROM BisCore.ElementGroupsMembers\n                      WHERE ${formIdBindings(\"SourceECInstanceId\", groupInformationElementKeys, bindings)}\n                    ),\n                    GroupGeometricElements (ECInstanceId, ECClassId) AS (\n                      SELECT ECInstanceId, ECClassId FROM GroupMembers\n                      UNION ALL\n                      SELECT r.ECInstanceId, r.ECClassId FROM GroupGeometricElements s\n                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n                    ),\n                    ElementGeometricElements (ECInstanceId, ECClassId) AS (\n                      SELECT ECInstanceId, ECClassId FROM BisCore.Element WHERE ${formIdBindings(\"ECInstanceId\", elementKeys, bindings)}\n                      UNION ALL\n                      SELECT r.ECInstanceId, r.ECClassId FROM ElementGeometricElements s\n                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n                    ),\n                    GeometricElementGeometricElements (ECInstanceId, ECClassId) AS (\n                      SELECT ECInstanceId, ECClassId FROM BisCore.GeometricElement WHERE ${formIdBindings(\"ECInstanceId\", geometricElementKeys, bindings)}\n                      UNION ALL\n                      SELECT r.ECInstanceId, r.ECClassId FROM GeometricElementGeometricElements s\n                        JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n                    )\n                   ${\n                     hasFunctionalElements\n                       ? `\n                   SELECT ECInstanceId FROM FunctionalElementChildGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\n                   UNION\n                   `\n                       : ``\n                   }\n                   SELECT ECInstanceId FROM GeometricElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\n                   UNION\n                   SELECT ECInstanceId FROM GroupGeometricElements WHERE ECClassId IS (BisCore.GeometricElement)\n                   UNION\n                   SELECT ECInstanceId FROM ElementGeometricElements WHERE ECClassId IS (BisCore.GeometricElement);`;\n        return from(this._queryExecutor.createQueryReader(query, bindings));\n      }),\n      map((row) => row.ECInstanceId as string),\n    );\n  }\n\n  private getHilitedFunctionalElementsQuery(functionalElements: string[], bindings: ECSqlBinding[]): string {\n    return `ChildFunctionalElements (ECInstanceId, ECClassId) AS (\n              SELECT ECInstanceId, ECClassId FROM Functional.FunctionalElement WHERE ${formIdBindings(\"ECInstanceId\", functionalElements, bindings)}\n              UNION ALL\n              SELECT r.ECInstanceId, r.ECClassId FROM ChildFunctionalElements s\n                JOIN Functional.FunctionalElement r ON r.Parent.Id = s.ECInstanceId\n            ),\n            PhysicalElements (ECInstanceId, ECClassId) AS (\n              SELECT r.SourceECInstanceId, r.SourceECClassId FROM ChildFunctionalElements s\n                JOIN Functional.PhysicalElementFulfillsFunction r ON r.TargetECInstanceId = s.ECInstanceId\n            ),\n            DrawingGraphicElements (ECInstanceId, ECClassId) AS (\n              SELECT r.SourceECInstanceId, r.SourceECClassId FROM ChildFunctionalElements s\n                JOIN Functional.DrawingGraphicRepresentsFunctionalElement r ON r.TargetECInstanceId = s.ECInstanceId\n            ),\n            PhysicalElementGeometricElements (ECInstanceId, ECClassId) AS (\n              SELECT ECInstanceId, ECClassId FROM PhysicalElements\n              UNION ALL\n              SELECT r.ECInstanceId, r.ECClassId FROM PhysicalElementGeometricElements s\n                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n            ),\n            DrawingGraphicElementGeometricElements (ECInstanceId, ECClassId) AS (\n              SELECT ECInstanceId, ECClassId FROM DrawingGraphicElements\n              UNION ALL\n              SELECT r.ECInstanceId, r.ECClassId FROM DrawingGraphicElementGeometricElements s\n                JOIN BisCore.Element r ON r.Parent.Id = s.ECInstanceId\n            ),\n            FunctionalElementChildGeometricElements (ECInstanceId, ECClassId) AS (\n              SELECT ECInstanceId, ECClassId FROM PhysicalElementGeometricElements\n              UNION\n              SELECT ECInstanceId, ECClassId FROM DrawingGraphicElementGeometricElements\n            ),`;\n  }\n\n  private async getClass(fullClassName: string): Promise<ECClass | undefined> {\n    const { schemaName, className } = parseFullClassName(fullClassName);\n    const schema = await this._metadataProvider.getSchema(schemaName);\n    if (!schema) {\n      return undefined;\n    }\n    const schemaClass = await schema.getClass(className);\n    if (!schemaClass) {\n      return undefined;\n    }\n    return schemaClass;\n  }\n}\n\nconst INSTANCE_TYPES = [\n  \"subject\",\n  \"model\",\n  \"category\",\n  \"subCategory\",\n  \"functionalElement\",\n  \"groupInformationElement\",\n  \"geometricElement\",\n  \"element\",\n  \"unknown\",\n] as const;\n\ntype InstanceIdType = (typeof INSTANCE_TYPES)[number];\n\ntype InstancesByType = {\n  [idType in InstanceIdType]: Observable<string>;\n};\n\nfunction unique<T>() {\n  return function (obs: Observable<T>): Observable<T> {\n    return obs.pipe(\n      scan(\n        (acc, val) => {\n          if (acc.set.has(val)) {\n            delete acc.val;\n            return acc;\n          }\n\n          acc.set.add(val);\n          acc.val = val;\n          return acc;\n        },\n        { set: new Set<T>() } as { set: Set<T>; val?: T },\n      ),\n      map(({ val }) => val),\n      filter((x): x is T => !!x),\n    );\n  };\n}\n"]}