Make "Save" button respect selected rows

Author: shati-patel

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

@@ -244,7 +244,7 @@ export const LibraryRow = ({
           <SectionDivider />
           <ButtonsContainer>
             <VSCodeButton onClick={handleSave} disabled={!hasUnsavedChanges}>
-              Save
+              {selectedSignatures.size === 0 ? "Save" : "Save selected"}
             </VSCodeButton>
           </ButtonsContainer>
         </>

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

@@ -216,15 +216,27 @@ export function ModelEditor({
   const onSaveAllClick = useCallback(() => {
     vscode.postMessage({
       t: "saveModeledMethods",
+      methodSignatures:
+        selectedSignatures.size === 0
+          ? undefined
+          : Array.from(selectedSignatures),
     });
-  }, []);
+  }, [selectedSignatures]);
 
-  const onSaveModelClick = useCallback((methodSignatures: string[]) => {
-    vscode.postMessage({
-      t: "saveModeledMethods",
-      methodSignatures,
-    });
-  }, []);
+  const onSaveModelClick = useCallback(
+    (methodSignatures: string[]) => {
+      vscode.postMessage({
+        t: "saveModeledMethods",
+        methodSignatures:
+          selectedSignatures.size === 0
+            ? methodSignatures
+            : methodSignatures.filter((signature) =>
+                selectedSignatures.has(signature),
+              ),
+      });
+    },
+    [selectedSignatures],
+  );
 
   const onGenerateFromSourceClick = useCallback(() => {
     vscode.postMessage({
@@ -327,7 +339,7 @@ export function ModelEditor({
                 onClick={onSaveAllClick}
                 disabled={modifiedSignatures.size === 0}
               >
-                Save all
+                {selectedSignatures.size === 0 ? "Save all" : "Save selected"}
               </VSCodeButton>
               <VSCodeButton appearance="secondary" onClick={onRefreshClick}>
                 Refresh