[project @ 1996-07-25 20:43:49 by partain]
[ghc-hetmet.git] / ghc / compiler / rename / RnSource.lhs
index 3831ec0..d650c01 100644 (file)
@@ -10,6 +10,7 @@ module RnSource ( rnSource, rnTyDecl, rnClassDecl, rnInstDecl, rnPolyType ) wher
 
 IMP_Ubiq()
 IMPORT_DELOOPER(RnLoop)                -- *check* the RnPass/RnExpr/RnBinds loop-breaking
+IMPORT_1_3(List(partition))
 
 import HsSyn
 import HsPragmas
@@ -17,26 +18,28 @@ import RdrHsSyn
 import RnHsSyn
 import RnMonad
 import RnBinds         ( rnTopBinds, rnMethodBinds )
-import RnUtils         ( lookupGlobalRnEnv, lubExportFlag )
+import RnUtils         ( getLocalsFromRnEnv, lookupGlobalRnEnv, lubExportFlag )
 
 import Bag             ( emptyBag, unitBag, consBag, unionManyBags, unionBags, listToBag, bagToList )
 import Class           ( derivableClassKeys )
-import CmdLineOpts     ( opt_CompilingPrelude )
+import CmdLineOpts     ( opt_CompilingGhcInternals )
 import ErrUtils                ( addErrLoc, addShortErrLocLine, addShortWarnLocLine )
 import FiniteMap       ( emptyFM, lookupFM, addListToFM_C )
+import Id              ( isDataCon, GenId{-instance NamedThing-} )
 import ListSetOps      ( unionLists, minusList )
 import Maybes          ( maybeToBool, catMaybes )
 import Name            ( isLocallyDefined, isLexVarId, getLocalName, ExportFlag(..), 
                          nameImportFlag, RdrName, pprNonSym, Name )
-import Outputable      -- ToDo:rm
-import PprStyle        -- ToDo:rm 
+import Outputable      ( Outputable(..){-instances-} )
+--import PprStyle      -- ToDo:rm 
 import Pretty
 import SrcLoc          ( SrcLoc )
+import TyCon           ( tyConDataCons, TyCon{-instance NamedThing-} )
 import Unique          ( Unique )
 import UniqFM          ( emptyUFM, addListToUFM_C, listToUFM, plusUFM, lookupUFM, eltsUFM )
