{-# LANGUAGE CPP #-} {- | Module : Distribution.Gentoo.GHC Description : Find GHC-related breakages on Gentoo. Copyright : (c) Ivan Lazar Miljenovic 2009 License : GPL-2 or later This module defines helper functions to find broken packages in GHC, or else find packages installed with older versions of GHC. -} module Distribution.Gentoo.GHC ( ghcVersion , ghcLoc , ghcLibDir , oldGhcPkgs , brokenPkgs , allInstalledPackages ) where import Distribution.Gentoo.Util import Distribution.Gentoo.Packages -- Cabal imports import Distribution.Simple.Utils(rawSystemStdInOut) import Distribution.Verbosity(silent) import Distribution.Package(PackageIdentifier, packageId) import Distribution.InstalledPackageInfo(InstalledPackageInfo_) import Distribution.Text(simpleParse) -- Other imports import Data.Char(isDigit) import Data.Either(partitionEithers) import Data.Maybe(fromJust, mapMaybe) import qualified Data.Map as Map import Data.Map(Map) import qualified Data.ByteString.Char8 as BS import System.FilePath((), takeExtension, pathSeparator) import System.Directory( canonicalizePath , doesDirectoryExist , findExecutable) import Control.Monad(foldM, liftM) import Output -- ----------------------------------------------------------------------------- -- Common helper utils, etc. -- Get only the first line of output rawSysStdOutLine :: FilePath -> [String] -> IO String rawSysStdOutLine app = liftM (head . lines) . rawCommand app rawCommand :: FilePath -> [String] -> IO String rawCommand cmd args = do (out,_,_) <- rawSystemStdInOut silent -- verbosity cmd -- program loc args -- args #if MIN_VERSION_Cabal(1,18,0) Nothing -- cabal-1.18+: new working dir Nothing -- cabal-1.18+: new environment #endif /* MIN_VERSION_Cabal(1,18,0) */ Nothing -- input text and binary mode False -- is output in binary mode return out -- Get the first line of output from calling GHC with the given -- arguments. ghcRawOut :: [String] -> IO String ghcRawOut args = ghcLoc >>= flip rawSysStdOutLine args -- Cheat with using fromJust since we know that GHC must be in $PATH -- somewhere, probably /usr/bin. ghcLoc :: IO FilePath ghcLoc = liftM fromJust $ findExecutable "ghc" -- The version of GHC installed. ghcVersion :: IO String ghcVersion = liftM (dropWhile (not . isDigit)) $ ghcRawOut ["--version"] -- The directory where GHC has all its libraries, etc. ghcLibDir :: IO FilePath ghcLibDir = canonicalizePath =<< ghcRawOut ["--print-libdir"] ghcPkgRawOut :: [String] -> IO String ghcPkgRawOut args = ghcPkgLoc >>= flip rawCommand args -- Cheat with using fromJust since we know that ghc-pkg must be in $PATH -- somewhere, probably /usr/bin. ghcPkgLoc :: IO FilePath ghcPkgLoc = liftM fromJust $ findExecutable "ghc-pkg" -- Return the Gentoo .conf files found in this GHC libdir confFiles :: FilePath -> IO [FilePath] confFiles dir = do let gDir = dir "gentoo" exists <- doesDirectoryExist gDir if exists then do conts <- getDirectoryContents' gDir return $ map (gDir ) $ filter isConf conts else return [] where isConf file = takeExtension file == ".conf" tryMaybe :: (a -> Maybe b) -> a -> Either a b tryMaybe f a = maybe (Left a) Right $ f a type ConfMap = Map PackageIdentifier FilePath -- Attempt to match the provided broken package to one of the -- installed packages. matchConf :: ConfMap -> PackageIdentifier -> Either PackageIdentifier FilePath matchConf = tryMaybe . flip Map.lookup -- Read in all Gentoo .conf files from the current GHC version and -- create a Map readConf :: IO ConfMap readConf = ghcLibDir >>= confFiles >>= foldM addConf Map.empty -- Add this .conf file to the Map addConf :: ConfMap -> FilePath -> IO ConfMap addConf cmp conf = do cnts <- BS.unpack `fmap` BS.readFile conf case (reads cnts) of [] -> return cmp -- ebuilds that have CABAL_CORE_LIB_GHC_PV set -- for this version of GHC will have a .conf -- file containing just [] [([],_)] -> return cmp rd -> do let nm = cfNm rd return $ Map.insert nm conf cmp where -- It's not InstalledPackageInfo, as it can't read the modules cfNm :: [([InstalledPackageInfo_ String], String)] -> PackageIdentifier cfNm = packageId . head . fst . head checkPkgs :: Verbosity -> ([PackageIdentifier], [FilePath]) -> IO ([Package],[PackageIdentifier],[FilePath]) checkPkgs v (pns,cnfs) = do pkgs <- haveFiles cnfs return (pkgs, pns, []) -- ----------------------------------------------------------------------------- -- Finding packages installed with other versions of GHC oldGhcPkgs :: Verbosity -> IO [Package] oldGhcPkgs v = do thisGhc <- ghcLibDir vsay v $ "oldGhcPkgs ghc lib: " ++ show thisGhc let thisGhc' = BS.pack thisGhc -- It would be nice to do this, but we can't assume -- some crazy user hasn't deleted one of these dirs -- libFronts' <- filterM doesDirectoryExist libFronts pkgs <- liftM notGHC $ checkLibDirs v thisGhc' libFronts return pkgs -- Find packages installed by other versions of GHC in this possible -- library directory. checkLibDirs :: Verbosity -> BSFilePath -> [BSFilePath] -> IO [Package] checkLibDirs v thisGhc libDirs = do vsay v $ "checkLibDir ghc libs: " ++ show (thisGhc, libDirs) pkgsHaveContent (hasDirMatching wanted) where wanted dir = isValid dir && (not . isInvalid) dir isValid dir = any (\ldir -> isGhcLibDir ldir dir) libDirs -- Invalid if it's this GHC isInvalid fp = fp == thisGhc || BS.isPrefixOf (thisGhc `BS.snoc` pathSeparator) fp -- A valid GHC library directory starting at libdir has a name of -- "ghc", then a hyphen and then a version number. isGhcLibDir :: BSFilePath -> BSFilePath -> Bool isGhcLibDir libdir dir = go ghcDirName where -- This is hacky because FilePath doesn't work on Bytestrings... libdir' = BS.snoc libdir pathSeparator ghcDirName = BS.pack "ghc" go dn = BS.isPrefixOf ghcDir dir -- Any possible version starts with a digit && isDigit (BS.index dir ghcDirLen) where ghcDir = flip BS.snoc '-' $ BS.append libdir' dn ghcDirLen = BS.length ghcDir -- The possible places GHC could have installed lib directories libFronts :: [BSFilePath] libFronts = map BS.pack $ do lib <- ["lib", "lib64"] return $ "/" "usr" lib -- ----------------------------------------------------------------------------- -- Finding broken packages in this install of GHC. brokenPkgs :: Verbosity -> IO ([Package],[PackageIdentifier],[FilePath]) brokenPkgs v = brokenConfs v >>= checkPkgs v -- .conf files from broken packages of this GHC version brokenConfs :: Verbosity -> IO ([PackageIdentifier], [FilePath]) brokenConfs v = do vsay v "brokenConfs: getting broken output from 'ghc-pkg'" brkn <- getBroken -- Check if we actually have to go look up files and -- do IO. vsay v $ "brokenConfs: resolving package names to gentoo equivalents. " ++ show (length brkn) ++ " are broken" if null brkn then return ([], []) else do vsay v "brokenConfs: reading '*.conf' files" cnfs <- readConf vsay v $ "brokenConfs: got " ++ show (Map.size cnfs) ++ " '*.conf' files" return $ partitionEithers $ map (matchConf cnfs) brkn -- Return the closure of all packages affected by breakage getBroken :: IO [PackageIdentifier] getBroken = liftM (mapMaybe simpleParse . words) $ ghcPkgRawOut ["check", "--simple-output"] -- ----------------------------------------------------------------------------- allInstalledPackages :: IO [Package] allInstalledPackages = do libDir <- ghcLibDir let libDir' = BS.pack libDir pkgs <- liftM notGHC $ pkgsHaveContent $ hasDirMatching (==libDir') return pkgs