module Distribution.Hackage.DB
( Hackage, readHackage, readHackage', parseHackage
, module Data.Map
, module Data.Version
, module Distribution.Package
, module Distribution.PackageDescription
)
where
import qualified Codec.Archive.Tar as Tar
import Data.ByteString.Lazy.Char8 ( ByteString )
import qualified Data.ByteString.Lazy.Char8 as BS8
import qualified Data.ByteString.Lazy as BSC
import Data.String.UTF8 ( toString, fromRep )
import Data.Map
import Data.Maybe
import Data.Version
import System.Directory
import System.FilePath
import Distribution.Package
import Distribution.Text
import Distribution.PackageDescription
import Distribution.PackageDescription.Parse ( parsePackageDescription, ParseResult(..) )
type Hackage = Map String (Map Version GenericPackageDescription)
readHackage :: IO Hackage
readHackage = do
homedir <- getHomeDirectory
readHackage' (joinPath [ homedir,
#ifdef IS_DARWIN
"Library", "Haskell", "repo-cache"
#else
".cabal", "packages"
#endif
, "hackage.haskell.org", "00-index.tar" ])
readHackage' :: FilePath -> IO Hackage
readHackage' = fmap parseHackage . BS8.readFile
parseHackage :: ByteString -> Hackage
parseHackage = Tar.foldEntries addEntry empty (error . show) . Tar.read
where
decodeUTF8 :: ByteString -> String
decodeUTF8 = toString . fromRep . BSC.unpack
addEntry :: Tar.Entry -> Hackage -> Hackage
addEntry e db = case splitDirectories (Tar.entryPath e) of
[".",".","@LongLink"] -> db
path@[name,vers,_] -> case Tar.entryContent e of
Tar.NormalFile buf _ -> add name vers buf db
_ -> error ("Hackage.DB.parseHackage: unexpected content type for " ++ show path)
_ -> db
add :: String -> String -> ByteString -> Hackage -> Hackage
add name version pkg = insertWith union name (singleton (pVersion version) (pPackage name pkg))
pPackage :: String -> ByteString -> GenericPackageDescription
pPackage name buf = case parsePackageDescription (decodeUTF8 buf) of
ParseOk _ a -> a
ParseFailed err -> error ("Hackage.DB.parseHackage: cannot parse cabal file " ++ show name ++ ": " ++ show err)
pVersion :: String -> Version
pVersion str = fromMaybe (error $ "Hackage.DB.parseHackage: cannot parse version " ++ show str) (simpleParse str)