feat: update code on reset

Author: heyzec

src/utils/editor.ts

@@ -17,6 +17,7 @@ export class Editor {
   private onChangeCallback?: (editor: Editor) => void;
 
   // Data associated with TextEditor
+  prepend: string;
   uri: string;
 
   // Metadata relating to this question
@@ -26,12 +27,14 @@ export class Editor {
 
   private constructor(
     editor: vscode.TextEditor,
+    prepend: string,
     uri: string,
     workspaceLocation: VscWorkspaceLocation,
     assessmentName: string,
     questionId: number,
   ) {
     this.editor = editor;
+    this.prepend = prepend;
     this.uri = uri;
     this.workspaceLocation = workspaceLocation;
     this.assessmentName = assessmentName;
@@ -123,6 +126,7 @@ export class Editor {
     // Create wrapper
     const self = new Editor(
       editor,
+      prepend,
       uri.toString(),
       workspaceLocation,
       assessmentName,
@@ -146,16 +150,18 @@ export class Editor {
     return self;
   }
 
-  async replace(code: string, tag: string = "") {
+  async replace(code: string) {
     const editor = this.editor;
-    // Don't replace if the code is the same
+    const contents = codeAddPrepend(code, this.prepend);
+
+    // In some sense, simulate a select all and paste
     editor.edit((editBuilder) => {
       editBuilder.replace(
         new vscode.Range(
           editor.document.positionAt(0),
           editor.document.positionAt(editor.document.getText().length),
         ),
-        code,
+        contents,
       );
     });
   }

src/utils/messageHandler.tsx

@@ -167,6 +167,12 @@ export class MessageHandler {
           context.globalState.update("courseId", courseId);
           treeDataProvider.refresh();
           break;
+        case MessageTypeNames.ResetEditor:
+          if (this.activeEditor) {
+            this.activeEditor.replace(message.initialCode);
+            this.panel?.reveal(vscode.ViewColumn.Two);
+          }
+          break;
       }
       console.log(`${Date.now()} Finish handleMessage: ${message.type}`);
     }

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,