aboutsummaryrefslogtreecommitdiff
path: root/hsm-log/Hsm/Log.hs
blob: 532191007dc0f710fc5ac185d4de939d74e5d190 (plain)
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
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
{-# LANGUAGE AllowAmbiguousTypes #-}
{-# LANGUAGE TypeFamilies #-}
{-# LANGUAGE UndecidableInstances #-}

module Hsm.Log
  ( Severity (Attention, Info, Trace)
  , Log
  , logMsg
  , runLog
  )
where

import Control.Monad (when)
import Data.Function (applyWhen)
import Data.Proxy (Proxy (Proxy))
import Data.Time.Clock (getCurrentTime)
import Data.Time.ISO8601 (formatISO8601Millis)
import Effectful (Dispatch (Static), DispatchOf, Eff, Effect, IOE, (:>))
import Effectful.Dispatch.Static
  ( SideEffects (WithSideEffects)
  , StaticRep
  , evalStaticRep
  , getStaticRep
  , unsafeEff_
  )
import GHC.TypeLits (KnownSymbol, Symbol, symbolVal)
import GHC.TypeLits.Printf (printf)
import String.ANSI (red)

data Severity
  = Attention
  | Info
  | Trace
  deriving (Eq, Ord, Show)

data Log (d :: Symbol) (a :: * -> *) (b :: *)

type instance DispatchOf (Log d) = Static WithSideEffects

newtype instance StaticRep (Log d)
  = Log Severity

logMsg
  :: forall d es
   . (KnownSymbol d, Log d :> es)
  => Severity
  -> String
  -> Eff es ()
logMsg severity message = do
  Log level <- getStaticRep
  unsafeEff_ . when (severity <= level) $ do
    time <- formatISO8601Millis <$> getCurrentTime
    putStrLn . applyWhen (severity == Attention) red $
      printf "%s %s [%s] %s" time (symbolVal $ Proxy @d) (show level) message

runLog
  :: forall d es a
   . IOE :> es
  => Severity
  -> Eff (Log d : es) a
  -> Eff es a
runLog = evalStaticRep . Log

class Logs (ds :: [Symbol]) (es :: [Effect]) where
  type Insert ds es :: [Effect]
  runLogs :: Severity -> Eff (Insert ds es) a -> Eff es a

instance Logs ('[] :: [Symbol]) (es :: [Effect]) where
  type Insert '[] es = es
  runLogs _ = id

instance
  (IOE :> Insert ds es, KnownSymbol d, Logs ds es)
  => Logs (d : ds :: [Symbol]) (es :: [Effect])
  where
  type Insert (d : ds) es = Log d : Insert ds es
  runLogs level = runLogs @ds level . runLog @d level