@itwin/presentation-frontend 5.0.0-dev.5 → 5.0.0-dev.50

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

@@ -6,6 +6,58 @@
 /** @packageDocumentation
  * @module UnifiedSelection
  */
+var __addDisposableResource = (this && this.__addDisposableResource) || function (env, value, async) {
+    if (value !== null && value !== void 0) {
+        if (typeof value !== "object" && typeof value !== "function") throw new TypeError("Object expected.");
+        var dispose, inner;
+        if (async) {
+            if (!Symbol.asyncDispose) throw new TypeError("Symbol.asyncDispose is not defined.");
+            dispose = value[Symbol.asyncDispose];
+        }
+        if (dispose === void 0) {
+            if (!Symbol.dispose) throw new TypeError("Symbol.dispose is not defined.");
+            dispose = value[Symbol.dispose];
+            if (async) inner = dispose;
+        }
+        if (typeof dispose !== "function") throw new TypeError("Object not disposable.");
+        if (inner) dispose = function() { try { inner.call(this); } catch (e) { return Promise.reject(e); } };
+        env.stack.push({ value: value, dispose: dispose, async: async });
+    }
+    else if (async) {
+        env.stack.push({ async: true });
+    }
+    return value;
+};
+var __disposeResources = (this && this.__disposeResources) || (function (SuppressedError) {
+    return function (env) {
+        function fail(e) {
+            env.error = env.hasError ? new SuppressedError(e, env.error, "An error was suppressed during disposal.") : e;
+            env.hasError = true;
+        }
+        var r, s = 0;
+        function next() {
+            while (r = env.stack.pop()) {
+                try {
+                    if (!r.async && s === 1) return s = 0, env.stack.push(r), Promise.resolve().then(next);
+                    if (r.dispose) {
+                        var result = r.dispose.call(r.value);
+                        if (r.async) return s |= 2, Promise.resolve(result).then(next, function(e) { fail(e); return next(); });
+                    }
+                    else s |= 1;
+                }
+                catch (e) {
+                    fail(e);
+                }
+            }
+            if (s === 1) return env.hasError ? Promise.reject(env.error) : Promise.resolve();
+            if (env.hasError) throw env.error;
+        }
+        return next();
+    };
+})(typeof SuppressedError === "function" ? SuppressedError : function (error, suppressed, message) {
+    var e = new Error(message);
+    return e.name = "SuppressedError", e.error = error, e.suppressed = suppressed, e;
+});
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.ToolSelectionSyncHandler = exports.SelectionManager = void 0;
 const rxjs_1 = require("rxjs");
