In Setup, tell haddock about GHC.Prim

[ghc-prim.git] / Setup.hs

diff --git a/Setup.hs b/Setup.hs
index 3c0051d..ebd4ce7 100644 (file)
--- a/Setup.hs
+++ b/Setup.hs
@@ -5,10 +5,13 @@ module Main (main) where
 
 import Control.Monad
 import Data.List
+import Data.Maybe
 import Distribution.PackageDescription
 import Distribution.Simple
 import Distribution.Simple.LocalBuildInfo
+import Distribution.Simple.Program
 import Distribution.Simple.Utils
+import Distribution.Text
 import System.Cmd
 import System.FilePath
 import System.Exit
@@ -16,15 +19,16 @@ import System.Directory
 import Control.Exception (try)
 
 main :: IO ()
-main = do let hooks = defaultUserHooks {
+main = do let hooks = simpleUserHooks {
                   regHook = addPrimModule
-                          $ regHook defaultUserHooks,
+                          $ regHook simpleUserHooks,
                   buildHook = build_primitive_sources
-                            $ buildHook defaultUserHooks,
+                            $ buildHook simpleUserHooks,
                   makefileHook = build_primitive_sources
-                               $ makefileHook defaultUserHooks,
-                  haddockHook = build_primitive_sources
-                               $ haddockHook defaultUserHooks }
+                               $ makefileHook simpleUserHooks,
+                  haddockHook = addPrimModuleForHaddock
+                              $ build_primitive_sources
+                              $ haddockHook simpleUserHooks }
           defaultMainWithHooks hooks
 
 type Hook a = PackageDescription -> LocalBuildInfo -> UserHooks -> a -> IO ()
@@ -38,11 +42,18 @@ addPrimModule f pd lbi uhs x =
            lbi' = lbi { localPkgDescr = lpd }
        f pd' lbi' uhs x
 
+addPrimModuleForHaddock :: Hook a -> Hook a
+addPrimModuleForHaddock f pd lbi uhs x =
+    do let pc = withPrograms lbi
+           pc' = userSpecifyArgs "haddock" ["GHC/Prim.hs"] pc
+           lbi' = lbi { withPrograms = pc' }
+       f pd lbi' uhs x
+
 addPrimModuleToPD :: PackageDescription -> PackageDescription
 addPrimModuleToPD pd =
     case library pd of
     Just lib ->
-        let ems = "GHC.Prim" : exposedModules lib
+        let ems = fromJust (simpleParse "GHC.Prim") : exposedModules lib
             lib' = lib { exposedModules = ems }
         in pd { library = Just lib' }
     Nothing ->