1 files changed, 40 insertions, 0 deletions
diff --git a/Propellor/Message.hs b/Propellor/Message.hs
new file mode 100644
index 00000000..a7ceff91
--- /dev/null
+++ b/Propellor/Message.hs
@@ -0,0 +1,40 @@
+module Propellor.Message where
+
+import System.Console.ANSI
+import System.IO
+
+import Propellor.Types
+
+-- | Shows a message while performing an action, with a colored status
+-- display.
+actionMessage :: ActionResult r => Desc -> IO r -> IO r
+actionMessage desc a = do
+	setTitle desc
+	showdesc
+	putStrLn "starting"
+	hFlush stdout
+
+	r <- a
+
+	let (msg, intensity, color) = getActionResult r
+	showdesc
+	setSGR [SetColor Foreground intensity color]
+	putStrLn msg
+	setSGR []
+	hFlush stdout
+
+	return r
+  where
+	showdesc = putStr $ desc ++ " ... "
+
+warningMessage :: String -> IO ()
+warningMessage s = do
+	setSGR [SetColor Foreground Vivid Red]
+	putStrLn $ "** warning: " ++ s
+	setSGR []
+	hFlush stdout
+
+errorMessage :: String -> IO a
+errorMessage s = do
+	warningMessage s
+	error "Propellor failed!"