58 lines
2 KiB
Haskell
58 lines
2 KiB
Haskell
{- git-annex command
|
|
-
|
|
- Copyright 2024 Joey Hess <id@joeyh.name>
|
|
-
|
|
- Licensed under the GNU AGPL version 3 or higher.
|
|
-}
|
|
|
|
{-# LANGUAGE OverloadedStrings #-}
|
|
|
|
module Command.ExtendCluster where
|
|
|
|
import Command
|
|
import qualified Annex
|
|
import Types.Cluster
|
|
import Config
|
|
import Types.GitConfig
|
|
import qualified Remote
|
|
import qualified Command.UpdateCluster
|
|
|
|
import qualified Data.Map as M
|
|
|
|
cmd :: Command
|
|
cmd = command "extendcluster" SectionSetup "add an gateway to a cluster"
|
|
(paramPair paramRemote paramName) (withParams seek)
|
|
|
|
seek :: CmdParams -> CommandSeek
|
|
seek (remotename:clustername:[]) = Remote.byName (Just clusterremotename) >>= \case
|
|
Just clusterremote -> Remote.byName (Just remotename) >>= \case
|
|
Just gatewayremote ->
|
|
case mkClusterUUID (Remote.uuid clusterremote) of
|
|
Just cu -> do
|
|
commandAction $ start cu clustername gatewayremote
|
|
Command.UpdateCluster.seek []
|
|
Nothing -> giveup $ clusterremotename
|
|
++ " is not a cluster remote."
|
|
Nothing -> giveup $ "No remote named " ++ remotename ++ " exists."
|
|
Nothing -> giveup $ "Expected to find a cluster remote named "
|
|
++ clusterremotename
|
|
++ " that is accessed via " ++ remotename
|
|
++ ", but there is no such remote. Perhaps you need to"
|
|
++ "git fetch from " ++ remotename
|
|
++ ", or git-annex updatecluster needs to be run there?"
|
|
where
|
|
clusterremotename = remotename ++ "-" ++ clustername
|
|
seek _ = giveup "Expected two parameters, gateway and clustername."
|
|
|
|
start :: ClusterUUID -> String -> Remote -> CommandStart
|
|
start cu clustername gatewayremote = starting "extendcluster" ai si $ do
|
|
myclusters <- annexClusters <$> Annex.getGitConfig
|
|
let setcus f = setConfig f (fromUUID (fromClusterUUID cu))
|
|
unless (M.member clustername myclusters) $ do
|
|
setcus $ annexConfig ("cluster." <> encodeBS clustername)
|
|
setcus $ remoteAnnexConfig gatewayremote $
|
|
remoteGitConfigKey ClusterGatewayField
|
|
next $ return True
|
|
where
|
|
ai = ActionItemOther (Just (UnquotedString clustername))
|
|
si = SeekInput [clustername]
|