From 25ff46d0eda89045145cd32b1b7144cfc5deb4db Mon Sep 17 00:00:00 2001 From: Foil-hat-guy Date: Sun, 20 Apr 2025 11:40:01 +0300 Subject: [PATCH] Short flag detection is added to the code. --- main.go | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/main.go b/main.go index 8dd5694..9f56727 100644 --- a/main.go +++ b/main.go @@ -15,11 +15,21 @@ const code = ` import ( "fmt" "os" + "strings" ) +func containFlag(arg string, sfortFlag string) bool { + if len(arg)>1 { + if arg[0] == "-"[0] && arg[1] != "-"[0] { + return strings.Contains(arg, sfortFlag) + } + } + return false +} + func init() { for _, arg := range os.Args { - if arg == "-v" || arg == "--version" { + if containFlag(arg, "v") || arg == "--version" { fmt.Println(gitVersion) os.Exit(0) }