461542956033f7dfd99ba4fbd53ce8618292b614
[ghc-hetmet.git] / libraries / installPackage.hs
1
2 import Distribution.PackageDescription
3 import Distribution.Simple
4 import Distribution.Simple.Configure
5 import Distribution.Simple.LocalBuildInfo
6 import Distribution.Simple.Program
7 import Distribution.Simple.Setup
8 import Distribution.Simple.Utils
9 import Distribution.Verbosity
10 import System.Environment
11
12 main :: IO ()
13 main
14   = do args <- getArgs
15        case args of
16            "register" : "--inplace" :args' ->
17                let verbosity = mkVerbosity args'
18                in doRegisterInplace verbosity
19            "install" : ghcpkg : ghcpkgconf : destdir : topdir :
20                     iprefix : ibindir : ilibdir : ilibexecdir : idynlibdir :
21                     idatadir : idocdir : ihtmldir : ihaddockdir :
22                     args' ->
23                let verbosity = mkVerbosity args'
24                in doInstall verbosity ghcpkg ghcpkgconf destdir topdir
25                             iprefix ibindir ilibdir ilibexecdir idynlibdir idatadir
26                             idocdir ihtmldir ihaddockdir
27            _ ->
28                error ("Bad arguments: " ++ show args)
29
30 mkVerbosity :: [String] -> Verbosity
31 mkVerbosity [] = normal
32 mkVerbosity ['-':'v':v] = let m = case v of
33                                       "" -> Nothing
34                                       _ -> Just v
35                           in flagToVerbosity m
36 mkVerbosity args = error ("Bad arguments: " ++ show args)
37
38 doRegisterInplace :: Verbosity -> IO ()
39 doRegisterInplace verbosity =
40        do lbi <- getConfig verbosity
41           let registerFlags = defaultRegisterFlags { regInPlace = toFlag True }
42               pd = localPkgDescr lbi
43               pd_reg = if pkgName (package pd) == "ghc-prim"
44                        then case library pd of
45                             Just lib ->
46                                 let ems = "GHC.Prim" : exposedModules lib
47                                     lib' = lib { exposedModules = ems }
48                                 in pd { library = Just lib' }
49                             Nothing ->
50                                 error "Expected a library, but none found"
51                        else pd
52           (regHook simpleUserHooks) pd_reg lbi simpleUserHooks registerFlags
53           return ()
54
55 doInstall :: Verbosity -> FilePath -> FilePath -> FilePath -> FilePath
56           -> FilePath -> FilePath -> FilePath -> FilePath -> FilePath
57           -> FilePath -> FilePath -> FilePath -> FilePath
58           -> IO ()
59 doInstall verbosity ghcpkg ghcpkgconf destdir topdir
60      iprefix ibindir ilibdir ilibexecdir idynlibdir idatadir
61      idocdir ihtmldir ihaddockdir =
62        do let userHooks = simpleUserHooks
63               copyto = if null destdir then NoCopyDest else CopyTo destdir
64               copyFlags = defaultCopyFlags {
65                               copyDest = toFlag copyto,
66                               copyVerbose = toFlag verbosity
67                           }
68               registerFlags = defaultRegisterFlags {
69                                   regPackageDB = toFlag GlobalPackageDB,
70                                   regVerbose = toFlag verbosity,
71                                   regGenScript = toFlag $ False,
72                                   regInPlace = toFlag $ False
73                               }
74           lbi <- getConfig verbosity
75           let pd = localPkgDescr lbi
76               i = installDirTemplates lbi
77               -- This is an almighty hack. We need to register
78               -- ghc-prim:GHC.Prim, but it doesn't exist, get built, get
79               -- haddocked, get copied, etc.
80               pd_reg = if pkgName (package pd) == "ghc-prim"
81                        then case library pd of
82                             Just lib ->
83                                 let ems = "GHC.Prim" : exposedModules lib
84                                     lib' = lib { exposedModules = ems }
85                                 in pd { library = Just lib' }
86                             Nothing ->
87                                 error "Expected a library, but none found"
88                        else pd
89               -- When coying, we need to actually give a concrete
90               -- directory to copy to rather than "$topdir"
91               toPathTemplate' = toPathTemplate . replaceTopdir topdir
92               i_copy = i { prefix       = toPathTemplate' iprefix,
93                            bindir       = toPathTemplate' ibindir,
94                            libdir       = toPathTemplate' ilibdir,
95                            dynlibdir    = toPathTemplate' idynlibdir,
96                            libexecdir   = toPathTemplate' ilibexecdir,
97                            datadir      = toPathTemplate' idatadir,
98                            docdir       = toPathTemplate' idocdir,
99                            htmldir      = toPathTemplate' ihtmldir,
100                            haddockdir   = toPathTemplate' ihaddockdir
101                          }
102               lbi_copy = lbi { installDirTemplates = i_copy }
103               -- When we run GHC we give it a $topdir that includes the
104               -- $compiler/lib/ part of libsubdir, so we only want the
105               -- $pkgid part in the package.conf file. This is a bit of
106               -- a hack, really.
107               progs = withPrograms lbi
108               prog = ConfiguredProgram {
109                          programId = programName ghcPkgProgram,
110                          programVersion = Nothing,
111                          programArgs = ["--force", "--global-conf", ghcpkgconf],
112                          programLocation = UserSpecified ghcpkg
113                      }
114               progs' = updateProgram prog progs
115               i_reg = i { prefix       = toPathTemplate iprefix,
116                           bindir       = toPathTemplate ibindir,
117                           libdir       = toPathTemplate ilibdir,
118                           dynlibdir    = toPathTemplate idynlibdir,
119                           libexecdir   = toPathTemplate ilibexecdir,
120                           datadir      = toPathTemplate idatadir,
121                           docdir       = toPathTemplate idocdir,
122                           htmldir      = toPathTemplate ihtmldir,
123                           haddockdir   = toPathTemplate ihaddockdir
124                         }
125               lbi_reg = lbi { installDirTemplates = i_reg,
126                               withPrograms = progs' }
127           (copyHook simpleUserHooks) pd     lbi_copy userHooks copyFlags
128           (regHook simpleUserHooks)  pd_reg lbi_reg  userHooks registerFlags
129           return ()
130
131 replaceTopdir :: FilePath -> FilePath -> FilePath
132 replaceTopdir topdir ('$':'t':'o':'p':'d':'i':'r':p) = topdir ++ p
133 replaceTopdir topdir ('$':'h':'t':'t':'p':'t':'o':'p':'d':'i':'r':p)
134     = topdir ++ p
135 replaceTopdir _ p = p
136
137 -- Get the build info, merging the setup-config and buildinfo files.
138 getConfig :: Verbosity -> IO LocalBuildInfo
139 getConfig verbosity = do
140     lbi <- getPersistBuildConfig
141     maybe_infoFile <- defaultHookedPackageDesc
142     case maybe_infoFile of
143         Nothing -> return lbi
144         Just infoFile -> do
145             hbi <- readHookedBuildInfo verbosity infoFile
146             return lbi { localPkgDescr = updatePackageDescription hbi (localPkgDescr lbi)}
147
148