smooth fork live fixes and updating moving of files and dirs

Author: christianalfoni

packages/app/src/app/overmind/effects/vscode/sandboxFsSync/utils.ts

@@ -1,4 +1,4 @@
-import { Module, SandboxFs, Directory } from '@codesandbox/common/lib/types';
+import { Directory, Module, SandboxFs } from '@codesandbox/common/lib/types';
 import { json } from 'overmind';
 
 export const writeFile = (fs: SandboxFs, module: Module) => {

packages/app/src/app/overmind/namespaces/editor/actions.ts

@@ -182,6 +182,8 @@ export const onOperationApplied: Action<{
     module,
     code,
   });
+
+  actions.editor.internal.updatePreviewCode();
 };
 
 export const codeChanged: Action<{

packages/app/src/app/overmind/namespaces/live/internalActions.ts

@@ -72,6 +72,16 @@ export const initialize: AsyncAction<string, Sandbox> = async (
     }
     state.editor.currentId = sandbox.id;
 
+    await effects.vscode.closeAllTabs();
+
+    state.editor.modulesByPath = effects.vscode.fs.create(
+      state.editor.currentSandbox
+    );
+
+    await effects.vscode.changeSandbox(sandbox);
+    effects.vscode.openModule(state.editor.currentModule);
+    effects.preview.executeCodeImmediately(true);
+
     effects.analytics.track('Live Session Joined', {});
     effects.live.listen(actions.live.liveMessageReceived);
 

packages/app/src/app/overmind/namespaces/live/liveMessageOperators.ts

@@ -184,16 +184,24 @@ export const onModuleUpdated: Operator<
   const moduleIndex = sandbox.modules.findIndex(
     moduleEntry => moduleEntry.shortid === data.moduleShortid
   );
+  const existingModule =
+    state.editor.sandboxes[sandbox.id].modules[moduleIndex];
+
+  if (existingModule.path !== data.module.path) {
+    effects.vscode.fs.rename(
+      state.editor.modulesByPath,
+      existingModule.path,
+      data.module.path
+    );
+  }
 
-  Object.assign(
-    state.editor.sandboxes[sandbox.id].modules[moduleIndex],
-    data.module
-  );
+  Object.assign(existingModule, data.module);
 
-  effects.vscode.fs.writeFile(
-    state.editor.modulesByPath,
-    state.editor.sandboxes[sandbox.id].modules[moduleIndex]
-  );
+  effects.vscode.fs.writeFile(state.editor.modulesByPath, existingModule);
+
+  if (state.editor.currentModuleShortid === data.moduleShortid) {
+    effects.vscode.openModule(existingModule);
+  }
 
   actions.editor.internal.updatePreviewCode();
 });
@@ -243,19 +251,30 @@ export const onDirectoryUpdated: Operator<
     directoryShortid: string;
     module: Directory; // Still very weird
   }>
-> = mutate(({ state }, { _isOwnMessage, data }) => {
+> = mutate(({ state, actions, effects }, { _isOwnMessage, data }) => {
   if (_isOwnMessage) {
     return;
   }
   const sandbox = state.editor.currentSandbox;
   const directoryIndex = sandbox.directories.findIndex(
     directoryEntry => directoryEntry.shortid === data.directoryShortid
   );
+  const existingDirectory =
+    state.editor.sandboxes[sandbox.id].directories[directoryIndex];
+  const hasChangedPath = existingDirectory.path !== data.module.path;
 
-  Object.assign(
-    state.editor.sandboxes[sandbox.id].directories[directoryIndex],
-    data.module
-  );
+  Object.assign(existingDirectory, data.module);
+
+  if (hasChangedPath) {
+    const prevCurrentModulePath = state.editor.currentModule.path;
+
+    state.editor.modulesByPath = effects.vscode.fs.create(sandbox);
+    actions.editor.internal.updatePreviewCode();
+
+    if (prevCurrentModulePath !== state.editor.currentModule.path) {
+      actions.editor.internal.setCurrentModule(state.editor.currentModule);
+    }
+  }
 });
 
 export const onDirectoryDeleted: Operator<