libraries/installPackage.hs

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