88a7fb5cbb
This removes all calls to inAnnex, except for some involving --batch. It may be that the batch code could get a similar speedup, but I don't know if people habitually pass a huge number of files through --batch that git-annex does not need to do anything to process, so I skipped it for now. A few calls to ifAnnexed remain, and might be worth doing more to convert. In particular, Command.Sync has one that would probably speed it up by a good amount. (also removed some dead code from Command.Lock)
57 lines
1.4 KiB
Haskell
57 lines
1.4 KiB
Haskell
{- git-annex command
|
|
-
|
|
- Copyright 2017 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
module Command.Inprogress where
|
|
|
|
import Command
|
|
import Annex.Transfer
|
|
|
|
import qualified Data.Set as S
|
|
|
|
cmd :: Command
|
|
cmd = noCommit $ noMessages $ command "inprogress" SectionQuery
|
|
"access files while they're being downloaded"
|
|
paramPaths (seek <$$> optParser)
|
|
|
|
data InprogressOptions = InprogressOptions
|
|
{ inprogressFiles :: CmdParams
|
|
, keyOptions :: Maybe KeyOptions
|
|
}
|
|
|
|
optParser :: CmdParamsDesc -> Parser InprogressOptions
|
|
optParser desc = InprogressOptions
|
|
<$> cmdParams desc
|
|
<*> optional (parseAllOption <|> parseSpecificKeyOption)
|
|
|
|
seek :: InprogressOptions -> CommandSeek
|
|
seek o = do
|
|
ts <- map (transferKey . fst) <$> getTransfers
|
|
case keyOptions o of
|
|
Just WantAllKeys ->
|
|
forM_ ts $ commandAction . start'
|
|
Just (WantSpecificKey k)
|
|
| k `elem` ts -> commandAction (start' k)
|
|
| otherwise -> commandAction stop
|
|
_ -> do
|
|
let s = S.fromList ts
|
|
withFilesInGitAnnex ww
|
|
(commandAction' (start s))
|
|
=<< workTreeItems ww (inprogressFiles o)
|
|
where
|
|
ww = WarnUnmatchLsFiles
|
|
|
|
start :: S.Set Key -> RawFilePath -> Key -> CommandStart
|
|
start s _file k
|
|
| S.member k s = start' k
|
|
| otherwise = stop
|
|
|
|
start' :: Key -> CommandStart
|
|
start' k = startingCustomOutput k $ do
|
|
tmpf <- fromRepo $ gitAnnexTmpObjectLocation k
|
|
whenM (liftIO $ doesFileExist tmpf) $
|
|
liftIO $ putStrLn tmpf
|
|
next $ return True
|