git-annex/Command/RecvKey.hs
Joey Hess c91929f693 add whenM and unlessM
Just more golfing.. I am pretty sure something in a library somewhere can
do this, but I have been unable to find it.
2011-05-17 03:13:11 -04:00

37 lines
796 B
Haskell

{- git-annex command
-
- Copyright 2010 Joey Hess <joey@kitenet.net>
-
- Licensed under the GNU GPL version 3 or higher.
-}
module Command.RecvKey where
import Control.Monad.State (liftIO)
import System.Exit
import Command
import CmdLine
import Content
import Utility
import RsyncFile
command :: [Command]
command = [repoCommand "recvkey" paramKey seek
"runs rsync in server mode to receive content"]
seek :: [CommandSeek]
seek = [withKeys start]
start :: CommandStartKey
start key = do
whenM (inAnnex key) $ error "key is already present in annex"
ok <- getViaTmp key (liftIO . rsyncServerReceive)
if ok
then do
-- forcibly quit after receiving one key,
-- and shutdown cleanly so queued git commands run
_ <- shutdown
liftIO exitSuccess
else liftIO exitFailure