From e0591f57c858e355686bdef7b2d3f4cf2ee2dc7b Mon Sep 17 00:00:00 2001
From: inhere <in.798@qq.com>
Date: Wed, 23 Oct 2024 16:04:51 +0800
Subject: [PATCH] :sparkles: up: cflag - app add new hook BeforeRun and call
 os.Exit on run error

---
 cflag/app.go | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/cflag/app.go b/cflag/app.go
index a36b0ee10..b18f5aa0b 100644
--- a/cflag/app.go
+++ b/cflag/app.go
@@ -30,6 +30,8 @@ type App struct {
 
 	// AfterHelpBuild hook
 	AfterHelpBuild func(buf *strutil.Buffer)
+	// BeforeRun hook func
+	BeforeRun func(c *Cmd) bool
 }
 
 // NewApp instance
@@ -97,6 +99,7 @@ func (a *App) Run() {
 	err := a.RunWithArgs(os.Args[1:])
 	if err != nil {
 		cliutil.Errorln("ERROR:", err)
+		os.Exit(1)
 	}
 }
 
@@ -112,7 +115,6 @@ func (a *App) RunWithArgs(args []string) error {
 	if name == "help" || name == "--help" || name == "-h" {
 		return a.showHelp()
 	}
-
 	if name[0] == '-' {
 		return fmt.Errorf("provide undefined flag option %q", name)
 	}
@@ -122,6 +124,9 @@ func (a *App) RunWithArgs(args []string) error {
 		return fmt.Errorf("input not exists command %q", name)
 	}
 
+	if a.BeforeRun != nil && !a.BeforeRun(cmd) {
+		return nil
+	}
 	return cmd.Parse(args[1:])
 }