git.iko.soy/stackage
on bb7e35e help sign in
stackage/Stackage/NarrowDatabase.hs
36 lines 1.6 kB history raw
1module Stackage.NarrowDatabase where
3import qualified Data.Map as Map
4import qualified Data.Set as Set
5import Prelude hiding (pi)
6import Stackage.Types
8-- | Narrow down the database to only the specified packages and all of
9-- their dependencies.
10narrowPackageDB :: PackageDB
11 -> Set (PackageName, Maintainer)
12 -> IO (Map PackageName BuildInfo)
13narrowPackageDB (PackageDB pdb) =
14 loop Map.empty . Set.map (\(name, maintainer) -> ([], name, maintainer))
15 where
16 loop result toProcess =
17 case Set.minView toProcess of
18 Nothing -> return result
19 Just ((users, p, maintainer), toProcess') ->
20 case Map.lookup p pdb of
21 Nothing
22 | null users -> error $ "Unknown package: " ++ show p
23 | otherwise -> loop result toProcess'
24 Just pi -> do
25 let users' = p:users
26 result' = Map.insert p BuildInfo
27 { biVersion = piVersion pi
28 , biUsers = users
29 , biMaintainer = maintainer
30 , biDeps = piDeps pi
31 } result
32 loop result' $ Set.foldl' (addDep users' result' maintainer) toProcess' $ Map.keysSet $ piDeps pi
33 addDep users result maintainer toProcess p =
34 case Map.lookup p result of
35 Nothing -> Set.insert (users, p, maintainer) toProcess
36 Just{} -> toProcess