Rewrite setInitialState to always set all fields

extensions/ql-vscode/src/model-editor/method-modeling/method-modeling-view-provider.ts

@@ -38,7 +38,7 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
   }
 
   protected override async onWebViewLoaded(): Promise<void> {
-    await Promise.all([this.setViewState(), this.setInitialState()]);
+    await this.setInitialState();
     this.registerToModelingEvents();
     this.registerToModelConfigEvents();
   }
@@ -104,29 +104,25 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
   }
 
   private async setInitialState(): Promise<void> {
-    if (this.modelingStore.hasStateForActiveDb()) {
+    await this.setViewState();
-      const selectedMethod = this.modelingStore.getSelectedMethodDetails();
-      if (selectedMethod) {
-        this.databaseItem = selectedMethod.databaseItem;
-        this.language = tryGetQueryLanguage(
-          selectedMethod.databaseItem.language,
-        );
-        this.method = selectedMethod.method;
 
-        await this.postMessage({
+    const stateForActiveDb = this.modelingStore.getStateForActiveDb();
-          t: "setSelectedMethod",
+    if (!stateForActiveDb) {
-          method: selectedMethod.method,
+      return;
-          modeledMethods: selectedMethod.modeledMethods,
+    }
-          isModified: selectedMethod.isModified,
-          isInProgress: selectedMethod.isInProgress,
-          processedByAutoModel: selectedMethod.processedByAutoModel,
-        });
-      }
 
-      await this.postMessage({
+    await this.setDatabaseItem(stateForActiveDb.databaseItem);
-        t: "setInModelingMode",
+
-        inModelingMode: true,
+    const selectedMethod = this.modelingStore.getSelectedMethodDetails();
-      });
+    if (selectedMethod) {
+      await this.setSelectedMethod(
+        stateForActiveDb.databaseItem,
+        selectedMethod.method,
+        selectedMethod.modeledMethods,
+        selectedMethod.isModified,
+        selectedMethod.isInProgress,
+        selectedMethod.processedByAutoModel,
+      );
     }
   }