Include modifiedMethodSignatures in SetModeledMethodsMessage

Author: robertbrignull

extensions/ql-vscode/src/common/interface-types.ts

@@ -531,6 +531,7 @@ interface SetMethodsMessage {
 interface SetModeledMethodsMessage {
   t: "setModeledMethods";
   methods: Record<string, ModeledMethod[]>;
+  modifiedMethodSignatures: string[];
 }
 
 interface SetModifiedMethodsMessage {

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

@@ -823,14 +823,8 @@ export class ModelEditorView extends AbstractWebview<
           await this.postMessage({
             t: "setModeledMethods",
             methods: event.modeledMethods,
+            modifiedMethodSignatures: [...event.modifiedMethodSignatures],
           });
-
-          if (event.modifiedMethodSignatures !== undefined) {
-            await this.postMessage({
-              t: "setModifiedMethods",
-              methodSignatures: [...event.modifiedMethodSignatures],
-            });
-          }
         }
       }),
     );

extensions/ql-vscode/src/view/model-editor/ModelEditor.tsx

@@ -196,6 +196,7 @@ export function ModelEditor({
             break;
           case "setModeledMethods":
             setModeledMethods(msg.methods);
+            setModifiedSignatures(new Set(msg.modifiedMethodSignatures));
             break;
           case "setModifiedMethods":
             setModifiedSignatures(new Set(msg.methodSignatures));