module Module
(
Module -- abstract, instance of Eq, Ord, Outputable
+ , ModuleName
+
+ , moduleNameString -- :: ModuleName -> EncodedString
+ , moduleNameUserString -- :: ModuleName -> UserString
+
, moduleString -- :: Module -> EncodedString
, moduleUserString -- :: Module -> UserString
- , moduleIfaceFlavour -- :: Module -> IfaceFlavour
- , moduleFS -- :: Module -> EncodedFS
-
- , mkBootModule -- :: Module -> Module
- , setModuleFlavour -- :: IfaceFlavour -> Module -> Module
+ , moduleName -- :: Module -> ModuleName
- , mkDynamicModule -- :: Module -> Module
+ , mkVanillaModule -- :: ModuleName -> Module
+ , mkThisModule -- :: ModuleName -> Module
+ , mkPrelModule -- :: UserString -> Module
+
, isDynamicModule -- :: Module -> Bool
+ , isPrelModule
, mkSrcModule
- , mkPrelModule -- :: UserString -> Module
- , mkSrcModuleFS -- :: UserFS -> Module
- , mkSysModuleFS -- :: EncodedFS -> IfaceFlavour -> Module
- , mkImportModuleFS -- :: UserFS -> IfaceFlavour -> Module
+ , mkSrcModuleFS -- :: UserFS -> ModuleName
+ , mkSysModuleFS -- :: EncodedFS -> ModuleName
- , pprModule
- , pprModuleSep
- , pprModuleBoot
+ , pprModule, pprModuleName
- -- IfaceFlavour
- , IfaceFlavour
- , hiFile
- , hiBootFile -- :: IfaceFlavour
- , mkDynFlavour -- :: Bool -> IfaceFlavour -> IfaceFlavour
+ -- DllFlavour
+ , DllFlavour, dll, notDll
- , bootFlavour -- :: IfaceFlavour -> Bool
+ -- ModFlavour
+ , ModFlavour,
+
+ -- Where to find a .hi file
+ , WhereFrom(..), SearchPath, mkSearchPath
+ , ModuleHiMap, mkModuleHiMaps
) where
#include "HsVersions.h"
import OccName
import Outputable
-import CmdLineOpts ( opt_Static )
-
+import FiniteMap
+import CmdLineOpts ( opt_Static, opt_CompilingPrelude, opt_WarnHiShadows, opt_HiMapSep )
+import Constants ( interfaceFileFormatVersion )
+import Maybes ( seqMaybe )
+import Maybe ( fromMaybe )
+import Directory ( doesFileExist )
+import DirUtils ( getDirectoryContents )
+import List ( intersperse )
+import Monad ( foldM )
+import IO ( hPutStrLn, stderr, isDoesNotExistError )
\end{code}
%* *
%************************************************************************
-The IfaceFlavour type is used mainly in an imported Name's Provenance
-to say whether the name comes from a regular .hi file, or whether it comes
-from a hand-written .hi-boot file. This is important, because it has to be
-propagated. Suppose
-
- C.hs imports B
- B.hs imports A
- A.hs imports C {-# SOURCE -#} ( f )
-
-Then in A.hi we may mention C.f, in an inlining. When compiling B we *must not*
-read C.f's details from C.hi, even if the latter happens to exist from an earlier
-compilation run. So we use the name "C!f" in A.hi, and when looking for an interface
-file with details of C!f we look in C.hi-boot. The "!" stuff is recorded in the
-IfaceFlavour in the Module of C.f in A.
-
-Not particularly beautiful, but it works.
-
A further twist to the tale is the support for dynamically linked libraries under
Win32. Here, dealing with the use of global variables that's residing in a DLL
requires special handling at the point of use (there's an extra level of indirection,
hiding in a DLL is explained elsewhere (ToDo: give renamer href here.)
\begin{code}
-data IfaceFlavour = HiFile -- The thing comes from a standard interface file
- -- or from the source file itself
- | HiBootFile -- ... or from a handwritten "hi-boot" interface file
-
- | HiDllFile -- The thing comes from a standard interface file, but
- -- it's corresponding object code is residing in a DLL.
- -- (see above.)
- deriving( Eq )
-
-hiFile = HiFile
-hiDllFile = HiDllFile
-hiBootFile = HiBootFile
-
--- badly named, isn't clear whether the boolean deals with
--- the 'bootedness' or the 'DLLedness'. ToDo: improve.
-mkDynFlavour :: Bool{-is really dyn?-} -> IfaceFlavour -> IfaceFlavour
-mkDynFlavour True HiFile = HiDllFile
-mkDynFlavour _ x = x
-
-instance Text IfaceFlavour where -- Just used in debug prints of lex tokens
- showsPrec n HiBootFile s = "!" ++ s
- showsPrec n HiFile s = s
- showsPrec n HiDllFile s = s
-
-bootFlavour :: IfaceFlavour -> Bool
-bootFlavour HiBootFile = True
-bootFlavour HiFile = False
-bootFlavour HiDllFile = False
+data DllFlavour = NotDll -- Ordinary module
+ | Dll -- The module's object code lives in a DLL.
+ deriving( Eq )
+
+dll = Dll
+notDll = NotDll
+
+instance Text DllFlavour where -- Just used in debug prints of lex tokens
+ showsPrec n NotDll s = s
+ showsPrec n Dll s = "dll " ++ s
\end{code}
%************************************************************************
%* *
-\subsection[Module]{The name of a module}
+\subsection{System/user module}
%* *
%************************************************************************
+We also track whether an imported module is from a 'system-ish' place. In this case
+we don't record the fact that this module depends on it, nor usages of things
+inside it.
+
+Apr 00: We want to record dependencies on all modules other than
+prelude modules else STG Hugs gets confused because it uses this
+info to know what modules to link. (Compiled GHC uses command line
+options to specify this.)
+
\begin{code}
-data Module = Module
- EncodedFS
- IfaceFlavour
+data ModFlavour = PrelMod -- A Prelude module
+ | UserMod -- Not library-ish
+\end{code}
+
+
+%************************************************************************
+%* *
+\subsection{Where from}
+%* *
+%************************************************************************
+
+The @WhereFrom@ type controls where the renamer looks for an interface file
+
+\begin{code}
+data WhereFrom = ImportByUser -- Ordinary user import: look for M.hi
+ | ImportByUserSource -- User {- SOURCE -}: look for M.hi-boot
+ | ImportBySystem -- Non user import. Look for M.hi if M is in
+ -- the module this module depends on, or is a system-ish module;
+ -- M.hi-boot otherwise
+
+instance Outputable WhereFrom where
+ ppr ImportByUser = empty
+ ppr ImportByUserSource = ptext SLIT("{- SOURCE -}")
+ ppr ImportBySystem = ptext SLIT("{- SYSTEM IMPORT -}")
+\end{code}
+
+
+%************************************************************************
+%* *
+\subsection{The name of a module}
+%* *
+%************************************************************************
+
+\begin{code}
+type ModuleName = EncodedFS
-- Haskell module names can include the quote character ',
-- so the module names have the z-encoding applied to them
+
+isPrelModuleName :: ModuleName -> Bool
+ -- True for names of prelude modules
+isPrelModuleName m = take 4 (_UNPK_ m) == "Prel"
+
+pprModuleName :: ModuleName -> SDoc
+pprModuleName nm = pprEncodedFS nm
+
+moduleNameString :: ModuleName -> EncodedString
+moduleNameString mod = _UNPK_ mod
+
+moduleNameUserString :: ModuleName -> UserString
+moduleNameUserString mod = decode (_UNPK_ mod)
+
+mkSrcModule :: UserString -> ModuleName
+mkSrcModule s = _PK_ (encode s)
+
+mkSrcModuleFS :: UserFS -> ModuleName
+mkSrcModuleFS s = encodeFS s
+
+mkSysModuleFS :: EncodedFS -> ModuleName
+mkSysModuleFS s = s
+\end{code}
+
+\begin{code}
+data Module = Module
+ ModuleName
+ ModFlavour
+ DllFlavour
\end{code}
\begin{code}
instance Outputable Module where
ppr = pprModule
--- Ignore the IfaceFlavour when comparing modules
instance Eq Module where
- (Module m1 _) == (Module m2 _) = m1 == m2
+ (Module m1 _ _) == (Module m2 _ _) = m1 == m2
instance Ord Module where
- (Module m1 _) `compare` (Module m2 _) = m1 `compare` m2
+ (Module m1 _ _) `compare` (Module m2 _ _) = m1 `compare` m2
\end{code}
\begin{code}
pprModule :: Module -> SDoc
-pprModule (Module mod _) = pprEncodedFS mod
-
-pprModuleSep, pprModuleBoot :: Module -> SDoc
-pprModuleSep (Module mod HiFile) = dot
-pprModuleSep (Module mod HiDllFile) = dot
-pprModuleSep (Module mod HiBootFile) = char '!'
-
-pprModuleBoot (Module mod HiFile) = empty
-pprModuleBoot (Module mod HiDllFile) = empty
-pprModuleBoot (Module mod HiBootFile) = char '!'
+pprModule (Module mod _ _) = getPprStyle $ \ sty ->
+ if userStyle sty then
+ text (moduleNameUserString mod)
+ else
+ pprModuleName mod
\end{code}
\begin{code}
-mkSrcModule :: UserString -> Module
-mkSrcModule s = Module (_PK_ (encode s)) HiFile
-
-mkPrelModule :: UserString -> Module
-mkPrelModule s = Module (_PK_ (encode s)) ilk
+mkModule :: FilePath -- Directory in which this module is
+ -> ModuleName -- Name of the module
+ -> DllFlavour
+ -> Module
+mkModule dir_path mod_nm is_dll
+ | isPrelModuleName mod_nm = mkPrelModule mod_nm
+ | otherwise = Module mod_nm UserMod is_dll
+ -- Make every module into a 'user module'
+ -- except those constructed by mkPrelModule
+
+
+mkVanillaModule :: ModuleName -> Module
+mkVanillaModule name = Module name UserMod dell
+ where
+ main_mod = mkSrcModuleFS SLIT("Main")
+
+ -- Main can never be in a DLL - need this
+ -- special case in order to correctly
+ -- compile PrelMain
+ dell | opt_Static || opt_CompilingPrelude ||
+ name == main_mod = NotDll
+ | otherwise = Dll
+
+
+mkThisModule :: ModuleName -> Module -- The module being comiled
+mkThisModule name =
+ Module name UserMod NotDll -- This is fine, a Dll flag is only
+ -- pinned on imported modules.
+
+mkPrelModule :: ModuleName -> Module
+mkPrelModule name = Module name sys dll
where
- ilk
- | opt_Static = HiFile
- | otherwise = HiDllFile
+ sys | opt_CompilingPrelude = UserMod
+ | otherwise = PrelMod
-mkSrcModuleFS :: UserFS -> Module
-mkSrcModuleFS s = Module (encodeFS s) HiFile
+ dll | opt_Static || opt_CompilingPrelude = NotDll
+ | otherwise = Dll
-mkImportModuleFS :: UserFS -> IfaceFlavour -> Module
-mkImportModuleFS s hif = Module (encodeFS s) hif
+moduleString :: Module -> EncodedString
+moduleString (Module mod _ _) = _UNPK_ mod
-mkSysModuleFS :: EncodedFS -> IfaceFlavour -> Module
-mkSysModuleFS s hif = Module s hif
+moduleName :: Module -> ModuleName
+moduleName (Module mod _ _) = mod
-mkBootModule :: Module -> Module
-mkBootModule (Module s _) = Module s HiBootFile
+moduleUserString :: Module -> UserString
+moduleUserString (Module mod _ _) = moduleNameUserString mod
+\end{code}
-mkDynamicModule :: Module -> Module
-mkDynamicModule (Module s HiFile) = Module s HiDllFile
-mkDynamicModule m = m
+\begin{code}
+isDynamicModule :: Module -> Bool
+isDynamicModule (Module _ _ Dll) = True
+isDynamicModule _ = False
-setModuleFlavour :: IfaceFlavour -> Module -> Module
-setModuleFlavour hif (Module n _) = Module n hif
+isPrelModule :: Module -> Bool
+isPrelModule (Module _ PrelMod _) = True
+isPrelModule _ = False
+\end{code}
-moduleString :: Module -> EncodedString
-moduleString (Module mod _) = _UNPK_ mod
-moduleFS :: Module -> EncodedFS
-moduleFS (Module mod _) = mod
+%************************************************************************
+%* *
+\subsection{Finding modules in the file system
+%* *
+%************************************************************************
-moduleUserString :: Module -> UserString
-moduleUserString (Module mod _) = decode (_UNPK_ mod)
+\begin{code}
+type ModuleHiMap = FiniteMap ModuleName (String, Module)
+ -- Mapping from module name to
+ -- * the file path of its corresponding interface file,
+ -- * the Module, decorated with it's properties
+\end{code}
-moduleIfaceFlavour :: Module -> IfaceFlavour
-moduleIfaceFlavour (Module _ hif) = hif
+(We allege that) it is quicker to build up a mapping from module names
+to the paths to their corresponding interface files once, than to search
+along the import part every time we slurp in a new module (which we
+do quite a lot of.)
+
+\begin{code}
+type SearchPath = [(String,String)] -- List of (directory,suffix) pairs to search
+ -- for interface files.
+
+mkModuleHiMaps :: SearchPath -> IO (ModuleHiMap, ModuleHiMap)
+mkModuleHiMaps dirs = foldM (getAllFilesMatching dirs) (env,env) dirs
+ where
+ env = emptyFM
+
+{- A pseudo file, currently "dLL_ifs.hi",
+ signals that the interface files
+ contained in a particular directory have got their
+ corresponding object codes stashed away in a DLL
+
+ This stuff is only needed to deal with Win32 DLLs,
+ and conceivably we conditionally compile in support
+ for handling it. (ToDo?)
+-}
+dir_contain_dll_his = "dLL_ifs.hi"
+
+getAllFilesMatching :: SearchPath
+ -> (ModuleHiMap, ModuleHiMap)
+ -> (FilePath, String)
+ -> IO (ModuleHiMap, ModuleHiMap)
+getAllFilesMatching dirs hims (dir_path, suffix) = ( do
+ -- fpaths entries do not have dir_path prepended
+ fpaths <- getDirectoryContents dir_path
+ is_dll <- catch
+ (if opt_Static || dir_path == "." then
+ return NotDll
+ else
+ do exists <- doesFileExist (dir_path ++ '/': dir_contain_dll_his)
+ return (if exists then Dll else NotDll)
+ )
+ (\ _ {-don't care-} -> return NotDll)
+ return (foldl (addModules is_dll) hims fpaths))
+ -- soft failure
+ `catch`
+ (\ err -> do
+ hPutStrLn stderr
+ ("Import path element `" ++ dir_path ++
+ if (isDoesNotExistError err) then
+ "' does not exist, ignoring."
+ else
+ "' couldn't read, ignoring.")
+
+ return hims
+ )
+ where
+ xiffus = reverse dotted_suffix
+ dotted_suffix = case suffix of
+ [] -> []
+ ('.':xs) -> suffix
+ ls -> '.':ls
+
+ hi_boot_version_xiffus =
+ reverse (show interfaceFileFormatVersion) ++ '-':hi_boot_xiffus
+ hi_boot_xiffus = "toob-ih." -- .hi-boot reversed!
+
+ addModules is_dll his@(hi_env, hib_env) filename = fromMaybe his $
+ FMAP add_hi (go xiffus rev_fname) `seqMaybe`
+
+ FMAP add_vhib (go hi_boot_version_xiffus rev_fname) `seqMaybe`
+ -- If there's a Foo.hi-boot-N file then override any Foo.hi-boot
+
+ FMAP add_hib (go hi_boot_xiffus rev_fname)
+ where
+ rev_fname = reverse filename
+ path = dir_path ++ '/':filename
+
+ -- In these functions file_nm is the base of the filename,
+ -- with the path and suffix both stripped off. The filename
+ -- is the *unencoded* module name (else 'make' gets confused).
+ -- But the domain of the HiMaps is ModuleName which is encoded.
+ add_hi file_nm = (add_to_map addNewOne hi_env file_nm, hib_env)
+ add_vhib file_nm = (hi_env, add_to_map overrideNew hib_env file_nm)
+ add_hib file_nm = (hi_env, add_to_map addNewOne hib_env file_nm)
+
+ add_to_map combiner env file_nm
+ = addToFM_C combiner env mod_nm (path, mkModule dir_path mod_nm is_dll)
+ where
+ mod_nm = mkSrcModuleFS file_nm
+
+ -- go prefix (prefix ++ stuff) == Just (reverse stuff)
+ go [] xs = Just (_PK_ (reverse xs))
+ go _ [] = Nothing
+ go (x:xs) (y:ys) | x == y = go xs ys
+ | otherwise = Nothing
+
+ addNewOne | opt_WarnHiShadows = conflict
+ | otherwise = stickWithOld
+
+ stickWithOld old new = old
+ overrideNew old new = new
+
+ conflict (old_path,mod) (new_path,_)
+ | old_path /= new_path =
+ pprTrace "Warning: " (text "Identically named interface files present on the import path, " $$
+ text (show old_path) <+> text "shadows" $$
+ text (show new_path) $$
+ text "on the import path: " <+>
+ text (concat (intersperse ":" (map fst dirs))))
+ (old_path,mod)
+ | otherwise = (old_path,mod) -- don't warn about innocous shadowings.
\end{code}
+
+%*********************************************************
+%* *
+\subsection{Making a search path}
+%* *
+%*********************************************************
+
+@mkSearchPath@ takes a string consisting of a colon-separated list
+of directories and corresponding suffixes, and turns it into a list
+of (directory, suffix) pairs. For example:
+
+\begin{verbatim}
+ mkSearchPath "foo%.hi:.%.p_hi:baz%.mc_hi"
+ = [("foo",".hi"),( ".", ".p_hi"), ("baz",".mc_hi")]
+\begin{verbatim}
+
\begin{code}
-isDynamicModule :: Module -> Bool
-isDynamicModule (Module _ HiDllFile) = True
-isDynamicModule _ = False
+mkSearchPath :: Maybe String -> SearchPath
+mkSearchPath Nothing = [(".",".hi")] -- ToDo: default should be to look in
+ -- the directory the module we're compiling
+ -- lives.
+mkSearchPath (Just s) = go s
+ where
+ go "" = []
+ go s =
+ case span (/= '%') s of
+ (dir,'%':rs) ->
+ case span (/= opt_HiMapSep) rs of
+ (hisuf,_:rest) -> (dir,hisuf):go rest
+ (hisuf,[]) -> [(dir,hisuf)]
\end{code}
+