-----------------------------------------------------------------------------
--- $Id: DriverMkDepend.hs,v 1.29 2003/07/17 12:04:53 simonmar Exp $
+-- $Id: DriverMkDepend.hs,v 1.36 2005/01/18 12:18:28 simonpj Exp $
--
-- GHC Driver
--
--
-----------------------------------------------------------------------------
-module DriverMkDepend where
+module DriverMkDepend (
+ doMkDependHSPhase, beginMkDependHS, endMkDependHS
+ ) where
#include "HsVersions.h"
-import GetImports ( getImports )
+import GetImports ( getImportsFromFile )
+import CmdLineOpts ( DynFlags )
import DriverState
import DriverUtil
import DriverFlags
+import Packages ( PackageIdH(..) )
import SysTools ( newTempName )
import qualified SysTools
-import Module ( ModuleName, ModLocation(..),
- moduleNameUserString, isHomeModule )
+import Module ( Module, ModLocation(..), moduleUserString)
import Finder ( findModule, hiBootExt, hiBootVerExt,
- mkHomeModLocation )
-import Util ( global )
+ mkHomeModLocation, FindResult(..) )
+import Util ( global, maybePrefixMatch )
import Panic
import DATA_IOREF ( IORef, readIORef, writeIORef )
GLOBAL_VAR(v_Dep_makefile_hdl, error "dep_makefile_hdl", Maybe Handle);
GLOBAL_VAR(v_Dep_tmp_file, error "dep_tmp_file", String);
GLOBAL_VAR(v_Dep_tmp_hdl, error "dep_tmp_hdl", Handle);
-GLOBAL_VAR(v_Dep_dir_contents, error "dep_dir_contents", [(String,[String])]);
depStartMarker = "# DO NOT DELETE: Beginning of Haskell dependencies"
depEndMarker = "# DO NOT DELETE: End of Haskell dependencies"
-- write the magic marker into the tmp file
hPutStrLn tmp_hdl depStartMarker
- -- cache the contents of all the import directories, for future
- -- reference.
- import_dirs <- readIORef v_Import_paths
- pkg_import_dirs <- getPackageImportPath
- import_dir_contents <- mapM softGetDirectoryContents import_dirs
- pkg_import_dir_contents <- mapM softGetDirectoryContents pkg_import_dirs
- writeIORef v_Dep_dir_contents
- (zip import_dirs import_dir_contents ++
- zip pkg_import_dirs pkg_import_dir_contents)
-
return ()
-doMkDependHSPhase basename suff input_fn
- = do src <- readFile input_fn
- let (import_sources, import_normals, mod_name) = getImports src
- (_, location') <- mkHomeModLocation mod_name "." basename suff
+doMkDependHSPhase dflags basename suff input_fn
+ = do (import_sources, import_normals, mod_name)
+ <- getImportsFromFile dflags input_fn
+ let orig_fn = basename ++ '.':suff
+ location' <- mkHomeModLocation mod_name orig_fn
-- take -ohi into account if present
ohi <- readIORef v_Output_hi
let location | Just fn <- ohi = location'{ ml_hi_file = fn }
| otherwise = location'
- let orig_fn = basename ++ '.':suff
- deps_sources <- mapM (findDependency True orig_fn) import_sources
- deps_normals <- mapM (findDependency False orig_fn) import_normals
+ deps_sources <- mapM (findDependency dflags True orig_fn) import_sources
+ deps_normals <- mapM (findDependency dflags False orig_fn) import_normals
let deps = deps_sources ++ deps_normals
osuf <- readIORef v_Object_suf
-
extra_suffixes <- readIORef v_Dep_suffixes
- let suffixes = osuf : map (++ ('_':osuf)) extra_suffixes
- ofiles = map (\suf -> basename ++ '.':suf) suffixes
-
- objs <- mapM odir_ify ofiles
+ let suffixes = map (++ ('_':osuf)) extra_suffixes
+ obj_file = ml_obj_file location
+ objs = obj_file : map (replaceFilenameSuffix obj_file) suffixes
-- Handle for file that accumulates dependencies
hdl <- readIORef v_Dep_tmp_hdl
escapeSpaces dep)
genDep (dep, True {- is an hi file -}) = do
hisuf <- readIORef v_Hi_suf
- let dep_base = remove_suffix '.' dep
- deps = (dep_base ++ hisuf)
- : map (\suf -> dep_base ++ suf ++ '_':hisuf) extra_suffixes
+ let
+ -- In order to construct hi files with alternate suffixes, we
+ -- now have to find the "basename" of the hi file. This is
+ -- difficult because we can't just split the hi filename
+ -- at the last dot - the hisuf might have dots in it. So we
+ -- check whether the hi filename ends in hisuf, and if it does,
+ -- we strip off hisuf, otherwise we strip everything after the
+ -- last dot.
+ dep_base
+ | Just rest <- maybePrefixMatch rev_hisuf rev_dep
+ = reverse rest
+ | otherwise
+ = remove_suffix '.' dep
+ where
+ rev_hisuf = reverse hisuf
+ rev_dep = reverse dep
+
+ deps = dep : map (\suf -> dep_base ++ suf ++ '_':hisuf)
+ extra_suffixes
-- length objs should be == length deps
sequence_ (zipWith (\o d -> hPutStrLn hdl (escapeSpaces o ++ " : " ++ escapeSpaces d)) objs deps)
-endMkDependHS :: IO ()
-endMkDependHS = do
+endMkDependHS :: DynFlags -> IO ()
+endMkDependHS dflags = do
makefile <- readIORef v_Dep_makefile
makefile_hdl <- readIORef v_Dep_makefile_hdl
tmp_file <- readIORef v_Dep_tmp_file
-- Create a backup of the original makefile
when (isJust makefile_hdl)
- (SysTools.copy ("Backing up " ++ makefile) makefile (makefile++".bak"))
+ (SysTools.copy dflags ("Backing up " ++ makefile)
+ makefile (makefile++".bak"))
-- Copy the new makefile in place
- SysTools.copy "Installing new makefile" tmp_file makefile
+ SysTools.copy dflags "Installing new makefile" tmp_file makefile
-findDependency :: Bool -> FilePath -> ModuleName -> IO (Maybe (String, Bool))
-findDependency is_source src imp = do
+findDependency :: DynFlags -> Bool -> FilePath -> Module -> IO (Maybe (String, Bool))
+findDependency dflags is_source src imp = do
excl_mods <- readIORef v_Dep_exclude_mods
include_prelude <- readIORef v_Dep_include_prelude
- let imp_mod = moduleNameUserString imp
+ let imp_mod = moduleUserString imp
if imp_mod `elem` excl_mods
then return Nothing
else do
- r <- findModule imp
+ r <- findModule dflags imp True{-explicit-}
case r of
- Right (mod,loc)
+ Found loc pkg
-- not in this package: we don't need a dependency
- | not (isHomeModule mod) && not include_prelude
+ | ExtPackage _ <- pkg, not include_prelude
-> return Nothing
-- normal import: just depend on the .hi file
then return (Just (boot_hi_file, not is_source))
else return (Just (hi_file, not is_source))
- Left _ -> throwDyn (ProgramError
+ _ -> throwDyn (ProgramError
(src ++ ": " ++ "can't locate import `" ++ imp_mod ++ "'" ++
if is_source then " (SOURCE import)" else ""))