@itwin/presentation-frontend 4.5.0-dev.8 → 4.5.0

package/lib/esm/presentation-frontend/selection/SelectionHandler.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionHandler.js","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHandler.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F;;GAEG;AAEH,OAAO,EAAE,cAAc,EAAe,MAAM,qBAAqB,CAAC;AAiClE;;;;;GAKG;AACH,MAAM,OAAO,gBAAgB;IAkB3B;;OAEG;IACH,YAAY,KAA4B;QAmBxC;;;WAGG;QACH,gEAAgE;QACtD,uBAAkB,GAAG,CAAC,GAA6B,EAAE,QAA4B,EAAQ,EAAE;YACnG,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC;gBAC3C,OAAO;YAET,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC,CAAC;QA9BA,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,cAAc,EAAE,CAAC;QACzC,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED;;;OAGG;IACI,OAAO;QACZ,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAgBD,0EAA0E;IAChE,YAAY,CAAC,GAA6B;QAClD,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM;YAC1B,OAAO,KAAK,CAAC;QACf,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kEAAkE;IAC3D,kBAAkB;QACvB,OAAO,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED;;;OAGG;IACI,YAAY,CAAC,KAAc;QAChC,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACvD,CAAC;IAED;;;;OAIG;IACI,cAAc,CAAC,IAAU,EAAE,QAAgB,CAAC;QACjD,IAAI,IAAI,CAAC,SAAS;YAChB,OAAO;QAET,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1F,CAAC;IAED;;;;OAIG;IACI,mBAAmB,CAAC,IAAU,EAAE,QAAgB,CAAC;QACtD,IAAI,IAAI,CAAC,SAAS;YAChB,OAAO;QAET,OAAO,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/F,CAAC;IAED;;;;OAIG;IACI,gBAAgB,CAAC,IAAU,EAAE,QAAgB,CAAC;QACnD,IAAI,IAAI,CAAC,SAAS;YAChB,OAAO;QAET,OAAO,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5F,CAAC;IAED;;;OAGG;IACI,cAAc,CAAC,QAAgB,CAAC;QACrC,IAAI,IAAI,CAAC,SAAS;YAChB,OAAO;QAET,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IACpF,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/** @packageDocumentation\n * @module UnifiedSelection\n */\n\nimport { DisposableList, IDisposable } from \"@itwin/core-bentley\";\nimport { IModelConnection } from \"@itwin/core-frontend\";\nimport { Keys, KeySet } from \"@itwin/presentation-common\";\nimport { ISelectionProvider } from \"./ISelectionProvider\";\nimport { SelectionChangeEventArgs, SelectionChangesListener } from \"./SelectionChangeEvent\";\nimport { SelectionManager } from \"./SelectionManager\";\n\n/**\n * Properties for creating a `SelectionHandler` instance.\n * @public\n */\nexport interface SelectionHandlerProps {\n  /** SelectionManager used to store overall selection. */\n  manager: SelectionManager;\n  /** iModel connection the selection changes will be associated with. */\n  imodel: IModelConnection;\n  /**\n   * Name of the selection handler. This is an identifier of what caused the\n   * selection to change, set as `SelectionChangeEventArgs.source` when firing\n   * selection change events. `SelectionHandler.shouldHandle` uses `name` to filter\n   * events that it doesn't need to handle.\n   */\n  name: string;\n  /**\n   * ID of presentation ruleset used by the component using this handler. The ID is set as\n   * `SelectionChangeEventArgs.rulesetId` when making selection changes and event\n   * listeners can use or ignore this information.\n   */\n  rulesetId?: string;\n  /** Callback function called when selection changes. */\n  onSelect?: SelectionChangesListener;\n}\n\n/**\n * A class that handles selection changes and helps to change\n * internal the selection state.\n *\n * @public\n */\nexport class SelectionHandler implements IDisposable {\n  private _inSelect: boolean;\n  private _disposables: DisposableList;\n\n  /** Selection manager used by this handler to manage selection */\n  public readonly manager: SelectionManager;\n  /** Name that's used as `SelectionChangeEventArgs.source` when making selection changes */\n  public name: string;\n  /** iModel whose selection is being handled */\n  public imodel: IModelConnection;\n  /**\n   * Id of a ruleset selection changes will be associated with.\n   * @see `SelectionHandlerProps.rulesetId`\n   */\n  public rulesetId?: string;\n  /** Callback function called when selection changes */\n  public onSelect?: SelectionChangesListener;\n\n  /**\n   * Constructor.\n   */\n  constructor(props: SelectionHandlerProps) {\n    this._inSelect = false;\n    this.manager = props.manager;\n    this._disposables = new DisposableList();\n    this.name = props.name;\n    this.rulesetId = props.rulesetId;\n    this.imodel = props.imodel;\n    this.onSelect = props.onSelect;\n    this._disposables.add(this.manager.selectionChange.addListener(this.onSelectionChanged));\n  }\n\n  /**\n   * Destructor. Must be called before disposing this object to make sure it cleans\n   * up correctly.\n   */\n  public dispose(): void {\n    this._disposables.dispose();\n  }\n\n  /**\n   * Called when the selection changes. Handles this callback by first checking whether\n   * the event should be handled at all (using the `shouldHandle` method) and then calling `onSelect`\n   */\n  // eslint-disable-next-line @typescript-eslint/naming-convention\n  protected onSelectionChanged = (evt: SelectionChangeEventArgs, provider: ISelectionProvider): void => {\n    if (!this.onSelect || !this.shouldHandle(evt))\n      return;\n\n    this._inSelect = true;\n    this.onSelect(evt, provider);\n    this._inSelect = false;\n  };\n\n  /** Called to check whether the event should be handled by this handler */\n  protected shouldHandle(evt: SelectionChangeEventArgs): boolean {\n    if (this.name === evt.source)\n      return false;\n    return true;\n  }\n\n  /** Get selection levels for the imodel managed by this handler */\n  public getSelectionLevels(): number[] {\n    return this.manager.getSelectionLevels(this.imodel);\n  }\n\n  /**\n   * Get selection for the imodel managed by this handler.\n   * @param level Level of the selection to get. Defaults to 0.\n   */\n  public getSelection(level?: number): Readonly<KeySet> {\n    return this.manager.getSelection(this.imodel, level);\n  }\n\n  /**\n   * Add to selection.\n   * @param keys The keys to add to selection.\n   * @param level Level of the selection.\n   */\n  public addToSelection(keys: Keys, level: number = 0): void {\n    if (this._inSelect)\n      return;\n\n    return this.manager.addToSelection(this.name, this.imodel, keys, level, this.rulesetId);\n  }\n\n  /**\n   * Remove from selection.\n   * @param keys The keys to remove from selection.\n   * @param level Level of the selection.\n   */\n  public removeFromSelection(keys: Keys, level: number = 0): void {\n    if (this._inSelect)\n      return;\n\n    return this.manager.removeFromSelection(this.name, this.imodel, keys, level, this.rulesetId);\n  }\n\n  /**\n   * Change selection.\n   * @param keys The keys indicating the new selection.\n   * @param level Level of the selection.\n   */\n  public replaceSelection(keys: Keys, level: number = 0): void {\n    if (this._inSelect)\n      return;\n\n    return this.manager.replaceSelection(this.name, this.imodel, keys, level, this.rulesetId);\n  }\n\n  /**\n   * Clear selection.\n   * @param level Level of the selection.\n   */\n  public clearSelection(level: number = 0): void {\n    if (this._inSelect)\n      return;\n\n    return this.manager.clearSelection(this.name, this.imodel, level, this.rulesetId);\n  }\n}\n"]}
+{"version":3,"file":"SelectionHandler.js","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHandler.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAChG;;GAEG;AAEH,OAAO,EAAE,cAAc,EAAe,MAAM,qBAAqB,CAAC;AAiClE;;;;;GAKG;AACH,MAAM,OAAO,gBAAgB;IAkB3B;;OAEG;IACH,YAAY,KAA4B;QAmBxC;;;WAGG;QACH,gEAAgE;QACtD,uBAAkB,GAAG,CAAC,GAA6B,EAAE,QAA4B,EAAQ,EAAE;YACnG,IAAI,CAAC,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,EAAE;gBAC7C,OAAO;aACR;YAED,IAAI,CAAC,SAAS,GAAG,IAAI,CAAC;YACtB,IAAI,CAAC,QAAQ,CAAC,GAAG,EAAE,QAAQ,CAAC,CAAC;YAC7B,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACzB,CAAC,CAAC;QA/BA,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC;QACvB,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,OAAO,CAAC;QAC7B,IAAI,CAAC,YAAY,GAAG,IAAI,cAAc,EAAE,CAAC;QACzC,IAAI,CAAC,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC;QACvB,IAAI,CAAC,SAAS,GAAG,KAAK,CAAC,SAAS,CAAC;QACjC,IAAI,CAAC,MAAM,GAAG,KAAK,CAAC,MAAM,CAAC;QAC3B,IAAI,CAAC,QAAQ,GAAG,KAAK,CAAC,QAAQ,CAAC;QAC/B,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,IAAI,CAAC,OAAO,CAAC,eAAe,CAAC,WAAW,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC,CAAC;IAC3F,CAAC;IAED;;;OAGG;IACI,OAAO;QACZ,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,CAAC;IAC9B,CAAC;IAiBD,0EAA0E;IAChE,YAAY,CAAC,GAA6B;QAClD,IAAI,IAAI,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM,EAAE;YAC5B,OAAO,KAAK,CAAC;SACd;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,kEAAkE;IAC3D,kBAAkB;QACvB,OAAO,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACtD,CAAC;IAED;;;OAGG;IACI,YAAY,CAAC,KAAc;QAChC,OAAO,IAAI,CAAC,OAAO,CAAC,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACvD,CAAC;IAED;;;;OAIG;IACI,cAAc,CAAC,IAAU,EAAE,QAAgB,CAAC;QACjD,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,OAAO;SACR;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC1F,CAAC;IAED;;;;OAIG;IACI,mBAAmB,CAAC,IAAU,EAAE,QAAgB,CAAC;QACtD,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,OAAO;SACR;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,mBAAmB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC/F,CAAC;IAED;;;;OAIG;IACI,gBAAgB,CAAC,IAAU,EAAE,QAAgB,CAAC;QACnD,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,OAAO;SACR;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IAC5F,CAAC;IAED;;;OAGG;IACI,cAAc,CAAC,QAAgB,CAAC;QACrC,IAAI,IAAI,CAAC,SAAS,EAAE;YAClB,OAAO;SACR;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,cAAc,CAAC,IAAI,CAAC,IAAI,EAAE,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,IAAI,CAAC,SAAS,CAAC,CAAC;IACpF,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n * See LICENSE.md in the project root for license terms and full copyright notice.\r\n *--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module UnifiedSelection\r\n */\r\n\r\nimport { DisposableList, IDisposable } from \"@itwin/core-bentley\";\r\nimport { IModelConnection } from \"@itwin/core-frontend\";\r\nimport { Keys, KeySet } from \"@itwin/presentation-common\";\r\nimport { ISelectionProvider } from \"./ISelectionProvider\";\r\nimport { SelectionChangeEventArgs, SelectionChangesListener } from \"./SelectionChangeEvent\";\r\nimport { SelectionManager } from \"./SelectionManager\";\r\n\r\n/**\r\n * Properties for creating a `SelectionHandler` instance.\r\n * @public\r\n */\r\nexport interface SelectionHandlerProps {\r\n  /** SelectionManager used to store overall selection. */\r\n  manager: SelectionManager;\r\n  /** iModel connection the selection changes will be associated with. */\r\n  imodel: IModelConnection;\r\n  /**\r\n   * Name of the selection handler. This is an identifier of what caused the\r\n   * selection to change, set as `SelectionChangeEventArgs.source` when firing\r\n   * selection change events. `SelectionHandler.shouldHandle` uses `name` to filter\r\n   * events that it doesn't need to handle.\r\n   */\r\n  name: string;\r\n  /**\r\n   * ID of presentation ruleset used by the component using this handler. The ID is set as\r\n   * `SelectionChangeEventArgs.rulesetId` when making selection changes and event\r\n   * listeners can use or ignore this information.\r\n   */\r\n  rulesetId?: string;\r\n  /** Callback function called when selection changes. */\r\n  onSelect?: SelectionChangesListener;\r\n}\r\n\r\n/**\r\n * A class that handles selection changes and helps to change\r\n * internal the selection state.\r\n *\r\n * @public\r\n */\r\nexport class SelectionHandler implements IDisposable {\r\n  private _inSelect: boolean;\r\n  private _disposables: DisposableList;\r\n\r\n  /** Selection manager used by this handler to manage selection */\r\n  public readonly manager: SelectionManager;\r\n  /** Name that's used as `SelectionChangeEventArgs.source` when making selection changes */\r\n  public name: string;\r\n  /** iModel whose selection is being handled */\r\n  public imodel: IModelConnection;\r\n  /**\r\n   * Id of a ruleset selection changes will be associated with.\r\n   * @see `SelectionHandlerProps.rulesetId`\r\n   */\r\n  public rulesetId?: string;\r\n  /** Callback function called when selection changes */\r\n  public onSelect?: SelectionChangesListener;\r\n\r\n  /**\r\n   * Constructor.\r\n   */\r\n  constructor(props: SelectionHandlerProps) {\r\n    this._inSelect = false;\r\n    this.manager = props.manager;\r\n    this._disposables = new DisposableList();\r\n    this.name = props.name;\r\n    this.rulesetId = props.rulesetId;\r\n    this.imodel = props.imodel;\r\n    this.onSelect = props.onSelect;\r\n    this._disposables.add(this.manager.selectionChange.addListener(this.onSelectionChanged));\r\n  }\r\n\r\n  /**\r\n   * Destructor. Must be called before disposing this object to make sure it cleans\r\n   * up correctly.\r\n   */\r\n  public dispose(): void {\r\n    this._disposables.dispose();\r\n  }\r\n\r\n  /**\r\n   * Called when the selection changes. Handles this callback by first checking whether\r\n   * the event should be handled at all (using the `shouldHandle` method) and then calling `onSelect`\r\n   */\r\n  // eslint-disable-next-line @typescript-eslint/naming-convention\r\n  protected onSelectionChanged = (evt: SelectionChangeEventArgs, provider: ISelectionProvider): void => {\r\n    if (!this.onSelect || !this.shouldHandle(evt)) {\r\n      return;\r\n    }\r\n\r\n    this._inSelect = true;\r\n    this.onSelect(evt, provider);\r\n    this._inSelect = false;\r\n  };\r\n\r\n  /** Called to check whether the event should be handled by this handler */\r\n  protected shouldHandle(evt: SelectionChangeEventArgs): boolean {\r\n    if (this.name === evt.source) {\r\n      return false;\r\n    }\r\n    return true;\r\n  }\r\n\r\n  /** Get selection levels for the imodel managed by this handler */\r\n  public getSelectionLevels(): number[] {\r\n    return this.manager.getSelectionLevels(this.imodel);\r\n  }\r\n\r\n  /**\r\n   * Get selection for the imodel managed by this handler.\r\n   * @param level Level of the selection to get. Defaults to 0.\r\n   */\r\n  public getSelection(level?: number): Readonly<KeySet> {\r\n    return this.manager.getSelection(this.imodel, level);\r\n  }\r\n\r\n  /**\r\n   * Add to selection.\r\n   * @param keys The keys to add to selection.\r\n   * @param level Level of the selection.\r\n   */\r\n  public addToSelection(keys: Keys, level: number = 0): void {\r\n    if (this._inSelect) {\r\n      return;\r\n    }\r\n\r\n    return this.manager.addToSelection(this.name, this.imodel, keys, level, this.rulesetId);\r\n  }\r\n\r\n  /**\r\n   * Remove from selection.\r\n   * @param keys The keys to remove from selection.\r\n   * @param level Level of the selection.\r\n   */\r\n  public removeFromSelection(keys: Keys, level: number = 0): void {\r\n    if (this._inSelect) {\r\n      return;\r\n    }\r\n\r\n    return this.manager.removeFromSelection(this.name, this.imodel, keys, level, this.rulesetId);\r\n  }\r\n\r\n  /**\r\n   * Change selection.\r\n   * @param keys The keys indicating the new selection.\r\n   * @param level Level of the selection.\r\n   */\r\n  public replaceSelection(keys: Keys, level: number = 0): void {\r\n    if (this._inSelect) {\r\n      return;\r\n    }\r\n\r\n    return this.manager.replaceSelection(this.name, this.imodel, keys, level, this.rulesetId);\r\n  }\r\n\r\n  /**\r\n   * Clear selection.\r\n   * @param level Level of the selection.\r\n   */\r\n  public clearSelection(level: number = 0): void {\r\n    if (this._inSelect) {\r\n      return;\r\n    }\r\n\r\n    return this.manager.clearSelection(this.name, this.imodel, level, this.rulesetId);\r\n  }\r\n}\r\n"]}

