Add --list-fields to list a host's currently set privdata fields.

3 files changed, 11 insertions, 0 deletions

diff --git a/src/Propellor/CmdLine.hs b/src/Propellor/CmdLine.hs
index c084355b..1027fd8d 100644
--- a/src/Propellor/CmdLine.hs
+++ b/src/Propellor/CmdLine.hs
@@ -30,6 +30,7 @@ usage = do
 		, "  propellor --set hostname field"
 		, "  propellor --dump hostname field"
 		, "  propellor --edit hostname field"
+		, "  propellor --list-fields hostname"
 		]
 	exitFailure
 
@@ -43,6 +44,7 @@ processCmdLine = go =<< getArgs
 	go ("--set":h:f:[]) = withprivfield f (return . Set h)
 	go ("--dump":h:f:[]) = withprivfield f (return . Dump h)
 	go ("--edit":h:f:[]) = withprivfield f (return . Edit h)
+	go ("--list-fields":h:[]) = return $ ListFields h
 	go ("--continue":s:[]) = case readish s of
 		Just cmdline -> return $ Continue cmdline
 		Nothing -> errorMessage "--continue serialization failure"
@@ -74,6 +76,7 @@ defaultMain hostlist = do
 	go _ (Set hn field) = setPrivData hn field
 	go _ (Dump hn field) = dumpPrivData hn field
 	go _ (Edit hn field) = editPrivData hn field
+	go _ (ListFields hn) = listPrivDataFields hn
 	go _ (AddKey keyid) = addKey keyid
 	go _ (Chain hn) = withhost hn $ \h -> do
 		r <- runPropellor h $ ensureProperties $ hostProperties h
diff --git a/src/Propellor/PrivData.hs b/src/Propellor/PrivData.hs
index fec6acc3..c2af4284 100644
--- a/src/Propellor/PrivData.hs
+++ b/src/Propellor/PrivData.hs
@@ -68,6 +68,13 @@ editPrivData host field = do
 		readFile f
 	setPrivDataTo host field v'
 
+listPrivDataFields :: HostName -> IO ()
+listPrivDataFields host = do
+	putStrLn (host ++ "'s currently set privdata fields:")
+	mapM_ list . M.keys =<< decryptPrivData host
+  where
+	list = putStrLn . ("\t" ++) . shellEscape . show
+
 setPrivDataTo :: HostName -> PrivDataField -> String -> IO ()
 setPrivDataTo host field value = do
 	makePrivDataDir
diff --git a/src/Propellor/Types.hs b/src/Propellor/Types.hs
index 740996be..59652f66 100644
--- a/src/Propellor/Types.hs
+++ b/src/Propellor/Types.hs
@@ -138,6 +138,7 @@ data CmdLine
 	| Set HostName PrivDataField
 	| Dump HostName PrivDataField
 	| Edit HostName PrivDataField
+	| ListFields HostName
 	| AddKey String
 	| Continue CmdLine
 	| Chain HostName