1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
|
{- 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
let seeker = AnnexedFileSeeker
{ startAction = start s
, checkContentPresent = Nothing
, usesLocationLog = False
}
withFilesInGitAnnex ww seeker
=<< workTreeItems ww (inprogressFiles o)
where
ww = WarnUnmatchLsFiles
start :: S.Set Key -> SeekInput -> RawFilePath -> Key -> CommandStart
start s _si _file k
| S.member k s = start' k
| otherwise = stop
start' :: Key -> CommandStart
start' k = startingCustomOutput k $ do
tmpf <- fromRawFilePath <$> fromRepo (gitAnnexTmpObjectLocation k)
whenM (liftIO $ doesFileExist tmpf) $
liftIO $ putStrLn tmpf
next $ return True
|
