module Development.Shake.Plus.Extended.Loaders (
batchLoad
, batchLoadWithin
, batchLoadWithin'
, batchLoadIndex
, batchLoadIndex'
) where
import Control.Comonad.Env as E
import Development.Shake.Plus.Core
import Development.Shake.Plus.Directory
import qualified Data.IxSet.Typed as Ix
import Path
import RIO
import qualified RIO.HashMap as HM
import Within
traverseToSnd :: Functor f => (a -> f b) -> a -> f (a, b)
traverseToSnd f a = (a,) <$> f a
batchLoad :: MonadAction m
=> Path b Dir
-> [FilePattern]
-> (Path b File -> m a)
-> m (HashMap (Path Rel File) a)
batchLoad dir pat f = do
xs <- getDirectoryFiles dir pat
xs' <- mapM (traverseToSnd $ f . (dir </>)) xs
return . HM.fromList $ xs'
batchLoadWithin :: MonadAction m
=> Within b [FilePattern]
-> (Within b (Path Rel File) -> m a)
-> m (Within b (HashMap (Path Rel File) a))
batchLoadWithin w f = do
xs <- getDirectoryFiles (E.ask w) (extract w)
xs' <- mapM (traverseToSnd $ f . (`within` E.ask w)) xs
return $ (`within` E.ask w) $ HM.fromList xs'
batchLoadWithin' :: MonadAction m
=> Within b [FilePattern]
-> (Within b (Path Rel File) -> m a)
-> m (HashMap (Within b (Path Rel File)) a)
batchLoadWithin' w f = do
xs <- getDirectoryFiles (E.ask w) (extract w)
xs' <- mapM (traverseToSnd $ f . (`within` E.ask w)) xs
return $ HM.fromList (first (`within` E.ask w) <$> xs')
batchLoadIndex :: (MonadAction m, Ix.Indexable ixs x)
=> (Path Rel File -> m x)
-> Path Rel Dir
-> [FilePattern]
-> m (Ix.IxSet ixs x)
batchLoadIndex rd dir fp = Ix.fromList . HM.elems <$> batchLoad dir fp rd
batchLoadIndex' :: (MonadAction m, Ix.Indexable ixs x)
=> Proxy ixs
-> (Path Rel File -> m x)
-> Path Rel Dir
-> [FilePattern]
-> m (Ix.IxSet ixs x)
batchLoadIndex' proxy rd dir fp = Ix.fromList . HM.elems <$> batchLoad dir fp rd