[project @ 2000-11-20 16:07:12 by simonpj]
[ghc-hetmet.git] / ghc / compiler / rename / RnNames.lhs
index dd44505..571ee3a 100644 (file)
@@ -5,7 +5,7 @@
 
 \begin{code}
 module RnNames (
-       getGlobalNames
+       getGlobalNames, exportsFromAvail
     ) where
 
 #include "HsVersions.h"
@@ -25,13 +25,12 @@ import RnEnv
 import RnMonad
 
 import FiniteMap
-import PrelNames       ( pRELUDE_Name, mAIN_Name, main_RDR )
+import PrelNames       ( pRELUDE_Name, mAIN_Name, main_RDR_Unqual, isUnboundName )
 import UniqFM          ( lookupUFM )
 import Bag             ( bagToList )
 import Module          ( ModuleName, moduleName, WhereFrom(..) )
 import NameSet
-import Name            ( Name, nameSrcLoc,
-                         setLocalNameSort, nameOccName,  nameEnvElts )
+import Name            ( Name, nameSrcLoc, nameOccName,  nameEnvElts )
 import HscTypes                ( Provenance(..), ImportReason(..), GlobalRdrEnv,
                          GenAvailInfo(..), AvailInfo, Avails, AvailEnv )
 import RdrName         ( RdrName, rdrNameOcc, setRdrNameOcc, mkRdrQual, mkRdrUnqual )
@@ -57,27 +56,13 @@ import List         ( partition )
 getGlobalNames :: Module -> RdrNameHsModule
               -> RnMG (GlobalRdrEnv,   -- Maps all in-scope things
                        GlobalRdrEnv,   -- Maps just *local* things
-                       Avails,         -- The exported stuff
-                       AvailEnv)       -- Maps a name to its parent AvailInfo
-                                       -- Just for in-scope things only
+                       ExportAvails)   -- The exported stuff
 
