aboutsummaryrefslogtreecommitdiff
path: root/src/Rahm/Desktop/Logger.hs
blob: 593f8205d585708fa27bf18dcd6a869616430ab7 (plain) (blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
module Rahm.Desktop.Logger
  ( getLogLevel,
    setLogLevel,
    logs,
    LogLevel (..),
  )
where

import Control.Monad (forM_, join, when)
import Data.Time.LocalTime (getZonedTime)
import Text.Printf (PrintfArg, PrintfType, printf)
import XMonad
  ( ExtensionClass (..),
    StateExtension (PersistentExtension),
    X,
    XConf (config),
    XConfig (logHook),
    asks,
    io,
  )
import qualified XMonad.Util.ExtensibleState as XS (get, put)

data LogLevel = Trace | Debug | Info | Warn | Error | Fatal
  deriving (Show, Read, Ord, Eq, Enum, Bounded)

newtype LoggerState = LoggerState
  { logLevel :: LogLevel
  }
  deriving (Show, Read, Eq)

instance ExtensionClass LoggerState where
  initialValue = LoggerState Info
  extensionType = PersistentExtension

class (PrintfType (Printf t)) => LoggerType t where
  type Printf t :: *
  gp :: Printf t -> (String -> X ()) -> t

instance (PrintfArg a, LoggerType b) => LoggerType (a -> b) where
  type Printf (a -> b) = a -> Printf b
  gp g f a = gp (g a) f

instance (a ~ ()) => LoggerType (X a) where
  type Printf (X a) = String
  gp str fn = fn str

getLogLevel :: X LogLevel
getLogLevel = logLevel <$> XS.get

setLogLevel :: LogLevel -> X ()
setLogLevel ll = do
  XS.put $ LoggerState ll
  join $ asks (logHook . config)

logs :: (LoggerType r) => LogLevel -> String -> r
logs ll fmt = gp (printf fmt) $ \ss -> do
  LoggerState ll' <- XS.get
  io $ do
    zoneTime <- getZonedTime
    when (ll >= ll') $
      forM_ (lines ss) $ \s ->
        putStrLn (printf "[%s %s] - %s" (take 23 $ show zoneTime) (show ll) s)