{-# LANGUAGE OverloadedStrings #-}

module Network.QUIC.Logger (
    Builder
  , DebugLogger
  , bhow
  , stdoutLogger
  , dirDebugLogger
  ) where

import System.FilePath
import System.Log.FastLogger
import Data.ByteString.Builder (byteString, toLazyByteString)
import qualified Data.ByteString.Char8 as C8
import qualified Data.ByteString.Lazy.Char8 as BL

import Network.QUIC.Imports
import Network.QUIC.Types

-- | A type for debug logger.
type DebugLogger = Builder -> IO ()

bhow :: Show a => a -> Builder
bhow :: forall a. Show a => a -> Builder
bhow = ByteString -> Builder
byteString forall b c a. (b -> c) -> (a -> b) -> a -> c
. String -> ByteString
C8.pack forall b c a. (b -> c) -> (a -> b) -> a -> c
. forall a. Show a => a -> String
show

stdoutLogger :: DebugLogger
stdoutLogger :: DebugLogger
stdoutLogger Builder
b = ByteString -> IO ()
BL.putStrLn forall a b. (a -> b) -> a -> b
$ Builder -> ByteString
toLazyByteString Builder
b

dirDebugLogger :: Maybe FilePath -> CID -> IO (DebugLogger, IO ())
dirDebugLogger :: Maybe String -> CID -> IO (DebugLogger, IO ())
dirDebugLogger Maybe String
Nothing CID
_ = do
    let dLog :: p -> m ()
dLog ~p
_ = forall (m :: * -> *) a. Monad m => a -> m a
return ()
        clean :: IO ()
clean = forall (m :: * -> *) a. Monad m => a -> m a
return ()
    forall (m :: * -> *) a. Monad m => a -> m a
return (forall {m :: * -> *} {p}. Monad m => p -> m ()
dLog, IO ()
clean)
dirDebugLogger (Just String
dir) CID
cid = do
    let file :: String
file = String
dir String -> String -> String
</> (forall a. Show a => a -> String
show CID
cid forall a. Semigroup a => a -> a -> a
<> String
".txt")
    (LogStr -> IO ()
fastlogger, IO ()
clean) <- forall v. LogType' v -> IO (v -> IO (), IO ())
newFastLogger1 (String -> BufSize -> LogType' LogStr
LogFileNoRotate String
file BufSize
4096)
    let dLog :: DebugLogger
dLog Builder
msg = do
            LogStr -> IO ()
fastlogger (forall msg. ToLogStr msg => msg -> LogStr
toLogStr Builder
msg forall a. Semigroup a => a -> a -> a
<> LogStr
"\n")
            DebugLogger
stdoutLogger Builder
msg
    forall (m :: * -> *) a. Monad m => a -> m a
return (DebugLogger
dLog, IO ()
clean)