Fix correction switching between modules

Author: Ives van Hoorne

packages/app/src/app/components/CodeEditor/Monaco/index.js

@@ -236,7 +236,11 @@ class MonacoEditor extends React.Component<Props, State> implements Editor {
     }
   };
 
-  changeModule = (newModule: Module, errors?: Array<ModuleError>) => {
+  changeModule = (
+    newModule: Module,
+    errors?: Array<ModuleError>,
+    corrections?: Array<ModuleCorrection>
+  ) => {
     const oldModule = this.currentModule;
 
     this.currentModule = newModule;
@@ -249,7 +253,13 @@ class MonacoEditor extends React.Component<Props, State> implements Editor {
         this.changeCode(newModule.code || '');
       }
 
-      if (errors) this.setErrors(errors);
+      if (errors) {
+        this.setErrors(errors);
+      }
+
+      if (corrections) {
+        this.setCorrections(corrections);
+      }
     });
   };
 

packages/app/src/app/components/CodeEditor/types.js

@@ -30,7 +30,11 @@ export interface Editor {
   updateModules?: () => any;
   changeSettings?: (settings: Settings) => any;
   changeDependencies?: (deps: Object) => any;
-  changeModule?: (module: Module, errors?: Array<ModuleError>) => any;
+  changeModule?: (
+    module: Module,
+    errors?: Array<ModuleError>,
+    corrections?: Array<ModuleCorrection>
+  ) => any;
   changeCode?: (code: string) => any;
 
   currentModule?: Module;

packages/app/src/app/pages/Sandbox/Editor/Content/index.js

@@ -191,9 +191,11 @@ class EditorPreview extends React.Component<Props, State> {
         }
         const editorModule = editor.currentModule;
 
-        if (newModule !== editorModule && editor.changeModule) {
+        const changeModule = editor.changeModule;
+        if (newModule !== editorModule && changeModule) {
           const errors = store.editor.errors.map(e => e);
-          editor.changeModule(newModule, errors);
+          const corrections = store.editor.corrections.map(e => e);
+          changeModule(newModule, errors, corrections);
         } else if (editor.changeCode) {
           editor.changeCode(newModule.code || '');
         }