git.iko.soy/typeable/stackage2nix
on v0.7.2 help sign in
typeable/stackage2nix/src/AllCabalHashes.hs
81 lines 3.1 kB history raw
1module AllCabalHashes where
3import Control.Lens hiding ((<.>))
4import Data.Aeson as A
5import Data.ByteString as BS
6import Data.ByteString.Lazy as BSL
7import Data.Map as M
8import Data.Text as T
9import Distribution.Nixpkgs.Hashes as NH
10import Distribution.Package
11import Distribution.PackageDescription
12import Distribution.PackageDescription.Parsec
13import Distribution.Text ( display )
14import Git
15import Git.Libgit2 as Libgit2
16import OpenSSL.Digest as SSL ( digest, digestByName )
17import System.FilePath
20type SHA1Hash = T.Text
21type SHA256Hash = String
23data Meta = Meta
24 { _mHashes :: !(Map String String)
25 , _mLocations :: ![String]
26 , _mPkgsize :: !Int
27 } deriving (Show)
29makeLenses ''Meta
31instance FromJSON Meta where
32 parseJSON (Object v) = Meta
33 <$> v .: "package-hashes"
34 <*> v .: "package-locations"
35 <*> v .: "package-size"
36 parseJSON o = fail ("invalid Cabal metadata: " ++ show o)
38readPackageByHash :: FilePath -> SHA1Hash -> IO (GenericPackageDescription, SHA256Hash)
39readPackageByHash repoDir sha1Hash = do
40 let repoOpts = defaultRepositoryOptions { repoPath = repoDir }
41 repo <- openLgRepository repoOpts
42 buf <- runLgRepository repo $ do
43 BlobObj (Blob _ contents) <- lookupObject =<< parseOid sha1Hash
44 case contents of
45 BlobString bs -> return bs
46 BlobStringLazy bsl -> return $ BSL.toStrict bsl
47 _ -> fail $ "Git SHA1 " ++ show sha1Hash ++ ": expected single Blob"
48 cabal <- let (_, res) = runParseResult (parseGenericPackageDescription buf)
49 in case res of Right a -> return a
50 Left (_, err) -> fail ("Git SHA1 " ++ show sha1Hash ++ ": " ++ show err)
51 let
52 hash = printSHA256 (digest (digestByName "sha256") buf)
53 pkg = setCabalFileHash hash cabal
54 return (pkg, hash)
56readPackageByName :: FilePath -> PackageIdentifier -> IO (GenericPackageDescription, SHA256Hash)
57readPackageByName repoDir (PackageIdentifier name version) = do
58 let cabalFile = repoDir </> unPackageName name </> display version </> unPackageName name <.> "cabal"
59 buf <- BS.readFile cabalFile
60 cabal <- let (_, res) = runParseResult (parseGenericPackageDescription buf)
61 in case res of Right a -> return a
62 Left (_, err) -> fail (cabalFile ++ ": " ++ show err)
63 let
64 hash = NH.printSHA256 (SSL.digest (SSL.digestByName "sha256") buf)
65 pkg = setCabalFileHash hash cabal
66 return (pkg, hash)
68setCabalFileHash :: SHA256Hash -> GenericPackageDescription -> GenericPackageDescription
69setCabalFileHash hash cabal = cabal
70 { packageDescription = (packageDescription cabal)
71 { customFieldsPD = ("X-Cabal-File-Hash", hash) : customFieldsPD (packageDescription cabal)
72 }
73 }
75readPackageMeta :: FilePath -> PackageIdentifier -> IO Meta
76readPackageMeta dirPrefix (PackageIdentifier name version) = do
77 let metaFile = dirPrefix </> unPackageName name </> display version </> unPackageName name <.> "json"
78 buf <- BS.readFile metaFile
79 case eitherDecodeStrict buf of
80 Left msg -> fail (metaFile ++ ": " ++ msg)
81 Right x -> return $ over (mHashes . ix "SHA256") (printSHA256 . packHex) x