100 lines
3.4 KiB
Haskell
100 lines
3.4 KiB
Haskell
|
module Cradle where
|
||
|
|
|
||
|
|
import Control.Applicative ((<$>))
|
||
|
|
import Control.Exception (throwIO)
|
||
|
|
import Control.Monad
|
||
|
|
import Data.List (isSuffixOf, intercalate)
|
||
|
|
import Distribution.Simple.Program (ghcProgram)
|
||
|
|
import Distribution.Simple.Program.Types (programName, programFindVersion)
|
||
|
|
import Distribution.Verbosity (silent)
|
||
|
|
import Distribution.Version (versionBranch)
|
||
|
|
import System.Directory
|
||
|
|
import System.FilePath ((</>),takeDirectory)
|
||
|
|
import Types
|
||
|
|
|
||
|
|
-- An error would be thrown
|
||
|
|
checkEnv :: Maybe FilePath -> IO Cradle
|
||
|
|
checkEnv (Just sbox) = do
|
||
|
|
(strver, ver) <- ghcVersion
|
||
|
|
conf <- checkPackageConf sbox strver
|
||
|
|
let confOpts = ghcPackageConfOptions ver conf
|
||
|
|
wdir <- getCurrentDirectory
|
||
|
|
cfiles <- cabalDir wdir
|
||
|
|
return $ case cfiles of
|
||
|
|
Nothing -> Cradle {
|
||
|
|
cradleCurrentDir = wdir
|
||
|
|
, cradleCabalDir = Nothing
|
||
|
|
, cradleCabalFile = Nothing
|
||
|
|
, cradlePackageConfOpts = Just confOpts
|
||
|
|
}
|
||
|
|
Just (cdir,cfile) -> Cradle {
|
||
|
|
cradleCurrentDir = wdir
|
||
|
|
, cradleCabalDir = Just cdir
|
||
|
|
, cradleCabalFile = Just cfile
|
||
|
|
, cradlePackageConfOpts = Just confOpts
|
||
|
|
}
|
||
|
|
checkEnv Nothing = do
|
||
|
|
(strver, ver) <- ghcVersion
|
||
|
|
wdir <- getCurrentDirectory
|
||
|
|
cfiles <- cabalDir wdir
|
||
|
|
case cfiles of
|
||
|
|
Nothing -> return $ Cradle {
|
||
|
|
cradleCurrentDir = wdir
|
||
|
|
, cradleCabalDir = Nothing
|
||
|
|
, cradleCabalFile = Nothing
|
||
|
|
, cradlePackageConfOpts = Nothing
|
||
|
|
}
|
||
|
|
Just (cdir,cfile) -> do
|
||
|
|
let sbox = cdir </> "cabal-dev/"
|
||
|
|
conf = packageConfName sbox strver
|
||
|
|
confOpts = ghcPackageConfOptions ver conf
|
||
|
|
exist <- doesFileExist conf
|
||
|
|
return $ Cradle {
|
||
|
|
cradleCurrentDir = wdir
|
||
|
|
, cradleCabalDir = Just cdir
|
||
|
|
, cradleCabalFile = Just cfile
|
||
|
|
, cradlePackageConfOpts = if exist then Just confOpts else Nothing
|
||
|
|
}
|
||
|
|
|
||
|
|
cabalDir :: FilePath -> IO (Maybe (FilePath,FilePath))
|
||
|
|
cabalDir dir = do
|
||
|
|
cnts <- (filter isCabal <$> getDirectoryContents dir)
|
||
|
|
>>= filterM (\file -> doesFileExist (dir </> file))
|
||
|
|
let dir' = takeDirectory dir
|
||
|
|
case cnts of
|
||
|
|
[] | dir' == dir -> return Nothing
|
||
|
|
| otherwise -> cabalDir dir'
|
||
|
|
cfile:_ -> return $ Just (dir,dir </> cfile)
|
||
|
|
where
|
||
|
|
isCabal name = ".cabal" `isSuffixOf` name && length name > 6
|
||
|
|
|
||
|
|
ghcVersion :: IO (String, Int)
|
||
|
|
ghcVersion = ghcVer >>= cook
|
||
|
|
where
|
||
|
|
ghcVer = programFindVersion ghcProgram silent (programName ghcProgram)
|
||
|
|
cook Nothing = throwIO $ userError $ "ghc not found"
|
||
|
|
cook (Just v)
|
||
|
|
| length vs < 2 = return (verstr, 0)
|
||
|
|
| otherwise = return (verstr, ver)
|
||
|
|
where
|
||
|
|
vs = versionBranch v
|
||
|
|
ver = (vs !! 0) * 100 + (vs !! 1)
|
||
|
|
verstr = intercalate "." . map show $ vs
|
||
|
|
|
||
|
|
packageConfName :: FilePath -> String -> FilePath
|
||
|
|
packageConfName path ver = path </> "packages-" ++ ver ++ ".conf"
|
||
|
|
|
||
|
|
checkPackageConf :: FilePath -> String -> IO FilePath
|
||
|
|
checkPackageConf path ver = do
|
||
|
|
let conf = packageConfName path ver
|
||
|
|
exist <- doesFileExist conf
|
||
|
|
if exist then
|
||
|
|
return conf
|
||
|
|
else
|
||
|
|
throwIO $ userError $ conf ++ " not found"
|
||
|
|
|
||
|
|
ghcPackageConfOptions :: Int -> String -> [String]
|
||
|
|
ghcPackageConfOptions ver file
|
||
|
|
| ver >= 706 = ["-package-db", file, "-no-user-package-conf"]
|
||
|
|
| otherwise = ["-package-conf", file, "-no-user-package-conf"]
|