@itwin/core-backend 5.7.0-dev.13 → 5.7.0-dev.14

package/lib/cjs/IModelDb.js

@@ -742,7 +742,7 @@ class IModelDb extends core_common_1.IModel {
     saveChanges(descriptionOrArgs) {
         if (this.openMode === core_bentley_1.OpenMode.Readonly)
             throw new core_common_1.IModelError(core_bentley_1.IModelStatus.ReadOnly, "IModelDb was opened read-only");
-        if (this instanceof BriefcaseDb) {
+        if (this.isBriefcaseDb()) {
             if (this.txns.isIndirectChanges) {
                 throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot save changes while in an indirect change scope");
             }
@@ -795,6 +795,21 @@ class IModelDb extends core_common_1.IModel {
     restartTxnSession() {
         return this[Symbols_1._nativeDb].restartTxnSession();
     }
+    /**
+     * Get the class full name from a class Id.
+     * @param classId the Id of the class to look up
+     * @returns the full name of the class (e.g. "BisCore:Element")
+     * @throws IModelError if the classId is invalid or the class is not found.
+     * @internal
+     */
+    getClassNameFromId(classId) {
+        if (!core_bentley_1.Id64.isValid(classId))
+            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, `Class Id ${classId} is invalid`);
+        const name = this[Symbols_1._nativeDb].classIdToName(classId);
+        if (name === undefined)
+            throw new core_common_1.IModelError(core_bentley_1.IModelStatus.NotFound, `Class not found: ${classId}`);
+        return name;
+    }
     /** Removes unused schemas from the database.
      *
      * If the removal was successful, the database is automatically saved to disk.
@@ -856,6 +871,9 @@ class IModelDb extends core_common_1.IModel {
                     }
                     callbackResources.cachedData = callbackResult.cachedData;
                 }
+                if (this.isBriefcaseDb() && IModelHost_1.IModelHost.useSemanticRebase) {
+                    this.saveChanges("Save changes from schema import pre callback");
+                }
             }
         }
         catch (callbackError) {
@@ -875,6 +893,9 @@ class IModelDb extends core_common_1.IModel {
         try {
             if (callback?.postSchemaImportCallback)
                 await callback.postSchemaImportCallback(context);
+            if (this.isBriefcaseDb() && IModelHost_1.IModelHost.useSemanticRebase) {
+                this.saveChanges("Save changes from schema import post callback");
+            }
         }
         catch (callbackError) {
             this.abandonChanges();
@@ -887,17 +908,31 @@ class IModelDb extends core_common_1.IModel {
     }
     /** Shared implementation for importing schemas from file or string. */
     async importSchemasInternal(schemas, options, nativeImportOp) {
-        if (this instanceof BriefcaseDb) {
+        // BriefcaseDb-specific validation checks
+        if (this.isBriefcaseDb()) {
             if (this.txns.rebaser.isRebasing) {
                 throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot import schemas while rebasing");
             }
             if (this.txns.isIndirectChanges) {
                 throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot import schemas while in an indirect change scope");
             }
+            // Additional checks when semantic rebase is enabled
+            if (IModelHost_1.IModelHost.useSemanticRebase) {
+                if (this[Symbols_1._nativeDb].hasUnsavedChanges()) {
+                    throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot import schemas with unsaved changes when useSemanticRebase flag is on");
+                }
+                if (this[Symbols_1._nativeDb].schemaSyncEnabled()) {
+                    throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot import schemas when schema sync is enabled and also useSemanticRebase flag is on");
+                }
+            }
         }
         if (options?.channelUpgrade) {
             try {
                 await this.channels.upgradeChannel(options.channelUpgrade, this, options.data);
+                // If semantic rebase is enabled and channel upgrade made changes, save them
+                if (this.isBriefcaseDb() && IModelHost_1.IModelHost.useSemanticRebase) {
+                    this.saveChanges();
+                }
             }
             catch (error) {
                 this.abandonChanges();
@@ -938,8 +973,16 @@ class IModelDb extends core_common_1.IModel {
                 schemaLockHeld: true,
                 ecSchemaXmlContext: maybeCustomNativeContext,
             };
-            if (this[Symbols_1._nativeDb].getITwinId() !== core_bentley_1.Guid.empty) // if this iModel is associated with an iTwin, importing schema requires the schema lock
-                await this.acquireSchemaLock();
+            // This check is different from isBriefcase in case of StandaloneDb, which is a briefcase but has no iTwinId.
+            if (this[Symbols_1._nativeDb].getITwinId() !== core_bentley_1.Guid.empty) { // if this iModel is associated with an iTwin, importing schema requires the schema lock
+                if (IModelHost_1.IModelHost.useSemanticRebase) {
+                    // Use shared lock for semantic rebase to allow concurrent schema imports
+                    await this.locks.acquireLocks({ shared: core_common_1.IModel.repositoryModelId });
+                }
+                else {
+                    await this.acquireSchemaLock();
+                }
+            }
             try {
                 nativeImportOp(schemas, nativeImportOptions);
             }
@@ -948,6 +991,17 @@ class IModelDb extends core_common_1.IModel {
             }
         }
         this.clearCaches();
+        // Store schemas for semantic rebase (BriefcaseDb only)
+        if (this.isBriefcaseDb() && IModelHost_1.IModelHost.useSemanticRebase) {
+            const lastSavedTxnProps = this.txns.getLastSavedTxnProps();
+            if (lastSavedTxnProps === undefined) {
+                throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "After schema import, no last saved transaction found");
+            }
+            if (lastSavedTxnProps.type !== "Schema" && lastSavedTxnProps.type !== "ECSchema") {
+                throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "After schema import, last saved transaction is not a Schema transaction");
+            }
+            BriefcaseManager_1.BriefcaseManager.storeSchemasForSemanticRebase(this, lastSavedTxnProps.id, schemas);
+        }
         if (options?.schemaImportCallbacks?.postSchemaImportCallback)
             await this.postSchemaImportCallback(options.schemaImportCallbacks, { iModel: this, resources: preSchemaImportCallbackResult, data: options.data });
     }
@@ -1433,7 +1487,7 @@ class IModelDb extends core_common_1.IModel {
      * @note This method should not be called from {TxnManager.withIndirectTxnModeAsync} or {TxnManager.withIndirectTxnMode}.
      */
     saveFileProperty(prop, strValue, blobVal) {
-        if (this instanceof BriefcaseDb) {
+        if (this.isBriefcaseDb()) {
             if (this.txns.isIndirectChanges) {
                 throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot save file property while in an indirect change scope");
             }
@@ -1445,7 +1499,7 @@ class IModelDb extends core_common_1.IModel {
      * @note This method should not be called from {TxnManager.withIndirectTxnModeAsync} or {TxnManager.withIndirectTxnMode}.
      */
     deleteFileProperty(prop) {
-        if (this instanceof BriefcaseDb) {
+        if (this.isBriefcaseDb()) {
             if (this.txns.isIndirectChanges) {
                 throw new core_common_1.IModelError(core_bentley_1.IModelStatus.BadRequest, "Cannot delete file property while in an indirect change scope");
             }
@@ -2950,6 +3004,7 @@ class BriefcaseDb extends IModelDb {
         this.clearCaches();
         this[Symbols_1._nativeDb].discardLocalChanges();
         this[Symbols_1._resetIModelDb]();
+        BriefcaseManager_1.BriefcaseManager.deleteRebaseFolders(this);
         if (args?.retainLocks) {
             return;
         }
@@ -3122,6 +3177,15 @@ class BriefcaseDb extends IModelDb {
         this.onOpened.raiseEvent(briefcaseDb, args);
         return briefcaseDb;
     }
+    /**
+     * Iterate through the existing transactions to see if any of them are schema transactions.
+     * @returns true if any schema transactions exist.
+     * @beta
+     */
+    checkIfSchemaTxnExists() {
+        const txnProps = Array.from(this.txns.queryTxns());
+        return txnProps.some((props) => props.type === "Schema" || props.type === "ECSchema");
+    }
     /**  This is called by native code when applying a changeset */
     onChangesetConflict(args) {
         // returning undefined will result in native handler to resolve conflict
@@ -3406,6 +3470,7 @@ class BriefcaseDb extends IModelDb {
             this.initializeIModelDb("pullMerge");
         });
         this.txns._onChangesPushed(this.changeset);
+        BriefcaseManager_1.BriefcaseManager.deleteRebaseFolders(this);
     }
     close(options) {
         if (this.isBriefcase && this.isOpen && !this.isReadonly && this.txns.rebaser.inProgress()) {