installPackage.hs 3.67 KB
Newer Older
1
2

import Distribution.PackageDescription
Ian Lynagh's avatar
Ian Lynagh committed
3
import Distribution.Program
4
5
6
7
8
9
10
11
12
import Distribution.Setup
import Distribution.Simple
import Distribution.Simple.Configure
import Distribution.Simple.LocalBuildInfo
import Distribution.Verbosity
import System.Environment

main :: IO ()
main = do args <- getArgs
Ian Lynagh's avatar
Ian Lynagh committed
13
          case args of
14
              destdir : pref : ghcpkg : ghcpkgconf : args' ->
Ian Lynagh's avatar
Ian Lynagh committed
15
                  let verbosity = case args' of
16
17
18
19
20
21
22
                              [] -> normal
                              ['-':'v':v] ->
                                  let m = case v of
                                              "" -> Nothing
                                              _ -> Just v
                                  in flagToVerbosity m
                              _ -> error ("Bad arguments: " ++ show args)
23
                  in doit destdir pref ghcpkg ghcpkgconf verbosity
Ian Lynagh's avatar
Ian Lynagh committed
24
25
26
              _ ->
                  error "Missing arguments"

27
28
doit :: FilePath -> FilePath -> FilePath -> FilePath -> Verbosity -> IO ()
doit destdir pref ghcpkg ghcpkgconf verbosity =
Ian Lynagh's avatar
Ian Lynagh committed
29
       do let userHooks = simpleUserHooks
30
31
              copyto = if null destdir then NoCopyDest else CopyTo destdir
              copyFlags = (emptyCopyFlags copyto) {
Ian Lynagh's avatar
Ian Lynagh committed
32
33
                              copyVerbose = verbosity
                          }
34
              registerFlags = emptyRegisterFlags {
Ian Lynagh's avatar
Ian Lynagh committed
35
                                  regUser = MaybeUserGlobal,
36
37
                                  regVerbose = verbosity,
                                  regGenScript = False,
Ian Lynagh's avatar
Ian Lynagh committed
38
                                  regInPlace = False
Ian Lynagh's avatar
Ian Lynagh committed
39
                              }
40
          lbi <- getPersistBuildConfig
41
          let pd = localPkgDescr lbi
Ian Lynagh's avatar
Ian Lynagh committed
42
              i = installDirTemplates lbi
43
44
              -- XXX This is an almighty hack, shadowing the base
              -- Setup.hs hack
45
46
47
48
              mkLib filt = case library pd of
                           Just lib ->
                               let ems = filter filt $ exposedModules lib
                               in lib {
49
                                      exposedModules = ems
50
51
52
53
54
55
56
                                   }
                           Nothing ->
                               error "Expected a library, but none found"
              -- There's no files for GHC.Prim, so we will fail if we
              -- try to copy them
              pd_copy = pd { library = Just (mkLib ("GHC.Prim" /=)) }
              pd_reg  = pd { library = Just (mkLib (const True)) }
Ian Lynagh's avatar
Ian Lynagh committed
57
58
              -- When coying, we need to actually give a concrete
              -- directory to copy to rather than "$topdir"
Ian Lynagh's avatar
Ian Lynagh committed
59
60
              i_copy = i { prefixDirTemplate = toPathTemplate pref }
              lbi_copy = lbi { installDirTemplates = i_copy }
Ian Lynagh's avatar
Ian Lynagh committed
61
62
63
64
              -- When we run GHC we give it a $topdir that includes the
              -- $compiler/lib/ part of libsubdir, so we only want the
              -- $pkgid part in the package.conf file. This is a bit of
              -- a hack, really.
Ian Lynagh's avatar
Ian Lynagh committed
65
66
67
68
69
70
71
72
73
74
75
              progs = withPrograms lbi
              prog = ConfiguredProgram {
                         programId = programName ghcPkgProgram,
                         programVersion = Nothing,
                         programArgs = ["--global-conf", ghcpkgconf],
                         programLocation = UserSpecified ghcpkg
                     }
              progs' = updateProgram prog progs
              i_reg = i { libSubdirTemplate = toPathTemplate "$pkgid" }
              lbi_reg = lbi { installDirTemplates = i_reg,
                              withPrograms = progs' }
76
77
          (copyHook simpleUserHooks) pd_copy lbi_copy userHooks copyFlags
          (regHook simpleUserHooks)  pd_reg  lbi_reg  userHooks registerFlags
Ian Lynagh's avatar
Ian Lynagh committed
78
          return ()
79