@itwin/presentation-backend 5.0.0-dev.98 → 5.1.0-dev.0

package/lib/cjs/presentation-backend/RulesetEmbedder.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"RulesetEmbedder.d.ts","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetEmbedder.ts"],"names":[],"mappings":"AAIA;;GAEG;AAIH,OAAO,EAML,MAAM,EACN,QAAQ,EAIT,MAAM,qBAAqB,CAAC;AAC7B,OAAO,EAAoB,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAenE,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAKrD;;;GAGG;AACH,UAAU,eAAe;IACvB,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACjD;AAED;;;GAGG;AACH,UAAU,eAAe;IACvB,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACjD;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC;;;;;;;;OAQG;IACH,IAAI,CAAC,EAAE,OAAO,GAAG,SAAS,GAAG,wBAAwB,GAAG,yBAAyB,CAAC;IAElF;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,KAAK,GAAG,WAAW,GAAG,OAAO,CAAC;IAEhD;;OAEG;IACH,cAAc,CAAC,EAAE,eAAe,CAAC;IAEjC;;OAEG;IACH,cAAc,CAAC,EAAE,eAAe,CAAC;CAClC;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC,kCAAkC;IAClC,MAAM,EAAE,QAAQ,CAAC;IACjB;;;OAGG;IACH,eAAe,CAAC,EAAE,UAAU,CAAC;CAC9B;AAED;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,OAAO,CAAW;IAC1B,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgG;IAC5H,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAuB;IACzD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAuB;IAE3D;;OAEG;gBACgB,KAAK,EAAE,oBAAoB;IAM9C;;;;;OAKG;IACU,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC;YA2EnF,aAAa;YAWb,gBAAgB;IAa9B;;OAEG;IACU,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAiBhC,uBAAuB;IAWrC,OAAO,CAAC,iBAAiB;IASzB,OAAO,CAAC,wBAAwB;IAShC,OAAO,CAAC,YAAY;YAYN,qBAAqB;YAWrB,yBAAyB;YAezB,aAAa;YAqBb,mCAAmC;YAcnC,aAAa;YAYb,WAAW;YAaX,aAAa;CAU5B"}
+{"version":3,"file":"RulesetEmbedder.d.ts","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetEmbedder.ts"],"names":[],"mappings":"AAIA;;GAEG;AAIH,OAAO,EAAoE,MAAM,EAAE,QAAQ,EAAkC,MAAM,qBAAqB,CAAC;AACzJ,OAAO,EAAU,UAAU,EAAE,MAAM,qBAAqB,CAAC;AAezD,OAAO,EAAE,OAAO,EAAE,MAAM,4BAA4B,CAAC;AAKrD;;;GAGG;AACH,UAAU,eAAe;IACvB,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACjD;AAED;;;GAGG;AACH,UAAU,eAAe;IACvB,cAAc,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;IACjD,aAAa,EAAE,CAAC,KAAK,EAAE,MAAM,KAAK,OAAO,CAAC,IAAI,CAAC,CAAC;CACjD;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC;;;;;;;;OAQG;IACH,IAAI,CAAC,EAAE,OAAO,GAAG,SAAS,GAAG,wBAAwB,GAAG,yBAAyB,CAAC;IAElF;;;;;;;OAOG;IACH,eAAe,CAAC,EAAE,KAAK,GAAG,WAAW,GAAG,OAAO,CAAC;IAEhD;;OAEG;IACH,cAAc,CAAC,EAAE,eAAe,CAAC;IAEjC;;OAEG;IACH,cAAc,CAAC,EAAE,eAAe,CAAC;CAClC;AAED;;;GAGG;AACH,MAAM,WAAW,oBAAoB;IACnC,kCAAkC;IAClC,MAAM,EAAE,QAAQ,CAAC;IACjB;;;OAGG;IACH,eAAe,CAAC,EAAE,UAAU,CAAC;CAC9B;AAED;;;GAGG;AACH,qBAAa,eAAe;IAC1B,OAAO,CAAC,OAAO,CAAW;IAC1B,OAAO,CAAC,gBAAgB,CAAa;IACrC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAgG;IAC5H,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAuB;IACzD,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAuB;IAE3D;;OAEG;gBACgB,KAAK,EAAE,oBAAoB;IAM9C;;;;;OAKG;IACU,aAAa,CAAC,OAAO,EAAE,OAAO,EAAE,OAAO,CAAC,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC;YA2EnF,aAAa;YAWb,gBAAgB;IAa9B;;OAEG;IACU,WAAW,IAAI,OAAO,CAAC,OAAO,EAAE,CAAC;YAchC,uBAAuB;IAWrC,OAAO,CAAC,iBAAiB;IASzB,OAAO,CAAC,wBAAwB;IAShC,OAAO,CAAC,YAAY;YAYN,qBAAqB;YAWrB,yBAAyB;YAezB,aAAa;YAqBb,mCAAmC;YAcnC,aAAa;YAYb,WAAW;YAaX,aAAa;CAU5B"}

package/lib/cjs/presentation-backend/RulesetEmbedder.js

