[project @ 2002-03-14 16:22:31 by simonmar]
[ghc-hetmet.git] / ghc / compiler / rename / Rename.lhs
index c99a63a..d9a4dcb 100644 (file)
@@ -5,8 +5,8 @@
 
 \begin{code}
 module Rename ( 
-       renameModule, renameStmt, renameRdrName, mkGlobalContext,
-       closeIfaceDecls, checkOldIface 
+       renameModule, RnResult(..), renameStmt, renameRdrName, mkGlobalContext,
+       closeIfaceDecls, checkOldIface, slurpIface
   ) where
 
 #include "HsVersions.h"
@@ -21,7 +21,7 @@ import RnHsSyn                ( RenamedHsDecl, RenamedTyClDecl, RenamedRuleDecl, RenamedInstDe
                          instDeclFVs, tyClDeclFVs, ruleDeclFVs
                        )
 
-import CmdLineOpts     ( DynFlags, DynFlag(..) )
+import CmdLineOpts     ( DynFlags, DynFlag(..), opt_InPackage )
 import RnMonad
 import RnExpr          ( rnStmt )
 import RnNames         ( getGlobalNames, exportsFromAvail )
@@ -39,14 +39,14 @@ import RnEnv                ( availsToNameSet,
                          warnUnusedLocalBinds, warnUnusedModules, 
                          lookupSrcName, getImplicitStmtFVs, 
                          getImplicitModuleFVs, newGlobalName, unQualInScope,
-                         ubiquitousNames, lookupOccRn, 
+                         ubiquitousNames, lookupOccRn, checkMain,
                          plusGlobalRdrEnv, mkGlobalRdrEnv
                        )
 import Module           ( Module, ModuleName, WhereFrom(..),
                          moduleNameUserString, moduleName,
                          moduleEnvElts
                        )
-import Name            ( Name, nameModule )
+import Name            ( Name, nameModule, isExternalName )
 import NameEnv
 import NameSet
 import RdrName         ( foldRdrEnv, isQual )
@@ -72,17 +72,17 @@ import List         ( partition, nub )
 %*********************************************************
 
 \begin{code}
-renameModule :: DynFlags
+renameModule :: DynFlags -> GhciMode
             -> HomeIfaceTable -> HomeSymbolTable
             -> PersistentCompilerState 
             -> Module -> RdrNameHsModule 
             -> IO (PersistentCompilerState, PrintUnqualified,
-                   Maybe (IsExported, ModIface, [RenamedHsDecl]))
+                   Maybe (IsExported, ModIface, RnResult))
        -- Nothing => some error occurred in the renamer
 
-renameModule dflags hit hst pcs this_module rdr_module
+renameModule dflags ghci_mode hit hst pcs this_module rdr_module
   = renameSource dflags hit hst pcs this_module $
-    rename this_module rdr_module
+    rename ghci_mode this_module rdr_module
 \end{code}
 
 \begin{code}
@@ -243,6 +243,31 @@ getModuleExports mod =
 
 %*********************************************************
 %*                                                      *
+\subsection{Slurp in a whole module eagerly}
+%*                                                      *
+%*********************************************************
+
+\begin{code}
+slurpIface
+       :: DynFlags -> HomeIfaceTable -> HomeSymbolTable
+       -> PersistentCompilerState -> Module
+       -> IO (PersistentCompilerState, PrintUnqualified, 
+              Maybe ([Name], [RenamedHsDecl]))
+slurpIface dflags hit hst pcs mod = 
+  renameSource dflags hit hst pcs iNTERACTIVE $
+
+    let mod_name = moduleName mod
+    in
+    loadInterface contextDoc mod_name ImportByUser `thenRn` \ iface ->
+    let fvs = availsToNameSet [ avail | (mn,avails) <- mi_exports iface, 
+                                       avail <- avails ]
+    in
+    slurpImpDecls fvs  `thenRn` \ rn_imp_decls ->
+    returnRn (alwaysQualify, Just (nameSetToList fvs, rn_imp_decls))
+\end{code}
+
+%*********************************************************
+%*                                                      *
 \subsection{The main function: rename}
 %*                                                      *
 %*********************************************************
@@ -275,9 +300,22 @@ renameSource dflags hit hst old_pcs this_module thing_inside
 \end{code}
 
 \begin{code}
