feat: add debug flags

optimism-java · Dec 16, 2024 · 57b0ada · 57b0ada
1 parent 4b51c2d
commit 57b0ada
Showing 1 changed file with 14 additions and 0 deletions.
diff --git 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() {