git-annex/Command/Wanted.hs
2024-08-11 14:37:18 -04:00

60 lines
1.6 KiB
Haskell

{- git-annex command
-
- Copyright 2013-2015 Joey Hess <id@joeyh.name>
-
- Licensed under the GNU AGPL version 3 or higher.
-}
module Command.Wanted where
import Command
import qualified Remote
import Logs.PreferredContent
import Types.StandardGroups
import Utility.SafeOutput
import qualified Data.Map as M
cmd :: Command
cmd = cmd' "wanted" "get or set preferred content expression"
preferredContentMapRaw
preferredContentSet
cmd'
:: String
-> String
-> Annex (M.Map UUID PreferredContentExpression)
-> (UUID -> PreferredContentExpression -> Annex ())
-> Command
cmd' name desc getter setter = noMessages $
command name SectionSetup desc pdesc
(withParams' seek completeRemotes)
where
pdesc = paramPair paramRepository (paramOptional paramExpression)
seek = withWords (commandAction . start)
start (rname:[]) = do
u <- Remote.nameToUUID rname
startingCustomOutput (ActionItemOther Nothing) $
performGet getter u
start ps@(rname:expr:[]) = do
u <- Remote.nameToUUID rname
let si = SeekInput ps
let ai = ActionItemOther (Just (UnquotedString rname))
startingUsualMessages name ai si $
performSet setter expr u
start _ = giveup "Specify a repository."
performGet :: Ord a => Annex (M.Map a PreferredContentExpression) -> a -> CommandPerform
performGet getter a = do
m <- getter
liftIO $ putStrLn $ safeOutput $ fromMaybe "" $ M.lookup a m
next $ return True
performSet :: (a -> PreferredContentExpression -> Annex ()) -> String -> a -> CommandPerform
performSet setter expr a = case checkPreferredContentExpression expr of
Just e -> giveup $ "Parse error: " ++ e
Nothing -> do
setter a expr
next $ return True