we basically need the entire sandbox at this point

Author: siddharthkp

packages/app/src/embed/components/App/index.js

@@ -229,12 +229,7 @@ export default class App extends React.PureComponent<
       sandbox.modules,
       sandbox.directories,
       currentModulePath,
-      findMainModule(
-        sandbox.modules,
-        sandbox.directories,
-        sandbox.entry,
-        sandbox.template
-      )
+      findMainModule(sandbox)
     );
   };
 

packages/app/src/embed/components/Content/index.tsx

@@ -387,12 +387,7 @@ export default class Content extends React.PureComponent<Props, State> {
     } = this.props;
 
     const mainModule = isInProjectView
-      ? findMainModule(
-          sandbox.modules,
-          sandbox.directories,
-          sandbox.entry,
-          sandbox.template
-        )
+      ? findMainModule(sandbox)
       : currentModule;
 
     if (!mainModule) throw new Error('Cannot find main module');

packages/common/src/sandbox/modules.ts

@@ -1,6 +1,6 @@
 import memoize from 'lodash/memoize';
-import { Module, Directory } from '../types';
-import getTemplateDefinition, { TemplateType } from '../templates';
+import { Module, Directory, Sandbox } from '../types';
+import getTemplateDefinition from '../templates';
 
 const compareTitle = (
   original: string,
@@ -249,22 +249,17 @@ export const isMainModule = (
   return path.replace('/', '') === entry;
 };
 
-export const findMainModule = (
-  modules: Module[],
-  directories: Directory[],
-  entry: string = 'index.js',
-  template?: TemplateType
-) => {
-  const resolve = resolveModuleWrapped({ modules, directories });
+export const findMainModule = (sandbox?: Sandbox) => {
+  const resolve = resolveModuleWrapped(sandbox);
 
   // first attempt: try loading the entry file if it exists
 
-  const entryModule = resolve(entry);
+  const entryModule = resolve(sandbox.entry);
   if (entryModule) return entryModule;
 
   // second attempt: try loading the first file that exists from
   // the list of possible defaults in the template defination
-  const templateDefinition = getTemplateDefinition(template);
+  const templateDefinition = getTemplateDefinition(sandbox.template);
 
   const defaultOpenedFiles = templateDefinition.getDefaultOpenedFiles({});
 
@@ -275,7 +270,7 @@ export const findMainModule = (
   if (defaultOpenModule) return defaultOpenModule;
 
   // third attempt: give up and load the first file in the list
-  return modules[0];
+  return sandbox.modules[0];
 };
 
 export const findCurrentModule = (