module Language.ATS.Package.Dependency (
fetchDeps
, buildHelper
) where
import qualified Codec.Archive.Tar as Tar
import Codec.Archive.Zip (ZipOption (..), extractFilesFromArchive, toArchive)
import qualified Codec.Compression.BZip as Bzip
import qualified Codec.Compression.GZip as Gzip
import qualified Codec.Compression.Lzma as Lzma
import Control.Concurrent.ParallelIO.Global
import Control.Lens
import Control.Monad
import Data.Binary (decode)
import Data.ByteString.Lazy (ByteString)
import qualified Data.ByteString.Lazy as BSL
import Data.Maybe (fromMaybe)
import Data.Semigroup (Semigroup (..))
import qualified Data.Text.Lazy as TL
import Development.Shake.ATS
import Dhall
import Language.ATS.Package.Config
import Language.ATS.Package.Error
import Language.ATS.Package.PackageSet
import Language.ATS.Package.Type
import Network.HTTP.Client hiding (host)
import Network.HTTP.Client.TLS (tlsManagerSettings)
import System.Directory
import System.Environment (getEnv)
import System.Posix.Files
import System.Process
fetchDeps :: CCompiler
-> [IO ()]
-> [String]
-> [String]
-> FilePath
-> Bool
-> IO ()
fetchDeps cc' setup' deps cdeps cfgPath b' =
unless (null deps && null cdeps && b') $ do
pkgSet <- TL.unpack . defaultPkgs . decode <$> BSL.readFile cfgPath
deps' <- join <$> setBuildPlan "ats" pkgSet deps
putStrLn "Checking ATS dependencies..."
d <- (<> "lib/") <$> pkgHome cc'
let libs' = fmap (buildHelper False) deps'
cdeps' <- join <$> setBuildPlan "c" pkgSet cdeps
let unpacked = fmap (over dirLens (TL.pack d <>)) cdeps'
clibs = fmap (buildHelper False) unpacked
parallel_ (setup' ++ libs' ++ clibs)
mapM_ (setup cc') unpacked
pkgHome :: CCompiler -> IO FilePath
pkgHome cc' = (++ ("/.atspkg/" ++ ccToDir cc')) <$> getEnv "HOME"
allSubdirs :: FilePath -> IO [FilePath]
allSubdirs [] = pure mempty
allSubdirs d = do
d' <- listDirectory d
let d'' = ((d <> "/") <>) <$> d'
ds <- filterM doesDirectoryExist d''
ds' <- mapM allSubdirs ds
pure $ join (ds : ds')
clibSetup :: CCompiler
-> String
-> FilePath
-> IO ()
clibSetup cc' lib' p = do
subdirs <- allSubdirs p
configurePath <- fromMaybe (p <> "/configure") <$> findFile subdirs "configure"
setFileMode configurePath ownerModes
h <- pkgHome cc'
let procEnv = Just [("CC", ccToString cc'), ("CFLAGS" :: String, "-I" <> h <> "include"), ("PATH", "/usr/bin:/bin")]
putStrLn $ "configuring " ++ lib' ++ "..."
void $ readCreateProcess ((proc configurePath ["--prefix", h, "--host", host]) { cwd = Just p, env = procEnv, std_err = CreatePipe }) ""
putStrLn $ "building " ++ lib' ++ "..."
void $ readCreateProcess ((proc "make" []) { cwd = Just p, std_err = CreatePipe }) ""
putStrLn $ "installing " ++ lib' ++ "..."
void $ readCreateProcess ((proc "make" ["install"]) { cwd = Just p, std_err = CreatePipe }) ""
setup :: CCompiler
-> ATSDependency
-> IO ()
setup cc' (ATSDependency lib' dirName' _ _ _) = do
lib'' <- (<> TL.unpack lib') <$> pkgHome cc'
b <- doesFileExist lib''
unless b $ do
clibSetup cc' (TL.unpack lib') (TL.unpack dirName')
writeFile lib'' ""
getCompressor :: Text -> IO (ByteString -> ByteString)
getCompressor s
| ".tar.gz" `TL.isSuffixOf` s || ".tgz" `TL.isSuffixOf` s = pure Gzip.decompress
| ".tar" `TL.isSuffixOf` s = pure id
| ".tar.xz" `TL.isSuffixOf` s = pure Lzma.decompress
| ".tar.bz2" `TL.isSuffixOf` s = pure Bzip.decompress
| otherwise = unrecognized (TL.unpack s)
tarResponse :: Text -> FilePath -> ByteString -> IO ()
tarResponse url' dirName response = do
compress <- getCompressor url'
let f = Tar.unpack dirName . Tar.read . compress
f response
zipResponse :: FilePath -> ByteString -> IO ()
zipResponse dirName response = do
let options = OptDestination dirName
extractFilesFromArchive [options] (toArchive response)
buildHelper :: Bool -> ATSDependency -> IO ()
buildHelper b (ATSDependency lib' dirName' url'' _ _) = do
let (lib, dirName, url') = (lib', dirName', url'') & each %~ TL.unpack
needsSetup <- not <$> doesDirectoryExist (dirName ++ if b then "/atspkg.dhall" else "")
when needsSetup $ do
putStrLn ("Fetching library " ++ lib ++ "...")
manager <- newManager tlsManagerSettings
initialRequest <- parseRequest url'
response <- responseBody <$> httpLbs (initialRequest { method = "GET" }) manager
putStrLn ("Unpacking library " ++ lib ++ "...")
if "zip" `TL.isSuffixOf` url'' then
zipResponse dirName response
else tarResponse url'' dirName response
needsMove <- doesDirectoryExist (dirName ++ "/package")
when needsMove $ do
renameDirectory (dirName ++ "/package") "tempdir"
removeDirectoryRecursive dirName
renameDirectory "tempdir" dirName