@@ -70,9 +70,9 @@ class RulesetEmbedder {
         await this.handleElementOperationPrerequisites();
         // find all rulesets with the same ID
         const rulesetsWithSameId = [];
-        const query = `
-      SELECT ECInstanceId, JsonProperties
-      FROM ${RulesetElements.Ruleset.schema.name}.${RulesetElements.Ruleset.className}
+        const query = `
+      SELECT ECInstanceId, JsonProperties
+      FROM ${RulesetElements.Ruleset.schema.name}.${RulesetElements.Ruleset.className}
       WHERE json_extract(JsonProperties, '$.jsonProperties.id') = :rulesetId`;
         const reader = this._imodel.createQueryReader(query, core_common_1.QueryBinder.from({ rulesetId: ruleset.id }), { rowFormat: core_common_1.QueryRowFormat.UseJsPropertyNames });
         while (await reader.step()) {
@@ -153,14 +153,11 @@ class RulesetEmbedder {
             return [];
         }
         const rulesetList = [];
-        this._imodel.withPreparedStatement(`SELECT ECInstanceId AS id FROM ${RulesetElements.Ruleset.classFullName}`, (statement) => {
-            while (core_bentley_1.DbResult.BE_SQLITE_ROW === statement.step()) {
-                const row = statement.getRow();
-                const rulesetElement = this._imodel.elements.getElement({ id: row.id });
-                const ruleset = rulesetElement.jsonProperties.jsonProperties;
-                rulesetList.push(ruleset);
-            }
-        });
+        for await (const row of this._imodel.createQueryReader(`SELECT ECInstanceId AS id FROM ${RulesetElements.Ruleset.classFullName}`)) {
+            const rulesetElement = this._imodel.elements.getElement({ id: row.id });
+            const ruleset = rulesetElement.jsonProperties.jsonProperties;
+            rulesetList.push(ruleset);
+        }
         return rulesetList;
     }
     async getOrCreateRulesetModel(callbacks) {

package/lib/cjs/presentation-backend/RulesetEmbedder.js.map

@@ -1 +1 @@
-{"version":3,"file":"RulesetEmbedder.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetEmbedder.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,2CAA6B;AAC7B,mCAA6E;AAC7E,sDAW6B;AAC7B,sDAAmE;AACnE,oDAa4B;AAE5B,oFAAwE;AACxE,6EAA+D;AAC/D,yCAA8C;AAuE9C;;;GAGG;AACH,MAAa,eAAe;IAClB,OAAO,CAAW;IAClB,gBAAgB,CAAa;IACpB,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,6BAAc,CAAC,eAAe,EAAE,iDAAiD,CAAC,CAAC;IAC3G,iBAAiB,GAAG,mBAAmB,CAAC;IACxC,mBAAmB,GAAG,mBAAmB,CAAC;IAE3D;;OAEG;IACH,YAAmB,KAA2B;QAC5C,8CAAiB,CAAC,cAAc,EAAE,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,eAAe,IAAI,oBAAM,CAAC,aAAa,CAAC;IACxE,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,aAAa,CAAC,OAAgB,EAAE,OAA8B;QACzE,MAAM,iBAAiB,GAAG,6BAA6B,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,cAAc,GAAG,IAAA,2BAAgB,EAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAEzD,oEAAoE;QACpE,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAEjD,qCAAqC;QACrC,MAAM,kBAAkB,GAInB,EAAE,CAAC;QACR,MAAM,KAAK,GAAG;;aAEL,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,IAAI,eAAe,CAAC,OAAO,CAAC,SAAS;6EACR,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,KAAK,EAAE,yBAAW,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACpJ,OAAO,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,wBAAwB,GAAe,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,eAAe,GAAY,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,cAAc,CAAC;YAC/E,kBAAkB,CAAC,IAAI,CAAC;gBACtB,EAAE,EAAE,wBAAwB;gBAC5B,OAAO,EAAE,eAAe;gBACxB,iBAAiB,EAAE,IAAA,2BAAgB,EAAC,eAAe,CAAC,OAAO,CAAC;aAC7D,CAAC,CAAC;QACL,CAAC;QAED,yCAAyC;QACzC,MAAM,UAAU,GACd,CAAC,iBAAiB,CAAC,IAAI,KAAK,SAAS,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;YACvE,CAAC,iBAAiB,CAAC,IAAI,KAAK,wBAAwB,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,iBAAiB,KAAK,cAAc,CAAC,CAAC;YACvI,CAAC,iBAAiB,CAAC,IAAI,KAAK,yBAAyB,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAA,YAAU,EAAC,KAAK,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;QACpJ,IAAI,UAAU,EAAE,CAAC;YACf,2FAA2F;YAC3F,MAAM,8BAA8B,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE;gBACjF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,IAAA,WAAS,EAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;oBACnH,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,OAAO,8BAA8B,CAAC,EAAE,CAAC;QAC3C,CAAC;QAED,yCAAyC;QACzC,MAAM,gBAAgB,GAAiB,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAG,CAAC,CAAU,EAAE,iBAAyB,EAAW,EAAE;YACtE,QAAQ,iBAAiB,CAAC,eAAe,EAAE,CAAC;gBAC1C,KAAK,KAAK;oBACR,OAAO,iBAAiB,KAAK,cAAc,CAAC;gBAC9C,KAAK,WAAW;oBACd,OAAO,iBAAiB,KAAK,cAAc,IAAI,IAAA,WAAS,EAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;YAChG,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QACF,kBAAkB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACnC,IAAI,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACzD,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QAEtD,qDAAqD;QACrD,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,KAAK,cAAc,CAAC,CAAC;QAChG,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,EAAE,OAAO,EAAE,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACtF,CAAC;QAED,sDAAsD;QACtD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACnF,MAAM,WAAW,GAAG,eAAe,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/F,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,iBAAiB,CAAC,cAAc,CAAC,CAAC;IAC9F,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAAqB,EAAE,OAAgB,EAAE,SAA2B;QAC9F,MAAM,sBAAsB,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoB,SAAS,CAAC,CAAC;QACjG,IAAA,qBAAM,EAAC,sBAAsB,KAAK,SAAS,CAAC,CAAC;QAC7C,sBAAsB,CAAC,cAAc,CAAC,cAAc,GAAG,OAAO,CAAC;QAE/D,MAAM,IAAI,CAAC,aAAa,CAAC,sBAAsB,EAAE,SAAS,CAAC,CAAC;QAE5D,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,sBAAsB,CAAC,EAAE,CAAC;IACnC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAgB,EAAE,KAAY,EAAE,WAAiB,EAAE,SAA2B;QAC3G,MAAM,KAAK,GAA2B;YACpC,KAAK,EAAE,KAAK,CAAC,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE,eAAe,CAAC,OAAO,CAAC,aAAa;YACpD,cAAc,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE;SAC5C,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,EAAE,CAAC;IACpB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,WAAW;QACtB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,IAAI,CAAC,OAAO,CAAC,qBAAqB,CAAC,kCAAkC,eAAe,CAAC,OAAO,CAAC,aAAa,EAAE,EAAE,CAAC,SAAyB,EAAE,EAAE;YAC1I,OAAO,uBAAQ,CAAC,aAAa,KAAK,SAAS,CAAC,IAAI,EAAE,EAAE,CAAC;gBACnD,MAAM,GAAG,GAAG,SAAS,CAAC,MAAM,EAAE,CAAC;gBAC/B,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;gBACxE,MAAM,OAAO,GAAG,cAAc,CAAC,cAAc,CAAC,cAAc,CAAC;gBAC7D,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;YAC5B,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,SAA2B;QAC/D,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC9C,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;YAC/B,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAC5F,OAAO,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;IACpE,CAAC;IAEO,iBAAiB;QACvB,MAAM,mBAAmB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC5D,IAAI,SAAS,KAAK,mBAAmB,EAAE,CAAC;YACtC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAEO,wBAAwB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACpC,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YAC1B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB,kCAAmB,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IACpJ,CAAC;IAEO,YAAY;QAClB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAa,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,yBAAW,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,IAAI,GAAG,IAAI,kBAAI,CAAC;YACpB,IAAI,EAAE,QAAQ,CAAC,EAAE;YACjB,KAAK,EAAE,MAAM,CAAC,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,mBAAmB;SAChC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAU,IAAI,CAAC,CAAC;IAC5D,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,mBAAwC,EAAE,SAA2B;QACvG,MAAM,UAAU,GAAe;YAC7B,cAAc,EAAE,mBAAmB;YACnC,IAAI,EAAE,IAAI,CAAC,iBAAiB;YAC5B,aAAa,EAAE,8BAAe,CAAC,aAAa;YAC5C,SAAS,EAAE,IAAI;SAChB,CAAC;QAEF,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,yBAAyB,CAAC,cAAuB,EAAE,SAA2B;QAC1F,MAAM,aAAa,GAAG,kCAAmB,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9G,MAAM,wBAAwB,GAAqC;YACjE,MAAM,EAAE;gBACN,EAAE,EAAE,cAAc,CAAC,EAAE;gBACrB,YAAY,EAAE,sCAAsC;aACrD;YACD,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,IAAI,EAAE,aAAa;YACnB,aAAa,EAAE,kCAAmB,CAAC,aAAa;SACjD,CAAC;QAEF,OAAO,IAAI,CAAC,aAAa,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;IACjE,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAA2B;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAa,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,yBAAW,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,WAAW,GAAG,IAAI,kBAAI,CAAC;YAC3B,IAAI,EAAE,QAAQ,CAAC,EAAE;YACjB,KAAK,EAAE,MAAM,CAAC,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,mBAAmB;SAChC,CAAC,CAAC;QACH,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,sBAAO,CAAC,aAAa;YACpC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE;gBACN,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,YAAY,EAAE,6BAA6B;aAC5C;YACD,IAAI,EAAE,WAAW;SAClB,CAAC;QAEF,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACtE,OAAO;QACT,CAAC;QAED,oCAAoC;QACpC,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;QAErD,uCAAuC;QACvC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,sBAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,8CAAiB,CAAC,QAAQ,CAAC,OAAO,EAAE,2BAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAE3H,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,aAAa,CAA8B,KAAa,EAAE,SAA2B;QACjG,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC3D,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5D,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,KAAiB,EAAE,SAA2B;QACtE,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACrD,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC;YACH,KAAK,CAAC,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1B,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAgB,EAAE,SAA2B;QACvE,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;CACF;AAtRD,0CAsRC;AAED,SAAS,6BAA6B,CAAC,OAA8B;IACnE,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,EAAE,IAAI,EAAE,wBAAwB,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC;IACtE,CAAC;IAED,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,wBAAwB;QAC9C,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,OAAO;QACnD,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,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/** @packageDocumentation\n * @module Core\n */\n\nimport * as path from \"path\";\nimport { gt as versionGt, gte as versionGte, lt as versionLt } from \"semver\";\nimport {\n  DefinitionElement,\n  DefinitionModel,\n  DefinitionPartition,\n  ECSqlStatement,\n  Element,\n  Entity,\n  IModelDb,\n  KnownLocations,\n  Model,\n  Subject,\n} from \"@itwin/core-backend\";\nimport { assert, DbResult, Id64String } from \"@itwin/core-bentley\";\nimport {\n  BisCodeSpec,\n  Code,\n  CodeScopeSpec,\n  CodeSpec,\n  DefinitionElementProps,\n  ElementProps,\n  IModel,\n  InformationPartitionElementProps,\n  ModelProps,\n  QueryBinder,\n  QueryRowFormat,\n  SubjectProps,\n} from \"@itwin/core-common\";\nimport { Ruleset } from \"@itwin/presentation-common\";\nimport { PresentationRules } from \"./domain/PresentationRulesDomain.js\";\nimport * as RulesetElements from \"./domain/RulesetElements.js\";\nimport { normalizeVersion } from \"./Utils.js\";\n\n/**\n * Interface for callbacks which will be called before and after Element/Model updates\n * @public\n */\ninterface UpdateCallbacks {\n  onBeforeUpdate: (props: Entity) => Promise<void>;\n  onAfterUpdate: (props: Entity) => Promise<void>;\n}\n\n/**\n * Interface for callbacks which will be called before and after Element/Model is inserted\n * @public\n */\ninterface InsertCallbacks {\n  onBeforeInsert: (props: Entity) => Promise<void>;\n  onAfterInsert: (props: Entity) => Promise<void>;\n}\n\n/**\n * Options for [[RulesetEmbedder.insertRuleset]] operation.\n * @public\n */\nexport interface RulesetInsertOptions {\n  /**\n   * When should insertion be skipped:\n   * - `same-id` - if iModel already contains a ruleset with the same id and **any** version\n   * - `same-id-and-version-eq` - if iModel already contains a ruleset with same id and version\n   * - `same-id-and-version-gte` - if iModel already contains a ruleset with same id and\n   * version greater or equal to version of the inserted ruleset.\n   *\n   * Defaults to `same-id-and-version-eq`.\n   */\n  skip?: \"never\" | \"same-id\" | \"same-id-and-version-eq\" | \"same-id-and-version-gte\";\n\n  /**\n   * Which existing versions of rulesets with same id should be replaced when we insert a new one:\n   * - `all` - replace all rulesets with same id.\n   * - `all-lower` - replace rulesets with same id and version lower than the version of inserted ruleset.\n   * - `exact` - replace only the ruleset whose id and version matches the inserted ruleset.\n   *\n   * Defaults to `exact`.\n   */\n  replaceVersions?: \"all\" | \"all-lower\" | \"exact\";\n\n  /**\n   * Callbacks that will be called before and after `Entity` updates\n   */\n  onEntityUpdate?: UpdateCallbacks;\n\n  /**\n   * Callbacks that will be called before and after `Entity` is inserted\n   */\n  onEntityInsert?: InsertCallbacks;\n}\n\n/**\n * Properties for creating a `RulesetEmbedder` instance.\n * @public\n */\nexport interface RulesetEmbedderProps {\n  /** iModel to embed rulesets to */\n  imodel: IModelDb;\n  /**\n   * An ID of existing subject under which presentation ruleset will be located or created.\n   * Defaults to 'IModel.rootSubjectId'.\n   */\n  parentSubjectId?: Id64String;\n}\n\n/**\n * An API for embedding presentation rulesets into iModels.\n * @public\n */\nexport class RulesetEmbedder {\n  private _imodel: IModelDb;\n  private _parentSubjectId: Id64String;\n  private readonly _schemaPath = path.join(KnownLocations.nativeAssetsDir, \"ECSchemas/Domain/PresentationRules.ecschema.xml\");\n  private readonly _rulesetModelName = \"PresentationRules\";\n  private readonly _rulesetSubjectName = \"PresentationRules\";\n\n  /**\n   * Constructs RulesetEmbedder\n   */\n  public constructor(props: RulesetEmbedderProps) {\n    PresentationRules.registerSchema();\n    this._imodel = props.imodel;\n    this._parentSubjectId = props.parentSubjectId ?? IModel.rootSubjectId;\n  }\n\n  /**\n   * Inserts a ruleset into iModel.\n   * @param ruleset Ruleset to insert.\n   * @param options Options for inserting a ruleset.\n   * @returns ID of inserted ruleset element or, if insertion was skipped, ID of existing ruleset with the same ID and highest version.\n   */\n  public async insertRuleset(ruleset: Ruleset, options?: RulesetInsertOptions): Promise<Id64String> {\n    const normalizedOptions = normalizeRulesetInsertOptions(options);\n    const rulesetVersion = normalizeVersion(ruleset.version);\n\n    // ensure imodel has PresentationRules schema and required CodeSpecs\n    await this.handleElementOperationPrerequisites();\n\n    // find all rulesets with the same ID\n    const rulesetsWithSameId: Array<{\n      ruleset: Ruleset;\n      id: Id64String;\n      normalizedVersion: string;\n    }> = [];\n    const query = `\n      SELECT ECInstanceId, JsonProperties\n      FROM ${RulesetElements.Ruleset.schema.name}.${RulesetElements.Ruleset.className}\n      WHERE json_extract(JsonProperties, '$.jsonProperties.id') = :rulesetId`;\n    const reader = this._imodel.createQueryReader(query, QueryBinder.from({ rulesetId: ruleset.id }), { rowFormat: QueryRowFormat.UseJsPropertyNames });\n    while (await reader.step()) {\n      const row = reader.current.toRow();\n      const existingRulesetElementId: Id64String = row.id;\n      const existingRuleset: Ruleset = JSON.parse(row.jsonProperties).jsonProperties;\n      rulesetsWithSameId.push({\n        id: existingRulesetElementId,\n        ruleset: existingRuleset,\n        normalizedVersion: normalizeVersion(existingRuleset.version),\n      });\n    }\n\n    // check if we need to do anything at all\n    const shouldSkip =\n      (normalizedOptions.skip === \"same-id\" && rulesetsWithSameId.length > 0) ||\n      (normalizedOptions.skip === \"same-id-and-version-eq\" && rulesetsWithSameId.some((entry) => entry.normalizedVersion === rulesetVersion)) ||\n      (normalizedOptions.skip === \"same-id-and-version-gte\" && rulesetsWithSameId.some((entry) => versionGte(entry.normalizedVersion, rulesetVersion)));\n    if (shouldSkip) {\n      // we're not inserting anything - return ID of the ruleset element with the highest version\n      const rulesetEntryWithHighestVersion = rulesetsWithSameId.reduce((highest, curr) => {\n        if (!highest.ruleset.version || (curr.ruleset.version && versionGt(curr.ruleset.version, highest.ruleset.version))) {\n          return curr;\n        }\n        return highest;\n      }, rulesetsWithSameId[0]);\n      return rulesetEntryWithHighestVersion.id;\n    }\n\n    // if requested, delete existing rulesets\n    const rulesetsToRemove: Id64String[] = [];\n    const shouldRemove = (_: Ruleset, normalizedVersion: string): boolean => {\n      switch (normalizedOptions.replaceVersions) {\n        case \"all\":\n          return normalizedVersion !== rulesetVersion;\n        case \"all-lower\":\n          return normalizedVersion !== rulesetVersion && versionLt(normalizedVersion, rulesetVersion);\n      }\n      return false;\n    };\n    rulesetsWithSameId.forEach((entry) => {\n      if (shouldRemove(entry.ruleset, entry.normalizedVersion)) {\n        rulesetsToRemove.push(entry.id);\n      }\n    });\n    this._imodel.elements.deleteElement(rulesetsToRemove);\n\n    // attempt to update ruleset with same ID and version\n    const exactMatch = rulesetsWithSameId.find((curr) => curr.normalizedVersion === rulesetVersion);\n    if (exactMatch !== undefined) {\n      return this.updateRuleset(exactMatch.id, ruleset, normalizedOptions.onEntityUpdate);\n    }\n\n    // no exact match found - insert a new ruleset element\n    const model = await this.getOrCreateRulesetModel(normalizedOptions.onEntityInsert);\n    const rulesetCode = RulesetElements.Ruleset.createRulesetCode(this._imodel, model.id, ruleset);\n    return this.insertNewRuleset(ruleset, model, rulesetCode, normalizedOptions.onEntityInsert);\n  }\n\n  private async updateRuleset(elementId: Id64String, ruleset: Ruleset, callbacks?: UpdateCallbacks) {\n    const existingRulesetElement = this._imodel.elements.tryGetElement<DefinitionElement>(elementId);\n    assert(existingRulesetElement !== undefined);\n    existingRulesetElement.jsonProperties.jsonProperties = ruleset;\n\n    await this.updateElement(existingRulesetElement, callbacks);\n\n    this._imodel.saveChanges();\n    return existingRulesetElement.id;\n  }\n\n  private async insertNewRuleset(ruleset: Ruleset, model: Model, rulesetCode: Code, callbacks?: InsertCallbacks): Promise<Id64String> {\n    const props: DefinitionElementProps = {\n      model: model.id,\n      code: rulesetCode,\n      classFullName: RulesetElements.Ruleset.classFullName,\n      jsonProperties: { jsonProperties: ruleset },\n    };\n\n    const element = await this.insertElement(props, callbacks);\n    this._imodel.saveChanges();\n    return element.id;\n  }\n\n  /**\n   * Get all rulesets embedded in the iModel.\n   */\n  public async getRulesets(): Promise<Ruleset[]> {\n    if (!this._imodel.containsClass(RulesetElements.Ruleset.classFullName)) {\n      return [];\n    }\n\n    const rulesetList: Ruleset[] = [];\n    this._imodel.withPreparedStatement(`SELECT ECInstanceId AS id FROM ${RulesetElements.Ruleset.classFullName}`, (statement: ECSqlStatement) => {\n      while (DbResult.BE_SQLITE_ROW === statement.step()) {\n        const row = statement.getRow();\n        const rulesetElement = this._imodel.elements.getElement({ id: row.id });\n        const ruleset = rulesetElement.jsonProperties.jsonProperties;\n        rulesetList.push(ruleset);\n      }\n    });\n    return rulesetList;\n  }\n\n  private async getOrCreateRulesetModel(callbacks?: InsertCallbacks): Promise<DefinitionModel> {\n    const rulesetModel = this.queryRulesetModel();\n    if (undefined !== rulesetModel) {\n      return rulesetModel;\n    }\n\n    const rulesetSubject = await this.insertSubject(callbacks);\n    const definitionPartition = await this.insertDefinitionPartition(rulesetSubject, callbacks);\n    return this.insertDefinitionModel(definitionPartition, callbacks);\n  }\n\n  private queryRulesetModel(): DefinitionModel | undefined {\n    const definitionPartition = this.queryDefinitionPartition();\n    if (undefined === definitionPartition) {\n      return undefined;\n    }\n\n    return this._imodel.models.getSubModel(definitionPartition.id);\n  }\n\n  private queryDefinitionPartition(): DefinitionPartition | undefined {\n    const subject = this.querySubject();\n    if (undefined === subject) {\n      return undefined;\n    }\n\n    return this._imodel.elements.tryGetElement<DefinitionPartition>(DefinitionPartition.createCode(this._imodel, subject.id, this._rulesetModelName));\n  }\n\n  private querySubject(): Subject | undefined {\n    const parent = this._imodel.elements.getElement<Subject>(this._parentSubjectId);\n    const codeSpec: CodeSpec = this._imodel.codeSpecs.getByName(BisCodeSpec.subject);\n    const code = new Code({\n      spec: codeSpec.id,\n      scope: parent.id,\n      value: this._rulesetSubjectName,\n    });\n\n    return this._imodel.elements.tryGetElement<Subject>(code);\n  }\n\n  private async insertDefinitionModel(definitionPartition: DefinitionPartition, callbacks?: InsertCallbacks): Promise<DefinitionModel> {\n    const modelProps: ModelProps = {\n      modeledElement: definitionPartition,\n      name: this._rulesetModelName,\n      classFullName: DefinitionModel.classFullName,\n      isPrivate: true,\n    };\n\n    return this.insertModel(modelProps, callbacks);\n  }\n\n  private async insertDefinitionPartition(rulesetSubject: Subject, callbacks?: InsertCallbacks): Promise<DefinitionPartition> {\n    const partitionCode = DefinitionPartition.createCode(this._imodel, rulesetSubject.id, this._rulesetModelName);\n    const definitionPartitionProps: InformationPartitionElementProps = {\n      parent: {\n        id: rulesetSubject.id,\n        relClassName: \"BisCore:SubjectOwnsPartitionElements\",\n      },\n      model: rulesetSubject.model,\n      code: partitionCode,\n      classFullName: DefinitionPartition.classFullName,\n    };\n\n    return this.insertElement(definitionPartitionProps, callbacks);\n  }\n\n  private async insertSubject(callbacks?: InsertCallbacks): Promise<Subject> {\n    const parent = this._imodel.elements.getElement<Subject>(this._parentSubjectId);\n    const codeSpec: CodeSpec = this._imodel.codeSpecs.getByName(BisCodeSpec.subject);\n    const subjectCode = new Code({\n      spec: codeSpec.id,\n      scope: parent.id,\n      value: this._rulesetSubjectName,\n    });\n    const subjectProps: SubjectProps = {\n      classFullName: Subject.classFullName,\n      model: parent.model,\n      parent: {\n        id: parent.id,\n        relClassName: \"BisCore:SubjectOwnsSubjects\",\n      },\n      code: subjectCode,\n    };\n\n    return this.insertElement(subjectProps, callbacks);\n  }\n\n  private async handleElementOperationPrerequisites(): Promise<void> {\n    if (this._imodel.containsClass(RulesetElements.Ruleset.classFullName)) {\n      return;\n    }\n\n    // import PresentationRules ECSchema\n    await this._imodel.importSchemas([this._schemaPath]);\n\n    // insert CodeSpec for ruleset elements\n    this._imodel.codeSpecs.insert(CodeSpec.create(this._imodel, PresentationRules.CodeSpec.Ruleset, CodeScopeSpec.Type.Model));\n\n    this._imodel.saveChanges();\n  }\n\n  private async insertElement<TProps extends ElementProps>(props: TProps, callbacks?: InsertCallbacks): Promise<Element> {\n    const element = this._imodel.elements.createElement(props);\n    /* c8 ignore next */\n    await callbacks?.onBeforeInsert(element);\n    try {\n      return this._imodel.elements.getElement(element.insert());\n    } finally {\n      /* c8 ignore next */\n      await callbacks?.onAfterInsert(element);\n    }\n  }\n\n  private async insertModel(props: ModelProps, callbacks?: InsertCallbacks): Promise<Model> {\n    const model = this._imodel.models.createModel(props);\n    /* c8 ignore next */\n    await callbacks?.onBeforeInsert(model);\n    try {\n      model.id = model.insert();\n      return model;\n    } finally {\n      /* c8 ignore next */\n      await callbacks?.onAfterInsert(model);\n    }\n  }\n\n  private async updateElement(element: Element, callbacks?: UpdateCallbacks) {\n    /* c8 ignore next */\n    await callbacks?.onBeforeUpdate(element);\n    try {\n      element.update();\n    } finally {\n      /* c8 ignore next */\n      await callbacks?.onAfterUpdate(element);\n    }\n  }\n}\n\nfunction normalizeRulesetInsertOptions(options?: RulesetInsertOptions): RulesetInsertOptions {\n  if (options === undefined) {\n    return { skip: \"same-id-and-version-eq\", replaceVersions: \"exact\" };\n  }\n\n  return {\n    skip: options.skip ?? \"same-id-and-version-eq\",\n    replaceVersions: options.replaceVersions ?? \"exact\",\n    onEntityUpdate: options.onEntityUpdate,\n    onEntityInsert: options.onEntityInsert,\n  };\n}\n"]}
+{"version":3,"file":"RulesetEmbedder.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetEmbedder.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;;;;;;;;;;;;;;;;;;;;;;;;AAEH,2CAA6B;AAC7B,mCAA6E;AAC7E,sDAAyJ;AACzJ,sDAAyD;AACzD,oDAa4B;AAE5B,oFAAwE;AACxE,6EAA+D;AAC/D,yCAA8C;AAuE9C;;;GAGG;AACH,MAAa,eAAe;IAClB,OAAO,CAAW;IAClB,gBAAgB,CAAa;IACpB,WAAW,GAAG,IAAI,CAAC,IAAI,CAAC,6BAAc,CAAC,eAAe,EAAE,iDAAiD,CAAC,CAAC;IAC3G,iBAAiB,GAAG,mBAAmB,CAAC;IACxC,mBAAmB,GAAG,mBAAmB,CAAC;IAE3D;;OAEG;IACH,YAAmB,KAA2B;QAC5C,8CAAiB,CAAC,cAAc,EAAE,CAAC;QACnC,IAAI,CAAC,OAAO,GAAG,KAAK,CAAC,MAAM,CAAC;QAC5B,IAAI,CAAC,gBAAgB,GAAG,KAAK,CAAC,eAAe,IAAI,oBAAM,CAAC,aAAa,CAAC;IACxE,CAAC;IAED;;;;;OAKG;IACI,KAAK,CAAC,aAAa,CAAC,OAAgB,EAAE,OAA8B;QACzE,MAAM,iBAAiB,GAAG,6BAA6B,CAAC,OAAO,CAAC,CAAC;QACjE,MAAM,cAAc,GAAG,IAAA,2BAAgB,EAAC,OAAO,CAAC,OAAO,CAAC,CAAC;QAEzD,oEAAoE;QACpE,MAAM,IAAI,CAAC,mCAAmC,EAAE,CAAC;QAEjD,qCAAqC;QACrC,MAAM,kBAAkB,GAInB,EAAE,CAAC;QACR,MAAM,KAAK,GAAG;;aAEL,eAAe,CAAC,OAAO,CAAC,MAAM,CAAC,IAAI,IAAI,eAAe,CAAC,OAAO,CAAC,SAAS;6EACR,CAAC;QAC1E,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,KAAK,EAAE,yBAAW,CAAC,IAAI,CAAC,EAAE,SAAS,EAAE,OAAO,CAAC,EAAE,EAAE,CAAC,EAAE,EAAE,SAAS,EAAE,4BAAc,CAAC,kBAAkB,EAAE,CAAC,CAAC;QACpJ,OAAO,MAAM,MAAM,CAAC,IAAI,EAAE,EAAE,CAAC;YAC3B,MAAM,GAAG,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YACnC,MAAM,wBAAwB,GAAe,GAAG,CAAC,EAAE,CAAC;YACpD,MAAM,eAAe,GAAY,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,cAAc,CAAC,CAAC,cAAc,CAAC;YAC/E,kBAAkB,CAAC,IAAI,CAAC;gBACtB,EAAE,EAAE,wBAAwB;gBAC5B,OAAO,EAAE,eAAe;gBACxB,iBAAiB,EAAE,IAAA,2BAAgB,EAAC,eAAe,CAAC,OAAO,CAAC;aAC7D,CAAC,CAAC;QACL,CAAC;QAED,yCAAyC;QACzC,MAAM,UAAU,GACd,CAAC,iBAAiB,CAAC,IAAI,KAAK,SAAS,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;YACvE,CAAC,iBAAiB,CAAC,IAAI,KAAK,wBAAwB,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,KAAK,CAAC,iBAAiB,KAAK,cAAc,CAAC,CAAC;YACvI,CAAC,iBAAiB,CAAC,IAAI,KAAK,yBAAyB,IAAI,kBAAkB,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAA,YAAU,EAAC,KAAK,CAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC,CAAC,CAAC;QACpJ,IAAI,UAAU,EAAE,CAAC;YACf,2FAA2F;YAC3F,MAAM,8BAA8B,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC,OAAO,EAAE,IAAI,EAAE,EAAE;gBACjF,IAAI,CAAC,OAAO,CAAC,OAAO,CAAC,OAAO,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,OAAO,IAAI,IAAA,WAAS,EAAC,IAAI,CAAC,OAAO,CAAC,OAAO,EAAE,OAAO,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,EAAE,CAAC;oBACnH,OAAO,IAAI,CAAC;gBACd,CAAC;gBACD,OAAO,OAAO,CAAC;YACjB,CAAC,EAAE,kBAAkB,CAAC,CAAC,CAAC,CAAC,CAAC;YAC1B,OAAO,8BAA8B,CAAC,EAAE,CAAC;QAC3C,CAAC;QAED,yCAAyC;QACzC,MAAM,gBAAgB,GAAiB,EAAE,CAAC;QAC1C,MAAM,YAAY,GAAG,CAAC,CAAU,EAAE,iBAAyB,EAAW,EAAE;YACtE,QAAQ,iBAAiB,CAAC,eAAe,EAAE,CAAC;gBAC1C,KAAK,KAAK;oBACR,OAAO,iBAAiB,KAAK,cAAc,CAAC;gBAC9C,KAAK,WAAW;oBACd,OAAO,iBAAiB,KAAK,cAAc,IAAI,IAAA,WAAS,EAAC,iBAAiB,EAAE,cAAc,CAAC,CAAC;YAChG,CAAC;YACD,OAAO,KAAK,CAAC;QACf,CAAC,CAAC;QACF,kBAAkB,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,EAAE;YACnC,IAAI,YAAY,CAAC,KAAK,CAAC,OAAO,EAAE,KAAK,CAAC,iBAAiB,CAAC,EAAE,CAAC;gBACzD,gBAAgB,CAAC,IAAI,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAClC,CAAC;QACH,CAAC,CAAC,CAAC;QACH,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,gBAAgB,CAAC,CAAC;QAEtD,qDAAqD;QACrD,MAAM,UAAU,GAAG,kBAAkB,CAAC,IAAI,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,IAAI,CAAC,iBAAiB,KAAK,cAAc,CAAC,CAAC;QAChG,IAAI,UAAU,KAAK,SAAS,EAAE,CAAC;YAC7B,OAAO,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,EAAE,EAAE,OAAO,EAAE,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACtF,CAAC;QAED,sDAAsD;QACtD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,iBAAiB,CAAC,cAAc,CAAC,CAAC;QACnF,MAAM,WAAW,GAAG,eAAe,CAAC,OAAO,CAAC,iBAAiB,CAAC,IAAI,CAAC,OAAO,EAAE,KAAK,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;QAC/F,OAAO,IAAI,CAAC,gBAAgB,CAAC,OAAO,EAAE,KAAK,EAAE,WAAW,EAAE,iBAAiB,CAAC,cAAc,CAAC,CAAC;IAC9F,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAAqB,EAAE,OAAgB,EAAE,SAA2B;QAC9F,MAAM,sBAAsB,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAoB,SAAS,CAAC,CAAC;QACjG,IAAA,qBAAM,EAAC,sBAAsB,KAAK,SAAS,CAAC,CAAC;QAC7C,sBAAsB,CAAC,cAAc,CAAC,cAAc,GAAG,OAAO,CAAC;QAE/D,MAAM,IAAI,CAAC,aAAa,CAAC,sBAAsB,EAAE,SAAS,CAAC,CAAC;QAE5D,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,sBAAsB,CAAC,EAAE,CAAC;IACnC,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAAC,OAAgB,EAAE,KAAY,EAAE,WAAiB,EAAE,SAA2B;QAC3G,MAAM,KAAK,GAA2B;YACpC,KAAK,EAAE,KAAK,CAAC,EAAE;YACf,IAAI,EAAE,WAAW;YACjB,aAAa,EAAE,eAAe,CAAC,OAAO,CAAC,aAAa;YACpD,cAAc,EAAE,EAAE,cAAc,EAAE,OAAO,EAAE;SAC5C,CAAC;QAEF,MAAM,OAAO,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,KAAK,EAAE,SAAS,CAAC,CAAC;QAC3D,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAC3B,OAAO,OAAO,CAAC,EAAE,CAAC;IACpB,CAAC;IAED;;OAEG;IACI,KAAK,CAAC,WAAW;QACtB,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACvE,OAAO,EAAE,CAAC;QACZ,CAAC;QAED,MAAM,WAAW,GAAc,EAAE,CAAC;QAClC,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,IAAI,CAAC,OAAO,CAAC,iBAAiB,CAAC,kCAAkC,eAAe,CAAC,OAAO,CAAC,aAAa,EAAE,CAAC,EAAE,CAAC;YAClI,MAAM,cAAc,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,EAAE,EAAE,CAAC,CAAC;YACxE,MAAM,OAAO,GAAG,cAAc,CAAC,cAAc,CAAC,cAAc,CAAC;YAC7D,WAAW,CAAC,IAAI,CAAC,OAAO,CAAC,CAAC;QAC5B,CAAC;QACD,OAAO,WAAW,CAAC;IACrB,CAAC;IAEO,KAAK,CAAC,uBAAuB,CAAC,SAA2B;QAC/D,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,EAAE,CAAC;QAC9C,IAAI,SAAS,KAAK,YAAY,EAAE,CAAC;YAC/B,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,cAAc,GAAG,MAAM,IAAI,CAAC,aAAa,CAAC,SAAS,CAAC,CAAC;QAC3D,MAAM,mBAAmB,GAAG,MAAM,IAAI,CAAC,yBAAyB,CAAC,cAAc,EAAE,SAAS,CAAC,CAAC;QAC5F,OAAO,IAAI,CAAC,qBAAqB,CAAC,mBAAmB,EAAE,SAAS,CAAC,CAAC;IACpE,CAAC;IAEO,iBAAiB;QACvB,MAAM,mBAAmB,GAAG,IAAI,CAAC,wBAAwB,EAAE,CAAC;QAC5D,IAAI,SAAS,KAAK,mBAAmB,EAAE,CAAC;YACtC,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,mBAAmB,CAAC,EAAE,CAAC,CAAC;IACjE,CAAC;IAEO,wBAAwB;QAC9B,MAAM,OAAO,GAAG,IAAI,CAAC,YAAY,EAAE,CAAC;QACpC,IAAI,SAAS,KAAK,OAAO,EAAE,CAAC;YAC1B,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAsB,kCAAmB,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,OAAO,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC,CAAC;IACpJ,CAAC;IAEO,YAAY;QAClB,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAa,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,yBAAW,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,IAAI,GAAG,IAAI,kBAAI,CAAC;YACpB,IAAI,EAAE,QAAQ,CAAC,EAAE;YACjB,KAAK,EAAE,MAAM,CAAC,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,mBAAmB;SAChC,CAAC,CAAC;QAEH,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAU,IAAI,CAAC,CAAC;IAC5D,CAAC;IAEO,KAAK,CAAC,qBAAqB,CAAC,mBAAwC,EAAE,SAA2B;QACvG,MAAM,UAAU,GAAe;YAC7B,cAAc,EAAE,mBAAmB;YACnC,IAAI,EAAE,IAAI,CAAC,iBAAiB;YAC5B,aAAa,EAAE,8BAAe,CAAC,aAAa;YAC5C,SAAS,EAAE,IAAI;SAChB,CAAC;QAEF,OAAO,IAAI,CAAC,WAAW,CAAC,UAAU,EAAE,SAAS,CAAC,CAAC;IACjD,CAAC;IAEO,KAAK,CAAC,yBAAyB,CAAC,cAAuB,EAAE,SAA2B;QAC1F,MAAM,aAAa,GAAG,kCAAmB,CAAC,UAAU,CAAC,IAAI,CAAC,OAAO,EAAE,cAAc,CAAC,EAAE,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;QAC9G,MAAM,wBAAwB,GAAqC;YACjE,MAAM,EAAE;gBACN,EAAE,EAAE,cAAc,CAAC,EAAE;gBACrB,YAAY,EAAE,sCAAsC;aACrD;YACD,KAAK,EAAE,cAAc,CAAC,KAAK;YAC3B,IAAI,EAAE,aAAa;YACnB,aAAa,EAAE,kCAAmB,CAAC,aAAa;SACjD,CAAC;QAEF,OAAO,IAAI,CAAC,aAAa,CAAC,wBAAwB,EAAE,SAAS,CAAC,CAAC;IACjE,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,SAA2B;QACrD,MAAM,MAAM,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAU,IAAI,CAAC,gBAAgB,CAAC,CAAC;QAChF,MAAM,QAAQ,GAAa,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,SAAS,CAAC,yBAAW,CAAC,OAAO,CAAC,CAAC;QACjF,MAAM,WAAW,GAAG,IAAI,kBAAI,CAAC;YAC3B,IAAI,EAAE,QAAQ,CAAC,EAAE;YACjB,KAAK,EAAE,MAAM,CAAC,EAAE;YAChB,KAAK,EAAE,IAAI,CAAC,mBAAmB;SAChC,CAAC,CAAC;QACH,MAAM,YAAY,GAAiB;YACjC,aAAa,EAAE,sBAAO,CAAC,aAAa;YACpC,KAAK,EAAE,MAAM,CAAC,KAAK;YACnB,MAAM,EAAE;gBACN,EAAE,EAAE,MAAM,CAAC,EAAE;gBACb,YAAY,EAAE,6BAA6B;aAC5C;YACD,IAAI,EAAE,WAAW;SAClB,CAAC;QAEF,OAAO,IAAI,CAAC,aAAa,CAAC,YAAY,EAAE,SAAS,CAAC,CAAC;IACrD,CAAC;IAEO,KAAK,CAAC,mCAAmC;QAC/C,IAAI,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,eAAe,CAAC,OAAO,CAAC,aAAa,CAAC,EAAE,CAAC;YACtE,OAAO;QACT,CAAC;QAED,oCAAoC;QACpC,MAAM,IAAI,CAAC,OAAO,CAAC,aAAa,CAAC,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;QAErD,uCAAuC;QACvC,IAAI,CAAC,OAAO,CAAC,SAAS,CAAC,MAAM,CAAC,sBAAQ,CAAC,MAAM,CAAC,IAAI,CAAC,OAAO,EAAE,8CAAiB,CAAC,QAAQ,CAAC,OAAO,EAAE,2BAAa,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC;QAE3H,IAAI,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;IAC7B,CAAC;IAEO,KAAK,CAAC,aAAa,CAA8B,KAAa,EAAE,SAA2B;QACjG,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;QAC3D,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAC,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAC5D,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,KAAiB,EAAE,SAA2B;QACtE,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC;QACrD,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,KAAK,CAAC,CAAC;QACvC,IAAI,CAAC;YACH,KAAK,CAAC,EAAE,GAAG,KAAK,CAAC,MAAM,EAAE,CAAC;YAC1B,OAAO,KAAK,CAAC;QACf,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,KAAK,CAAC,CAAC;QACxC,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,aAAa,CAAC,OAAgB,EAAE,SAA2B;QACvE,oBAAoB;QACpB,MAAM,SAAS,EAAE,cAAc,CAAC,OAAO,CAAC,CAAC;QACzC,IAAI,CAAC;YACH,OAAO,CAAC,MAAM,EAAE,CAAC;QACnB,CAAC;gBAAS,CAAC;YACT,oBAAoB;YACpB,MAAM,SAAS,EAAE,aAAa,CAAC,OAAO,CAAC,CAAC;QAC1C,CAAC;IACH,CAAC;CACF;AAnRD,0CAmRC;AAED,SAAS,6BAA6B,CAAC,OAA8B;IACnE,IAAI,OAAO,KAAK,SAAS,EAAE,CAAC;QAC1B,OAAO,EAAE,IAAI,EAAE,wBAAwB,EAAE,eAAe,EAAE,OAAO,EAAE,CAAC;IACtE,CAAC;IAED,OAAO;QACL,IAAI,EAAE,OAAO,CAAC,IAAI,IAAI,wBAAwB;QAC9C,eAAe,EAAE,OAAO,CAAC,eAAe,IAAI,OAAO;QACnD,cAAc,EAAE,OAAO,CAAC,cAAc;QACtC,cAAc,EAAE,OAAO,CAAC,cAAc;KACvC,CAAC;AACJ,CAAC","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 Core\r\n */\r\n\r\nimport * as path from \"path\";\r\nimport { gt as versionGt, gte as versionGte, lt as versionLt } from \"semver\";\r\nimport { DefinitionElement, DefinitionModel, DefinitionPartition, Element, Entity, IModelDb, KnownLocations, Model, Subject } from \"@itwin/core-backend\";\r\nimport { assert, Id64String } from \"@itwin/core-bentley\";\r\nimport {\r\n  BisCodeSpec,\r\n  Code,\r\n  CodeScopeSpec,\r\n  CodeSpec,\r\n  DefinitionElementProps,\r\n  ElementProps,\r\n  IModel,\r\n  InformationPartitionElementProps,\r\n  ModelProps,\r\n  QueryBinder,\r\n  QueryRowFormat,\r\n  SubjectProps,\r\n} from \"@itwin/core-common\";\r\nimport { Ruleset } from \"@itwin/presentation-common\";\r\nimport { PresentationRules } from \"./domain/PresentationRulesDomain.js\";\r\nimport * as RulesetElements from \"./domain/RulesetElements.js\";\r\nimport { normalizeVersion } from \"./Utils.js\";\r\n\r\n/**\r\n * Interface for callbacks which will be called before and after Element/Model updates\r\n * @public\r\n */\r\ninterface UpdateCallbacks {\r\n  onBeforeUpdate: (props: Entity) => Promise<void>;\r\n  onAfterUpdate: (props: Entity) => Promise<void>;\r\n}\r\n\r\n/**\r\n * Interface for callbacks which will be called before and after Element/Model is inserted\r\n * @public\r\n */\r\ninterface InsertCallbacks {\r\n  onBeforeInsert: (props: Entity) => Promise<void>;\r\n  onAfterInsert: (props: Entity) => Promise<void>;\r\n}\r\n\r\n/**\r\n * Options for [[RulesetEmbedder.insertRuleset]] operation.\r\n * @public\r\n */\r\nexport interface RulesetInsertOptions {\r\n  /**\r\n   * When should insertion be skipped:\r\n   * - `same-id` - if iModel already contains a ruleset with the same id and **any** version\r\n   * - `same-id-and-version-eq` - if iModel already contains a ruleset with same id and version\r\n   * - `same-id-and-version-gte` - if iModel already contains a ruleset with same id and\r\n   * version greater or equal to version of the inserted ruleset.\r\n   *\r\n   * Defaults to `same-id-and-version-eq`.\r\n   */\r\n  skip?: \"never\" | \"same-id\" | \"same-id-and-version-eq\" | \"same-id-and-version-gte\";\r\n\r\n  /**\r\n   * Which existing versions of rulesets with same id should be replaced when we insert a new one:\r\n   * - `all` - replace all rulesets with same id.\r\n   * - `all-lower` - replace rulesets with same id and version lower than the version of inserted ruleset.\r\n   * - `exact` - replace only the ruleset whose id and version matches the inserted ruleset.\r\n   *\r\n   * Defaults to `exact`.\r\n   */\r\n  replaceVersions?: \"all\" | \"all-lower\" | \"exact\";\r\n\r\n  /**\r\n   * Callbacks that will be called before and after `Entity` updates\r\n   */\r\n  onEntityUpdate?: UpdateCallbacks;\r\n\r\n  /**\r\n   * Callbacks that will be called before and after `Entity` is inserted\r\n   */\r\n  onEntityInsert?: InsertCallbacks;\r\n}\r\n\r\n/**\r\n * Properties for creating a `RulesetEmbedder` instance.\r\n * @public\r\n */\r\nexport interface RulesetEmbedderProps {\r\n  /** iModel to embed rulesets to */\r\n  imodel: IModelDb;\r\n  /**\r\n   * An ID of existing subject under which presentation ruleset will be located or created.\r\n   * Defaults to 'IModel.rootSubjectId'.\r\n   */\r\n  parentSubjectId?: Id64String;\r\n}\r\n\r\n/**\r\n * An API for embedding presentation rulesets into iModels.\r\n * @public\r\n */\r\nexport class RulesetEmbedder {\r\n  private _imodel: IModelDb;\r\n  private _parentSubjectId: Id64String;\r\n  private readonly _schemaPath = path.join(KnownLocations.nativeAssetsDir, \"ECSchemas/Domain/PresentationRules.ecschema.xml\");\r\n  private readonly _rulesetModelName = \"PresentationRules\";\r\n  private readonly _rulesetSubjectName = \"PresentationRules\";\r\n\r\n  /**\r\n   * Constructs RulesetEmbedder\r\n   */\r\n  public constructor(props: RulesetEmbedderProps) {\r\n    PresentationRules.registerSchema();\r\n    this._imodel = props.imodel;\r\n    this._parentSubjectId = props.parentSubjectId ?? IModel.rootSubjectId;\r\n  }\r\n\r\n  /**\r\n   * Inserts a ruleset into iModel.\r\n   * @param ruleset Ruleset to insert.\r\n   * @param options Options for inserting a ruleset.\r\n   * @returns ID of inserted ruleset element or, if insertion was skipped, ID of existing ruleset with the same ID and highest version.\r\n   */\r\n  public async insertRuleset(ruleset: Ruleset, options?: RulesetInsertOptions): Promise<Id64String> {\r\n    const normalizedOptions = normalizeRulesetInsertOptions(options);\r\n    const rulesetVersion = normalizeVersion(ruleset.version);\r\n\r\n    // ensure imodel has PresentationRules schema and required CodeSpecs\r\n    await this.handleElementOperationPrerequisites();\r\n\r\n    // find all rulesets with the same ID\r\n    const rulesetsWithSameId: Array<{\r\n      ruleset: Ruleset;\r\n      id: Id64String;\r\n      normalizedVersion: string;\r\n    }> = [];\r\n    const query = `\r\n      SELECT ECInstanceId, JsonProperties\r\n      FROM ${RulesetElements.Ruleset.schema.name}.${RulesetElements.Ruleset.className}\r\n      WHERE json_extract(JsonProperties, '$.jsonProperties.id') = :rulesetId`;\r\n    const reader = this._imodel.createQueryReader(query, QueryBinder.from({ rulesetId: ruleset.id }), { rowFormat: QueryRowFormat.UseJsPropertyNames });\r\n    while (await reader.step()) {\r\n      const row = reader.current.toRow();\r\n      const existingRulesetElementId: Id64String = row.id;\r\n      const existingRuleset: Ruleset = JSON.parse(row.jsonProperties).jsonProperties;\r\n      rulesetsWithSameId.push({\r\n        id: existingRulesetElementId,\r\n        ruleset: existingRuleset,\r\n        normalizedVersion: normalizeVersion(existingRuleset.version),\r\n      });\r\n    }\r\n\r\n    // check if we need to do anything at all\r\n    const shouldSkip =\r\n      (normalizedOptions.skip === \"same-id\" && rulesetsWithSameId.length > 0) ||\r\n      (normalizedOptions.skip === \"same-id-and-version-eq\" && rulesetsWithSameId.some((entry) => entry.normalizedVersion === rulesetVersion)) ||\r\n      (normalizedOptions.skip === \"same-id-and-version-gte\" && rulesetsWithSameId.some((entry) => versionGte(entry.normalizedVersion, rulesetVersion)));\r\n    if (shouldSkip) {\r\n      // we're not inserting anything - return ID of the ruleset element with the highest version\r\n      const rulesetEntryWithHighestVersion = rulesetsWithSameId.reduce((highest, curr) => {\r\n        if (!highest.ruleset.version || (curr.ruleset.version && versionGt(curr.ruleset.version, highest.ruleset.version))) {\r\n          return curr;\r\n        }\r\n        return highest;\r\n      }, rulesetsWithSameId[0]);\r\n      return rulesetEntryWithHighestVersion.id;\r\n    }\r\n\r\n    // if requested, delete existing rulesets\r\n    const rulesetsToRemove: Id64String[] = [];\r\n    const shouldRemove = (_: Ruleset, normalizedVersion: string): boolean => {\r\n      switch (normalizedOptions.replaceVersions) {\r\n        case \"all\":\r\n          return normalizedVersion !== rulesetVersion;\r\n        case \"all-lower\":\r\n          return normalizedVersion !== rulesetVersion && versionLt(normalizedVersion, rulesetVersion);\r\n      }\r\n      return false;\r\n    };\r\n    rulesetsWithSameId.forEach((entry) => {\r\n      if (shouldRemove(entry.ruleset, entry.normalizedVersion)) {\r\n        rulesetsToRemove.push(entry.id);\r\n      }\r\n    });\r\n    this._imodel.elements.deleteElement(rulesetsToRemove);\r\n\r\n    // attempt to update ruleset with same ID and version\r\n    const exactMatch = rulesetsWithSameId.find((curr) => curr.normalizedVersion === rulesetVersion);\r\n    if (exactMatch !== undefined) {\r\n      return this.updateRuleset(exactMatch.id, ruleset, normalizedOptions.onEntityUpdate);\r\n    }\r\n\r\n    // no exact match found - insert a new ruleset element\r\n    const model = await this.getOrCreateRulesetModel(normalizedOptions.onEntityInsert);\r\n    const rulesetCode = RulesetElements.Ruleset.createRulesetCode(this._imodel, model.id, ruleset);\r\n    return this.insertNewRuleset(ruleset, model, rulesetCode, normalizedOptions.onEntityInsert);\r\n  }\r\n\r\n  private async updateRuleset(elementId: Id64String, ruleset: Ruleset, callbacks?: UpdateCallbacks) {\r\n    const existingRulesetElement = this._imodel.elements.tryGetElement<DefinitionElement>(elementId);\r\n    assert(existingRulesetElement !== undefined);\r\n    existingRulesetElement.jsonProperties.jsonProperties = ruleset;\r\n\r\n    await this.updateElement(existingRulesetElement, callbacks);\r\n\r\n    this._imodel.saveChanges();\r\n    return existingRulesetElement.id;\r\n  }\r\n\r\n  private async insertNewRuleset(ruleset: Ruleset, model: Model, rulesetCode: Code, callbacks?: InsertCallbacks): Promise<Id64String> {\r\n    const props: DefinitionElementProps = {\r\n      model: model.id,\r\n      code: rulesetCode,\r\n      classFullName: RulesetElements.Ruleset.classFullName,\r\n      jsonProperties: { jsonProperties: ruleset },\r\n    };\r\n\r\n    const element = await this.insertElement(props, callbacks);\r\n    this._imodel.saveChanges();\r\n    return element.id;\r\n  }\r\n\r\n  /**\r\n   * Get all rulesets embedded in the iModel.\r\n   */\r\n  public async getRulesets(): Promise<Ruleset[]> {\r\n    if (!this._imodel.containsClass(RulesetElements.Ruleset.classFullName)) {\r\n      return [];\r\n    }\r\n\r\n    const rulesetList: Ruleset[] = [];\r\n    for await (const row of this._imodel.createQueryReader(`SELECT ECInstanceId AS id FROM ${RulesetElements.Ruleset.classFullName}`)) {\r\n      const rulesetElement = this._imodel.elements.getElement({ id: row.id });\r\n      const ruleset = rulesetElement.jsonProperties.jsonProperties;\r\n      rulesetList.push(ruleset);\r\n    }\r\n    return rulesetList;\r\n  }\r\n\r\n  private async getOrCreateRulesetModel(callbacks?: InsertCallbacks): Promise<DefinitionModel> {\r\n    const rulesetModel = this.queryRulesetModel();\r\n    if (undefined !== rulesetModel) {\r\n      return rulesetModel;\r\n    }\r\n\r\n    const rulesetSubject = await this.insertSubject(callbacks);\r\n    const definitionPartition = await this.insertDefinitionPartition(rulesetSubject, callbacks);\r\n    return this.insertDefinitionModel(definitionPartition, callbacks);\r\n  }\r\n\r\n  private queryRulesetModel(): DefinitionModel | undefined {\r\n    const definitionPartition = this.queryDefinitionPartition();\r\n    if (undefined === definitionPartition) {\r\n      return undefined;\r\n    }\r\n\r\n    return this._imodel.models.getSubModel(definitionPartition.id);\r\n  }\r\n\r\n  private queryDefinitionPartition(): DefinitionPartition | undefined {\r\n    const subject = this.querySubject();\r\n    if (undefined === subject) {\r\n      return undefined;\r\n    }\r\n\r\n    return this._imodel.elements.tryGetElement<DefinitionPartition>(DefinitionPartition.createCode(this._imodel, subject.id, this._rulesetModelName));\r\n  }\r\n\r\n  private querySubject(): Subject | undefined {\r\n    const parent = this._imodel.elements.getElement<Subject>(this._parentSubjectId);\r\n    const codeSpec: CodeSpec = this._imodel.codeSpecs.getByName(BisCodeSpec.subject);\r\n    const code = new Code({\r\n      spec: codeSpec.id,\r\n      scope: parent.id,\r\n      value: this._rulesetSubjectName,\r\n    });\r\n\r\n    return this._imodel.elements.tryGetElement<Subject>(code);\r\n  }\r\n\r\n  private async insertDefinitionModel(definitionPartition: DefinitionPartition, callbacks?: InsertCallbacks): Promise<DefinitionModel> {\r\n    const modelProps: ModelProps = {\r\n      modeledElement: definitionPartition,\r\n      name: this._rulesetModelName,\r\n      classFullName: DefinitionModel.classFullName,\r\n      isPrivate: true,\r\n    };\r\n\r\n    return this.insertModel(modelProps, callbacks);\r\n  }\r\n\r\n  private async insertDefinitionPartition(rulesetSubject: Subject, callbacks?: InsertCallbacks): Promise<DefinitionPartition> {\r\n    const partitionCode = DefinitionPartition.createCode(this._imodel, rulesetSubject.id, this._rulesetModelName);\r\n    const definitionPartitionProps: InformationPartitionElementProps = {\r\n      parent: {\r\n        id: rulesetSubject.id,\r\n        relClassName: \"BisCore:SubjectOwnsPartitionElements\",\r\n      },\r\n      model: rulesetSubject.model,\r\n      code: partitionCode,\r\n      classFullName: DefinitionPartition.classFullName,\r\n    };\r\n\r\n    return this.insertElement(definitionPartitionProps, callbacks);\r\n  }\r\n\r\n  private async insertSubject(callbacks?: InsertCallbacks): Promise<Subject> {\r\n    const parent = this._imodel.elements.getElement<Subject>(this._parentSubjectId);\r\n    const codeSpec: CodeSpec = this._imodel.codeSpecs.getByName(BisCodeSpec.subject);\r\n    const subjectCode = new Code({\r\n      spec: codeSpec.id,\r\n      scope: parent.id,\r\n      value: this._rulesetSubjectName,\r\n    });\r\n    const subjectProps: SubjectProps = {\r\n      classFullName: Subject.classFullName,\r\n      model: parent.model,\r\n      parent: {\r\n        id: parent.id,\r\n        relClassName: \"BisCore:SubjectOwnsSubjects\",\r\n      },\r\n      code: subjectCode,\r\n    };\r\n\r\n    return this.insertElement(subjectProps, callbacks);\r\n  }\r\n\r\n  private async handleElementOperationPrerequisites(): Promise<void> {\r\n    if (this._imodel.containsClass(RulesetElements.Ruleset.classFullName)) {\r\n      return;\r\n    }\r\n\r\n    // import PresentationRules ECSchema\r\n    await this._imodel.importSchemas([this._schemaPath]);\r\n\r\n    // insert CodeSpec for ruleset elements\r\n    this._imodel.codeSpecs.insert(CodeSpec.create(this._imodel, PresentationRules.CodeSpec.Ruleset, CodeScopeSpec.Type.Model));\r\n\r\n    this._imodel.saveChanges();\r\n  }\r\n\r\n  private async insertElement<TProps extends ElementProps>(props: TProps, callbacks?: InsertCallbacks): Promise<Element> {\r\n    const element = this._imodel.elements.createElement(props);\r\n    /* c8 ignore next */\r\n    await callbacks?.onBeforeInsert(element);\r\n    try {\r\n      return this._imodel.elements.getElement(element.insert());\r\n    } finally {\r\n      /* c8 ignore next */\r\n      await callbacks?.onAfterInsert(element);\r\n    }\r\n  }\r\n\r\n  private async insertModel(props: ModelProps, callbacks?: InsertCallbacks): Promise<Model> {\r\n    const model = this._imodel.models.createModel(props);\r\n    /* c8 ignore next */\r\n    await callbacks?.onBeforeInsert(model);\r\n    try {\r\n      model.id = model.insert();\r\n      return model;\r\n    } finally {\r\n      /* c8 ignore next */\r\n      await callbacks?.onAfterInsert(model);\r\n    }\r\n  }\r\n\r\n  private async updateElement(element: Element, callbacks?: UpdateCallbacks) {\r\n    /* c8 ignore next */\r\n    await callbacks?.onBeforeUpdate(element);\r\n    try {\r\n      element.update();\r\n    } finally {\r\n      /* c8 ignore next */\r\n      await callbacks?.onAfterUpdate(element);\r\n    }\r\n  }\r\n}\r\n\r\nfunction normalizeRulesetInsertOptions(options?: RulesetInsertOptions): RulesetInsertOptions {\r\n  if (options === undefined) {\r\n    return { skip: \"same-id-and-version-eq\", replaceVersions: \"exact\" };\r\n  }\r\n\r\n  return {\r\n    skip: options.skip ?? \"same-id-and-version-eq\",\r\n    replaceVersions: options.replaceVersions ?? \"exact\",\r\n    onEntityUpdate: options.onEntityUpdate,\r\n    onEntityInsert: options.onEntityInsert,\r\n  };\r\n}\r\n"]}

package/lib/cjs/presentation-backend/RulesetManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"RulesetManager.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetManager.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAEH,oEAAwE;AA6BxE;;;GAGG;AACH,MAAa,kBAAkB;IACrB,kBAAkB,CAAiC;IACnD,mBAAmB,GAAG,IAAI,GAAG,EAA6B,CAAC;IAEnE,YAAY,iBAAiD;QAC3D,IAAI,CAAC,kBAAkB,GAAG,iBAAiB,CAAC;IAC9C,CAAC;IAED;;OAEG;IACI,GAAG,CAAC,EAAU;QACnB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,uBAAuB,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;QACjF,MAAM,aAAa,GAA0B,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACjF,IAAI,CAAC,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;YAC/B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IACjI,CAAC;IAED;;OAEG;IACI,GAAG,CAAC,OAAgB;QACzB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9D,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,OAA6C;QACzD,IAAI,SAAS,EAAE,iBAAyB,CAAC;QACzC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACvB,iBAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YACvB,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,MAAM,CAAC;IACtF,CAAC;IAED;;OAEG;IACI,KAAK;QACV,IAAI,CAAC,kBAAkB,EAAE,CAAC,aAAa,EAAE,CAAC;QAC1C,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;IAEO,WAAW,CAAC,OAAgB,EAAE,IAAY,EAAE,WAAiD;QACnG,MAAM,UAAU,GAAG,IAAI,uCAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACxD,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AApED,gDAoEC","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 Core\n */\n\nimport { RegisteredRuleset, Ruleset } from \"@itwin/presentation-common\";\nimport { NativePlatformDefinition } from \"./NativePlatform.js\";\n\n/**\n * Presentation ruleset registry.\n * @public\n */\nexport interface RulesetManager {\n  /**\n   * Get a ruleset with the specified id.\n   */\n  get(id: string): RegisteredRuleset | undefined;\n\n  /**\n   * Register the supplied ruleset\n   */\n  add(ruleset: Ruleset): RegisteredRuleset;\n\n  /**\n   * Unregister the supplied ruleset\n   */\n  remove(ruleset: RegisteredRuleset | [string, string]): boolean;\n\n  /**\n   * Remove all rulesets registered in this session.\n   */\n  clear(): void;\n}\n\n/**\n * Presentation ruleset registry implementation.\n * @internal\n */\nexport class RulesetManagerImpl implements RulesetManager {\n  private _getNativePlatform: () => NativePlatformDefinition;\n  private _registeredRulesets = new Map<string, RegisteredRuleset>();\n\n  constructor(getNativePlatform: () => NativePlatformDefinition) {\n    this._getNativePlatform = getNativePlatform;\n  }\n\n  /**\n   * Get a ruleset with the specified id.\n   */\n  public get(id: string): RegisteredRuleset | undefined {\n    const foundRuleset = this._registeredRulesets.get(id);\n    if (foundRuleset) {\n      return foundRuleset;\n    }\n\n    const serializedRulesetsArray = this._getNativePlatform().getRulesets(id).result;\n    const rulesetsArray: RulesetResponseJson[] = JSON.parse(serializedRulesetsArray);\n    if (0 === rulesetsArray.length) {\n      return undefined;\n    }\n    return this.saveRuleset(rulesetsArray[0].ruleset, rulesetsArray[0].hash, (ruleset: RegisteredRuleset) => this.remove(ruleset));\n  }\n\n  /**\n   * Register the supplied ruleset\n   */\n  public add(ruleset: Ruleset): RegisteredRuleset {\n    const foundRuleset = this._registeredRulesets.get(ruleset.id);\n    if (foundRuleset) {\n      return foundRuleset;\n    }\n\n    const hash = this._getNativePlatform().addRuleset(JSON.stringify(ruleset)).result;\n    return this.saveRuleset(ruleset, hash, (r: RegisteredRuleset) => this.remove(r));\n  }\n\n  /**\n   * Unregister the supplied ruleset\n   */\n  public remove(ruleset: RegisteredRuleset | [string, string]): boolean {\n    let rulesetId, rulesetIdentifier: string;\n    if (Array.isArray(ruleset)) {\n      rulesetId = ruleset[0];\n      rulesetIdentifier = ruleset[1];\n    } else {\n      rulesetId = ruleset.id;\n      rulesetIdentifier = ruleset.uniqueIdentifier;\n    }\n\n    this._registeredRulesets.delete(rulesetId);\n    return this._getNativePlatform().removeRuleset(rulesetId, rulesetIdentifier).result;\n  }\n\n  /**\n   * Remove all rulesets registered in this session.\n   */\n  public clear(): void {\n    this._getNativePlatform().clearRulesets();\n    this._registeredRulesets.clear();\n  }\n\n  private saveRuleset(ruleset: Ruleset, hash: string, disposeFunc: (ruleset: RegisteredRuleset) => void) {\n    const newRuleset = new RegisteredRuleset(ruleset, hash, disposeFunc);\n    this._registeredRulesets.set(newRuleset.id, newRuleset);\n    return newRuleset;\n  }\n}\n\ninterface RulesetResponseJson {\n  ruleset: Ruleset;\n  hash: string;\n}\n"]}
+{"version":3,"file":"RulesetManager.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetManager.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAEH,oEAAwE;AA6BxE;;;GAGG;AACH,MAAa,kBAAkB;IACrB,kBAAkB,CAAiC;IACnD,mBAAmB,GAAG,IAAI,GAAG,EAA6B,CAAC;IAEnE,YAAY,iBAAiD;QAC3D,IAAI,CAAC,kBAAkB,GAAG,iBAAiB,CAAC;IAC9C,CAAC;IAED;;OAEG;IACI,GAAG,CAAC,EAAU;QACnB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,EAAE,CAAC,CAAC;QACtD,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,uBAAuB,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,MAAM,CAAC;QACjF,MAAM,aAAa,GAA0B,IAAI,CAAC,KAAK,CAAC,uBAAuB,CAAC,CAAC;QACjF,IAAI,CAAC,KAAK,aAAa,CAAC,MAAM,EAAE,CAAC;YAC/B,OAAO,SAAS,CAAC;QACnB,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC,aAAa,CAAC,CAAC,CAAC,CAAC,OAAO,EAAE,aAAa,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,OAA0B,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC,CAAC;IACjI,CAAC;IAED;;OAEG;IACI,GAAG,CAAC,OAAgB;QACzB,MAAM,YAAY,GAAG,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC,CAAC;QAC9D,IAAI,YAAY,EAAE,CAAC;YACjB,OAAO,YAAY,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC;QAClF,OAAO,IAAI,CAAC,WAAW,CAAC,OAAO,EAAE,IAAI,EAAE,CAAC,CAAoB,EAAE,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,CAAC,CAAC,CAAC;IACnF,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,OAA6C;QACzD,IAAI,SAAS,EAAE,iBAAyB,CAAC;QACzC,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,EAAE,CAAC;YAC3B,SAAS,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YACvB,iBAAiB,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;QACjC,CAAC;aAAM,CAAC;YACN,SAAS,GAAG,OAAO,CAAC,EAAE,CAAC;YACvB,iBAAiB,GAAG,OAAO,CAAC,gBAAgB,CAAC;QAC/C,CAAC;QAED,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;QAC3C,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,aAAa,CAAC,SAAS,EAAE,iBAAiB,CAAC,CAAC,MAAM,CAAC;IACtF,CAAC;IAED;;OAEG;IACI,KAAK;QACV,IAAI,CAAC,kBAAkB,EAAE,CAAC,aAAa,EAAE,CAAC;QAC1C,IAAI,CAAC,mBAAmB,CAAC,KAAK,EAAE,CAAC;IACnC,CAAC;IAEO,WAAW,CAAC,OAAgB,EAAE,IAAY,EAAE,WAAiD;QACnG,MAAM,UAAU,GAAG,IAAI,uCAAiB,CAAC,OAAO,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QACrE,IAAI,CAAC,mBAAmB,CAAC,GAAG,CAAC,UAAU,CAAC,EAAE,EAAE,UAAU,CAAC,CAAC;QACxD,OAAO,UAAU,CAAC;IACpB,CAAC;CACF;AApED,gDAoEC","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 Core\r\n */\r\n\r\nimport { RegisteredRuleset, Ruleset } from \"@itwin/presentation-common\";\r\nimport { NativePlatformDefinition } from \"./NativePlatform.js\";\r\n\r\n/**\r\n * Presentation ruleset registry.\r\n * @public\r\n */\r\nexport interface RulesetManager {\r\n  /**\r\n   * Get a ruleset with the specified id.\r\n   */\r\n  get(id: string): RegisteredRuleset | undefined;\r\n\r\n  /**\r\n   * Register the supplied ruleset\r\n   */\r\n  add(ruleset: Ruleset): RegisteredRuleset;\r\n\r\n  /**\r\n   * Unregister the supplied ruleset\r\n   */\r\n  remove(ruleset: RegisteredRuleset | [string, string]): boolean;\r\n\r\n  /**\r\n   * Remove all rulesets registered in this session.\r\n   */\r\n  clear(): void;\r\n}\r\n\r\n/**\r\n * Presentation ruleset registry implementation.\r\n * @internal\r\n */\r\nexport class RulesetManagerImpl implements RulesetManager {\r\n  private _getNativePlatform: () => NativePlatformDefinition;\r\n  private _registeredRulesets = new Map<string, RegisteredRuleset>();\r\n\r\n  constructor(getNativePlatform: () => NativePlatformDefinition) {\r\n    this._getNativePlatform = getNativePlatform;\r\n  }\r\n\r\n  /**\r\n   * Get a ruleset with the specified id.\r\n   */\r\n  public get(id: string): RegisteredRuleset | undefined {\r\n    const foundRuleset = this._registeredRulesets.get(id);\r\n    if (foundRuleset) {\r\n      return foundRuleset;\r\n    }\r\n\r\n    const serializedRulesetsArray = this._getNativePlatform().getRulesets(id).result;\r\n    const rulesetsArray: RulesetResponseJson[] = JSON.parse(serializedRulesetsArray);\r\n    if (0 === rulesetsArray.length) {\r\n      return undefined;\r\n    }\r\n    return this.saveRuleset(rulesetsArray[0].ruleset, rulesetsArray[0].hash, (ruleset: RegisteredRuleset) => this.remove(ruleset));\r\n  }\r\n\r\n  /**\r\n   * Register the supplied ruleset\r\n   */\r\n  public add(ruleset: Ruleset): RegisteredRuleset {\r\n    const foundRuleset = this._registeredRulesets.get(ruleset.id);\r\n    if (foundRuleset) {\r\n      return foundRuleset;\r\n    }\r\n\r\n    const hash = this._getNativePlatform().addRuleset(JSON.stringify(ruleset)).result;\r\n    return this.saveRuleset(ruleset, hash, (r: RegisteredRuleset) => this.remove(r));\r\n  }\r\n\r\n  /**\r\n   * Unregister the supplied ruleset\r\n   */\r\n  public remove(ruleset: RegisteredRuleset | [string, string]): boolean {\r\n    let rulesetId, rulesetIdentifier: string;\r\n    if (Array.isArray(ruleset)) {\r\n      rulesetId = ruleset[0];\r\n      rulesetIdentifier = ruleset[1];\r\n    } else {\r\n      rulesetId = ruleset.id;\r\n      rulesetIdentifier = ruleset.uniqueIdentifier;\r\n    }\r\n\r\n    this._registeredRulesets.delete(rulesetId);\r\n    return this._getNativePlatform().removeRuleset(rulesetId, rulesetIdentifier).result;\r\n  }\r\n\r\n  /**\r\n   * Remove all rulesets registered in this session.\r\n   */\r\n  public clear(): void {\r\n    this._getNativePlatform().clearRulesets();\r\n    this._registeredRulesets.clear();\r\n  }\r\n\r\n  private saveRuleset(ruleset: Ruleset, hash: string, disposeFunc: (ruleset: RegisteredRuleset) => void) {\r\n    const newRuleset = new RegisteredRuleset(ruleset, hash, disposeFunc);\r\n    this._registeredRulesets.set(newRuleset.id, newRuleset);\r\n    return newRuleset;\r\n  }\r\n}\r\n\r\ninterface RulesetResponseJson {\r\n  ruleset: Ruleset;\r\n  hash: string;\r\n}\r\n"]}

package/lib/cjs/presentation-backend/RulesetVariablesManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"RulesetVariablesManager.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetVariablesManager.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAGH,oEAA+E;AA4E/E;;;GAGG;AACH,MAAa,2BAA2B;IAC9B,kBAAkB,CAAiC;IACnD,UAAU,CAAS;IAE3B,YAAY,cAA8C,EAAE,SAAiB;QAC3E,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAEM,gBAAgB,CAAC,UAAkB,EAAE,IAAwB,EAAE,KAAoB;QACxF,IAAI,CAAC,kBAAkB,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAC9F,CAAC;IAEM,gBAAgB,CAAC,UAAkB,EAAE,IAAwB;QAClE,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,MAAM,CAAC;IACrG,CAAC;IAEM,QAAQ,CAAC,UAAkB,EAAE,IAAwB,EAAE,KAAoB;QAChF,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAgB,CAAC,CAAC;YACpD,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAmB,CAAC,CAAC;YACvD,KAAK,wCAAkB,CAAC,SAAS;gBAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAqB,CAAC,CAAC;YAC1D,KAAK,wCAAkB,CAAC,GAAG;gBACzB,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,KAAe,CAAC,CAAC;YAClD,KAAK,wCAAkB,CAAC,QAAQ;gBAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAiB,CAAC,CAAC;YACrD,KAAK,wCAAkB,CAAC,MAAM;gBAC5B,OAAO,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,KAAe,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,UAAkB;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,QAAQ,CAAC,UAAkB,EAAE,IAAwB;QAC1D,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,SAAS;gBAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnC,KAAK,wCAAkB,CAAC,GAAG;gBACzB,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACjC,KAAK,wCAAkB,CAAC,QAAQ;gBAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,MAAM;gBAC5B,OAAO,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,SAAS,CAAC,UAAkB;QACjC,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,MAAM,CAAW,CAAC;IAChF,CAAC;IAED;;OAEG;IACI,SAAS,CAAC,UAAkB,EAAE,KAAa;QAChD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACtE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,CAAY,CAAC;IAC/E,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAc;QAC/C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACpE,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,UAAkB;QAC9B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,GAAG,CAAW,CAAC;IAC7E,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,UAAkB,EAAE,KAAa;QAC7C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACnE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,QAAQ,CAAa,CAAC;IACpF,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAe;QAChD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACxE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,CAAe,CAAC;IAClF,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAiB;QAClD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACpE,CAAC;IAED;;;OAGG;IACI,QAAQ,CAAC,UAAkB;QAChC,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,SAAS,CAAiB,CAAC;IACzF,CAAC;IAED;;OAEG;IACI,QAAQ,CAAC,UAAkB,EAAE,KAAmB;QACrD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC;CACF;AAhJD,kEAgJC","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 Core\n */\n\nimport { Id64String } from \"@itwin/core-bentley\";\nimport { VariableValue, VariableValueTypes } from \"@itwin/presentation-common\";\nimport { NativePlatformDefinition } from \"./NativePlatform.js\";\n\n/**\n * Presentation ruleset variables registry.\n * @public\n */\nexport interface RulesetVariablesManager {\n  /** Gets a value of the specified type and ID */\n  getValue(variableId: string, type: VariableValueTypes): VariableValue;\n  /** Sets a value of the specified type */\n  setValue(variableId: string, type: VariableValueTypes, value: VariableValue): void;\n  /** Unsets variable with given id. */\n  unset(variableId: string): void;\n\n  /**\n   * Retrieves `string` variable value.\n   * Returns empty string if variable does not exist or does not convert to string.\n   */\n  getString(variableId: string): string;\n  /**\n   * Sets `string` variable value\n   */\n  setString(variableId: string, value: string): void;\n\n  /**\n   * Retrieves `boolean` variable value.\n   * Returns `false` if variable does not exist or does not convert to boolean.\n   */\n  getBool(variableId: string): boolean;\n  /**\n   * Sets `boolean` variable value\n   */\n  setBool(variableId: string, value: boolean): void;\n\n  /**\n   * Retrieves `number` variable value.\n   * Returns `0` if variable does not exist or does not convert to integer.\n   */\n  getInt(variableId: string): number;\n  /**\n   * Sets `number` variable value\n   */\n  setInt(variableId: string, value: number): void;\n\n  /**\n   * Retrieves `number[]` variable value.\n   * Returns empty array if variable does not exist or does not convert to integer array.\n   */\n  getInts(variableId: string): number[];\n  /**\n   * Sets `number[]` variable value\n   */\n  setInts(variableId: string, value: number[]): void;\n\n  /**\n   * Retrieves `Id64String` variable value.\n   * Returns invalid Id64String if variable does not exist or does not convert to Id64String.\n   */\n  getId64(variableId: string): Id64String;\n  /**\n   * Sets `Id64String` variable value\n   */\n  setId64(variableId: string, value: Id64String): void;\n\n  /**\n   * Retrieves `Id64String[]` variable value.\n   * Returns empty array if variable does not exist or does not convert to Id64String array.\n   */\n  getId64s(variableId: string): Id64String[];\n  /**\n   * Sets `Id64String[]` variable value\n   */\n  setId64s(variableId: string, value: Id64String[]): void;\n}\n\n/**\n * Presentation ruleset variables registry implementation.\n * @internal\n */\nexport class RulesetVariablesManagerImpl implements RulesetVariablesManager {\n  private _getNativePlatform: () => NativePlatformDefinition;\n  private _rulesetId: string;\n\n  constructor(getNativeAddon: () => NativePlatformDefinition, rulesetId: string) {\n    this._getNativePlatform = getNativeAddon;\n    this._rulesetId = rulesetId;\n  }\n\n  public setValueInternal(variableId: string, type: VariableValueTypes, value: VariableValue): void {\n    this._getNativePlatform().setRulesetVariableValue(this._rulesetId, variableId, type, value);\n  }\n\n  public getValueInternal(variableId: string, type: VariableValueTypes): VariableValue {\n    return this._getNativePlatform().getRulesetVariableValue(this._rulesetId, variableId, type).result;\n  }\n\n  public setValue(variableId: string, type: VariableValueTypes, value: VariableValue): void {\n    switch (type) {\n      case VariableValueTypes.Bool:\n        return this.setBool(variableId, value as boolean);\n      case VariableValueTypes.Id64:\n        return this.setId64(variableId, value as Id64String);\n      case VariableValueTypes.Id64Array:\n        return this.setId64s(variableId, value as Id64String[]);\n      case VariableValueTypes.Int:\n        return this.setInt(variableId, value as number);\n      case VariableValueTypes.IntArray:\n        return this.setInts(variableId, value as number[]);\n      case VariableValueTypes.String:\n        return this.setString(variableId, value as string);\n    }\n  }\n\n  public unset(variableId: string): void {\n    this._getNativePlatform().unsetRulesetVariableValue(this._rulesetId, variableId);\n  }\n\n  public getValue(variableId: string, type: VariableValueTypes): VariableValue {\n    switch (type) {\n      case VariableValueTypes.Bool:\n        return this.getBool(variableId);\n      case VariableValueTypes.Id64:\n        return this.getId64(variableId);\n      case VariableValueTypes.Id64Array:\n        return this.getId64s(variableId);\n      case VariableValueTypes.Int:\n        return this.getInt(variableId);\n      case VariableValueTypes.IntArray:\n        return this.getInts(variableId);\n      case VariableValueTypes.String:\n        return this.getString(variableId);\n    }\n  }\n\n  /**\n   * Retrieves `string` variable value.\n   * Returns empty string if variable does not exist or does not convert to string.\n   */\n  public getString(variableId: string): string {\n    return this.getValueInternal(variableId, VariableValueTypes.String) as string;\n  }\n\n  /**\n   * Sets `string` variable value\n   */\n  public setString(variableId: string, value: string): void {\n    this.setValueInternal(variableId, VariableValueTypes.String, value);\n  }\n\n  /**\n   * Retrieves `boolean` variable value.\n   * Returns `false` if variable does not exist or does not convert to boolean.\n   */\n  public getBool(variableId: string): boolean {\n    return this.getValueInternal(variableId, VariableValueTypes.Bool) as boolean;\n  }\n\n  /**\n   * Sets `boolean` variable value\n   */\n  public setBool(variableId: string, value: boolean): void {\n    this.setValueInternal(variableId, VariableValueTypes.Bool, value);\n  }\n\n  /**\n   * Retrieves `number` variable value.\n   * Returns `0` if variable does not exist or does not convert to integer.\n   */\n  public getInt(variableId: string): number {\n    return this.getValueInternal(variableId, VariableValueTypes.Int) as number;\n  }\n\n  /**\n   * Sets `number` variable value\n   */\n  public setInt(variableId: string, value: number): void {\n    this.setValueInternal(variableId, VariableValueTypes.Int, value);\n  }\n\n  /**\n   * Retrieves `number[]` variable value.\n   * Returns empty array if variable does not exist or does not convert to integer array.\n   */\n  public getInts(variableId: string): number[] {\n    return this.getValueInternal(variableId, VariableValueTypes.IntArray) as number[];\n  }\n\n  /**\n   * Sets `number[]` variable value\n   */\n  public setInts(variableId: string, value: number[]): void {\n    this.setValueInternal(variableId, VariableValueTypes.IntArray, value);\n  }\n\n  /**\n   * Retrieves `Id64String` variable value.\n   * Returns invalid Id64String if variable does not exist or does not convert to Id64String.\n   */\n  public getId64(variableId: string): Id64String {\n    return this.getValueInternal(variableId, VariableValueTypes.Id64) as Id64String;\n  }\n\n  /**\n   * Sets `Id64String` variable value\n   */\n  public setId64(variableId: string, value: Id64String): void {\n    this.setValueInternal(variableId, VariableValueTypes.Id64, value);\n  }\n\n  /**\n   * Retrieves `Id64String[]` variable value.\n   * Returns empty array if variable does not exist or does not convert to Id64String array.\n   */\n  public getId64s(variableId: string): Id64String[] {\n    return this.getValueInternal(variableId, VariableValueTypes.Id64Array) as Id64String[];\n  }\n\n  /**\n   * Sets `Id64String[]` variable value\n   */\n  public setId64s(variableId: string, value: Id64String[]): void {\n    this.setValueInternal(variableId, VariableValueTypes.Id64Array, value);\n  }\n}\n"]}
+{"version":3,"file":"RulesetVariablesManager.js","sourceRoot":"","sources":["../../../src/presentation-backend/RulesetVariablesManager.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG;;GAEG;;;AAGH,oEAA+E;AA4E/E;;;GAGG;AACH,MAAa,2BAA2B;IAC9B,kBAAkB,CAAiC;IACnD,UAAU,CAAS;IAE3B,YAAY,cAA8C,EAAE,SAAiB;QAC3E,IAAI,CAAC,kBAAkB,GAAG,cAAc,CAAC;QACzC,IAAI,CAAC,UAAU,GAAG,SAAS,CAAC;IAC9B,CAAC;IAEM,gBAAgB,CAAC,UAAkB,EAAE,IAAwB,EAAE,KAAoB;QACxF,IAAI,CAAC,kBAAkB,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,EAAE,KAAK,CAAC,CAAC;IAC9F,CAAC;IAEM,gBAAgB,CAAC,UAAkB,EAAE,IAAwB;QAClE,OAAO,IAAI,CAAC,kBAAkB,EAAE,CAAC,uBAAuB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,EAAE,IAAI,CAAC,CAAC,MAAM,CAAC;IACrG,CAAC;IAEM,QAAQ,CAAC,UAAkB,EAAE,IAAwB,EAAE,KAAoB;QAChF,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAgB,CAAC,CAAC;YACpD,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAmB,CAAC,CAAC;YACvD,KAAK,wCAAkB,CAAC,SAAS;gBAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,EAAE,KAAqB,CAAC,CAAC;YAC1D,KAAK,wCAAkB,CAAC,GAAG;gBACzB,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,EAAE,KAAe,CAAC,CAAC;YAClD,KAAK,wCAAkB,CAAC,QAAQ;gBAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,EAAE,KAAiB,CAAC,CAAC;YACrD,KAAK,wCAAkB,CAAC,MAAM;gBAC5B,OAAO,IAAI,CAAC,SAAS,CAAC,UAAU,EAAE,KAAe,CAAC,CAAC;QACvD,CAAC;IACH,CAAC;IAEM,KAAK,CAAC,UAAkB;QAC7B,IAAI,CAAC,kBAAkB,EAAE,CAAC,yBAAyB,CAAC,IAAI,CAAC,UAAU,EAAE,UAAU,CAAC,CAAC;IACnF,CAAC;IAEM,QAAQ,CAAC,UAAkB,EAAE,IAAwB;QAC1D,QAAQ,IAAI,EAAE,CAAC;YACb,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,IAAI;gBAC1B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,SAAS;gBAC/B,OAAO,IAAI,CAAC,QAAQ,CAAC,UAAU,CAAC,CAAC;YACnC,KAAK,wCAAkB,CAAC,GAAG;gBACzB,OAAO,IAAI,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC;YACjC,KAAK,wCAAkB,CAAC,QAAQ;gBAC9B,OAAO,IAAI,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;YAClC,KAAK,wCAAkB,CAAC,MAAM;gBAC5B,OAAO,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;QACtC,CAAC;IACH,CAAC;IAED;;;OAGG;IACI,SAAS,CAAC,UAAkB;QACjC,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,MAAM,CAAW,CAAC;IAChF,CAAC;IAED;;OAEG;IACI,SAAS,CAAC,UAAkB,EAAE,KAAa;QAChD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,MAAM,EAAE,KAAK,CAAC,CAAC;IACtE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,CAAY,CAAC;IAC/E,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAc;QAC/C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACpE,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,UAAkB;QAC9B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,GAAG,CAAW,CAAC;IAC7E,CAAC;IAED;;OAEG;IACI,MAAM,CAAC,UAAkB,EAAE,KAAa;QAC7C,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;IACnE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,QAAQ,CAAa,CAAC;IACpF,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAe;QAChD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,QAAQ,EAAE,KAAK,CAAC,CAAC;IACxE,CAAC;IAED;;;OAGG;IACI,OAAO,CAAC,UAAkB;QAC/B,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,CAAe,CAAC;IAClF,CAAC;IAED;;OAEG;IACI,OAAO,CAAC,UAAkB,EAAE,KAAiB;QAClD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,IAAI,EAAE,KAAK,CAAC,CAAC;IACpE,CAAC;IAED;;;OAGG;IACI,QAAQ,CAAC,UAAkB;QAChC,OAAO,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,SAAS,CAAiB,CAAC;IACzF,CAAC;IAED;;OAEG;IACI,QAAQ,CAAC,UAAkB,EAAE,KAAmB;QACrD,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,wCAAkB,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IACzE,CAAC;CACF;AAhJD,kEAgJC","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 Core\r\n */\r\n\r\nimport { Id64String } from \"@itwin/core-bentley\";\r\nimport { VariableValue, VariableValueTypes } from \"@itwin/presentation-common\";\r\nimport { NativePlatformDefinition } from \"./NativePlatform.js\";\r\n\r\n/**\r\n * Presentation ruleset variables registry.\r\n * @public\r\n */\r\nexport interface RulesetVariablesManager {\r\n  /** Gets a value of the specified type and ID */\r\n  getValue(variableId: string, type: VariableValueTypes): VariableValue;\r\n  /** Sets a value of the specified type */\r\n  setValue(variableId: string, type: VariableValueTypes, value: VariableValue): void;\r\n  /** Unsets variable with given id. */\r\n  unset(variableId: string): void;\r\n\r\n  /**\r\n   * Retrieves `string` variable value.\r\n   * Returns empty string if variable does not exist or does not convert to string.\r\n   */\r\n  getString(variableId: string): string;\r\n  /**\r\n   * Sets `string` variable value\r\n   */\r\n  setString(variableId: string, value: string): void;\r\n\r\n  /**\r\n   * Retrieves `boolean` variable value.\r\n   * Returns `false` if variable does not exist or does not convert to boolean.\r\n   */\r\n  getBool(variableId: string): boolean;\r\n  /**\r\n   * Sets `boolean` variable value\r\n   */\r\n  setBool(variableId: string, value: boolean): void;\r\n\r\n  /**\r\n   * Retrieves `number` variable value.\r\n   * Returns `0` if variable does not exist or does not convert to integer.\r\n   */\r\n  getInt(variableId: string): number;\r\n  /**\r\n   * Sets `number` variable value\r\n   */\r\n  setInt(variableId: string, value: number): void;\r\n\r\n  /**\r\n   * Retrieves `number[]` variable value.\r\n   * Returns empty array if variable does not exist or does not convert to integer array.\r\n   */\r\n  getInts(variableId: string): number[];\r\n  /**\r\n   * Sets `number[]` variable value\r\n   */\r\n  setInts(variableId: string, value: number[]): void;\r\n\r\n  /**\r\n   * Retrieves `Id64String` variable value.\r\n   * Returns invalid Id64String if variable does not exist or does not convert to Id64String.\r\n   */\r\n  getId64(variableId: string): Id64String;\r\n  /**\r\n   * Sets `Id64String` variable value\r\n   */\r\n  setId64(variableId: string, value: Id64String): void;\r\n\r\n  /**\r\n   * Retrieves `Id64String[]` variable value.\r\n   * Returns empty array if variable does not exist or does not convert to Id64String array.\r\n   */\r\n  getId64s(variableId: string): Id64String[];\r\n  /**\r\n   * Sets `Id64String[]` variable value\r\n   */\r\n  setId64s(variableId: string, value: Id64String[]): void;\r\n}\r\n\r\n/**\r\n * Presentation ruleset variables registry implementation.\r\n * @internal\r\n */\r\nexport class RulesetVariablesManagerImpl implements RulesetVariablesManager {\r\n  private _getNativePlatform: () => NativePlatformDefinition;\r\n  private _rulesetId: string;\r\n\r\n  constructor(getNativeAddon: () => NativePlatformDefinition, rulesetId: string) {\r\n    this._getNativePlatform = getNativeAddon;\r\n    this._rulesetId = rulesetId;\r\n  }\r\n\r\n  public setValueInternal(variableId: string, type: VariableValueTypes, value: VariableValue): void {\r\n    this._getNativePlatform().setRulesetVariableValue(this._rulesetId, variableId, type, value);\r\n  }\r\n\r\n  public getValueInternal(variableId: string, type: VariableValueTypes): VariableValue {\r\n    return this._getNativePlatform().getRulesetVariableValue(this._rulesetId, variableId, type).result;\r\n  }\r\n\r\n  public setValue(variableId: string, type: VariableValueTypes, value: VariableValue): void {\r\n    switch (type) {\r\n      case VariableValueTypes.Bool:\r\n        return this.setBool(variableId, value as boolean);\r\n      case VariableValueTypes.Id64:\r\n        return this.setId64(variableId, value as Id64String);\r\n      case VariableValueTypes.Id64Array:\r\n        return this.setId64s(variableId, value as Id64String[]);\r\n      case VariableValueTypes.Int:\r\n        return this.setInt(variableId, value as number);\r\n      case VariableValueTypes.IntArray:\r\n        return this.setInts(variableId, value as number[]);\r\n      case VariableValueTypes.String:\r\n        return this.setString(variableId, value as string);\r\n    }\r\n  }\r\n\r\n  public unset(variableId: string): void {\r\n    this._getNativePlatform().unsetRulesetVariableValue(this._rulesetId, variableId);\r\n  }\r\n\r\n  public getValue(variableId: string, type: VariableValueTypes): VariableValue {\r\n    switch (type) {\r\n      case VariableValueTypes.Bool:\r\n        return this.getBool(variableId);\r\n      case VariableValueTypes.Id64:\r\n        return this.getId64(variableId);\r\n      case VariableValueTypes.Id64Array:\r\n        return this.getId64s(variableId);\r\n      case VariableValueTypes.Int:\r\n        return this.getInt(variableId);\r\n      case VariableValueTypes.IntArray:\r\n        return this.getInts(variableId);\r\n      case VariableValueTypes.String:\r\n        return this.getString(variableId);\r\n    }\r\n  }\r\n\r\n  /**\r\n   * Retrieves `string` variable value.\r\n   * Returns empty string if variable does not exist or does not convert to string.\r\n   */\r\n  public getString(variableId: string): string {\r\n    return this.getValueInternal(variableId, VariableValueTypes.String) as string;\r\n  }\r\n\r\n  /**\r\n   * Sets `string` variable value\r\n   */\r\n  public setString(variableId: string, value: string): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.String, value);\r\n  }\r\n\r\n  /**\r\n   * Retrieves `boolean` variable value.\r\n   * Returns `false` if variable does not exist or does not convert to boolean.\r\n   */\r\n  public getBool(variableId: string): boolean {\r\n    return this.getValueInternal(variableId, VariableValueTypes.Bool) as boolean;\r\n  }\r\n\r\n  /**\r\n   * Sets `boolean` variable value\r\n   */\r\n  public setBool(variableId: string, value: boolean): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.Bool, value);\r\n  }\r\n\r\n  /**\r\n   * Retrieves `number` variable value.\r\n   * Returns `0` if variable does not exist or does not convert to integer.\r\n   */\r\n  public getInt(variableId: string): number {\r\n    return this.getValueInternal(variableId, VariableValueTypes.Int) as number;\r\n  }\r\n\r\n  /**\r\n   * Sets `number` variable value\r\n   */\r\n  public setInt(variableId: string, value: number): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.Int, value);\r\n  }\r\n\r\n  /**\r\n   * Retrieves `number[]` variable value.\r\n   * Returns empty array if variable does not exist or does not convert to integer array.\r\n   */\r\n  public getInts(variableId: string): number[] {\r\n    return this.getValueInternal(variableId, VariableValueTypes.IntArray) as number[];\r\n  }\r\n\r\n  /**\r\n   * Sets `number[]` variable value\r\n   */\r\n  public setInts(variableId: string, value: number[]): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.IntArray, value);\r\n  }\r\n\r\n  /**\r\n   * Retrieves `Id64String` variable value.\r\n   * Returns invalid Id64String if variable does not exist or does not convert to Id64String.\r\n   */\r\n  public getId64(variableId: string): Id64String {\r\n    return this.getValueInternal(variableId, VariableValueTypes.Id64) as Id64String;\r\n  }\r\n\r\n  /**\r\n   * Sets `Id64String` variable value\r\n   */\r\n  public setId64(variableId: string, value: Id64String): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.Id64, value);\r\n  }\r\n\r\n  /**\r\n   * Retrieves `Id64String[]` variable value.\r\n   * Returns empty array if variable does not exist or does not convert to Id64String array.\r\n   */\r\n  public getId64s(variableId: string): Id64String[] {\r\n    return this.getValueInternal(variableId, VariableValueTypes.Id64Array) as Id64String[];\r\n  }\r\n\r\n  /**\r\n   * Sets `Id64String[]` variable value\r\n   */\r\n  public setId64s(variableId: string, value: Id64String[]): void {\r\n    this.setValueInternal(variableId, VariableValueTypes.Id64Array, value);\r\n  }\r\n}\r\n"]}

package/lib/cjs/presentation-backend/SelectionScopesHelper.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScopesHelper.d.ts","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAoB,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAEjE,OAAO,EACL,8BAA8B,EAG9B,MAAM,EAGN,cAAc,EACf,MAAM,4BAA4B,CAAC;AAGpC;;;;;GAKG;AACH,qBAAa,qBAAqB;IAEhC,OAAO;WAEO,kBAAkB,IAAI,cAAc,EAAE;IAgBpD,OAAO,CAAC,MAAM,CAAC,aAAa;IAW5B,OAAO,CAAC,MAAM,CAAC,uBAAuB;IAWtC,OAAO,CAAC,MAAM,CAAC,wBAAwB;IAcvC,OAAO,CAAC,MAAM,CAAC,qBAAqB;IAcpC,OAAO,CAAC,MAAM,CAAC,8BAA8B;IAuB7C,OAAO,CAAC,MAAM,CAAC,oCAAoC;IAYnD,OAAO,CAAC,MAAM,CAAC,uBAAuB;IAgBtC,OAAO,CAAC,MAAM,CAAC,iCAAiC;IA2BhD,OAAO,CAAC,MAAM,CAAC,kCAAkC;IA6BjD,OAAO,CAAC,MAAM,CAAC,qCAAqC;WA6BhC,gBAAgB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,8BAA8B,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CAsB/H"}
+{"version":3,"file":"SelectionScopesHelper.d.ts","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":"AAKA;;GAEG;AAEH,OAAO,EAAoB,QAAQ,EAAE,MAAM,qBAAqB,CAAC;AAGjE,OAAO,EACL,8BAA8B,EAG9B,MAAM,EAGN,cAAc,EACf,MAAM,4BAA4B,CAAC;AAGpC;;;;;GAKG;AACH,qBAAa,qBAAqB;IAEhC,OAAO;WAEO,kBAAkB,IAAI,cAAc,EAAE;IAgBpD,OAAO,CAAC,MAAM,CAAC,aAAa;mBAWP,uBAAuB;mBASvB,wBAAwB;mBAYxB,qBAAqB;mBAYrB,8BAA8B;mBAuB9B,oCAAoC;mBAYpC,uBAAuB;mBAoBvB,iCAAiC;mBAyBjC,kCAAkC;mBA2BlC,qCAAqC;WA2BtC,gBAAgB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAAE,EAAE,8BAA8B,CAAC,QAAQ,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC;CAsB/H"}

package/lib/cjs/presentation-backend/SelectionScopesHelper.js

@@ -10,6 +10,7 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.SelectionScopesHelper = void 0;
 const core_bentley_1 = require("@itwin/core-bentley");
+const core_common_1 = require("@itwin/core-common");
 const presentation_common_1 = require("@itwin/presentation-common");
 const Utils_js_1 = require("./Utils.js");
 /**
@@ -46,62 +47,60 @@ class SelectionScopesHelper {
         }
         return (0, Utils_js_1.getElementKey)(iModel, currId);
     }
-    static computeElementSelection(iModel, elementIds, ancestorLevel) {
+    static async computeElementSelection(iModel, elementIds, ancestorLevel) {
         const parentKeys = new presentation_common_1.KeySet();
-        elementIds.forEach(skipTransients((id) => {
+        await forEachNonTransientId(elementIds, async (id) => {
             const key = this.getElementKey(iModel, id, ancestorLevel);
             key && parentKeys.add(key);
-        }));
+        });
         return parentKeys;
     }
-    static computeCategorySelection(iModel, ids) {
+    static async computeCategorySelection(iModel, ids) {
         const categoryKeys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
+        await forEachNonTransientId(ids, async (id) => {
             const el = iModel.elements.tryGetElement(id);
             const category = el?.category ? iModel.elements.tryGetElementProps(el.category) : undefined;
             if (category) {
                 categoryKeys.add({ className: category.classFullName, id: category.id });
             }
-        }));
+        });
         return categoryKeys;
     }
-    static computeModelSelection(iModel, ids) {
+    static async computeModelSelection(iModel, ids) {
         const modelKeys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
+        await forEachNonTransientId(ids, async (id) => {
             const el = iModel.elements.tryGetElementProps(id);
             const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;
             if (model) {
                 modelKeys.add({ className: model.classFullName, id: model.id });
             }
-        }));
+        });
         return modelKeys;
     }
-    static getRelatedFunctionalElementKey(imodel, graphicalElementId) {
-        const query = `
-      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId
-        FROM bis.Element e
-        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId
-        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId
-        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)
-        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId
-        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id
-       WHERE e.ECInstanceId = ?
-      `;
-        return imodel.withPreparedStatement(query, (stmt) => {
-            stmt.bindId(1, graphicalElementId);
-            if (core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step()) {
-                const row = stmt.getRow();
-                if (row.funcElClassName && row.funcElId) {
-                    return { className: row.funcElClassName.replace(".", ":"), id: row.funcElId };
-                }
+    static async getRelatedFunctionalElementKey(imodel, graphicalElementId) {
+        const query = `
+      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId
+        FROM bis.Element e
+        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId
+        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId
+        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)
+        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId
+        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id
+      WHERE e.ECInstanceId = ?
+    `;
+        const bindings = new core_common_1.QueryBinder();
+        bindings.bindId(1, graphicalElementId);
+        for await (const row of imodel.createQueryReader(query, bindings)) {
+            if (row.funcElClassName && row.funcElId) {
+                return { className: row.funcElClassName.replace(".", ":"), id: row.funcElId };
             }
-            return undefined;
-        });
+        }
+        return undefined;
     }
-    static findFirstRelatedFunctionalElementKey(imodel, graphicalElementId) {
+    static async findFirstRelatedFunctionalElementKey(imodel, graphicalElementId) {
         let currId = graphicalElementId;
         while (currId) {
-            const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);
+            const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(imodel, currId);
             if (relatedFunctionalKey) {
                 return relatedFunctionalKey;
             }
@@ -109,28 +108,30 @@ class SelectionScopesHelper {
         }
         return undefined;
     }
-    static elementClassDerivesFrom(imodel, elementId, baseClassFullName) {
-        const query = `
-      SELECT 1
-      FROM bis.Element e
-      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId
-      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId
-      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id
-      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?
-      `;
-        return imodel.withPreparedStatement(query, (stmt) => {
-            stmt.bindId(1, elementId);
-            stmt.bindString(2, baseClassFullName);
-            return core_bentley_1.DbResult.BE_SQLITE_ROW === stmt.step();
-        });
+    static async elementClassDerivesFrom(imodel, elementId, baseClassFullName) {
+        const query = `
+      SELECT 1
+      FROM bis.Element e
+      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId
+      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId
+      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id
+      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?
+    `;
+        const bindings = new core_common_1.QueryBinder();
+        bindings.bindId(1, elementId);
+        bindings.bindString(2, baseClassFullName);
+        for await (const _ of imodel.createQueryReader(query, bindings)) {
+            return true;
+        }
+        return false;
     }
-    static computeFunctionalElementSelection(iModel, ids) {
+    static async computeFunctionalElementSelection(iModel, ids) {
         const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
-            const is3d = this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
+        await forEachNonTransientId(ids, async (id) => {
+            const is3d = await this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
             if (!is3d) {
                 // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);
                 if (firstFunctionalKey) {
                     keys.add(firstFunctionalKey);
                     return;
@@ -139,23 +140,23 @@ class SelectionScopesHelper {
             let keyToAdd;
             if (is3d) {
                 // if we're computing scope for a 3d element, try to switch to its related functional element
-                keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);
+                keyToAdd = await this.getRelatedFunctionalElementKey(iModel, id);
             }
             if (!keyToAdd) {
                 keyToAdd = (0, Utils_js_1.getElementKey)(iModel, id);
             }
             keyToAdd && keys.add(keyToAdd);
-        }));
+        });
         return keys;
     }
-    static computeFunctionalAssemblySelection(iModel, ids) {
+    static async computeFunctionalAssemblySelection(iModel, ids) {
         const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
+        await forEachNonTransientId(ids, async (id) => {
             let idToGetAssemblyFor = id;
-            const is3d = this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
+            const is3d = await this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
             if (!is3d) {
                 // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);
                 if (firstFunctionalKey) {
                     idToGetAssemblyFor = firstFunctionalKey.id;
                 }
@@ -165,23 +166,23 @@ class SelectionScopesHelper {
             let keyToAdd = assemblyKey;
             if (is3d && keyToAdd) {
                 // if we're computing scope for a 3d element, try to switch to its related functional element
-                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
+                const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
                 if (relatedFunctionalKey) {
                     keyToAdd = relatedFunctionalKey;
                 }
             }
             keyToAdd && keys.add(keyToAdd);
-        }));
+        });
         return keys;
     }
-    static computeFunctionalTopAssemblySelection(iModel, ids) {
+    static async computeFunctionalTopAssemblySelection(iModel, ids) {
         const keys = new presentation_common_1.KeySet();
-        ids.forEach(skipTransients((id) => {
+        await forEachNonTransientId(ids, async (id) => {
             let idToGetAssemblyFor = id;
-            const is3d = this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
+            const is3d = await this.elementClassDerivesFrom(iModel, id, "BisCore.GeometricElement3d");
             if (!is3d) {
                 // if the input is not a 3d element, we try to find the first related functional element
-                const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);
+                const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);
                 if (firstFunctionalKey) {
                     idToGetAssemblyFor = firstFunctionalKey.id;
                 }
@@ -191,13 +192,13 @@ class SelectionScopesHelper {
             let keyToAdd = topAssemblyKey;
             if (is3d && keyToAdd) {
                 // if we're computing scope for a 3d element, try to switch to its related functional element
-                const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
+                const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);
                 if (relatedFunctionalKey) {
                     keyToAdd = relatedFunctionalKey;
                 }
             }
             keyToAdd && keys.add(keyToAdd);
-        }));
+        });
         return keys;
     }
     static async computeSelection({ imodel, scope, elementIds }) {
@@ -224,11 +225,7 @@ class SelectionScopesHelper {
     }
 }
 exports.SelectionScopesHelper = SelectionScopesHelper;
-const skipTransients = (callback) => {
-    return (id) => {
-        if (!core_bentley_1.Id64.isTransient(id)) {
-            callback(id);
-        }
-    };
-};
+async function forEachNonTransientId(ids, callback) {
+    await Promise.all(ids.filter((id) => !core_bentley_1.Id64.isTransient(id)).map(callback));
+}
 //# sourceMappingURL=SelectionScopesHelper.js.map

package/lib/cjs/presentation-backend/SelectionScopesHelper.js.map

@@ -1 +1 @@
-{"version":3,"file":"SelectionScopesHelper.js","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,qDAAqD;AACrD;;GAEG;;;AAGH,sDAAiE;AACjE,oEAQoC;AACpC,yCAA2C;AAE3C;;;;;GAKG;AACH,MAAa,qBAAqB;IAChC,oBAAoB;IACpB,gBAAuB,CAAC;IAEjB,MAAM,CAAC,kBAAkB;QAC9B,MAAM,oBAAoB,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,WAAmB,EAAkB,EAAE,CAAC,CAAC;YACrG,EAAE,EAAE,OAAO;YACX,KAAK;YACL,WAAW;SACZ,CAAC,CAAC;QACH,OAAO;YACL,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,2BAA2B,CAAC;YACvE,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,qCAAqC,CAAC;YACnF,oBAAoB,CAAC,cAAc,EAAE,cAAc,EAAE,iDAAiD,CAAC;YACvG,gHAAgH;YAChH,wGAAwG;YACxG,+FAA+F;SAChG,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,MAAgB,EAAE,SAAqB,EAAE,aAAqB;QACzF,IAAI,MAAM,GAAG,SAAS,CAAC;QACvB,IAAI,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QACtE,OAAO,QAAQ,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,GAAG,QAAQ,CAAC;YAClB,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;YAClE,EAAE,aAAa,CAAC;QAClB,CAAC;QACD,OAAO,IAAA,wBAAa,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,UAAwB,EAAE,aAAqB;QACtG,MAAM,UAAU,GAAG,IAAI,4BAAM,EAAE,CAAC;QAChC,UAAU,CAAC,OAAO,CAChB,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YACpB,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;YAC1D,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CACH,CAAC;QACF,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,wBAAwB,CAAC,MAAgB,EAAE,GAAiB;QACzE,MAAM,YAAY,GAAG,IAAI,4BAAM,EAAE,CAAC;QAClC,GAAG,CAAC,OAAO,CACT,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YACpB,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAmB,EAAE,CAAC,CAAC;YAC/D,MAAM,QAAQ,GAAG,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC5F,IAAI,QAAQ,EAAE,CAAC;gBACb,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAG,EAAE,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,qBAAqB,CAAC,MAAgB,EAAE,GAAiB;QACtE,MAAM,SAAS,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC/B,GAAG,CAAC,OAAO,CACT,cAAc,CAAC,CAAC,EAAE,EAAE,EAAE;YACpB,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACxE,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,aAAa,EAAE,EAAE,EAAE,KAAK,CAAC,EAAG,EAAE,CAAC,CAAC;YACnE,CAAC;QACH,CAAC,CAAC,CACH,CAAC;QACF,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,8BAA8B,CAAC,MAAgB,EAAE,kBAA8B;QAC5F,MAAM,KAAK,GAAG;;;;;;;;;OASX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAA2B,EAAE;YAC3E,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;YACnC,IAAI,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,EAAE,CAAC;gBAC3C,MAAM,GAAG,GAAG,IAAI,CAAC,MAAM,EAAE,CAAC;gBAC1B,IAAI,GAAG,CAAC,eAAe,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;oBACxC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;gBAChF,CAAC;YACH,CAAC;YACD,OAAO,SAAS,CAAC;QACnB,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,oCAAoC,CAAC,MAAgB,EAAE,kBAA8B;QAClG,IAAI,MAAM,GAA2B,kBAAkB,CAAC;QACxD,OAAO,MAAM,EAAE,CAAC;YACd,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACjF,IAAI,oBAAoB,EAAE,CAAC;gBACzB,OAAO,oBAAoB,CAAC;YAC9B,CAAC;YACD,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QAClE,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,uBAAuB,CAAC,MAAgB,EAAE,SAAqB,EAAE,iBAAyB;QACvG,MAAM,KAAK,GAAG;;;;;;;OAOX,CAAC;QACJ,OAAO,MAAM,CAAC,qBAAqB,CAAC,KAAK,EAAE,CAAC,IAAI,EAAW,EAAE;YAC3D,IAAI,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;YAC1B,IAAI,CAAC,UAAU,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;YACtC,OAAO,uBAAQ,CAAC,aAAa,KAAK,IAAI,CAAC,IAAI,EAAE,CAAC;QAChD,CAAC,CAAC,CAAC;IACL,CAAC;IAEO,MAAM,CAAC,iCAAiC,CAAC,MAAgB,EAAE,GAAiB;QAClF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CACT,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YAC1B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YACpF,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBAC7B,OAAO;gBACT,CAAC;YACH,CAAC;YACD,IAAI,QAAiC,CAAC;YACtC,IAAI,IAAI,EAAE,CAAC;gBACT,6FAA6F;gBAC7F,QAAQ,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YAC7D,CAAC;YACD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAA,wBAAa,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACvC,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CACH,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,kCAAkC,CAAC,MAAgB,EAAE,GAAiB;QACnF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CACT,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YAC1B,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YACpF,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;gBAC7C,CAAC;YACH,CAAC;YACD,0EAA0E;YAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YACtE,IAAI,QAAQ,GAAG,WAAW,CAAC;YAC3B,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;gBACrB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB,EAAE,CAAC;oBACzB,QAAQ,GAAG,oBAAoB,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CACH,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,qCAAqC,CAAC,MAAgB,EAAE,GAAiB;QACtF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,GAAG,CAAC,OAAO,CACT,cAAc,CAAC,CAAC,EAAE,EAAQ,EAAE;YAC1B,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YACpF,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACjF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;gBAC7C,CAAC;YACH,CAAC;YACD,8EAA8E;YAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC/F,IAAI,QAAQ,GAAG,cAAc,CAAC;YAC9B,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;gBACrB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,oBAAoB,EAAE,CAAC;oBACzB,QAAQ,GAAG,oBAAoB,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CACH,CAAC;QACF,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAA4C;QAC1G,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;YACjB,KAAK,SAAS;gBACZ,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAG,KAAoC,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;YACpH,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YAC7D,KAAK,cAAc;gBACjB,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACnF,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAC3D,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACxD,KAAK,YAAY,CAAC;YAClB,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,iCAAiC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACpE,KAAK,qBAAqB;gBACxB,OAAO,IAAI,CAAC,kCAAkC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACrE,KAAK,yBAAyB;gBAC5B,OAAO,IAAI,CAAC,qCAAqC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC1E,CAAC;QACD,MAAM,IAAI,uCAAiB,CAAC,wCAAkB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;CACF;AApOD,sDAoOC;AAED,MAAM,cAAc,GAAG,CAAC,QAAkC,EAAE,EAAE;IAC5D,OAAO,CAAC,EAAc,EAAE,EAAE;QACxB,IAAI,CAAC,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC,EAAE,CAAC;YAC1B,QAAQ,CAAC,EAAE,CAAC,CAAC;QACf,CAAC;IACH,CAAC,CAAC;AACJ,CAAC,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/* eslint-disable @typescript-eslint/no-deprecated */\n/** @packageDocumentation\n * @module Core\n */\n\nimport { GeometricElement, IModelDb } from \"@itwin/core-backend\";\nimport { DbResult, Id64, Id64String } from \"@itwin/core-bentley\";\nimport {\n  ComputeSelectionRequestOptions,\n  ElementSelectionScopeProps,\n  InstanceKey,\n  KeySet,\n  PresentationError,\n  PresentationStatus,\n  SelectionScope,\n} from \"@itwin/presentation-common\";\nimport { getElementKey } from \"./Utils.js\";\n\n/**\n * Contains helper methods for computing selection scopes. Will get removed\n * once rules-driven scopes are implemented.\n *\n * @internal\n */\nexport class SelectionScopesHelper {\n  /* c8 ignore next */\n  private constructor() {}\n\n  public static getSelectionScopes(): SelectionScope[] {\n    const createSelectionScope = (scopeId: string, label: string, description: string): SelectionScope => ({\n      id: scopeId,\n      label,\n      description,\n    });\n    return [\n      createSelectionScope(\"element\", \"Element\", \"Select the picked element\"),\n      createSelectionScope(\"assembly\", \"Assembly\", \"Select parent of the picked element\"),\n      createSelectionScope(\"top-assembly\", \"Top Assembly\", \"Select the topmost parent of the picked element\"),\n      // WIP: temporarily comment-out \"category\" and \"model\" scopes since we can't hilite contents of them fast enough\n      // createSelectionScope(\"category\", \"Category\", \"Select all elements in the picked element's category\"),\n      // createSelectionScope(\"model\", \"Model\", \"Select all elements in the picked element's model\"),\n    ];\n  }\n\n  private static getElementKey(iModel: IModelDb, elementId: Id64String, ancestorLevel: number) {\n    let currId = elementId;\n    let parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\n    while (parentId && ancestorLevel !== 0) {\n      currId = parentId;\n      parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\n      --ancestorLevel;\n    }\n    return getElementKey(iModel, currId);\n  }\n\n  private static computeElementSelection(iModel: IModelDb, elementIds: Id64String[], ancestorLevel: number) {\n    const parentKeys = new KeySet();\n    elementIds.forEach(\n      skipTransients((id) => {\n        const key = this.getElementKey(iModel, id, ancestorLevel);\n        key && parentKeys.add(key);\n      }),\n    );\n    return parentKeys;\n  }\n\n  private static computeCategorySelection(iModel: IModelDb, ids: Id64String[]) {\n    const categoryKeys = new KeySet();\n    ids.forEach(\n      skipTransients((id) => {\n        const el = iModel.elements.tryGetElement<GeometricElement>(id);\n        const category = el?.category ? iModel.elements.tryGetElementProps(el.category) : undefined;\n        if (category) {\n          categoryKeys.add({ className: category.classFullName, id: category.id! });\n        }\n      }),\n    );\n    return categoryKeys;\n  }\n\n  private static computeModelSelection(iModel: IModelDb, ids: Id64String[]) {\n    const modelKeys = new KeySet();\n    ids.forEach(\n      skipTransients((id) => {\n        const el = iModel.elements.tryGetElementProps(id);\n        const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;\n        if (model) {\n          modelKeys.add({ className: model.classFullName, id: model.id! });\n        }\n      }),\n    );\n    return modelKeys;\n  }\n\n  private static getRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\n    const query = `\n      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId\n        FROM bis.Element e\n        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId\n        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId\n        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)\n        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId\n        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id\n       WHERE e.ECInstanceId = ?\n      `;\n    return imodel.withPreparedStatement(query, (stmt): InstanceKey | undefined => {\n      stmt.bindId(1, graphicalElementId);\n      if (DbResult.BE_SQLITE_ROW === stmt.step()) {\n        const row = stmt.getRow();\n        if (row.funcElClassName && row.funcElId) {\n          return { className: row.funcElClassName.replace(\".\", \":\"), id: row.funcElId };\n        }\n      }\n      return undefined;\n    });\n  }\n\n  private static findFirstRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): InstanceKey | undefined {\n    let currId: Id64String | undefined = graphicalElementId;\n    while (currId) {\n      const relatedFunctionalKey = this.getRelatedFunctionalElementKey(imodel, currId);\n      if (relatedFunctionalKey) {\n        return relatedFunctionalKey;\n      }\n      currId = imodel.elements.tryGetElementProps(currId)?.parent?.id;\n    }\n    return undefined;\n  }\n\n  private static elementClassDerivesFrom(imodel: IModelDb, elementId: Id64String, baseClassFullName: string): boolean {\n    const query = `\n      SELECT 1\n      FROM bis.Element e\n      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId\n      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId\n      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id\n      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?\n      `;\n    return imodel.withPreparedStatement(query, (stmt): boolean => {\n      stmt.bindId(1, elementId);\n      stmt.bindString(2, baseClassFullName);\n      return DbResult.BE_SQLITE_ROW === stmt.step();\n    });\n  }\n\n  private static computeFunctionalElementSelection(iModel: IModelDb, ids: Id64String[]) {\n    const keys = new KeySet();\n    ids.forEach(\n      skipTransients((id): void => {\n        const is3d = this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\n        if (!is3d) {\n          // if the input is not a 3d element, we try to find the first related functional element\n          const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\n          if (firstFunctionalKey) {\n            keys.add(firstFunctionalKey);\n            return;\n          }\n        }\n        let keyToAdd: InstanceKey | undefined;\n        if (is3d) {\n          // if we're computing scope for a 3d element, try to switch to its related functional element\n          keyToAdd = this.getRelatedFunctionalElementKey(iModel, id);\n        }\n        if (!keyToAdd) {\n          keyToAdd = getElementKey(iModel, id);\n        }\n        keyToAdd && keys.add(keyToAdd);\n      }),\n    );\n    return keys;\n  }\n\n  private static computeFunctionalAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\n    const keys = new KeySet();\n    ids.forEach(\n      skipTransients((id): void => {\n        let idToGetAssemblyFor = id;\n        const is3d = this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\n        if (!is3d) {\n          // if the input is not a 3d element, we try to find the first related functional element\n          const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\n          if (firstFunctionalKey) {\n            idToGetAssemblyFor = firstFunctionalKey.id;\n          }\n        }\n        // find the assembly of either the given element or the functional element\n        const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);\n        let keyToAdd = assemblyKey;\n        if (is3d && keyToAdd) {\n          // if we're computing scope for a 3d element, try to switch to its related functional element\n          const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\n          if (relatedFunctionalKey) {\n            keyToAdd = relatedFunctionalKey;\n          }\n        }\n        keyToAdd && keys.add(keyToAdd);\n      }),\n    );\n    return keys;\n  }\n\n  private static computeFunctionalTopAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\n    const keys = new KeySet();\n    ids.forEach(\n      skipTransients((id): void => {\n        let idToGetAssemblyFor = id;\n        const is3d = this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\n        if (!is3d) {\n          // if the input is not a 3d element, we try to find the first related functional element\n          const firstFunctionalKey = this.findFirstRelatedFunctionalElementKey(iModel, id);\n          if (firstFunctionalKey) {\n            idToGetAssemblyFor = firstFunctionalKey.id;\n          }\n        }\n        // find the top assembly of either the given element or the functional element\n        const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);\n        let keyToAdd = topAssemblyKey;\n        if (is3d && keyToAdd) {\n          // if we're computing scope for a 3d element, try to switch to its related functional element\n          const relatedFunctionalKey = this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\n          if (relatedFunctionalKey) {\n            keyToAdd = relatedFunctionalKey;\n          }\n        }\n        keyToAdd && keys.add(keyToAdd);\n      }),\n    );\n    return keys;\n  }\n\n  public static async computeSelection({ imodel, scope, elementIds }: ComputeSelectionRequestOptions<IModelDb>): Promise<KeySet> {\n    switch (scope.id) {\n      case \"element\":\n        return this.computeElementSelection(imodel, elementIds, (scope as ElementSelectionScopeProps).ancestorLevel ?? 0);\n      case \"assembly\":\n        return this.computeElementSelection(imodel, elementIds, 1);\n      case \"top-assembly\":\n        return this.computeElementSelection(imodel, elementIds, Number.MAX_SAFE_INTEGER);\n      case \"category\":\n        return this.computeCategorySelection(imodel, elementIds);\n      case \"model\":\n        return this.computeModelSelection(imodel, elementIds);\n      case \"functional\":\n      case \"functional-element\":\n        return this.computeFunctionalElementSelection(imodel, elementIds);\n      case \"functional-assembly\":\n        return this.computeFunctionalAssemblySelection(imodel, elementIds);\n      case \"functional-top-assembly\":\n        return this.computeFunctionalTopAssemblySelection(imodel, elementIds);\n    }\n    throw new PresentationError(PresentationStatus.InvalidArgument, \"scopeId\");\n  }\n}\n\nconst skipTransients = (callback: (id: Id64String) => void) => {\n  return (id: Id64String) => {\n    if (!Id64.isTransient(id)) {\n      callback(id);\n    }\n  };\n};\n"]}
+{"version":3,"file":"SelectionScopesHelper.js","sourceRoot":"","sources":["../../../src/presentation-backend/SelectionScopesHelper.ts"],"names":[],"mappings":";AAAA;;;gGAGgG;AAChG,qDAAqD;AACrD;;GAEG;;;AAGH,sDAAkE;AAClE,oDAAiD;AACjD,oEAQoC;AACpC,yCAA2C;AAE3C;;;;;GAKG;AACH,MAAa,qBAAqB;IAChC,oBAAoB;IACpB,gBAAwB,CAAC;IAElB,MAAM,CAAC,kBAAkB;QAC9B,MAAM,oBAAoB,GAAG,CAAC,OAAe,EAAE,KAAa,EAAE,WAAmB,EAAkB,EAAE,CAAC,CAAC;YACrG,EAAE,EAAE,OAAO;YACX,KAAK;YACL,WAAW;SACZ,CAAC,CAAC;QACH,OAAO;YACL,oBAAoB,CAAC,SAAS,EAAE,SAAS,EAAE,2BAA2B,CAAC;YACvE,oBAAoB,CAAC,UAAU,EAAE,UAAU,EAAE,qCAAqC,CAAC;YACnF,oBAAoB,CAAC,cAAc,EAAE,cAAc,EAAE,iDAAiD,CAAC;YACvG,gHAAgH;YAChH,wGAAwG;YACxG,+FAA+F;SAChG,CAAC;IACJ,CAAC;IAEO,MAAM,CAAC,aAAa,CAAC,MAAgB,EAAE,SAAqB,EAAE,aAAqB;QACzF,IAAI,MAAM,GAAG,SAAS,CAAC;QACvB,IAAI,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QACtE,OAAO,QAAQ,IAAI,aAAa,KAAK,CAAC,EAAE,CAAC;YACvC,MAAM,GAAG,QAAQ,CAAC;YAClB,QAAQ,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;YAClE,EAAE,aAAa,CAAC;QAClB,CAAC;QACD,OAAO,IAAA,wBAAa,EAAC,MAAM,EAAE,MAAM,CAAC,CAAC;IACvC,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,MAAgB,EAAE,UAAwB,EAAE,aAAqB;QAC5G,MAAM,UAAU,GAAG,IAAI,4BAAM,EAAE,CAAC;QAChC,MAAM,qBAAqB,CAAC,UAAU,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YACnD,MAAM,GAAG,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,EAAE,EAAE,aAAa,CAAC,CAAC;YAC1D,GAAG,IAAI,UAAU,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;QAC7B,CAAC,CAAC,CAAC;QACH,OAAO,UAAU,CAAC;IACpB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,MAAgB,EAAE,GAAiB;QAC/E,MAAM,YAAY,GAAG,IAAI,4BAAM,EAAE,CAAC;QAClC,MAAM,qBAAqB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,aAAa,CAAmB,EAAE,CAAC,CAAC;YAC/D,MAAM,QAAQ,GAAG,EAAE,EAAE,QAAQ,CAAC,CAAC,CAAC,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,QAAQ,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YAC5F,IAAI,QAAQ,EAAE,CAAC;gBACb,YAAY,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,EAAE,QAAQ,CAAC,EAAG,EAAE,CAAC,CAAC;YAC5E,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,YAAY,CAAC;IACtB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,qBAAqB,CAAC,MAAgB,EAAE,GAAiB;QAC5E,MAAM,SAAS,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC/B,MAAM,qBAAqB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;YAClD,MAAM,KAAK,GAAG,EAAE,CAAC,CAAC,CAAC,MAAM,CAAC,MAAM,CAAC,gBAAgB,CAAC,EAAE,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;YACxE,IAAI,KAAK,EAAE,CAAC;gBACV,SAAS,CAAC,GAAG,CAAC,EAAE,SAAS,EAAE,KAAK,CAAC,aAAa,EAAE,EAAE,EAAE,KAAK,CAAC,EAAG,EAAE,CAAC,CAAC;YACnE,CAAC;QACH,CAAC,CAAC,CAAC;QACH,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,8BAA8B,CAAC,MAAgB,EAAE,kBAA8B;QAClG,MAAM,KAAK,GAAG;;;;;;;;;KASb,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,yBAAW,EAAE,CAAC;QACnC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,kBAAkB,CAAC,CAAC;QAEvC,IAAI,KAAK,EAAE,MAAM,GAAG,IAAI,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC;YAClE,IAAI,GAAG,CAAC,eAAe,IAAI,GAAG,CAAC,QAAQ,EAAE,CAAC;gBACxC,OAAO,EAAE,SAAS,EAAE,GAAG,CAAC,eAAe,CAAC,OAAO,CAAC,GAAG,EAAE,GAAG,CAAC,EAAE,EAAE,EAAE,GAAG,CAAC,QAAQ,EAAE,CAAC;YAChF,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,oCAAoC,CAAC,MAAgB,EAAE,kBAA8B;QACxG,IAAI,MAAM,GAA2B,kBAAkB,CAAC;QACxD,OAAO,MAAM,EAAE,CAAC;YACd,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACvF,IAAI,oBAAoB,EAAE,CAAC;gBACzB,OAAO,oBAAoB,CAAC;YAC9B,CAAC;YACD,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC,kBAAkB,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,EAAE,CAAC;QAClE,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,uBAAuB,CAAC,MAAgB,EAAE,SAAqB,EAAE,iBAAyB;QAC7G,MAAM,KAAK,GAAG;;;;;;;KAOb,CAAC;QAEF,MAAM,QAAQ,GAAG,IAAI,yBAAW,EAAE,CAAC;QACnC,QAAQ,CAAC,MAAM,CAAC,CAAC,EAAE,SAAS,CAAC,CAAC;QAC9B,QAAQ,CAAC,UAAU,CAAC,CAAC,EAAE,iBAAiB,CAAC,CAAC;QAE1C,IAAI,KAAK,EAAE,MAAM,CAAC,IAAI,MAAM,CAAC,iBAAiB,CAAC,KAAK,EAAE,QAAQ,CAAC,EAAE,CAAC;YAChE,OAAO,IAAI,CAAC;QACd,CAAC;QACD,OAAO,KAAK,CAAC;IACf,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,iCAAiC,CAAC,MAAgB,EAAE,GAAiB;QACxF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,MAAM,qBAAqB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YAC1F,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACvF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,IAAI,CAAC,GAAG,CAAC,kBAAkB,CAAC,CAAC;oBAC7B,OAAO;gBACT,CAAC;YACH,CAAC;YACD,IAAI,QAAiC,CAAC;YACtC,IAAI,IAAI,EAAE,CAAC;gBACT,6FAA6F;gBAC7F,QAAQ,GAAG,MAAM,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACnE,CAAC;YACD,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,QAAQ,GAAG,IAAA,wBAAa,EAAC,MAAM,EAAE,EAAE,CAAC,CAAC;YACvC,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,kCAAkC,CAAC,MAAgB,EAAE,GAAiB;QACzF,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,MAAM,qBAAqB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YAC1F,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACvF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;gBAC7C,CAAC;YACH,CAAC;YACD,0EAA0E;YAC1E,MAAM,WAAW,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,CAAC,CAAC,CAAC;YACtE,IAAI,QAAQ,GAAG,WAAW,CAAC;YAC3B,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;gBACrB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAC5F,IAAI,oBAAoB,EAAE,CAAC;oBACzB,QAAQ,GAAG,oBAAoB,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,qCAAqC,CAAC,MAAgB,EAAE,GAAiB;QAC5F,MAAM,IAAI,GAAG,IAAI,4BAAM,EAAE,CAAC;QAC1B,MAAM,qBAAqB,CAAC,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,EAAE;YAC5C,IAAI,kBAAkB,GAAG,EAAE,CAAC;YAC5B,MAAM,IAAI,GAAG,MAAM,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,EAAE,EAAE,4BAA4B,CAAC,CAAC;YAC1F,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,wFAAwF;gBACxF,MAAM,kBAAkB,GAAG,MAAM,IAAI,CAAC,oCAAoC,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;gBACvF,IAAI,kBAAkB,EAAE,CAAC;oBACvB,kBAAkB,GAAG,kBAAkB,CAAC,EAAE,CAAC;gBAC7C,CAAC;YACH,CAAC;YACD,8EAA8E;YAC9E,MAAM,cAAc,GAAG,IAAI,CAAC,aAAa,CAAC,MAAM,EAAE,kBAAkB,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YAC/F,IAAI,QAAQ,GAAG,cAAc,CAAC;YAC9B,IAAI,IAAI,IAAI,QAAQ,EAAE,CAAC;gBACrB,6FAA6F;gBAC7F,MAAM,oBAAoB,GAAG,MAAM,IAAI,CAAC,8BAA8B,CAAC,MAAM,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;gBAC5F,IAAI,oBAAoB,EAAE,CAAC;oBACzB,QAAQ,GAAG,oBAAoB,CAAC;gBAClC,CAAC;YACH,CAAC;YACD,QAAQ,IAAI,IAAI,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QACjC,CAAC,CAAC,CAAC;QACH,OAAO,IAAI,CAAC;IACd,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,gBAAgB,CAAC,EAAE,MAAM,EAAE,KAAK,EAAE,UAAU,EAA4C;QAC1G,QAAQ,KAAK,CAAC,EAAE,EAAE,CAAC;YACjB,KAAK,SAAS;gBACZ,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAG,KAAoC,CAAC,aAAa,IAAI,CAAC,CAAC,CAAC;YACpH,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC,CAAC;YAC7D,KAAK,cAAc;gBACjB,OAAO,IAAI,CAAC,uBAAuB,CAAC,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,gBAAgB,CAAC,CAAC;YACnF,KAAK,UAAU;gBACb,OAAO,IAAI,CAAC,wBAAwB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YAC3D,KAAK,OAAO;gBACV,OAAO,IAAI,CAAC,qBAAqB,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACxD,KAAK,YAAY,CAAC;YAClB,KAAK,oBAAoB;gBACvB,OAAO,IAAI,CAAC,iCAAiC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACpE,KAAK,qBAAqB;gBACxB,OAAO,IAAI,CAAC,kCAAkC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;YACrE,KAAK,yBAAyB;gBAC5B,OAAO,IAAI,CAAC,qCAAqC,CAAC,MAAM,EAAE,UAAU,CAAC,CAAC;QAC1E,CAAC;QACD,MAAM,IAAI,uCAAiB,CAAC,wCAAkB,CAAC,eAAe,EAAE,SAAS,CAAC,CAAC;IAC7E,CAAC;CACF;AA5ND,sDA4NC;AAED,KAAK,UAAU,qBAAqB,CAAC,GAAc,EAAE,QAA2C;IAC9F,MAAM,OAAO,CAAC,GAAG,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC,mBAAI,CAAC,WAAW,CAAC,EAAE,CAAC,CAAC,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,CAAC;AAC7E,CAAC","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/* eslint-disable @typescript-eslint/no-deprecated */\r\n/** @packageDocumentation\r\n * @module Core\r\n */\r\n\r\nimport { GeometricElement, IModelDb } from \"@itwin/core-backend\";\r\nimport { Id64, Id64Array, Id64String } from \"@itwin/core-bentley\";\r\nimport { QueryBinder } from \"@itwin/core-common\";\r\nimport {\r\n  ComputeSelectionRequestOptions,\r\n  ElementSelectionScopeProps,\r\n  InstanceKey,\r\n  KeySet,\r\n  PresentationError,\r\n  PresentationStatus,\r\n  SelectionScope,\r\n} from \"@itwin/presentation-common\";\r\nimport { getElementKey } from \"./Utils.js\";\r\n\r\n/**\r\n * Contains helper methods for computing selection scopes. Will get removed\r\n * once rules-driven scopes are implemented.\r\n *\r\n * @internal\r\n */\r\nexport class SelectionScopesHelper {\r\n  /* c8 ignore next */\r\n  private constructor() { }\r\n\r\n  public static getSelectionScopes(): SelectionScope[] {\r\n    const createSelectionScope = (scopeId: string, label: string, description: string): SelectionScope => ({\r\n      id: scopeId,\r\n      label,\r\n      description,\r\n    });\r\n    return [\r\n      createSelectionScope(\"element\", \"Element\", \"Select the picked element\"),\r\n      createSelectionScope(\"assembly\", \"Assembly\", \"Select parent of the picked element\"),\r\n      createSelectionScope(\"top-assembly\", \"Top Assembly\", \"Select the topmost parent of the picked element\"),\r\n      // WIP: temporarily comment-out \"category\" and \"model\" scopes since we can't hilite contents of them fast enough\r\n      // createSelectionScope(\"category\", \"Category\", \"Select all elements in the picked element's category\"),\r\n      // createSelectionScope(\"model\", \"Model\", \"Select all elements in the picked element's model\"),\r\n    ];\r\n  }\r\n\r\n  private static getElementKey(iModel: IModelDb, elementId: Id64String, ancestorLevel: number) {\r\n    let currId = elementId;\r\n    let parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    while (parentId && ancestorLevel !== 0) {\r\n      currId = parentId;\r\n      parentId = iModel.elements.tryGetElementProps(currId)?.parent?.id;\r\n      --ancestorLevel;\r\n    }\r\n    return getElementKey(iModel, currId);\r\n  }\r\n\r\n  private static async computeElementSelection(iModel: IModelDb, elementIds: Id64String[], ancestorLevel: number) {\r\n    const parentKeys = new KeySet();\r\n    await forEachNonTransientId(elementIds, async (id) => {\r\n      const key = this.getElementKey(iModel, id, ancestorLevel);\r\n      key && parentKeys.add(key);\r\n    });\r\n    return parentKeys;\r\n  }\r\n\r\n  private static async computeCategorySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const categoryKeys = new KeySet();\r\n    await forEachNonTransientId(ids, async (id) => {\r\n      const el = iModel.elements.tryGetElement<GeometricElement>(id);\r\n      const category = el?.category ? iModel.elements.tryGetElementProps(el.category) : undefined;\r\n      if (category) {\r\n        categoryKeys.add({ className: category.classFullName, id: category.id! });\r\n      }\r\n    });\r\n    return categoryKeys;\r\n  }\r\n\r\n  private static async computeModelSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const modelKeys = new KeySet();\r\n    await forEachNonTransientId(ids, async (id) => {\r\n      const el = iModel.elements.tryGetElementProps(id);\r\n      const model = el ? iModel.models.tryGetModelProps(el.model) : undefined;\r\n      if (model) {\r\n        modelKeys.add({ className: model.classFullName, id: model.id! });\r\n      }\r\n    });\r\n    return modelKeys;\r\n  }\r\n\r\n  private static async getRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): Promise<InstanceKey | undefined> {\r\n    const query = `\r\n      SELECT funcSchemaDef.Name || '.' || funcClassDef.Name funcElClassName, fe.ECInstanceId funcElId\r\n        FROM bis.Element e\r\n        LEFT JOIN func.PhysicalElementFulfillsFunction rel1 ON rel1.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.DrawingGraphicRepresentsFunctionalElement rel2 ON rel2.SourceECInstanceId = e.ECInstanceId\r\n        LEFT JOIN func.FunctionalElement fe ON fe.ECInstanceId IN (rel1.TargetECInstanceId, rel2.TargetECInstanceId)\r\n        INNER JOIN meta.ECClassDef funcClassDef ON funcClassDef.ECInstanceId = fe.ECClassId\r\n        INNER JOIN meta.ECSchemaDef funcSchemaDef ON funcSchemaDef.ECInstanceId = funcClassDef.Schema.Id\r\n      WHERE e.ECInstanceId = ?\r\n    `;\r\n\r\n    const bindings = new QueryBinder();\r\n    bindings.bindId(1, graphicalElementId);\r\n\r\n    for await (const row of imodel.createQueryReader(query, bindings)) {\r\n      if (row.funcElClassName && row.funcElId) {\r\n        return { className: row.funcElClassName.replace(\".\", \":\"), id: row.funcElId };\r\n      }\r\n    }\r\n    return undefined;\r\n  }\r\n\r\n  private static async findFirstRelatedFunctionalElementKey(imodel: IModelDb, graphicalElementId: Id64String): Promise<InstanceKey | undefined> {\r\n    let currId: Id64String | undefined = graphicalElementId;\r\n    while (currId) {\r\n      const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(imodel, currId);\r\n      if (relatedFunctionalKey) {\r\n        return relatedFunctionalKey;\r\n      }\r\n      currId = imodel.elements.tryGetElementProps(currId)?.parent?.id;\r\n    }\r\n    return undefined;\r\n  }\r\n\r\n  private static async elementClassDerivesFrom(imodel: IModelDb, elementId: Id64String, baseClassFullName: string): Promise<boolean> {\r\n    const query = `\r\n      SELECT 1\r\n      FROM bis.Element e\r\n      INNER JOIN meta.ClassHasAllBaseClasses baseClassRels ON baseClassRels.SourceECInstanceId = e.ECClassId\r\n      INNER JOIN meta.ECClassDef baseClass ON baseClass.ECInstanceId = baseClassRels.TargetECInstanceId\r\n      INNER JOIN meta.ECSchemaDef baseSchema ON baseSchema.ECInstanceId = baseClass.Schema.Id\r\n      WHERE e.ECInstanceId = ? AND (baseSchema.Name || ':' || baseClass.Name) = ?\r\n    `;\r\n\r\n    const bindings = new QueryBinder();\r\n    bindings.bindId(1, elementId);\r\n    bindings.bindString(2, baseClassFullName);\r\n\r\n    for await (const _ of imodel.createQueryReader(query, bindings)) {\r\n      return true;\r\n    }\r\n    return false;\r\n  }\r\n\r\n  private static async computeFunctionalElementSelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    await forEachNonTransientId(ids, async (id) => {\r\n      const is3d = await this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey) {\r\n          keys.add(firstFunctionalKey);\r\n          return;\r\n        }\r\n      }\r\n      let keyToAdd: InstanceKey | undefined;\r\n      if (is3d) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        keyToAdd = await this.getRelatedFunctionalElementKey(iModel, id);\r\n      }\r\n      if (!keyToAdd) {\r\n        keyToAdd = getElementKey(iModel, id);\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    });\r\n    return keys;\r\n  }\r\n\r\n  private static async computeFunctionalAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    await forEachNonTransientId(ids, async (id) => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = await this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey) {\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n        }\r\n      }\r\n      // find the assembly of either the given element or the functional element\r\n      const assemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, 1);\r\n      let keyToAdd = assemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey) {\r\n          keyToAdd = relatedFunctionalKey;\r\n        }\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    });\r\n    return keys;\r\n  }\r\n\r\n  private static async computeFunctionalTopAssemblySelection(iModel: IModelDb, ids: Id64String[]) {\r\n    const keys = new KeySet();\r\n    await forEachNonTransientId(ids, async (id) => {\r\n      let idToGetAssemblyFor = id;\r\n      const is3d = await this.elementClassDerivesFrom(iModel, id, \"BisCore.GeometricElement3d\");\r\n      if (!is3d) {\r\n        // if the input is not a 3d element, we try to find the first related functional element\r\n        const firstFunctionalKey = await this.findFirstRelatedFunctionalElementKey(iModel, id);\r\n        if (firstFunctionalKey) {\r\n          idToGetAssemblyFor = firstFunctionalKey.id;\r\n        }\r\n      }\r\n      // find the top assembly of either the given element or the functional element\r\n      const topAssemblyKey = this.getElementKey(iModel, idToGetAssemblyFor, Number.MAX_SAFE_INTEGER);\r\n      let keyToAdd = topAssemblyKey;\r\n      if (is3d && keyToAdd) {\r\n        // if we're computing scope for a 3d element, try to switch to its related functional element\r\n        const relatedFunctionalKey = await this.getRelatedFunctionalElementKey(iModel, keyToAdd.id);\r\n        if (relatedFunctionalKey) {\r\n          keyToAdd = relatedFunctionalKey;\r\n        }\r\n      }\r\n      keyToAdd && keys.add(keyToAdd);\r\n    });\r\n    return keys;\r\n  }\r\n\r\n  public static async computeSelection({ imodel, scope, elementIds }: ComputeSelectionRequestOptions<IModelDb>): Promise<KeySet> {\r\n    switch (scope.id) {\r\n      case \"element\":\r\n        return this.computeElementSelection(imodel, elementIds, (scope as ElementSelectionScopeProps).ancestorLevel ?? 0);\r\n      case \"assembly\":\r\n        return this.computeElementSelection(imodel, elementIds, 1);\r\n      case \"top-assembly\":\r\n        return this.computeElementSelection(imodel, elementIds, Number.MAX_SAFE_INTEGER);\r\n      case \"category\":\r\n        return this.computeCategorySelection(imodel, elementIds);\r\n      case \"model\":\r\n        return this.computeModelSelection(imodel, elementIds);\r\n      case \"functional\":\r\n      case \"functional-element\":\r\n        return this.computeFunctionalElementSelection(imodel, elementIds);\r\n      case \"functional-assembly\":\r\n        return this.computeFunctionalAssemblySelection(imodel, elementIds);\r\n      case \"functional-top-assembly\":\r\n        return this.computeFunctionalTopAssemblySelection(imodel, elementIds);\r\n    }\r\n    throw new PresentationError(PresentationStatus.InvalidArgument, \"scopeId\");\r\n  }\r\n}\r\n\r\nasync function forEachNonTransientId(ids: Id64Array, callback: (id: Id64String) => Promise<void>): Promise<void> {\r\n  await Promise.all(ids.filter((id) => !Id64.isTransient(id)).map(callback));\r\n}\r\n"]}