% Copyright (C) 2008 David Roundy
%
% This program is free software; you can redistribute it and/or modify
% it under the terms of the GNU General Public License as published by
% the Free Software Foundation; either version 2, or (at your option)
% any later version.
%
% This program is distributed in the hope that it will be useful,
% but WITHOUT ANY WARRANTY; without even the implied warranty of
% MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
% GNU General Public License for more details.
%
% You should have received a copy of the GNU General Public License
% along with this program; see the file COPYING. If not, write to
% the Free Software Foundation, Inc., 51 Franklin Street, Fifth Floor,
% Boston, MA 02110-1301, USA.
\darcsCommand{transfer-mode}
\begin{code}
module Darcs.Commands.TransferMode ( transferMode ) where
import Prelude hiding ( catch )
import Control.Exception.Extensible ( catch )
import System.IO ( stdout, hFlush )
import Darcs.Utils ( withCurrentDirectory, prettyException )
import Darcs.Commands ( DarcsCommand(..), nodefaults )
import Darcs.Arguments ( DarcsFlag, workingRepoDir )
import Darcs.Repository ( amInRepository )
import Progress ( setProgressMode )
import Darcs.Global ( darcsdir )
import qualified Data.ByteString as B (hPut, readFile, length, ByteString)
transferModeDescription :: String
transferModeDescription = "Internal command for efficient ssh transfers."
transferModeHelp :: String
transferModeHelp =
"When pulling from or pushing to a remote repository over ssh, if both\n" ++
"the local and remote ends have Darcs 2, the `transfer-mode' command\n" ++
"will be invoked on the remote end. This allows Darcs to intelligently\n" ++
"transfer information over a single ssh connection.\n" ++
"\n" ++
"If either end runs Darcs 1, a separate ssh connection will be created\n" ++
"for each transfer. As well as being less efficient, this means users\n" ++
"who do not run ssh-agent will be prompted for the ssh password tens or\n" ++
"hundreds of times!\n"
transferMode :: DarcsCommand
transferMode = DarcsCommand {commandName = "transfer-mode",
commandHelp = transferModeHelp,
commandDescription = transferModeDescription,
commandExtraArgs = 0,
commandExtraArgHelp = [],
commandGetArgPossibilities = return [],
commandCommand = transferModeCmd,
commandPrereq = amInRepository,
commandArgdefaults = nodefaults,
commandAdvancedOptions = [],
commandBasicOptions = [workingRepoDir]}
transferModeCmd :: [DarcsFlag] -> [String] -> IO ()
transferModeCmd _ _ = do setProgressMode False
putStrLn "Hello user, I am darcs transfer mode"
hFlush stdout
withCurrentDirectory darcsdir $ transfer
transfer :: IO ()
transfer = do 'g':'e':'t':' ':fn <- getLine
x <- readfile fn
case x of
Right c -> do putStrLn $ "got " ++ fn
putStrLn $ show $ B.length c
B.hPut stdout c
hFlush stdout
Left e -> do putStrLn $ "error " ++ fn
putStrLn $ show e
hFlush stdout
transfer
readfile :: String -> IO (Either String B.ByteString)
readfile fn = (Right `fmap` B.readFile fn) `catch` (\e -> return $ Left (prettyException e))
\end{code}