correct the documentation for newForeignPtr

[haskell-directory.git] / Setup.hs

diff --git a/Setup.hs b/Setup.hs
index 8064d78..693b16d 100644 (file)
--- a/Setup.hs
+++ b/Setup.hs
@@ -8,11 +8,15 @@ module Main (main) where
 
 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.FilePath
+import System.Info
 
 main :: IO ()
 main = do args <- getArgs
@@ -23,12 +27,15 @@ 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,
+                               $ filter_modules_hook
+                               $ makefileHook defaultUserHooks,
+                  regHook = add_extra_libs
+                          $ regHook defaultUserHooks,
                   instHook = filter_modules_hook
                            $ instHook defaultUserHooks }
           withArgs args'' $ defaultMainWithHooks hooks
@@ -65,6 +72,21 @@ type PostConfHook = Args -> ConfigFlags -> PackageDescription -> LocalBuildInfo
 
 -- 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 uhs x
  = do let lib' = case library pd of
@@ -116,3 +138,21 @@ add_extra_deps f pd cf
           _ ->
               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
+