#include "HsVersions.h"
module RnNames (
- getGlobalNames,
- GlobalNameInfo(..)
+ getGlobalNames
) where
-import PreludeGlaST ( returnPrimIO, thenPrimIO, MutableVar(..) )
-
-import Ubiq
-
-import HsSyn
-import RdrHsSyn
-import RnHsSyn
-
-import ParseIface ( ParsedIface )
+IMP_Ubiq()
+
+import CmdLineOpts ( opt_SourceUnchanged )
+import HsSyn ( HsModule(..), HsDecl(..), FixityDecl(..), Fixity, Fake, InPat, IE(..), HsTyVar,
+ TyDecl, ClassDecl, InstDecl, DefaultDecl, ImportDecl(..), HsBinds, IfaceSig
+ )
+import HsBinds ( collectTopBinders )
+import HsImpExp ( ieName )
+import RdrHsSyn ( RdrNameHsDecl(..), RdrName(..), RdrNameIE(..), SYN_IE(RdrNameImportDecl),
+ SYN_IE(RdrNameHsModule), SYN_IE(RdrNameFixityDecl),
+ rdrNameOcc
+ )
+import RnHsSyn ( RenamedHsModule(..), RenamedFixityDecl(..) )
+import RnIfaces ( getInterfaceExports, getDeclBinders, checkUpToDate )
+import RnEnv
import RnMonad
-import RnIfaces ( IfaceCache(..), cachedIface )
-import RnUtils ( RnEnv(..), emptyRnEnv, extendGlobalRnEnv, qualNameErr, dupNamesErr )
-
-import Bag ( emptyBag, unitBag, unionBags, unionManyBags, mapBag, listToBag, bagToList )
-import ErrUtils ( Error(..), Warning(..), addShortErrLocLine )
-import FiniteMap ( fmToList )
-import Name ( RdrName(..), Name, isQual, mkTopLevName,
- mkImportedName, nameExportFlag,
- getLocalName, getSrcLoc, pprNonOp
- )
-import PrelInfo ( BuiltinNames(..), BuiltinKeys(..) )
-import PrelMods ( fromPrelude )
+import FiniteMap
+import PrelMods
+import UniqFM ( UniqFM, emptyUFM, addListToUFM_C, lookupUFM )
+import Bag ( Bag, bagToList )
+import Maybes ( maybeToBool, expectJust )
+import Name
import Pretty
-import SrcLoc ( SrcLoc )
-import UniqSupply ( splitUniqSupply )
-import Util ( equivClasses, panic )
+import PprStyle ( PprStyle(..) )
+import Util ( panic, pprTrace )
\end{code}
-\begin{code}
-type GlobalNameInfo = (BuiltinNames,
- BuiltinKeys,
- Name -> ExportFlag,
- Name -> [RdrName])
-
-type RnM_Info s r = RnMonad GlobalNameInfo s r
-
-getGlobalNames ::
- IfaceCache
- -> GlobalNameInfo
- -> UniqSupply
- -> RdrNameHsModule
- -> PrimIO (RnEnv,
- [Module], -- directly imported modules
- Bag (Module,(RnName,ExportFlag)), -- unqualified imports from module
- Bag RenamedFixityDecl, -- imported fixity decls
- Bag Error,
- Bag Warning)
-
-getGlobalNames iface_var info us
- (HsModule mod _ _ imports _ ty_decls _ cls_decls _ _ _ binds _ _)
- = case initRn True mod emptyRnEnv us1
- (setExtraRn info $
- getSourceNames ty_decls cls_decls binds)
- of { ((src_vals, src_tcs), src_errs, src_warns) ->
-
- getImportedNames iface_var info us2 imports `thenPrimIO`
- \ (imp_vals, imp_tcs, imp_mods, unqual_imps, imp_fixes, imp_errs, imp_warns) ->
- let
- unqual_vals = mapBag (\rn -> (Unqual (getLocalName rn), rn)) src_vals
- unqual_tcs = mapBag (\rn -> (Unqual (getLocalName rn), rn)) src_tcs
+%************************************************************************
+%* *
+\subsection{Get global names}
+%* *
+%************************************************************************
+
+\begin{code}
+getGlobalNames :: RdrNameHsModule
+ -> RnMG (Maybe (ExportEnv, RnEnv, [AvailInfo]))
+ -- Nothing <=> no need to recompile
+
+getGlobalNames m@(HsModule this_mod _ exports imports _ _ mod_loc)
+ = fixRn (\ ~(rec_exp_fn, _) ->
+
+ -- PROCESS LOCAL DECLS
+ -- Do these *first* so that the correct provenance gets
+ -- into the global name cache.
+ importsFromLocalDecls rec_exp_fn m `thenRn` \ (local_rn_env, local_mod_avails) ->
+
+ -- PROCESS IMPORT DECLS
+ mapAndUnzipRn importsFromImportDecl all_imports
+ `thenRn` \ (imp_rn_envs, imp_avails_s) ->
+
+ -- CHECK FOR EARLY EXIT
+ checkEarlyExit this_mod `thenRn` \ early_exit ->
+ if early_exit then
+ returnRn (junk_exp_fn, Nothing)
+ else
+
+ -- COMBINE RESULTS
+ -- We put the local env first, so that a local provenance
+ -- "wins", even if a module imports itself.
+ foldlRn plusRnEnv emptyRnEnv imp_rn_envs `thenRn` \ imp_rn_env ->
+ plusRnEnv local_rn_env imp_rn_env `thenRn` \ rn_env ->
+ let
+ all_avails :: ModuleAvails
+ all_avails = foldr plusModuleAvails local_mod_avails imp_avails_s
+ local_avails = expectJust "getGlobalNames" (lookupModuleAvails local_mod_avails this_mod)
+ in
+
+ -- PROCESS EXPORT LISTS
+ exportsFromAvail this_mod exports all_avails rn_env
+ `thenRn` \ (export_fn, export_env) ->
+
+ returnRn (export_fn, Just (export_env, rn_env, local_avails))
+ ) `thenRn` \ (_, result) ->
+ returnRn result
+ where
+ junk_exp_fn = error "RnNames:export_fn"
- all_vals = bagToList (unqual_vals `unionBags` imp_vals)
- all_tcs = bagToList (unqual_tcs `unionBags` imp_tcs)
+ all_imports = prel_imports ++ imports
- (all_env, dups) = extendGlobalRnEnv emptyRnEnv all_vals all_tcs
+ prel_imports | this_mod == pRELUDE ||
+ explicit_prelude_import = []
- dup_errs = map dup_err (equivClasses cmp_rdr (bagToList dups))
- cmp_rdr (rdr1,_,_) (rdr2,_,_) = cmp rdr1 rdr2
- dup_err ((rdr,rn,rn'):rest) = globalDupNamesErr rdr (rn:rn': [rn|(_,_,rn)<-rest])
+ | otherwise = [ImportDecl pRELUDE
+ False {- Not qualified -}
+ Nothing {- No "as" -}
+ Nothing {- No import list -}
+ mod_loc]
+
+ explicit_prelude_import
+ = not (null [ () | (ImportDecl mod qual _ _ _) <- imports, mod == pRELUDE ])
+\end{code}
+
+\begin{code}
+checkEarlyExit mod
+ = if not opt_SourceUnchanged then
+ -- Source code changed; look no further
+ returnRn False
+ else
+ -- Unchanged source; look further
+ -- We check for
+ -- (a) errors so far. These can arise if a module imports
+ -- something that's no longer exported by the imported module
+ -- (b) usage information up to date
+ checkErrsRn `thenRn` \ no_errs_so_far ->
+ checkUpToDate mod `thenRn` \ up_to_date ->
+ returnRn (no_errs_so_far && up_to_date)
+\end{code}
+
- all_errs = src_errs `unionBags` imp_errs `unionBags` listToBag dup_errs
- all_warns = src_warns `unionBags` imp_warns
+\begin{code}
+importsFromImportDecl :: RdrNameImportDecl
+ -> RnMG (RnEnv, ModuleAvails)
+
+ -- Check for "import M ()", and then don't even look at M.
+ -- This makes sense, and is actually rather useful for the Prelude.
+importsFromImportDecl (ImportDecl mod qual as_mod (Just (False,[])) loc)
+ = returnRn (emptyRnEnv, emptyModuleAvails)
+
+importsFromImportDecl (ImportDecl mod qual as_mod import_spec loc)
+ = pushSrcLocRn loc $
+ getInterfaceExports mod `thenRn` \ (avails, fixities) ->
+ filterImports mod import_spec avails `thenRn` \ filtered_avails ->
+ let
+ filtered_avails' = [ Avail (set_name_prov n) (map set_name_prov ns)
+ | Avail n ns <- filtered_avails
+ ]
+ fixities' = [ (occ,fixity,provenance) | (occ,fixity) <- fixities ]
in
- returnPrimIO (all_env, bagToList imp_mods, unqual_imps, imp_fixes, all_errs, all_warns)
- }
+ qualifyImports mod qual as_mod (ExportEnv filtered_avails' fixities')
where
- (us1, us2) = splitUniqSupply us
+ set_name_prov name = setNameProvenance name provenance
+ provenance = Imported mod loc
\end{code}
-*********************************************************
-* *
-\subsection{Top-level source names}
-* *
-*********************************************************
\begin{code}
-getSourceNames ::
- [RdrNameTyDecl]
- -> [RdrNameClassDecl]
- -> RdrNameHsBinds
- -> RnM_Info s (Bag RnName, -- values
- Bag RnName) -- tycons/classes
-
-getSourceNames ty_decls cls_decls binds
- = mapAndUnzipRn getTyDeclNames ty_decls `thenRn` \ (tycon_s, constrs_s) ->
- mapAndUnzipRn getClassNames cls_decls `thenRn` \ (cls_s, cls_ops_s) ->
- getTopBindsNames binds `thenRn` \ bind_names ->
- returnRn (unionManyBags constrs_s `unionBags`
- unionManyBags cls_ops_s `unionBags` bind_names,
- listToBag tycon_s `unionBags` listToBag cls_s)
-
-
-getTyDeclNames :: RdrNameTyDecl
- -> RnM_Info s (RnName, Bag RnName) -- tycon and constrs
-
-getTyDeclNames (TyData _ tycon _ condecls _ _ src_loc)
- = newGlobalName src_loc Nothing tycon `thenRn` \ tycon_name ->
- mapRn (getConDeclName (Just (nameExportFlag tycon_name)))
- condecls `thenRn` \ con_names ->
- returnRn (RnData tycon_name con_names,
- listToBag (map (\ n -> RnConstr n tycon_name) con_names))
-
-getTyDeclNames (TyNew _ tycon _ condecls _ _ src_loc)
- = newGlobalName src_loc Nothing tycon `thenRn` \ tycon_name ->
- mapRn (getConDeclName (Just (nameExportFlag tycon_name)))
- condecls `thenRn` \ con_names ->
- returnRn (RnData tycon_name con_names,
- listToBag (map (\ n -> RnConstr n tycon_name) con_names))
-
-getTyDeclNames (TySynonym tycon _ _ src_loc)
- = newGlobalName src_loc Nothing tycon `thenRn` \ tycon_name ->
- returnRn (RnSyn tycon_name, emptyBag)
-
-getConDeclName exp (ConDecl con _ src_loc)
- = newGlobalName src_loc exp con
-getConDeclName exp (ConOpDecl _ op _ src_loc)
- = newGlobalName src_loc exp op
-getConDeclName exp (NewConDecl con _ src_loc)
- = newGlobalName src_loc exp con
-getConDeclName exp (RecConDecl con fields src_loc)
- = panic "getConDeclName:RecConDecl"
- newGlobalName src_loc exp con
-
-
-getClassNames :: RdrNameClassDecl
- -> RnM_Info s (RnName, Bag RnName) -- class and class ops
-
-getClassNames (ClassDecl _ cname _ sigs _ _ src_loc)
- = newGlobalName src_loc Nothing cname `thenRn` \ class_name ->
- getClassOpNames (Just (nameExportFlag class_name))
- sigs `thenRn` \ op_names ->
- returnRn (RnClass class_name op_names,
- listToBag (map (\ n -> RnClassOp n class_name) op_names))
-
-getClassOpNames exp []
- = returnRn []
-getClassOpNames exp (ClassOpSig op _ _ src_loc : sigs)
- = newGlobalName src_loc exp op `thenRn` \ op_name ->
- getClassOpNames exp sigs `thenRn` \ op_names ->
- returnRn (op_name : op_names)
-getClassOpNames exp (_ : sigs)
- = getClassOpNames exp sigs
+importsFromLocalDecls rec_exp_fn (HsModule mod _ _ _ fix_decls decls _)
+ = foldlRn getLocalDeclBinders [] decls `thenRn` \ avails ->
+ mapRn fixityFromFixDecl fix_decls `thenRn` \ fixities ->
+ qualifyImports mod
+ False -- Not qualified
+ Nothing -- No "as M" part
+ (ExportEnv avails fixities)
+ where
+ newLocalName rdr_name loc
+ = newLocallyDefinedGlobalName mod (rdrNameOcc rdr_name) rec_exp_fn loc
+
+ getLocalDeclBinders avails (ValD binds)
+ = mapRn do_one (bagToList (collectTopBinders binds)) `thenRn` \ val_avails ->
+ returnRn (val_avails ++ avails)
+
+ getLocalDeclBinders avails decl
+ = getDeclBinders newLocalName decl `thenRn` \ avail ->
+ returnRn (avail : avails)
+
+ do_one (rdr_name, loc)
+ = newLocalName rdr_name loc `thenRn` \ name ->
+ returnRn (Avail name [])
\end{code}
-*********************************************************
-* *
-\subsection{Bindings}
-* *
-*********************************************************
+%************************************************************************
+%* *
+\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}
-getTopBindsNames :: RdrNameHsBinds
- -> RnM_Info s (Bag RnName)
-
-getTopBindsNames binds = doBinds binds
-
-doBinds EmptyBinds = returnRn emptyBag
-doBinds (SingleBind bind) = doBind bind
-doBinds (BindWith bind sigs) = doBind bind
-doBinds (ThenBinds binds1 binds2)
- = andRn unionBags (doBinds binds1) (doBinds binds2)
-
-doBind EmptyBind = returnRn emptyBag
-doBind (NonRecBind mbind) = doMBinds mbind
-doBind (RecBind mbind) = doMBinds mbind
-
-doMBinds EmptyMonoBinds = returnRn emptyBag
-doMBinds (PatMonoBind pat grhss_and_binds locn) = doPat locn pat
-doMBinds (FunMonoBind p_name _ _ locn) = doName locn p_name
-doMBinds (AndMonoBinds mbinds1 mbinds2)
- = andRn unionBags (doMBinds mbinds1) (doMBinds mbinds2)
-
-doPats locn pats
- = mapRn (doPat locn) pats `thenRn` \ pats_s ->
- returnRn (unionManyBags pats_s)
-
-doPat locn WildPatIn = returnRn emptyBag
-doPat locn (LitPatIn _) = returnRn emptyBag
-doPat locn (LazyPatIn pat) = doPat locn pat
-doPat locn (VarPatIn var) = doName locn var
-doPat locn (NegPatIn pat) = doPat locn pat
-doPat locn (ParPatIn pat) = doPat locn pat
-doPat locn (ListPatIn pats) = doPats locn pats
-doPat locn (TuplePatIn pats) = doPats locn pats
-doPat locn (ConPatIn name pats) = doPats locn pats
-doPat locn (ConOpPatIn p1 op p2)
- = andRn unionBags (doPat locn p1) (doPat locn p2)
-doPat locn (AsPatIn as_name pat)
- = andRn unionBags (doName locn as_name) (doPat locn pat)
-doPat locn (RecPatIn name fields)
- = mapRn (doField locn) fields `thenRn` \ fields_s ->
- returnRn (unionManyBags fields_s)
-
-doField locn (_, pat, _) = doPat locn pat
-
-doName locn rdr
- = newGlobalName locn Nothing rdr `thenRn` \ name ->
- returnRn (unitBag (RnName name))
+filterImports :: Module
+ -> Maybe (Bool, [RdrNameIE]) -- Import spec; True => hidin
+ -> [AvailInfo] -- What's available
+ -> RnMG [AvailInfo] -- What's actually imported
+ -- Complains if import spec mentions things the
+ -- module doesn't export
+
+filterImports mod Nothing imports
+ = returnRn imports
+
+filterImports mod (Just (want_hiding, import_items)) avails
+ = -- Check that each import item mentions things that are actually available
+ mapRn check_import_item import_items `thenRn_`
+
+ -- Return filtered environment; no need to filter fixities
+ returnRn (map new_avail avails)
+
+ where
+ import_fm :: FiniteMap OccName RdrNameIE
+ import_fm = listToFM [(ieOcc ie, ie) | ie <- import_items]
+
+ avail_fm :: FiniteMap OccName AvailInfo
+ avail_fm = listToFM [(nameOccName name, avail) | avail@(Avail name ns) <- avails]
+
+ new_avail NotAvailable = NotAvailable
+ new_avail avail@(Avail name _)
+ | not in_import_items && want_hiding = avail
+ | not in_import_items && not want_hiding = NotAvailable
+ | in_import_items && want_hiding = NotAvailable
+ | in_import_items && not want_hiding = filtered_avail
+ where
+ maybe_import_item = lookupFM import_fm (nameOccName name)
+ in_import_items = maybeToBool maybe_import_item
+ Just import_item = maybe_import_item
+ filtered_avail = filterAvail import_item avail
+
+ check_import_item :: RdrNameIE -> RnMG ()
+ check_import_item item
+ = checkRn (maybeToBool maybe_matching_avail && sub_names_ok item avail)
+ (badImportItemErr mod item)
+ where
+ item_name = ieOcc item
+ maybe_matching_avail = lookupFM avail_fm item_name
+ Just avail = maybe_matching_avail
+
+ sub_names_ok (IEVar _) _ = True
+ sub_names_ok (IEThingAbs _) _ = True
+ sub_names_ok (IEThingAll _) _ = True
+ sub_names_ok (IEThingWith _ wanted) (Avail _ has) = all ((`elem` has_list) . rdrNameOcc) wanted
+ where
+ has_list = map nameOccName has
+ sub_names_ok other1 other2 = False
\end{code}
-*********************************************************
-* *
-\subsection{Creating a new global name}
-* *
-*********************************************************
+
+
+%************************************************************************
+%* *
+\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 qaulified names. It also turns the @Names@ in the @ExportEnv@ into
+fully fledged @Names@.
\begin{code}
-newGlobalName :: SrcLoc -> Maybe ExportFlag
- -> RdrName -> RnM_Info s Name
-
-newGlobalName locn maybe_exp rdr
- = getExtraRn `thenRn` \ (_,_,exp_fn,occ_fn) ->
- getModuleRn `thenRn` \ mod ->
- getSourceRn `thenRn` \ source ->
- rnGetUnique `thenRn` \ u ->
- let
- src_unqual = getLocalName rdr
+qualifyImports :: Module -- Improrted module
+ -> Bool -- True <=> qualified import
+ -> Maybe Module -- Optional "as M" part
+ -> ExportEnv -- What's imported
+ -> RnMG (RnEnv, ModuleAvails)
+
+qualifyImports this_mod qual as_mod (ExportEnv avails fixities)
+ = -- Make the qualified-name environments, checking of course for clashes
+ foldlRn add_name emptyNameEnv avails `thenRn` \ name_env ->
+ foldlRn (add_fixity name_env) emptyFixityEnv fixities `thenRn` \ fixity_env ->
+
+ -- Deal with the "qualified" part; if not qualifies then add unqualfied bindings
+ if qual then
+ returnRn (RnEnv name_env fixity_env, mod_avail_env)
+ else
+ returnRn (RnEnv (unQualify name_env) (unQualify fixity_env), mod_avail_env)
+
+ where
+ mod_avail_env = unitFM this_mod avails
+
+ add_name name_env NotAvailable = returnRn name_env
+ add_name name_env (Avail n ns) = foldlRn add_one name_env (n : ns)
+
+ add_one :: NameEnv -> Name -> RnMG NameEnv
+ add_one env name = addOneToNameEnvRn env (Qual this_mod occ_name) name
+ where
+ occ_name = nameOccName name
+
+ add_fixity name_env fixity_env (occ_name, fixity, provenance)
+ | maybeToBool (lookupFM name_env qual_name) -- The name is imported
+ = addOneToFixityEnvRn fixity_env qual_name (fixity,provenance)
+ | otherwise -- It ain't imported
+ = returnRn fixity_env
+ where
+ qual_name = Qual this_mod occ_name
+\end{code}
- src_orig = if fromPrelude mod
- then (Unqual src_unqual)
- else (Qual mod src_unqual)
+unQualify adds an Unqual binding for every existing Qual binding.
- exp = case maybe_exp of
- Just exp -> exp
- Nothing -> exp_fn n
+\begin{code}
+unQualify :: FiniteMap RdrName elt -> FiniteMap RdrName elt
+unQualify fm = addListToFM fm [(Unqual occ, elt) | (Qual _ occ, elt) <- fmToList fm]
+\end{code}
- n = if source then
- mkTopLevName u src_orig locn exp (occ_fn n)
- else
- mkImportedName u rdr locn exp (occ_fn n)
- in
- addErrIfRn (source && isQual rdr)
- (qualNameErr "name in definition" (rdr, locn)) `thenRn_`
- returnRn n
+%************************************************************************
+%* *
+\subsection{Local declarations}
+%* *
+%************************************************************************
+
+
+\begin{code}
+fixityFromFixDecl :: RdrNameFixityDecl -> RnMG (OccName, Fixity, Provenance)
+
+fixityFromFixDecl (FixityDecl rdr_name fixity loc)
+ = returnRn (rdrNameOcc rdr_name, fixity, LocalDef (panic "export-flag") loc)
\end{code}
-*********************************************************
-* *
-\subsection{Imported names}
-* *
-*********************************************************
+
+%************************************************************************
+%* *
+\subsection{Export list processing
+%* *
+%************************************************************************
+
+The @AvailEnv@ type is just used internally in @exportsFromAvail@.
+When exporting we need to combine the availabilities for a particular
+exported thing, and we also need to check for name clashes -- that
+is: two exported things must have different @OccNames@.
\begin{code}
-getImportedNames ::
- IfaceCache
- -> GlobalNameInfo -- builtin and knot name info
- -> UniqSupply
- -> [RdrNameImportDecl] -- import declarations
- -> PrimIO (Bag (RdrName,RnName), -- imported values in scope
- Bag (RdrName,RnName), -- imported tycons/classes in scope
- Bag Module, -- directly imported modules
- Bag (Module,(RnName,ExportFlag)), -- unqualified imports from module
- Bag RenamedFixityDecl, -- fixity info for imported names
- Bag Error,
- Bag Warning)
-
-getImportedNames iface_var info us imports
- = returnPrimIO (builtin_vals, builtin_tcs, emptyBag, emptyBag, emptyBag, emptyBag, emptyBag)
- where
- -- For now jsut add the builtin names ...
- (b_names,_,_,_) = info
- builtin_vals = listToBag [(Unqual s, rn) | (s,rn) <- fmToList b_names, not (isRnTyCon rn)]
- builtin_tcs = listToBag [(Unqual s, rn) | (s,rn) <- fmToList b_names, isRnTyCon rn]
+type AvailEnv = FiniteMap OccName (RdrNameIE, AvailInfo)
+ -- The FM maps each OccName to the RdrNameIE that gave rise to it,
+ -- for error reporting, as well as to its AvailInfo
+
+emptyAvailEnv = emptyFM
+
+unitAvailEnv :: RdrNameIE -> AvailInfo -> AvailEnv
+unitAvailEnv ie NotAvailable
+ = emptyFM
+unitAvailEnv ie avail@(Avail n ns)
+ = unitFM (nameOccName n) (ie,avail)
+
+plusAvailEnv a1 a2
+ = mapRn (addErrRn.availClashErr) (conflictsFM bad_avail a1 a2) `thenRn_`
+ returnRn (plusFM_C plus_avail a1 a2)
+
+listToAvailEnv :: RdrNameIE -> [AvailInfo] -> RnM s d AvailEnv
+listToAvailEnv ie items
+ = foldlRn plusAvailEnv emptyAvailEnv (map (unitAvailEnv ie) items)
+
+bad_avail (ie1,Avail n1 _) (ie2,Avail n2 _) = n1 /= n2 -- Same OccName, different Name
+plus_avail (ie1,a1) (ie2,a2) = (ie1, a1 `plusAvail` a2)
\end{code}
\begin{code}
-globalDupNamesErr rdr rns sty
- = ppHang (ppBesides [pprNonOp sty rdr, ppStr " multiply defined:"])
- 4 (ppAboves (map pp_def rns))
+exportsFromAvail :: Module
+ -> Maybe [RdrNameIE] -- Export spec
+ -> ModuleAvails
+ -> RnEnv
+ -> RnMG (Name -> ExportFlag, ExportEnv)
+ -- Complains if two distinct exports have same OccName
+ -- Complains about exports items not in scope
+exportsFromAvail this_mod Nothing all_avails rn_env
+ = exportsFromAvail this_mod (Just [IEModuleContents this_mod]) all_avails rn_env
+
+exportsFromAvail this_mod (Just export_items) all_avails (RnEnv name_env fixity_env)
+ = mapRn exports_from_item export_items `thenRn` \ avail_envs ->
+ foldlRn plusAvailEnv emptyAvailEnv avail_envs `thenRn` \ export_avail_env ->
+ let
+ export_avails = map snd (eltsFM export_avail_env)
+ export_fixities = mk_exported_fixities (availsToNameSet export_avails)
+ export_fn = mk_export_fn export_avails
+ in
+ returnRn (export_fn, ExportEnv export_avails export_fixities)
+
+ where
+ full_avail_env :: UniqFM AvailInfo
+ full_avail_env = addListToUFM_C plusAvail emptyUFM
+ [(name,avail) | avail@(Avail name _) <- concat (eltsFM all_avails)]
+ -- NB: full_avail_env won't contain bindings for data constructors and class ops,
+ -- which is right and proper; attempts to export them on their own will provoke an error
+
+ exports_from_item :: RdrNameIE -> RnMG AvailEnv
+ exports_from_item ie@(IEModuleContents mod)
+ = case lookupFM all_avails mod of
+ Nothing -> failWithRn emptyAvailEnv (modExportErr mod)
+ Just avails -> addOccurrenceNames Compulsory [n | Avail n _ <- avails] `thenRn_`
+ listToAvailEnv ie avails
+
+ exports_from_item ie
+ | not (maybeToBool maybe_in_scope)
+ = failWithRn emptyAvailEnv (unknownNameErr (ieName ie))
+
+#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 PprDebug name)
+ returnRn emptyAvailEnv
+#endif
+
+ | not enough_avail
+ = failWithRn emptyAvailEnv (exportItemErr ie export_avail)
+
+ | otherwise -- Phew! It's OK!
+ = addOccurrenceName Compulsory name `thenRn_`
+ returnRn (unitAvailEnv ie export_avail)
+ where
+ maybe_in_scope = lookupNameEnv name_env (ieName ie)
+ Just name = maybe_in_scope
+ maybe_avail = lookupUFM full_avail_env name
+ Just avail = maybe_avail
+ export_avail = filterAvail ie avail
+ enough_avail = case export_avail of {NotAvailable -> False; other -> True}
+
+ -- We export a fixity iff we export a thing with the same (qualified) RdrName
+ mk_exported_fixities :: NameSet -> [(OccName, Fixity, Provenance)]
+ mk_exported_fixities exports
+ = [ (rdrNameOcc rdr_name, fixity, prov)
+ | (rdr_name, (fixity, prov)) <- fmToList fixity_env,
+ export_fixity name_env exports rdr_name
+ ]
+
+mk_export_fn :: [AvailInfo] -> (Name -> ExportFlag)
+mk_export_fn avails
+ = \name -> if name `elemNameSet` exported_names
+ then Exported
+ else NotExported
where
- pp_def rn = addShortErrLocLine (getSrcLoc rn) (\ sty -> ppr sty rn) sty
+ exported_names :: NameSet
+ exported_names = availsToNameSet avails
+
+export_fixity :: NameEnv -> NameSet -> RdrName -> Bool
+export_fixity name_env exports (Unqual _)
+ = False -- The qualified fixity is always there as well
+export_fixity name_env exports rdr_name@(Qual _ occ)
+ = case lookupFM name_env rdr_name of
+ Just fixity_name -> fixity_name `elemNameSet` exports
+ -- Check whether the exported thing is
+ -- the one to which the fixity attaches
+ other -> False -- Not even in scope
+\end{code}
+
+
+%************************************************************************
+%* *
+\subsection{Errors}
+%* *
+%************************************************************************
- -- ToDo: print import src locs for imported names
+\begin{code}
+ieOcc ie = rdrNameOcc (ieName ie)
+
+badImportItemErr mod ie sty
+ = ppSep [ppStr "Module", pprModule sty mod, ppStr "does not export", ppr sty ie]
+
+modExportErr mod sty
+ = ppCat [ ppStr "Unknown module in export list: module", ppPStr mod]
+
+exportItemErr export_item NotAvailable sty
+ = ppSep [ ppStr "Export item not in scope:", ppr sty export_item ]
+
+exportItemErr export_item avail sty
+ = ppHang (ppStr "Export item not fully in scope:")
+ 4 (ppAboves [ppCat [ppStr "Wanted: ", ppr sty export_item],
+ ppCat [ppStr "Available: ", ppr sty (ieOcc export_item), pprAvail sty avail]])
+
+availClashErr (occ_name, ((ie1,avail1), (ie2,avail2))) sty
+ = ppHang (ppCat [ppStr "Conflicting exports for local name: ", ppr sty occ_name])
+ 4 (ppAboves [ppr sty ie1, ppr sty ie2])
\end{code}
+