Avoid saving redundant default data (Fixes #44)

Author: celonymire

src/utils/BaseViewProvider.ts

@@ -58,7 +58,7 @@ export default abstract class<ProviderMessageType, WebviewMessageType>
 		return data;
 	}
 
-	writeStorage(file: string, data: object) {
+	writeStorage(file: string, data?: object) {
 		const fileData = this._context.workspaceState.get(this.view, {});
 		this._context.workspaceState.update(this.view, {
 			...fileData,

src/views/judge/provider/JudgeViewProvider.ts

@@ -267,6 +267,11 @@ export default class extends BaseViewProvider<ProviderMessage, WebviewMessage> {
 		if (!file) {
 			return;
 		}
+		if (this._state.size === 0 && this._timeLimit === 0) {
+			// everything is defaulted, might as well not save it
+			super.writeStorage(file, undefined);
+			return;
+		}
 
 		const testcases: ITestcase[] = [];
 		for (const testcase of this._state.values()) {

src/views/stress/message.ts

@@ -35,6 +35,7 @@ export enum ProviderMessageType {
 	STOP = 2,
 	VIEW = 3,
 	ADD = 4,
+	RESET = 5,
 }
 export interface ILoadedMessage {
 	type: ProviderMessageType.LOADED;
@@ -53,9 +54,13 @@ export interface IAddMessage {
 	type: ProviderMessageType.ADD;
 	id: number;
 }
+export interface IResetMessage {
+	type: ProviderMessageType.RESET;
+}
 export type ProviderMessage =
 	| ILoadedMessage
 	| IRunMessage
 	| IStopMessage
 	| IViewMessage
-	| IAddMessage;
+	| IAddMessage
+	| IResetMessage;

src/views/stress/provider/StressViewProvider.ts

@@ -57,6 +57,9 @@ export default class extends BaseViewProvider<ProviderMessage, WebviewMessage> {
 			case ProviderMessageType.ADD:
 				this._add(msg);
 				break;
+			case ProviderMessageType.RESET:
+				this._reset();
+				break;
 		}
 	}
 
@@ -385,20 +388,41 @@ export default class extends BaseViewProvider<ProviderMessage, WebviewMessage> {
 		});
 	}
 
+	private _reset() {
+		const file = vscode.window.activeTextEditor?.document.fileName;
+		if (!file) {
+			return;
+		}
+
+		for (let i = 0; i < 3; i++) {
+			this._state[i].data.reset();
+			this._state[i].status = Status.NA;
+		}
+		super._postMessage({ type: WebviewMessageType.CLEAR });
+		this._saveState();
+	}
+
 	private _saveState() {
 		const file = vscode.window.activeTextEditor?.document.fileName;
 		if (!file) {
 			return;
 		}
 
+		let isDefault = true;
+		for (const state of this._state) {
+			isDefault &&= state.data.data === '';
+			isDefault &&= state.status === Status.NA;
+		}
 		super.writeStorage(
 			file,
-			this._state.map<IData>((value) => {
-				return {
-					data: value.data.data,
-					status: value.status,
-				};
-			}),
+			isDefault
+				? undefined
+				: this._state.map<IData>((value) => {
+						return {
+							data: value.data.data,
+							status: value.status,
+						};
+					}),
 		);
 	}
 

src/views/stress/webview/App.tsx

@@ -31,6 +31,7 @@ const expand = (id: number) =>
 	postProviderMessage({ type: ProviderMessageType.VIEW, id });
 const add = (id: number) =>
 	postProviderMessage({ type: ProviderMessageType.ADD, id });
+const reset = () => postProviderMessage({ type: ProviderMessageType.RESET });
 
 window.addEventListener('message', (event: MessageEvent<WebviewMessage>) => {
 	switch (event.data.type) {
@@ -61,6 +62,7 @@ function handleClear() {
 	batch(() => {
 		for (let i = 0; i < 3; i++) {
 			state[i].data = '';
+			state[i].status = Status.NA;
 		}
 	});
 }
@@ -116,6 +118,14 @@ export default function App() {
 							<div class="w-6 shrink-0" />
 							<div class="flex justify-start gap-x-2 bg-zinc-800 grow">
 								{button}
+								<button
+									type="button"
+									class="text-base leading-tight px-3 w-fit display-font"
+									style={{ backgroundColor: BLUE_COLOR }}
+									onClick={reset}
+								>
+									reset
+								</button>
 							</div>
 						</div>
 					</div>