From c9e1fc68b39ea12be7bb0b3fa60846f717d64170 Mon Sep 17 00:00:00 2001
From: OpenWaygate <27287694+OpenWaygate@users.noreply.github.com>
Date: Thu, 4 Sep 2025 00:12:19 +0800
Subject: [PATCH] :sparkles: Support root dir and log level

---
 pkg/vars.go | 21 +++++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/pkg/vars.go b/pkg/vars.go
index b3f2bb1..c29bbcd 100644
--- a/pkg/vars.go
+++ b/pkg/vars.go
@@ -21,15 +21,32 @@ var (
 
 func init() {
 	var err error
-	Root, err = os.OpenRoot("/")
+	rootDir, ok := os.LookupEnv("YUTU_ROOT")
+	if !ok {
+		rootDir = "/"
+	}
+	Root, err = os.OpenRoot(rootDir)
 	if err != nil {
 		panic(err)
 	}
 
+	logLevel := slog.LevelInfo
+	if lvl, ok := os.LookupEnv("YUTU_LOG_LEVEL"); ok {
+		switch lvl {
+		case "DEBUG", "debug":
+			logLevel = slog.LevelDebug
+		case "INFO", "info":
+			logLevel = slog.LevelInfo
+		case "WARN", "warn":
+			logLevel = slog.LevelWarn
+		case "ERROR", "error":
+			logLevel = slog.LevelError
+		}
+	}
 	Logger = slog.New(
 		slog.NewTextHandler(
 			os.Stdout, &slog.HandlerOptions{
-				Level: slog.LevelInfo,
+				Level: logLevel,
 			},
 		),
 	)