Send message to model editor view on revealing method

This will call a method on the correct model editor view when the user
clicks on "Review in editor". This does not yet do anything to the view;
this will be added in a follow-up commit.

Author: koesie10

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

@@ -4,14 +4,23 @@ import { DisposableObject } from "../../common/disposable-object";
 import { MethodModelingViewProvider } from "./method-modeling-view-provider";
 import { Method } from "../method";
 import { ModelingStore } from "../modeling-store";
+import { ModelEditorViewTracker } from "../model-editor-view-tracker";
 
 export class MethodModelingPanel extends DisposableObject {
   private readonly provider: MethodModelingViewProvider;
 
-  constructor(app: App, modelingStore: ModelingStore) {
+  constructor(
+    app: App,
+    modelingStore: ModelingStore,
+    editorViewTracker: ModelEditorViewTracker,
+  ) {
     super();
 
-    this.provider = new MethodModelingViewProvider(app, modelingStore);
+    this.provider = new MethodModelingViewProvider(
+      app,
+      modelingStore,
+      editorViewTracker,
+    );
     this.push(
       window.registerWebviewViewProvider(
         MethodModelingViewProvider.viewType,

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

@@ -8,9 +8,10 @@ import { extLogger } from "../../common/logging/vscode/loggers";
 import { App } from "../../common/app";
 import { redactableError } from "../../common/errors";
 import { Method } from "../method";
-import { ModelingStore } from "../modeling-store";
+import { DbModelingState, ModelingStore } from "../modeling-store";
 import { AbstractWebviewViewProvider } from "../../common/vscode/abstract-webview-view-provider";
 import { assertNever } from "../../common/helpers-pure";
+import { ModelEditorViewTracker } from "../model-editor-view-tracker";
 
 export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
   ToMethodModelingMessage,
@@ -23,6 +24,7 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
   constructor(
     app: App,
     private readonly modelingStore: ModelingStore,
+    private readonly editorViewTracker: ModelEditorViewTracker,
   ) {
     super(app, "method-modeling");
   }
@@ -78,23 +80,45 @@ export class MethodModelingViewProvider extends AbstractWebviewViewProvider<
         break;
 
       case "setModeledMethod": {
-        const activeState = this.modelingStore.getStateForActiveDb();
-        if (!activeState) {
-          throw new Error("No active state found in modeling store");
-        }
+        const activeState = this.ensureActiveState();
+
         this.modelingStore.updateModeledMethod(
           activeState.databaseItem,
           msg.method,
         );
         break;
       }
       case "revealInModelEditor":
+        await this.revealInModelEditor(msg.method);
+
         break;
       default:
         assertNever(msg);
     }
   }
 
+  private async revealInModelEditor(method: Method): Promise<void> {
+    const activeState = this.ensureActiveState();
+
+    const views = this.editorViewTracker.getViews(
+      activeState.databaseItem.databaseUri.toString(),
+    );
+    if (views.length === 0) {
+      return;
+    }
+
+    await Promise.all(views.map((view) => view.revealMethod(method)));
+  }
+
+  private ensureActiveState(): DbModelingState {
+    const activeState = this.modelingStore.getStateForActiveDb();
+    if (!activeState) {
+      throw new Error("No active state found in modeling store");
+    }
+
+    return activeState;
+  }
+
   private registerToModelingStoreEvents(): void {
     this.push(
       this.modelingStore.onModeledMethodsChanged(async (e) => {

extensions/ql-vscode/src/model-editor/model-editor-module.ts

@@ -46,7 +46,7 @@ export class ModelEditorModule extends DisposableObject {
       new MethodsUsagePanel(this.modelingStore, cliServer),
     );
     this.methodModelingPanel = this.push(
-      new MethodModelingPanel(app, this.modelingStore),
+      new MethodModelingPanel(app, this.modelingStore, this.editorViewTracker),
     );
 
     this.registerToModelingStoreEvents();

extensions/ql-vscode/src/model-editor/model-editor-view-tracker.ts

@@ -1,5 +1,9 @@
+import { Method } from "./method";
+
 interface ModelEditorViewInterface {
   databaseUri: string;
+
+  revealMethod(method: Method): Promise<void>;
 }
 
 export class ModelEditorViewTracker<

extensions/ql-vscode/src/model-editor/model-editor-view.ts

@@ -346,6 +346,10 @@ export class ModelEditorView extends AbstractWebview<
     return this.databaseItem.databaseUri.toString();
   }
 
+  public async revealMethod(method: Method): Promise<void> {
+    void this.app.logger.log(`Revealing method ${JSON.stringify(method)}`);
+  }
+
   private async setViewState(): Promise<void> {
     const showLlmButton =
       this.databaseItem.language === "java" && showLlmGeneration();

extensions/ql-vscode/src/model-editor/modeling-store.ts

@@ -6,7 +6,7 @@ import { Method, Usage } from "./method";
 import { ModeledMethod } from "./modeled-method";
 import { INITIAL_HIDE_MODELED_METHODS_VALUE } from "./shared/hide-modeled-methods";
 
-interface DbModelingState {
+export interface DbModelingState {
   databaseItem: DatabaseItem;
   methods: Method[];
   hideModeledMethods: boolean;