git-annex/Command/InitRemote.hs

117 lines
3.1 KiB
Haskell
Raw Normal View History

2011-03-29 03:22:31 +00:00
{- git-annex command
-
- Copyright 2011 Joey Hess <joey@kitenet.net>
-
- Licensed under the GNU GPL version 3 or higher.
-}
module Command.InitRemote where
import qualified Data.Map as M
import Control.Monad (when)
import Control.Monad.State (liftIO)
import Data.Maybe
import Data.String.Utils
2011-03-29 03:22:31 +00:00
import Command
2011-03-29 18:55:59 +00:00
import qualified Annex
2011-03-29 03:22:31 +00:00
import qualified Remote
import qualified Types.Remote as R
2011-03-29 18:55:59 +00:00
import qualified GitRepo as Git
import Utility
import Types
2011-03-29 03:22:31 +00:00
import UUID
import Messages
command :: [Command]
command = [repoCommand "initremote"
(paramPair paramName $
paramOptional $ paramRepeating $ paramKeyValue) seek
"sets up a special (non-git) remote"]
seek :: [CommandSeek]
seek = [withWords start]
2011-03-29 03:22:31 +00:00
start :: CommandStartWords
start ws = notBareRepo $ do
when (null ws) $ needname
2011-03-29 18:55:59 +00:00
(u, c) <- findByName name
let fullconfig = M.union config c
t <- findType fullconfig
2011-03-29 03:22:31 +00:00
showStart "initremote" name
2011-05-15 06:02:46 +00:00
next $ perform t u $ M.union config c
2011-03-29 03:22:31 +00:00
where
name = head ws
2011-04-15 19:09:36 +00:00
config = Remote.keyValToConfig $ tail ws
needname = do
let err s = error $ "Specify a name for the remote. " ++ s
names <- remoteNames
if null names
then err ""
else err $ "Either a new name, or one of these existing special remotes: " ++ join " " names
2011-03-29 03:22:31 +00:00
perform :: R.RemoteType Annex -> UUID -> R.RemoteConfig -> CommandPerform
2011-03-29 18:55:59 +00:00
perform t u c = do
c' <- R.setup t u c
2011-05-15 06:02:46 +00:00
next $ cleanup u c'
2011-03-29 18:55:59 +00:00
cleanup :: UUID -> R.RemoteConfig -> CommandCleanup
2011-03-29 18:55:59 +00:00
cleanup u c = do
Remote.configSet u c
g <- Annex.gitRepo
logfile <- Remote.remoteLog
liftIO $ Git.run g "add" [File logfile]
liftIO $ Git.run g "commit"
[ Params "-q --allow-empty -m"
, Param "git annex initremote"
, File logfile
]
return True
2011-03-29 17:49:54 +00:00
2011-03-29 18:55:59 +00:00
{- Look up existing remote's UUID and config by name, or generate a new one -}
findByName :: String -> Annex (UUID, R.RemoteConfig)
2011-03-29 18:55:59 +00:00
findByName name = do
m <- Remote.readRemoteLog
2011-05-15 06:49:43 +00:00
maybe generate return $ findByName' name m
where
generate = do
2011-03-29 18:55:59 +00:00
uuid <- liftIO $ genUUID
2011-05-15 06:49:43 +00:00
return (uuid, M.insert nameKey name M.empty)
2011-03-29 18:55:59 +00:00
findByName' :: String -> M.Map UUID R.RemoteConfig -> Maybe (UUID, R.RemoteConfig)
2011-03-29 18:55:59 +00:00
findByName' n m = if null matches then Nothing else Just $ head matches
2011-03-29 17:49:54 +00:00
where
matches = filter (matching . snd) $ M.toList m
matching c = case M.lookup nameKey c of
Nothing -> False
Just n'
| n' == n -> True
| otherwise -> False
remoteNames :: Annex [String]
remoteNames = do
m <- Remote.readRemoteLog
return $ catMaybes $ map ((M.lookup nameKey) . snd) $ M.toList m
2011-03-29 18:55:59 +00:00
{- find the specified remote type -}
findType :: R.RemoteConfig -> Annex (R.RemoteType Annex)
2011-05-15 06:49:43 +00:00
findType config = maybe unspecified specified $ M.lookup typeKey config
where
unspecified = error "Specify the type of remote with type="
specified s = case filter (findtype s) Remote.remoteTypes of
2011-03-29 18:55:59 +00:00
[] -> error $ "Unknown remote type " ++ s
(t:_) -> return t
findtype s i = R.typename i == s
2011-03-29 18:55:59 +00:00
2011-03-29 17:49:54 +00:00
{- The name of a configured remote is stored in its config using this key. -}
nameKey :: String
nameKey = "name"
2011-03-29 18:55:59 +00:00
{- The type of a remote is stored in its config using this key. -}
typeKey :: String
typeKey = "type"