import Control.Monad
import Data.List
-import Distribution.Simple
import Distribution.PackageDescription
import Distribution.Setup
+import Distribution.Simple
import Distribution.Simple.LocalBuildInfo
+import Distribution.Simple.Utils
+import System.Cmd
import System.Environment
-import System.Exit
+import System.FilePath
+import System.Info
main :: IO ()
main = do args <- getArgs
$ confHook defaultUserHooks,
postConf = add_configure_options confArgs
$ postConf defaultUserHooks,
- buildHook = add_ghc_options ghcArgs
+ buildHook = build_primitive_sources
+ $ add_ghc_options ghcArgs
$ filter_modules_hook
$ buildHook defaultUserHooks,
+ makefileHook = add_ghc_options ghcArgs
+ $ filter_modules_hook
+ $ makefileHook defaultUserHooks,
+ regHook = add_extra_libs
+ $ regHook defaultUserHooks,
instHook = filter_modules_hook
$ instHook defaultUserHooks }
withArgs args'' $ defaultMainWithHooks hooks
| x == y = removePrefix xs ys
| otherwise = Nothing
-type Hook a = PackageDescription -> LocalBuildInfo -> Maybe UserHooks -> a
- -> IO ()
+type Hook a = PackageDescription -> LocalBuildInfo -> UserHooks -> a -> IO ()
type ConfHook = PackageDescription -> ConfigFlags -> IO LocalBuildInfo
type PostConfHook = Args -> ConfigFlags -> PackageDescription -> LocalBuildInfo
- -> IO ExitCode
+ -> IO ()
-- type PDHook = PackageDescription -> ConfigFlags -> IO ()
+build_primitive_sources :: Hook a -> Hook a
+build_primitive_sources f pd lbi uhs x
+ = do when (compilerFlavor (compiler lbi) == GHC) $ do
+ let genprimopcode = joinPath ["..", "..", "utils",
+ "genprimopcode", "genprimopcode"]
+ primops = joinPath ["..", "..", "compiler", "prelude",
+ "primops.txt"]
+ primhs = joinPath ["GHC", "Prim.hs"]
+ primopwrappers = joinPath ["GHC", "PrimopWrappers.hs"]
+ maybeExit $ system (genprimopcode ++ " --make-haskell-source < "
+ ++ primops ++ " > " ++ primhs)
+ maybeExit $ system (genprimopcode ++ " --make-haskell-wrappers < "
+ ++ primops ++ " > " ++ primopwrappers)
+ f pd lbi uhs x
+
add_ghc_options :: [String] -> Hook a -> Hook a
-add_ghc_options args f pd lbi muhs x
+add_ghc_options args f pd lbi uhs x
= do let lib' = case library pd of
Just lib ->
let bi = libBuildInfo lib
in lib { libBuildInfo = bi' }
Nothing -> error "Expected a library"
pd' = pd { library = Just lib' }
- f pd' lbi muhs x
+ f pd' lbi uhs x
add_configure_options :: [String] -> PostConfHook -> PostConfHook
add_configure_options args f as cfs pd lbi
= f (as ++ args) cfs pd lbi
filter_modules_hook :: Hook a -> Hook a
-filter_modules_hook f pd lbi muhs x
+filter_modules_hook f pd lbi uhs x
= let build_filter = case compilerFlavor $ compiler lbi of
GHC -> forGHCBuild
_ -> isPortableBuild
in lib { exposedModules = ems }
Nothing -> error "Expected a library"
pd' = pd { library = Just lib' }
- in f pd' lbi muhs x
+ in f pd' lbi uhs x
isPortableBuild :: String -> Bool
isPortableBuild s
| "GHC" `isPrefixOf` s = False
| "Data.Generics" `isPrefixOf` s = False
- | otherwise = s `notElem` ["Foreign.Concurrent", "System.Process"]
+ | otherwise = s `notElem` ["Foreign.Concurrent"]
forGHCBuild :: String -> Bool
forGHCBuild = ("GHC.Prim" /=)
_ ->
return lbi
+add_extra_libs :: Hook a -> Hook a
+add_extra_libs f pd lbi uhs x
+ = let pd' = if (os == "mingw32") && (compilerFlavor (compiler lbi) == GHC)
+ then case library pd of
+ Just lib ->
+ let lib_bi = libBuildInfo lib
+ lib_bi' = lib_bi { extraLibs = "wsock32"
+ : "msvcrt"
+ : "kernel32"
+ : "user32"
+ : "shell32"
+ : extraLibs lib_bi }
+ lib' = lib { libBuildInfo = lib_bi' }
+ in pd { library = Just lib' }
+ Nothing -> error "Expected a library"
+ else pd
+ in f pd' lbi uhs x
+