@@ -1,20 +1,28 @@
1−#!/usr/bin/runhaskell
21\begin{code}
3−{-# OPTIONS_GHC -Wall #-}
2+{-# LANGUAGE CPP #-}
3+#ifndef MIN_VERSION_Cabal
4+#define MIN_VERSION_Cabal(x,y,z) 0
5+#endif
46module Main (main) where
57
68import Data.List ( nub )
7−import Data.Version ( showVersion )
8−import Distribution.Package ( PackageName(PackageName), PackageId, InstalledPackageId, packageVersion, packageName )
9+import Distribution.Package ( InstalledPackageId )
10+import Distribution.Package ( PackageId, Package (..), packageVersion, packageName )
911import Distribution.PackageDescription ( PackageDescription(), TestSuite(..) )
1012import Distribution.Simple ( defaultMainWithHooks, UserHooks(..), simpleUserHooks )
11−import Distribution.Simple.Utils ( rewriteFile, createDirectoryIfMissingVerbose )
13+import Distribution.Simple.Utils ( rewriteFile, createDirectoryIfMissingVerbose, copyFiles )
1214import Distribution.Simple.BuildPaths ( autogenModulesDir )
13−import Distribution.Simple.Setup ( BuildFlags(buildVerbosity), fromFlag )
14−import Distribution.Simple.LocalBuildInfo ( withLibLBI, withTestLBI, LocalBuildInfo(), ComponentLocalBuildInfo(componentPackageDeps) )
15−import Distribution.Verbosity ( Verbosity )
15+import Distribution.Simple.Setup ( BuildFlags(buildVerbosity), Flag(..), fromFlag, HaddockFlags(haddockDistPref))
16+import Distribution.Simple.LocalBuildInfo ( withLibLBI, withTestLBI, LocalBuildInfo(), ComponentLocalBuildInfo(componentPackageDeps), compiler, buildDir )
17+import Distribution.Simple.Compiler ( showCompilerId )
18+import Distribution.Text ( display , simpleParse )
19+import Distribution.Verbosity ( Verbosity, normal )
1620import System.FilePath ( (</>) )
1721
22+#if MIN_VERSION_Cabal(1,25,0)
23+import Distribution.Simple.BuildPaths ( autogenComponentModulesDir )
24+#endif
25+
1826main :: IO ()
1927main = defaultMainWithHooks simpleUserHooks
2028 { buildHook = \pkg lbi hooks flags -> do
@@ -22,25 +30,62 @@ main = defaultMainWithHooks simpleUserHooks
2230 buildHook simpleUserHooks pkg lbi hooks flags
2331 }
2432
33+haddockOutputDir :: Package p => HaddockFlags -> p -> FilePath
34+haddockOutputDir flags pkg = destDir where
35+ baseDir = case haddockDistPref flags of
36+ NoFlag -> "."
37+ Flag x -> x
38+ destDir = baseDir </> "doc" </> "html" </> display (packageName pkg)
39+
2540generateBuildModule :: Verbosity -> PackageDescription -> LocalBuildInfo -> IO ()
2641generateBuildModule verbosity pkg lbi = do
2742 let dir = autogenModulesDir lbi
43+ let bdir = buildDir lbi
2844 createDirectoryIfMissingVerbose verbosity True dir
2945 withLibLBI pkg lbi $ \_ libcfg -> do
46+ -- We need the directory with library's cabal_macros.h!
47+#if MIN_VERSION_Cabal(1,25,0)
48+ let dir2 = autogenComponentModulesDir lbi libcfg
49+#else
50+ let dir2 = dir
51+#endif
3052 withTestLBI pkg lbi $ \suite suitecfg -> do
31− rewriteFile (dir </> "Build_" ++ testName suite ++ ".hs") $ unlines
32− [ "module Build_" ++ testName suite ++ " where"
53+ -- when (testName suite == "doctest") $ we need IsString instance
54+ rewriteFile (dir </> "Build_doctests.hs") $ unlines
55+ [ "module Build_doctests where"
3356 , ""
3457 , "autogen_dir :: String"
3558 , "autogen_dir = " ++ show dir
3659 , ""
37− , "deps :: [String]"
38− , "deps = " ++ (show $ formatdeps (testDeps libcfg suitecfg))
60+ , "component_autogen_dir :: String"
61+ , "component_autogen_dir = " ++ show dir2
62+ , ""
63+ , "build_dir :: String"
64+ , "build_dir = " ++ show bdir
65+ , ""
66+ -- -package-id etc. flags
67+ , "pkgs :: [String]"
68+ , "pkgs = " ++ (show $ formatDeps $ testDeps libcfg suitecfg)
69+ , ""
70+ , "compiler :: String"
71+ , "compiler = " ++ (show $ showCompilerId $ compiler lbi)
72+ , ""
73+ , "isOldCompiler :: Bool"
74+ , "isOldCompiler = " ++ show isOldCompiler
3975 ]
4076 where
41− formatdeps = map (formatone . snd)
42− formatone p = case packageName p of
43− PackageName n -> n ++ "-" ++ showVersion (packageVersion p)
77+ -- we do this check in Setup, as then doctests don't need to depend on Cabal
78+ isOldCompiler = maybe False id $ do
79+ a <- simpleParse $ showCompilerId $ compiler lbi
80+ b <- simpleParse "7.5"
81+ return $ packageVersion (a :: PackageId) < b
82+
83+ formatDeps = map formatOne
84+ formatOne (installedPkgId, pkgId)
85+ -- The problem is how different cabal executables handle package databases
86+ -- when doctests depend on the library
87+ | packageId pkg == pkgId = "-package=" ++ display pkgId
88+ | otherwise = "-package-id=" ++ display installedPkgId
4489
4590testDeps :: ComponentLocalBuildInfo -> ComponentLocalBuildInfo -> [(InstalledPackageId, PackageId)]
4691testDeps xs ys = nub $ componentPackageDeps xs ++ componentPackageDeps ys