@itwin/core-backend 5.0.0-dev.11 → 5.0.0-dev.110

package/lib/cjs/CheckpointManager.js

@@ -7,7 +7,7 @@
  * @module iModels
  */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CheckpointManager = exports.V1CheckpointManager = exports.V2CheckpointManager = exports.Downloads = exports.ProgressStatus = void 0;
+exports.CheckpointManager = exports.V2CheckpointManager = exports.Downloads = exports.ProgressStatus = void 0;
 // cspell:ignore BLOCKCACHE
 const path = require("path");
 const imodeljs_native_1 = require("@bentley/imodeljs-native");
@@ -34,6 +34,7 @@ var ProgressStatus;
 })(ProgressStatus || (exports.ProgressStatus = ProgressStatus = {}));
 /** @internal */
 class Downloads {
+    static _active = new Map();
     static async process(job, fn) {
         const jobName = job.request.localFile; // save this, it can change inside call to `fn`!
         this._active.set(jobName, job);
@@ -58,12 +59,18 @@ class Downloads {
     }
 }
 exports.Downloads = Downloads;
-Downloads._active = new Map();
 /**
  * Utility class for opening V2 checkpoints from cloud containers, and also for downloading them.
  * @internal
 */
 class V2CheckpointManager {
+    static cloudCacheName = "Checkpoints";
+    static _cloudCache;
+    static containers = new Map();
+    /** used by HubMock
+     * @internal
+     */
+    static [Symbols_1._mockCheckpoint];
     static getFolder() {
         const cloudCachePath = path.join(BriefcaseManager_1.BriefcaseManager.cacheDir, V2CheckpointManager.cloudCacheName);
         if (!(IModelJsFs_1.IModelJsFs.existsSync(cloudCachePath))) {
@@ -104,7 +111,7 @@ class V2CheckpointManager {
             // the sasToken is checked for refresh (before it expires) on every Rpc request using that user's accessToken. For Ipc, the
             // accessToken in the checkpoint request is undefined, and the sasToken is requested by IModelHost.getAccessToken(). It is refreshed on a timer.
             if (undefined === checkpoint.accessToken) {
-                tokenFn = async () => (await IModelHost_1.IModelHost.hubAccess.queryV2Checkpoint(checkpoint))?.sasToken ?? "";
+                tokenFn = async () => (await IModelHost_1.IModelHost[Symbols_1._hubAccess].queryV2Checkpoint(checkpoint))?.sasToken ?? "";
                 tokenRefreshSeconds = undefined;
             }
             container = CloudSqlite_1.CloudSqlite.createCloudContainer({ ...this.toCloudContainerProps(v2Props), tokenRefreshSeconds, logId: process.env.POD_NAME, tokenFn });
@@ -113,9 +120,11 @@ class V2CheckpointManager {
         return container;
     }
     static async attach(checkpoint) {
+        if (this[Symbols_1._mockCheckpoint]) // used by HubMock
+            return { dbName: this[Symbols_1._mockCheckpoint].mockAttach(checkpoint), container: undefined };
         let v2props;
         try {
-            v2props = await IModelHost_1.IModelHost.hubAccess.queryV2Checkpoint(checkpoint);
+            v2props = await IModelHost_1.IModelHost[Symbols_1._hubAccess].queryV2Checkpoint(checkpoint);
             if (!v2props)
                 throw new Error("no checkpoint");
         }
@@ -125,6 +134,8 @@ class V2CheckpointManager {
         try {
             const container = this.getContainer(v2props, checkpoint);
             const dbName = v2props.dbName;
+            // Use the new token from the recently queried v2 checkpoint just incase the one we currently have is expired.
+            container.accessToken = v2props.sasToken;
             if (!container.isConnected)
                 container.connect(this.cloudCache);
             container.checkForChanges();
@@ -158,14 +169,19 @@ class V2CheckpointManager {
             throw new core_common_1.IModelError(e.errorNumber, error);
         }
     }
+    /** @internal */
     static async performDownload(job) {
         const request = job.request;
-        const v2props = await IModelHost_1.IModelHost.hubAccess.queryV2Checkpoint({ ...request.checkpoint, allowPreceding: true });
-        if (!v2props)
-            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.NotFound, "V2 checkpoint not found");
-        CheckpointManager.onDownloadV2.raiseEvent(job);
-        const container = CloudSqlite_1.CloudSqlite.createCloudContainer(this.toCloudContainerProps(v2props));
-        await CloudSqlite_1.CloudSqlite.transferDb("download", container, { dbName: v2props.dbName, localFileName: request.localFile, onProgress: request.onProgress });
+        if (this[Symbols_1._mockCheckpoint])
+            this[Symbols_1._mockCheckpoint].mockDownload(request);
+        else {
+            const v2props = await IModelHost_1.IModelHost[Symbols_1._hubAccess].queryV2Checkpoint({ ...request.checkpoint, allowPreceding: true });
+            if (!v2props)
+                throw new core_common_1.IModelError(core_bentley_1.IModelStatus.NotFound, "V2 checkpoint not found");
+            CheckpointManager.onDownloadV2.raiseEvent(job);
+            const container = CloudSqlite_1.CloudSqlite.createCloudContainer(this.toCloudContainerProps(v2props));
+            await CloudSqlite_1.CloudSqlite.transferDb("download", container, { dbName: v2props.dbName, localFileName: request.localFile, onProgress: request.onProgress });
+        }
         return request.checkpoint.changeset.id;
     }
     /** Fully download a V2 checkpoint to a local file that can be used to create a briefcase or to work offline.
@@ -177,71 +193,21 @@ class V2CheckpointManager {
     }
 }
 exports.V2CheckpointManager = V2CheckpointManager;
-V2CheckpointManager.cloudCacheName = "Checkpoints";
-V2CheckpointManager.containers = new Map();
-/** Utility class to deal with downloading V1 checkpoints from iModelHub.
- * @internal
- */
-class V1CheckpointManager {
-    static getFolder(iModelId) {
-        return path.join(BriefcaseManager_1.BriefcaseManager.getIModelPath(iModelId), "checkpoints");
-    }
-    static getFileName(checkpoint) {
-        const changesetId = checkpoint.changeset.id || "first";
-        return path.join(this.getFolder(checkpoint.iModelId), `${changesetId}.bim`);
-    }
-    static async getCheckpointDb(request) {
-        const db = IModelDb_1.SnapshotDb.tryFindByKey(CheckpointManager.getKey(request.checkpoint));
-        return (undefined !== db) ? db : Downloads.download(request, async (job) => this.downloadAndOpen(job));
-    }
-    /** Download a V1 checkpoint */
-    static async downloadCheckpoint(request) {
-        return Downloads.download(request, async (job) => this.performDownload(job));
-    }
-    static openCheckpointV1(fileName, checkpoint) {
-        const snapshot = IModelDb_1.SnapshotDb.openFile(fileName, { key: CheckpointManager.getKey(checkpoint) });
-        snapshot._iTwinId = checkpoint.iTwinId;
-        return snapshot;
-    }
-    static async downloadAndOpen(job) {
-        const db = CheckpointManager.tryOpenLocalFile(job.request);
-        if (db)
-            return db;
-        await this.performDownload(job);
-        await CheckpointManager.updateToRequestedVersion(job.request);
-        return this.openCheckpointV1(job.request.localFile, job.request.checkpoint);
-    }
-    static async performDownload(job) {
-        CheckpointManager.onDownloadV1.raiseEvent(job);
-        // eslint-disable-next-line @typescript-eslint/no-deprecated
-        return (await IModelHost_1.IModelHost.hubAccess.downloadV1Checkpoint(job.request)).id;
-    }
-}
-exports.V1CheckpointManager = V1CheckpointManager;
 /** @internal  */
 class CheckpointManager {
+    static onDownloadV2 = new core_bentley_1.BeEvent();
     static getKey(checkpoint) { return `${checkpoint.iModelId}:${checkpoint.changeset.id}`; }
     static async doDownload(request) {
-        try {
-            // first see if there's a V2 checkpoint available.
-            const stopwatch = new core_bentley_1.StopWatch(`[${request.checkpoint.changeset.id}]`, true);
-            core_bentley_1.Logger.logInfo(loggerCategory, `Starting download of V2 checkpoint with id ${stopwatch.description}`);
-            const changesetId = await V2CheckpointManager.downloadCheckpoint(request);
-            core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded V2 checkpoint with id ${stopwatch.description} (${stopwatch.elapsedSeconds} seconds)`);
-            if (changesetId !== request.checkpoint.changeset.id)
-                core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded previous v2 checkpoint because requested checkpoint not found.`, { requestedChangesetId: request.checkpoint.changeset.id, iModelId: request.checkpoint.iModelId, changesetId, iTwinId: request.checkpoint.iTwinId });
-            else
-                core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded v2 checkpoint.`, { iModelId: request.checkpoint.iModelId, changesetId: request.checkpoint.changeset.id, iTwinId: request.checkpoint.iTwinId });
-            return changesetId;
-        }
-        catch (error) {
-            if (error.errorNumber === core_bentley_1.IModelStatus.NotFound) { // No V2 checkpoint available, try a v1 checkpoint
-                const changeset = await V1CheckpointManager.downloadCheckpoint(request);
-                core_bentley_1.Logger.logWarning(loggerCategory, `Got an error downloading v2 checkpoint, but downloaded v1 checkpoint successfully!`, { error, iModelId: request.checkpoint.iModelId, iTwinId: request.checkpoint.iTwinId, requestedChangesetId: request.checkpoint.changeset.id, changesetId: changeset });
-                return changeset;
-            }
-            throw error; // most likely, was aborted
-        }
+        // first see if there's a V2 checkpoint available.
+        const stopwatch = new core_bentley_1.StopWatch(`[${request.checkpoint.changeset.id}]`, true);
+        core_bentley_1.Logger.logInfo(loggerCategory, `Starting download of V2 checkpoint with id ${stopwatch.description}`);
+        const changesetId = await V2CheckpointManager.downloadCheckpoint(request);
+        core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded V2 checkpoint with id ${stopwatch.description} (${stopwatch.elapsedSeconds} seconds)`);
+        if (changesetId !== request.checkpoint.changeset.id)
+            core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded previous v2 checkpoint because requested checkpoint not found.`, { requestedChangesetId: request.checkpoint.changeset.id, iModelId: request.checkpoint.iModelId, changesetId, iTwinId: request.checkpoint.iTwinId });
+        else
+            core_bentley_1.Logger.logInfo(loggerCategory, `Downloaded v2 checkpoint.`, { iModelId: request.checkpoint.iModelId, changesetId: request.checkpoint.changeset.id, iTwinId: request.checkpoint.iTwinId });
+        return changesetId;
     }
     static async updateToRequestedVersion(request) {
         const checkpoint = request.checkpoint;
@@ -266,7 +232,7 @@ class CheckpointManager {
                 if (currentChangeset.id !== checkpoint.changeset.id) {
                     const accessToken = checkpoint.accessToken;
                     const toIndex = checkpoint.changeset.index ??
-                        (await IModelHost_1.IModelHost.hubAccess.getChangesetFromVersion({ accessToken, iModelId: checkpoint.iModelId, version: core_common_1.IModelVersion.asOfChangeSet(checkpoint.changeset.id) })).index;
+                        (await IModelHost_1.IModelHost[Symbols_1._hubAccess].getChangesetFromVersion({ accessToken, iModelId: checkpoint.iModelId, version: core_common_1.IModelVersion.asOfChangeSet(checkpoint.changeset.id) })).index;
                     await BriefcaseManager_1.BriefcaseManager.pullAndApplyChangesets(db, { accessToken, toIndex });
                 }
                 else {
@@ -345,35 +311,17 @@ class CheckpointManager {
             IModelJsFs_1.IModelJsFs.removeSync(fileName);
         return isValid;
     }
-    /** try to open an existing local file to satisfy a download request */
-    static tryOpenLocalFile(request) {
-        const checkpoint = request.checkpoint;
-        if (this.verifyCheckpoint(checkpoint, request.localFile))
-            return V1CheckpointManager.openCheckpointV1(request.localFile, checkpoint);
-        // check a list of aliases for finding checkpoints downloaded to non-default locations (e.g. from older versions)
-        if (request.aliasFiles) {
-            for (const alias of request.aliasFiles) {
-                if (this.verifyCheckpoint(checkpoint, alias)) {
-                    request.localFile = alias;
-                    return V1CheckpointManager.openCheckpointV1(alias, checkpoint);
-                }
-            }
-        }
-        return undefined;
-    }
     static async toCheckpointProps(args) {
-        const changeset = args.changeset ?? await IModelHost_1.IModelHost.hubAccess.getLatestChangeset({ ...args, accessToken: await IModelHost_1.IModelHost.getAccessToken() });
+        const changeset = args.changeset ?? await IModelHost_1.IModelHost[Symbols_1._hubAccess].getLatestChangeset({ ...args, accessToken: await IModelHost_1.IModelHost.getAccessToken() });
         return {
             iModelId: args.iModelId,
             iTwinId: args.iTwinId,
             changeset: {
                 index: changeset.index,
-                id: changeset.id ?? (await IModelHost_1.IModelHost.hubAccess.queryChangeset({ ...args, changeset, accessToken: await IModelHost_1.IModelHost.getAccessToken() })).id,
+                id: changeset.id ?? (await IModelHost_1.IModelHost[Symbols_1._hubAccess].queryChangeset({ ...args, changeset, accessToken: await IModelHost_1.IModelHost.getAccessToken() })).id,
             },
         };
     }
 }
 exports.CheckpointManager = CheckpointManager;
-CheckpointManager.onDownloadV1 = new core_bentley_1.BeEvent();
-CheckpointManager.onDownloadV2 = new core_bentley_1.BeEvent();
 //# sourceMappingURL=CheckpointManager.js.map

package/lib/cjs/CheckpointManager.js.map

@@ -1 +1 @@
-{"version":3,"file":"CheckpointManager.js","sourceRoot":"","sources":["../../src/CheckpointManager.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;AAEH,2BAA2B;AAE3B,6BAA6B;AAC7B,8DAAgE;AAChE,sDAA4J;AAC5J,oDAE4B;AAE5B,mEAAgE;AAChE,yDAAsD;AACtD,+CAA4C;AAC5C,6CAA0C;AAC1C,6CAA0C;AAC1C,yCAAkD;AAClD,8DAAyD;AACzD,gDAA+C;AAE/C,MAAM,cAAc,GAAG,6CAAqB,CAAC,QAAQ,CAAC;AAyBtD;;GAEG;AACH,IAAY,cAKX;AALD,WAAY,cAAc;IACxB,yBAAyB;IACzB,2DAAY,CAAA;IACZ,sBAAsB;IACtB,qDAAS,CAAA;AACX,CAAC,EALW,cAAc,8BAAd,cAAc,QAKzB;AAmCD,gBAAgB;AAChB,MAAa,SAAS;IAGZ,MAAM,CAAC,KAAK,CAAC,OAAO,CAAI,GAAgB,EAAE,EAAoC;QACpF,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,gDAAgD;QACvF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,QAAuB;QAChD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAI,OAAwB,EAAE,UAA4C;QACpG,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,SAAS,KAAK,GAAG;YACnB,OAAO,GAAG,CAAC,OAAO,CAAC;QAErB,uBAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QACtD,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC;QAClB,OAAO,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;IACrD,CAAC;;AA1BH,8BA2BC;AA1BgB,iBAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;AA4B1D;;;EAGE;AACF,MAAa,mBAAmB;IAKvB,MAAM,CAAC,SAAS;QACrB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,mCAAgB,CAAC,QAAQ,EAAE,mBAAmB,CAAC,cAAc,CAAC,CAAC;QAChG,IAAI,CAAC,CAAC,uBAAU,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;YAC7C,uBAAU,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,gIAAgI;IACzH,MAAM,CAAC,OAAO;QACnB,KAAK,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,IAAI,KAAK,CAAC,WAAW;gBACnB,KAAK,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,yBAAW,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,OAAO,EAAE,CAAC;QAClE,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAEO,MAAM,KAAK,UAAU;QAC3B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,QAAQ,GAAuB,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACxF,mFAAmF;YACnF,IAAI,CAAC,CAAC,uBAAU,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;gBACjE,QAAQ,GAAG,SAAS,CAAC,CAAC,2CAA2C;YAEnE,IAAI,CAAC,WAAW,GAAG,yBAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QACtH,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,qKAAqK;IAC7J,MAAM,CAAC,qBAAqB,CAAC,IAA6B;QAChE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC,WAAW,wBAAwB,EAAE,WAAW,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;IACrI,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,OAAgC,EAAE,UAA2B;QACvF,IAAI,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,OAAmF,CAAC;YACxF,IAAI,mBAAmB,GAAuB,CAAC,CAAC,CAAC;YACjD,yIAAyI;YACzI,2HAA2H;YAC3H,gJAAgJ;YAChJ,IAAI,SAAS,KAAK,UAAU,CAAC,WAAW,EAAE,CAAC;gBACzC,OAAO,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,uBAAU,CAAC,SAAS,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;gBACjG,mBAAmB,GAAG,SAAS,CAAC;YAClC,CAAC;YACD,SAAS,GAAG,yBAAW,CAAC,oBAAoB,CAAC,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,mBAAmB,EAAE,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;YACpJ,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA2B;QACpD,IAAI,OAA4C,CAAC;QACjD,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,uBAAU,CAAC,SAAS,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACnE,IAAI,CAAC,OAAO;gBACV,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,QAAQ,EAAE,iCAAiC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/F,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAC9B,IAAI,CAAC,SAAS,CAAC,WAAW;gBACxB,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrC,SAAS,CAAC,eAAe,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,uBAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC;gBAC/E,MAAM,iBAAiB,GAAG,CAAC,IAAY,EAAE,UAAkB,EAAE,EAAE,CAAC,uBAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,wBAAwB,IAAI,EAAE,EAAE,UAAU,CAAC,CAAC;gBACvJ,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBAClD,MAAM,SAAS,GAAG,iBAAiB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,+DAA+D;gBACtH,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI,OAAO,CAAC,WAAW,IAAI,SAAS,IAAI,OAAO,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;oBACtG,MAAM,WAAW,GAAG,KAAK,EAAE,QAAmC,EAAE,EAAE;wBAChE,MAAM,SAAS,GAAG,IAAI,wBAAS,CAAC,IAAI,SAAS,CAAC,WAAW,IAAI,MAAM,GAAG,EAAE,IAAI,CAAC,CAAC;wBAC9E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,wBAAwB,SAAS,CAAC,WAAW,EAAE,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;wBACvH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC;wBACpC,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,eAAe,SAAS,CAAC,WAAW,aAAa,IAAI,KAAK,SAAS,CAAC,cAAc,WAAW,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;oBACvK,CAAC,CAAC;oBACF,mEAAmE;oBACnE,WAAW,CAAC,yBAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;gBACnH,CAAC;qBAAM,CAAC;oBACN,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2DAA2D,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAChM,CAAC;YACH,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,+BAA+B,CAAC,CAAC,OAAO,EAAE,CAAC;YACzD,IAAI,UAAU,CAAC,QAAQ;gBACrB,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YAEzC,MAAM,IAAI,yBAAW,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAgB;QACnD,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC5B,MAAM,OAAO,GAAwC,MAAM,uBAAU,CAAC,SAAS,CAAC,iBAAiB,CAAC,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;QACnJ,IAAI,CAAC,OAAO;YACV,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC;QAE1E,iBAAiB,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC/C,MAAM,SAAS,GAAG,yBAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;QACxF,MAAM,yBAAW,CAAC,UAAU,CAAC,UAAU,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;QAClJ,OAAO,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;IACzC,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAwB;QAC7D,OAAO,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5F,CAAC;;AA5HH,kDA6HC;AA5HwB,kCAAc,GAAG,aAAa,CAAC;AAEvC,8BAAU,GAAG,IAAI,GAAG,EAAsC,CAAC;AA4H5E;;GAEG;AACH,MAAa,mBAAmB;IACvB,MAAM,CAAC,SAAS,CAAC,QAAoB;QAC1C,OAAO,IAAI,CAAC,IAAI,CAAC,mCAAgB,CAAC,aAAa,CAAC,QAAQ,CAAC,EAAE,aAAa,CAAC,CAAC;IAC5E,CAAC;IAEM,MAAM,CAAC,WAAW,CAAC,UAA2B;QACnD,MAAM,WAAW,GAAG,UAAU,CAAC,SAAS,CAAC,EAAE,IAAI,OAAO,CAAC;QACvD,OAAO,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,GAAG,WAAW,MAAM,CAAC,CAAC;IAC9E,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,OAAwB;QAC1D,MAAM,EAAE,GAAG,qBAAU,CAAC,YAAY,CAAC,iBAAiB,CAAC,MAAM,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC,CAAC;QACjF,OAAO,CAAC,SAAS,KAAK,EAAE,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IACtH,CAAC;IAED,+BAA+B;IACxB,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAwB;QAC7D,OAAO,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5F,CAAC;IAEM,MAAM,CAAC,gBAAgB,CAAC,QAAuB,EAAE,UAA2B;QACjF,MAAM,QAAQ,GAAG,qBAAU,CAAC,QAAQ,CAAC,QAAQ,EAAE,EAAE,GAAG,EAAE,iBAAiB,CAAC,MAAM,CAAC,UAAU,CAAC,EAAE,CAAC,CAAC;QAC7F,QAAgB,CAAC,QAAQ,GAAG,UAAU,CAAC,OAAO,CAAC;QAChD,OAAO,QAAQ,CAAC;IAClB,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAgB;QACnD,MAAM,EAAE,GAAG,iBAAiB,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC3D,IAAI,EAAE;YACJ,OAAO,EAAE,CAAC;QACZ,MAAM,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC;QAChC,MAAM,iBAAiB,CAAC,wBAAwB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;QAC9D,OAAO,IAAI,CAAC,gBAAgB,CAAC,GAAG,CAAC,OAAO,CAAC,SAAS,EAAE,GAAG,CAAC,OAAO,CAAC,UAAU,CAAC,CAAC;IAC9E,CAAC;IAEO,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAgB;QACnD,iBAAiB,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;QAC/C,4DAA4D;QAC5D,OAAO,CAAC,MAAM,uBAAU,CAAC,SAAS,CAAC,oBAAoB,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,EAAE,CAAC;IAC3E,CAAC;CACF;AAxCD,kDAwCC;AAED,iBAAiB;AACjB,MAAa,iBAAiB;IAGrB,MAAM,CAAC,MAAM,CAAC,UAA2B,IAAI,OAAO,GAAG,UAAU,CAAC,QAAQ,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAwB;QACtD,IAAI,CAAC;YACH,kDAAkD;YAClD,MAAM,SAAS,GAAG,IAAI,wBAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;YAC9E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,8CAA8C,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;YACtG,MAAM,WAAW,GAAG,MAAM,mBAAmB,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;YAC1E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,oCAAoC,SAAS,CAAC,WAAW,KAAK,SAAS,CAAC,cAAc,WAAW,CAAC,CAAC;YAClI,IAAI,WAAW,KAAK,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;gBACjD,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2EAA2E,EAAE,EAAE,oBAAoB,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;;gBAEhQ,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2BAA2B,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;YAC5L,OAAO,WAAW,CAAC;QACrB,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YACpB,IAAI,KAAK,CAAC,WAAW,KAAK,2BAAY,CAAC,QAAQ,EAAE,CAAC,CAAC,kDAAkD;gBACnG,MAAM,SAAS,GAAG,MAAM,mBAAmB,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;gBACxE,qBAAM,CAAC,UAAU,CAAC,cAAc,EAAE,oFAAoF,EAAE,EAAE,KAAK,EAAE,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,oBAAoB,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,CAAC,CAAC;gBAC9R,OAAO,SAAS,CAAC;YACnB,CAAC;YACD,MAAM,KAAK,CAAC,CAAC,2BAA2B;QAC1C,CAAC;IACH,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,OAAwB;QACnE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACtC,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;QACrC,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC;QAClH,IAAI,CAAC;YACH,4BAA4B;YAC5B,MAAM,YAAY,GAAG,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,CAAC,IAAI,uBAAQ,CAAC,KAAK,CAAC,CAAC,0CAA0C;YAC/H,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,EAAE,uBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,yDAAyD;YACtH,MAAM,EAAE,GAAG,qBAAU,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC;YACzD,MAAM,QAAQ,GAAG,EAAE,CAAC,mBAAS,CAAC,CAAC;YAC/B,IAAI,CAAC;gBAEH,IAAI,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC;oBAC9B,qBAAM,CAAC,UAAU,CAAC,cAAc,EAAE,4CAA4C,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;oBACjG,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,CAAC;gBAED,IAAI,QAAQ,CAAC,cAAc,EAAE,KAAK,8BAAgB,CAAC,UAAU;oBAC3D,QAAQ,CAAC,gBAAgB,CAAC,8BAAgB,CAAC,UAAU,CAAC,CAAC;gBAEzD,iBAAiB,CAAC,uBAAuB,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBAC1D,iCAAiC;gBACjC,MAAM,gBAAgB,GAAiC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;gBACtF,IAAI,gBAAgB,CAAC,EAAE,KAAK,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC;oBACpD,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;oBAC3C,MAAM,OAAO,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK;wBACxC,CAAC,MAAM,uBAAU,CAAC,SAAS,CAAC,uBAAuB,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,2BAAa,CAAC,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;oBAC5K,MAAM,mCAAgB,CAAC,sBAAsB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC9E,CAAC;qBAAM,CAAC;oBACN,2FAA2F;oBAC3F,gBAAgB,CAAC,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,KAAM,CAAC,CAAC,+DAA+D;oBACrH,QAAQ,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,oEAAoE;gBAChI,EAAE,CAAC,WAAW,EAAE,CAAC;gBACjB,EAAE,CAAC,KAAK,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAEpB,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,4CAA4C,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC/F,uBAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAElC,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,SAAS,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,cAAc,EAAE,CAAC;gBAC3K,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,+EAA+E,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAClI,mCAAgB,CAAC,6BAA6B,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,uGAAuG;IAChG,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAwB;QAC7D,IAAI,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC;YAC9D,OAAO;QAET,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvC,IAAI,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC;oBACrD,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;oBAC1B,OAAO;gBACT,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,6FAA6F;IACtF,MAAM,CAAC,uBAAuB,CAAC,UAA2B,EAAE,UAAsB;QACvF,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC;QAEjF,MAAM,QAAQ,GAAG,UAAU,CAAC,mBAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,QAAQ,CAAC,mBAAmB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,mBAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QACxD,IAAI,QAAQ,KAAK,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,IAAI,QAAQ,CAAC,UAAU,EAAE;gBACvB,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,gBAAgB,EAAE,mDAAmD,CAAC,CAAC;YAE5G,qBAAM,CAAC,UAAU,CAAC,cAAc,EAAE,gGAAgG,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YAChL,MAAM,kBAAkB,GAAG,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC/D,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;YACxC,UAAkB,CAAC,SAAS,GAAG,kBAAkB,CAAC;YACnD,wEAAwE;YACxE,QAAQ,CAAC,cAAc,CAAC,mBAAmB,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;YAC7D,IAAI,SAAS,KAAK,WAAW,CAAC,KAAK;gBACjC,QAAQ,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,mBAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtD,IAAI,OAAO,KAAK,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC;YAChD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,gBAAgB,EAAE,mDAAmD,CAAC,CAAC;IAC9G,CAAC;IAED,4DAA4D;IACrD,MAAM,CAAC,gBAAgB,CAAC,UAA2B,EAAE,QAAuB;QACjF,IAAI,CAAC,uBAAU,CAAC,UAAU,CAAC,QAAQ,CAAC;YAClC,OAAO,KAAK,CAAC;QAEf,MAAM,QAAQ,GAAG,IAAI,6BAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnD,IAAI,CAAC;YACH,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,uBAAQ,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,KAAK,QAAQ,CAAC,WAAW,EAAE,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,KAAK,QAAQ,CAAC,mBAAmB,EAAE,CAAC,EAAE,CAAC;QAChI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACrB,IAAI,CAAC,OAAO;YACV,uBAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAElC,OAAO,OAAO,CAAC;IACjB,CAAC;IAED,uEAAuE;IAChE,MAAM,CAAC,gBAAgB,CAAC,OAAwB;QACrD,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACtC,IAAI,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC;YACtD,OAAO,mBAAmB,CAAC,gBAAgB,CAAC,OAAO,CAAC,SAAS,EAAE,UAAU,CAAC,CAAC;QAE7E,iHAAiH;QACjH,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvC,IAAI,IAAI,CAAC,gBAAgB,CAAC,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC;oBAC7C,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;oBAC1B,OAAO,mBAAmB,CAAC,gBAAgB,CAAC,KAAK,EAAE,UAAU,CAAC,CAAC;gBACjE,CAAC;YACH,CAAC;QACH,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAwB;QAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,uBAAU,CAAC,SAAS,CAAC,kBAAkB,CAAC,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,MAAM,uBAAU,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAE/I,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE;gBACT,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,EAAE,EAAE,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,uBAAU,CAAC,SAAS,CAAC,cAAc,CAAC,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,uBAAU,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE;aAC3I;SACF,CAAC;IACJ,CAAC;;AA3KH,8CA4KC;AA3KwB,8BAAY,GAAG,IAAI,sBAAO,EAA8B,CAAC;AACzD,8BAAY,GAAG,IAAI,sBAAO,EAA8B,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 iModels\r\n */\r\n\r\n// cspell:ignore BLOCKCACHE\r\n\r\nimport * as path from \"path\";\r\nimport { NativeLoggerCategory } from \"@bentley/imodeljs-native\";\r\nimport { AccessToken, BeEvent, ChangeSetStatus, Guid, GuidString, IModelStatus, Logger, LogLevel, Mutable, OpenMode, StopWatch } from \"@itwin/core-bentley\";\r\nimport {\r\n  BriefcaseIdValue, ChangesetId, ChangesetIdWithIndex, ChangesetIndexAndId, IModelError, IModelVersion, LocalDirName, LocalFileName, OpenCheckpointArgs,\r\n} from \"@itwin/core-common\";\r\nimport { V2CheckpointAccessProps } from \"./BackendHubAccess\";\r\nimport { BackendLoggerCategory } from \"./BackendLoggerCategory\";\r\nimport { BriefcaseManager } from \"./BriefcaseManager\";\r\nimport { CloudSqlite } from \"./CloudSqlite\";\r\nimport { IModelHost } from \"./IModelHost\";\r\nimport { IModelJsFs } from \"./IModelJsFs\";\r\nimport { SnapshotDb, TokenArg } from \"./IModelDb\";\r\nimport { IModelNative } from \"./internal/NativePlatform\";\r\nimport { _nativeDb } from \"./internal/Symbols\";\r\n\r\nconst loggerCategory = BackendLoggerCategory.IModelDb;\r\n\r\n/**\r\n * Properties of a checkpoint\r\n * @public\r\n */\r\nexport interface CheckpointProps extends TokenArg {\r\n  readonly expectV2?: boolean;\r\n\r\n  /** iTwin that the iModel belongs to */\r\n  readonly iTwinId: GuidString;\r\n\r\n  /** Id of the iModel */\r\n  readonly iModelId: GuidString;\r\n\r\n  /** changeset for the checkpoint */\r\n  readonly changeset: ChangesetIdWithIndex;\r\n\r\n  /** If true, then the latest successful v2 checkpoint at or before the provided changeset will be returned when calling queryV2Checkpoint.  */\r\n  readonly allowPreceding?: boolean;\r\n\r\n  /** The number of seconds before the current token expires to attempt to reacquire a new token. Default is 1 hour. */\r\n  readonly reattachSafetySeconds?: number;\r\n}\r\n\r\n/** Return value from [[ProgressFunction]].\r\n *  @public\r\n */\r\nexport enum ProgressStatus {\r\n  /** Continue download. */\r\n  Continue = 0,\r\n  /** Abort download. */\r\n  Abort = 1,\r\n}\r\n\r\n/** Called to show progress during a download. If this function returns non-zero, the download is aborted.\r\n *  @public\r\n */\r\nexport type ProgressFunction = (loaded: number, total: number) => ProgressStatus;\r\n\r\n/** The parameters that specify a request to download a checkpoint file from iModelHub.\r\n * @internal\r\n */\r\nexport interface DownloadRequest {\r\n  /** name of local file to hold the downloaded data. */\r\n  localFile: LocalFileName;\r\n\r\n  /** A list of full fileName paths to test before downloading. If a valid file exists by one of these names,\r\n   * no download is performed and `localFile` is updated to reflect the fact that the file exists with that name.\r\n   * This can be used, for example, to look for checkpoints from previous versions if the naming strategy changes.\r\n   */\r\n  readonly aliasFiles?: ReadonlyArray<string>;\r\n\r\n  /** Properties of the checkpoint to be downloaded */\r\n  readonly checkpoint: CheckpointProps;\r\n\r\n  /** If present, this function will be called to indicate progress as the briefcase is downloaded. If this\r\n   * function returns a non-zero value, the download is aborted.\r\n   */\r\n  readonly onProgress?: ProgressFunction;\r\n}\r\n\r\n/** @internal */\r\nexport interface DownloadJob {\r\n  request: DownloadRequest;\r\n  promise?: Promise<any>;\r\n}\r\n\r\n/** @internal */\r\nexport class Downloads {\r\n  private static _active = new Map<string, DownloadJob>();\r\n\r\n  private static async process<T>(job: DownloadJob, fn: (job: DownloadJob) => Promise<T>) {\r\n    const jobName = job.request.localFile; // save this, it can change inside call to `fn`!\r\n    this._active.set(jobName, job);\r\n    try {\r\n      return await fn(job);\r\n    } finally {\r\n      this._active.delete(jobName);\r\n    }\r\n  }\r\n\r\n  public static isInProgress(pathName: LocalFileName): DownloadJob | undefined {\r\n    return this._active.get(pathName);\r\n  }\r\n\r\n  public static async download<T>(request: DownloadRequest, downloadFn: (job: DownloadJob) => Promise<T>) {\r\n    const pathName = request.localFile;\r\n    let job = this.isInProgress(pathName);\r\n    if (undefined !== job)\r\n      return job.promise;\r\n\r\n    IModelJsFs.recursiveMkDirSync(path.dirname(pathName));\r\n    job = { request };\r\n    return job.promise = this.process(job, downloadFn);\r\n  }\r\n}\r\n\r\n/**\r\n * Utility class for opening V2 checkpoints from cloud containers, and also for downloading them.\r\n * @internal\r\n*/\r\nexport class V2CheckpointManager {\r\n  public static readonly cloudCacheName = \"Checkpoints\";\r\n  private static _cloudCache?: CloudSqlite.CloudCache;\r\n  private static containers = new Map<string, CloudSqlite.CloudContainer>();\r\n\r\n  public static getFolder(): LocalDirName {\r\n    const cloudCachePath = path.join(BriefcaseManager.cacheDir, V2CheckpointManager.cloudCacheName);\r\n    if (!(IModelJsFs.existsSync(cloudCachePath))) {\r\n      IModelJsFs.recursiveMkDirSync(cloudCachePath);\r\n    }\r\n    return cloudCachePath;\r\n  }\r\n\r\n  /* only used by tests that reset the state of the v2CheckpointManager. all dbs should be closed before calling this function. */\r\n  public static cleanup(): void {\r\n    for (const [_, value] of this.containers.entries()) {\r\n      if (value.isConnected)\r\n        value.disconnect({ detach: true });\r\n    }\r\n\r\n    CloudSqlite.CloudCaches.dropCache(this.cloudCacheName)?.destroy();\r\n    this._cloudCache = undefined;\r\n    this.containers.clear();\r\n  }\r\n\r\n  private static get cloudCache(): CloudSqlite.CloudCache {\r\n    if (!this._cloudCache) {\r\n      let cacheDir: string | undefined = process.env.CHECKPOINT_CACHE_DIR ?? this.getFolder();\r\n      // See if there is a daemon running, otherwise use profile directory for cloudCache\r\n      if (!(IModelJsFs.existsSync(path.join(cacheDir, \"portnumber.bcv\"))))\r\n        cacheDir = undefined; // no daemon running, use profile directory\r\n\r\n      this._cloudCache = CloudSqlite.CloudCaches.getCache({ cacheName: this.cloudCacheName, cacheDir, cacheSize: \"50G\" });\r\n    }\r\n    return this._cloudCache;\r\n  }\r\n\r\n  /** Member names differ slightly between the V2Checkpoint api and the CloudSqlite api. Add aliases `accessName` for `accountName` and `accessToken` for `sasToken` */\r\n  private static toCloudContainerProps(from: V2CheckpointAccessProps): CloudSqlite.ContainerAccessProps {\r\n    return { ...from, baseUri: `https://${from.accountName}.blob.core.windows.net`, accessToken: from.sasToken, storageType: \"azure\" };\r\n  }\r\n\r\n  private static getContainer(v2Props: V2CheckpointAccessProps, checkpoint: CheckpointProps) {\r\n    let container = this.containers.get(v2Props.containerId);\r\n    if (undefined === container) {\r\n      let tokenFn: ((args: CloudSqlite.RequestTokenArgs) => Promise<AccessToken>) | undefined;\r\n      let tokenRefreshSeconds: number | undefined = -1;\r\n      // from Rpc, the accessToken in the checkpoint request is from the current user. It is used to request the sasToken for the container and\r\n      // the sasToken is checked for refresh (before it expires) on every Rpc request using that user's accessToken. For Ipc, the\r\n      // accessToken in the checkpoint request is undefined, and the sasToken is requested by IModelHost.getAccessToken(). It is refreshed on a timer.\r\n      if (undefined === checkpoint.accessToken) {\r\n        tokenFn = async () => (await IModelHost.hubAccess.queryV2Checkpoint(checkpoint))?.sasToken ?? \"\";\r\n        tokenRefreshSeconds = undefined;\r\n      }\r\n      container = CloudSqlite.createCloudContainer({ ...this.toCloudContainerProps(v2Props), tokenRefreshSeconds, logId: process.env.POD_NAME, tokenFn });\r\n      this.containers.set(v2Props.containerId, container);\r\n    }\r\n    return container;\r\n  }\r\n\r\n  public static async attach(checkpoint: CheckpointProps): Promise<{ dbName: string, container: CloudSqlite.CloudContainer }> {\r\n    let v2props: V2CheckpointAccessProps | undefined;\r\n    try {\r\n      v2props = await IModelHost.hubAccess.queryV2Checkpoint(checkpoint);\r\n      if (!v2props)\r\n        throw new Error(\"no checkpoint\");\r\n    } catch (err: any) {\r\n      throw new IModelError(IModelStatus.NotFound, `V2 checkpoint not found: err: ${err.message}`);\r\n    }\r\n\r\n    try {\r\n      const container = this.getContainer(v2props, checkpoint);\r\n      const dbName = v2props.dbName;\r\n      if (!container.isConnected)\r\n        container.connect(this.cloudCache);\r\n      container.checkForChanges();\r\n      const dbStats = container.queryDatabase(dbName);\r\n      if (IModelHost.appWorkspace.settings.getBoolean(\"Checkpoints/prefetch\", false)) {\r\n        const getPrefetchConfig = (name: string, defaultVal: number) => IModelHost.appWorkspace.settings.getNumber(`Checkpoints/prefetch/${name}`, defaultVal);\r\n        const minRequests = getPrefetchConfig(\"minRequests\", 3);\r\n        const maxRequests = getPrefetchConfig(\"maxRequests\", 6);\r\n        const timeout = getPrefetchConfig(\"timeout\", 100);\r\n        const maxBlocks = getPrefetchConfig(\"maxBlocks\", 500); // default size of 2GB. Assumes a checkpoint block size of 4MB.\r\n        if (dbStats?.totalBlocks !== undefined && dbStats.totalBlocks <= maxBlocks && dbStats.nPrefetch === 0) {\r\n          const logPrefetch = async (prefetch: CloudSqlite.CloudPrefetch) => {\r\n            const stopwatch = new StopWatch(`[${container.containerId}/${dbName}]`, true);\r\n            Logger.logInfo(loggerCategory, `Starting prefetch of ${stopwatch.description}`, { minRequests, maxRequests, timeout });\r\n            const done = await prefetch.promise;\r\n            Logger.logInfo(loggerCategory, `Prefetch of ${stopwatch.description} complete=${done} (${stopwatch.elapsedSeconds} seconds)`, { minRequests, maxRequests, timeout });\r\n          };\r\n          // eslint-disable-next-line @typescript-eslint/no-floating-promises\r\n          logPrefetch(CloudSqlite.startCloudPrefetch(container, dbName, { minRequests, nRequests: maxRequests, timeout }));\r\n        } else {\r\n          Logger.logInfo(loggerCategory, `Skipping prefetch due to size limits or ongoing prefetch.`, { maxBlocks, numPrefetches: dbStats?.nPrefetch, totalBlocksInDb: dbStats?.totalBlocks, v2props });\r\n        }\r\n      }\r\n      return { dbName, container };\r\n    } catch (e: any) {\r\n      const error = `Cloud cache connect failed: ${e.message}`;\r\n      if (checkpoint.expectV2)\r\n        Logger.logError(loggerCategory, error);\r\n\r\n      throw new IModelError(e.errorNumber, error);\r\n    }\r\n  }\r\n\r\n  private static async performDownload(job: DownloadJob): Promise<ChangesetId> {\r\n    const request = job.request;\r\n    const v2props: V2CheckpointAccessProps | undefined = await IModelHost.hubAccess.queryV2Checkpoint({ ...request.checkpoint, allowPreceding: true });\r\n    if (!v2props)\r\n      throw new IModelError(IModelStatus.NotFound, \"V2 checkpoint not found\");\r\n\r\n    CheckpointManager.onDownloadV2.raiseEvent(job);\r\n    const container = CloudSqlite.createCloudContainer(this.toCloudContainerProps(v2props));\r\n    await CloudSqlite.transferDb(\"download\", container, { dbName: v2props.dbName, localFileName: request.localFile, onProgress: request.onProgress });\r\n    return request.checkpoint.changeset.id;\r\n  }\r\n\r\n  /** Fully download a V2 checkpoint to a local file that can be used to create a briefcase or to work offline.\r\n   * @returns a Promise that is resolved when the download completes with the changesetId of the downloaded checkpoint (which will\r\n   * be the same as the requested changesetId or the most recent checkpoint before it.)\r\n   */\r\n  public static async downloadCheckpoint(request: DownloadRequest): Promise<ChangesetId> {\r\n    return Downloads.download(request, async (job: DownloadJob) => this.performDownload(job));\r\n  }\r\n}\r\n\r\n/** Utility class to deal with downloading V1 checkpoints from iModelHub.\r\n * @internal\r\n */\r\nexport class V1CheckpointManager {\r\n  public static getFolder(iModelId: GuidString): LocalDirName {\r\n    return path.join(BriefcaseManager.getIModelPath(iModelId), \"checkpoints\");\r\n  }\r\n\r\n  public static getFileName(checkpoint: CheckpointProps): LocalFileName {\r\n    const changesetId = checkpoint.changeset.id || \"first\";\r\n    return path.join(this.getFolder(checkpoint.iModelId), `${changesetId}.bim`);\r\n  }\r\n\r\n  public static async getCheckpointDb(request: DownloadRequest): Promise<SnapshotDb> {\r\n    const db = SnapshotDb.tryFindByKey(CheckpointManager.getKey(request.checkpoint));\r\n    return (undefined !== db) ? db : Downloads.download(request, async (job: DownloadJob) => this.downloadAndOpen(job));\r\n  }\r\n\r\n  /** Download a V1 checkpoint */\r\n  public static async downloadCheckpoint(request: DownloadRequest): Promise<ChangesetId> {\r\n    return Downloads.download(request, async (job: DownloadJob) => this.performDownload(job));\r\n  }\r\n\r\n  public static openCheckpointV1(fileName: LocalFileName, checkpoint: CheckpointProps) {\r\n    const snapshot = SnapshotDb.openFile(fileName, { key: CheckpointManager.getKey(checkpoint) });\r\n    (snapshot as any)._iTwinId = checkpoint.iTwinId;\r\n    return snapshot;\r\n  }\r\n\r\n  private static async downloadAndOpen(job: DownloadJob) {\r\n    const db = CheckpointManager.tryOpenLocalFile(job.request);\r\n    if (db)\r\n      return db;\r\n    await this.performDownload(job);\r\n    await CheckpointManager.updateToRequestedVersion(job.request);\r\n    return this.openCheckpointV1(job.request.localFile, job.request.checkpoint);\r\n  }\r\n\r\n  private static async performDownload(job: DownloadJob): Promise<ChangesetId> {\r\n    CheckpointManager.onDownloadV1.raiseEvent(job);\r\n    // eslint-disable-next-line @typescript-eslint/no-deprecated\r\n    return (await IModelHost.hubAccess.downloadV1Checkpoint(job.request)).id;\r\n  }\r\n}\r\n\r\n/** @internal  */\r\nexport class CheckpointManager {\r\n  public static readonly onDownloadV1 = new BeEvent<(job: DownloadJob) => void>();\r\n  public static readonly onDownloadV2 = new BeEvent<(job: DownloadJob) => void>();\r\n  public static getKey(checkpoint: CheckpointProps) { return `${checkpoint.iModelId}:${checkpoint.changeset.id}`; }\r\n\r\n  private static async doDownload(request: DownloadRequest): Promise<ChangesetId> {\r\n    try {\r\n      // first see if there's a V2 checkpoint available.\r\n      const stopwatch = new StopWatch(`[${request.checkpoint.changeset.id}]`, true);\r\n      Logger.logInfo(loggerCategory, `Starting download of V2 checkpoint with id ${stopwatch.description}`);\r\n      const changesetId = await V2CheckpointManager.downloadCheckpoint(request);\r\n      Logger.logInfo(loggerCategory, `Downloaded V2 checkpoint with id ${stopwatch.description} (${stopwatch.elapsedSeconds} seconds)`);\r\n      if (changesetId !== request.checkpoint.changeset.id)\r\n        Logger.logInfo(loggerCategory, `Downloaded previous v2 checkpoint because requested checkpoint not found.`, { requestedChangesetId: request.checkpoint.changeset.id, iModelId: request.checkpoint.iModelId, changesetId, iTwinId: request.checkpoint.iTwinId });\r\n      else\r\n        Logger.logInfo(loggerCategory, `Downloaded v2 checkpoint.`, { iModelId: request.checkpoint.iModelId, changesetId: request.checkpoint.changeset.id, iTwinId: request.checkpoint.iTwinId });\r\n      return changesetId;\r\n    } catch (error: any) {\r\n      if (error.errorNumber === IModelStatus.NotFound) { // No V2 checkpoint available, try a v1 checkpoint\r\n        const changeset = await V1CheckpointManager.downloadCheckpoint(request);\r\n        Logger.logWarning(loggerCategory, `Got an error downloading v2 checkpoint, but downloaded v1 checkpoint successfully!`, { error, iModelId: request.checkpoint.iModelId, iTwinId: request.checkpoint.iTwinId, requestedChangesetId: request.checkpoint.changeset.id, changesetId: changeset });\r\n        return changeset;\r\n      }\r\n      throw error; // most likely, was aborted\r\n    }\r\n  }\r\n\r\n  public static async updateToRequestedVersion(request: DownloadRequest) {\r\n    const checkpoint = request.checkpoint;\r\n    const targetFile = request.localFile;\r\n    const traceInfo = { iTwinId: checkpoint.iTwinId, iModelId: checkpoint.iModelId, changeset: checkpoint.changeset };\r\n    try {\r\n      // Open checkpoint for write\r\n      const prevLogLevel = Logger.getLevel(NativeLoggerCategory.SQLite) ?? LogLevel.Error; // Get log level before we set it to None.\r\n      Logger.setLevel(NativeLoggerCategory.SQLite, LogLevel.None); // Ignores noisy error messages when applying changesets.\r\n      const db = SnapshotDb.openForApplyChangesets(targetFile);\r\n      const nativeDb = db[_nativeDb];\r\n      try {\r\n\r\n        if (nativeDb.hasPendingTxns()) {\r\n          Logger.logWarning(loggerCategory, \"Checkpoint with Txns found - deleting them\", () => traceInfo);\r\n          nativeDb.deleteAllTxns();\r\n        }\r\n\r\n        if (nativeDb.getBriefcaseId() !== BriefcaseIdValue.Unassigned)\r\n          nativeDb.resetBriefcaseId(BriefcaseIdValue.Unassigned);\r\n\r\n        CheckpointManager.validateCheckpointGuids(checkpoint, db);\r\n        // Apply change sets if necessary\r\n        const currentChangeset: Mutable<ChangesetIndexAndId> = nativeDb.getCurrentChangeset();\r\n        if (currentChangeset.id !== checkpoint.changeset.id) {\r\n          const accessToken = checkpoint.accessToken;\r\n          const toIndex = checkpoint.changeset.index ??\r\n            (await IModelHost.hubAccess.getChangesetFromVersion({ accessToken, iModelId: checkpoint.iModelId, version: IModelVersion.asOfChangeSet(checkpoint.changeset.id) })).index;\r\n          await BriefcaseManager.pullAndApplyChangesets(db, { accessToken, toIndex });\r\n        } else {\r\n          // make sure the parent changeset index is saved in the file - old versions didn't have it.\r\n          currentChangeset.index = checkpoint.changeset.index!; // eslint-disable-line @typescript-eslint/no-non-null-assertion\r\n          nativeDb.saveLocalValue(\"parentChangeSet\", JSON.stringify(currentChangeset));\r\n        }\r\n      } finally {\r\n        Logger.setLevel(NativeLoggerCategory.SQLite, prevLogLevel); // Set logging to what it was before we started applying changesets.\r\n        db.saveChanges();\r\n        db.close();\r\n      }\r\n    } catch (error: any) {\r\n\r\n      Logger.logError(loggerCategory, \"Error downloading checkpoint - deleting it\", () => traceInfo);\r\n      IModelJsFs.removeSync(targetFile);\r\n\r\n      if (error.errorNumber === ChangeSetStatus.CorruptedChangeStream || error.errorNumber === ChangeSetStatus.InvalidId || error.errorNumber === ChangeSetStatus.InvalidVersion) {\r\n        Logger.logError(loggerCategory, \"Detected potential corruption of change sets. Deleting them to enable retries\", () => traceInfo);\r\n        BriefcaseManager.deleteChangeSetsFromLocalDisk(checkpoint.iModelId);\r\n      }\r\n      throw error;\r\n    }\r\n  }\r\n\r\n  /** Download a checkpoint file from iModelHub into a local file specified in the request parameters. */\r\n  public static async downloadCheckpoint(request: DownloadRequest): Promise<void> {\r\n    if (this.verifyCheckpoint(request.checkpoint, request.localFile))\r\n      return;\r\n\r\n    if (request.aliasFiles) {\r\n      for (const alias of request.aliasFiles) {\r\n        if (this.verifyCheckpoint(request.checkpoint, alias)) {\r\n          request.localFile = alias;\r\n          return;\r\n        }\r\n      }\r\n    }\r\n\r\n    await this.doDownload(request);\r\n    return this.updateToRequestedVersion(request);\r\n  }\r\n\r\n  /** checks a file's dbGuid & iTwinId for consistency, and updates the dbGuid when possible */\r\n  public static validateCheckpointGuids(checkpoint: CheckpointProps, snapshotDb: SnapshotDb) {\r\n    const traceInfo = { iTwinId: checkpoint.iTwinId, iModelId: checkpoint.iModelId };\r\n\r\n    const nativeDb = snapshotDb[_nativeDb];\r\n    const dbChangeset = nativeDb.getCurrentChangeset();\r\n    const iModelId = Guid.normalize(nativeDb.getIModelId());\r\n    if (iModelId !== Guid.normalize(checkpoint.iModelId)) {\r\n      if (nativeDb.isReadonly())\r\n        throw new IModelError(IModelStatus.ValidationFailed, \"iModelId is not properly set up in the checkpoint\");\r\n\r\n      Logger.logWarning(loggerCategory, \"iModelId is not properly set up in the checkpoint. Updated checkpoint to the correct iModelId.\", () => ({ ...traceInfo, dbGuid: iModelId }));\r\n      const iModelIdNormalized = Guid.normalize(checkpoint.iModelId);\r\n      nativeDb.setIModelId(iModelIdNormalized);\r\n      (snapshotDb as any)._iModelId = iModelIdNormalized;\r\n      // Required to reset the ChangeSetId because setDbGuid clears the value.\r\n      nativeDb.saveLocalValue(\"ParentChangeSetId\", dbChangeset.id);\r\n      if (undefined !== dbChangeset.index)\r\n        nativeDb.saveLocalValue(\"parentChangeSet\", JSON.stringify(dbChangeset));\r\n    }\r\n\r\n    const iTwinId = Guid.normalize(nativeDb.getITwinId());\r\n    if (iTwinId !== Guid.normalize(checkpoint.iTwinId))\r\n      throw new IModelError(IModelStatus.ValidationFailed, \"iTwinId was not properly set up in the checkpoint\");\r\n  }\r\n\r\n  /** @returns true if the file is the checkpoint requested */\r\n  public static verifyCheckpoint(checkpoint: CheckpointProps, fileName: LocalFileName): boolean {\r\n    if (!IModelJsFs.existsSync(fileName))\r\n      return false;\r\n\r\n    const nativeDb = new IModelNative.platform.DgnDb();\r\n    try {\r\n      nativeDb.openIModel(fileName, OpenMode.Readonly);\r\n    } catch {\r\n      return false;\r\n    }\r\n\r\n    const isValid = checkpoint.iModelId === nativeDb.getIModelId() && checkpoint.changeset.id === nativeDb.getCurrentChangeset().id;\r\n    nativeDb.closeFile();\r\n    if (!isValid)\r\n      IModelJsFs.removeSync(fileName);\r\n\r\n    return isValid;\r\n  }\r\n\r\n  /** try to open an existing local file to satisfy a download request */\r\n  public static tryOpenLocalFile(request: DownloadRequest): SnapshotDb | undefined {\r\n    const checkpoint = request.checkpoint;\r\n    if (this.verifyCheckpoint(checkpoint, request.localFile))\r\n      return V1CheckpointManager.openCheckpointV1(request.localFile, checkpoint);\r\n\r\n    // check a list of aliases for finding checkpoints downloaded to non-default locations (e.g. from older versions)\r\n    if (request.aliasFiles) {\r\n      for (const alias of request.aliasFiles) {\r\n        if (this.verifyCheckpoint(checkpoint, alias)) {\r\n          request.localFile = alias;\r\n          return V1CheckpointManager.openCheckpointV1(alias, checkpoint);\r\n        }\r\n      }\r\n    }\r\n    return undefined;\r\n  }\r\n\r\n  public static async toCheckpointProps(args: OpenCheckpointArgs): Promise<CheckpointProps> {\r\n    const changeset = args.changeset ?? await IModelHost.hubAccess.getLatestChangeset({ ...args, accessToken: await IModelHost.getAccessToken() });\r\n\r\n    return {\r\n      iModelId: args.iModelId,\r\n      iTwinId: args.iTwinId,\r\n      changeset: {\r\n        index: changeset.index,\r\n        id: changeset.id ?? (await IModelHost.hubAccess.queryChangeset({ ...args, changeset, accessToken: await IModelHost.getAccessToken() })).id,\r\n      },\r\n    };\r\n  }\r\n}\r\n"]}
+{"version":3,"file":"CheckpointManager.js","sourceRoot":"","sources":["../../src/CheckpointManager.ts"],"names":[],"mappings":";AAAA;;;+FAG+F;AAC/F;;GAEG;;;AAEH,2BAA2B;AAE3B,6BAA6B;AAC7B,8DAAgE;AAChE,sDAA4J;AAC5J,oDAE4B;AAE5B,mEAAgE;AAChE,yDAAsD;AACtD,+CAA4C;AAC5C,6CAA0C;AAC1C,6CAA0C;AAC1C,yCAAkD;AAClD,8DAAyD;AACzD,gDAA4E;AAE5E,MAAM,cAAc,GAAG,6CAAqB,CAAC,QAAQ,CAAC;AA+BtD;;GAEG;AACH,IAAY,cAKX;AALD,WAAY,cAAc;IACxB,yBAAyB;IACzB,2DAAY,CAAA;IACZ,sBAAsB;IACtB,qDAAS,CAAA;AACX,CAAC,EALW,cAAc,8BAAd,cAAc,QAKzB;AAmCD,gBAAgB;AAChB,MAAa,SAAS;IACZ,MAAM,CAAC,OAAO,GAAG,IAAI,GAAG,EAAuB,CAAC;IAEhD,MAAM,CAAC,KAAK,CAAC,OAAO,CAAI,GAAgB,EAAE,EAAoC;QACpF,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC,SAAS,CAAC,CAAC,gDAAgD;QACvF,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,CAAC;QAC/B,IAAI,CAAC;YACH,OAAO,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC;QACvB,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,OAAO,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;QAC/B,CAAC;IACH,CAAC;IAEM,MAAM,CAAC,YAAY,CAAC,QAAuB;QAChD,OAAO,IAAI,CAAC,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;IACpC,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,QAAQ,CAAI,OAAwB,EAAE,UAA4C;QACpG,MAAM,QAAQ,GAAG,OAAO,CAAC,SAAS,CAAC;QACnC,IAAI,GAAG,GAAG,IAAI,CAAC,YAAY,CAAC,QAAQ,CAAC,CAAC;QACtC,IAAI,SAAS,KAAK,GAAG;YACnB,OAAO,GAAG,CAAC,OAAO,CAAC;QAErB,uBAAU,CAAC,kBAAkB,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC;QACtD,GAAG,GAAG,EAAE,OAAO,EAAE,CAAC;QAClB,OAAO,GAAG,CAAC,OAAO,GAAG,IAAI,CAAC,OAAO,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC;IACrD,CAAC;;AA1BH,8BA2BC;AAED;;;EAGE;AACF,MAAa,mBAAmB;IACvB,MAAM,CAAU,cAAc,GAAG,aAAa,CAAC;IAC9C,MAAM,CAAC,WAAW,CAA0B;IAC5C,MAAM,CAAC,UAAU,GAAG,IAAI,GAAG,EAAsC,CAAC;IAE1E;;OAEG;IACI,MAAM,CAAC,CAAC,yBAAe,CAAC,CAAkB;IAG1C,MAAM,CAAC,SAAS;QACrB,MAAM,cAAc,GAAG,IAAI,CAAC,IAAI,CAAC,mCAAgB,CAAC,QAAQ,EAAE,mBAAmB,CAAC,cAAc,CAAC,CAAC;QAChG,IAAI,CAAC,CAAC,uBAAU,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC,EAAE,CAAC;YAC7C,uBAAU,CAAC,kBAAkB,CAAC,cAAc,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,cAAc,CAAC;IACxB,CAAC;IAED,gIAAgI;IACzH,MAAM,CAAC,OAAO;QACnB,KAAK,MAAM,CAAC,CAAC,EAAE,KAAK,CAAC,IAAI,IAAI,CAAC,UAAU,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,IAAI,KAAK,CAAC,WAAW;gBACnB,KAAK,CAAC,UAAU,CAAC,EAAE,MAAM,EAAE,IAAI,EAAE,CAAC,CAAC;QACvC,CAAC;QAED,yBAAW,CAAC,WAAW,CAAC,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,EAAE,OAAO,EAAE,CAAC;QAClE,IAAI,CAAC,WAAW,GAAG,SAAS,CAAC;QAC7B,IAAI,CAAC,UAAU,CAAC,KAAK,EAAE,CAAC;IAC1B,CAAC;IAEO,MAAM,KAAK,UAAU;QAC3B,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;YACtB,IAAI,QAAQ,GAAuB,OAAO,CAAC,GAAG,CAAC,oBAAoB,IAAI,IAAI,CAAC,SAAS,EAAE,CAAC;YACxF,mFAAmF;YACnF,IAAI,CAAC,CAAC,uBAAU,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,gBAAgB,CAAC,CAAC,CAAC;gBACjE,QAAQ,GAAG,SAAS,CAAC,CAAC,2CAA2C;YAEnE,IAAI,CAAC,WAAW,GAAG,yBAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,EAAE,SAAS,EAAE,IAAI,CAAC,cAAc,EAAE,QAAQ,EAAE,SAAS,EAAE,KAAK,EAAE,CAAC,CAAC;QACtH,CAAC;QACD,OAAO,IAAI,CAAC,WAAW,CAAC;IAC1B,CAAC;IAED,qKAAqK;IAC7J,MAAM,CAAC,qBAAqB,CAAC,IAA6B;QAChE,OAAO,EAAE,GAAG,IAAI,EAAE,OAAO,EAAE,WAAW,IAAI,CAAC,WAAW,wBAAwB,EAAE,WAAW,EAAE,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC;IACrI,CAAC;IAEO,MAAM,CAAC,YAAY,CAAC,OAAgC,EAAE,UAA2B;QACvF,IAAI,SAAS,GAAG,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QACzD,IAAI,SAAS,KAAK,SAAS,EAAE,CAAC;YAC5B,IAAI,OAAmF,CAAC;YACxF,IAAI,mBAAmB,GAAuB,CAAC,CAAC,CAAC;YACjD,yIAAyI;YACzI,2HAA2H;YAC3H,gJAAgJ;YAChJ,IAAI,SAAS,KAAK,UAAU,CAAC,WAAW,EAAE,CAAC;gBACzC,OAAO,GAAG,KAAK,IAAI,EAAE,CAAC,CAAC,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC,EAAE,QAAQ,IAAI,EAAE,CAAC;gBACnG,mBAAmB,GAAG,SAAS,CAAC;YAClC,CAAC;YACD,SAAS,GAAG,yBAAW,CAAC,oBAAoB,CAAC,EAAE,GAAG,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,EAAE,mBAAmB,EAAE,KAAK,EAAE,OAAO,CAAC,GAAG,CAAC,QAAQ,EAAE,OAAO,EAAE,CAAC,CAAC;YACpJ,IAAI,CAAC,UAAU,CAAC,GAAG,CAAC,OAAO,CAAC,WAAW,EAAE,SAAS,CAAC,CAAC;QACtD,CAAC;QACD,OAAO,SAAS,CAAC;IACnB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,MAAM,CAAC,UAA2B;QACpD,IAAI,IAAI,CAAC,yBAAe,CAAC,EAAE,kBAAkB;YAC3C,OAAO,EAAE,MAAM,EAAE,IAAI,CAAC,yBAAe,CAAC,CAAC,UAAU,CAAC,UAAU,CAAC,EAAE,SAAS,EAAE,SAAS,EAAE,CAAC;QAExF,IAAI,OAA4C,CAAC;QACjD,IAAI,CAAC;YACH,OAAO,GAAG,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC,UAAU,CAAC,CAAC;YACrE,IAAI,CAAC,OAAO;gBACV,MAAM,IAAI,KAAK,CAAC,eAAe,CAAC,CAAC;QACrC,CAAC;QAAC,OAAO,GAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,QAAQ,EAAE,iCAAiC,GAAG,CAAC,OAAO,EAAE,CAAC,CAAC;QAC/F,CAAC;QAED,IAAI,CAAC;YACH,MAAM,SAAS,GAAG,IAAI,CAAC,YAAY,CAAC,OAAO,EAAE,UAAU,CAAC,CAAC;YACzD,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC;YAC9B,8GAA8G;YAC9G,SAAS,CAAC,WAAW,GAAG,OAAO,CAAC,QAAQ,CAAC;YACzC,IAAI,CAAC,SAAS,CAAC,WAAW;gBACxB,SAAS,CAAC,OAAO,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC;YACrC,SAAS,CAAC,eAAe,EAAE,CAAC;YAC5B,MAAM,OAAO,GAAG,SAAS,CAAC,aAAa,CAAC,MAAM,CAAC,CAAC;YAChD,IAAI,uBAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,UAAU,CAAC,sBAAsB,EAAE,KAAK,CAAC,EAAE,CAAC;gBAC/E,MAAM,iBAAiB,GAAG,CAAC,IAAY,EAAE,UAAkB,EAAE,EAAE,CAAC,uBAAU,CAAC,YAAY,CAAC,QAAQ,CAAC,SAAS,CAAC,wBAAwB,IAAI,EAAE,EAAE,UAAU,CAAC,CAAC;gBACvJ,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,WAAW,GAAG,iBAAiB,CAAC,aAAa,EAAE,CAAC,CAAC,CAAC;gBACxD,MAAM,OAAO,GAAG,iBAAiB,CAAC,SAAS,EAAE,GAAG,CAAC,CAAC;gBAClD,MAAM,SAAS,GAAG,iBAAiB,CAAC,WAAW,EAAE,GAAG,CAAC,CAAC,CAAC,+DAA+D;gBACtH,IAAI,OAAO,EAAE,WAAW,KAAK,SAAS,IAAI,OAAO,CAAC,WAAW,IAAI,SAAS,IAAI,OAAO,CAAC,SAAS,KAAK,CAAC,EAAE,CAAC;oBACtG,MAAM,WAAW,GAAG,KAAK,EAAE,QAAmC,EAAE,EAAE;wBAChE,MAAM,SAAS,GAAG,IAAI,wBAAS,CAAC,IAAI,SAAS,CAAC,WAAW,IAAI,MAAM,GAAG,EAAE,IAAI,CAAC,CAAC;wBAC9E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,wBAAwB,SAAS,CAAC,WAAW,EAAE,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;wBACvH,MAAM,IAAI,GAAG,MAAM,QAAQ,CAAC,OAAO,CAAC;wBACpC,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,eAAe,SAAS,CAAC,WAAW,aAAa,IAAI,KAAK,SAAS,CAAC,cAAc,WAAW,EAAE,EAAE,WAAW,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;oBACvK,CAAC,CAAC;oBACF,mEAAmE;oBACnE,WAAW,CAAC,yBAAW,CAAC,kBAAkB,CAAC,SAAS,EAAE,MAAM,EAAE,EAAE,WAAW,EAAE,SAAS,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC,CAAC;gBACnH,CAAC;qBAAM,CAAC;oBACN,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2DAA2D,EAAE,EAAE,SAAS,EAAE,aAAa,EAAE,OAAO,EAAE,SAAS,EAAE,eAAe,EAAE,OAAO,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAChM,CAAC;YACH,CAAC;YACD,OAAO,EAAE,MAAM,EAAE,SAAS,EAAE,CAAC;QAC/B,CAAC;QAAC,OAAO,CAAM,EAAE,CAAC;YAChB,MAAM,KAAK,GAAG,+BAA+B,CAAC,CAAC,OAAO,EAAE,CAAC;YACzD,IAAI,UAAU,CAAC,QAAQ;gBACrB,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,KAAK,CAAC,CAAC;YAEzC,MAAM,IAAI,yBAAW,CAAC,CAAC,CAAC,WAAW,EAAE,KAAK,CAAC,CAAC;QAC9C,CAAC;IACH,CAAC;IAED,gBAAgB;IACR,MAAM,CAAC,KAAK,CAAC,eAAe,CAAC,GAAgB;QACnD,MAAM,OAAO,GAAG,GAAG,CAAC,OAAO,CAAC;QAC5B,IAAI,IAAI,CAAC,yBAAe,CAAC;YACvB,IAAI,CAAC,yBAAe,CAAC,CAAC,YAAY,CAAC,OAAO,CAAC,CAAC;aACzC,CAAC;YACJ,MAAM,OAAO,GAAwC,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,iBAAiB,CAAC,EAAE,GAAG,OAAO,CAAC,UAAU,EAAE,cAAc,EAAE,IAAI,EAAE,CAAC,CAAC;YACrJ,IAAI,CAAC,OAAO;gBACV,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,QAAQ,EAAE,yBAAyB,CAAC,CAAC;YAE1E,iBAAiB,CAAC,YAAY,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC;YAC/C,MAAM,SAAS,GAAG,yBAAW,CAAC,oBAAoB,CAAC,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC,CAAC;YACxF,MAAM,yBAAW,CAAC,UAAU,CAAC,UAAU,EAAE,SAAS,EAAE,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,aAAa,EAAE,OAAO,CAAC,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,CAAC,CAAC;QACpJ,CAAC;QACD,OAAO,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC;IACzC,CAAC;IAED;;;OAGG;IACI,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAwB;QAC7D,OAAO,SAAS,CAAC,QAAQ,CAAC,OAAO,EAAE,KAAK,EAAE,GAAgB,EAAE,EAAE,CAAC,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,CAAC,CAAC;IAC5F,CAAC;;AA5IH,kDA6IC;AAED,iBAAiB;AACjB,MAAa,iBAAiB;IACrB,MAAM,CAAU,YAAY,GAAG,IAAI,sBAAO,EAA8B,CAAC;IACzE,MAAM,CAAC,MAAM,CAAC,UAA2B,IAAI,OAAO,GAAG,UAAU,CAAC,QAAQ,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC;IAEzG,MAAM,CAAC,KAAK,CAAC,UAAU,CAAC,OAAwB;QACtD,kDAAkD;QAClD,MAAM,SAAS,GAAG,IAAI,wBAAS,CAAC,IAAI,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,GAAG,EAAE,IAAI,CAAC,CAAC;QAC9E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,8CAA8C,SAAS,CAAC,WAAW,EAAE,CAAC,CAAC;QACtG,MAAM,WAAW,GAAG,MAAM,mBAAmB,CAAC,kBAAkB,CAAC,OAAO,CAAC,CAAC;QAC1E,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,oCAAoC,SAAS,CAAC,WAAW,KAAK,SAAS,CAAC,cAAc,WAAW,CAAC,CAAC;QAClI,IAAI,WAAW,KAAK,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE;YACjD,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2EAA2E,EAAE,EAAE,oBAAoB,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;;YAEhQ,qBAAM,CAAC,OAAO,CAAC,cAAc,EAAE,2BAA2B,EAAE,EAAE,QAAQ,EAAE,OAAO,CAAC,UAAU,CAAC,QAAQ,EAAE,WAAW,EAAE,OAAO,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,OAAO,EAAE,OAAO,CAAC,UAAU,CAAC,OAAO,EAAE,CAAC,CAAC;QAC5L,OAAO,WAAW,CAAC;IACrB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,wBAAwB,CAAC,OAAwB;QACnE,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC;QACtC,MAAM,UAAU,GAAG,OAAO,CAAC,SAAS,CAAC;QACrC,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,SAAS,EAAE,UAAU,CAAC,SAAS,EAAE,CAAC;QAClH,IAAI,CAAC;YACH,4BAA4B;YAC5B,MAAM,YAAY,GAAG,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,CAAC,IAAI,uBAAQ,CAAC,KAAK,CAAC,CAAC,0CAA0C;YAC/H,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,EAAE,uBAAQ,CAAC,IAAI,CAAC,CAAC,CAAC,yDAAyD;YACtH,MAAM,EAAE,GAAG,qBAAU,CAAC,sBAAsB,CAAC,UAAU,CAAC,CAAC;YACzD,MAAM,QAAQ,GAAG,EAAE,CAAC,mBAAS,CAAC,CAAC;YAC/B,IAAI,CAAC;gBAEH,IAAI,QAAQ,CAAC,cAAc,EAAE,EAAE,CAAC;oBAC9B,qBAAM,CAAC,UAAU,CAAC,cAAc,EAAE,4CAA4C,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;oBACjG,QAAQ,CAAC,aAAa,EAAE,CAAC;gBAC3B,CAAC;gBAED,IAAI,QAAQ,CAAC,cAAc,EAAE,KAAK,8BAAgB,CAAC,UAAU;oBAC3D,QAAQ,CAAC,gBAAgB,CAAC,8BAAgB,CAAC,UAAU,CAAC,CAAC;gBAEzD,iBAAiB,CAAC,uBAAuB,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBAC1D,iCAAiC;gBACjC,MAAM,gBAAgB,GAAiC,QAAQ,CAAC,mBAAmB,EAAE,CAAC;gBACtF,IAAI,gBAAgB,CAAC,EAAE,KAAK,UAAU,CAAC,SAAS,CAAC,EAAE,EAAE,CAAC;oBACpD,MAAM,WAAW,GAAG,UAAU,CAAC,WAAW,CAAC;oBAC3C,MAAM,OAAO,GAAG,UAAU,CAAC,SAAS,CAAC,KAAK;wBACxC,CAAC,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,uBAAuB,CAAC,EAAE,WAAW,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,OAAO,EAAE,2BAAa,CAAC,aAAa,CAAC,UAAU,CAAC,SAAS,CAAC,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,KAAK,CAAC;oBAC9K,MAAM,mCAAgB,CAAC,sBAAsB,CAAC,EAAE,EAAE,EAAE,WAAW,EAAE,OAAO,EAAE,CAAC,CAAC;gBAC9E,CAAC;qBAAM,CAAC;oBACN,2FAA2F;oBAC3F,gBAAgB,CAAC,KAAK,GAAG,UAAU,CAAC,SAAS,CAAC,KAAM,CAAC,CAAC,+DAA+D;oBACrH,QAAQ,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,gBAAgB,CAAC,CAAC,CAAC;gBAC/E,CAAC;YACH,CAAC;oBAAS,CAAC;gBACT,qBAAM,CAAC,QAAQ,CAAC,sCAAoB,CAAC,MAAM,EAAE,YAAY,CAAC,CAAC,CAAC,oEAAoE;gBAChI,EAAE,CAAC,WAAW,EAAE,CAAC;gBACjB,EAAE,CAAC,KAAK,EAAE,CAAC;YACb,CAAC;QACH,CAAC;QAAC,OAAO,KAAU,EAAE,CAAC;YAEpB,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,4CAA4C,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;YAC/F,uBAAU,CAAC,UAAU,CAAC,UAAU,CAAC,CAAC;YAElC,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,qBAAqB,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,SAAS,IAAI,KAAK,CAAC,WAAW,KAAK,8BAAe,CAAC,cAAc,EAAE,CAAC;gBAC3K,qBAAM,CAAC,QAAQ,CAAC,cAAc,EAAE,+EAA+E,EAAE,GAAG,EAAE,CAAC,SAAS,CAAC,CAAC;gBAClI,mCAAgB,CAAC,6BAA6B,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YACtE,CAAC;YACD,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;IAED,uGAAuG;IAChG,MAAM,CAAC,KAAK,CAAC,kBAAkB,CAAC,OAAwB;QAC7D,IAAI,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,EAAE,OAAO,CAAC,SAAS,CAAC;YAC9D,OAAO;QAET,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;YACvB,KAAK,MAAM,KAAK,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;gBACvC,IAAI,IAAI,CAAC,gBAAgB,CAAC,OAAO,CAAC,UAAU,EAAE,KAAK,CAAC,EAAE,CAAC;oBACrD,OAAO,CAAC,SAAS,GAAG,KAAK,CAAC;oBAC1B,OAAO;gBACT,CAAC;YACH,CAAC;QACH,CAAC;QAED,MAAM,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,CAAC;QAC/B,OAAO,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,CAAC;IAChD,CAAC;IAED,6FAA6F;IACtF,MAAM,CAAC,uBAAuB,CAAC,UAA2B,EAAE,UAAsB;QACvF,MAAM,SAAS,GAAG,EAAE,OAAO,EAAE,UAAU,CAAC,OAAO,EAAE,QAAQ,EAAE,UAAU,CAAC,QAAQ,EAAE,CAAC;QAEjF,MAAM,QAAQ,GAAG,UAAU,CAAC,mBAAS,CAAC,CAAC;QACvC,MAAM,WAAW,GAAG,QAAQ,CAAC,mBAAmB,EAAE,CAAC;QACnD,MAAM,QAAQ,GAAG,mBAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,WAAW,EAAE,CAAC,CAAC;QACxD,IAAI,QAAQ,KAAK,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,EAAE,CAAC;YACrD,IAAI,QAAQ,CAAC,UAAU,EAAE;gBACvB,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,gBAAgB,EAAE,mDAAmD,CAAC,CAAC;YAE5G,qBAAM,CAAC,UAAU,CAAC,cAAc,EAAE,gGAAgG,EAAE,GAAG,EAAE,CAAC,CAAC,EAAE,GAAG,SAAS,EAAE,MAAM,EAAE,QAAQ,EAAE,CAAC,CAAC,CAAC;YAChL,MAAM,kBAAkB,GAAG,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;YAC/D,QAAQ,CAAC,WAAW,CAAC,kBAAkB,CAAC,CAAC;YACxC,UAAkB,CAAC,SAAS,GAAG,kBAAkB,CAAC;YACnD,wEAAwE;YACxE,QAAQ,CAAC,cAAc,CAAC,mBAAmB,EAAE,WAAW,CAAC,EAAE,CAAC,CAAC;YAC7D,IAAI,SAAS,KAAK,WAAW,CAAC,KAAK;gBACjC,QAAQ,CAAC,cAAc,CAAC,iBAAiB,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC,CAAC,CAAC;QAC5E,CAAC;QAED,MAAM,OAAO,GAAG,mBAAI,CAAC,SAAS,CAAC,QAAQ,CAAC,UAAU,EAAE,CAAC,CAAC;QACtD,IAAI,OAAO,KAAK,mBAAI,CAAC,SAAS,CAAC,UAAU,CAAC,OAAO,CAAC;YAChD,MAAM,IAAI,yBAAW,CAAC,2BAAY,CAAC,gBAAgB,EAAE,mDAAmD,CAAC,CAAC;IAC9G,CAAC;IAED,4DAA4D;IACrD,MAAM,CAAC,gBAAgB,CAAC,UAA2B,EAAE,QAAuB;QACjF,IAAI,CAAC,uBAAU,CAAC,UAAU,CAAC,QAAQ,CAAC;YAClC,OAAO,KAAK,CAAC;QAEf,MAAM,QAAQ,GAAG,IAAI,6BAAY,CAAC,QAAQ,CAAC,KAAK,EAAE,CAAC;QACnD,IAAI,CAAC;YACH,QAAQ,CAAC,UAAU,CAAC,QAAQ,EAAE,uBAAQ,CAAC,QAAQ,CAAC,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACP,OAAO,KAAK,CAAC;QACf,CAAC;QAED,MAAM,OAAO,GAAG,UAAU,CAAC,QAAQ,KAAK,QAAQ,CAAC,WAAW,EAAE,IAAI,UAAU,CAAC,SAAS,CAAC,EAAE,KAAK,QAAQ,CAAC,mBAAmB,EAAE,CAAC,EAAE,CAAC;QAChI,QAAQ,CAAC,SAAS,EAAE,CAAC;QACrB,IAAI,CAAC,OAAO;YACV,uBAAU,CAAC,UAAU,CAAC,QAAQ,CAAC,CAAC;QAElC,OAAO,OAAO,CAAC;IACjB,CAAC;IAEM,MAAM,CAAC,KAAK,CAAC,iBAAiB,CAAC,IAAwB;QAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,SAAS,IAAI,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,kBAAkB,CAAC,EAAE,GAAG,IAAI,EAAE,WAAW,EAAE,MAAM,uBAAU,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC;QAEjJ,OAAO;YACL,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,OAAO,EAAE,IAAI,CAAC,OAAO;YACrB,SAAS,EAAE;gBACT,KAAK,EAAE,SAAS,CAAC,KAAK;gBACtB,EAAE,EAAE,SAAS,CAAC,EAAE,IAAI,CAAC,MAAM,uBAAU,CAAC,oBAAU,CAAC,CAAC,cAAc,CAAC,EAAE,GAAG,IAAI,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,uBAAU,CAAC,cAAc,EAAE,EAAE,CAAC,CAAC,CAAC,EAAE;aAC7I;SACF,CAAC;IACJ,CAAC;;AA/IH,8CAgJC","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 iModels\r\n */\r\n\r\n// cspell:ignore BLOCKCACHE\r\n\r\nimport * as path from \"path\";\r\nimport { NativeLoggerCategory } from \"@bentley/imodeljs-native\";\r\nimport { AccessToken, BeEvent, ChangeSetStatus, Guid, GuidString, IModelStatus, Logger, LogLevel, Mutable, OpenMode, StopWatch } from \"@itwin/core-bentley\";\r\nimport {\r\n  BriefcaseIdValue, ChangesetId, ChangesetIdWithIndex, ChangesetIndexAndId, IModelError, IModelVersion, LocalDirName, LocalFileName, OpenCheckpointArgs,\r\n} from \"@itwin/core-common\";\r\nimport { V2CheckpointAccessProps } from \"./BackendHubAccess\";\r\nimport { BackendLoggerCategory } from \"./BackendLoggerCategory\";\r\nimport { BriefcaseManager } from \"./BriefcaseManager\";\r\nimport { CloudSqlite } from \"./CloudSqlite\";\r\nimport { IModelHost } from \"./IModelHost\";\r\nimport { IModelJsFs } from \"./IModelJsFs\";\r\nimport { SnapshotDb, TokenArg } from \"./IModelDb\";\r\nimport { IModelNative } from \"./internal/NativePlatform\";\r\nimport { _hubAccess, _mockCheckpoint, _nativeDb } from \"./internal/Symbols\";\r\n\r\nconst loggerCategory = BackendLoggerCategory.IModelDb;\r\n\r\n/** @internal */\r\nexport interface MockCheckpoint {\r\n  mockAttach(checkpoint: CheckpointProps): string;\r\n  mockDownload(_request: DownloadRequest): void;\r\n}\r\n\r\n/**\r\n * Properties of a checkpoint\r\n * @public\r\n */\r\nexport interface CheckpointProps extends TokenArg {\r\n  readonly expectV2?: boolean;\r\n\r\n  /** iTwin that the iModel belongs to */\r\n  readonly iTwinId: GuidString;\r\n\r\n  /** Id of the iModel */\r\n  readonly iModelId: GuidString;\r\n\r\n  /** changeset for the checkpoint */\r\n  readonly changeset: ChangesetIdWithIndex;\r\n\r\n  /** If true, then the latest successful v2 checkpoint at or before the provided changeset will be returned when calling queryV2Checkpoint.  */\r\n  readonly allowPreceding?: boolean;\r\n\r\n  /** The number of seconds before the current token expires to attempt to reacquire a new token. Default is 1 hour. */\r\n  readonly reattachSafetySeconds?: number;\r\n}\r\n\r\n/** Return value from [[ProgressFunction]].\r\n *  @public\r\n */\r\nexport enum ProgressStatus {\r\n  /** Continue download. */\r\n  Continue = 0,\r\n  /** Abort download. */\r\n  Abort = 1,\r\n}\r\n\r\n/** Called to show progress during a download. If this function returns non-zero, the download is aborted.\r\n *  @public\r\n */\r\nexport type ProgressFunction = (loaded: number, total: number) => ProgressStatus;\r\n\r\n/** The parameters that specify a request to download a checkpoint file from iModelHub.\r\n * @internal\r\n */\r\nexport interface DownloadRequest {\r\n  /** name of local file to hold the downloaded data. */\r\n  localFile: LocalFileName;\r\n\r\n  /** A list of full fileName paths to test before downloading. If a valid file exists by one of these names,\r\n   * no download is performed and `localFile` is updated to reflect the fact that the file exists with that name.\r\n   * This can be used, for example, to look for checkpoints from previous versions if the naming strategy changes.\r\n   */\r\n  readonly aliasFiles?: ReadonlyArray<string>;\r\n\r\n  /** Properties of the checkpoint to be downloaded */\r\n  readonly checkpoint: CheckpointProps;\r\n\r\n  /** If present, this function will be called to indicate progress as the briefcase is downloaded. If this\r\n   * function returns a non-zero value, the download is aborted.\r\n   */\r\n  readonly onProgress?: ProgressFunction;\r\n}\r\n\r\n/** @internal */\r\nexport interface DownloadJob {\r\n  request: DownloadRequest;\r\n  promise?: Promise<any>;\r\n}\r\n\r\n/** @internal */\r\nexport class Downloads {\r\n  private static _active = new Map<string, DownloadJob>();\r\n\r\n  private static async process<T>(job: DownloadJob, fn: (job: DownloadJob) => Promise<T>) {\r\n    const jobName = job.request.localFile; // save this, it can change inside call to `fn`!\r\n    this._active.set(jobName, job);\r\n    try {\r\n      return await fn(job);\r\n    } finally {\r\n      this._active.delete(jobName);\r\n    }\r\n  }\r\n\r\n  public static isInProgress(pathName: LocalFileName): DownloadJob | undefined {\r\n    return this._active.get(pathName);\r\n  }\r\n\r\n  public static async download<T>(request: DownloadRequest, downloadFn: (job: DownloadJob) => Promise<T>) {\r\n    const pathName = request.localFile;\r\n    let job = this.isInProgress(pathName);\r\n    if (undefined !== job)\r\n      return job.promise;\r\n\r\n    IModelJsFs.recursiveMkDirSync(path.dirname(pathName));\r\n    job = { request };\r\n    return job.promise = this.process(job, downloadFn);\r\n  }\r\n}\r\n\r\n/**\r\n * Utility class for opening V2 checkpoints from cloud containers, and also for downloading them.\r\n * @internal\r\n*/\r\nexport class V2CheckpointManager {\r\n  public static readonly cloudCacheName = \"Checkpoints\";\r\n  private static _cloudCache?: CloudSqlite.CloudCache;\r\n  private static containers = new Map<string, CloudSqlite.CloudContainer>();\r\n\r\n  /** used by HubMock\r\n   * @internal\r\n   */\r\n  public static [_mockCheckpoint]?: MockCheckpoint;\r\n\r\n\r\n  public static getFolder(): LocalDirName {\r\n    const cloudCachePath = path.join(BriefcaseManager.cacheDir, V2CheckpointManager.cloudCacheName);\r\n    if (!(IModelJsFs.existsSync(cloudCachePath))) {\r\n      IModelJsFs.recursiveMkDirSync(cloudCachePath);\r\n    }\r\n    return cloudCachePath;\r\n  }\r\n\r\n  /* only used by tests that reset the state of the v2CheckpointManager. all dbs should be closed before calling this function. */\r\n  public static cleanup(): void {\r\n    for (const [_, value] of this.containers.entries()) {\r\n      if (value.isConnected)\r\n        value.disconnect({ detach: true });\r\n    }\r\n\r\n    CloudSqlite.CloudCaches.dropCache(this.cloudCacheName)?.destroy();\r\n    this._cloudCache = undefined;\r\n    this.containers.clear();\r\n  }\r\n\r\n  private static get cloudCache(): CloudSqlite.CloudCache {\r\n    if (!this._cloudCache) {\r\n      let cacheDir: string | undefined = process.env.CHECKPOINT_CACHE_DIR ?? this.getFolder();\r\n      // See if there is a daemon running, otherwise use profile directory for cloudCache\r\n      if (!(IModelJsFs.existsSync(path.join(cacheDir, \"portnumber.bcv\"))))\r\n        cacheDir = undefined; // no daemon running, use profile directory\r\n\r\n      this._cloudCache = CloudSqlite.CloudCaches.getCache({ cacheName: this.cloudCacheName, cacheDir, cacheSize: \"50G\" });\r\n    }\r\n    return this._cloudCache;\r\n  }\r\n\r\n  /** Member names differ slightly between the V2Checkpoint api and the CloudSqlite api. Add aliases `accessName` for `accountName` and `accessToken` for `sasToken` */\r\n  private static toCloudContainerProps(from: V2CheckpointAccessProps): CloudSqlite.ContainerAccessProps {\r\n    return { ...from, baseUri: `https://${from.accountName}.blob.core.windows.net`, accessToken: from.sasToken, storageType: \"azure\" };\r\n  }\r\n\r\n  private static getContainer(v2Props: V2CheckpointAccessProps, checkpoint: CheckpointProps) {\r\n    let container = this.containers.get(v2Props.containerId);\r\n    if (undefined === container) {\r\n      let tokenFn: ((args: CloudSqlite.RequestTokenArgs) => Promise<AccessToken>) | undefined;\r\n      let tokenRefreshSeconds: number | undefined = -1;\r\n      // from Rpc, the accessToken in the checkpoint request is from the current user. It is used to request the sasToken for the container and\r\n      // the sasToken is checked for refresh (before it expires) on every Rpc request using that user's accessToken. For Ipc, the\r\n      // accessToken in the checkpoint request is undefined, and the sasToken is requested by IModelHost.getAccessToken(). It is refreshed on a timer.\r\n      if (undefined === checkpoint.accessToken) {\r\n        tokenFn = async () => (await IModelHost[_hubAccess].queryV2Checkpoint(checkpoint))?.sasToken ?? \"\";\r\n        tokenRefreshSeconds = undefined;\r\n      }\r\n      container = CloudSqlite.createCloudContainer({ ...this.toCloudContainerProps(v2Props), tokenRefreshSeconds, logId: process.env.POD_NAME, tokenFn });\r\n      this.containers.set(v2Props.containerId, container);\r\n    }\r\n    return container;\r\n  }\r\n\r\n  public static async attach(checkpoint: CheckpointProps): Promise<{ dbName: string, container: CloudSqlite.CloudContainer | undefined }> {\r\n    if (this[_mockCheckpoint]) // used by HubMock\r\n      return { dbName: this[_mockCheckpoint].mockAttach(checkpoint), container: undefined };\r\n\r\n    let v2props: V2CheckpointAccessProps | undefined;\r\n    try {\r\n      v2props = await IModelHost[_hubAccess].queryV2Checkpoint(checkpoint);\r\n      if (!v2props)\r\n        throw new Error(\"no checkpoint\");\r\n    } catch (err: any) {\r\n      throw new IModelError(IModelStatus.NotFound, `V2 checkpoint not found: err: ${err.message}`);\r\n    }\r\n\r\n    try {\r\n      const container = this.getContainer(v2props, checkpoint);\r\n      const dbName = v2props.dbName;\r\n      // Use the new token from the recently queried v2 checkpoint just incase the one we currently have is expired.\r\n      container.accessToken = v2props.sasToken;\r\n      if (!container.isConnected)\r\n        container.connect(this.cloudCache);\r\n      container.checkForChanges();\r\n      const dbStats = container.queryDatabase(dbName);\r\n      if (IModelHost.appWorkspace.settings.getBoolean(\"Checkpoints/prefetch\", false)) {\r\n        const getPrefetchConfig = (name: string, defaultVal: number) => IModelHost.appWorkspace.settings.getNumber(`Checkpoints/prefetch/${name}`, defaultVal);\r\n        const minRequests = getPrefetchConfig(\"minRequests\", 3);\r\n        const maxRequests = getPrefetchConfig(\"maxRequests\", 6);\r\n        const timeout = getPrefetchConfig(\"timeout\", 100);\r\n        const maxBlocks = getPrefetchConfig(\"maxBlocks\", 500); // default size of 2GB. Assumes a checkpoint block size of 4MB.\r\n        if (dbStats?.totalBlocks !== undefined && dbStats.totalBlocks <= maxBlocks && dbStats.nPrefetch === 0) {\r\n          const logPrefetch = async (prefetch: CloudSqlite.CloudPrefetch) => {\r\n            const stopwatch = new StopWatch(`[${container.containerId}/${dbName}]`, true);\r\n            Logger.logInfo(loggerCategory, `Starting prefetch of ${stopwatch.description}`, { minRequests, maxRequests, timeout });\r\n            const done = await prefetch.promise;\r\n            Logger.logInfo(loggerCategory, `Prefetch of ${stopwatch.description} complete=${done} (${stopwatch.elapsedSeconds} seconds)`, { minRequests, maxRequests, timeout });\r\n          };\r\n          // eslint-disable-next-line @typescript-eslint/no-floating-promises\r\n          logPrefetch(CloudSqlite.startCloudPrefetch(container, dbName, { minRequests, nRequests: maxRequests, timeout }));\r\n        } else {\r\n          Logger.logInfo(loggerCategory, `Skipping prefetch due to size limits or ongoing prefetch.`, { maxBlocks, numPrefetches: dbStats?.nPrefetch, totalBlocksInDb: dbStats?.totalBlocks, v2props });\r\n        }\r\n      }\r\n      return { dbName, container };\r\n    } catch (e: any) {\r\n      const error = `Cloud cache connect failed: ${e.message}`;\r\n      if (checkpoint.expectV2)\r\n        Logger.logError(loggerCategory, error);\r\n\r\n      throw new IModelError(e.errorNumber, error);\r\n    }\r\n  }\r\n\r\n  /** @internal */\r\n  private static async performDownload(job: DownloadJob): Promise<ChangesetId> {\r\n    const request = job.request;\r\n    if (this[_mockCheckpoint])\r\n      this[_mockCheckpoint].mockDownload(request);\r\n    else {\r\n      const v2props: V2CheckpointAccessProps | undefined = await IModelHost[_hubAccess].queryV2Checkpoint({ ...request.checkpoint, allowPreceding: true });\r\n      if (!v2props)\r\n        throw new IModelError(IModelStatus.NotFound, \"V2 checkpoint not found\");\r\n\r\n      CheckpointManager.onDownloadV2.raiseEvent(job);\r\n      const container = CloudSqlite.createCloudContainer(this.toCloudContainerProps(v2props));\r\n      await CloudSqlite.transferDb(\"download\", container, { dbName: v2props.dbName, localFileName: request.localFile, onProgress: request.onProgress });\r\n    }\r\n    return request.checkpoint.changeset.id;\r\n  }\r\n\r\n  /** Fully download a V2 checkpoint to a local file that can be used to create a briefcase or to work offline.\r\n   * @returns a Promise that is resolved when the download completes with the changesetId of the downloaded checkpoint (which will\r\n   * be the same as the requested changesetId or the most recent checkpoint before it.)\r\n   */\r\n  public static async downloadCheckpoint(request: DownloadRequest): Promise<ChangesetId> {\r\n    return Downloads.download(request, async (job: DownloadJob) => this.performDownload(job));\r\n  }\r\n}\r\n\r\n/** @internal  */\r\nexport class CheckpointManager {\r\n  public static readonly onDownloadV2 = new BeEvent<(job: DownloadJob) => void>();\r\n  public static getKey(checkpoint: CheckpointProps) { return `${checkpoint.iModelId}:${checkpoint.changeset.id}`; }\r\n\r\n  private static async doDownload(request: DownloadRequest): Promise<ChangesetId> {\r\n    // first see if there's a V2 checkpoint available.\r\n    const stopwatch = new StopWatch(`[${request.checkpoint.changeset.id}]`, true);\r\n    Logger.logInfo(loggerCategory, `Starting download of V2 checkpoint with id ${stopwatch.description}`);\r\n    const changesetId = await V2CheckpointManager.downloadCheckpoint(request);\r\n    Logger.logInfo(loggerCategory, `Downloaded V2 checkpoint with id ${stopwatch.description} (${stopwatch.elapsedSeconds} seconds)`);\r\n    if (changesetId !== request.checkpoint.changeset.id)\r\n      Logger.logInfo(loggerCategory, `Downloaded previous v2 checkpoint because requested checkpoint not found.`, { requestedChangesetId: request.checkpoint.changeset.id, iModelId: request.checkpoint.iModelId, changesetId, iTwinId: request.checkpoint.iTwinId });\r\n    else\r\n      Logger.logInfo(loggerCategory, `Downloaded v2 checkpoint.`, { iModelId: request.checkpoint.iModelId, changesetId: request.checkpoint.changeset.id, iTwinId: request.checkpoint.iTwinId });\r\n    return changesetId;\r\n  }\r\n\r\n  public static async updateToRequestedVersion(request: DownloadRequest) {\r\n    const checkpoint = request.checkpoint;\r\n    const targetFile = request.localFile;\r\n    const traceInfo = { iTwinId: checkpoint.iTwinId, iModelId: checkpoint.iModelId, changeset: checkpoint.changeset };\r\n    try {\r\n      // Open checkpoint for write\r\n      const prevLogLevel = Logger.getLevel(NativeLoggerCategory.SQLite) ?? LogLevel.Error; // Get log level before we set it to None.\r\n      Logger.setLevel(NativeLoggerCategory.SQLite, LogLevel.None); // Ignores noisy error messages when applying changesets.\r\n      const db = SnapshotDb.openForApplyChangesets(targetFile);\r\n      const nativeDb = db[_nativeDb];\r\n      try {\r\n\r\n        if (nativeDb.hasPendingTxns()) {\r\n          Logger.logWarning(loggerCategory, \"Checkpoint with Txns found - deleting them\", () => traceInfo);\r\n          nativeDb.deleteAllTxns();\r\n        }\r\n\r\n        if (nativeDb.getBriefcaseId() !== BriefcaseIdValue.Unassigned)\r\n          nativeDb.resetBriefcaseId(BriefcaseIdValue.Unassigned);\r\n\r\n        CheckpointManager.validateCheckpointGuids(checkpoint, db);\r\n        // Apply change sets if necessary\r\n        const currentChangeset: Mutable<ChangesetIndexAndId> = nativeDb.getCurrentChangeset();\r\n        if (currentChangeset.id !== checkpoint.changeset.id) {\r\n          const accessToken = checkpoint.accessToken;\r\n          const toIndex = checkpoint.changeset.index ??\r\n            (await IModelHost[_hubAccess].getChangesetFromVersion({ accessToken, iModelId: checkpoint.iModelId, version: IModelVersion.asOfChangeSet(checkpoint.changeset.id) })).index;\r\n          await BriefcaseManager.pullAndApplyChangesets(db, { accessToken, toIndex });\r\n        } else {\r\n          // make sure the parent changeset index is saved in the file - old versions didn't have it.\r\n          currentChangeset.index = checkpoint.changeset.index!; // eslint-disable-line @typescript-eslint/no-non-null-assertion\r\n          nativeDb.saveLocalValue(\"parentChangeSet\", JSON.stringify(currentChangeset));\r\n        }\r\n      } finally {\r\n        Logger.setLevel(NativeLoggerCategory.SQLite, prevLogLevel); // Set logging to what it was before we started applying changesets.\r\n        db.saveChanges();\r\n        db.close();\r\n      }\r\n    } catch (error: any) {\r\n\r\n      Logger.logError(loggerCategory, \"Error downloading checkpoint - deleting it\", () => traceInfo);\r\n      IModelJsFs.removeSync(targetFile);\r\n\r\n      if (error.errorNumber === ChangeSetStatus.CorruptedChangeStream || error.errorNumber === ChangeSetStatus.InvalidId || error.errorNumber === ChangeSetStatus.InvalidVersion) {\r\n        Logger.logError(loggerCategory, \"Detected potential corruption of change sets. Deleting them to enable retries\", () => traceInfo);\r\n        BriefcaseManager.deleteChangeSetsFromLocalDisk(checkpoint.iModelId);\r\n      }\r\n      throw error;\r\n    }\r\n  }\r\n\r\n  /** Download a checkpoint file from iModelHub into a local file specified in the request parameters. */\r\n  public static async downloadCheckpoint(request: DownloadRequest): Promise<void> {\r\n    if (this.verifyCheckpoint(request.checkpoint, request.localFile))\r\n      return;\r\n\r\n    if (request.aliasFiles) {\r\n      for (const alias of request.aliasFiles) {\r\n        if (this.verifyCheckpoint(request.checkpoint, alias)) {\r\n          request.localFile = alias;\r\n          return;\r\n        }\r\n      }\r\n    }\r\n\r\n    await this.doDownload(request);\r\n    return this.updateToRequestedVersion(request);\r\n  }\r\n\r\n  /** checks a file's dbGuid & iTwinId for consistency, and updates the dbGuid when possible */\r\n  public static validateCheckpointGuids(checkpoint: CheckpointProps, snapshotDb: SnapshotDb) {\r\n    const traceInfo = { iTwinId: checkpoint.iTwinId, iModelId: checkpoint.iModelId };\r\n\r\n    const nativeDb = snapshotDb[_nativeDb];\r\n    const dbChangeset = nativeDb.getCurrentChangeset();\r\n    const iModelId = Guid.normalize(nativeDb.getIModelId());\r\n    if (iModelId !== Guid.normalize(checkpoint.iModelId)) {\r\n      if (nativeDb.isReadonly())\r\n        throw new IModelError(IModelStatus.ValidationFailed, \"iModelId is not properly set up in the checkpoint\");\r\n\r\n      Logger.logWarning(loggerCategory, \"iModelId is not properly set up in the checkpoint. Updated checkpoint to the correct iModelId.\", () => ({ ...traceInfo, dbGuid: iModelId }));\r\n      const iModelIdNormalized = Guid.normalize(checkpoint.iModelId);\r\n      nativeDb.setIModelId(iModelIdNormalized);\r\n      (snapshotDb as any)._iModelId = iModelIdNormalized;\r\n      // Required to reset the ChangeSetId because setDbGuid clears the value.\r\n      nativeDb.saveLocalValue(\"ParentChangeSetId\", dbChangeset.id);\r\n      if (undefined !== dbChangeset.index)\r\n        nativeDb.saveLocalValue(\"parentChangeSet\", JSON.stringify(dbChangeset));\r\n    }\r\n\r\n    const iTwinId = Guid.normalize(nativeDb.getITwinId());\r\n    if (iTwinId !== Guid.normalize(checkpoint.iTwinId))\r\n      throw new IModelError(IModelStatus.ValidationFailed, \"iTwinId was not properly set up in the checkpoint\");\r\n  }\r\n\r\n  /** @returns true if the file is the checkpoint requested */\r\n  public static verifyCheckpoint(checkpoint: CheckpointProps, fileName: LocalFileName): boolean {\r\n    if (!IModelJsFs.existsSync(fileName))\r\n      return false;\r\n\r\n    const nativeDb = new IModelNative.platform.DgnDb();\r\n    try {\r\n      nativeDb.openIModel(fileName, OpenMode.Readonly);\r\n    } catch {\r\n      return false;\r\n    }\r\n\r\n    const isValid = checkpoint.iModelId === nativeDb.getIModelId() && checkpoint.changeset.id === nativeDb.getCurrentChangeset().id;\r\n    nativeDb.closeFile();\r\n    if (!isValid)\r\n      IModelJsFs.removeSync(fileName);\r\n\r\n    return isValid;\r\n  }\r\n\r\n  public static async toCheckpointProps(args: OpenCheckpointArgs): Promise<CheckpointProps> {\r\n    const changeset = args.changeset ?? await IModelHost[_hubAccess].getLatestChangeset({ ...args, accessToken: await IModelHost.getAccessToken() });\r\n\r\n    return {\r\n      iModelId: args.iModelId,\r\n      iTwinId: args.iTwinId,\r\n      changeset: {\r\n        index: changeset.index,\r\n        id: changeset.id ?? (await IModelHost[_hubAccess].queryChangeset({ ...args, changeset, accessToken: await IModelHost.getAccessToken() })).id,\r\n      },\r\n    };\r\n  }\r\n}\r\n"]}

package/lib/cjs/ClassRegistry.d.ts

@@ -6,13 +6,43 @@ import { EntityMetaData, IModelError } from "@itwin/core-common";
 import { Entity } from "./Entity";
 import { IModelDb } from "./IModelDb";
 import { Schema } from "./Schema";
+/** Maintains the mapping between the name of a BIS [ECClass]($ecschema-metadata) (in "schema:class" format) and the JavaScript [[Entity]] class that implements it.
+ * @public
+ */
+export declare class EntityJsClassMap {
+    private readonly _classMap;
+    /** @internal */
+    has(classFullName: string): boolean;
+    /** @internal */
+    get(classFullName: string): typeof Entity | undefined;
+    /** @internal */
+    set(classFullName: string, entityClass: typeof Entity): void;
+    /** @internal */
+    delete(classFullName: string): boolean;
+    /** @internal */
+    clear(): void;
+    /** @internal */
+    [Symbol.iterator](): IterableIterator<[string, typeof Entity]>;
+    /**
+     * Registers a single `entityClass` defined in the specified `schema`.
+     * This method registers the class globally. To register a class for a specific iModel, use [[IModelDb.jsClassMap]].
+     *
+     * @param entityClass - The JavaScript class that implements the BIS [ECClass](@itwin/core-common) to be registered.
+     * @param schema - The schema that contains the `entityClass`.
+     *
+     * @throws Error if the class is already registered.
+     *
+     * @public
+     */
+    register(entityClass: typeof Entity, schema: typeof Schema): void;
+}
 /** Maintains the mapping between the name of a BIS [ECClass]($ecschema-metadata) (in "schema:class" format) and the JavaScript [[Entity]] class that implements it.
  * Applications or modules that supply their own Entity subclasses should use [[registerModule]] or [[register]] at startup
  * to establish their mappings.
  * @public
  */
 export declare class ClassRegistry {
-    private static readonly _classMap;
+    private static readonly _globalClassMap;
     /** @internal */
     static isNotFoundError(err: any): boolean;
     /** @internal */
@@ -65,7 +95,7 @@ export declare class ClassRegistry {
      * @return true if the class was unregistered
      * @internal
      */
-    static unregisterCLass(classFullName: string): boolean;
+    static unregisterClass(classFullName: string): boolean;
     /** Unregister all classes from a schema.
      * This function is not normally needed, but is useful for cases where a generated *proxy* schema needs to be replaced by the *real* schema.
      * @param schema Name of the schema to unregister
@@ -77,6 +107,20 @@ export declare class ClassRegistry {
  * A cache that records the mapping between class names and class metadata.
  * @see [[IModelDb.classMetaDataRegistry]] to access the registry for a specific iModel.
  * @internal
+ * @deprecated in 5.0. Please use `schemaContext` from the `iModel` instead.
+ *
+ * @example
+ * @
+ * Current Usage:
+ * ```ts
+ * const metaData: EntityMetaData | undefined = iModel.classMetaDataRegistry.find("SchemaName:ClassName");
+ * ```
+ *
+ * Replacement:
+ * ```ts
+ * const entityMetaData: EntityClass | undefined = iModel.schemaContext.getSchemaItemSync("SchemaName.ClassName", EntityClass);
+ * const relationshipMetaData: RelationshipClass | undefined = iModel.schemaContext.getSchemaItemSync("SchemaName", "ClassName", RelationshipClass);
+ * ```
  */
 export declare class MetaDataRegistry {
     private _registry;

package/lib/cjs/ClassRegistry.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ClassRegistry.d.ts","sourceRoot":"","sources":["../../src/ClassRegistry.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAkB,UAAU,EAAwB,MAAM,qBAAqB,CAAC;AACvF,OAAO,EAAE,cAAc,EAAsB,WAAW,EAAkB,MAAM,oBAAoB,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,MAAM,EAAW,MAAM,UAAU,CAAC;AAO3C;;;;GAIG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,SAAS,CAAoC;IACrE,gBAAgB;WACF,eAAe,CAAC,GAAG,EAAE,GAAG;IACtC,gBAAgB;WACF,yBAAyB,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW;IACvE;;;OAGG;WACW,QAAQ,CAAC,WAAW,EAAE,OAAO,MAAM,EAAE,MAAM,EAAE,OAAO,MAAM;IAYxE,yEAAyE;IACzE,OAAO,CAAC,MAAM,CAAC,mBAAmB;IAqBlC;;;;;;OAMG;WACW,aAAa,CAAC,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,GAAG,MAAM;IAkB9E;;OAEG;IACH,OAAO,CAAC,MAAM,CAAC,sBAAsB;IAyGrC;;;;OAIG;WACW,cAAc,CAAC,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,MAAM;IAQlE;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,aAAa;IAa5B;;;;OAIG;WACW,mBAAmB,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,MAAM,GAAG,SAAS;IAInF;;;;OAIG;WACW,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,OAAO,MAAM;IAM9E;;;;;OAKG;WACW,eAAe,CAAC,aAAa,EAAE,MAAM;IACnD;;;;OAIG;WACW,qBAAqB,CAAC,MAAM,EAAE,OAAO,MAAM;CAM1D;AAED;;;;GAIG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,cAAc,CAAiC;IAEvD,wCAAwC;IACjC,IAAI,CAAC,aAAa,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAIvD,aAAa,CAAC,OAAO,EAAE,UAAU,GAAG,cAAc,GAAG,SAAS;IAKrE,gCAAgC;IACzB,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,GAAG,IAAI;CAKlE"}
+{"version":3,"file":"ClassRegistry.d.ts","sourceRoot":"","sources":["../../src/ClassRegistry.ts"],"names":[],"mappings":"AAIA;;GAEG;AAEH,OAAO,EAAkB,UAAU,EAAwB,MAAM,qBAAqB,CAAC;AACvF,OAAO,EAAE,cAAc,EAAsB,WAAW,EAAkB,MAAM,oBAAoB,CAAC;AACrG,OAAO,EAAE,MAAM,EAAE,MAAM,UAAU,CAAC;AAClC,OAAO,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAC;AACtC,OAAO,EAAE,MAAM,EAAW,MAAM,UAAU,CAAC;AAO3C;;GAEG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAoC;IAE9D,gBAAgB;IACT,GAAG,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO;IAI1C,gBAAgB;IACT,GAAG,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,MAAM,GAAG,SAAS;IAI5D,gBAAgB;IACT,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,WAAW,EAAE,OAAO,MAAM,GAAG,IAAI;IAInE,gBAAgB;IACT,MAAM,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO;IAI7C,gBAAgB;IACT,KAAK,IAAI,IAAI;IAIpB,gBAAgB;IACT,CAAC,MAAM,CAAC,QAAQ,CAAC,IAAI,gBAAgB,CAAC,CAAC,MAAM,EAAE,OAAO,MAAM,CAAC,CAAC;IAIrE;;;;;;;;;;OAUG;IACI,QAAQ,CAAC,WAAW,EAAE,OAAO,MAAM,EAAE,MAAM,EAAE,OAAO,MAAM,GAAG,IAAI;CAUzE;AAED;;;;GAIG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,MAAM,CAAC,QAAQ,CAAC,eAAe,CAA0B;IACjE,gBAAgB;WACF,eAAe,CAAC,GAAG,EAAE,GAAG;IACtC,gBAAgB;WACF,yBAAyB,CAAC,SAAS,EAAE,MAAM,GAAG,WAAW;IACvE;;;OAGG;WACW,QAAQ,CAAC,WAAW,EAAE,OAAO,MAAM,EAAE,MAAM,EAAE,OAAO,MAAM;IAIxE,yEAAyE;IACzE,OAAO,CAAC,MAAM,CAAC,mBAAmB;IAqBlC;;;;;;OAMG;WACW,aAAa,CAAC,MAAM,EAAE,QAAQ,EAAE,eAAe,EAAE,MAAM,GAAG,MAAM;IAkB9E;;OAEG;IAEH,OAAO,CAAC,MAAM,CAAC,sBAAsB;IA2GrC;;;;OAIG;WACW,cAAc,CAAC,SAAS,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,MAAM;IAQlE;;;OAGG;IACH,OAAO,CAAC,MAAM,CAAC,aAAa;IAc5B;;;;OAIG;WACW,mBAAmB,CAAC,aAAa,EAAE,MAAM,GAAG,OAAO,MAAM,GAAG,SAAS;IAInF;;;;OAIG;WACW,QAAQ,CAAC,aAAa,EAAE,MAAM,EAAE,MAAM,EAAE,QAAQ,GAAG,OAAO,MAAM;IAK9E;;;;;OAKG;WACW,eAAe,CAAC,aAAa,EAAE,MAAM;IAEnD;;;;OAIG;WACW,qBAAqB,CAAC,MAAM,EAAE,OAAO,MAAM;CAM1D;AAED;;;;;;;;;;;;;;;;;;GAkBG;AACH,qBAAa,gBAAgB;IAE3B,OAAO,CAAC,SAAS,CAAqC;IACtD,OAAO,CAAC,cAAc,CAAiC;IAEvD,wCAAwC;IAEjC,IAAI,CAAC,aAAa,EAAE,MAAM,GAAG,cAAc,GAAG,SAAS;IAKvD,aAAa,CAAC,OAAO,EAAE,UAAU,GAAG,cAAc,GAAG,SAAS;IAKrE,gCAAgC;IAEzB,GAAG,CAAC,aAAa,EAAE,MAAM,EAAE,QAAQ,EAAE,cAAc,GAAG,IAAI;CAKlE"}