518 lines
18 KiB
Haskell
518 lines
18 KiB
Haskell
|
{-# LANGUAGE CPP #-}
|
||
|
{-# LANGUAGE DataKinds #-}
|
||
|
{-# LANGUAGE TypeApplications #-}
|
||
|
{-# LANGUAGE FlexibleContexts #-}
|
||
|
{-# LANGUAGE OverloadedStrings #-}
|
||
|
{-# LANGUAGE TemplateHaskell #-}
|
||
|
{-# LANGUAGE QuasiQuotes #-}
|
||
|
{-# LANGUAGE DuplicateRecordFields #-}
|
||
|
{-# LANGUAGE RankNTypes #-}
|
||
|
|
||
|
module GHCup.OptParse.Compile where
|
||
|
|
||
|
|
||
|
import GHCup
|
||
|
import GHCup.Errors
|
||
|
import GHCup.Utils.File
|
||
|
import GHCup.Types
|
||
|
import GHCup.Types.Optics
|
||
|
import GHCup.Utils
|
||
|
import GHCup.Utils.Logger
|
||
|
import GHCup.OptParse.Common
|
||
|
import GHCup.Utils.String.QQ
|
||
|
|
||
|
#if !MIN_VERSION_base(4,13,0)
|
||
|
import Control.Monad.Fail ( MonadFail )
|
||
|
#endif
|
||
|
import Codec.Archive ( ArchiveResult )
|
||
|
import Control.Concurrent (threadDelay)
|
||
|
import Control.Monad.Reader
|
||
|
import Control.Monad.Trans.Resource
|
||
|
import Data.Bifunctor
|
||
|
import Data.Functor
|
||
|
import Data.Maybe
|
||
|
import Data.Versions ( Version, prettyVer, version )
|
||
|
import Data.Text ( Text )
|
||
|
import Haskus.Utils.Variant.Excepts
|
||
|
import Options.Applicative hiding ( style )
|
||
|
import Options.Applicative.Help.Pretty ( text )
|
||
|
import Prelude hiding ( appendFile )
|
||
|
import System.Exit
|
||
|
import Text.PrettyPrint.HughesPJClass ( prettyShow )
|
||
|
|
||
|
import qualified Data.Text as T
|
||
|
import Control.Exception.Safe (MonadMask)
|
||
|
import System.FilePath (isPathSeparator)
|
||
|
import Text.Read (readEither)
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
----------------
|
||
|
--[ Commands ]--
|
||
|
----------------
|
||
|
|
||
|
|
||
|
data CompileCommand = CompileGHC GHCCompileOptions
|
||
|
| CompileHLS HLSCompileOptions
|
||
|
|
||
|
|
||
|
|
||
|
---------------
|
||
|
--[ Options ]--
|
||
|
---------------
|
||
|
|
||
|
|
||
|
data GHCCompileOptions = GHCCompileOptions
|
||
|
{ targetGhc :: Either Version GitBranch
|
||
|
, bootstrapGhc :: Either Version FilePath
|
||
|
, jobs :: Maybe Int
|
||
|
, buildConfig :: Maybe FilePath
|
||
|
, patchDir :: Maybe FilePath
|
||
|
, crossTarget :: Maybe Text
|
||
|
, addConfArgs :: [Text]
|
||
|
, setCompile :: Bool
|
||
|
, ovewrwiteVer :: Maybe Version
|
||
|
, buildFlavour :: Maybe String
|
||
|
, hadrian :: Bool
|
||
|
, isolateDir :: Maybe FilePath
|
||
|
}
|
||
|
|
||
|
data HLSCompileOptions = HLSCompileOptions
|
||
|
{ targetHLS :: Either Version GitBranch
|
||
|
, jobs :: Maybe Int
|
||
|
, setCompile :: Bool
|
||
|
, ovewrwiteVer :: Maybe Version
|
||
|
, isolateDir :: Maybe FilePath
|
||
|
, cabalProject :: Maybe FilePath
|
||
|
, cabalProjectLocal :: Maybe FilePath
|
||
|
, patchDir :: Maybe FilePath
|
||
|
, targetGHCs :: [ToolVersion]
|
||
|
}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
---------------
|
||
|
--[ Parsers ]--
|
||
|
---------------
|
||
|
|
||
|
|
||
|
compileP :: Parser CompileCommand
|
||
|
compileP = subparser
|
||
|
( command
|
||
|
"ghc"
|
||
|
( CompileGHC
|
||
|
<$> info
|
||
|
(ghcCompileOpts <**> helper)
|
||
|
( progDesc "Compile GHC from source"
|
||
|
<> footerDoc (Just $ text compileFooter)
|
||
|
)
|
||
|
)
|
||
|
<> command
|
||
|
"hls"
|
||
|
( CompileHLS
|
||
|
<$> info
|
||
|
(hlsCompileOpts <**> helper)
|
||
|
( progDesc "Compile HLS from source"
|
||
|
<> footerDoc (Just $ text compileHLSFooter)
|
||
|
)
|
||
|
)
|
||
|
)
|
||
|
where
|
||
|
compileFooter = [s|Discussion:
|
||
|
Compiles and installs the specified GHC version into
|
||
|
a self-contained "~/.ghcup/ghc/<ghcver>" directory
|
||
|
and symlinks the ghc binaries to "~/.ghcup/bin/<binary>-<ghcver>".
|
||
|
|
||
|
This also allows building a cross-compiler. Consult the documentation
|
||
|
first: <https://gitlab.haskell.org/ghc/ghc/-/wikis/building/cross-compiling#configuring-the-build>
|
||
|
|
||
|
ENV variables:
|
||
|
Various toolchain variables will be passed onto the ghc build system,
|
||
|
such as: CC, LD, OBJDUMP, NM, AR, RANLIB.
|
||
|
|
||
|
Examples:
|
||
|
# compile from known version
|
||
|
ghcup compile ghc -j 4 -v 8.4.2 -b 8.2.2
|
||
|
# compile from git commit/reference
|
||
|
ghcup compile ghc -j 4 -g master -b 8.2.2
|
||
|
# specify path to bootstrap ghc
|
||
|
ghcup compile ghc -j 4 -v 8.4.2 -b /usr/bin/ghc-8.2.2
|
||
|
# build cross compiler
|
||
|
ghcup compile ghc -j 4 -v 8.4.2 -b 8.2.2 -x armv7-unknown-linux-gnueabihf --config $(pwd)/build.mk -- --enable-unregisterised|]
|
||
|
|
||
|
compileHLSFooter = [s|Discussion:
|
||
|
Compiles and installs the specified HLS version.
|
||
|
The last argument is a list of GHC versions to compile for.
|
||
|
These need to be available in PATH prior to compilation.
|
||
|
|
||
|
Examples:
|
||
|
ghcup compile hls -v 1.4.0 -j 12 8.10.5 8.10.7 9.0.1|]
|
||
|
|
||
|
|
||
|
ghcCompileOpts :: Parser GHCCompileOptions
|
||
|
ghcCompileOpts =
|
||
|
GHCCompileOptions
|
||
|
<$> ((Left <$> option
|
||
|
(eitherReader
|
||
|
(first (const "Not a valid version") . version . T.pack)
|
||
|
)
|
||
|
(short 'v' <> long "version" <> metavar "VERSION" <> help
|
||
|
"The tool version to compile"
|
||
|
)
|
||
|
) <|>
|
||
|
(Right <$> (GitBranch <$> option
|
||
|
str
|
||
|
(short 'g' <> long "git-ref" <> metavar "GIT_REFERENCE" <> help
|
||
|
"The git commit/branch/ref to build from"
|
||
|
) <*>
|
||
|
optional (option str (short 'r' <> long "repository" <> metavar "GIT_REPOSITORY" <> help "The git repository to build from (defaults to GHC upstream)"))
|
||
|
)))
|
||
|
<*> option
|
||
|
(eitherReader
|
||
|
(\x ->
|
||
|
(bimap (const "Not a valid version") Left . version . T.pack $ x) <|> (if isPathSeparator (head x) then pure $ Right x else Left "Not an absolute Path")
|
||
|
)
|
||
|
)
|
||
|
( short 'b'
|
||
|
<> long "bootstrap-ghc"
|
||
|
<> metavar "BOOTSTRAP_GHC"
|
||
|
<> help
|
||
|
"The GHC version (or full path) to bootstrap with (must be installed)"
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader (readEither @Int))
|
||
|
(short 'j' <> long "jobs" <> metavar "JOBS" <> help
|
||
|
"How many jobs to use for make"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
str
|
||
|
(short 'c' <> long "config" <> metavar "CONFIG" <> help
|
||
|
"Absolute path to build config file"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
str
|
||
|
(short 'p' <> long "patchdir" <> metavar "PATCH_DIR" <> help
|
||
|
"Absolute path to patch directory (applies all .patch and .diff files in order using -p1)"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
str
|
||
|
(short 'x' <> long "cross-target" <> metavar "CROSS_TARGET" <> help
|
||
|
"Build cross-compiler for this platform"
|
||
|
)
|
||
|
)
|
||
|
<*> many (argument str (metavar "CONFIGURE_ARGS" <> help "Additional arguments to configure, prefix with '-- ' (longopts)"))
|
||
|
<*> flag
|
||
|
False
|
||
|
True
|
||
|
(long "set" <> help
|
||
|
"Set as active version after install"
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader
|
||
|
(first (const "Not a valid version") . version . T.pack)
|
||
|
)
|
||
|
(short 'o' <> long "overwrite-version" <> metavar "OVERWRITE_VERSION" <> help
|
||
|
"Allows to overwrite the finally installed VERSION with a different one, e.g. when you build 8.10.4 with your own patches, you might want to set this to '8.10.4-p1'"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
str
|
||
|
(short 'f' <> long "flavour" <> metavar "BUILD_FLAVOUR" <> help
|
||
|
"Set the compile build flavour (this value depends on the build system type: 'make' vs 'hadrian')"
|
||
|
)
|
||
|
)
|
||
|
<*> switch
|
||
|
(long "hadrian" <> help "Use the hadrian build system instead of make (only git versions seem to be properly supported atm)"
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader isolateParser)
|
||
|
( short 'i'
|
||
|
<> long "isolate"
|
||
|
<> metavar "DIR"
|
||
|
<> help "install in an isolated directory instead of the default one, no symlinks to this installation will be made"
|
||
|
)
|
||
|
)
|
||
|
|
||
|
hlsCompileOpts :: Parser HLSCompileOptions
|
||
|
hlsCompileOpts =
|
||
|
HLSCompileOptions
|
||
|
<$> ((Left <$> option
|
||
|
(eitherReader
|
||
|
(first (const "Not a valid version") . version . T.pack)
|
||
|
)
|
||
|
(short 'v' <> long "version" <> metavar "VERSION" <> help
|
||
|
"The tool version to compile"
|
||
|
)
|
||
|
) <|>
|
||
|
(Right <$> (GitBranch <$> option
|
||
|
str
|
||
|
(short 'g' <> long "git-ref" <> metavar "GIT_REFERENCE" <> help
|
||
|
"The git commit/branch/ref to build from"
|
||
|
) <*>
|
||
|
optional (option str (short 'r' <> long "repository" <> metavar "GIT_REPOSITORY" <> help "The git repository to build from (defaults to GHC upstream)"))
|
||
|
)))
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader (readEither @Int))
|
||
|
(short 'j' <> long "jobs" <> metavar "JOBS" <> help
|
||
|
"How many jobs to use for make"
|
||
|
)
|
||
|
)
|
||
|
<*> flag
|
||
|
False
|
||
|
True
|
||
|
(long "set" <> help
|
||
|
"Set as active version after install"
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader
|
||
|
(first (const "Not a valid version") . version . T.pack)
|
||
|
)
|
||
|
(short 'o' <> long "overwrite-version" <> metavar "OVERWRITE_VERSION" <> help
|
||
|
"Allows to overwrite the finally installed VERSION with a different one, e.g. when you build 8.10.4 with your own patches, you might want to set this to '8.10.4-p1'"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader isolateParser)
|
||
|
( short 'i'
|
||
|
<> long "isolate"
|
||
|
<> metavar "DIR"
|
||
|
<> help "install in an isolated directory instead of the default one, no symlinks to this installation will be made"
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
str
|
||
|
(long "cabal-project" <> metavar "CABAL_PROJECT" <> help
|
||
|
"If relative, specifies the path to cabal.project inside the unpacked HLS tarball/checkout. If absolute, will copy the file over."
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader absolutePathParser)
|
||
|
(long "cabal-project-local" <> metavar "CABAL_PROJECT_LOCAL" <> help
|
||
|
"Absolute path to a cabal.project.local to be used for the build. Will be copied over."
|
||
|
)
|
||
|
)
|
||
|
<*> optional
|
||
|
(option
|
||
|
(eitherReader absolutePathParser)
|
||
|
(short 'p' <> long "patchdir" <> metavar "PATCH_DIR" <> help
|
||
|
"Absolute path to patch directory (applies all .patch and .diff files in order using -p1)"
|
||
|
)
|
||
|
)
|
||
|
<*> some (toolVersionArgument Nothing (Just GHC))
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
---------------------------
|
||
|
--[ Effect interpreters ]--
|
||
|
---------------------------
|
||
|
|
||
|
|
||
|
type GHCEffects = '[ AlreadyInstalled
|
||
|
, BuildFailed
|
||
|
, DigestError
|
||
|
, GPGError
|
||
|
, DownloadFailed
|
||
|
, GHCupSetError
|
||
|
, NoDownload
|
||
|
, NotFoundInPATH
|
||
|
, PatchFailed
|
||
|
, UnknownArchive
|
||
|
, TarDirDoesNotExist
|
||
|
, NotInstalled
|
||
|
, DirNotEmpty
|
||
|
, ArchiveResult
|
||
|
, FileDoesNotExistError
|
||
|
, HadrianNotFound
|
||
|
, InvalidBuildConfig
|
||
|
, ProcessError
|
||
|
, CopyError
|
||
|
, BuildFailed
|
||
|
]
|
||
|
type HLSEffects = '[ AlreadyInstalled
|
||
|
, BuildFailed
|
||
|
, DigestError
|
||
|
, GPGError
|
||
|
, DownloadFailed
|
||
|
, GHCupSetError
|
||
|
, NoDownload
|
||
|
, NotFoundInPATH
|
||
|
, PatchFailed
|
||
|
, UnknownArchive
|
||
|
, TarDirDoesNotExist
|
||
|
, TagNotFound
|
||
|
, NextVerNotFound
|
||
|
, NoToolVersionSet
|
||
|
, NotInstalled
|
||
|
, DirNotEmpty
|
||
|
, ArchiveResult
|
||
|
]
|
||
|
|
||
|
|
||
|
|
||
|
runCompileGHC :: (MonadUnliftIO m, MonadIO m)
|
||
|
=> (ReaderT AppState m (VEither GHCEffects a) -> m (VEither GHCEffects a))
|
||
|
-> Excepts GHCEffects (ResourceT (ReaderT AppState m)) a
|
||
|
-> m (VEither GHCEffects a)
|
||
|
runCompileGHC runAppState =
|
||
|
runAppState
|
||
|
. runResourceT
|
||
|
. runE
|
||
|
@GHCEffects
|
||
|
|
||
|
runCompileHLS :: (MonadUnliftIO m, MonadIO m)
|
||
|
=> (ReaderT AppState m (VEither HLSEffects a) -> m (VEither HLSEffects a))
|
||
|
-> Excepts HLSEffects (ResourceT (ReaderT AppState m)) a
|
||
|
-> m (VEither HLSEffects a)
|
||
|
runCompileHLS runAppState =
|
||
|
runAppState
|
||
|
. runResourceT
|
||
|
. runE
|
||
|
@HLSEffects
|
||
|
|
||
|
|
||
|
|
||
|
------------------
|
||
|
--[ Entrypoint ]--
|
||
|
------------------
|
||
|
|
||
|
|
||
|
|
||
|
compile :: ( Monad m
|
||
|
, MonadMask m
|
||
|
, MonadUnliftIO m
|
||
|
, MonadFail m
|
||
|
)
|
||
|
=> CompileCommand
|
||
|
-> Settings
|
||
|
-> (forall eff a . ReaderT AppState m (VEither eff a) -> m (VEither eff a))
|
||
|
-> (ReaderT LeanAppState m () -> m ())
|
||
|
-> m ExitCode
|
||
|
compile compileCommand settings runAppState runLogger = do
|
||
|
VRight Dirs{ .. } <- runAppState (VRight <$> getDirs)
|
||
|
case compileCommand of
|
||
|
(CompileHLS HLSCompileOptions { .. }) -> do
|
||
|
runCompileHLS runAppState (do
|
||
|
case targetHLS of
|
||
|
Left targetVer -> do
|
||
|
GHCupInfo { _ghcupDownloads = dls } <- lift getGHCupInfo
|
||
|
let vi = getVersionInfo targetVer HLS dls
|
||
|
forM_ (_viPreCompile =<< vi) $ \msg -> do
|
||
|
lift $ logInfo msg
|
||
|
lift $ logInfo
|
||
|
"...waiting for 5 seconds, you can still abort..."
|
||
|
liftIO $ threadDelay 5000000 -- for compilation, give the user a sec to intervene
|
||
|
Right _ -> pure ()
|
||
|
ghcs <- liftE $ forM targetGHCs (\ghc -> fmap (_tvVersion . fst) . fromVersion (Just ghc) $ GHC)
|
||
|
targetVer <- liftE $ compileHLS
|
||
|
targetHLS
|
||
|
ghcs
|
||
|
jobs
|
||
|
ovewrwiteVer
|
||
|
isolateDir
|
||
|
cabalProject
|
||
|
cabalProjectLocal
|
||
|
patchDir
|
||
|
GHCupInfo { _ghcupDownloads = dls } <- lift getGHCupInfo
|
||
|
let vi = getVersionInfo targetVer HLS dls
|
||
|
when setCompile $ void $ liftE $
|
||
|
setHLS targetVer
|
||
|
pure (vi, targetVer)
|
||
|
)
|
||
|
>>= \case
|
||
|
VRight (vi, tv) -> do
|
||
|
runLogger $ logInfo
|
||
|
"HLS successfully compiled and installed"
|
||
|
forM_ (_viPostInstall =<< vi) $ \msg ->
|
||
|
runLogger $ logInfo msg
|
||
|
liftIO $ putStr (T.unpack $ prettyVer tv)
|
||
|
pure ExitSuccess
|
||
|
VLeft err@(V (BuildFailed tmpdir _)) -> do
|
||
|
case keepDirs settings of
|
||
|
Never -> runLogger $ logError $ T.pack $ prettyShow err
|
||
|
_ -> runLogger (logError $ T.pack (prettyShow err) <> "\n" <>
|
||
|
"Check the logs at " <> T.pack logsDir <> " and the build directory "
|
||
|
<> T.pack tmpdir <> " for more clues." <> "\n" <>
|
||
|
"Make sure to clean up " <> T.pack tmpdir <> " afterwards.")
|
||
|
pure $ ExitFailure 9
|
||
|
VLeft e -> do
|
||
|
runLogger $ logError $ T.pack $ prettyShow e
|
||
|
pure $ ExitFailure 9
|
||
|
(CompileGHC GHCCompileOptions { hadrian = True, crossTarget = Just _ }) -> do
|
||
|
runLogger $ logError "Hadrian cross compile support is not yet implemented!"
|
||
|
pure $ ExitFailure 9
|
||
|
(CompileGHC GHCCompileOptions {..}) ->
|
||
|
runCompileGHC runAppState (do
|
||
|
case targetGhc of
|
||
|
Left targetVer -> do
|
||
|
GHCupInfo { _ghcupDownloads = dls } <- lift getGHCupInfo
|
||
|
let vi = getVersionInfo targetVer GHC dls
|
||
|
forM_ (_viPreCompile =<< vi) $ \msg -> do
|
||
|
lift $ logInfo msg
|
||
|
lift $ logInfo
|
||
|
"...waiting for 5 seconds, you can still abort..."
|
||
|
liftIO $ threadDelay 5000000 -- for compilation, give the user a sec to intervene
|
||
|
Right _ -> pure ()
|
||
|
targetVer <- liftE $ compileGHC
|
||
|
(first (GHCTargetVersion crossTarget) targetGhc)
|
||
|
ovewrwiteVer
|
||
|
bootstrapGhc
|
||
|
jobs
|
||
|
buildConfig
|
||
|
patchDir
|
||
|
addConfArgs
|
||
|
buildFlavour
|
||
|
hadrian
|
||
|
isolateDir
|
||
|
GHCupInfo { _ghcupDownloads = dls } <- lift getGHCupInfo
|
||
|
let vi = getVersionInfo (_tvVersion targetVer) GHC dls
|
||
|
when setCompile $ void $ liftE $
|
||
|
setGHC targetVer SetGHCOnly
|
||
|
pure (vi, targetVer)
|
||
|
)
|
||
|
>>= \case
|
||
|
VRight (vi, tv) -> do
|
||
|
runLogger $ logInfo
|
||
|
"GHC successfully compiled and installed"
|
||
|
forM_ (_viPostInstall =<< vi) $ \msg ->
|
||
|
runLogger $ logInfo msg
|
||
|
liftIO $ putStr (T.unpack $ tVerToText tv)
|
||
|
pure ExitSuccess
|
||
|
VLeft (V (AlreadyInstalled _ v)) -> do
|
||
|
runLogger $ logWarn $
|
||
|
"GHC ver " <> prettyVer v <> " already installed; if you really want to reinstall it, you may want to run 'ghcup install ghc --force " <> prettyVer v <> "'"
|
||
|
pure ExitSuccess
|
||
|
VLeft (V (DirNotEmpty fp)) -> do
|
||
|
runLogger $ logWarn $
|
||
|
"Install directory " <> T.pack fp <> " is not empty. Use 'ghcup install ghc --isolate " <> T.pack fp <> " --force ..." <> "' to install regardless."
|
||
|
pure $ ExitFailure 3
|
||
|
VLeft err@(V (BuildFailed tmpdir _)) -> do
|
||
|
case keepDirs settings of
|
||
|
Never -> runLogger $ logError $ T.pack $ prettyShow err
|
||
|
_ -> runLogger (logError $ T.pack (prettyShow err) <> "\n" <>
|
||
|
"Check the logs at " <> T.pack logsDir <> " and the build directory "
|
||
|
<> T.pack tmpdir <> " for more clues." <> "\n" <>
|
||
|
"Make sure to clean up " <> T.pack tmpdir <> " afterwards.")
|
||
|
pure $ ExitFailure 9
|
||
|
VLeft e -> do
|
||
|
runLogger $ logError $ T.pack $ prettyShow e
|
||
|
pure $ ExitFailure 9
|