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