package/lib/esm/presentation-frontend/selection/SelectionHelper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionHelper.d.ts","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHelper.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAE,GAAG,EAAE,IAAI,EAAW,MAAM,4BAA4B,CAAC;AAEhE;;;GAGG;AACH,qBAAa,eAAe;IAG1B,OAAO;IAEP;;;;OAIG;WACW,mBAAmB,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,GAAG,EAAE;CAe/D"}
+{"version":3,"file":"SelectionHelper.d.ts","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHelper.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAE,GAAG,EAAE,IAAI,EAAW,MAAM,4BAA4B,CAAC;AAEhE;;;GAGG;AACH,qBAAa,eAAe;IAE1B,OAAO;IAEP;;;;OAIG;WACW,mBAAmB,CAAC,IAAI,EAAE,QAAQ,CAAC,IAAI,CAAC,GAAG,GAAG,EAAE;CAe/D"}

package/lib/esm/presentation-frontend/selection/SelectionHelper.js

@@ -1,7 +1,7 @@
 /*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
+ * 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
  */
@@ -22,10 +22,12 @@ export class SelectionHelper {
         const result = new Array();
         keys.forEach((key) => {
             if (Key.isNodeKey(key)) {
-                if (NodeKey.isInstancesNodeKey(key))
+                if (NodeKey.isInstancesNodeKey(key)) {
                     result.push(...key.instanceKeys);
-                else
+                }
+                else {
                     result.push(key);
+                }
             }
             else {
                 result.push(key);

package/lib/esm/presentation-frontend/selection/SelectionHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionHelper.js","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHelper.ts"],"names":[],"mappings":"AAAA;;;+FAG+F;AAC/F;;GAEG;AAEH,OAAO,EAAE,GAAG,EAAQ,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAEhE;;;GAGG;AACH,MAAM,OAAO,eAAe;IAE1B,uBAAuB;IACvB,gBAAwB,CAAC;IAEzB;;;;OAIG;IACI,MAAM,CAAC,mBAAmB,CAAC,IAAoB;QACpD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAO,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAQ,EAAE,EAAE;YACxB,IAAI,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;gBACtB,IAAI,OAAO,CAAC,kBAAkB,CAAC,GAAG,CAAC;oBACjC,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC;;oBAEjC,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;aACpB;iBAAM;gBACL,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;aAClB;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;CAEF","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/** @packageDocumentation\n * @module UnifiedSelection\n */\n\nimport { Key, Keys, NodeKey } from \"@itwin/presentation-common\";\n\n/**\n * Helper class for working with selection.\n * @public\n */\nexport class SelectionHelper {\n\n  // istanbul ignore next\n  private constructor() { }\n\n  /**\n   * Re-map the given keyset for selection. This means all instance node keys get converted\n   * to instance keys, because in that case we want to select instances instead of nodes. All\n   * other types of keys ar left as is.\n   */\n  public static getKeysForSelection(keys: Readonly<Keys>): Key[] {\n    const result = new Array<Key>();\n    keys.forEach((key: Key) => {\n      if (Key.isNodeKey(key)) {\n        if (NodeKey.isInstancesNodeKey(key))\n          result.push(...key.instanceKeys);\n        else\n          result.push(key);\n      } else {\n        result.push(key);\n      }\n    });\n    return result;\n  }\n\n}\n"]}
+{"version":3,"file":"SelectionHelper.js","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionHelper.ts"],"names":[],"mappings":"AAAA;;;gGAGgG;AAChG;;GAEG;AAEH,OAAO,EAAE,GAAG,EAAQ,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAEhE;;;GAGG;AACH,MAAM,OAAO,eAAe;IAC1B,uBAAuB;IACvB,gBAAuB,CAAC;IAExB;;;;OAIG;IACI,MAAM,CAAC,mBAAmB,CAAC,IAAoB;QACpD,MAAM,MAAM,GAAG,IAAI,KAAK,EAAO,CAAC;QAChC,IAAI,CAAC,OAAO,CAAC,CAAC,GAAQ,EAAE,EAAE;YACxB,IAAI,GAAG,CAAC,SAAS,CAAC,GAAG,CAAC,EAAE;gBACtB,IAAI,OAAO,CAAC,kBAAkB,CAAC,GAAG,CAAC,EAAE;oBACnC,MAAM,CAAC,IAAI,CAAC,GAAG,GAAG,CAAC,YAAY,CAAC,CAAC;iBAClC;qBAAM;oBACL,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;iBAClB;aACF;iBAAM;gBACL,MAAM,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;aAClB;QACH,CAAC,CAAC,CAAC;QACH,OAAO,MAAM,CAAC;IAChB,CAAC;CACF","sourcesContent":["/*---------------------------------------------------------------------------------------------\r\n * Copyright (c) Bentley Systems, Incorporated. All rights reserved.\r\n * See LICENSE.md in the project root for license terms and full copyright notice.\r\n *--------------------------------------------------------------------------------------------*/\r\n/** @packageDocumentation\r\n * @module UnifiedSelection\r\n */\r\n\r\nimport { Key, Keys, NodeKey } from \"@itwin/presentation-common\";\r\n\r\n/**\r\n * Helper class for working with selection.\r\n * @public\r\n */\r\nexport class SelectionHelper {\r\n  // istanbul ignore next\r\n  private constructor() {}\r\n\r\n  /**\r\n   * Re-map the given keyset for selection. This means all instance node keys get converted\r\n   * to instance keys, because in that case we want to select instances instead of nodes. All\r\n   * other types of keys ar left as is.\r\n   */\r\n  public static getKeysForSelection(keys: Readonly<Keys>): Key[] {\r\n    const result = new Array<Key>();\r\n    keys.forEach((key: Key) => {\r\n      if (Key.isNodeKey(key)) {\r\n        if (NodeKey.isInstancesNodeKey(key)) {\r\n          result.push(...key.instanceKeys);\r\n        } else {\r\n          result.push(key);\r\n        }\r\n      } else {\r\n        result.push(key);\r\n      }\r\n    });\r\n    return result;\r\n  }\r\n}\r\n"]}

package/lib/esm/presentation-frontend/selection/SelectionManager.d.ts

@@ -4,6 +4,7 @@
 import { Id64Arg, IDisposable } from "@itwin/core-bentley";
 import { IModelConnection } from "@itwin/core-frontend";
 import { Keys, KeySet, SelectionScope, SelectionScopeProps } from "@itwin/presentation-common";
+import { SelectionStorage } from "@itwin/unified-selection";
 import { HiliteSet } from "./HiliteSetProvider";
 import { ISelectionProvider } from "./ISelectionProvider";
 import { SelectionChangeEvent } from "./SelectionChangeEvent";
@@ -15,15 +16,26 @@ import { SelectionScopesManager } from "./SelectionScopesManager";
 export interface SelectionManagerProps {
     /** A manager for [selection scopes]($docs/presentation/unified-selection/index#selection-scopes) */
     scopes: SelectionScopesManager;
+    /**
+     * Custom unified selection storage to be used by [[SelectionManager]]. If not provided [[SelectionManager]] creates
+     * and maintains storage.
+     */
+    selectionStorage?: SelectionStorage;
 }
 /**
  * The selection manager which stores the overall selection.
  * @public
  */
 export declare class SelectionManager implements ISelectionProvider {
-    private _selectionContainerMap;
+    private _selectionStorage;
     private _imodelToolSelectionSyncHandlers;
     private _hiliteSetProviders;
+    private _ownsStorage;
+    private _knownIModels;
+    private _currentSelection;
+    private _selectionChanges;
+    private _selectionEventsSubscription;
+    private _listeners;
     /** An event which gets broadcasted on selection changes */
     readonly selectionChange: SelectionChangeEvent;
     /** Manager for [selection scopes]($docs/presentation/unified-selection/index#selection-scopes) */
@@ -32,8 +44,8 @@ export declare class SelectionManager implements ISelectionProvider {
      * Creates an instance of SelectionManager.
      */
     constructor(props: SelectionManagerProps);
+    dispose(): void;
     private onConnectionClose;
-    private getContainer;
     /** @internal */
     getToolSelectionSyncHandler(imodel: IModelConnection): ToolSelectionSyncHandler | undefined;
     /**
@@ -47,7 +59,13 @@ export declare class SelectionManager implements ISelectionProvider {
     suspendIModelToolSelectionSync(imodel: IModelConnection): IDisposable;
     /** Get the selection levels currently stored in this manager for the specified imodel */
     getSelectionLevels(imodel: IModelConnection): number[];
-    /** Get the selection currently stored in this manager */
+    /**
+     * Get the selection currently stored in this manager
+     *
+     * @note Calling immediately after `add*`|`replace*`|`remove*`|`clear*` method call does not guarantee
+     * that returned `KeySet` will include latest changes. Listen for `selectionChange` event to get the
+     * latest selection after changes.
+     */
     getSelection(imodel: IModelConnection, level?: number): Readonly<KeySet>;
     private handleEvent;
     /**
@@ -126,6 +144,7 @@ export declare class SelectionManager implements ISelectionProvider {
      */
     getHiliteSetIterator(imodel: IModelConnection): AsyncIterableIterator<HiliteSet>;
     private getHiliteSetProvider;
+    private streamSelectionEvents;
 }
 /** @internal */
 export declare const TRANSIENT_ELEMENT_CLASSNAME = "/TRANSIENT";

package/lib/esm/presentation-frontend/selection/SelectionManager.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionManager.d.ts","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionManager.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAQ,OAAO,EAAa,WAAW,EAAS,MAAM,qBAAqB,CAAC;AACnF,OAAO,EAAE,gBAAgB,EAA4C,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAqB,IAAI,EAAE,MAAM,EAAE,cAAc,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAClH,OAAO,EAAE,SAAS,EAAqB,MAAM,qBAAqB,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAiD,MAAM,wBAAwB,CAAC;AAC7G,OAAO,EAA6B,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAE7F;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,oGAAoG;IACpG,MAAM,EAAE,sBAAsB,CAAC;CAChC;AAED;;;GAGG;AACH,qBAAa,gBAAiB,YAAW,kBAAkB;IACzD,OAAO,CAAC,sBAAsB,CAAmD;IACjF,OAAO,CAAC,gCAAgC,CAA+F;IACvI,OAAO,CAAC,mBAAmB,CAAkD;IAE7E,2DAA2D;IAC3D,SAAgB,eAAe,EAAE,oBAAoB,CAAC;IAEtD,kGAAkG;IAClG,SAAgB,MAAM,EAAE,sBAAsB,CAAC;IAE/C;;OAEG;gBACS,KAAK,EAAE,qBAAqB;IAQxC,OAAO,CAAC,iBAAiB;IAMzB,OAAO,CAAC,YAAY;IASpB,gBAAgB;IAET,2BAA2B,CAAC,MAAM,EAAE,gBAAgB;IAE3D;;OAEG;IACI,8BAA8B,CAAC,MAAM,EAAE,gBAAgB,EAAE,IAAI,UAAO;IAqB3E;;;OAGG;IACI,8BAA8B,CAAC,MAAM,EAAE,gBAAgB,GAAG,WAAW;IAU5E,yFAAyF;IAClF,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,GAAG,MAAM,EAAE;IAI7D,yDAAyD;IAClD,YAAY,CAAC,MAAM,EAAE,gBAAgB,EAAE,KAAK,GAAE,MAAU,GAAG,QAAQ,CAAC,MAAM,CAAC;IAIlF,OAAO,CAAC,WAAW;IA8BnB;;;;;;;OAOG;IACI,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAaxH;;;;;;;OAOG;IACI,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa7H;;;;;;;OAOG;IACI,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa1H;;;;;;OAMG;IACI,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa5G;;;;;;;;OAQG;IACU,uBAAuB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKxM;;;;;;;;OAQG;IACU,4BAA4B,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK7M;;;;;;;;OAQG;IACU,yBAAyB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,GAAG,EAAE,OAAO,EAAE,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAK1M;;;OAGG;IACU,YAAY,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC;IAIvE;;;OAGG;IACI,oBAAoB,CAAC,MAAM,EAAE,gBAAgB;IAIpD,OAAO,CAAC,oBAAoB;CAQ7B;AAuCD,gBAAgB;AAChB,eAAO,MAAM,2BAA2B,eAAe,CAAC;AAExD,gBAAgB;AAChB,qBAAa,wBAAyB,YAAW,WAAW;IAE1D,OAAO,CAAC,oBAAoB,CAAU;IACtC,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,OAAO,CAAmB;IAClC,OAAO,CAAC,uCAAuC,CAAa;IAC5D,OAAO,CAAC,cAAc,CAA2B;IAC1C,WAAW,CAAC,EAAE,OAAO,CAAC;gBAEV,MAAM,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB;IAMxE,OAAO;IAId,+BAA+B;IAC/B,IAAW,aAAa,gBAAgD;IAGxE,OAAO,CAAC,sBAAsB,CAqD5B;CACH"}
+{"version":3,"file":"SelectionManager.d.ts","sourceRoot":"","sources":["../../../../src/presentation-frontend/selection/SelectionManager.ts"],"names":[],"mappings":"AAIA;;GAEG;AAGH,OAAO,EAAQ,OAAO,EAAa,WAAW,EAAS,MAAM,qBAAqB,CAAC;AACnF,OAAO,EAAE,gBAAgB,EAA4C,MAAM,sBAAsB,CAAC;AAClG,OAAO,EAAoD,IAAI,EAAE,MAAM,EAAW,cAAc,EAAE,mBAAmB,EAAE,MAAM,4BAA4B,CAAC;AAC1J,OAAO,EAKL,gBAAgB,EAGjB,MAAM,0BAA0B,CAAC;AAElC,OAAO,EAAE,SAAS,EAAqB,MAAM,qBAAqB,CAAC;AACnE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAiD,MAAM,wBAAwB,CAAC;AAC7G,OAAO,EAA6B,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAE7F;;;GAGG;AACH,MAAM,WAAW,qBAAqB;IACpC,oGAAoG;IACpG,MAAM,EAAE,sBAAsB,CAAC;IAC/B;;;OAGG;IACH,gBAAgB,CAAC,EAAE,gBAAgB,CAAC;CACrC;AAED;;;GAGG;AACH,qBAAa,gBAAiB,YAAW,kBAAkB;IACzD,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,gCAAgC,CAA+F;IACvI,OAAO,CAAC,mBAAmB,CAAkD;IAC7E,OAAO,CAAC,YAAY,CAAU;IAE9B,OAAO,CAAC,aAAa,CAAuC;IAC5D,OAAO,CAAC,iBAAiB,CAAiC;IAC1D,OAAO,CAAC,iBAAiB,CAAkD;IAC3E,OAAO,CAAC,4BAA4B,CAAe;IACnD,OAAO,CAAC,UAAU,CAAyB;IAE3C,2DAA2D;IAC3D,SAAgB,eAAe,EAAE,oBAAoB,CAAC;IAEtD,kGAAkG;IAClG,SAAgB,MAAM,EAAE,sBAAsB,CAAC;IAE/C;;OAEG;gBACS,KAAK,EAAE,qBAAqB;IAmBjC,OAAO;IAKd,OAAO,CAAC,iBAAiB;IAUzB,gBAAgB;IAET,2BAA2B,CAAC,MAAM,EAAE,gBAAgB;IAI3D;;OAEG;IACI,8BAA8B,CAAC,MAAM,EAAE,gBAAgB,EAAE,IAAI,UAAO;IAqB3E;;;OAGG;IACI,8BAA8B,CAAC,MAAM,EAAE,gBAAgB,GAAG,WAAW;IAW5E,yFAAyF;IAClF,kBAAkB,CAAC,MAAM,EAAE,gBAAgB,GAAG,MAAM,EAAE;IAI7D;;;;;;OAMG;IACI,YAAY,CAAC,MAAM,EAAE,gBAAgB,EAAE,KAAK,GAAE,MAAU,GAAG,QAAQ,CAAC,MAAM,CAAC;IAIlF,OAAO,CAAC,WAAW;IAgCnB;;;;;;;OAOG;IACI,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAaxH;;;;;;;OAOG;IACI,mBAAmB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa7H;;;;;;;OAOG;IACI,gBAAgB,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa1H;;;;;;OAMG;IACI,cAAc,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,EAAE,gBAAgB,EAAE,KAAK,GAAE,MAAU,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,IAAI;IAa5G;;;;;;;;OAQG;IACU,uBAAuB,CAClC,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,gBAAgB,EACxB,GAAG,EAAE,OAAO,EACZ,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EACpD,KAAK,GAAE,MAAU,EACjB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAKhB;;;;;;;;OAQG;IACU,4BAA4B,CACvC,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,gBAAgB,EACxB,GAAG,EAAE,OAAO,EACZ,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EACpD,KAAK,GAAE,MAAU,EACjB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAKhB;;;;;;;;OAQG;IACU,yBAAyB,CACpC,MAAM,EAAE,MAAM,EACd,MAAM,EAAE,gBAAgB,EACxB,GAAG,EAAE,OAAO,EACZ,KAAK,EAAE,mBAAmB,GAAG,cAAc,GAAG,MAAM,EACpD,KAAK,GAAE,MAAU,EACjB,SAAS,CAAC,EAAE,MAAM,GACjB,OAAO,CAAC,IAAI,CAAC;IAKhB;;;OAGG;IACU,YAAY,CAAC,MAAM,EAAE,gBAAgB,GAAG,OAAO,CAAC,SAAS,CAAC;IAIvE;;;OAGG;IACI,oBAAoB,CAAC,MAAM,EAAE,gBAAgB;IAIpD,OAAO,CAAC,oBAAoB;IAS5B,OAAO,CAAC,qBAAqB;CA6B9B;AAED,gBAAgB;AAChB,eAAO,MAAM,2BAA2B,eAAe,CAAC;AAExD,gBAAgB;AAChB,qBAAa,wBAAyB,YAAW,WAAW;IAC1D,OAAO,CAAC,oBAAoB,CAAU;IACtC,OAAO,CAAC,iBAAiB,CAAmB;IAC5C,OAAO,CAAC,OAAO,CAAmB;IAClC,OAAO,CAAC,uCAAuC,CAAa;IAC5D,OAAO,CAAC,cAAc,CAA2B;IAC1C,WAAW,CAAC,EAAE,OAAO,CAAC;gBAEV,MAAM,EAAE,gBAAgB,EAAE,gBAAgB,EAAE,gBAAgB;IAMxE,OAAO;IAId,+BAA+B;IAC/B,IAAW,aAAa,gBAEvB;IAGD,OAAO,CAAC,sBAAsB,CA4D5B;CACH"}

package/lib/esm/presentation-frontend/selection/SelectionManager.js

@@ -1,13 +1,16 @@
 /*---------------------------------------------------------------------------------------------
-* Copyright (c) Bentley Systems, Incorporated. All rights reserved.
-* See LICENSE.md in the project root for license terms and full copyright notice.
-*--------------------------------------------------------------------------------------------*/
+ * 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 { defer, EMPTY, mergeMap, of, Subject, takeUntil, tap } from "rxjs";
 import { Id64, using } from "@itwin/core-bentley";
 import { IModelConnection, SelectionSetEventType } from "@itwin/core-frontend";
-import { AsyncTasksTracker, KeySet } from "@itwin/presentation-common";
+import { AsyncTasksTracker, KeySet, NodeKey } from "@itwin/presentation-common";
+import { createStorage, } from "@itwin/unified-selection";
+import { Presentation } from "../Presentation";
 import { HiliteSetProvider } from "./HiliteSetProvider";
 import { SelectionChangeEvent, SelectionChangeType } from "./SelectionChangeEvent";
 import { createSelectionScopeProps } from "./SelectionScopesManager";
@@ -20,31 +23,43 @@ export class SelectionManager {
      * Creates an instance of SelectionManager.
      */
     constructor(props) {
-        this._selectionContainerMap = new Map();
         this._imodelToolSelectionSyncHandlers = new Map();
         this._hiliteSetProviders = new Map();
+        this._knownIModels = new Map();
+        this._currentSelection = new CurrentSelectionStorage();
+        this._selectionChanges = new Subject();
+        this._listeners = [];
         this.selectionChange = new SelectionChangeEvent();
         this.scopes = props.scopes;
-        IModelConnection.onClose.addListener((imodel) => {
+        this._selectionStorage = props.selectionStorage ?? createStorage();
+        this._ownsStorage = props.selectionStorage === undefined;
+        this._selectionStorage.selectionChangeEvent.addListener((args) => this._selectionChanges.next(args));
+        this._selectionEventsSubscription = this.streamSelectionEvents();
+        this._listeners.push(IModelConnection.onOpen.addListener((imodel) => {
+            this._knownIModels.set(imodel.key, imodel);
+        }));
+        this._listeners.push(IModelConnection.onClose.addListener((imodel) => {
             this.onConnectionClose(imodel);
-        });
+        }));
+    }
+    dispose() {
+        this._selectionEventsSubscription.unsubscribe();
+        this._listeners.forEach((dispose) => dispose());
     }
     onConnectionClose(imodel) {
-        this.clearSelection("Connection Close Event", imodel);
-        this._selectionContainerMap.delete(imodel);
         this._hiliteSetProviders.delete(imodel);
-    }
-    getContainer(imodel) {
-        let selectionContainer = this._selectionContainerMap.get(imodel);
-        if (!selectionContainer) {
-            selectionContainer = new SelectionContainer();
-            this._selectionContainerMap.set(imodel, selectionContainer);
+        this._knownIModels.delete(imodel.key);
+        this._currentSelection.clear(imodel.key);
+        if (this._ownsStorage) {
+            this.clearSelection("Connection Close Event", imodel);
+            this._selectionStorage.clearStorage({ iModelKey: imodel.key });
         }
-        return selectionContainer;
     }
     /** @internal */
     // istanbul ignore next
-    getToolSelectionSyncHandler(imodel) { return this._imodelToolSelectionSyncHandlers.get(imodel)?.handler; }
+    getToolSelectionSyncHandler(imodel) {
+        return this._imodelToolSelectionSyncHandlers.get(imodel)?.handler;
+    }
     /**
      * Request the manager to sync with imodel's tool selection (see `IModelConnection.selectionSet`).
      */
@@ -77,46 +92,57 @@ export class SelectionManager {
      */
     suspendIModelToolSelectionSync(imodel) {
         const registration = this._imodelToolSelectionSyncHandlers.get(imodel);
-        if (!registration)
+        if (!registration) {
             return { dispose: () => { } };
+        }
         const wasSuspended = registration.handler.isSuspended;
         registration.handler.isSuspended = true;
         return { dispose: () => (registration.handler.isSuspended = wasSuspended) };
     }
     /** Get the selection levels currently stored in this manager for the specified imodel */
     getSelectionLevels(imodel) {
-        return this.getContainer(imodel).getSelectionLevels();
+        return this._selectionStorage.getSelectionLevels({ iModelKey: imodel.key });
     }
-    /** Get the selection currently stored in this manager */
+    /**
+     * Get the selection currently stored in this manager
+     *
+     * @note Calling immediately after `add*`|`replace*`|`remove*`|`clear*` method call does not guarantee
+     * that returned `KeySet` will include latest changes. Listen for `selectionChange` event to get the
+     * latest selection after changes.
+     */
     getSelection(imodel, level = 0) {
-        return this.getContainer(imodel).getSelection(level);
+        return this._currentSelection.getSelection(imodel.key, level);
     }
     handleEvent(evt) {
-        const container = this.getContainer(evt.imodel);
-        const selectedItemsSet = container.getSelection(evt.level);
-        const guidBefore = selectedItemsSet.guid;
         switch (evt.changeType) {
             case SelectionChangeType.Add:
-                selectedItemsSet.add(evt.keys);
+                this._selectionStorage.addToSelection({
+                    iModelKey: evt.imodel.key,
+                    source: evt.source,
+                    level: evt.level,
+                    selectables: keysToSelectable(evt.imodel, evt.keys),
+                });
                 break;
             case SelectionChangeType.Remove:
-                selectedItemsSet.delete(evt.keys);
+                this._selectionStorage.removeFromSelection({
+                    iModelKey: evt.imodel.key,
+                    source: evt.source,
+                    level: evt.level,
+                    selectables: keysToSelectable(evt.imodel, evt.keys),
+                });
                 break;
             case SelectionChangeType.Replace:
-                if (selectedItemsSet.size !== evt.keys.size || !selectedItemsSet.hasAll(evt.keys)) {
-                    // note: the above check is only needed to avoid changing
-                    // guid of the keyset if we're replacing keyset with the same keys
-                    selectedItemsSet.clear().add(evt.keys);
-                }
+                this._selectionStorage.replaceSelection({
+                    iModelKey: evt.imodel.key,
+                    source: evt.source,
+                    level: evt.level,
+                    selectables: keysToSelectable(evt.imodel, evt.keys),
+                });
                 break;
             case SelectionChangeType.Clear:
-                selectedItemsSet.clear();
+                this._selectionStorage.clearSelection({ iModelKey: evt.imodel.key, source: evt.source, level: evt.level });
                 break;
         }
-        if (selectedItemsSet.guid === guidBefore)
-            return;
-        container.clear(evt.level + 1);
-        this.selectionChange.raiseEvent(evt, this);
     }
     /**
      * Add keys to the selection
@@ -258,36 +284,30 @@ export class SelectionManager {
         }
         return provider;
     }
-}
-/** @internal */
-class SelectionContainer {
-    constructor() {
-        this._selectedItemsSetMap = new Map();
-    }
-    getSelection(level) {
-        let selectedItemsSet = this._selectedItemsSetMap.get(level);
-        if (!selectedItemsSet) {
-            selectedItemsSet = new KeySet();
-            this._selectedItemsSetMap.set(level, selectedItemsSet);
-        }
-        return selectedItemsSet;
-    }
-    getSelectionLevels() {
-        const levels = new Array();
-        for (const entry of this._selectedItemsSetMap.entries()) {
-            if (!entry[1].isEmpty)
-                levels.push(entry[0]);
-        }
-        return levels.sort();
-    }
-    clear(level) {
-        const keys = this._selectedItemsSetMap.keys();
-        for (const key of keys) {
-            if (key >= level) {
-                const selectedItemsSet = this._selectedItemsSetMap.get(key);
-                selectedItemsSet.clear();
-            }
-        }
+    streamSelectionEvents() {
+        return this._selectionChanges
+            .pipe(mergeMap((args) => {
+            const currentSelectables = this._selectionStorage.getSelection({ iModelKey: args.iModelKey, level: args.level });
+            return this._currentSelection.computeSelection(args.iModelKey, args.level, currentSelectables, args.selectables).pipe(mergeMap(({ level, changedSelection }) => {
+                const imodel = this._knownIModels.get(args.iModelKey);
+                if (!imodel) {
+                    return EMPTY;
+                }
+                return of({
+                    imodel,
+                    keys: changedSelection,
+                    level,
+                    source: args.source,
+                    timestamp: args.timestamp,
+                    changeType: getChangeType(args.changeType),
+                });
+            }));
+        }))
+            .subscribe({
+            next: (args) => {
+                this.selectionChange.raiseEvent(args, this);
+            },
+        });
     }
 }
 /** @internal */
@@ -300,12 +320,14 @@ export class ToolSelectionSyncHandler {
         // eslint-disable-next-line @typescript-eslint/naming-convention
         this.onToolSelectionChanged = async (ev) => {
             // ignore selection change event if the handler is suspended
-            if (this.isSuspended)
+            if (this.isSuspended) {
                 return;
+            }
             // this component only cares about its own imodel
             const imodel = ev.set.iModel;
-            if (imodel !== this._imodel)
+            if (imodel !== this._imodel) {
                 return;
+            }
             // determine the level of selection changes
             // wip: may want to allow selecting at different levels?
             const selectionLevel = 0;
@@ -354,18 +376,23 @@ export class ToolSelectionSyncHandler {
         this._imodelToolSelectionListenerDisposeFunc();
     }
     /** note: used only it tests */
-    get pendingAsyncs() { return this._asyncsTracker.pendingAsyncs; }
+    get pendingAsyncs() {
+        return this._asyncsTracker.pendingAsyncs;
+    }
 }
 const parseIds = (ids) => {
     let allPersistent = true;
     let allTransient = true;
     for (const id of Id64.iterable(ids)) {
-        if (Id64.isTransient(id))
+        if (Id64.isTransient(id)) {
             allPersistent = false;
-        else
+        }
+        else {
             allTransient = false;
-        if (!allPersistent && !allTransient)
+        }
+        if (!allPersistent && !allTransient) {
             break;
+        }
     }
     // avoid making a copy if ids are only persistent or only transient
     if (allPersistent) {
@@ -379,16 +406,19 @@ const parseIds = (ids) => {
     const persistentElementIds = [];
     const transientElementIds = [];
     for (const id of Id64.iterable(ids)) {
-        if (Id64.isTransient(id))
+        if (Id64.isTransient(id)) {
             transientElementIds.push(id);
-        else
+        }
+        else {
             persistentElementIds.push(id);
+        }
     }
     return { persistent: persistentElementIds, transient: transientElementIds };
 };
 function addTransientKeys(transientIds, keys) {
-    for (const id of Id64.iterable(transientIds))
+    for (const id of Id64.iterable(transientIds)) {
         keys.add({ className: TRANSIENT_ELEMENT_CLASSNAME, id });
+    }
 }
 /** @internal */
 class ScopedSelectionChanger {
@@ -417,4 +447,194 @@ class ScopedSelectionChanger {
         this.manager.replaceSelection(this.name, this.imodel, keys, level);
     }
 }
+/** Stores current selection in `KeySet` format per iModel.  */
+class CurrentSelectionStorage {
+    constructor() {
+        this._currentSelection = new Map();
+    }
+    getCurrentSelectionStorage(imodelKey) {
+        let storage = this._currentSelection.get(imodelKey);
+        if (!storage) {
+            storage = new IModelSelectionStorage();
+            this._currentSelection.set(imodelKey, storage);
+        }
+        return storage;
+    }
+    getSelection(imodelKey, level) {
+        return this.getCurrentSelectionStorage(imodelKey).getSelection(level);
+    }
+    clear(imodelKey) {
+        this._currentSelection.delete(imodelKey);
+    }
+    computeSelection(imodelKey, level, currSelectables, changedSelectables) {
+        return this.getCurrentSelectionStorage(imodelKey).computeSelection(level, currSelectables, changedSelectables);
+    }
+}
+/**
+ * Computes and stores current selection in `KeySet` format.
+ * It always stores result of latest resolved call to `computeSelection`.
+ */
+class IModelSelectionStorage {
+    constructor() {
+        this._currentSelection = new Map();
+    }
+    getSelection(level) {
+        let entry = this._currentSelection.get(level);
+        if (!entry) {
+            entry = { value: new KeySet(), ongoingComputationDisposers: new Set() };
+            this._currentSelection.set(level, entry);
+        }
+        return entry.value;
+    }
+    clearSelections(level) {
+        const clearedLevels = [];
+        for (const [storedLevel] of this._currentSelection.entries()) {
+            if (storedLevel > level) {
+                clearedLevels.push(storedLevel);
+            }
+        }
+        clearedLevels.forEach((storedLevel) => {
+            const entry = this._currentSelection.get(storedLevel);
+            // istanbul ignore if
+            if (!entry) {
+                return;
+            }
+            for (const disposer of entry.ongoingComputationDisposers) {
+                disposer.next();
+            }
+            this._currentSelection.delete(storedLevel);
+        });
+    }
+    addDisposer(level, disposer) {
+        const entry = this._currentSelection.get(level);
+        if (!entry) {
+            this._currentSelection.set(level, { value: new KeySet(), ongoingComputationDisposers: new Set([disposer]) });
+            return;
+        }
+        entry.ongoingComputationDisposers.add(disposer);
+    }
+    setSelection(level, keys, disposer) {
+        const currEntry = this._currentSelection.get(level);
+        // istanbul ignore else
+        if (currEntry) {
+            currEntry.ongoingComputationDisposers.delete(disposer);
+        }
+        this._currentSelection.set(level, {
+            value: keys,
+            ongoingComputationDisposers: currEntry?.ongoingComputationDisposers ?? /* istanbul ignore next */ new Set(),
+        });
+    }
+    computeSelection(level, currSelectables, changedSelectables) {
+        this.clearSelections(level);
+        const prevComputationsDisposers = [...(this._currentSelection.get(level)?.ongoingComputationDisposers ?? [])];
+        const currDisposer = new Subject();
+        this.addDisposer(level, currDisposer);
+        return defer(async () => {
+            const convertedSelectables = [];
+            const [current, changed] = await Promise.all([
+                selectablesToKeys(currSelectables, convertedSelectables),
+                selectablesToKeys(changedSelectables, convertedSelectables),
+            ]);
+            const currentSelection = new KeySet([...current.keys, ...current.selectableKeys.flatMap((selectable) => selectable.keys)]);
+            const changedSelection = new KeySet([...changed.keys, ...changed.selectableKeys.flatMap((selectable) => selectable.keys)]);
+            return {
+                level,
+                currentSelection,
+                changedSelection,
+            };
+        }).pipe(takeUntil(currDisposer), tap({
+            next: (val) => {
+                prevComputationsDisposers.forEach((disposer) => disposer.next());
+                this.setSelection(val.level, val.currentSelection, currDisposer);
+            },
+        }));
+    }
+}
+function keysToSelectable(imodel, keys) {
+    const selectables = [];
+    keys.forEach((key) => {
+        if ("id" in key) {
+            selectables.push(key);
+            return;
+        }
+        const customSelectable = {
+            identifier: key.pathFromRoot.join("/"),
+            data: key,
+            loadInstanceKeys: () => createInstanceKeysIterator(imodel, key),
+        };
+        selectables.push(customSelectable);
+    });
+    return selectables;
+}
+async function selectablesToKeys(selectables, convertedList) {
+    const keys = [];
+    const selectableKeys = [];
+    for (const [className, ids] of selectables.instanceKeys) {
+        for (const id of ids) {
+            keys.push({ id, className });
+        }
+    }
+    for (const [_, selectable] of selectables.custom) {
+        if (isNodeKey(selectable.data)) {
+            selectableKeys.push({ identifier: selectable.identifier, keys: [selectable.data] });
+            continue;
+        }
+        const converted = convertedList.find((con) => con.identifier === selectable.identifier);
+        if (converted) {
+            selectableKeys.push(converted);
+            continue;
+        }
+        const newConverted = { identifier: selectable.identifier, keys: [] };
+        convertedList.push(newConverted);
+        for await (const instanceKey of selectable.loadInstanceKeys()) {
+            newConverted.keys.push(instanceKey);
+        }
+        selectableKeys.push(newConverted);
+    }
+    return { keys, selectableKeys };
+}
+async function* createInstanceKeysIterator(imodel, nodeKey) {
+    if (NodeKey.isInstancesNodeKey(nodeKey)) {
+        for (const key of nodeKey.instanceKeys) {
+            yield key;
+        }
+        return;
+    }
+    const content = await Presentation.presentation.getContentInstanceKeys({
+        imodel,
+        keys: new KeySet([nodeKey]),
+        rulesetOrId: {
+            id: "grouped-instances",
+            rules: [
+                {
+                    ruleType: "Content",
+                    specifications: [
+                        {
+                            specType: "SelectedNodeInstances",
+                        },
+                    ],
+                },
+            ],
+        },
+    });
+    for await (const key of content.items()) {
+        yield key;
+    }
+}
+function isNodeKey(data) {
+    const key = data;
+    return key.pathFromRoot !== undefined && key.type !== undefined;
+}
+function getChangeType(type) {
+    switch (type) {
+        case "add":
+            return SelectionChangeType.Add;
+        case "remove":
+            return SelectionChangeType.Remove;
+        case "replace":
+            return SelectionChangeType.Replace;
+        case "clear":
+            return SelectionChangeType.Clear;
+    }
+}
 //# sourceMappingURL=SelectionManager.js.map