This commit is contained in:
Joey Hess 2014-11-18 18:44:24 -04:00
parent 47bcd4e130
commit 66466a953d
2 changed files with 14 additions and 15 deletions

View File

@ -3,10 +3,6 @@ module Propellor.CmdLine where
import System.Environment (getArgs) import System.Environment (getArgs)
import Data.List import Data.List
import System.Exit import System.Exit
import System.Log.Logger
import System.Log.Formatter
import System.Log.Handler (setFormatter, LogHandler)
import System.Log.Handler.Simple
import System.PosixCompat import System.PosixCompat
import Control.Exception (bracket) import Control.Exception (bracket)
import System.Posix.IO import System.Posix.IO
@ -343,14 +339,3 @@ gitPush hin hout = void $ fromstdin `concurrently` tostdout
B.hPut toh b B.hPut toh b
hFlush toh hFlush toh
connect fromh toh connect fromh toh
checkDebugMode :: IO ()
checkDebugMode = go =<< getEnv "PROPELLOR_DEBUG"
where
go (Just "1") = do
f <- setFormatter
<$> streamHandler stderr DEBUG
<*> pure (simpleLogFormatter "[$time] $msg")
updateGlobalLogger rootLoggerName $
setLevel DEBUG . setHandlers [f]
go _ = noop

View File

@ -5,6 +5,9 @@ module Propellor.Message where
import System.Console.ANSI import System.Console.ANSI
import System.IO import System.IO
import System.Log.Logger import System.Log.Logger
import System.Log.Formatter
import System.Log.Handler (setFormatter, LogHandler)
import System.Log.Handler.Simple
import "mtl" Control.Monad.Reader import "mtl" Control.Monad.Reader
import Data.Maybe import Data.Maybe
import Control.Applicative import Control.Applicative
@ -98,3 +101,14 @@ colorLine h intensity color msg = do
-- | Causes a debug message to be displayed when PROPELLOR_DEBUG=1 -- | Causes a debug message to be displayed when PROPELLOR_DEBUG=1
debug :: [String] -> IO () debug :: [String] -> IO ()
debug = debugM "propellor" . unwords debug = debugM "propellor" . unwords
checkDebugMode :: IO ()
checkDebugMode = go =<< getEnv "PROPELLOR_DEBUG"
where
go (Just "1") = do
f <- setFormatter
<$> streamHandler stderr DEBUG
<*> pure (simpleLogFormatter "[$time] $msg")
updateGlobalLogger rootLoggerName $
setLevel DEBUG . setHandlers [f]
go _ = noop