propellor spin

2 files changed, 39 insertions, 3 deletions

diff --git a/src/Propellor/PrivData.hs b/src/Propellor/PrivData.hs
index ac7228f8..10965fe3 100644
--- a/src/Propellor/PrivData.hs
+++ b/src/Propellor/PrivData.hs
@@ -26,6 +26,7 @@ import Utility.SafeCommand
 import Utility.Misc
 import Utility.FileMode
 import Utility.Env
+import Utility.Table
 
 -- | Allows a Property to access the value of a specific PrivDataField,
 -- for use in a specific Context.
@@ -90,10 +91,17 @@ editPrivData field context = do
 listPrivDataFields :: IO ()
 listPrivDataFields = do
 	m <- decryptPrivData
-	putStrLn ("\nAll currently set privdata fields:")
-	mapM_ list $ M.keys m
+	putStrLn "\n"
+	let rows = map mkrow (M.keys m)
+	let table = tableWithHeader header rows
+	putStr $ unlines $ formatTable table
   where
-	list = putStrLn . ("\t" ++) . shellEscape . show
+  	header = ["Field", "Context", "Hosts"]
+	mkrow (field, (Context context)) =
+		[ shellEscape $ show field
+		, shellEscape context
+		, "xxx"
+		]
 
 setPrivDataTo :: PrivDataField -> Context -> PrivData -> IO ()
 setPrivDataTo field context value = do
diff --git a/src/Utility/Table.hs b/src/Utility/Table.hs
new file mode 100644
index 00000000..910038e8
--- /dev/null
+++ b/src/Utility/Table.hs
@@ -0,0 +1,28 @@
+{- text based table generation
+ -
+ - Copyright 2014 Joey Hess <joey@kitenet.net>
+ -
+ - License: BSD-2-clause
+ -}
+
+module Utility.Table where
+
+type Table = [[String]]
+
+-- | A table with a header that is set off with lines under each
+-- header item.
+tableWithHeader :: [String] -> [[String]] -> Table
+tableWithHeader header rows = header : map linesep header : rows
+  where
+	linesep = map (const '-')
+
+-- | Formats a table to lines, automatically padding rows to the same size.
+formatTable :: Table -> [String]
+formatTable table = map (\r -> unwords (map pad (zip r rowsizes))) table
+  where
+	pad (cell, size) = cell ++ take (size - length cell) padding
+	padding = repeat ' '
+	rowsizes = sumrows (map (map length) table)
+	sumrows [] = repeat 0
+	sumrows [r] = r
+	sumrows (r1:r2:rs) = sumrows $ map (uncurry max) (zip r1 r2) : rs