@@ -28,34 +80,41 @@ class SelectionManager {
     constructor(props) {
         this._imodelToolSelectionSyncHandlers = new Map();
         this._hiliteSetProviders = new Map();
-        this._knownIModels = new Map();
+        this._knownIModels = new Set();
         this._currentSelection = new CurrentSelectionStorage();
         this._selectionChanges = new rxjs_1.Subject();
         this._listeners = [];
         this.selectionChange = new SelectionChangeEvent_1.SelectionChangeEvent();
         this.scopes = props.scopes;
-        this._selectionStorage = props.selectionStorage ?? (0, unified_selection_1.createStorage)();
+        this.selectionStorage = props.selectionStorage ?? (0, unified_selection_1.createStorage)();
+        this._imodelKeyFactory = props.imodelKeyFactory ?? ((imodel) => (imodel.key.length ? imodel.key : imodel.name));
         this._ownsStorage = props.selectionStorage === undefined;
-        this._selectionStorage.selectionChangeEvent.addListener((args) => this._selectionChanges.next(args));
+        this.selectionStorage.selectionChangeEvent.addListener((args) => this._selectionChanges.next(args));
         this._selectionEventsSubscription = this.streamSelectionEvents();
         this._listeners.push(core_frontend_1.IModelConnection.onOpen.addListener((imodel) => {
-            this._knownIModels.set(imodel.key, imodel);
+            this._knownIModels.add(imodel);
         }));
         this._listeners.push(core_frontend_1.IModelConnection.onClose.addListener((imodel) => {
             this.onConnectionClose(imodel);
         }));
     }
-    dispose() {
+    [Symbol.dispose]() {
         this._selectionEventsSubscription.unsubscribe();
         this._listeners.forEach((dispose) => dispose());
     }
+    /** @deprecated in 5.0. Use [Symbol.dispose] instead. */
+    // istanbul ignore next
+    dispose() {
+        this[Symbol.dispose]();
+    }
     onConnectionClose(imodel) {
+        const imodelKey = this._imodelKeyFactory(imodel);
         this._hiliteSetProviders.delete(imodel);
-        this._knownIModels.delete(imodel.key);
-        this._currentSelection.clear(imodel.key);
+        this._knownIModels.delete(imodel);
+        this._currentSelection.clear(imodelKey);
         if (this._ownsStorage) {
             this.clearSelection("Connection Close Event", imodel);
-            this._selectionStorage.clearStorage({ iModelKey: imodel.key });
+            this.selectionStorage.clearStorage({ imodelKey });
         }
     }
     /** @internal */
@@ -84,27 +143,30 @@ class SelectionManager {
                 }
                 else {
                     this._imodelToolSelectionSyncHandlers.delete(imodel);
-                    registration.handler.dispose();
+                    registration.handler[Symbol.dispose]();
                 }
             }
         }
     }
     /**
-     * Temporarily suspends tool selection synchronization until the returned `IDisposable`
+     * Temporarily suspends tool selection synchronization until the returned `Disposable`
      * is disposed.
      */
     suspendIModelToolSelectionSync(imodel) {
         const registration = this._imodelToolSelectionSyncHandlers.get(imodel);
         if (!registration) {
-            return { dispose: () => { } };
+            const noop = () => { };
+            return { [Symbol.dispose]: noop, dispose: noop };
         }
         const wasSuspended = registration.handler.isSuspended;
         registration.handler.isSuspended = true;
-        return { dispose: () => (registration.handler.isSuspended = wasSuspended) };
+        const doDispose = () => (registration.handler.isSuspended = wasSuspended);
+        return { [Symbol.dispose]: doDispose, dispose: doDispose };
     }
     /** Get the selection levels currently stored in this manager for the specified imodel */
     getSelectionLevels(imodel) {
-        return this._selectionStorage.getSelectionLevels({ iModelKey: imodel.key });
+        const imodelKey = this._imodelKeyFactory(imodel);
+        return this.selectionStorage.getSelectionLevels({ imodelKey });
     }
     /**
      * Get the selection currently stored in this manager
@@ -114,39 +176,39 @@ class SelectionManager {
      * latest selection after changes.
      */
     getSelection(imodel, level = 0) {
-        return this._currentSelection.getSelection(imodel.key, level);
+        const imodelKey = this._imodelKeyFactory(imodel);
+        return this._currentSelection.getSelection(imodelKey, level);
     }
     handleEvent(evt) {
-        if (!this._knownIModels.has(evt.imodel.key)) {
-            this._knownIModels.set(evt.imodel.key, evt.imodel);
-        }
+        const imodelKey = this._imodelKeyFactory(evt.imodel);
+        this._knownIModels.add(evt.imodel);
         switch (evt.changeType) {
             case SelectionChangeEvent_1.SelectionChangeType.Add:
-                this._selectionStorage.addToSelection({
-                    iModelKey: evt.imodel.key,
+                this.selectionStorage.addToSelection({
+                    imodelKey,
                     source: evt.source,
                     level: evt.level,
                     selectables: keysToSelectable(evt.imodel, evt.keys),
                 });
                 break;
             case SelectionChangeEvent_1.SelectionChangeType.Remove:
-                this._selectionStorage.removeFromSelection({
-                    iModelKey: evt.imodel.key,
+                this.selectionStorage.removeFromSelection({
+                    imodelKey,
                     source: evt.source,
                     level: evt.level,
                     selectables: keysToSelectable(evt.imodel, evt.keys),
                 });
                 break;
             case SelectionChangeEvent_1.SelectionChangeType.Replace:
-                this._selectionStorage.replaceSelection({
-                    iModelKey: evt.imodel.key,
+                this.selectionStorage.replaceSelection({
+                    imodelKey,
                     source: evt.source,
                     level: evt.level,
                     selectables: keysToSelectable(evt.imodel, evt.keys),
                 });
                 break;
             case SelectionChangeEvent_1.SelectionChangeType.Clear:
-                this._selectionStorage.clearSelection({ iModelKey: evt.imodel.key, source: evt.source, level: evt.level });
+                this.selectionStorage.clearSelection({ imodelKey, source: evt.source, level: evt.level });
                 break;
         }
     }
@@ -293,9 +355,9 @@ class SelectionManager {
     streamSelectionEvents() {
         return this._selectionChanges
             .pipe((0, rxjs_1.mergeMap)((args) => {
-            const currentSelectables = this._selectionStorage.getSelection({ iModelKey: args.imodelKey, level: args.level });
+            const currentSelectables = this.selectionStorage.getSelection({ imodelKey: args.imodelKey, level: args.level });
             return this._currentSelection.computeSelection(args.imodelKey, args.level, currentSelectables, args.selectables).pipe((0, rxjs_1.mergeMap)(({ level, changedSelection }) => {
-                const imodel = this._knownIModels.get(args.imodelKey);
+                const imodel = findIModel(this._knownIModels, this._imodelKeyFactory, args.imodelKey);
                 // istanbul ignore if
                 if (!imodel) {
                     return rxjs_1.EMPTY;
@@ -318,66 +380,80 @@ class SelectionManager {
     }
 }
 exports.SelectionManager = SelectionManager;
+function findIModel(set, imodelKeyFactory, key) {
+    for (const imodel of set) {
+        if (imodelKeyFactory(imodel) === key) {
+            return imodel;
+        }
+    }
+    return undefined;
+}
 /** @internal */
 class ToolSelectionSyncHandler {
     constructor(imodel, logicalSelection) {
         this._selectionSourceName = "Tool";
         this._asyncsTracker = new presentation_common_1.AsyncTasksTracker();
         this.onToolSelectionChanged = async (ev) => {
-            // ignore selection change event if the handler is suspended
-            if (this.isSuspended) {
-                return;
-            }
-            // this component only cares about its own imodel
-            const imodel = ev.set.iModel;
-            if (imodel !== this._imodel) {
-                return;
-            }
-            // determine the level of selection changes
-            // wip: may want to allow selecting at different levels?
-            const selectionLevel = 0;
-            let ids;
-            switch (ev.type) {
-                case core_frontend_1.SelectionSetEventType.Add:
-                    ids = ev.added;
-                    break;
-                case core_frontend_1.SelectionSetEventType.Replace:
-                    ids = ev.set.elements;
-                    break;
-                default:
-                    ids = ev.removed;
-                    break;
-            }
-            // we're always using scoped selection changer even if the scope is set to "element" - that
-            // makes sure we're adding to selection keys with concrete classes and not "BisCore:Element", which
-            // we can't because otherwise our keys compare fails (presentation components load data with
-            // concrete classes)
-            const changer = new ScopedSelectionChanger(this._selectionSourceName, this._imodel, this._logicalSelection, (0, SelectionScopesManager_1.createSelectionScopeProps)(this._logicalSelection.scopes.activeScope));
-            // we know what to do immediately on `clear` events
-            if (core_frontend_1.SelectionSetEventType.Clear === ev.type) {
-                await changer.clear(selectionLevel);
-                return;
-            }
-            const parsedIds = parseIds(ids);
-            await (0, core_bentley_1.using)(this._asyncsTracker.trackAsyncTask(), async (_r) => {
+            const env_1 = { stack: [], error: void 0, hasError: false };
+            try {
+                // ignore selection change event if the handler is suspended
+                if (this.isSuspended) {
+                    return;
+                }
+                // this component only cares about its own imodel
+                const imodel = ev.set.iModel;
+                if (imodel !== this._imodel) {
+                    return;
+                }
+                // determine the level of selection changes
+                // wip: may want to allow selecting at different levels?
+                const selectionLevel = 0;
+                let ids;
+                switch (ev.type) {
+                    case core_frontend_1.SelectionSetEventType.Add:
+                        ids = ev.additions;
+                        break;
+                    case core_frontend_1.SelectionSetEventType.Replace:
+                        ids = ev.set.active;
+                        break;
+                    default:
+                        ids = ev.removals;
+                        break;
+                }
+                // we're always using scoped selection changer even if the scope is set to "element" - that
+                // makes sure we're adding to selection keys with concrete classes and not "BisCore:Element", which
+                // we can't because otherwise our keys compare fails (presentation components load data with
+                // concrete classes)
+                const changer = new ScopedSelectionChanger(this._selectionSourceName, this._imodel, this._logicalSelection, (0, SelectionScopesManager_1.createSelectionScopeProps)(this._logicalSelection.scopes.activeScope));
+                const _r = __addDisposableResource(env_1, this._asyncsTracker.trackAsyncTask(), false);
                 switch (ev.type) {
                     case core_frontend_1.SelectionSetEventType.Add:
-                        await changer.add(parsedIds.transient, parsedIds.persistent, selectionLevel);
+                        await changer.add(ids, selectionLevel);
                         break;
                     case core_frontend_1.SelectionSetEventType.Replace:
-                        await changer.replace(parsedIds.transient, parsedIds.persistent, selectionLevel);
+                        await changer.replace(ids, selectionLevel);
                         break;
                     case core_frontend_1.SelectionSetEventType.Remove:
-                        await changer.remove(parsedIds.transient, parsedIds.persistent, selectionLevel);
+                        await changer.remove(ids, selectionLevel);
+                        break;
+                    case core_frontend_1.SelectionSetEventType.Clear:
+                        await changer.clear(selectionLevel);
                         break;
                 }
-            });
+            }
+            catch (e_1) {
+                env_1.error = e_1;
+                env_1.hasError = true;
+            }
+            finally {
+                __disposeResources(env_1);
+            }
         };
         this._imodel = imodel;
         this._logicalSelection = logicalSelection;
         this._imodelToolSelectionListenerDisposeFunc = imodel.selectionSet.onChanged.addListener(this.onToolSelectionChanged);
     }
-    dispose() {
+    [Symbol.dispose]() {
         this._imodelToolSelectionListenerDisposeFunc();
     }
     /** note: used only it tests */
@@ -386,7 +462,7 @@ class ToolSelectionSyncHandler {
     }
 }
 exports.ToolSelectionSyncHandler = ToolSelectionSyncHandler;
-const parseIds = (ids) => {
+const parseElementIds = (ids) => {
     let allPersistent = true;
     let allTransient = true;
     for (const id of core_bentley_1.Id64.iterable(ids)) {
@@ -421,12 +497,11 @@ const parseIds = (ids) => {
     }
     return { persistent: persistentElementIds, transient: transientElementIds };
 };
-function addTransientKeys(transientIds, keys) {
-    for (const id of core_bentley_1.Id64.iterable(transientIds)) {
-        keys.add({ className: unified_selection_1.TRANSIENT_ELEMENT_CLASSNAME, id });
+function addKeys(target, className, ids) {
+    for (const id of core_bentley_1.Id64.iterable(ids)) {
+        target.add({ className, id });
     }
 }
-/** @internal */
 class ScopedSelectionChanger {
     constructor(name, imodel, manager, scope) {
         this.name = name;
@@ -437,21 +512,33 @@ class ScopedSelectionChanger {
     async clear(level) {
         this.manager.clearSelection(this.name, this.imodel, level);
     }
-    async add(transientIds, persistentIds, level) {
-        const keys = await this.manager.scopes.computeSelection(this.imodel, persistentIds, this.scope);
-        addTransientKeys(transientIds, keys);
+    async add(ids, level) {
+        const keys = await this.#computeSelection(ids);
         this.manager.addToSelection(this.name, this.imodel, keys, level);
     }
-    async remove(transientIds, persistentIds, level) {
-        const keys = await this.manager.scopes.computeSelection(this.imodel, persistentIds, this.scope);
-        addTransientKeys(transientIds, keys);
+    async remove(ids, level) {
+        const keys = await this.#computeSelection(ids);
         this.manager.removeFromSelection(this.name, this.imodel, keys, level);
     }
-    async replace(transientIds, persistentIds, level) {
-        const keys = await this.manager.scopes.computeSelection(this.imodel, persistentIds, this.scope);
-        addTransientKeys(transientIds, keys);
+    async replace(ids, level) {
+        const keys = await this.#computeSelection(ids);
         this.manager.replaceSelection(this.name, this.imodel, keys, level);
     }
+    async #computeSelection(ids) {
+        let keys = new presentation_common_1.KeySet();
+        if (ids.elements) {
+            const { persistent, transient } = parseElementIds(ids.elements);
+            keys = await this.manager.scopes.computeSelection(this.imodel, persistent, this.scope);
+            addKeys(keys, unified_selection_1.TRANSIENT_ELEMENT_CLASSNAME, transient);
+        }
+        if (ids.models) {
+            addKeys(keys, "BisCore.Model", ids.models);
+        }
+        if (ids.subcategories) {
+            addKeys(keys, "BisCore.SubCategory", ids.subcategories);
+        }
+        return keys;
+    }
 }
 /** Stores current selection in `KeySet` format per iModel.  */
 class CurrentSelectionStorage {