Add support for dynmically setting log level

Author: peacekeeper

src/main/java/com/godiddy/cli/GodiddyCLIApplication.java

@@ -1,7 +1,10 @@
 package com.godiddy.cli;
 
+import com.godiddy.cli.config.LogLevel;
+import org.apache.logging.log4j.Level;
 import org.apache.logging.log4j.LogManager;
 import org.apache.logging.log4j.Logger;
+import org.apache.logging.log4j.core.config.Configurator;
 import org.fusesource.jansi.AnsiConsole;
 import picocli.CommandLine;
 
@@ -13,11 +16,35 @@ public class GodiddyCLIApplication {
 
     public static void main(String... args) {
         log.info("Starting application: " + Arrays.asList(args));
+        configureLogLevel();
         AnsiConsole.systemInstall();
         int exitCode = new CommandLine(new GodiddyRootCommand()).execute(args);
         System.out.println();
         log.info("Exit code: " + exitCode);
         AnsiConsole.systemUninstall();
         System.exit(exitCode);
     }
+
+    private static void configureLogLevel() {
+        LogLevel.Value logLevel = LogLevel.getLog();
+        switch (logLevel) {
+            case warn -> {
+                Configurator.setLevel("com.godiddy.cli", Level.WARN);
+                Configurator.setLevel("com.danubetech", Level.WARN);
+            }
+            case standard -> {
+                Configurator.setLevel("com.godiddy.cli", Level.INFO);
+                Configurator.setLevel("com.danubetech", Level.WARN);
+            }
+            case info -> {
+                Configurator.setLevel("com.godiddy.cli", Level.INFO);
+                Configurator.setLevel("com.danubetech", Level.INFO);
+            }
+            case debug -> {
+                Configurator.setLevel("com.godiddy.cli", Level.DEBUG);
+                Configurator.setLevel("com.danubetech", Level.DEBUG);
+            }
+            default -> throw new RuntimeException("Invalid log level: " + logLevel);
+        }
+    }
 }

src/main/java/com/godiddy/cli/Main.java

@@ -4,6 +4,7 @@
 import com.godiddy.cli.config.Formatting;
 import com.godiddy.cli.config.Headers;
 import com.godiddy.cli.config.Kms;
+import com.godiddy.cli.config.LogLevel;
 
 public class CLIConfig {
 
@@ -55,6 +56,18 @@ public static void setKms(Kms.Value kms) {
         }
     }
 
+    public static LogLevel.Value getLogLevel() {
+        return CLIStorage.get("logLevel") == null ? null : LogLevel.Value.valueOf(CLIStorage.get("logLevel"));
+    }
+
+    public static void setLogLevel(LogLevel.Value logLevel) {
+        if (logLevel == null) {
+            CLIStorage.remove("logLevel");
+        } else {
+            CLIStorage.put("logLevel", logLevel.name());
+        }
+    }
+
     public static String getWalletServiceBase() {
         return CLIStorage.get("walletServiceBase");
     }

src/main/java/com/godiddy/cli/clistorage/cliconfig/CLIConfig.java

@@ -0,0 +1,54 @@
+package com.godiddy.cli.commands.config;
+
+import com.godiddy.cli.clistorage.cliconfig.CLIConfig;
+import com.godiddy.cli.config.LogLevel;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+import picocli.CommandLine;
+
+import java.util.concurrent.Callable;
+
+@CommandLine.Command(
+        name = "loglevel",
+        description = "Get or set the log level. Default value: " + LogLevel.DEFAULT_LOGLEVEL + ".",
+        mixinStandardHelpOptions = true
+)
+public class ConfigLogLevelCommand extends ConfigAbstractCommand implements Callable<Integer> {
+
+    private static final Logger log = LogManager.getLogger(ConfigLogLevelCommand.class);
+
+    @CommandLine.Parameters(
+            index = "0",
+            description = "The log level. Valid values: ${COMPLETION-CANDIDATES}. Default value: " + LogLevel.DEFAULT_LOGLEVEL + ".",
+            arity = "0..1"
+    )
+    LogLevel.Value logLevel;
+
+    @Override
+    public Integer call() throws Exception {
+        log.trace("Parameter 'logLevel': " + this.logLevel);
+        if (Boolean.TRUE.equals(this.delete)) {
+            CLIConfig.setLogLevel(null);
+            System.out.println("Log level setting successfully deleted.");
+        } else {
+            if (this.logLevel == null) {
+                LogLevel.Value logLevel = CLIConfig.getLogLevel();
+                if (logLevel == null) {
+                    System.out.println("No log level set.");
+                } else {
+                    System.out.println("Log level: " + logLevel);
+                }
+            } else {
+                LogLevel.Value logLevel = this.logLevel;
+                LogLevel.Value predefinedLogLevel = LogLevel.PREDEFINED_LOGLEVEL.get(logLevel.name());
+                if (predefinedLogLevel != null) {
+                    System.out.println("Using predefined log level value '" + predefinedLogLevel + "' for parameter value '" + logLevel + "'.");
+                    logLevel = predefinedLogLevel;
+                }
+                CLIConfig.setLogLevel(logLevel);
+                System.out.println("Log level successfully set: " + logLevel);
+            }
+        }
+        return 0;
+    }
+}

src/main/java/com/godiddy/cli/commands/config/ConfigLogLevelCommand.java

@@ -15,6 +15,7 @@
                 ConfigFormattingCommand.class,
                 ConfigHeadersCommand.class,
                 ConfigKmsCommand.class,
+                ConfigLogLevelCommand.class,
                 ConfigWalletServiceBaseCommand.class,
                 ConfigVaultEndpointCommand.class,
                 ConfigVaultTokenCommand.class

src/main/java/com/godiddy/cli/commands/config/ConfigRootCommand.java

@@ -0,0 +1,34 @@
+package com.godiddy.cli.config;
+
+import com.godiddy.cli.clistorage.cliconfig.CLIConfig;
+import org.apache.logging.log4j.LogManager;
+import org.apache.logging.log4j.Logger;
+
+import java.util.Map;
+import java.util.Objects;
+
+public class LogLevel {
+
+    private static final Logger log = LogManager.getLogger(LogLevel.class);
+
+    public enum Value {
+        warn,
+        standard,
+        info,
+        debug,
+        def
+    };
+
+    public static final String DEFAULT_LOGLEVEL = "standard";
+    public static final Map<String, Value> PREDEFINED_LOGLEVEL = Map.of(
+            "def", Value.valueOf(DEFAULT_LOGLEVEL)
+    );
+
+    public static Value getLog() {
+        Value logLevel = Objects.requireNonNullElse(CLIConfig.getLogLevel(), Value.valueOf(DEFAULT_LOGLEVEL));
+        if (Value.valueOf(DEFAULT_LOGLEVEL).equals(logLevel)) {
+            log.info("Using default log level: " + DEFAULT_LOGLEVEL);
+        }
+        return logLevel;
+    }
+}