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) }