-import UniqSet         ( UniqSet(..) )
+import UniqSet         ( SYN_IE(UniqSet) )
 import Util            ( isIn, isn'tIn, thenCmp, sortLt, removeDups, mapAndUnzip3, cmpPString,
-                         assertPanic, pprTrace{-ToDo:rm-} )
+                         panic, assertPanic{- , pprTrace ToDo:rm-} )
 \end{code}
 
 rnSource `renames' the source module and export list.
@@ -60,6 +63,8 @@ rnSource :: [Module]                  -- imported modules
         -> RdrNameHsModule
         -> RnM s (RenamedHsModule,
                   Name -> ExportFlag,          -- export info
+                  ([(Name, ExportFlag)],       -- export module X stuff
+                   [(Name, ExportFlag)]),
                   Bag (RnName, RdrName))       -- occurrence info
 
 rnSource imp_mods unqual_imps imp_fixes
@@ -70,7 +75,7 @@ rnSource imp_mods unqual_imps imp_fixes
 
   = pushSrcLocRn src_loc $
 
-    rnExports (mod:imp_mods) unqual_imps exports       `thenRn` \ exported_fn ->
+    rnExports (mod:imp_mods) unqual_imps exports       `thenRn` \ (exported_fn, module_dotdots) ->
     rnFixes fixes                                      `thenRn` \ src_fixes ->
     let
        all_fixes     = src_fixes ++ bagToList imp_fixes
@@ -96,7 +101,7 @@ rnSource imp_mods unqual_imps imp_fixes
                new_ty_decls new_specdata_sigs new_class_decls
                new_inst_decls new_specinst_sigs new_defaults
                new_binds [] src_loc,
-             exported_fn,
+             exported_fn, module_dotdots,
              occ_info
             )
   where
@@ -115,13 +120,20 @@ rnSource imp_mods unqual_imps imp_fixes
 rnExports :: [Module]
          -> Bag (Module,RnName)
          -> Maybe [RdrNameIE]
-         -> RnM s (Name -> ExportFlag)
+         -> RnM s (Name -> ExportFlag,    -- main export-flag fun
+                   ([(Name,ExportFlag)],  -- info about "module X" exports
+                    [(Name,ExportFlag)])
+                  )
 
 rnExports mods unqual_imps Nothing
-  = returnRn (\n -> if isLocallyDefined n then ExportAll else NotExported)
+  = returnRn (\n -> if isLocallyDefined n then ExportAll else NotExported
+            , ([], [])
+            )
 
 rnExports mods unqual_imps (Just exps)
-  = mapAndUnzipRn (rnIE mods) exps `thenRn` \ (mod_maybes, exp_bags) ->
+  = getModuleRn                           `thenRn` \ this_mod ->
+    getRnEnv                      `thenRn` \ rn_env ->
+    mapAndUnzipRn (rnIE mods) exps `thenRn` \ (mod_maybes, exp_bags) ->
     let 
        (tc_bags, val_bags) = unzip exp_bags
        tc_names  = bagToList (unionManyBags tc_bags)
@@ -134,34 +146,50 @@ rnExports mods unqual_imps (Just exps)
        cmp_fst (x,_) (y,_) = x `cmp` y
 
        (uniq_mods, dup_mods) = removeDups cmpPString exp_mods
+       (expmods_this, expmods_imps) = partition (== this_mod) uniq_mods
 
-       -- Get names for exported modules
+       -- Get names for "module This_Mod" export
+       (this_tcs, this_vals)
+         = if null expmods_this 
+           then ([], [])
+           else getLocalsFromRnEnv rn_env
 
+       -- Get names for exported imported modules
        (mod_tcs, mod_vals, empty_mods)
-         = case mapAndUnzip3 get_mod_names uniq_mods of
+         = case mapAndUnzip3 get_mod_names expmods_imps of
              (tcs, vals, emptys) -> (concat tcs, concat vals, catMaybes emptys)
                
        (unqual_tcs, unqual_vals) = partition (isRnTyConOrClass.snd) (bagToList unqual_imps)
 
         get_mod_names mod
-         = (tcs, vals, empty_mod)
+         = --pprTrace "get_mod_names" (ppAboves [ppPStr mod, interpp'SP PprDebug (map fst tcs), interpp'SP PprDebug (map fst vals)]) $
+           (tcs, vals, empty_mod)
           where
             tcs  = [(getName rn, nameImportFlag (getName rn))
                   | (mod',rn) <- unqual_tcs, mod == mod']
             vals = [(getName rn, nameImportFlag (getName rn))
-                  | (mod',rn) <- unqual_vals, mod == mod']
+                  | (mod',rn) <- unqual_vals, mod == mod', fun_looking rn]
            empty_mod = if null tcs && null vals
                        then Just mod
                        else Nothing
                                                            
+           -- fun_looking: must avoid class ops and data constructors
+           -- and record fieldnames
+           fun_looking (RnName    _) = True
+           fun_looking (WiredInId i) = not (isDataCon i)
+           fun_looking _             = False
+
        -- Build finite map of exported names to export flag
        tc_map0  = addListToUFM_C lub_expflag emptyUFM (map pair_fst tc_names)
-       tc_map   = addListToUFM_C lub_expflag tc_map0  (map pair_fst mod_tcs)
+       tc_map1  = addListToUFM_C lub_expflag tc_map0  (map pair_fst mod_tcs)
+       tc_map   = addListToUFM_C lub_expflag tc_map1  (map (pair_fst.exp_all) this_tcs)
        
         val_map0 = addListToUFM_C lub_expflag emptyUFM (map pair_fst val_names)
-        val_map  = addListToUFM_C lub_expflag val_map0 (map pair_fst mod_vals)
+        val_map1 = addListToUFM_C lub_expflag val_map0 (map pair_fst mod_vals)
+        val_map  = addListToUFM_C lub_expflag val_map1 (map (pair_fst.exp_all) this_vals)
 
-       pair_fst p@(f,_) = (f,p)
+       pair_fst pr@(n,_) = (n,pr)
+       exp_all rn = (getName rn, ExportAll)
        lub_expflag (n, flag1) (_, flag2) = (n, lubExportFlag flag1 flag2)
 
        -- Check for exporting of duplicate local names
@@ -174,8 +202,8 @@ rnExports mods unqual_imps (Just exps)
        -- Build export flag function
        final_exp_map = plusUFM tc_map val_map
        exp_fn n = case lookupUFM final_exp_map n of
-                    Nothing       -> NotExported
-                    Just (_,flag) -> flag
+                     Nothing       -> NotExported
+                     Just (_,flag) -> flag
     in
     getSrcLocRn                                                        `thenRn` \ src_loc ->
     mapRn (addWarnRn . dupNameExportWarn  src_loc) dup_tc_names        `thenRn_`
@@ -184,28 +212,43 @@ rnExports mods unqual_imps (Just exps)
     mapRn (addWarnRn . emptyModExportWarn src_loc) empty_mods          `thenRn_`
     mapRn (addErrRn  . dupLocalsExportErr src_loc) dup_tc_locals       `thenRn_`
     mapRn (addErrRn  . dupLocalsExportErr src_loc) dup_val_locals      `thenRn_`
-    returnRn exp_fn
+    returnRn (exp_fn, (mod_vals, mod_tcs))
+
+------------------------------------
+-- rename an "IE" in the export list
 
+rnIE ::        [Module]    -- this module and all the (directly?) imported modules
+     -> RdrNameIE
+     -> RnM s (
+           Maybe Module,               -- Just m => a "module X" export item
+           (Bag (Name, ExportFlag),    -- Exported tycons/classes
+            Bag (Name, ExportFlag)))   -- Exported values
 
 rnIE mods (IEVar name)
   = lookupValue name   `thenRn` \ rn ->
     checkIEVar rn      `thenRn` \ exps ->
     returnRn (Nothing, exps)
   where
-    checkIEVar (RnName n)         = returnRn (emptyBag, unitBag (n,ExportAll))
+    checkIEVar (RnName    n)      = returnRn (emptyBag, unitBag (n,ExportAll))
+    checkIEVar (WiredInId i)     = returnRn (emptyBag, unitBag (getName i, ExportAll))
     checkIEVar rn@(RnClassOp _ _) = getSrcLocRn `thenRn` \ src_loc ->
                                    failButContinueRn (emptyBag, emptyBag) (classOpExportErr rn src_loc)
-    checkIEVar rn                = returnRn (emptyBag, emptyBag)
+    checkIEVar rn@(RnField _ _)          = getSrcLocRn `thenRn` \ src_loc ->
+                                   failButContinueRn (emptyBag, emptyBag) (fieldExportErr rn src_loc)
+    checkIEVar rn                = --pprTrace "rnIE:IEVar:panic? ToDo?:" (ppr PprDebug rn) $
+                                   returnRn (emptyBag, emptyBag)
 
 rnIE mods (IEThingAbs name)
   = lookupTyConOrClass name    `thenRn` \ rn ->
     checkIEAbs rn              `thenRn` \ exps ->
     returnRn (Nothing, exps)
   where
-    checkIEAbs (RnSyn n)      = returnRn (unitBag (n,ExportAbs), emptyBag)
-    checkIEAbs (RnData n _ _) = returnRn (unitBag (n,ExportAbs), emptyBag)
-    checkIEAbs (RnClass n _)  = returnRn (unitBag (n,ExportAbs), emptyBag)
-    checkIEAbs rn             = returnRn (emptyBag, emptyBag)
+    checkIEAbs (RnSyn n)       = returnRn (unitBag (n,ExportAbs), emptyBag)
+    checkIEAbs (RnData n _ _)  = returnRn (unitBag (n,ExportAbs), emptyBag)
+    checkIEAbs (RnClass n _)   = returnRn (unitBag (n,ExportAbs), emptyBag)
+    checkIEAbs (WiredInTyCon t) = returnRn (unitBag (getName t,ExportAbs), emptyBag)
+    checkIEAbs rn               = --pprTrace "rnIE:IEAbs:panic? ToDo?:" (ppr PprDebug rn) $
+                                 returnRn (emptyBag, emptyBag)
 
 rnIE mods (IEThingAll name)
   = lookupTyConOrClass name    `thenRn` \ rn ->
@@ -213,14 +256,24 @@ rnIE mods (IEThingAll name)
     checkImportAll rn           `thenRn_`
     returnRn (Nothing, exps)
   where
-    checkIEAll (RnData n cons fields) = returnRn (unitBag (exp_all n), listToBag (map exp_all cons)
-                                                                        `unionBags`
-                                                                      listToBag (map exp_all fields))
-    checkIEAll (RnClass n ops)        = returnRn (unitBag (exp_all n), listToBag (map exp_all ops))
-    checkIEAll rn@(RnSyn n)           = getSrcLocRn `thenRn` \ src_loc ->
-                                       warnAndContinueRn (unitBag (n, ExportAbs), emptyBag)
-                                           (synAllExportErr False{-warning-} rn src_loc)
-    checkIEAll rn                     = returnRn (emptyBag, emptyBag)
+    checkIEAll (RnData n cons fields)
+      = returnRn (unitBag (exp_all n),
+           listToBag (map exp_all cons) `unionBags` listToBag (map exp_all fields))
+
+    checkIEAll (WiredInTyCon t)
+      = returnRn (unitBag (exp_all (getName t)), listToBag (map exp_all cons))
+      where
+       cons   = map getName (tyConDataCons t)
+
+    checkIEAll (RnClass n ops)
+      = returnRn (unitBag (exp_all n), listToBag (map exp_all ops))
+    checkIEAll rn@(RnSyn n)
+      = getSrcLocRn `thenRn` \ src_loc ->
+       warnAndContinueRn (unitBag (n, ExportAbs), emptyBag)
+                         (synAllExportErr False{-warning-} rn src_loc)
+
+    checkIEAll rn = --pprTrace "rnIE:IEAll:panic? ToDo?:" (ppr PprDebug rn) $
+                   returnRn (emptyBag, emptyBag)
 
     exp_all n = (n, ExportAll)
 
@@ -246,8 +299,10 @@ rnIE mods (IEThingWith name names)
     checkIEWith rn@(RnSyn _) rns
        = getSrcLocRn `thenRn` \ src_loc ->
          failButContinueRn (emptyBag, emptyBag) (synAllExportErr True{-error-} rn src_loc)
+    checkIEWith (WiredInTyCon _) rns = panic "RnSource.rnIE:checkIEWith:WiredInTyCon:ToDo (boring)"
     checkIEWith rn rns
-       = returnRn (emptyBag, emptyBag)
+       = --pprTrace "rnIE:IEWith:panic? ToDo?:" (ppr PprDebug rn) $
+         returnRn (emptyBag, emptyBag)
 
     exp_all n = (n, ExportAll)
 
@@ -590,8 +645,8 @@ rnFixes fixities
        rn_fixity_pieces mk_fixity name i fix
          = getRnEnv `thenRn` \ env ->
              case lookupGlobalRnEnv env name of
-               Just res | isLocallyDefined res || opt_CompilingPrelude
-                 -- the opt_CompilingPrelude thing is a *HACK* to get (:)'s
+               Just res | isLocallyDefined res -- || opt_CompilingGhcInternals
+                 -- the opt_CompilingGhcInternals thing is a *HACK* to get (:)'s
                  -- fixity decl to go through.  It has a builtin name, which
                  -- doesn't respond to isLocallyDefined...  sigh.
                  -> returnRn (Just (mk_fixity res i))
@@ -716,7 +771,11 @@ dupLocalsExportErr locn locals@((str,_):_)
 
 classOpExportErr op locn
   = addShortErrLocLine locn $ \ sty ->
-    ppBesides [ppStr "class operation `", ppr sty op, ppStr "' can only be exported with class"]
+    ppBesides [ppStr "class operation `", ppr sty op, ppStr "' can only be exported with its class"]
+
+fieldExportErr op locn
+  = addShortErrLocLine locn $ \ sty ->
+    ppBesides [ppStr "field name `", ppr sty op, ppStr "' can only be exported with its data type"]
 
 synAllExportErr is_error syn locn
   = (if is_error then addShortErrLocLine else addShortWarnLocLine) locn $ \ sty ->