Fix: reset button functionality

Author: shirsho-12

src/utils/editor.ts

@@ -199,4 +199,19 @@ export class Editor {
   ) {
     this.onChangeCallback = callback;
   }
+
+  reset(prepend: string, initialCode: string) {
+    this.log(`EXTENSION: Editor's reset called`);
+    this.replace(
+      prepend !== ""
+        ? [
+            "// PREPEND -- DO NOT EDIT",
+            prepend,
+            "// END PREPEND",
+            initialCode,
+          ].join("\n")
+        : initialCode,
+      "reset",
+    );
+  }
 }

src/utils/messageHandler.tsx

@@ -110,7 +110,12 @@ export class MessageHandler {
           sendToFrontend(this.panel, msg);
           break;
         }
-
+        case MessageTypeNames.ResetEditor:
+          if (this.activeEditor) {
+            this.activeEditor.reset("", message.initialCode);
+            this.panel?.reveal(vscode.ViewColumn.Two);
+          }
+          break;
         case MessageTypeNames.NewEditor:
           this.activeEditor = await Editor.create(
             message.workspaceLocation,

src/utils/messages.ts

@@ -48,6 +48,13 @@ const Messages = createMessages({
   EvalEditor: (workspaceLocation: VscWorkspaceLocation) => ({
     workspaceLocation: workspaceLocation,
   }),
+  ResetEditor: (
+    workspaceLocation: VscWorkspaceLocation,
+    initialCode: string,
+  ) => ({
+    workspaceLocation,
+    initialCode,
+  }),
   NotifyAssessmentsOverview: (
     assessmentOverviews: VscAssessmentOverview[],
     courseId: number,