Typecheck main in PSCi (purescript#2163)

paf31 · paf31 · commit 2bafbae807e9 · 2016-05-28T13:23:25.000-07:00
diff --git a/src/Language/PureScript/Interactive/Module.hs b/src/Language/PureScript/Interactive/Module.hs
@@ -45,14 +45,27 @@ createTemporaryModule :: Bool -> PSCiState -> P.Expr -> P.Module
 createTemporaryModule exec PSCiState{psciImportedModules = imports, psciLetBindings = lets} val =
   let
     moduleName    = P.ModuleName [P.ProperName "$PSCI"]
+    effModuleName = P.moduleNameFromString "Control.Monad.Eff"
+    effImport     = (effModuleName, P.Implicit, Just (P.ModuleName [P.ProperName "$Eff"]))
     supportImport = (supportModuleName, P.Implicit, Just (P.ModuleName [P.ProperName "$Support"]))
     eval          = P.Var (P.Qualified (Just (P.ModuleName [P.ProperName "$Support"])) (P.Ident "eval"))
     mainValue     = P.App eval (P.Var (P.Qualified Nothing (P.Ident "it")))
     itDecl        = P.ValueDeclaration (P.Ident "it") P.Public [] $ Right val
+    typeDecl      = P.TypeDeclaration (P.Ident "$main")
+                      (P.TypeApp
+                        (P.TypeApp
+                          (P.TypeConstructor
+                            (P.Qualified (Just (P.ModuleName [P.ProperName "$Eff"])) (P.ProperName "Eff")))
+                              (P.TypeWildcard internalSpan))
+                                (P.TypeWildcard internalSpan))
     mainDecl      = P.ValueDeclaration (P.Ident "$main") P.Public [] $ Right mainValue
-    decls         = if exec then [itDecl, mainDecl] else [itDecl]
+    decls         = if exec then [itDecl, typeDecl, mainDecl] else [itDecl]
+    internalSpan  = P.internalModuleSourceSpan "<internal>"
   in
-    P.Module (P.internalModuleSourceSpan "<internal>") [] moduleName ((importDecl `map` (supportImport : imports)) ++ lets ++ decls) Nothing
+    P.Module internalSpan
+             [] moduleName
+             ((importDecl `map` (effImport : supportImport : imports)) ++ lets ++ decls)
+             Nothing
 
 
 -- |
