1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55
|
module Main (main) where
import Distribution.Simple
import Distribution.Simple.Setup
import Distribution.PackageDescription
import Distribution.Version
import Distribution.Simple.LocalBuildInfo
import Distribution.Simple.Program
import Distribution.Verbosity
import Data.Char (isSpace)
import Data.List (dropWhile,reverse)
import Distribution.Types.UnqualComponentName
flag :: String -> FlagName
flag = mkFlagName
unqualComponentName :: String -> UnqualComponentName
unqualComponentName = mkUnqualComponentName
main :: IO ()
main = defaultMainWithHooks simpleUserHooks {
confHook = \pkg flags -> do
if lookup (flag "use-pkg-config")
(unFlagAssignment (configConfigurationsFlags flags)) == Just True
then do
confHook simpleUserHooks pkg flags
else do
lbi <- confHook simpleUserHooks pkg flags
bi <- psqlBuildInfo lbi
return lbi {
localPkgDescr = updatePackageDescription
(Just bi, [(unqualComponentName "runtests", bi)]) (localPkgDescr lbi)
}
}
psqlBuildInfo :: LocalBuildInfo -> IO BuildInfo
psqlBuildInfo lbi = do
(pgconfigProg, _) <- requireProgram verbosity
(simpleProgram "pg_config") (withPrograms lbi)
let pgconfig = getProgramOutput verbosity pgconfigProg
incDir <- pgconfig ["--includedir"]
libDir <- pgconfig ["--libdir"]
return emptyBuildInfo {
extraLibDirs = [strip libDir],
includeDirs = [strip incDir]
}
where
verbosity = normal -- honestly, this is a hack
strip x = dropWhile isSpace $ reverse $ dropWhile isSpace $ reverse x
|