-%\r
-% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998\r
-%\r
-\section[RnNames]{Extracting imported and top-level names in scope}\r
-\r
-\begin{code}\r
-module RnNames (\r
- getGlobalNames\r
- ) where\r
-\r
-#include "HsVersions.h"\r
-\r
-import CmdLineOpts ( opt_NoImplicitPrelude, opt_WarnDuplicateExports, \r
- opt_SourceUnchanged, opt_WarnUnusedBinds\r
- )\r
-\r
-import HsSyn ( HsModule(..), HsDecl(..), TyClDecl(..),\r
- IE(..), ieName, \r
- ForeignDecl(..), ForKind(..), isDynamic,\r
- FixitySig(..), Sig(..), ImportDecl(..),\r
- collectTopBinders\r
- )\r
-import RdrHsSyn ( RdrNameIE, RdrNameImportDecl,\r
- RdrNameHsModule, RdrNameHsDecl\r
- )\r
-import RnIfaces ( getInterfaceExports, getDeclBinders,\r
- recordSlurp, checkUpToDate\r
- )\r
-import RnEnv\r
-import RnMonad\r
-\r
-import FiniteMap\r
-import PrelMods\r
-import PrelInfo ( main_RDR )\r
-import UniqFM ( lookupUFM )\r
-import Bag ( bagToList )\r
-import Maybes ( maybeToBool )\r
-import Module ( ModuleName, mkThisModule, pprModuleName, WhereFrom(..) )\r
-import NameSet\r
-import Name ( Name, ExportFlag(..), ImportReason(..), Provenance(..),\r
- isLocallyDefined, setNameProvenance,\r
- nameOccName, getSrcLoc, pprProvenance, getNameProvenance\r
- )\r
-import RdrName ( RdrName, rdrNameOcc, mkRdrQual, mkRdrUnqual, isQual )\r
-import SrcLoc ( SrcLoc )\r
-import NameSet ( elemNameSet, emptyNameSet )\r
-import Outputable\r
-import Unique ( getUnique )\r
-import Util ( removeDups, equivClassesByUniq, sortLt )\r
-import List ( partition )\r
-\end{code}\r
-\r
-\r
-\r
-%************************************************************************\r
-%* *\r
-\subsection{Get global names}\r
-%* *\r
-%************************************************************************\r
-\r
-\begin{code}\r
-getGlobalNames :: RdrNameHsModule\r
- -> RnMG (Maybe (ExportEnv, \r
- GlobalRdrEnv,\r
- FixityEnv, -- Fixities for local decls only\r
- NameEnv AvailInfo -- Maps a name to its parent AvailInfo\r
- -- Just for in-scope things only\r
- ))\r
- -- Nothing => no need to recompile\r
-\r
-getGlobalNames (HsModule this_mod _ exports imports decls mod_loc)\r
- = -- These two fix-loops are to get the right\r
- -- provenance information into a Name\r
- fixRn (\ ~(rec_gbl_env, rec_exported_avails, _) ->\r
-\r
- let\r
- rec_unqual_fn :: Name -> Bool -- Is this chap in scope unqualified?\r
- rec_unqual_fn = unQualInScope rec_gbl_env\r
-\r
- rec_exp_fn :: Name -> ExportFlag\r
- rec_exp_fn = mk_export_fn (availsToNameSet rec_exported_avails)\r
- in\r
- setModuleRn this_mod $\r
-\r
- -- PROCESS LOCAL DECLS\r
- -- Do these *first* so that the correct provenance gets\r
- -- into the global name cache.\r
- importsFromLocalDecls this_mod rec_exp_fn decls `thenRn` \ (local_gbl_env, local_mod_avails) ->\r
-\r
- -- PROCESS IMPORT DECLS\r
- -- Do the non {- SOURCE -} ones first, so that we get a helpful\r
- -- warning for {- SOURCE -} ones that are unnecessary\r
- let\r
- (source, ordinary) = partition is_source_import all_imports\r
- is_source_import (ImportDecl _ ImportByUserSource _ _ _ _) = True\r
- is_source_import other = False\r
- in\r
- mapAndUnzipRn (importsFromImportDecl rec_unqual_fn) ordinary `thenRn` \ (imp_gbl_envs1, imp_avails_s1) ->\r
- mapAndUnzipRn (importsFromImportDecl rec_unqual_fn) source `thenRn` \ (imp_gbl_envs2, imp_avails_s2) ->\r
-\r
- -- COMBINE RESULTS\r
- -- We put the local env second, so that a local provenance\r
- -- "wins", even if a module imports itself.\r
- let\r
- gbl_env :: GlobalRdrEnv\r
- imp_gbl_env = foldr plusGlobalRdrEnv emptyRdrEnv (imp_gbl_envs2 ++ imp_gbl_envs1)\r
- gbl_env = imp_gbl_env `plusGlobalRdrEnv` local_gbl_env\r
-\r
- all_avails :: ExportAvails\r
- all_avails = foldr plusExportAvails local_mod_avails (imp_avails_s2 ++ imp_avails_s1)\r
- in\r
-\r
- -- TRY FOR EARLY EXIT\r
- -- We can't go for an early exit before this because we have to check\r
- -- for name clashes. Consider:\r
- --\r
- -- module A where module B where\r
- -- import B h = True\r
- -- f = h\r
- --\r
- -- Suppose I've compiled everything up, and then I add a\r
- -- new definition to module B, that defines "f".\r
- --\r
- -- Then I must detect the name clash in A before going for an early\r
- -- exit. The early-exit code checks what's actually needed from B\r
- -- to compile A, and of course that doesn't include B.f. That's\r
- -- why we wait till after the plusEnv stuff to do the early-exit.\r
- checkEarlyExit this_mod `thenRn` \ up_to_date ->\r
- if up_to_date then\r
- returnRn (gbl_env, junk_exp_fn, Nothing)\r
- else\r
- \r
- -- RECORD BETTER PROVENANCES IN THE CACHE\r
- -- The names in the envirnoment have better provenances (e.g. imported on line x)\r
- -- than the names in the name cache. We update the latter now, so that we\r
- -- we start renaming declarations we'll get the good names\r
- -- The isQual is because the qualified name is always in scope\r
- updateProvenances (concat [names | (rdr_name, names) <- rdrEnvToList imp_gbl_env, \r
- isQual rdr_name]) `thenRn_`\r
-\r
- -- PROCESS EXPORT LISTS\r
- exportsFromAvail this_mod exports all_avails gbl_env `thenRn` \ exported_avails ->\r
-\r
- -- DONE\r
- returnRn (gbl_env, exported_avails, Just all_avails)\r
- ) `thenRn` \ (gbl_env, exported_avails, maybe_stuff) ->\r
-\r
- case maybe_stuff of {\r
- Nothing -> returnRn Nothing ;\r
- Just all_avails ->\r
-\r
- traceRn (text "updateProv" <+> fsep (map ppr (rdrEnvElts gbl_env))) `thenRn_`\r
- \r
- -- DEAL WITH FIXITIES\r
- fixitiesFromLocalDecls gbl_env decls `thenRn` \ local_fixity_env ->\r
- let\r
- -- Export only those fixities that are for names that are\r
- -- (a) defined in this module\r
- -- (b) exported\r
- exported_fixities :: [(Name,Fixity)]\r
- exported_fixities = [(name,fixity) | FixitySig name fixity _ <- nameEnvElts local_fixity_env,\r
- isLocallyDefined name\r
- ]\r
- in\r
- traceRn (text "fixity env" <+> vcat (map ppr (nameEnvElts local_fixity_env))) `thenRn_`\r
-\r
- --- TIDY UP \r
- let\r
- export_env = ExportEnv exported_avails exported_fixities\r
- (_, global_avail_env) = all_avails\r
- in\r
- returnRn (Just (export_env, gbl_env, local_fixity_env, global_avail_env))\r
- }\r
- where\r
- junk_exp_fn = error "RnNames:export_fn"\r
-\r
- all_imports = prel_imports ++ imports\r
-\r
- -- NB: opt_NoImplicitPrelude is slightly different to import Prelude ();\r
- -- because the former doesn't even look at Prelude.hi for instance declarations,\r
- -- whereas the latter does.\r
- prel_imports | this_mod == pRELUDE_Name ||\r
- explicit_prelude_import ||\r
- opt_NoImplicitPrelude\r
- = []\r
-\r
- | otherwise = [ImportDecl pRELUDE_Name\r
- ImportByUser\r
- False {- Not qualified -}\r
- Nothing {- No "as" -}\r
- Nothing {- No import list -}\r
- mod_loc]\r
- \r
- explicit_prelude_import\r
- = not (null [ () | (ImportDecl mod _ _ _ _ _) <- imports, mod == pRELUDE_Name ])\r
-\end{code}\r
- \r
-\begin{code}\r
-checkEarlyExit mod\r
- = checkErrsRn `thenRn` \ no_errs_so_far ->\r
- if not no_errs_so_far then\r
- -- Found errors already, so exit now\r
- returnRn True\r
- else\r
-\r
- traceRn (text "Considering whether compilation is required...") `thenRn_`\r
- if not opt_SourceUnchanged then\r
- -- Source code changed and no errors yet... carry on \r
- traceRn (nest 4 (text "source file changed or recompilation check turned off")) `thenRn_` \r
- returnRn False\r
- else\r
-\r
- -- Unchanged source, and no errors yet; see if usage info\r
- -- up to date, and exit if so\r
- checkUpToDate mod `thenRn` \ up_to_date ->\r
- putDocRn (text "Compilation" <+> \r
- text (if up_to_date then "IS NOT" else "IS") <+>\r
- text "required") `thenRn_`\r
- returnRn up_to_date\r
-\end{code}\r
- \r
-\begin{code}\r
-importsFromImportDecl :: (Name -> Bool) -- OK to omit qualifier\r
- -> RdrNameImportDecl\r
- -> RnMG (GlobalRdrEnv, \r
- ExportAvails) \r
-\r
-importsFromImportDecl is_unqual (ImportDecl imp_mod_name from qual_only as_mod import_spec iloc)\r
- = pushSrcLocRn iloc $\r
- getInterfaceExports imp_mod_name from `thenRn` \ (imp_mod, avails) ->\r
-\r
- if null avails then\r
- -- If there's an error in getInterfaceExports, (e.g. interface\r
- -- file not found) we get lots of spurious errors from 'filterImports'\r
- returnRn (emptyRdrEnv, mkEmptyExportAvails imp_mod_name)\r
- else\r
-\r
- filterImports imp_mod_name import_spec avails `thenRn` \ (filtered_avails, hides, explicits) ->\r
-\r
- -- We 'improve' the provenance by setting\r
- -- (a) the import-reason field, so that the Name says how it came into scope\r
- -- including whether it's explicitly imported\r
- -- (b) the print-unqualified field\r
- -- But don't fiddle with wired-in things or we get in a twist\r
- let\r
- improve_prov name = setNameProvenance name (NonLocalDef (UserImport imp_mod iloc (is_explicit name)) \r
- (is_unqual name))\r
- is_explicit name = name `elemNameSet` explicits\r
- in\r
- qualifyImports imp_mod_name\r
- (not qual_only) -- Maybe want unqualified names\r
- as_mod hides\r
- filtered_avails improve_prov `thenRn` \ (rdr_name_env, mod_avails) ->\r
-\r
- returnRn (rdr_name_env, mod_avails)\r
-\end{code}\r
-\r
-\r
-\begin{code}\r
-importsFromLocalDecls mod_name rec_exp_fn decls\r
- = mapRn (getLocalDeclBinders newLocalName) decls `thenRn` \ avails_s ->\r
-\r
- let\r
- avails = concat avails_s\r
-\r
- all_names :: [Name] -- All the defns; no dups eliminated\r
- all_names = [name | avail <- avails, name <- availNames avail]\r
-\r
- dups :: [[Name]]\r
- dups = filter non_singleton (equivClassesByUniq getUnique all_names)\r
- where\r
- non_singleton (x1:x2:xs) = True\r
- non_singleton other = False\r
- in\r
- -- Check for duplicate definitions\r
- mapRn_ (addErrRn . dupDeclErr) dups `thenRn_` \r
-\r
- -- Record that locally-defined things are available\r
- mapRn_ (recordSlurp Nothing) avails `thenRn_`\r
-\r
- -- Build the environment\r
- qualifyImports mod_name \r
- True -- Want unqualified names\r
- Nothing -- no 'as M'\r
- [] -- Hide nothing\r
- avails\r
- (\n -> n)\r
-\r
- where\r
- newLocalName rdr_name loc = newLocalTopBinder mod (rdrNameOcc rdr_name)\r
- rec_exp_fn loc\r
- mod = mkThisModule mod_name\r
-\r
-getLocalDeclBinders :: (RdrName -> SrcLoc -> RnMG Name) -- New-name function\r
- -> RdrNameHsDecl\r
- -> RnMG Avails\r
-getLocalDeclBinders new_name (ValD binds)\r
- = mapRn do_one (bagToList (collectTopBinders binds))\r
- where\r
- do_one (rdr_name, loc) = new_name rdr_name loc `thenRn` \ name ->\r
- returnRn (Avail name)\r
-\r
- -- foreign declarations\r
-getLocalDeclBinders new_name (ForD (ForeignDecl nm kind _ dyn _ loc))\r
- | binds_haskell_name kind dyn\r
- = new_name nm loc `thenRn` \ name ->\r
- returnRn [Avail name]\r
-\r
- | otherwise\r
- = returnRn []\r
-\r
-getLocalDeclBinders new_name decl\r
- = getDeclBinders new_name decl `thenRn` \ maybe_avail ->\r
- case maybe_avail of\r
- Nothing -> returnRn [] -- Instance decls and suchlike\r
- Just avail -> returnRn [avail]\r
-\r
-binds_haskell_name (FoImport _) _ = True\r
-binds_haskell_name FoLabel _ = True\r
-binds_haskell_name FoExport ext_nm = isDynamic ext_nm\r
-\r
-fixitiesFromLocalDecls :: GlobalRdrEnv -> [RdrNameHsDecl] -> RnMG FixityEnv\r
-fixitiesFromLocalDecls gbl_env decls\r
- = foldlRn getFixities emptyNameEnv decls\r
- where\r
- getFixities :: FixityEnv -> RdrNameHsDecl -> RnMG FixityEnv\r
- getFixities acc (FixD fix)\r
- = fix_decl acc fix\r
-\r
- getFixities acc (TyClD (ClassDecl _ _ _ sigs _ _ _ _ _ _))\r
- = foldlRn fix_decl acc [sig | FixSig sig <- sigs]\r
- -- Get fixities from class decl sigs too.\r
- getFixities acc other_decl\r
- = returnRn acc\r
-\r
- fix_decl acc sig@(FixitySig rdr_name fixity loc)\r
- = -- Check for fixity decl for something not declared\r
- case lookupRdrEnv gbl_env rdr_name of {\r
- Nothing | opt_WarnUnusedBinds \r
- -> pushSrcLocRn loc (addWarnRn (unusedFixityDecl rdr_name fixity)) `thenRn_`\r
- returnRn acc \r
- | otherwise -> returnRn acc ;\r
- \r
- Just (name:_) ->\r
-\r
- -- Check for duplicate fixity decl\r
- case lookupNameEnv acc name of {\r
- Just (FixitySig _ _ loc') -> addErrRn (dupFixityDecl rdr_name loc loc') `thenRn_`\r
- returnRn acc ;\r
-\r
- Nothing -> returnRn (addToNameEnv acc name (FixitySig name fixity loc))\r
- }}\r
-\end{code}\r
-\r
-%************************************************************************\r
-%* *\r
-\subsection{Filtering imports}\r
-%* *\r
-%************************************************************************\r
-\r
-@filterImports@ takes the @ExportEnv@ telling what the imported module makes\r
-available, and filters it through the import spec (if any).\r
-\r
-\begin{code}\r
-filterImports :: ModuleName -- The module being imported\r
- -> Maybe (Bool, [RdrNameIE]) -- Import spec; True => hiding\r
- -> [AvailInfo] -- What's available\r
- -> RnMG ([AvailInfo], -- What's actually imported\r
- [AvailInfo], -- What's to be hidden (the unqualified version, that is)\r
- NameSet) -- What was imported explicitly\r
-\r
- -- Complains if import spec mentions things that the module doesn't export\r
- -- Warns/informs if import spec contains duplicates.\r
-filterImports mod Nothing imports\r
- = returnRn (imports, [], emptyNameSet)\r
-\r
-filterImports mod (Just (want_hiding, import_items)) avails\r
- = mapMaybeRn check_item import_items `thenRn` \ avails_w_explicits ->\r
- let\r
- (item_avails, explicits_s) = unzip avails_w_explicits\r
- explicits = foldl addListToNameSet emptyNameSet explicits_s\r
- in\r
- if want_hiding \r
- then \r
- -- All imported; item_avails to be hidden\r
- returnRn (avails, item_avails, emptyNameSet)\r
- else\r
- -- Just item_avails imported; nothing to be hidden\r
- returnRn (item_avails, [], explicits)\r
- where\r
- import_fm :: FiniteMap OccName AvailInfo\r
- import_fm = listToFM [ (nameOccName name, avail) \r
- | avail <- avails,\r
- name <- availNames avail]\r
- -- Even though availNames returns data constructors too,\r
- -- they won't make any difference because naked entities like T\r
- -- in an import list map to TcOccs, not VarOccs.\r
-\r
- check_item item@(IEModuleContents _)\r
- = addErrRn (badImportItemErr mod item) `thenRn_`\r
- returnRn Nothing\r
-\r
- check_item item\r
- | not (maybeToBool maybe_in_import_avails) ||\r
- not (maybeToBool maybe_filtered_avail)\r
- = addErrRn (badImportItemErr mod item) `thenRn_`\r
- returnRn Nothing\r
-\r
- | dodgy_import = addWarnRn (dodgyImportWarn mod item) `thenRn_`\r
- returnRn (Just (filtered_avail, explicits))\r
-\r
- | otherwise = returnRn (Just (filtered_avail, explicits))\r
- \r
- where\r
- wanted_occ = rdrNameOcc (ieName item)\r
- maybe_in_import_avails = lookupFM import_fm wanted_occ\r
-\r
- Just avail = maybe_in_import_avails\r
- maybe_filtered_avail = filterAvail item avail\r
- Just filtered_avail = maybe_filtered_avail\r
- explicits | dot_dot = [availName filtered_avail]\r
- | otherwise = availNames filtered_avail\r
-\r
- dot_dot = case item of \r
- IEThingAll _ -> True\r
- other -> False\r
-\r
- dodgy_import = case (item, avail) of\r
- (IEThingAll _, AvailTC _ [n]) -> True\r
- -- This occurs when you import T(..), but\r
- -- only export T abstractly. The single [n]\r
- -- in the AvailTC is the type or class itself\r
- \r
- other -> False\r
-\end{code}\r
-\r
-\r
-\r
-%************************************************************************\r
-%* *\r
-\subsection{Qualifiying imports}\r
-%* *\r
-%************************************************************************\r
-\r
-@qualifyImports@ takes the @ExportEnv@ after filtering through the import spec\r
-of an import decl, and deals with producing an @RnEnv@ with the \r
-right qualified names. It also turns the @Names@ in the @ExportEnv@ into\r
-fully fledged @Names@.\r
-\r
-\begin{code}\r
-qualifyImports :: ModuleName -- Imported module\r
- -> Bool -- True <=> want unqualified import\r
- -> Maybe ModuleName -- Optional "as M" part \r
- -> [AvailInfo] -- What's to be hidden\r
- -> Avails -- Whats imported and how\r
- -> (Name -> Name) -- Improves the provenance on imported things\r
- -> RnMG (GlobalRdrEnv, ExportAvails)\r
- -- NB: the Names in ExportAvails don't have the improve-provenance\r
- -- function applied to them\r
- -- We could fix that, but I don't think it matters\r
-\r
-qualifyImports this_mod unqual_imp as_mod hides\r
- avails improve_prov\r
- = \r
- -- Make the name environment. We're talking about a \r
- -- single module here, so there must be no name clashes.\r
- -- In practice there only ever will be if it's the module\r
- -- being compiled.\r
- let\r
- -- Add the things that are available\r
- name_env1 = foldl add_avail emptyRdrEnv avails\r
-\r
- -- Delete things that are hidden\r
- name_env2 = foldl del_avail name_env1 hides\r
-\r
- -- Create the export-availability info\r
- export_avails = mkExportAvails qual_mod unqual_imp name_env2 avails\r
- in\r
- returnRn (name_env2, export_avails)\r
-\r
- where\r
- qual_mod = case as_mod of\r
- Nothing -> this_mod\r
- Just another_name -> another_name\r
-\r
- add_avail :: GlobalRdrEnv -> AvailInfo -> GlobalRdrEnv\r
- add_avail env avail = foldl add_name env (availNames avail)\r
-\r
- add_name env name\r
- | unqual_imp = env2\r
- | otherwise = env1\r
- where\r
- env1 = addOneToGlobalRdrEnv env (mkRdrQual qual_mod occ) better_name\r
- env2 = addOneToGlobalRdrEnv env1 (mkRdrUnqual occ) better_name\r
- occ = nameOccName name\r
- better_name = improve_prov name\r
-\r
- del_avail env avail = foldl delOneFromGlobalRdrEnv env rdr_names\r
- where\r
- rdr_names = map (mkRdrUnqual . nameOccName) (availNames avail)\r
-\end{code}\r
-\r
-\r
-%************************************************************************\r
-%* *\r
-\subsection{Export list processing\r
-%* *\r
-%************************************************************************\r
-\r
-Processing the export list.\r
-\r
-You might think that we should record things that appear in the export list as\r
-``occurrences'' (using addOccurrenceName), but you'd be wrong. We do check (here)\r
-that they are in scope, but there is no need to slurp in their actual declaration\r
-(which is what addOccurrenceName forces). Indeed, doing so would big trouble when\r
-compiling PrelBase, because it re-exports GHC, which includes takeMVar#, whose type\r
-includes ConcBase.StateAndSynchVar#, and so on...\r
-\r
-\begin{code}\r
-type ExportAccum -- The type of the accumulating parameter of\r
- -- the main worker function in exportsFromAvail\r
- = ([ModuleName], -- 'module M's seen so far\r
- ExportOccMap, -- Tracks exported occurrence names\r
- NameEnv AvailInfo) -- The accumulated exported stuff, kept in an env\r
- -- so we can common-up related AvailInfos\r
-\r
-type ExportOccMap = FiniteMap OccName (Name, RdrNameIE)\r
- -- Tracks what a particular exported OccName\r
- -- in an export list refers to, and which item\r
- -- it came from. It's illegal to export two distinct things\r
- -- that have the same occurrence name\r
-\r
-\r
-exportsFromAvail :: ModuleName\r
- -> Maybe [RdrNameIE] -- Export spec\r
- -> ExportAvails\r
- -> GlobalRdrEnv \r
- -> RnMG Avails\r
- -- Complains if two distinct exports have same OccName\r
- -- Warns about identical exports.\r
- -- Complains about exports items not in scope\r
-exportsFromAvail this_mod Nothing export_avails global_name_env\r
- = exportsFromAvail this_mod true_exports export_avails global_name_env\r
- where\r
- true_exports = Just $ if this_mod == mAIN_Name\r
- then [IEVar main_RDR]\r
- -- export Main.main *only* unless otherwise specified,\r
- else [IEModuleContents this_mod]\r
- -- but for all other modules export everything.\r
-\r
-exportsFromAvail this_mod (Just export_items) \r
- (mod_avail_env, entity_avail_env)\r
- global_name_env\r
- = foldlRn exports_from_item\r
- ([], emptyFM, emptyNameEnv) export_items `thenRn` \ (_, _, export_avail_map) ->\r
- let\r
- export_avails :: [AvailInfo]\r
- export_avails = nameEnvElts export_avail_map\r
- in\r
- returnRn export_avails\r
-\r
- where\r
- exports_from_item :: ExportAccum -> RdrNameIE -> RnMG ExportAccum\r
-\r
- exports_from_item acc@(mods, occs, avails) ie@(IEModuleContents mod)\r
- | mod `elem` mods -- Duplicate export of M\r
- = warnCheckRn opt_WarnDuplicateExports\r
- (dupModuleExport mod) `thenRn_`\r
- returnRn acc\r
-\r
- | otherwise\r
- = case lookupFM mod_avail_env mod of\r
- Nothing -> failWithRn acc (modExportErr mod)\r
- Just mod_avails -> foldlRn (check_occs ie) occs mod_avails `thenRn` \ occs' ->\r
- let\r
- avails' = foldl add_avail avails mod_avails\r
- in\r
- returnRn (mod:mods, occs', avails')\r
-\r
- exports_from_item acc@(mods, occs, avails) ie\r
- | not (maybeToBool maybe_in_scope) \r
- = failWithRn acc (unknownNameErr (ieName ie))\r
-\r
- | not (null dup_names)\r
- = addNameClashErrRn rdr_name (name:dup_names) `thenRn_`\r
- returnRn acc\r
-\r
-#ifdef DEBUG\r
- -- I can't see why this should ever happen; if the thing is in scope\r
- -- at all it ought to have some availability\r
- | not (maybeToBool maybe_avail)\r
- = pprTrace "exportsFromAvail: curious Nothing:" (ppr name)\r
- returnRn acc\r
-#endif\r
-\r
- | not enough_avail\r
- = failWithRn acc (exportItemErr ie)\r
-\r
- | otherwise -- Phew! It's OK! Now to check the occurrence stuff!\r
- = check_occs ie occs export_avail `thenRn` \ occs' ->\r
- returnRn (mods, occs', add_avail avails export_avail)\r
-\r
- where\r
- rdr_name = ieName ie\r
- maybe_in_scope = lookupFM global_name_env rdr_name\r
- Just (name:dup_names) = maybe_in_scope\r
- maybe_avail = lookupUFM entity_avail_env name\r
- Just avail = maybe_avail\r
- maybe_export_avail = filterAvail ie avail\r
- enough_avail = maybeToBool maybe_export_avail\r
- Just export_avail = maybe_export_avail\r
-\r
-add_avail avails avail = addToNameEnv_C plusAvail avails (availName avail) avail\r
-\r
-check_occs :: RdrNameIE -> ExportOccMap -> AvailInfo -> RnMG ExportOccMap\r
-check_occs ie occs avail \r
- = foldlRn check occs (availNames avail)\r
- where\r
- check occs name\r
- = case lookupFM occs name_occ of\r
- Nothing -> returnRn (addToFM occs name_occ (name, ie))\r
- Just (name', ie') \r
- | name == name' -> -- Duplicate export\r
- warnCheckRn opt_WarnDuplicateExports\r
- (dupExportWarn name_occ ie ie') `thenRn_`\r
- returnRn occs\r
-\r
- | otherwise -> -- Same occ name but different names: an error\r
- failWithRn occs (exportClashErr name_occ ie ie')\r
- where\r
- name_occ = nameOccName name\r
- \r
-mk_export_fn :: NameSet -> (Name -> ExportFlag)\r
-mk_export_fn exported_names\r
- = \name -> if name `elemNameSet` exported_names\r
- then Exported\r
- else NotExported\r
-\end{code}\r
-\r
-%************************************************************************\r
-%* *\r
-\subsection{Errors}\r
-%* *\r
-%************************************************************************\r
-\r
-\begin{code}\r
-badImportItemErr mod ie\r
- = sep [ptext SLIT("Module"), quotes (pprModuleName mod), \r
- ptext SLIT("does not export"), quotes (ppr ie)]\r
-\r
-dodgyImportWarn mod (IEThingAll tc)\r
- = sep [ptext SLIT("Module") <+> quotes (pprModuleName mod) <+> ptext SLIT("exports") <+> quotes (ppr tc), \r
- ptext SLIT("with no constructors/class operations;"),\r
- ptext SLIT("yet it is imported with a (..)")]\r
-\r
-modExportErr mod\r
- = hsep [ ptext SLIT("Unknown module in export list: module"), quotes (pprModuleName mod)]\r
-\r
-exportItemErr export_item\r
- = sep [ ptext SLIT("Bad export item"), quotes (ppr export_item)]\r
-\r
-exportClashErr occ_name ie1 ie2\r
- = hsep [ptext SLIT("The export items"), quotes (ppr ie1), ptext SLIT("and"), quotes (ppr ie2),\r
- ptext SLIT("create conflicting exports for"), quotes (ppr occ_name)]\r
-\r
-dupDeclErr (n:ns)\r
- = vcat [ptext SLIT("Multiple declarations of") <+> quotes (ppr n),\r
- nest 4 (vcat (map pp sorted_ns))]\r
- where\r
- sorted_ns = sortLt occ'ed_before (n:ns)\r
-\r
- occ'ed_before a b = LT == compare (getSrcLoc a) (getSrcLoc b)\r
-\r
- pp n = pprProvenance (getNameProvenance n)\r
-\r
-dupExportWarn occ_name ie1 ie2\r
- = hsep [quotes (ppr occ_name), \r
- ptext SLIT("is exported by"), quotes (ppr ie1),\r
- ptext SLIT("and"), quotes (ppr ie2)]\r
-\r
-dupModuleExport mod\r
- = hsep [ptext SLIT("Duplicate"),\r
- quotes (ptext SLIT("Module") <+> pprModuleName mod), \r
- ptext SLIT("in export list")]\r
-\r
-unusedFixityDecl rdr_name fixity\r
- = hsep [ptext SLIT("Unused fixity declaration for"), quotes (ppr rdr_name)]\r
-\r
-dupFixityDecl rdr_name loc1 loc2\r
- = vcat [ptext SLIT("Multiple fixity declarations for") <+> quotes (ppr rdr_name),\r
- ptext SLIT("at ") <+> ppr loc1,\r
- ptext SLIT("and") <+> ppr loc2]\r
-\r
-\end{code}\r
+%
+% (c) The GRASP/AQUA Project, Glasgow University, 1992-1998
+%
+\section[RnNames]{Extracting imported and top-level names in scope}
+
+\begin{code}
+module RnNames (
+ getGlobalNames
+ ) where
+
+#include "HsVersions.h"
+
+import CmdLineOpts ( opt_NoImplicitPrelude, opt_WarnDuplicateExports,
+ opt_SourceUnchanged, opt_WarnUnusedBinds
+ )
+
+import HsSyn ( HsModule(..), HsDecl(..), TyClDecl(..),
+ IE(..), ieName,
+ ForeignDecl(..), ForKind(..), isDynamicExtName,
+ FixitySig(..), Sig(..), ImportDecl(..),
+ collectTopBinders
+ )
+import RdrHsSyn ( RdrNameIE, RdrNameImportDecl,
+ RdrNameHsModule, RdrNameHsDecl
+ )
+import RnIfaces ( getInterfaceExports, getDeclBinders, getDeclSysBinders,
+ recordLocalSlurps, checkModUsage, findAndReadIface, outOfDate
+ )
+import RnEnv
+import RnMonad
+
+import FiniteMap
+import PrelInfo ( pRELUDE_Name, mAIN_Name, main_RDR )
+import UniqFM ( lookupUFM )
+import Bag ( bagToList )
+import Module ( ModuleName, mkThisModule, pprModuleName, WhereFrom(..) )
+import NameSet
+import Name ( Name, ExportFlag(..), ImportReason(..), Provenance(..),
+ isLocallyDefined, setNameProvenance,
+ nameOccName, getSrcLoc, pprProvenance, getNameProvenance,
+ nameEnvElts
+ )
+import RdrName ( RdrName, rdrNameOcc, setRdrNameOcc, mkRdrQual, mkRdrUnqual, isQual, isUnqual )
+import OccName ( setOccNameSpace, dataName )
+import NameSet ( elemNameSet, emptyNameSet )
+import Outputable
+import Maybes ( maybeToBool, catMaybes, mapMaybe )
+import UniqFM ( emptyUFM, listToUFM, plusUFM_C )
+import Util ( removeDups, equivClassesByUniq, sortLt )
+import List ( partition )
+\end{code}
+
+
+
+%************************************************************************
+%* *
+\subsection{Get global names}
+%* *
+%************************************************************************
+
+\begin{code}
+getGlobalNames :: RdrNameHsModule
+ -> RnMG (Maybe (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
+ Maybe ParsedIface -- The old interface file, if any
+ ))
+ -- Nothing => no need to recompile
+
+getGlobalNames (HsModule this_mod _ exports imports decls _ mod_loc)
+ = -- These two fix-loops are to get the right
+ -- provenance information into a Name
+ fixRn ( \ ~(Just (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 -> ExportFlag
+ rec_exp_fn = mk_export_fn (availsToNameSet rec_export_avails)
+ in
+
+ -- 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) ->
+
+ -- PROCESS IMPORT DECLS
+ -- Do the non {- SOURCE -} ones first, so that we get a helpful
+ -- warning for {- SOURCE -} ones that are unnecessary
+ let
+ (source, ordinary) = partition is_source_import all_imports
+ is_source_import (ImportDecl _ ImportByUserSource _ _ _ _) = True
+ is_source_import other = False
+ 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) ->
+
+ -- COMBINE RESULTS
+ -- We put the local env second, so that a local provenance
+ -- "wins", even if a module imports itself.
+ let
+ gbl_env :: GlobalRdrEnv
+ imp_gbl_env = foldr plusGlobalRdrEnv emptyRdrEnv (imp_gbl_envs2 ++ imp_gbl_envs1)
+ gbl_env = imp_gbl_env `plusGlobalRdrEnv` local_gbl_env
+
+ all_avails :: ExportAvails
+ all_avails = foldr plusExportAvails local_mod_avails (imp_avails_s2 ++ imp_avails_s1)
+ (_, global_avail_env) = all_avails
+ in
+
+ -- TRY FOR EARLY EXIT
+ -- We can't go for an early exit before this because we have to check
+ -- for name clashes. Consider:
+ --
+ -- module A where module B where
+ -- import B h = True
+ -- f = h
+ --
+ -- Suppose I've compiled everything up, and then I add a
+ -- new definition to module B, that defines "f".
+ --
+ -- Then I must detect the name clash in A before going for an early
+ -- exit. The early-exit code checks what's actually needed from B
+ -- to compile A, and of course that doesn't include B.f. That's
+ -- why we wait till after the plusEnv stuff to do the early-exit.
+
+ -- Check For eacly exit
+ checkErrsRn `thenRn` \ no_errs_so_far ->
+ if not no_errs_so_far then
+ -- Found errors already, so exit now
+ returnRn Nothing
+ else
+ checkEarlyExit this_mod `thenRn` \ (up_to_date, old_iface) ->
+ if up_to_date then
+ -- Interface files are sufficiently unchanged
+ putDocRn (text "Compilation IS NOT required") `thenRn_`
+ returnRn Nothing
+ else
+
+ -- RECORD BETTER PROVENANCES IN THE CACHE
+ -- The names in the envirnoment have better provenances (e.g. imported on line x)
+ -- than the names in the name cache. We update the latter now, so that we
+ -- we start renaming declarations we'll get the good names
+ -- The isQual is because the qualified name is always in scope
+ updateProvenances (concat [names | (rdr_name, names) <- rdrEnvToList gbl_env,
+ isQual rdr_name]) `thenRn_`
+
+ -- PROCESS EXPORT LISTS
+ exportsFromAvail this_mod exports all_avails gbl_env `thenRn` \ export_avails ->
+
+
+ -- ALL DONE
+ returnRn (Just (gbl_env, local_gbl_env, export_avails, global_avail_env, old_iface))
+ )
+ where
+ all_imports = prel_imports ++ imports
+
+ -- NB: opt_NoImplicitPrelude is slightly different to import Prelude ();
+ -- because the former doesn't even look at Prelude.hi for instance declarations,
+ -- whereas the latter does.
+ prel_imports | this_mod == pRELUDE_Name ||
+ explicit_prelude_import ||
+ opt_NoImplicitPrelude
+ = []
+
+ | otherwise = [ImportDecl pRELUDE_Name
+ ImportByUser
+ False {- Not qualified -}
+ Nothing {- No "as" -}
+ Nothing {- No import list -}
+ mod_loc]
+
+ explicit_prelude_import
+ = not (null [ () | (ImportDecl mod _ _ _ _ _) <- imports, mod == pRELUDE_Name ])
+\end{code}
+
+\begin{code}
+checkEarlyExit mod_name
+ = traceRn (text "Considering whether compilation is required...") `thenRn_`
+
+ -- Read the old interface file, if any, for the module being compiled
+ findAndReadIface doc_str mod_name False {- Not hi-boot -} `thenRn` \ maybe_iface ->
+
+ -- CHECK WHETHER WE HAVE IT ALREADY
+ case maybe_iface of
+ Left err -> -- Old interface file not found, so we'd better bail out
+ traceRn (vcat [ptext SLIT("No old interface file for") <+> pprModuleName mod_name,
+ err]) `thenRn_`
+ returnRn (outOfDate, Nothing)
+
+ Right iface
+ | not opt_SourceUnchanged
+ -> -- Source code changed
+ traceRn (nest 4 (text "source file changed or recompilation check turned off")) `thenRn_`
+ returnRn (False, Just iface)
+
+ | otherwise
+ -> -- Source code unchanged and no errors yet... carry on
+ checkModUsage (pi_usages iface) `thenRn` \ up_to_date ->
+ returnRn (up_to_date, Just iface)
+ where
+ -- Only look in current directory, with suffix .hi
+ doc_str = sep [ptext SLIT("need usage info from"), pprModuleName mod_name]
+\end{code}
+
+\begin{code}
+importsFromImportDecl :: (Name -> Bool) -- OK to omit qualifier
+ -> RdrNameImportDecl
+ -> RnMG (GlobalRdrEnv,
+ ExportAvails)
+
+importsFromImportDecl is_unqual (ImportDecl imp_mod_name from qual_only as_mod import_spec iloc)
+ = pushSrcLocRn iloc $
+ getInterfaceExports imp_mod_name from `thenRn` \ (imp_mod, avails) ->
+
+ if null avails then
+ -- If there's an error in getInterfaceExports, (e.g. interface
+ -- file not found) we get lots of spurious errors from 'filterImports'
+ returnRn (emptyRdrEnv, mkEmptyExportAvails imp_mod_name)
+ else
+
+ filterImports imp_mod_name import_spec avails `thenRn` \ (filtered_avails, hides, explicits) ->
+
+ qualifyImports imp_mod_name
+ (not qual_only) -- Maybe want unqualified names
+ as_mod hides
+ (improveAvails imp_mod iloc explicits
+ is_unqual filtered_avails)
+
+
+improveAvails imp_mod iloc explicits is_unqual avails
+ -- We 'improve' the provenance by setting
+ -- (a) the import-reason field, so that the Name says how it came into scope
+ -- including whether it's explicitly imported
+ -- (b) the print-unqualified field
+ = map improve_avail avails
+ where
+ improve_avail (Avail n) = Avail (improve n)
+ improve_avail (AvailTC n ns) = AvailTC (improve n) (map improve ns)
+
+ improve name = setNameProvenance name
+ (NonLocalDef (UserImport imp_mod iloc (is_explicit name))
+ (is_unqual name))
+ is_explicit name = name `elemNameSet` explicits
+\end{code}
+
+
+\begin{code}
+importsFromLocalDecls mod_name rec_exp_fn decls
+ = mapRn (getLocalDeclBinders mod rec_exp_fn) decls `thenRn` \ avails_s ->
+
+ let
+ avails = concat avails_s
+
+ all_names :: [Name] -- All the defns; no dups eliminated
+ all_names = [name | avail <- avails, name <- availNames avail]
+
+ dups :: [[Name]]
+ (_, dups) = removeDups compare all_names
+ in
+ -- Check for duplicate definitions
+ mapRn_ (addErrRn . dupDeclErr) dups `thenRn_`
+
+ -- Record that locally-defined things are available
+ recordLocalSlurps avails `thenRn_`
+
+ -- Build the environment
+ qualifyImports mod_name
+ True -- Want unqualified names
+ Nothing -- no 'as M'
+ [] -- Hide nothing
+ avails
+
+ where
+ mod = mkThisModule mod_name
+
+getLocalDeclBinders :: Module -> (Name -> ExportFlag)
+ -> RdrNameHsDecl -> RnMG Avails
+getLocalDeclBinders mod rec_exp_fn (ValD binds)
+ = mapRn do_one (bagToList (collectTopBinders binds))
+ where
+ do_one (rdr_name, loc) = newLocalName mod rec_exp_fn rdr_name loc `thenRn` \ name ->
+ returnRn (Avail name)
+
+getLocalDeclBinders mod rec_exp_fn decl
+ = getDeclBinders (newLocalName mod rec_exp_fn) decl `thenRn` \ maybe_avail ->
+ case maybe_avail of
+ Nothing -> returnRn [] -- Instance decls and suchlike
+ Just avail -> returnRn [avail]
+
+newLocalName mod rec_exp_fn rdr_name loc
+ = check_unqual rdr_name loc `thenRn_`
+ newTopBinder mod (rdrNameOcc rdr_name) `thenRn` \ name ->
+ returnRn (setNameProvenance name (LocalDef loc (rec_exp_fn name)))
+ where
+ -- There should never be a qualified name in a binding position (except in instance decls)
+ -- The parser doesn't check this because the same parser parses instance decls
+ check_unqual rdr_name loc
+ | isUnqual rdr_name = returnRn ()
+ | otherwise = qualNameErr (text "the binding for" <+> quotes (ppr rdr_name))
+ (rdr_name,loc)
+\end{code}
+
+
+%************************************************************************
+%* *
+\subsection{Filtering imports}
+%* *
+%************************************************************************
+
+@filterImports@ takes the @ExportEnv@ telling what the imported module makes
+available, and filters it through the import spec (if any).
+
+\begin{code}
+filterImports :: ModuleName -- The module being imported
+ -> Maybe (Bool, [RdrNameIE]) -- Import spec; True => hiding
+ -> [AvailInfo] -- What's available
+ -> RnMG ([AvailInfo], -- What's actually imported
+ [AvailInfo], -- What's to be hidden
+ -- (the unqualified version, that is)
+ -- (We need to return both the above sets, because
+ -- the qualified version is never hidden; so we can't
+ -- implement hiding by reducing what's imported.)
+ NameSet) -- What was imported explicitly
+
+ -- Complains if import spec mentions things that the module doesn't export
+ -- Warns/informs if import spec contains duplicates.
+filterImports mod Nothing imports
+ = returnRn (imports, [], emptyNameSet)
+
+filterImports mod (Just (want_hiding, import_items)) avails
+ = flatMapRn get_item import_items `thenRn` \ avails_w_explicits ->
+ let
+ (item_avails, explicits_s) = unzip avails_w_explicits
+ explicits = foldl addListToNameSet emptyNameSet explicits_s
+ in
+ if want_hiding
+ then
+ -- All imported; item_avails to be hidden
+ returnRn (avails, item_avails, emptyNameSet)
+ else
+ -- Just item_avails imported; nothing to be hidden
+ returnRn (item_avails, [], explicits)
+ where
+ import_fm :: FiniteMap OccName AvailInfo
+ import_fm = listToFM [ (nameOccName name, avail)
+ | avail <- avails,
+ name <- availNames avail]
+ -- Even though availNames returns data constructors too,
+ -- 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_`
+ returnRn []
+
+ get_item item@(IEModuleContents _) = bale_out item
+
+ get_item item@(IEThingAll _)
+ = case check_item item of
+ Nothing -> bale_out item
+ Just avail@(AvailTC _ [n]) -> -- This occurs when you import T(..), but
+ -- only export T abstractly. The single [n]
+ -- in the AvailTC is the type or class itself
+ addWarnRn (dodgyImportWarn mod item) `thenRn_`
+ returnRn [(avail, [availName avail])]
+ Just avail -> returnRn [(avail, [availName avail])]
+
+ get_item item@(IEThingAbs n)
+ | want_hiding -- hiding( C )
+ -- Here the 'C' can be a data constructor *or* a type/class
+ = case catMaybes [check_item item, check_item (IEThingAbs data_n)] of
+ [] -> bale_out item
+ avails -> returnRn [(a, []) | a <- avails]
+ -- The 'explicits' list is irrelevant when hiding
+ where
+ data_n = setRdrNameOcc n (setOccNameSpace (rdrNameOcc n) dataName)
+
+ get_item item
+ = case check_item item of
+ Nothing -> bale_out item
+ Just avail -> returnRn [(avail, availNames avail)]
+
+ check_item item
+ | not (maybeToBool maybe_in_import_avails) ||
+ not (maybeToBool maybe_filtered_avail)
+ = Nothing
+
+ | otherwise
+ = Just filtered_avail
+
+ where
+ wanted_occ = rdrNameOcc (ieName item)
+ maybe_in_import_avails = lookupFM import_fm wanted_occ
+
+ Just avail = maybe_in_import_avails
+ maybe_filtered_avail = filterAvail item avail
+ Just filtered_avail = maybe_filtered_avail
+\end{code}
+
+
+
+%************************************************************************
+%* *
+\subsection{Qualifiying imports}
+%* *
+%************************************************************************
+
+@qualifyImports@ takes the @ExportEnv@ after filtering through the import spec
+of an import decl, and deals with producing an @RnEnv@ with the
+right qualified names. It also turns the @Names@ in the @ExportEnv@ into
+fully fledged @Names@.
+
+\begin{code}
+qualifyImports :: ModuleName -- Imported module
+ -> Bool -- True <=> want unqualified import
+ -> Maybe ModuleName -- Optional "as M" part
+ -> [AvailInfo] -- What's to be hidden
+ -> Avails -- Whats imported and how
+ -> RnMG (GlobalRdrEnv, ExportAvails)
+
+qualifyImports this_mod unqual_imp as_mod hides avails
+ =
+ -- Make the name environment. We're talking about a
+ -- single module here, so there must be no name clashes.
+ -- In practice there only ever will be if it's the module
+ -- being compiled.
+ let
+ -- Add the things that are available
+ name_env1 = foldl add_avail emptyRdrEnv avails
+
+ -- Delete things that are hidden
+ name_env2 = foldl del_avail name_env1 hides
+
+ -- Create the export-availability info
+ export_avails = mkExportAvails qual_mod unqual_imp name_env2 avails
+ in
+ returnRn (name_env2, export_avails)
+
+ where
+ qual_mod = case as_mod of
+ Nothing -> this_mod
+ Just another_name -> another_name
+
+ add_avail :: GlobalRdrEnv -> AvailInfo -> GlobalRdrEnv
+ add_avail env avail = foldl add_name env (availNames avail)
+
+ add_name env name
+ | unqual_imp = env2
+ | otherwise = env1
+ where
+ env1 = addOneToGlobalRdrEnv env (mkRdrQual qual_mod occ) name
+ env2 = addOneToGlobalRdrEnv env1 (mkRdrUnqual occ) name
+ occ = nameOccName name
+
+ del_avail env avail = foldl delOneFromGlobalRdrEnv env rdr_names
+ where
+ rdr_names = map (mkRdrUnqual . nameOccName) (availNames avail)
+
+
+mkEmptyExportAvails :: ModuleName -> ExportAvails
+mkEmptyExportAvails mod_name = (unitFM mod_name [], emptyUFM)
+
+mkExportAvails :: ModuleName -> Bool -> GlobalRdrEnv -> [AvailInfo] -> ExportAvails
+mkExportAvails mod_name unqual_imp name_env avails
+ = (mod_avail_env, entity_avail_env)
+ where
+ mod_avail_env = unitFM mod_name unqual_avails
+
+ -- unqual_avails is the Avails that are visible in *unqualfied* form
+ -- (1.4 Report, Section 5.1.1)
+ -- For example, in
+ -- import T hiding( f )
+ -- we delete f from avails
+
+ unqual_avails | not unqual_imp = [] -- Short cut when no unqualified imports
+ | otherwise = mapMaybe prune avails
+
+ prune (Avail n) | unqual_in_scope n = Just (Avail n)
+ prune (Avail n) | otherwise = Nothing
+ prune (AvailTC n ns) | null uqs = Nothing
+ | otherwise = Just (AvailTC n uqs)
+ where
+ uqs = filter unqual_in_scope ns
+
+ unqual_in_scope n = unQualInScope name_env n
+
+ entity_avail_env = listToUFM [ (name,avail) | avail <- avails,
+ name <- availNames avail]
+
+plusExportAvails :: ExportAvails -> ExportAvails -> ExportAvails
+plusExportAvails (m1, e1) (m2, e2)
+ = (plusFM_C (++) m1 m2, plusAvailEnv e1 e2)
+ -- ToDo: wasteful: we do this once for each constructor!
+\end{code}
+
+
+%************************************************************************
+%* *
+\subsection{Export list processing}
+%* *
+%************************************************************************
+
+Processing the export list.
+
+You might think that we should record things that appear in the export list
+as ``occurrences'' (using @addOccurrenceName@), but you'd be wrong.
+We do check (here) that they are in scope,
+but there is no need to slurp in their actual declaration
+(which is what @addOccurrenceName@ forces).
+
+Indeed, doing so would big trouble when
+compiling @PrelBase@, because it re-exports @GHC@, which includes @takeMVar#@,
+whose type includes @ConcBase.StateAndSynchVar#@, and so on...
+
+\begin{code}
+type ExportAccum -- The type of the accumulating parameter of
+ -- the main worker function in exportsFromAvail
+ = ([ModuleName], -- 'module M's seen so far
+ ExportOccMap, -- Tracks exported occurrence names
+ AvailEnv) -- The accumulated exported stuff, kept in an env
+ -- so we can common-up related AvailInfos
+
+type ExportOccMap = FiniteMap OccName (Name, RdrNameIE)
+ -- Tracks what a particular exported OccName
+ -- in an export list refers to, and which item
+ -- it came from. It's illegal to export two distinct things
+ -- that have the same occurrence name
+
+
+exportsFromAvail :: ModuleName
+ -> Maybe [RdrNameIE] -- Export spec
+ -> ExportAvails
+ -> GlobalRdrEnv
+ -> RnMG Avails
+ -- Complains if two distinct exports have same OccName
+ -- Warns about identical exports.
+ -- Complains about exports items not in scope
+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]
+ -- export Main.main *only* unless otherwise specified,
+ else [IEModuleContents this_mod]
+ -- but for all other modules export everything.
+
+exportsFromAvail this_mod (Just export_items)
+ (mod_avail_env, entity_avail_env)
+ global_name_env
+ = foldlRn exports_from_item
+ ([], emptyFM, emptyAvailEnv) export_items `thenRn` \ (_, _, export_avail_map) ->
+ let
+ export_avails :: [AvailInfo]
+ export_avails = nameEnvElts export_avail_map
+ in
+ returnRn export_avails
+
+ where
+ exports_from_item :: ExportAccum -> RdrNameIE -> RnMG ExportAccum
+
+ exports_from_item acc@(mods, occs, avails) ie@(IEModuleContents mod)
+ | mod `elem` mods -- Duplicate export of M
+ = warnCheckRn opt_WarnDuplicateExports
+ (dupModuleExport mod) `thenRn_`
+ returnRn acc
+
+ | otherwise
+ = case lookupFM mod_avail_env mod of
+ Nothing -> failWithRn acc (modExportErr mod)
+ Just mod_avails -> foldlRn (check_occs ie) occs mod_avails
+ `thenRn` \ occs' ->
+ let
+ avails' = foldl addAvail avails mod_avails
+ in
+ returnRn (mod:mods, occs', avails')
+
+ exports_from_item acc@(mods, occs, avails) ie
+ | not (maybeToBool maybe_in_scope)
+ = failWithRn acc (unknownNameErr (ieName ie))
+
+ | not (null dup_names)
+ = addNameClashErrRn rdr_name (name:dup_names) `thenRn_`
+ returnRn acc
+
+#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
+
+ | not enough_avail
+ = failWithRn acc (exportItemErr ie)
+
+ | otherwise -- 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: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
+
+check_occs :: RdrNameIE -> ExportOccMap -> AvailInfo -> RnMG ExportOccMap
+check_occs ie occs avail
+ = foldlRn check occs (availNames avail)
+ where
+ check occs name
+ = case lookupFM occs name_occ of
+ Nothing -> returnRn (addToFM occs name_occ (name, ie))
+ Just (name', ie')
+ | name == name' -> -- Duplicate export
+ warnCheckRn opt_WarnDuplicateExports
+ (dupExportWarn name_occ ie ie')
+ `thenRn_` returnRn occs
+
+ | otherwise -> -- Same occ name but different names: an error
+ failWithRn occs (exportClashErr name_occ ie ie')
+ where
+ name_occ = nameOccName name
+
+mk_export_fn :: NameSet -> (Name -> ExportFlag)
+mk_export_fn exported_names
+ = \name -> if name `elemNameSet` exported_names
+ then Exported
+ else NotExported
+\end{code}
+
+%************************************************************************
+%* *
+\subsection{Errors}
+%* *
+%************************************************************************
+
+\begin{code}
+badImportItemErr mod ie
+ = sep [ptext SLIT("Module"), quotes (pprModuleName mod),
+ ptext SLIT("does not export"), quotes (ppr ie)]
+
+dodgyImportWarn mod item = dodgyMsg (ptext SLIT("import")) item
+dodgyExportWarn item = dodgyMsg (ptext SLIT("export")) item
+
+dodgyMsg kind item@(IEThingAll tc)
+ = sep [ ptext SLIT("The") <+> kind <+> ptext SLIT("item") <+> quotes (ppr item),
+ ptext SLIT("suggests that") <+> quotes (ppr tc) <+> ptext SLIT("has constructor or class methods"),
+ ptext SLIT("but it has none; it is a type synonym or abstract type or class") ]
+
+modExportErr mod
+ = hsep [ ptext SLIT("Unknown module in export list: module"), quotes (pprModuleName mod)]
+
+exportItemErr export_item
+ = sep [ ptext SLIT("The export item") <+> quotes (ppr export_item),
+ ptext SLIT("attempts to export constructors or class methods that are not visible here") ]
+
+exportClashErr occ_name ie1 ie2
+ = hsep [ptext SLIT("The export items"), quotes (ppr ie1)
+ ,ptext SLIT("and"), quotes (ppr ie2)
+ ,ptext SLIT("create conflicting exports for"), quotes (ppr occ_name)]
+
+dupDeclErr (n:ns)
+ = vcat [ptext SLIT("Multiple declarations of") <+> quotes (ppr n),
+ nest 4 (vcat (map pp sorted_ns))]
+ where
+ sorted_ns = sortLt occ'ed_before (n:ns)
+
+ occ'ed_before a b = LT == compare (getSrcLoc a) (getSrcLoc b)
+
+ pp n = pprProvenance (getNameProvenance n)
+
+dupExportWarn occ_name ie1 ie2
+ = hsep [quotes (ppr occ_name),
+ ptext SLIT("is exported by"), quotes (ppr ie1),
+ ptext SLIT("and"), quotes (ppr ie2)]
+
+dupModuleExport mod
+ = hsep [ptext SLIT("Duplicate"),
+ quotes (ptext SLIT("Module") <+> pprModuleName mod),
+ ptext SLIT("in export list")]
+\end{code}