installPackage.hs 5.49 KB
Newer Older
1 2 3 4 5

import Distribution.PackageDescription
import Distribution.Simple
import Distribution.Simple.Configure
import Distribution.Simple.LocalBuildInfo
6 7
import Distribution.Simple.Program
import Distribution.Simple.Setup
8
import Distribution.Simple.Utils
9 10 11 12
import Distribution.Verbosity
import System.Environment

main :: IO ()
Ian Lynagh's avatar
Ian Lynagh committed
13 14 15
main
  = do args <- getArgs
       case args of
16 17
           destdir : ipref : ibindir : ilibdir : ilibexecdir
                   : idatadir : idocdir : ihtmldir_copy : ihtmldir_reg
18
                   : ghcpkg : ghcpkgconf : args' ->
Ian Lynagh's avatar
Ian Lynagh committed
19 20 21 22 23 24 25 26
               let verbosity = case args' of
                           [] -> normal
                           ['-':'v':v] ->
                               let m = case v of
                                           "" -> Nothing
                                           _ -> Just v
                               in flagToVerbosity m
                           _ -> error ("Bad arguments: " ++ show args)
27 28 29
               in doit destdir ipref ibindir ilibdir
                       ilibexecdir idatadir idocdir
                       ihtmldir_copy ihtmldir_reg
30
                       ghcpkg ghcpkgconf verbosity
Ian Lynagh's avatar
Ian Lynagh committed
31 32
           _ ->
               error "Missing arguments"
Ian Lynagh's avatar
Ian Lynagh committed
33

34
doit :: FilePath -> FilePath -> FilePath -> FilePath -> FilePath
35
     -> FilePath -> FilePath -> FilePath -> FilePath -> FilePath
36
     -> FilePath
Ian Lynagh's avatar
Ian Lynagh committed
37 38
     -> Verbosity
     -> IO ()
39 40
doit destdir ipref ibindir ilibdir ilibexecdir idatadir idocdir
     ihtmldir_copy ihtmldir_reg ghcpkg ghcpkgconf verbosity =
Ian Lynagh's avatar
Ian Lynagh committed
41
       do let userHooks = simpleUserHooks
42 43
              copyto = if null destdir then NoCopyDest else CopyTo destdir
              copyFlags = (emptyCopyFlags copyto) {
Ian Lynagh's avatar
Ian Lynagh committed
44 45
                              copyVerbose = verbosity
                          }
46
              registerFlags = emptyRegisterFlags {
47
                                  regPackageDB = Just GlobalPackageDB,
48 49
                                  regVerbose = verbosity,
                                  regGenScript = False,
Ian Lynagh's avatar
Ian Lynagh committed
50
                                  regInPlace = False
Ian Lynagh's avatar
Ian Lynagh committed
51
                              }
52
          lbi <- getConfig verbosity
53
          let pd = localPkgDescr lbi
Ian Lynagh's avatar
Ian Lynagh committed
54
              i = installDirTemplates lbi
55 56
              -- XXX This is an almighty hack, shadowing the base
              -- Setup.hs hack
57 58 59 60
              mkLib filt = case library pd of
                           Just lib ->
                               let ems = filter filt $ exposedModules lib
                               in lib {
61
                                      exposedModules = ems
62 63 64 65 66 67 68
                                   }
                           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
69 70
              -- When coying, we need to actually give a concrete
              -- directory to copy to rather than "$topdir"
71
              i_copy = i { prefixDirTemplate  = toPathTemplate ipref,
72 73 74 75 76
                           binDirTemplate     = toPathTemplate ibindir,
                           libDirTemplate     = toPathTemplate ilibdir,
                           libexecDirTemplate = toPathTemplate ilibexecdir,
                           dataDirTemplate    = toPathTemplate idatadir,
                           docDirTemplate     = toPathTemplate idocdir,
77
                           htmlDirTemplate    = toPathTemplate ihtmldir_copy
Ian Lynagh's avatar
Ian Lynagh committed
78
                         }
Ian Lynagh's avatar
Ian Lynagh committed
79
              lbi_copy = lbi { installDirTemplates = i_copy }
Ian Lynagh's avatar
Ian Lynagh committed
80 81 82 83
              -- 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
84 85 86 87
              progs = withPrograms lbi
              prog = ConfiguredProgram {
                         programId = programName ghcPkgProgram,
                         programVersion = Nothing,
sven.panne@aedion.de's avatar
sven.panne@aedion.de committed
88
                         programArgs = ["--force", "--global-conf", ghcpkgconf],
Ian Lynagh's avatar
Ian Lynagh committed
89 90 91
                         programLocation = UserSpecified ghcpkg
                     }
              progs' = updateProgram prog progs
92 93 94 95 96 97 98 99
              i_reg = i { prefixDirTemplate  = toPathTemplate ipref,
                          binDirTemplate     = toPathTemplate ibindir,
                          libDirTemplate     = toPathTemplate ilibdir,
                          libexecDirTemplate = toPathTemplate ilibexecdir,
                          dataDirTemplate    = toPathTemplate idatadir,
                          docDirTemplate     = toPathTemplate idocdir,
                          htmlDirTemplate    = toPathTemplate ihtmldir_reg
                        }
Ian Lynagh's avatar
Ian Lynagh committed
100 101
              lbi_reg = lbi { installDirTemplates = i_reg,
                              withPrograms = progs' }
102 103
          (copyHook simpleUserHooks) pd_copy lbi_copy userHooks copyFlags
          (regHook simpleUserHooks)  pd_reg  lbi_reg  userHooks registerFlags
Ian Lynagh's avatar
Ian Lynagh committed
104
          return ()
105

106 107 108 109 110 111 112 113 114 115 116 117
-- Get the build info, merging the setup-config and buildinfo files.
getConfig :: Verbosity -> IO LocalBuildInfo
getConfig verbosity = do
    lbi <- getPersistBuildConfig
    maybe_infoFile <- defaultHookedPackageDesc
    case maybe_infoFile of
        Nothing -> return lbi
        Just infoFile -> do
            hbi <- readHookedBuildInfo verbosity infoFile
            return lbi { localPkgDescr = updatePackageDescription hbi (localPkgDescr lbi)}