@@ -2,9 +2,6 @@
22module Control.Logger
33 ( Logger
44 , loggerContext
5− , scrubber
6− , censor
7− , censorWith
85 , logMsg
96 , logMsgWith
107 , logInfo
@@ -35,20 +32,13 @@ module Control.Logger
3532 , compactCallStack
3633 , st
3734 , ToObject(..)
38− , withScrubber
39− , censoringWith
40− , censoring
4135 ) where
4236
4337import Control.Has
44−import Control.Lens hiding (censoring)
4538import Control.Logger.Internal
4639import Control.Logger.Orphans ()
47−import Control.Monad
4840import Control.Monad.Catch
49−import Control.Monad.IO.Class
5041import qualified Data.List as List
51−import Data.Monoid
5242import Data.Text (Text)
5343import qualified Data.Text as Text
5444import Data.Time
@@ -56,6 +46,10 @@ import GHC.Stack
5646import Katip (ToObject (..))
5747import System.Log.FastLogger
5848import Text.Shakespeare.Text (st)
49+#if MIN_VERSION_base(4,18,0)
50+import Control.Monad
51+#else
52+#endif
5953
6054logError
6155 :: (LoggingMonad r m)
@@ -109,50 +103,6 @@ logErrorWith
109103 -> m ()
110104logErrorWith logger msg = withFrozenCallStack (logMsgWith logger Error msg)
111105
112−safeReplace :: Text -> Text -> Text -> Text
113−safeReplace what with where_
114− | Text.null what = where_
115− | otherwise = Text.replace what with where_
116−
117−-- | Add a some text to scrubber
118−censorWith :: Text -> Text -> Logger -> Logger
119−censorWith replacement needle l =
120− l & scrubber <>~ Endo (safeReplace needle replacement)
121−
122−-- | Censor with "***"
123−censor :: Text -> Logger -> Logger
124−censor = censorWith "***"
125−
126−-- | Run an action with adjusted scrubber
127−withScrubber
128− :: forall r m a. (Has Logger r, MonadReader r m)
129− => (Endo Text -> Endo Text)
130− -> m a
131− -> m a
132−withScrubber scrubF = local expandScrubber
133− where
134− expandScrubber :: r -> r
135− expandScrubber = over part (over scrubber scrubF)
136−
137−-- | Run an action while censoring additional text from logs
138−censoringWith
139− :: (Has Logger r, MonadReader r m)
140− => Text
141− -> Text
142− -> m a
143− -> m a
144−censoringWith replacement txt = withScrubber (<> censored)
145− where
146− censored = Endo (safeReplace txt replacement)
147−
148−-- | Run an action while censoring additional text from logs
149−censoring
150− :: (Has Logger r, MonadReader r m)
151− => Text
152− -> m a
153− -> m a
154−censoring = censoringWith "***"
155−
156106-- | Log exceptions occured in computation and just ignore them.
157107tryLogError
158108 :: (MonadIO m, MonadCatch m, Has Logger r, MonadReader r m)
@@ -203,14 +153,14 @@ fileLogger :: MonadIO m => FilePath -> m Logger
203153fileLogger fp = fastFunc <$> liftIO (newFileLoggerSet defaultBufSize fp)
204154
205155silentLogger :: Logger
206−silentLogger = Logger mempty mempty (\_ _ _ _ -> return ())
156+silentLogger = Logger mempty (\_ _ _ _ -> return ())
207157
208158-- | Logger which flushes buffer after each message. This is useful with
209159-- stderrLogger.
210160flushingFastFunc
211161 :: LogSeverity -> LoggerSet -> Logger
212162flushingFastFunc minLogLevel loggerSet =
213− Logger mempty mempty $ \_ stack logLevel msg ->
163+ Logger mempty $ \_ stack logLevel msg ->
214164 when (logLevel >= minLogLevel) $ do
215165 ts <- getCurrentTime
216166 pushLogStr loggerSet
@@ -234,14 +184,14 @@ flushingStderrLogger =
234184-- | Make our outdated 'Logger' of fast-logger's 'LoggerSet'
235185fastFunc :: LoggerSet -> Logger
236186fastFunc loggerSet =
237− Logger mempty mempty $ \_ _ _ msg -> pushLogStr loggerSet $ toLogStr msg
187+ Logger mempty $ \_ _ _ msg -> pushLogStr loggerSet $ toLogStr msg
238188
239189type LogSource = Text
240190
241191-- | Default logger tests and same things.
242192defaultFastFunc :: LogSource -> LogSeverity -> LoggerSet -> Logger
243193defaultFastFunc loc minLogLevel loggerSet =
244− Logger mempty mempty $ \_ stack s msg -> when (s >= minLogLevel) $ do
194+ Logger mempty $ \_ stack s msg -> when (s >= minLogLevel) $ do
245195 ts <- getCurrentTime
246196 pushLogStr loggerSet $ formatLogStr ts stack loc s (toLogStr msg)
247197