summaryrefslogtreecommitdiff
path: root/Command/LockContent.hs
blob: 72b2bb096e9b0bd9ed389acd1ddd2f02b6c92761 (plain)
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
{- git-annex-shell command
 -
 - Copyright 2015 Joey Hess <id@joeyh.name>
 -
 - Licensed under the GNU GPL version 3 or higher.
 -}

module Command.LockContent where

import Common.Annex
import Command
import Annex.Content
import Types.Key
import Remote.Helper.Ssh (contentLockedMarker)

cmd :: Command
cmd = noCommit $ 
	command "lockcontent" SectionPlumbing 
		"locks key's content in the annex, preventing it being dropped"
		paramKey
		(withParams seek)

seek :: CmdParams -> CommandSeek
seek = withWords start

-- First, lock the content. Then, make sure the content is actually
-- present, and print out a "1". Wait for the caller to send a line before
-- dropping the lock.
start :: [String] -> CommandStart
start [ks] = do
	ok <- lockContentShared k (const locksuccess)
		`catchNonAsync` (const $ return False)
	liftIO $ if ok
		then exitSuccess
		else exitFailure
  where
	k = fromMaybe (error "bad key") (file2key ks)
	locksuccess = ifM (inAnnex k)
		( liftIO $ do
			putStrLn contentLockedMarker
			hFlush stdout
			_ <- getLine
			return True
		, return False
		)
start _ = error "Specify exactly 1 key."