Skip to content

Commit 6a70594

Browse files
committed
feat: new flag '-v'
1 parent 460ca91 commit 6a70594

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

main.go

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,8 @@ package main
22

33
import (
44
"flag"
5+
"fmt"
6+
"os"
57

68
"github.com/WindomZ/leetcode-init/leetcode"
79
)
@@ -12,13 +14,15 @@ const (
1214
usageURL = "the url of leetcode problem."
1315
usageMarkdown = "load and rendering markdown template, and save to TEMPLATE.md."
1416
usageHelp = "prints a usage message documenting all defined command-line flags."
17+
usageVersion = "prints version information."
1518
)
1619

1720
var (
1821
keyFlag string
1922
urlFlag string
2023
markdownFlag string
2124
helpFlag bool
25+
versionFlag bool
2226
)
2327

2428
func main() {
@@ -27,6 +31,7 @@ func main() {
2731
flag.StringVar(&urlFlag, "u", "", usageURL)
2832
flag.StringVar(&markdownFlag, "m", "", usageMarkdown)
2933
flag.BoolVar(&helpFlag, "h", false, usageHelp)
34+
flag.BoolVar(&versionFlag, "v", false, usageVersion)
3035

3136
flag.Parse()
3237

@@ -35,6 +40,11 @@ func main() {
3540
return
3641
}
3742

43+
if versionFlag {
44+
fmt.Fprintf(os.Stderr, "%s version: v1.0.0\n", os.Args[0])
45+
return
46+
}
47+
3848
var problem *leetcode.Problem
3949
if urlFlag != "" {
4050
problem = leetcode.NewProblemByURI(leetcode.LanguageGo,

0 commit comments

Comments
 (0)