[project @ 2003-01-06 15:30:14 by simonpj]
[ghc-hetmet.git] / ghc / compiler / typecheck / TcRnDriver.lhs
index 04b0ca3..e0a07c2 100644 (file)
@@ -16,13 +16,14 @@ module TcRnDriver (
 #include "HsVersions.h"
 
 #ifdef GHCI
-import {-# SOURCE #-} TcSplice( tcSpliceDecls )
+import {-# SOURCE #-} TcSplice ( tcSpliceDecls )
+import               DsMeta   ( templateHaskellNames )
 #endif
 
 import CmdLineOpts     ( DynFlag(..), opt_PprStyle_Debug, dopt )
-import HsSyn           ( HsModule(..), HsBinds(..), MonoBinds(..), HsDecl(..), HsExpr(..),
+import HsSyn           ( HsModule(..), HsBinds(..), MonoBinds(..), HsExpr(..),
                          Stmt(..), Pat(VarPat), HsStmtContext(..), RuleDecl(..),
-                         HsGroup(..),
+                         HsGroup(..), SpliceDecl(..),
                          mkSimpleMatch, placeHolderType, toHsType, andMonoBinds,
                          isSrcRule, collectStmtsBinders
                        )
@@ -37,7 +38,7 @@ import MkId           ( unsafeCoerceId )
 import RdrName         ( RdrName, getRdrName, mkUnqual, mkRdrUnqual, 
                          lookupRdrEnv, elemRdrEnv )
 
-import RnHsSyn         ( RenamedHsDecl, RenamedStmt, RenamedTyClDecl, 
+import RnHsSyn         ( RenamedStmt, RenamedTyClDecl, 
                          ruleDeclFVs, instDeclFVs, tyClDeclFVs )
 import TcHsSyn         ( TypecheckedHsExpr, TypecheckedRuleDecl,
                          zonkTopBinds, zonkTopDecls, mkHsLet,
@@ -73,7 +74,7 @@ import TcTyClsDecls   ( tcTyAndClassDecls )
 import RnNames         ( rnImports, exportsFromAvail, reportUnusedNames )
 import RnIfaces                ( slurpImpDecls, checkVersions, RecompileRequired, outOfDate )
 import RnHiFiles       ( readIface, loadOldIface )
-import RnEnv           ( lookupSrcName, lookupOccRn,
+import RnEnv           ( lookupSrcName, lookupOccRn, plusGlobalRdrEnv,
                          ubiquitousNames, implicitModuleFVs, implicitStmtFVs, dataTcOccs )
 import RnExpr          ( rnStmts, rnExpr )
 import RnNames         ( importsFromLocalDecls )
@@ -88,7 +89,7 @@ import ErrUtils               ( mkDumpDoc, showPass )
 import Id              ( Id, mkLocalId, isLocalId, idName, idType, idUnfolding, setIdLocalExported )
 import IdInfo          ( GlobalIdDetails(..) )
 import Var             ( Var, setGlobalIdDetails )
-import Module           ( Module, moduleName, moduleUserString )
+import Module           ( Module, moduleName, moduleUserString, moduleEnvElts )
 import Name            ( Name, isExternalName, getSrcLoc, nameOccName )
 import NameEnv         ( delListFromNameEnv )
 import NameSet
@@ -100,7 +101,7 @@ import HscTypes             ( PersistentCompilerState(..), InteractiveContext(..),
                          ModIface, ModDetails(..), ModGuts(..),
                          HscEnv(..), 
                          ModIface(..), ModDetails(..), IfaceDecls(..),
-                         GhciMode(..), 
+                         GhciMode(..), noDependencies,
                          Deprecations(..), plusDeprecs,
                          emptyGlobalRdrEnv,
                          GenAvailInfo(Avail), availsToNameSet, 
@@ -112,7 +113,7 @@ import HscTypes             ( PersistentCompilerState(..), InteractiveContext(..),
 #ifdef GHCI
 import RdrName         ( rdrEnvElts )
 import RnHiFiles       ( loadInterface )
-import RnEnv           ( mkGlobalRdrEnv, plusGlobalRdrEnv )
+import RnEnv           ( mkGlobalRdrEnv )
 import HscTypes                ( GlobalRdrElt(..), GlobalRdrEnv, ImportReason(..), Provenance(..), 
                          isLocalGRE )
 #endif
@@ -145,9 +146,9 @@ tcRnModule hsc_env pcs
    do {        -- Deal with imports; sets tcg_rdr_env, tcg_imports
        (rdr_env, imports) <- rnImports import_decls ;
        updGblEnv ( \ gbl -> gbl { tcg_rdr_env = rdr_env,
-                                  tcg_imports = imports }) 
+                                  tcg_imports = tcg_imports gbl `plusImportAvails` imports }) 
                     $ do {
-       traceRn (text "rn1") ;
+       traceRn (text "rn1" <+> ppr (imp_dep_mods imports)) ;
                -- Fail if there are any errors so far
                -- The error printing (if needed) takes advantage 
                -- of the tcg_env we have now set
@@ -172,7 +173,6 @@ tcRnModule hsc_env pcs
        updGblEnv (\gbl -> gbl { tcg_deprecs = tcg_deprecs gbl `plusDeprecs` mod_deprecs })
                  $ do {
 
-       traceRn (text "rn4") ;
                -- Process the export list
        export_avails <- exportsFromAvail exports ;
        updGblEnv (\gbl -> gbl { tcg_exports = export_avails })
@@ -261,8 +261,11 @@ tcRnStmt :: HscEnv -> PersistentCompilerState
         -> RdrNameStmt
         -> IO (PersistentCompilerState, 
                Maybe (InteractiveContext, [Name], TypecheckedHsExpr))
-               -- The returned [Id] is the same as the input except for
+               -- The returned [Name] is the same as the input except for
                -- ExprStmt, in which case the returned [Name] is [itName]
+               --
+               -- The returned TypecheckedHsExpr is of type IO [ () ],
+               -- a list of the bound values, coerced to ().
 
 tcRnStmt hsc_env pcs ictxt rdr_stmt
   = initTc hsc_env pcs iNTERACTIVE $ 
@@ -486,9 +489,16 @@ tcRnThing hsc_env pcs ictxt rdr_name
        do { addMessages (head msgs_s) ; failM }
     else do {
 
-    mapM_ addMessages msgs_s ; -- Add deprecation warnings
-    mapM tcLookupGlobal names  -- and lookup up the entities
-    }}
+       -- Add deprecation warnings
+    mapM_ addMessages msgs_s ; 
+
+       -- Slurp in the supporting declarations
+    tcg_env <- importSupportingDecls (mkFVs names) ;
+    setGblEnv tcg_env $ do {
+
+       -- And lookup up the entities
+    mapM tcLookupGlobal names
+    }}}
 \end{code}
 
 
@@ -557,6 +567,7 @@ tcRnExtCore hsc_env pcs
        mod_guts = ModGuts {    mg_module   = this_mod,
                                mg_usages   = [],       -- ToDo: compute usage
                                mg_dir_imps = [],       -- ??
+                               mg_deps     = noDependencies,   -- ??
                                mg_exports  = my_exports,
                                mg_types    = final_type_env,
                                mg_insts    = tcg_insts tcg_env,
@@ -594,16 +605,25 @@ tcRnSrcDecls ds
        -- Type check the decls up to, but not including, the first splice
        (tcg_env, src_fvs1) <- tcRnGroup first_group ;
 
-       -- If there is no splice, we're done
-       case group_tail of
-          Nothing -> return (tcg_env, src_fvs1)
-          Just (splice_expr, rest_ds) -> do {
+       -- Bale out if errors; for example, error recovery when checking
+       -- the RHS of 'main' can mean that 'main' is not in the envt for 
+       -- the subsequent checkMain test
+       failIfErrsM ;
 
+       -- If there is no splice, we're done
+       case group_tail of {
+          Nothing -> return (tcg_env, src_fvs1) ;
+          Just (SpliceDecl splice_expr splice_loc, rest_ds) -> 
+#ifndef GHCI
+       failWithTc (text "Can't do a top-level splice; need a bootstrapped compiler")
+#else
        setGblEnv tcg_env $ do {
-               
+
        -- Rename the splice expression, and get its supporting decls
-       (rn_splice_expr, fvs) <- initRn SourceMode (rnExpr splice_expr) ;
-       tcg_env <- importSupportingDecls fvs ;
+       (rn_splice_expr, fvs) <- initRn SourceMode $
+                                addSrcLoc splice_loc $
+                                rnExpr splice_expr ;
+       tcg_env <- importSupportingDecls (fvs `plusFV` templateHaskellNames) ;
        setGblEnv tcg_env $ do {
 
        -- Execute the splice
@@ -613,7 +633,9 @@ tcRnSrcDecls ds
        (tcg_env, src_fvs2) <- tcRnSrcDecls (spliced_decls ++ rest_ds) ;
 
        return (tcg_env, src_fvs1 `plusFV` src_fvs2)
-    }}}}
+    }}
+#endif /* GHCI */
+    }}
 \end{code}
 
 
@@ -658,6 +680,8 @@ rnTopSrcDecls group
                                                  tcg_imports gbl }) 
                     $ do {
 
+       failIfErrsM ;   -- No point in continuing if (say) we have duplicate declarations
+
                -- Rename the source decls
        (tcg_env, rn_src_decls, src_fvs) <- initRn SourceMode (rnSrcDecls group) ;
        setGblEnv tcg_env $ do {
@@ -678,15 +702,9 @@ rnTopSrcDecls group
 ------------------------------------------------
 tcTopSrcDecls :: HsGroup Name -> TcM TcGblEnv
 tcTopSrcDecls rn_decls
- = fixM (\ unf_env -> do {     
-       -- Loop back the final environment, including the fully zonked
-       -- versions of bindings from this module.  In the presence of mutual
-       -- recursion, interface type signatures may mention variables defined
-       -- in this module, which is why the knot is so big
-
-                       -- Do the main work
-       ((tcg_env, binds, rules, fords), lie) <- getLIE (
-               tc_src_decls unf_env rn_decls
+ = do {                        -- Do the main work
+       ((tcg_env, lcl_env, binds, rules, fords), lie) <- getLIE (
+               tc_src_decls rn_decls
            ) ;
 
             -- tcSimplifyTop deals with constant or ambiguous InstIds.  
@@ -696,24 +714,29 @@ tcTopSrcDecls rn_decls
             -- type.  (Usually, ambiguous type variables are resolved
             -- during the generalisation step.)
         traceTc (text "Tc8") ;
-       inst_binds <- setGblEnv tcg_env (tcSimplifyTop lie) ;
+       inst_binds <- setGblEnv tcg_env $
+                     setLclTypeEnv lcl_env $
+                     tcSimplifyTop lie ;
                -- The setGblEnv exposes the instances to tcSimplifyTop
+               -- The setLclTypeEnv exposes the local Ids, so that
+               -- we get better error messages (monomorphism restriction)
 
            -- Backsubstitution.  This must be done last.
            -- Even tcSimplifyTop may do some unification.
         traceTc (text "Tc9") ;
-       (ids, binds', fords', rules') <- zonkTopDecls (binds `andMonoBinds` inst_binds)
-                                                     rules fords ;
+       (bind_ids, binds', fords', rules') <- zonkTopDecls (binds `andMonoBinds` inst_binds)
+                                                          rules fords ;
 
-       let { tcg_env' = tcg_env { tcg_type_env = extendTypeEnvWithIds (tcg_type_env tcg_env) ids,
+       let { tcg_env' = tcg_env { tcg_type_env = extendTypeEnvWithIds (tcg_type_env tcg_env) 
+                                                                      bind_ids,
                                   tcg_binds = tcg_binds tcg_env `andMonoBinds` binds',
                                   tcg_rules = tcg_rules tcg_env ++ rules',
                                   tcg_fords = tcg_fords tcg_env ++ fords' } } ;
        
        return tcg_env' 
-    })
+    }
 
-tc_src_decls unf_env 
+tc_src_decls
        (HsGroup { hs_tyclds = tycl_decls, 
                   hs_instds = inst_decls,
                   hs_fords  = foreign_decls,
@@ -722,7 +745,7 @@ tc_src_decls unf_env
                   hs_valds  = val_binds })
  = do {                -- Type-check the type and class decls, and all imported decls
         traceTc (text "Tc2") ;
-       tcg_env <- tcTyClDecls unf_env tycl_decls ;
+       tcg_env <- tcTyClDecls tycl_decls ;
        setGblEnv tcg_env       $ do {
 
                -- Source-language instances, including derivings,
@@ -782,13 +805,12 @@ tc_src_decls unf_env
                          cls_dm_binds   `AndMonoBinds`
                          foe_binds } ;
 
-       return (tcg_env, all_binds, src_rules, foe_decls)
+       return (tcg_env, lcl_env, all_binds, src_rules, foe_decls)
      }}}}}}}}}
 \end{code}
 
 \begin{code}
-tcTyClDecls :: RecTcGblEnv
-           -> [RenamedTyClDecl]
+tcTyClDecls :: [RenamedTyClDecl]
            -> TcM TcGblEnv
 
 -- tcTyClDecls deals with 
@@ -799,11 +821,7 @@ tcTyClDecls :: RecTcGblEnv
 -- persistent compiler state to reflect the things imported from
 -- other modules
 
-tcTyClDecls unf_env tycl_decls
-  -- (unf_env :: RecTcGblEnv) is used for type-checking interface pragmas
-  -- which is done lazily [ie failure just drops the pragma
-  -- without having any global-failure effect].
-
+tcTyClDecls tycl_decls
   = checkNoErrs $
        -- tcTyAndClassDecls recovers internally, but if anything gave rise to
        -- an error we'd better stop now, to avoid a cascade
@@ -811,18 +829,12 @@ tcTyClDecls unf_env tycl_decls
     traceTc (text "TyCl1")             `thenM_`
     tcTyAndClassDecls tycl_decls       `thenM` \ tycl_things ->
     tcExtendGlobalEnv tycl_things      $
-    
-       -- Interface type signatures
-       -- We tie a knot so that the Ids read out of interfaces are in scope
-       --   when we read their pragmas.
-       -- What we rely on is that pragmas are typechecked lazily; if
-       --   any type errors are found (ie there's an inconsistency)
-       --   we silently discard the pragma
-    traceTc (text "TyCl2")                     `thenM_`
-    tcInterfaceSigs unf_env tycl_decls         `thenM` \ sig_ids ->
-    tcExtendGlobalValEnv sig_ids               $
-    
-    getGblEnv          -- Return the TcLocals environment
+
+    traceTc (text "TyCl2")             `thenM_`
+    tcInterfaceSigs tycl_decls         `thenM` \ tcg_env ->
+       -- Returns the extended environment
+
+    returnM tcg_env
 \end{code}    
 
 
@@ -922,7 +934,7 @@ typecheckIfaceDecls (HsGroup { hs_tyclds = tycl_decls,
                               hs_instds = inst_decls,
                               hs_ruleds = rule_decls })
  = do {                -- Typecheck the type, class, and interface-sig decls
-       tcg_env <- fixM (\ unf_env -> tcTyClDecls unf_env tycl_decls) ;
+       tcg_env <- tcTyClDecls tycl_decls ;
        setGblEnv tcg_env               $ do {
        
        -- Typecheck the instance decls, and rules
@@ -1157,11 +1169,14 @@ tcCoreDump mod_guts
 pprTcGblEnv :: TcGblEnv -> SDoc
 pprTcGblEnv (TcGblEnv { tcg_type_env = type_env, 
                        tcg_insts    = dfun_ids, 
-                       tcg_rules    = rules })
+                       tcg_rules    = rules,
+                       tcg_imports  = imports })
   = vcat [ ppr_types dfun_ids type_env
         , ppr_insts dfun_ids
         , vcat (map ppr rules)
-        , ppr_gen_tycons (typeEnvTyCons type_env)]
+        , ppr_gen_tycons (typeEnvTyCons type_env)
+        , ppr (moduleEnvElts (imp_dep_mods imports))
+        , ppr (imp_dep_pkgs imports)]
 
 pprModGuts :: ModGuts -> SDoc
 pprModGuts (ModGuts { mg_types = type_env,