Register.hs 23.7 KB
Newer Older
1 2 3
{-# LANGUAGE FlexibleContexts #-}
{-# LANGUAGE RankNTypes #-}

4 5
-----------------------------------------------------------------------------
-- |
ijones's avatar
ijones committed
6
-- Module      :  Distribution.Simple.Register
7
-- Copyright   :  Isaac Jones 2003-2004
8
-- License     :  BSD3
9
--
Duncan Coutts's avatar
Duncan Coutts committed
10
-- Maintainer  :  cabal-devel@haskell.org
ijones's avatar
ijones committed
11
-- Portability :  portable
12
--
Duncan Coutts's avatar
Duncan Coutts committed
13 14 15 16 17 18 19 20
-- This module deals with registering and unregistering packages. There are a
-- couple ways it can do this, one is to do it directly. Another is to generate
-- a script that can be run later to do it. The idea here being that the user
-- is shielded from the details of what command to use for package registration
-- for a particular compiler. In practice this aspect was not especially
-- popular so we also provide a way to simply generate the package registration
-- file which then must be manually passed to @ghc-pkg@. It is possible to
-- generate registration information for where the package is to be installed,
Ian D. Bollinger's avatar
Ian D. Bollinger committed
21
-- or alternatively to register the package in place in the build tree. The
Duncan Coutts's avatar
Duncan Coutts committed
22 23 24 25 26 27
-- latter is occasionally handy, and will become more important when we try to
-- build multi-package systems.
--
-- This module does not delegate anything to the per-compiler modules but just
-- mixes it all in in this module, which is rather unsatisfactory. The script
-- generation and the unregister feature are not well used or tested.
28

29
module Distribution.Simple.Register (
Duncan Coutts's avatar
Duncan Coutts committed
30 31 32
    register,
    unregister,

33
    internalPackageDBPath,
34
    createInternalPackageDB,
35

36
    initPackageDB,
37 38 39 40
    doesPackageDBExist,
    createPackageDB,
    deletePackageDB,

refold's avatar
refold committed
41
    invokeHcPkg,
Duncan Coutts's avatar
Duncan Coutts committed
42
    registerPackage,
43
    generateRegistrationInfo,
Duncan Coutts's avatar
Duncan Coutts committed
44 45 46
    inplaceInstalledPackageInfo,
    absoluteInstalledPackageInfo,
    generalInstalledPackageInfo,
47
  ) where
48

49 50 51
import Prelude ()
import Distribution.Compat.Prelude

52 53 54
import Distribution.Types.TargetInfo
import Distribution.Types.LocalBuildInfo

55
import Distribution.Simple.LocalBuildInfo
56
import Distribution.Simple.BuildPaths
57 58 59 60 61

import qualified Distribution.Simple.GHC   as GHC
import qualified Distribution.Simple.GHCJS as GHCJS
import qualified Distribution.Simple.LHC   as LHC
import qualified Distribution.Simple.UHC   as UHC
62
import qualified Distribution.Simple.HaskellSuite as HaskellSuite
63

64
import Distribution.Simple.BuildTarget
65
import Distribution.Simple.Compiler
Duncan Coutts's avatar
Duncan Coutts committed
66 67 68
import Distribution.Simple.Program
import Distribution.Simple.Program.Script
import qualified Distribution.Simple.Program.HcPkg as HcPkg
69
import Distribution.Simple.Setup
Duncan Coutts's avatar
Duncan Coutts committed
70
import Distribution.PackageDescription
71
import Distribution.Package
72
import qualified Distribution.InstalledPackageInfo as IPI
73
import Distribution.InstalledPackageInfo (InstalledPackageInfo)
74
import Distribution.Simple.Utils
75
import Distribution.System
76
import Distribution.Text
Duncan Coutts's avatar
Duncan Coutts committed
77
import Distribution.Verbosity as Verbosity
78
import Distribution.Version
79
import Distribution.Compat.Graph (IsNode(nodeKey))
80

81
import System.FilePath ((</>), (<.>), isAbsolute)
Duncan Coutts's avatar
Duncan Coutts committed
82
import System.Directory
ijones's avatar
ijones committed
83

84
import Data.List (partition)
85
import qualified Data.ByteString.Lazy.Char8 as BS.Char8
86

simonmar's avatar
simonmar committed
87 88 89
-- -----------------------------------------------------------------------------
-- Registration

ijones's avatar
ijones committed
90
register :: PackageDescription -> LocalBuildInfo
ijones's avatar
ijones committed
91
         -> RegisterFlags -- ^Install in the user's database?; verbose
ijones's avatar
ijones committed
92
         -> IO ()
93 94 95 96 97 98
register pkg_descr lbi flags =
   -- Duncan originally asked for us to not register/install files
   -- when there was no public library.  But with per-component
   -- configure, we legitimately need to install internal libraries
   -- so that we can get them.  So just unconditionally install.
   doRegister
99 100
 where
  doRegister = do
101
    targets <- readTargetInfos verbosity pkg_descr lbi (regArgs flags)
102 103 104

    -- It's important to register in build order, because ghc-pkg
    -- will complain if a dependency is not registered.
105 106
    let maybeGenerateOne target
            | CLib lib <- targetComponent target
107 108
            = fmap Just (generateOne pkg_descr lib lbi clbi flags)
            | otherwise = return Nothing
109
          where clbi = targetCLBI target
110 111 112 113

    ipis <-
      if fromFlag (regAssumeDepsUpToDate flags)
        then
114 115 116
          case targets of
            [target] -> do
                mb_ipi <- maybeGenerateOne target
117 118 119 120 121 122
                case mb_ipi of
                    Nothing -> die "Cannot --assume-deps-up-to-date register non-library target"
                    Just ipi -> return [ipi]
            [] -> die "In --assume-deps-up-to-date mode you must specify a target"
            _ -> die "In --assume-deps-up-to-date mode you can only register a single target"
        else fmap catMaybes
123
           . traverse maybeGenerateOne
124
           $ neededTargetsInBuildOrder' pkg_descr lbi (map nodeKey targets)
125 126 127 128
    registerAll pkg_descr lbi flags ipis
    return ()
   where
    verbosity = fromFlag (regVerbosity flags)
129 130 131 132 133

generateOne :: PackageDescription -> Library -> LocalBuildInfo -> ComponentLocalBuildInfo
            -> RegisterFlags
            -> IO InstalledPackageInfo
generateOne pkg lib lbi clbi regFlags
134
  = do
135
    absPackageDBs    <- absolutePackageDBPaths packageDbs
136
    installedPkgInfo <- generateRegistrationInfo
137
                           verbosity pkg lib lbi clbi inplace reloc distPref
138
                           (registrationPackageDB absPackageDBs)
139
    info verbosity (IPI.showInstalledPackageInfo installedPkgInfo)
140 141 142 143 144 145 146 147 148 149 150
    return installedPkgInfo
  where
    inplace   = fromFlag (regInPlace regFlags)
    reloc     = relocatable lbi
    -- FIXME: there's really no guarantee this will work.
    -- registering into a totally different db stack can
    -- fail if dependencies cannot be satisfied.
    packageDbs = nub $ withPackageDB lbi
                    ++ maybeToList (flagToMaybe  (regPackageDB regFlags))
    distPref  = fromFlag (regDistPref regFlags)
    verbosity = fromFlag (regVerbosity regFlags)
151

152 153 154 155 156
registerAll :: PackageDescription -> LocalBuildInfo -> RegisterFlags
            -> [InstalledPackageInfo]
            -> IO ()
registerAll pkg lbi regFlags ipis
  = do
157
    when (fromFlag (regPrintId regFlags)) $ do
158
      for_ ipis $ \installedPkgInfo ->
159 160 161
        -- Only print the public library's IPI
        when (IPI.sourcePackageId installedPkgInfo == packageId pkg) $
          putStrLn (display (IPI.installedUnitId installedPkgInfo))
162

163 164
     -- Three different modes:
    case () of
165 166
     _ | modeGenerateRegFile   -> writeRegistrationFileOrDirectory
       | modeGenerateRegScript -> writeRegisterScript
167 168
       | otherwise             -> do
           setupMessage verbosity "Registering" (packageId pkg)
169
           for_ ipis $ \installedPkgInfo ->
170 171
               registerPackage verbosity (compiler lbi) (withPrograms lbi)
                               HcPkg.NoMultiInstance packageDbs installedPkgInfo
Duncan Coutts's avatar
Duncan Coutts committed
172 173 174 175 176 177 178 179

  where
    modeGenerateRegFile = isJust (flagToMaybe (regGenPkgConf regFlags))
    regFile             = fromMaybe (display (packageId pkg) <.> "conf")
                                    (fromFlag (regGenPkgConf regFlags))

    modeGenerateRegScript = fromFlag (regGenScript regFlags)

Duncan Coutts's avatar
Duncan Coutts committed
180 181 182 183 184
    -- FIXME: there's really no guarantee this will work.
    -- registering into a totally different db stack can
    -- fail if dependencies cannot be satisfied.
    packageDbs = nub $ withPackageDB lbi
                    ++ maybeToList (flagToMaybe  (regPackageDB regFlags))
Duncan Coutts's avatar
Duncan Coutts committed
185 186
    verbosity = fromFlag (regVerbosity regFlags)

187 188 189 190 191
    writeRegistrationFileOrDirectory = do
      -- Handles overwriting both directory and file
      deletePackageDB regFile
      case ipis of
        [installedPkgInfo] -> do
192
          info verbosity ("Creating package registration file: " ++ regFile)
193 194
          writeUTF8File regFile (IPI.showInstalledPackageInfo installedPkgInfo)
        _ -> do
195
          info verbosity ("Creating package registration directory: " ++ regFile)
196 197 198 199 200
          createDirectory regFile
          let num_ipis = length ipis
              lpad m xs = replicate (m - length ys) '0' ++ ys
                  where ys = take m xs
              number i = lpad (length (show num_ipis)) (show i)
201
          for_ (zip ([1..] :: [Int]) ipis) $ \(i, installedPkgInfo) ->
202 203 204 205
            writeUTF8File (regFile </> (number i ++ "-" ++ display (IPI.installedUnitId installedPkgInfo)))
                          (IPI.showInstalledPackageInfo installedPkgInfo)

    writeRegisterScript =
Duncan Coutts's avatar
Duncan Coutts committed
206
      case compilerFlavor (compiler lbi) of
207 208 209 210 211
        JHC -> notice verbosity "Registration scripts not needed for jhc"
        UHC -> notice verbosity "Registration scripts not needed for uhc"
        _   -> withHcPkg
               "Registration scripts are not implemented for this compiler"
               (compiler lbi) (withPrograms lbi)
212
               (writeHcPkgRegisterScript verbosity ipis packageDbs)
Duncan Coutts's avatar
Duncan Coutts committed
213 214


215 216
generateRegistrationInfo :: Verbosity
                         -> PackageDescription
Duncan Coutts's avatar
Duncan Coutts committed
217 218 219 220
                         -> Library
                         -> LocalBuildInfo
                         -> ComponentLocalBuildInfo
                         -> Bool
221
                         -> Bool
Duncan Coutts's avatar
Duncan Coutts committed
222
                         -> FilePath
223
                         -> PackageDB
Duncan Coutts's avatar
Duncan Coutts committed
224
                         -> IO InstalledPackageInfo
225
generateRegistrationInfo verbosity pkg lib lbi clbi inplace reloc distPref packageDb = do
Duncan Coutts's avatar
Duncan Coutts committed
226 227
  --TODO: eliminate pwd!
  pwd <- getCurrentDirectory
228

229
  --TODO: the method of setting the UnitId is compiler specific
230
  --      this aspect should be delegated to a per-compiler helper.
231
  let comp = compiler lbi
232 233 234 235
      lbi' = lbi {
                withPackageDB = withPackageDB lbi
                    ++ [SpecificPackageDB (internalPackageDBPath lbi distPref)]
             }
236
  abi_hash <-
237
    case compilerFlavor comp of
238
     GHC | compilerVersion comp >= mkVersion [6,11] -> do
239
            fmap mkAbiHash $ GHC.libAbiHash verbosity pkg lbi' lib clbi
240
     GHCJS -> do
241 242
            fmap mkAbiHash $ GHCJS.libAbiHash verbosity pkg lbi' lib clbi
     _ -> return (mkAbiHash "")
243

Christiaan Baaij's avatar
Christiaan Baaij committed
244 245 246
  installedPkgInfo <-
    if inplace
      then return (inplaceInstalledPackageInfo pwd distPref
247
                     pkg abi_hash lib lbi clbi)
Christiaan Baaij's avatar
Christiaan Baaij committed
248 249
    else if reloc
      then relocRegistrationInfo verbosity
250
                     pkg lib lbi clbi abi_hash packageDb
Christiaan Baaij's avatar
Christiaan Baaij committed
251
      else return (absoluteInstalledPackageInfo
252
                     pkg abi_hash lib lbi clbi)
253

254

255
  return installedPkgInfo{ IPI.abiHash = abi_hash }
256

257 258 259 260 261
relocRegistrationInfo :: Verbosity
                      -> PackageDescription
                      -> Library
                      -> LocalBuildInfo
                      -> ComponentLocalBuildInfo
262
                      -> AbiHash
263 264
                      -> PackageDB
                      -> IO InstalledPackageInfo
265
relocRegistrationInfo verbosity pkg lib lbi clbi abi_hash packageDb =
266
  case (compilerFlavor (compiler lbi)) of
267 268
    GHC -> do fs <- GHC.pkgRoot verbosity lbi packageDb
              return (relocatableInstalledPackageInfo
269
                        pkg abi_hash lib lbi clbi fs)
270 271
    _   -> die "Distribution.Simple.Register.relocRegistrationInfo: \
               \not implemented for this compiler"
272

273
initPackageDB :: Verbosity -> Compiler -> ProgramDb -> FilePath -> IO ()
274
initPackageDB verbosity comp progdb dbPath =
275
    createPackageDB verbosity comp progdb False dbPath
276

277
-- | Create an empty package DB at the specified location.
278
createPackageDB :: Verbosity -> Compiler -> ProgramDb -> Bool
279 280 281 282 283 284 285 286 287 288 289
                -> FilePath -> IO ()
createPackageDB verbosity comp progdb preferCompat dbPath =
    case compilerFlavor comp of
      GHC   -> HcPkg.init (GHC.hcPkgInfo   progdb) verbosity preferCompat dbPath
      GHCJS -> HcPkg.init (GHCJS.hcPkgInfo progdb) verbosity False dbPath
      LHC   -> HcPkg.init (LHC.hcPkgInfo   progdb) verbosity False dbPath
      UHC   -> return ()
      HaskellSuite _ -> HaskellSuite.initPackageDB verbosity progdb dbPath
      _              -> die $ "Distribution.Simple.Register.createPackageDB: "
                           ++ "not implemented for this compiler"

290
doesPackageDBExist :: FilePath -> NoCallStackIO Bool
291 292 293 294 295 296 297
doesPackageDBExist dbPath = do
    -- currently one impl for all compiler flavours, but could change if needed
    dir_exists <- doesDirectoryExist dbPath
    if dir_exists
        then return True
        else doesFileExist dbPath

298
deletePackageDB :: FilePath -> NoCallStackIO ()
299 300 301 302 303 304 305
deletePackageDB dbPath = do
    -- currently one impl for all compiler flavours, but could change if needed
    dir_exists <- doesDirectoryExist dbPath
    if dir_exists
        then removeDirectoryRecursive dbPath
        else do file_exists <- doesFileExist dbPath
                when file_exists $ removeFile dbPath
refold's avatar
refold committed
306 307 308

-- | Run @hc-pkg@ using a given package DB stack, directly forwarding the
-- provided command-line arguments to it.
309
invokeHcPkg :: Verbosity -> Compiler -> ProgramDb -> PackageDBStack
refold's avatar
refold committed
310
                -> [String] -> IO ()
Mikhail Glushenkov's avatar
Mikhail Glushenkov committed
311 312
invokeHcPkg verbosity comp progdb dbStack extraArgs =
  withHcPkg "invokeHcPkg" comp progdb
313 314
    (\hpi -> HcPkg.invoke hpi verbosity dbStack extraArgs)

315
withHcPkg :: String -> Compiler -> ProgramDb
316
          -> (HcPkg.HcPkgInfo -> IO a) -> IO a
Mikhail Glushenkov's avatar
Mikhail Glushenkov committed
317
withHcPkg name comp progdb f =
318
  case compilerFlavor comp of
Mikhail Glushenkov's avatar
Mikhail Glushenkov committed
319 320 321
    GHC   -> f (GHC.hcPkgInfo progdb)
    GHCJS -> f (GHCJS.hcPkgInfo progdb)
    LHC   -> f (LHC.hcPkgInfo progdb)
322 323
    _     -> die ("Distribution.Simple.Register." ++ name ++ ":\
                  \not implemented for this compiler")
324

Duncan Coutts's avatar
Duncan Coutts committed
325
registerPackage :: Verbosity
326
                -> Compiler
327
                -> ProgramDb
328
                -> HcPkg.MultiInstance
329
                -> PackageDBStack
330
                -> InstalledPackageInfo
Duncan Coutts's avatar
Duncan Coutts committed
331
                -> IO ()
332
registerPackage verbosity comp progdb multiInstance packageDbs installedPkgInfo =
333
  case compilerFlavor comp of
334 335
    GHC   -> GHC.registerPackage   verbosity progdb multiInstance packageDbs installedPkgInfo
    GHCJS -> GHCJS.registerPackage verbosity progdb multiInstance packageDbs installedPkgInfo
336
    _ | HcPkg.MultiInstance == multiInstance
337
          -> die "Registering multiple package instances is not yet supported for this compiler"
338 339
    LHC   -> LHC.registerPackage   verbosity      progdb packageDbs installedPkgInfo
    UHC   -> UHC.registerPackage   verbosity comp progdb packageDbs installedPkgInfo
340
    JHC   -> notice verbosity "Registering for jhc (nothing to do)"
341
    HaskellSuite {} ->
342
      HaskellSuite.registerPackage verbosity      progdb packageDbs installedPkgInfo
Duncan Coutts's avatar
Duncan Coutts committed
343 344 345
    _    -> die "Registering is not implemented for this compiler"

writeHcPkgRegisterScript :: Verbosity
346
                         -> [InstalledPackageInfo]
347
                         -> PackageDBStack
348
                         -> HcPkg.HcPkgInfo
Duncan Coutts's avatar
Duncan Coutts committed
349
                         -> IO ()
350 351 352 353 354 355 356 357
writeHcPkgRegisterScript verbosity ipis packageDbs hpi = do
  let genScript installedPkgInfo =
          let invocation  = HcPkg.reregisterInvocation hpi Verbosity.normal
                              packageDbs (Right installedPkgInfo)
          in invocationAsSystemScript buildOS invocation
      scripts = map genScript ipis
      -- TODO: Do something more robust here
      regScript = unlines scripts
Duncan Coutts's avatar
Duncan Coutts committed
358

359
  info verbosity ("Creating package registration script: " ++ regScriptFileName)
360
  writeUTF8File regScriptFileName regScript
Duncan Coutts's avatar
Duncan Coutts committed
361 362 363 364 365 366
  setFileExecutable regScriptFileName

regScriptFileName :: FilePath
regScriptFileName = case buildOS of
                        Windows -> "register.bat"
                        _       -> "register.sh"
simonmar's avatar
simonmar committed
367

368

369 370 371
-- -----------------------------------------------------------------------------
-- Making the InstalledPackageInfo

Duncan Coutts's avatar
Duncan Coutts committed
372 373 374 375 376 377 378
-- | Construct 'InstalledPackageInfo' for a library in a package, given a set
-- of installation directories.
--
generalInstalledPackageInfo
  :: ([FilePath] -> [FilePath]) -- ^ Translate relative include dir paths to
                                -- absolute paths.
  -> PackageDescription
379
  -> AbiHash
Duncan Coutts's avatar
Duncan Coutts committed
380
  -> Library
381
  -> LocalBuildInfo
Duncan Coutts's avatar
Duncan Coutts committed
382 383 384
  -> ComponentLocalBuildInfo
  -> InstallDirs FilePath
  -> InstalledPackageInfo
385
generalInstalledPackageInfo adjustRelIncDirs pkg abi_hash lib lbi clbi installDirs =
386
  IPI.InstalledPackageInfo {
387 388 389
    IPI.sourcePackageId    = (packageId   pkg) {
                                pkgName = componentCompatPackageName clbi
                             },
390
    IPI.installedUnitId    = componentUnitId clbi,
391
    IPI.instantiatedWith   = componentInstantiatedWith clbi,
392
    IPI.compatPackageKey   = componentCompatPackageKey clbi,
Duncan Coutts's avatar
Duncan Coutts committed
393 394 395 396 397 398 399
    IPI.license            = license     pkg,
    IPI.copyright          = copyright   pkg,
    IPI.maintainer         = maintainer  pkg,
    IPI.author             = author      pkg,
    IPI.stability          = stability   pkg,
    IPI.homepage           = homepage    pkg,
    IPI.pkgUrl             = pkgUrl      pkg,
400
    IPI.synopsis           = synopsis    pkg,
Duncan Coutts's avatar
Duncan Coutts committed
401 402
    IPI.description        = description pkg,
    IPI.category           = category    pkg,
403
    IPI.abiHash            = abi_hash,
Duncan Coutts's avatar
Duncan Coutts committed
404
    IPI.exposed            = libExposed  lib,
405
    IPI.exposedModules     = componentExposedModules clbi,
Duncan Coutts's avatar
Duncan Coutts committed
406
    IPI.hiddenModules      = otherModules bi,
407
    IPI.trusted            = IPI.trusted IPI.emptyInstalledPackageInfo,
Duncan Coutts's avatar
Duncan Coutts committed
408
    IPI.importDirs         = [ libdir installDirs | hasModules ],
409 410
    -- Note. the libsubdir and datasubdir templates have already been expanded
    -- into libdir and datadir.
Duncan Coutts's avatar
Duncan Coutts committed
411 412 413
    IPI.libraryDirs        = if hasLibrary
                               then libdir installDirs : extraLibDirs bi
                               else                      extraLibDirs bi,
414
    IPI.dataDir            = datadir installDirs,
415
    IPI.hsLibraries        = if hasLibrary
416
                               then [getHSLibraryName (componentUnitId clbi)]
417
                               else [],
Duncan Coutts's avatar
Duncan Coutts committed
418
    IPI.extraLibraries     = extraLibs bi,
419
    IPI.extraGHCiLibraries = extraGHCiLibs bi,
Duncan Coutts's avatar
Duncan Coutts committed
420 421
    IPI.includeDirs        = absinc ++ adjustRelIncDirs relinc,
    IPI.includes           = includes bi,
422 423 424
                             --TODO: unclear what the root cause of the
                             -- duplication is, but we nub it here for now:
    IPI.depends            = ordNub $ map fst (componentPackageDeps clbi),
Duncan Coutts's avatar
Duncan Coutts committed
425 426 427 428 429
    IPI.ccOptions          = [], -- Note. NOT ccOptions bi!
                                 -- We don't want cc-options to be propagated
                                 -- to C compilations in other packages.
    IPI.ldOptions          = ldOptions bi,
    IPI.frameworks         = frameworks bi,
430
    IPI.frameworkDirs      = extraFrameworkDirs bi,
Duncan Coutts's avatar
Duncan Coutts committed
431
    IPI.haddockInterfaces  = [haddockdir installDirs </> haddockName pkg],
432 433
    IPI.haddockHTMLs       = [htmldir installDirs],
    IPI.pkgRoot            = Nothing
Duncan Coutts's avatar
Duncan Coutts committed
434 435 436 437
  }
  where
    bi = libBuildInfo lib
    (absinc, relinc) = partition isAbsolute (includeDirs bi)
438
    hasModules = not $ null (allLibModules lib clbi)
Duncan Coutts's avatar
Duncan Coutts committed
439
    hasLibrary = hasModules || not (null (cSources bi))
440 441 442
                            || (not (null (jsSources bi)) &&
                                compilerFlavor (compiler lbi) == GHCJS)

Ian D. Bollinger's avatar
Ian D. Bollinger committed
443
-- | Construct 'InstalledPackageInfo' for a library that is in place in the
Duncan Coutts's avatar
Duncan Coutts committed
444 445
-- build tree.
--
Ian D. Bollinger's avatar
Ian D. Bollinger committed
446
-- This function knows about the layout of in place packages.
Duncan Coutts's avatar
Duncan Coutts committed
447 448 449 450
--
inplaceInstalledPackageInfo :: FilePath -- ^ top of the build tree
                            -> FilePath -- ^ location of the dist tree
                            -> PackageDescription
451
                            -> AbiHash
Duncan Coutts's avatar
Duncan Coutts committed
452 453 454 455
                            -> Library
                            -> LocalBuildInfo
                            -> ComponentLocalBuildInfo
                            -> InstalledPackageInfo
456
inplaceInstalledPackageInfo inplaceDir distPref pkg abi_hash lib lbi clbi =
457
    generalInstalledPackageInfo adjustRelativeIncludeDirs
458
                                pkg abi_hash lib lbi clbi installDirs
Duncan Coutts's avatar
Duncan Coutts committed
459
  where
refold's avatar
refold committed
460
    adjustRelativeIncludeDirs = map (inplaceDir </>)
461
    libTargetDir = componentBuildDir lbi clbi
Duncan Coutts's avatar
Duncan Coutts committed
462
    installDirs =
463
      (absoluteComponentInstallDirs pkg lbi (componentUnitId clbi) NoCopyDest) {
464
        libdir     = inplaceDir </> libTargetDir,
465
        datadir    = inplaceDir </> dataDir pkg,
Duncan Coutts's avatar
Duncan Coutts committed
466 467 468 469 470 471 472 473 474 475 476 477 478 479
        docdir     = inplaceDocdir,
        htmldir    = inplaceHtmldir,
        haddockdir = inplaceHtmldir
      }
    inplaceDocdir  = inplaceDir </> distPref </> "doc"
    inplaceHtmldir = inplaceDocdir </> "html" </> display (packageName pkg)


-- | Construct 'InstalledPackageInfo' for the final install location of a
-- library package.
--
-- This function knows about the layout of installed packages.
--
absoluteInstalledPackageInfo :: PackageDescription
480
                             -> AbiHash
Duncan Coutts's avatar
Duncan Coutts committed
481 482 483 484
                             -> Library
                             -> LocalBuildInfo
                             -> ComponentLocalBuildInfo
                             -> InstalledPackageInfo
485
absoluteInstalledPackageInfo pkg abi_hash lib lbi clbi =
486
    generalInstalledPackageInfo adjustReativeIncludeDirs
487
                                pkg abi_hash lib lbi clbi installDirs
Duncan Coutts's avatar
Duncan Coutts committed
488 489 490 491 492 493 494
  where
    -- For installed packages we install all include files into one dir,
    -- whereas in the build tree they may live in multiple local dirs.
    adjustReativeIncludeDirs _
      | null (installIncludes bi) = []
      | otherwise                 = [includedir installDirs]
    bi = libBuildInfo lib
495
    installDirs = absoluteComponentInstallDirs pkg lbi (componentUnitId clbi) NoCopyDest
Duncan Coutts's avatar
Duncan Coutts committed
496

497 498

relocatableInstalledPackageInfo :: PackageDescription
499
                                -> AbiHash
500 501 502
                                -> Library
                                -> LocalBuildInfo
                                -> ComponentLocalBuildInfo
503
                                -> FilePath
504
                                -> InstalledPackageInfo
505
relocatableInstalledPackageInfo pkg abi_hash lib lbi clbi pkgroot =
506
    generalInstalledPackageInfo adjustReativeIncludeDirs
507
                                pkg abi_hash lib lbi clbi installDirs
508 509 510 511 512 513 514
  where
    -- For installed packages we install all include files into one dir,
    -- whereas in the build tree they may live in multiple local dirs.
    adjustReativeIncludeDirs _
      | null (installIncludes bi) = []
      | otherwise                 = [includedir installDirs]
    bi = libBuildInfo lib
515 516

    installDirs = fmap (("${pkgroot}" </>) . shortRelativePath pkgroot)
517
                $ absoluteComponentInstallDirs pkg lbi (componentUnitId clbi) NoCopyDest
518

simonmar's avatar
simonmar committed
519 520
-- -----------------------------------------------------------------------------
-- Unregistration
521

ijones's avatar
ijones committed
522
unregister :: PackageDescription -> LocalBuildInfo -> RegisterFlags -> IO ()
Duncan Coutts's avatar
Duncan Coutts committed
523 524 525
unregister pkg lbi regFlags = do
  let pkgid     = packageId pkg
      genScript = fromFlag (regGenScript regFlags)
526
      verbosity = fromFlag (regVerbosity regFlags)
Duncan Coutts's avatar
Duncan Coutts committed
527
      packageDb = fromFlagOrDefault (registrationPackageDB (withPackageDB lbi))
528
                                    (regPackageDB regFlags)
529 530 531 532 533 534 535
      unreg hpi =
        let invocation = HcPkg.unregisterInvocation
                           hpi Verbosity.normal packageDb pkgid
        in if genScript
             then writeFileAtomic unregScriptFileName
                    (BS.Char8.pack $ invocationAsSystemScript buildOS invocation)
             else runProgramInvocation verbosity invocation
Duncan Coutts's avatar
Duncan Coutts committed
536
  setupMessage verbosity "Unregistering" pkgid
537 538
  withHcPkg "unregistering is only implemented for GHC and GHCJS"
    (compiler lbi) (withPrograms lbi) unreg
ijones's avatar
ijones committed
539

Duncan Coutts's avatar
Duncan Coutts committed
540 541 542 543
unregScriptFileName :: FilePath
unregScriptFileName = case buildOS of
                          Windows -> "unregister.bat"
                          _       -> "unregister.sh"
544 545 546 547 548 549

internalPackageDBPath :: LocalBuildInfo -> FilePath -> FilePath
internalPackageDBPath lbi distPref =
      case compilerFlavor (compiler lbi) of
        UHC -> UHC.inplacePackageDbPath lbi
        _   -> distPref </> "package.conf.inplace"
550 551 552 553 554 555 556 557 558 559 560 561

-- | Initialize a new package db file for libraries defined
-- internally to the package.
createInternalPackageDB :: Verbosity -> LocalBuildInfo -> FilePath
                        -> IO PackageDB
createInternalPackageDB verbosity lbi distPref = do
    existsAlready <- doesPackageDBExist dbPath
    when existsAlready $ deletePackageDB dbPath
    createPackageDB verbosity (compiler lbi) (withPrograms lbi) False dbPath
    return (SpecificPackageDB dbPath)
  where
    dbPath = internalPackageDBPath lbi distPref