Add --help flag support
Currently executing `boltbrowser --help` creates a file with name `--help` in the current working directory, which is counterintuitive. This commit adds command-line flags support to boltbrowser and changes it to print out the usage message instead.
This commit is contained in:
parent
3c22c6227b
commit
46e67d5848
@ -1,6 +1,7 @@
|
||||
package main
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
|
||||
@ -19,11 +20,19 @@ var memBolt *BoltDB
|
||||
|
||||
var currentFilename string
|
||||
|
||||
func init() {
|
||||
flag.Usage = func() {
|
||||
fmt.Fprintf(os.Stdout, "Usage: %s <filename(s)>\n", ProgramName)
|
||||
}
|
||||
}
|
||||
|
||||
func main() {
|
||||
var err error
|
||||
|
||||
if len(os.Args) < 2 {
|
||||
fmt.Printf("Usage: %s <filename(s)>\n", ProgramName)
|
||||
flag.Parse()
|
||||
|
||||
if flag.NArg() == 0 {
|
||||
flag.Usage()
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
@ -35,7 +44,7 @@ func main() {
|
||||
style := defaultStyle()
|
||||
termbox.SetOutputMode(termbox.Output256)
|
||||
|
||||
databaseFiles := os.Args[1:]
|
||||
databaseFiles := flag.Args()
|
||||
for _, databaseFile := range databaseFiles {
|
||||
currentFilename = databaseFile
|
||||
db, err = bolt.Open(databaseFile, 0600, nil)
|
||||
|
Loading…
Reference in New Issue
Block a user