-getGlobalNames this_mod (HsModule _ _ exports imports decls _ mod_loc)
-  =    -- These two fix-loops are to get the right
-       -- provenance information into a Name
-    fixRn ( \ ~(rec_gbl_env, _, rec_export_avails, _) ->
-
-       let
-          rec_unqual_fn :: Name -> Bool        -- Is this chap in scope unqualified?
-          rec_unqual_fn = unQualInScope rec_gbl_env
-
-          rec_exp_fn :: Name -> Bool
-          rec_exp_fn = mk_export_fn (availsToNameSet rec_export_avails)
-       in
-
-               -- PROCESS LOCAL DECLS
+getGlobalNames this_mod (HsModule _ _ _ imports decls _ mod_loc)
+  =            -- PROCESS LOCAL DECLS
                -- Do these *first* so that the correct provenance gets
                -- into the global name cache.
-       importsFromLocalDecls this_mod rec_exp_fn decls         `thenRn` \ (local_gbl_env, local_mod_avails) ->
+       importsFromLocalDecls this_mod decls            `thenRn` \ (local_gbl_env, local_mod_avails) ->
 
                -- PROCESS IMPORT DECLS
                -- Do the non {- SOURCE -} ones first, so that we get a helpful
@@ -88,9 +73,11 @@ getGlobalNames this_mod (HsModule _ _ exports imports decls _ mod_loc)
          (source, ordinary) = partition is_source_import all_imports
          is_source_import (ImportDecl _ ImportByUserSource _ _ _ _) = True
          is_source_import other                                     = False
+
+         get_imports = importsFromImportDecl this_mod_name
        in
-       mapAndUnzipRn (importsFromImportDecl rec_unqual_fn) ordinary    `thenRn` \ (imp_gbl_envs1, imp_avails_s1) ->
-       mapAndUnzipRn (importsFromImportDecl rec_unqual_fn) source      `thenRn` \ (imp_gbl_envs2, imp_avails_s2) ->
+       mapAndUnzipRn get_imports ordinary      `thenRn` \ (imp_gbl_envs1, imp_avails_s1) ->
+       mapAndUnzipRn get_imports source        `thenRn` \ (imp_gbl_envs2, imp_avails_s2) ->
 
                -- COMBINE RESULTS
                -- We put the local env second, so that a local provenance
@@ -102,21 +89,10 @@ getGlobalNames this_mod (HsModule _ _ exports imports decls _ mod_loc)
 
            all_avails :: ExportAvails
            all_avails = foldr plusExportAvails local_mod_avails (imp_avails_s2 ++ imp_avails_s1)
-
-           (_, global_avail_env) = all_avails
        in
 
-               -- PROCESS EXPORT LIST (but not if we've had errors already)
-       checkErrsRn             `thenRn` \ no_errs_so_far ->
-       (if no_errs_so_far then
-           exportsFromAvail this_mod_name exports all_avails gbl_env
-        else
-           returnRn []
-       )                                               `thenRn` \ export_avails ->
-       
                -- ALL DONE
-       returnRn (gbl_env, local_gbl_env, export_avails, global_avail_env)
-   )
+       returnRn (gbl_env, local_gbl_env, all_avails)
   where
     this_mod_name = moduleName this_mod
 
@@ -141,12 +117,12 @@ getGlobalNames this_mod (HsModule _ _ exports imports decls _ mod_loc)
 \end{code}
        
 \begin{code}
-importsFromImportDecl :: (Name -> Bool)                -- OK to omit qualifier
+importsFromImportDecl :: ModuleName
                      -> RdrNameImportDecl
                      -> RnMG (GlobalRdrEnv, 
                               ExportAvails) 
 
-importsFromImportDecl is_unqual (ImportDecl imp_mod_name from qual_only as_mod import_spec iloc)
+importsFromImportDecl this_mod_name (ImportDecl imp_mod_name from qual_only as_mod import_spec iloc)
   = pushSrcLocRn iloc $
     getInterfaceExports imp_mod_name from      `thenRn` \ (imp_mod, avails_by_module) ->
 
@@ -158,13 +134,31 @@ importsFromImportDecl is_unqual (ImportDecl imp_mod_name from qual_only as_mod i
 
     let
        avails :: Avails
-       avails = concat (map snd avails_by_module)
+       avails = [ avail | (mod_name, avails) <- avails_by_module,
+                          mod_name /= this_mod_name,
+                          avail <- avails ]
+       -- If the module exports anything defined in this module, just ignore it.
+       -- Reason: otherwise it looks as if there are two local definition sites
+       -- for the thing, and an error gets reported.  Easiest thing is just to
+       -- filter them out up front. This situation only arises if a module
+       -- imports itself, or another module that imported it.  (Necessarily,
+       -- this invoves a loop.)  
+       --
+       -- Tiresome consequence: if you say
+       --      module A where
+       --         import B( AType )
+       --         type AType = ...
+       --
+       --      module B( AType ) where
+       --         import {-# SOURCE #-} A( AType )
+       --
+       -- then you'll get a 'B does not export AType' message.  Oh well.
+
     in
-    filterImports imp_mod_name import_spec avails      `thenRn` \ (filtered_avails, hides, explicits) ->
+    filterImports imp_mod_name from import_spec avails `thenRn` \ (filtered_avails, hides, explicits) ->
 
     let
        mk_provenance name = NonLocalDef (UserImport imp_mod iloc (name `elemNameSet` explicits)) 
-                                        (is_unqual name)
     in
 
     qualifyImports imp_mod_name
@@ -176,8 +170,8 @@ importsFromImportDecl is_unqual (ImportDecl imp_mod_name from qual_only as_mod i
 
 
 \begin{code}
-importsFromLocalDecls this_mod rec_exp_fn decls
-  = mapRn (getLocalDeclBinders this_mod rec_exp_fn) decls      `thenRn` \ avails_s ->
+importsFromLocalDecls this_mod decls
+  = mapRn (getLocalDeclBinders this_mod) decls `thenRn` \ avails_s ->
 
     let
        avails = concat avails_s
@@ -189,10 +183,10 @@ importsFromLocalDecls this_mod rec_exp_fn decls
        (_, dups) = removeDups compare all_names
     in
        -- Check for duplicate definitions
-    mapRn_ (addErrRn . dupDeclErr) dups                `thenRn_` 
+    mapRn_ (addErrRn . dupDeclErr) dups                        `thenRn_` 
 
        -- Record that locally-defined things are available
-    recordLocalSlurps avails                   `thenRn_`
+    recordLocalSlurps (availsToNameSet avails)         `thenRn_`
 
        -- Build the environment
     qualifyImports (moduleName this_mod)
@@ -204,9 +198,8 @@ importsFromLocalDecls this_mod rec_exp_fn decls
 
 ---------------------------
 getLocalDeclBinders :: Module 
-                   -> (Name -> Bool)   -- Whether exported
                    -> RdrNameHsDecl -> RnMG Avails
-getLocalDeclBinders mod rec_exp_fn (TyClD tycl_decl)
+getLocalDeclBinders mod (TyClD tycl_decl)
   =    -- For type and class decls, we generate Global names, with
        -- no export indicator.  They need to be global because they get
        -- permanently bound into the TyCons and Classes.  They don't need
@@ -214,14 +207,16 @@ getLocalDeclBinders mod rec_exp_fn (TyClD tycl_decl)
     getTyClDeclBinders mod tycl_decl   `thenRn` \ avail ->
     returnRn [avail]
 
-getLocalDeclBinders mod rec_exp_fn (ValD binds)
-  = mapRn (newLocalBinder mod rec_exp_fn) 
-         (bagToList (collectTopBinders binds))
+getLocalDeclBinders mod (ValD binds)
+  = mapRn new (bagToList (collectTopBinders binds))
+  where
+    new (rdr_name, loc) = newTopBinder mod rdr_name loc        `thenRn` \ name ->
+                         returnRn (Avail name)
 
-getLocalDeclBinders mod rec_exp_fn (ForD (ForeignDecl nm kind _ ext_nm _ loc))
+getLocalDeclBinders mod (ForD (ForeignDecl nm kind _ ext_nm _ loc))
   | binds_haskell_name kind
-  = newLocalBinder mod rec_exp_fn (nm, loc)        `thenRn` \ avail ->
-    returnRn [avail]
+  = newTopBinder mod nm loc        `thenRn` \ name ->
+    returnRn [Avail name]
 
   | otherwise          -- a foreign export
   = returnRn []
@@ -230,17 +225,11 @@ getLocalDeclBinders mod rec_exp_fn (ForD (ForeignDecl nm kind _ ext_nm _ loc))
     binds_haskell_name FoLabel      = True
     binds_haskell_name FoExport     = isDynamicExtName ext_nm
 
-getLocalDeclBinders mod rec_exp_fn (FixD _)    = returnRn []
-getLocalDeclBinders mod rec_exp_fn (DeprecD _) = returnRn []
-getLocalDeclBinders mod rec_exp_fn (DefD _)    = returnRn []
-getLocalDeclBinders mod rec_exp_fn (InstD _)   = returnRn []
-getLocalDeclBinders mod rec_exp_fn (RuleD _)   = returnRn []
-
----------------------------
-newLocalBinder mod rec_exp_fn (rdr_name, loc)
-  =    -- Generate a local name, and with a suitable export indicator
-    newTopBinder mod rdr_name loc      `thenRn` \ name ->
-    returnRn (Avail (setLocalNameSort name (rec_exp_fn name)))
+getLocalDeclBinders mod (FixD _)    = returnRn []
+getLocalDeclBinders mod (DeprecD _) = returnRn []
+getLocalDeclBinders mod (DefD _)    = returnRn []
+getLocalDeclBinders mod (InstD _)   = returnRn []
+getLocalDeclBinders mod (RuleD _)   = returnRn []
 \end{code}
 
 
@@ -255,6 +244,7 @@ available, and filters it through the import spec (if any).
 
 \begin{code}
 filterImports :: ModuleName                    -- The module being imported
+             -> WhereFrom                      -- Tells whether it's a {-# SOURCE #-} import
              -> Maybe (Bool, [RdrNameIE])      -- Import spec; True => hiding
              -> [AvailInfo]                    -- What's available
              -> RnMG ([AvailInfo],             -- What's actually imported
@@ -267,10 +257,10 @@ filterImports :: ModuleName                       -- The module being imported
 
        -- Complains if import spec mentions things that the module doesn't export
         -- Warns/informs if import spec contains duplicates.
-filterImports mod Nothing imports
+filterImports mod from Nothing imports
   = returnRn (imports, [], emptyNameSet)
 
-filterImports mod (Just (want_hiding, import_items)) total_avails
+filterImports mod from (Just (want_hiding, import_items)) total_avails
   = flatMapRn get_item import_items            `thenRn` \ avails_w_explicits ->
     let
        (item_avails, explicits_s) = unzip avails_w_explicits
@@ -292,7 +282,7 @@ filterImports mod (Just (want_hiding, import_items)) total_avails
        -- they won't make any difference because naked entities like T
        -- in an import list map to TcOccs, not VarOccs.
 
-    bale_out item = addErrRn (badImportItemErr mod item)       `thenRn_`
+    bale_out item = addErrRn (badImportItemErr mod from item)  `thenRn_`
                    returnRn []
 
     get_item item@(IEModuleContents _) = bale_out item
@@ -483,7 +473,7 @@ exportsFromAvail this_mod Nothing export_avails global_name_env
   = exportsFromAvail this_mod true_exports export_avails global_name_env
   where
     true_exports = Just $ if this_mod == mAIN_Name
-                          then [IEVar main_RDR]
+                          then [IEVar main_RDR_Unqual]
                                -- export Main.main *only* unless otherwise specified,
                           else [IEModuleContents this_mod]
                                -- but for all other modules export everything.
@@ -520,46 +510,38 @@ exportsFromAvail this_mod (Just export_items)
                                   returnRn (mod:mods, occs', avails')
 
     exports_from_item warn_dups acc@(mods, occs, avails) ie
-       | not (maybeToBool maybe_in_scope) 
-       = failWithRn acc (unknownNameErr (ieName ie))
-
-       | not (null dup_names)
-       = addNameClashErrRn rdr_name ((name,prov):dup_names)    `thenRn_`
-         returnRn acc
+       = lookupSrcName global_name_env (ieName ie)     `thenRn` \ name -> 
 
-#ifdef DEBUG
-       -- I can't see why this should ever happen; if the thing is in scope
-       -- at all it ought to have some availability
-       | not (maybeToBool maybe_avail)
-       = pprTrace "exportsFromAvail: curious Nothing:" (ppr name)
-         returnRn acc
-#endif
+               -- See what's available in the current environment
+         case lookupUFM entity_avail_env name of {
+           Nothing ->  -- Presumably this happens because lookupSrcName didn't find
+                       -- the name and returned an unboundName, which won't be in
+                       -- the entity_avail_env, of course
+                       WARN( not (isUnboundName name), ppr name )
+                       returnRn acc ;
 
-       | not enough_avail
-       = failWithRn acc (exportItemErr ie)
+           Just avail ->
 
-       | otherwise     -- Phew!  It's OK!  Now to check the occurrence stuff!
+               -- Filter out the bits we want
+         case filterAvail ie avail of {
+           Nothing ->  -- Not enough availability
+                          failWithRn acc (exportItemErr ie) ;
 
+           Just export_avail ->        
 
-       = warnCheckRn (ok_item ie avail) (dodgyExportWarn ie)   `thenRn_`
+               -- Phew!  It's OK!  Now to check the occurrence stuff!
+         warnCheckRn (ok_item ie avail) (dodgyExportWarn ie)   `thenRn_`
           check_occs ie occs export_avail                      `thenRn` \ occs' ->
          returnRn (mods, occs', addAvail avails export_avail)
+         }}
+
+
 
-       where
-         rdr_name        = ieName ie
-          maybe_in_scope  = lookupFM global_name_env rdr_name
-         Just ((name,prov):dup_names) = maybe_in_scope
-         maybe_avail        = lookupUFM entity_avail_env name
-         Just avail         = maybe_avail
-         maybe_export_avail = filterAvail ie avail
-         enough_avail       = maybeToBool maybe_export_avail
-         Just export_avail  = maybe_export_avail
-
-    ok_item (IEThingAll _) (AvailTC _ [n]) = False
-               -- This occurs when you import T(..), but
-               -- only export T abstractly.  The single [n]
-               -- in the AvailTC is the type or class itself
-    ok_item _ _ = True
+ok_item (IEThingAll _) (AvailTC _ [n]) = False
+  -- This occurs when you import T(..), but
+  -- only export T abstractly.  The single [n]
+  -- in the AvailTC is the type or class itself
+ok_item _ _ = True
 
 check_occs :: RdrNameIE -> ExportOccMap -> AvailInfo -> RnMG ExportOccMap
 check_occs ie occs avail 
@@ -579,9 +561,6 @@ check_occs ie occs avail
                                failWithRn occs (exportClashErr name_occ ie ie')
       where
        name_occ = nameOccName name
-       
-mk_export_fn :: NameSet -> (Name -> Bool)      -- True => exported
-mk_export_fn exported_names = \name ->  name `elemNameSet` exported_names
 \end{code}
 
 %************************************************************************
@@ -591,9 +570,13 @@ mk_export_fn exported_names = \name ->  name `elemNameSet` exported_names
 %************************************************************************
 
 \begin{code}
-badImportItemErr mod ie
-  = sep [ptext SLIT("Module"), quotes (ppr mod), 
+badImportItemErr mod from ie
+  = sep [ptext SLIT("Module"), quotes (ppr mod), source_import,
         ptext SLIT("does not export"), quotes (ppr ie)]
+  where
+    source_import = case from of
+                     ImportByUserSource -> ptext SLIT("(hi-boot interface)")
+                     other              -> empty
 
 dodgyImportWarn mod item = dodgyMsg (ptext SLIT("import")) item
 dodgyExportWarn     item = dodgyMsg (ptext SLIT("export")) item