From 4e7928efb6a311f64f3bf3239b5f8e8ecfa6317c Mon Sep 17 00:00:00 2001
From: Joey Hess <joey@kitenet.net>
Date: Tue, 1 Apr 2014 18:42:32 -0400
Subject: [PATCH] propellor spin

---
 Propellor/PrivData.hs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/Propellor/PrivData.hs b/Propellor/PrivData.hs
index d97a772..e768ae9 100644
--- a/Propellor/PrivData.hs
+++ b/Propellor/PrivData.hs
@@ -33,7 +33,7 @@ getPrivData field = do
 setPrivData :: HostName -> PrivDataField -> IO ()
 setPrivData host field = do
 	putStrLn "Enter private data on stdin; ctrl-D when done:"
-	value <- hGetContentsStrict stdin
+	value <- chomp <$> hGetContentsStrict stdin
 	makePrivDataDir
 	let f = privDataFile host
 	m <- fromMaybe M.empty . readish <$> gpgDecrypt f
@@ -41,6 +41,10 @@ setPrivData host field = do
 	gpgEncrypt f (show m')
 	putStrLn "Private data set."
 	void $ boolSystem "git" [Param "add", File f]
+  where
+	chomp s
+		| end s == "\n" = chomp (beginning s)
+		| otherwise = s
 
 makePrivDataDir :: IO ()
 makePrivDataDir = createDirectoryIfMissing False privDataDir