From 57b0ada55a76b3150dad60a12d9ced8bed701c29 Mon Sep 17 00:00:00 2001
From: fearlessfe <505380967@qq.com>
Date: Mon, 16 Dec 2024 22:42:21 +0800
Subject: [PATCH] feat: add debug flags

---
 cmd/shisui/main.go | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/cmd/shisui/main.go b/cmd/shisui/main.go
index 93fab6f..8e476d7 100644
--- a/cmd/shisui/main.go
+++ b/cmd/shisui/main.go
@@ -109,6 +109,20 @@ func init() {
 	app.Action = shisui
 	app.Flags = slices.Concat(portalProtocolFlags, historyRpcFlags, metricsFlags, debug.Flags)
 	flags.AutoEnvVars(app.Flags, "SHISUI")
+
+	app.Before = func(ctx *cli.Context) error {
+		flags.MigrateGlobalFlags(ctx)
+		if err := debug.Setup(ctx); err != nil {
+			return err
+		}
+		flags.CheckEnvVars(ctx, app.Flags, "SHISUI")
+		return nil
+	}
+
+	app.After = func(ctx *cli.Context) error {
+		debug.Exit()
+		return nil
+	}
 }
 
 func main() {