Short flag detection is added to the code.
This commit is contained in:
parent
593e665606
commit
25ff46d0ed
1 changed files with 11 additions and 1 deletions
12
main.go
12
main.go
|
@ -15,11 +15,21 @@ const code = `
|
||||||
import (
|
import (
|
||||||
"fmt"
|
"fmt"
|
||||||
"os"
|
"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() {
|
func init() {
|
||||||
for _, arg := range os.Args {
|
for _, arg := range os.Args {
|
||||||
if arg == "-v" || arg == "--version" {
|
if containFlag(arg, "v") || arg == "--version" {
|
||||||
fmt.Println(gitVersion)
|
fmt.Println(gitVersion)
|
||||||
os.Exit(0)
|
os.Exit(0)
|
||||||
}
|
}
|
||||||
|
|
Loading…
Add table
Reference in a new issue