-rename :: Module -> RdrNameHsModule 
-       -> RnMG (PrintUnqualified, Maybe (IsExported, ModIface, [RenamedHsDecl]))
-rename this_module contents@(HsModule _ _ exports imports local_decls mod_deprec loc)
+data RnResult  -- A RenamedModule ia passed from renamer to typechecker
+  = RnResult { rr_mod      :: Module,    -- Same as in the ModIface, 
+              rr_fixities :: FixityEnv,  -- but convenient to have it here
+
+              rr_main :: Maybe Name,     -- Just main, for module Main, 
+                                         -- Nothing for other modules
+
+              rr_decls :: [RenamedHsDecl]      
+                       -- The other declarations of the module
+                       -- Fixity and deprecations have already been slurped out
+    }                  -- and are now in the ModIface for the module
+
+rename :: GhciMode -> Module -> RdrNameHsModule 
+       -> RnMG (PrintUnqualified, Maybe (IsExported, ModIface, RnResult))
+rename ghci_mode this_module 
+       contents@(HsModule _ _ exports imports local_decls mod_deprec loc)
   = pushSrcLocRn loc           $
 
        -- FIND THE GLOBAL NAME ENVIRONMENT
@@ -327,6 +365,26 @@ rename this_module contents@(HsModule _ _ exports imports local_decls mod_deprec
     rnSourceDecls gbl_env global_avail_env 
                  local_fixity_env local_decls          `thenRn` \ (rn_local_decls, source_fvs) ->
 
+       -- GET ANY IMPLICIT FREE VARIALBES
+    getImplicitModuleFVs rn_local_decls          `thenRn` \ implicit_fvs ->
+    checkMain ghci_mode mod_name gbl_env  `thenRn` \ (maybe_main_name, main_fvs, implicit_main_fvs) ->
+    let
+       export_fvs = availsToNameSet export_avails
+       used_fvs   = source_fvs `plusFV` export_fvs `plusFV` main_fvs
+               -- The export_fvs make the exported names look just as if they
+               -- occurred in the source program.  For the reasoning, see the
+               -- comments with RnIfaces.mkImportInfo
+               -- It also helps reportUnusedNames, which of course must not complain
+               -- that 'f' isn't mentioned if it is mentioned in the export list
+
+       needed_fvs = implicit_fvs `plusFV` implicit_main_fvs `plusFV` used_fvs
+               -- It's important to do the "plus" this way round, so that
+               -- when compiling the prelude, locally-defined (), Bool, etc
+               -- override the implicit ones. 
+
+    in
+    traceRn (text "Needed FVs:" <+> fsep (map ppr (nameSetToList needed_fvs))) `thenRn_`
+
        -- EXIT IF ERRORS FOUND
        -- We exit here if there are any errors in the source, *before*
        -- we attempt to slurp the decls from the interfaces, otherwise
@@ -340,25 +398,7 @@ rename this_module contents@(HsModule _ _ exports imports local_decls mod_deprec
     else
 
        -- SLURP IN ALL THE NEEDED DECLARATIONS
-       -- Find out what re-bindable names to use for desugaring
-    getImplicitModuleFVs mod_name rn_local_decls       `thenRn` \ implicit_fvs ->
-    let
-       export_fvs  = availsToNameSet export_avails
-       source_fvs2 = source_fvs `plusFV` export_fvs
-               -- The export_fvs make the exported names look just as if they
-               -- occurred in the source program.  For the reasoning, see the
-               -- comments with RnIfaces.mkImportInfo
-               -- It also helps reportUnusedNames, which of course must not complain
-               -- that 'f' isn't mentioned if it is mentioned in the export list
-
-       source_fvs3 = implicit_fvs `plusFV` source_fvs2
-               -- It's important to do the "plus" this way round, so that
-               -- when compiling the prelude, locally-defined (), Bool, etc
-               -- override the implicit ones. 
-
-    in
-    traceRn (text "Source FVs:" <+> fsep (map ppr (nameSetToList source_fvs3)))        `thenRn_`
-    slurpImpDecls source_fvs3                  `thenRn` \ rn_imp_decls ->
+    slurpImpDecls needed_fvs                   `thenRn` \ rn_imp_decls ->
     rnDump rn_imp_decls rn_local_decls         `thenRn_` 
 
        -- GENERATE THE VERSION/USAGE INFO
@@ -377,7 +417,21 @@ rename this_module contents@(HsModule _ _ exports imports local_decls mod_deprec
        
        final_decls = rn_local_decls ++ rn_imp_decls
 
+       -- In interactive mode, we don't want to discard any top-level
+       -- entities at all (eg. do not inline them away during
+       -- simplification), and retain them all in the TypeEnv so they are
+       -- available from the command line.
+       --
+       -- isExternalName separates the user-defined top-level names from those
+       -- introduced by the type checker.
+       dont_discard :: Name -> Bool
+       dont_discard | ghci_mode == Interactive = isExternalName
+                    | otherwise                = (`elemNameSet` exported_names)
+
+       exported_names    = availsToNameSet export_avails
+
        mod_iface = ModIface {  mi_module   = this_module,
+                               mi_package  = opt_InPackage,
                                mi_version  = initialVersionInfo,
                                mi_usages   = my_usages,
                                mi_boot     = False,
@@ -389,18 +443,20 @@ rename this_module contents@(HsModule _ _ exports imports local_decls mod_deprec
                                mi_decls    = panic "mi_decls"
                    }
 
-       is_exported name  = name `elemNameSet` exported_names
-       exported_names    = availsToNameSet export_avails
+       rn_result = RnResult { rr_mod      = this_module,
+                              rr_fixities = fixities,
+                              rr_decls    = final_decls,
+                              rr_main     = maybe_main_name }
     in
 
        -- REPORT UNUSED NAMES, AND DEBUG DUMP 
     reportUnusedNames mod_iface print_unqualified 
                      imports full_avail_env gbl_env
-                     source_fvs2 rn_imp_decls          `thenRn_`
-               -- NB: source_fvs2: include exports (else we get bogus 
+                     used_fvs rn_imp_decls             `thenRn_`
+               -- NB: used_fvs: include exports (else we get bogus 
                --     warnings of unused things) but not implicit FVs.
 
-    returnRn (print_unqualified, Just (is_exported, mod_iface, final_decls))
+    returnRn (print_unqualified, Just (dont_discard, mod_iface, rn_result))
   where
     mod_name = moduleName this_module
 \end{code}
@@ -486,13 +542,14 @@ checkOldIface :: GhciMode
               -> DynFlags
              -> HomeIfaceTable -> HomeSymbolTable
              -> PersistentCompilerState
+             -> Module
              -> FilePath
              -> Bool                   -- Source unchanged
              -> Maybe ModIface         -- Old interface from compilation manager, if any
              -> IO (PersistentCompilerState, Bool, (RecompileRequired, Maybe ModIface))
                                -- True <=> errors happened
 
-checkOldIface ghci_mode dflags hit hst pcs iface_path source_unchanged maybe_iface
+checkOldIface ghci_mode dflags hit hst pcs mod iface_path source_unchanged maybe_iface
     = runRn dflags hit hst pcs (panic "Bogus module") $
 
        -- CHECK WHETHER THE SOURCE HAS CHANGED
@@ -506,9 +563,10 @@ checkOldIface ghci_mode dflags hit hst pcs iface_path source_unchanged maybe_ifa
          returnRn (outOfDate, maybe_iface)
     else
 
+    setModuleRn mod $
     case maybe_iface of
        Just old_iface -> -- Use the one we already have
-                         setModuleRn (mi_module old_iface) (check_versions old_iface)
+                         check_versions old_iface
 
        Nothing -- try and read it from a file
           -> readIface iface_path      `thenRn` \ read_result ->
@@ -519,9 +577,18 @@ checkOldIface ghci_mode dflags hit hst pcs iface_path source_unchanged maybe_ifa
                                   $$ nest 4 err) `thenRn_`
                           returnRn (outOfDate, Nothing)
 
-               Right parsed_iface
-                      -> setModuleRn (pi_mod parsed_iface) $
-                         loadOldIface parsed_iface `thenRn` \ m_iface ->
+               Right parsed_iface ->
+                     let read_mod_name = pi_mod parsed_iface
+                         wanted_mod_name = moduleName mod
+                     in
+                     if (wanted_mod_name /= read_mod_name) then
+                        traceHiDiffsRn (
+                           text "Existing interface file has wrong module name: "
+                                <> quotes (ppr read_mod_name)
+                               ) `thenRn_`
+                        returnRn (outOfDate, Nothing)
+                     else
+                         loadOldIface mod parsed_iface `thenRn` \ m_iface ->
                          check_versions m_iface
     where
        check_versions :: ModIface -> RnMG (RecompileRequired, Maybe ModIface)
@@ -538,11 +605,10 @@ I think the following function should now have a more representative name,
 but what?
 
 \begin{code}
-loadOldIface :: ParsedIface -> RnMG ModIface
+loadOldIface :: Module -> ParsedIface -> RnMG ModIface
 
-loadOldIface parsed_iface
+loadOldIface mod parsed_iface
   = let iface = parsed_iface 
-        mod = pi_mod iface
     in
     initIfaceRnMS mod (
        loadHomeDecls (pi_decls iface)  `thenRn` \ decls ->
@@ -564,7 +630,8 @@ loadOldIface parsed_iface
 
        decls = mkIfaceDecls new_decls new_rules new_insts
 
-       mod_iface = ModIface { mi_module = mod, mi_version = version,
+       mod_iface = ModIface { mi_module = mod, mi_package = pi_pkg parsed_iface,
+                              mi_version = version,
                               mi_exports = avails, mi_usages  = usages,
                               mi_boot = False, mi_orphan = pi_orphan iface, 
                               mi_fixities = fix_env, mi_deprecs = deprec_env,