Add minSeverity option to local-cluster CLI

cardano-foundation · May 9, 2024 · 2ee86f5 · 2ee86f5
1 parent 4e9f63b
commit 2ee86f5
Show file tree

Hide file tree

Showing 2 changed files with 32 additions and 1 deletion.
diff --git a/lib/local-cluster/exe/local-cluster.hs b/lib/local-cluster/exe/local-cluster.hs
@@ -242,6 +242,7 @@ main = withUtf8 $ do
         , clusterLogs
         , nodeToClientSocket
         , httpService
+        , minSeverity
         } <-
         parseCommandLineOptions
     evalContT $ do
@@ -250,7 +251,7 @@ main = withUtf8 $ do
             ContT
                 $ newToTextTracer
                     (toFilePath . absFileOf <$> clusterLogs)
-                    Nothing
+                    minSeverity
 
         -- Create a temporary directory for the cluster
         clusterPath <-

diff --git a/lib/local-cluster/lib/Cardano/Wallet/Launch/Cluster/CommandLine.hs b/lib/local-cluster/lib/Cardano/Wallet/Launch/Cluster/CommandLine.hs
@@ -11,6 +11,9 @@ where
 
 import Prelude
 
+import Cardano.BM.Data.Severity
+    ( Severity
+    )
 import Cardano.Wallet.Launch.Cluster.FileOf
     ( Absolutizer (..)
     , DirOf (..)
@@ -51,10 +54,13 @@ import System.Path
     ( absRel
     )
 
+import qualified Cardano.BM.Data.Severity as Severity
+
 data CommandLineOptions = CommandLineOptions
     { clusterConfigsDir :: DirOf "cluster-configs"
     , clusterDir :: Maybe (DirOf "cluster")
     , clusterLogs :: Maybe (FileOf "cluster-logs")
+    , minSeverity :: Maybe Severity
     , nodeToClientSocket :: FileOf "node-to-client-socket"
     , httpService :: ServiceConfiguration
     }
@@ -69,12 +75,36 @@ parseCommandLineOptions = do
                 <$> clusterConfigsDirParser absolutizer
                 <*> clusterDirParser absolutizer
                 <*> clusterLogsParser absolutizer
+                <*> minSeverityParser
                 <*> nodeToClientSocketParser absolutizer
                 <*> monitoringParser
                 <**> helper
             )
             (progDesc "Local Cluster for testing")
 
+minSeverityParser :: Parser (Maybe Severity)
+minSeverityParser =
+    optional
+        $ option
+            parse
+            ( long "min-severity"
+                <> metavar "MIN_SEVERITY"
+                <> help "Minimum severity level for logging"
+            )
+  where
+    parse = do
+        s :: String <- auto
+        case s of
+            "Debug" -> pure Severity.Debug
+            "Info" -> pure Severity.Info
+            "Notice" -> pure Severity.Notice
+            "Warning" -> pure Severity.Warning
+            "Error" -> pure Severity.Error
+            "Critical" -> pure Severity.Critical
+            "Alert" -> pure Severity.Alert
+            "Emergency" -> pure Severity.Emergency
+            _ -> fail "Invalid severity level"
+
 monitoringParser :: Parser ServiceConfiguration
 monitoringParser =
     mkServiceConfiguration