Add "Save All" keybind (Fixes #39)

Author: sam20908

package.json

@@ -201,6 +201,11 @@
 				"title": "Delete All Testcases",
 				"category": "Fast Olympic Coding"
 			},
+			{
+				"command": "fastolympiccoding.saveAll",
+				"title": "Save All Testcases",
+				"category": "Fast Olympic Coding"
+			},
 			{
 				"command": "fastolympiccoding.stressTest",
 				"title": "Run Stress Test",
@@ -238,6 +243,11 @@
 				"key": "ctrl+alt+d",
 				"when": "view.fastolympiccoding.judge.visible"
 			},
+			{
+				"command": "fastolympiccoding.saveAll",
+				"key": "ctrl+alt+s",
+				"when": "view.fastolympiccoding.judge.visible"
+			},
 			{
 				"command": "fastolympiccoding.stressTest",
 				"key": "ctrl+alt+g",
@@ -285,4 +295,4 @@
 		"nu-observables": "^0.0.7",
 		"preact": "^10.26.5"
 	}
-}
+}

src/extension.ts

@@ -102,6 +102,13 @@ function registerCommands(context: vscode.ExtensionContext): void {
 		),
 	);
 
+		context.subscriptions.push(
+		vscode.commands.registerTextEditorCommand(
+			'fastolympiccoding.saveAll',
+			() => judgeViewProvider.saveAll(),
+		),
+	);
+
 	context.subscriptions.push(
 		vscode.commands.registerTextEditorCommand(
 			'fastolympiccoding.stressTest',

src/views/judge/message.ts

@@ -66,8 +66,9 @@ export enum WebviewMessageType {
 	SET = 1,
 	STDIO = 2,
 	DELETE = 3,
-	SHOW = 4,
-	INITIAL_STATE = 5,
+	SAVE_ALL = 4,
+	SHOW = 5,
+	INITIAL_STATE = 6,
 }
 export interface INewMessage {
 	type: WebviewMessageType.NEW;
@@ -89,6 +90,9 @@ export interface IDeleteMessage {
 	type: WebviewMessageType.DELETE;
 	id: number;
 }
+export interface ISaveAllMessage {
+	type: WebviewMessageType.SAVE_ALL;
+}
 export interface IShowMessage {
 	type: WebviewMessageType.SHOW;
 	visible: boolean;
@@ -102,5 +106,6 @@ export type WebviewMessage =
 	| ISetMessage
 	| IStdioMessage
 	| IDeleteMessage
+	| ISaveAllMessage
 	| IShowMessage
 	| IInitialState;

src/views/judge/provider/JudgeViewProvider.ts

@@ -222,6 +222,10 @@ export default class extends BaseViewProvider<ProviderMessage, WebviewMessage> {
 		}
 	}
 
+	saveAll() {
+		super._postMessage({ type: WebviewMessageType.SAVE_ALL });
+	}
+
 	private _nextTestcase() {
 		void this._run(this._addTestcase(), true);
 	}

src/views/judge/webview/App.tsx

@@ -36,6 +36,9 @@ window.addEventListener('message', (msg: MessageEvent<WebviewMessage>) => {
 		case WebviewMessageType.DELETE:
 			handleDelete(msg.data);
 			break;
+		case WebviewMessageType.SAVE_ALL:
+			handleSaveAll();
+			break;
 		case WebviewMessageType.SHOW:
 			handleShow(msg.data);
 			break;
@@ -91,6 +94,24 @@ function handleDelete({ id }: IDeleteMessage) {
 	testcases.delete(id);
 }
 
+function handleSaveAll() {
+	for (const [id, testcase] of testcases) {
+		if (testcase.status === Status.EDITING) {
+			const stdin = testcase.stdin;
+			const acceptedStdout = testcase.acceptedStdout;
+			// the extension host will send shortened version of both of these
+			testcase.stdin = '';
+			testcase.acceptedStdout = '';
+			postProviderMessage({
+				type: ProviderMessageType.SAVE,
+				id,
+				stdin,
+				acceptedStdout,
+			});
+		}
+	}
+}
+
 function handleShow({ visible }: IShowMessage) {
 	show.value = visible;
 }