41 lines
1 KiB
Haskell
41 lines
1 KiB
Haskell
{- 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 Command
|
|
import Annex.Content
|
|
import Remote.Helper.Ssh (contentLockedMarker)
|
|
import Utility.SimpleProtocol
|
|
|
|
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 (commandAction . start)
|
|
|
|
-- First, lock the content, then print out "OK".
|
|
-- 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 (giveup "bad key") (deserializeKey ks)
|
|
locksuccess = liftIO $ do
|
|
putStrLn contentLockedMarker
|
|
hFlush stdout
|
|
_ <- getProtocolLine stdin
|
|
return True
|
|
start _ = giveup "Specify exactly 1 key."
|