60 lines
1.9 KiB
Haskell
60 lines
1.9 KiB
Haskell
|
{- git-annex command, used internally by old versions of assistant;
|
||
|
|
- kept around for now so running daemons don't break when upgraded
|
||
|
|
-
|
||
|
|
- Copyright 2012 Joey Hess <joey@kitenet.net>
|
||
|
|
-
|
||
|
|
- Licensed under the GNU GPL version 3 or higher.
|
||
|
|
-}
|
||
|
|
|
||
|
|
module Command.TransferKey where
|
||
|
|
|
||
|
|
import Common.Annex
|
||
|
|
import Command
|
||
|
|
import Annex.Content
|
||
|
|
import Logs.Location
|
||
|
|
import Logs.Transfer
|
||
|
|
import qualified Remote
|
||
|
|
import Types.Remote
|
||
|
|
import qualified Command.Move
|
||
|
|
import qualified Option
|
||
|
|
|
||
|
|
def :: [Command]
|
||
|
|
def = [withOptions options $
|
||
|
|
noCommit $ command "transferkey" paramKey seek SectionPlumbing
|
||
|
|
"transfers a key from or to a remote"]
|
||
|
|
|
||
|
|
options :: [Option]
|
||
|
|
options = [fileOption, Command.Move.fromOption, Command.Move.toOption]
|
||
|
|
|
||
|
|
fileOption :: Option
|
||
|
|
fileOption = Option.field [] "file" paramFile "the associated file"
|
||
|
|
|
||
|
|
seek :: [CommandSeek]
|
||
|
|
seek = [withField Command.Move.toOption Remote.byNameWithUUID $ \to ->
|
||
|
|
withField Command.Move.fromOption Remote.byNameWithUUID $ \from ->
|
||
|
|
withField fileOption return $ \file ->
|
||
|
|
withKeys $ start to from file]
|
||
|
|
|
||
|
|
start :: Maybe Remote -> Maybe Remote -> AssociatedFile -> Key -> CommandStart
|
||
|
|
start to from file key =
|
||
|
|
case (from, to) of
|
||
|
|
(Nothing, Just dest) -> next $ toPerform dest key file
|
||
|
|
(Just src, Nothing) -> next $ fromPerform src key file
|
||
|
|
_ -> error "specify either --from or --to"
|
||
|
|
|
||
|
|
toPerform :: Remote -> Key -> AssociatedFile -> CommandPerform
|
||
|
|
toPerform remote key file = go $
|
||
|
|
upload (uuid remote) key file forwardRetry $ \p -> do
|
||
|
|
ok <- Remote.storeKey remote key file p
|
||
|
|
when ok $
|
||
|
|
Remote.logStatus remote key InfoPresent
|
||
|
|
return ok
|
||
|
|
|
||
|
|
fromPerform :: Remote -> Key -> AssociatedFile -> CommandPerform
|
||
|
|
fromPerform remote key file = go $
|
||
|
|
download (uuid remote) key file forwardRetry $ \p ->
|
||
|
|
getViaTmp key $ \t -> Remote.retrieveKeyFile remote key file t p
|
||
|
|
|
||
|
|
go :: Annex Bool -> CommandPerform
|
||
|
|
go a = ifM a ( liftIO exitSuccess, liftIO exitFailure)
|