BkrLogging.hs

module System.Bkr.BkrLogging ( setupLogging
                      , logDebug
                      , logNotice
                      , logCritical
                      ) where

import System.Bkr.BkrConfig (getLogFileLocation,  getLogFileMaximumSize)

import System.Log.Logger
import System.Log.Handler.Simple
import System.Log.Formatter
import System.Log.Handler (setFormatter)
import Data.Maybe (fromJust)
import System.Posix.Files (getFileStatus, fileSize)
import System.Directory (removeFile, doesFileExist)
import Control.Monad (when)

--import System.Log.Handler.Growl

setupLogging :: Priority -> IO ()
setupLogging logLevel = do
     -- Get log file location
     logFilePath <- getLogFileLocation
     --print $ show logFilePath
     -- Check if we got a log file, if not set up logging to stderr otherwise set up file logging
     if logFilePath == Nothing
        then do
              putStrLn "Could not find log file, please make sure to set the log file path correctly in bkr settings. Will log to stderr."
              updateGlobalLogger "bkrfile" (setLevel logLevel)
        else do
              -- If the log file exists, check that is hasn't reached maximum size and recycle if it has
              logFileExists <- doesFileExist (fromJust logFilePath)
              when logFileExists (do
                                   logFileMaxSize <- getLogFileMaximumSize
                                   fileStatus <- getFileStatus (fromJust logFilePath)
                                   let logFileSize = fileSize fileStatus
                                   when ((read (show logFileSize) :: Int) > logFileMaxSize) (do
                                                                                              removeFile $ fromJust logFilePath
                                                                                              print "The log file reached maximum allowed size and has been recycled."))

              -- File handler
              h <- fileHandler (fromJust logFilePath) DEBUG >>= \lh -> return $ setFormatter lh (tfLogFormatter "%F %X.%q" "$time|$loggername|$prio: $msg")
        
              -- Update global logger with handler and log level
              updateGlobalLogger "bkrfile" (addHandler h)
              updateGlobalLogger "bkrfile" (setLevel logLevel)
        

logDebug :: String -> IO ()
logDebug = debugM "bkrfile"

logNotice :: String -> IO ()
logNotice = noticeM "bkrfile"

logCritical :: String -> IO ()
logCritical = criticalM "bkrfile"