From e9d16e77a3e1358efb71d0f7ce948fe824100aec Mon Sep 17 00:00:00 2001
From: Lea Anthony <lea.anthony@gmail.com>
Date: Mon, 25 Jan 2021 21:42:31 +1100
Subject: [PATCH] Add support for `loglevel` flag in debug builds

---
 v2/internal/app/debug.go | 25 +++++++++++++++++++++++++
 1 file changed, 25 insertions(+)

diff --git a/v2/internal/app/debug.go b/v2/internal/app/debug.go
index 7bcabc3c9..41be6bd27 100644
--- a/v2/internal/app/debug.go
+++ b/v2/internal/app/debug.go
@@ -2,11 +2,36 @@
 
 package app
 
+import (
+	"flag"
+	"github.com/wailsapp/wails/v2/pkg/logger"
+	"strings"
+)
+
 // Init initialises the application for a debug environment
 func (a *App) Init() error {
 	// Indicate debug mode
 	a.debug = true
 	// Enable dev tools
 	a.options.DevTools = true
+
+	// Set log levels
+	greeting := flag.String("loglevel", "debug", "Loglevel to use - Trace, Debug, Info, Warning, Error")
+	flag.Parse()
+	if len(*greeting) > 0 {
+		switch strings.ToLower(*greeting) {
+		case "trace":
+			a.logger.SetLogLevel(logger.TRACE)
+		case "info":
+			a.logger.SetLogLevel(logger.INFO)
+		case "warning":
+			a.logger.SetLogLevel(logger.WARNING)
+		case "error":
+			a.logger.SetLogLevel(logger.ERROR)
+		default:
+			a.logger.SetLogLevel(logger.DEBUG)
+		}
+	}
+
 	return nil
 }