read filename from argv
This commit is contained in:
parent
6506a07253
commit
0be30e3222
19
main.go
19
main.go
|
@ -71,9 +71,28 @@ func getMode() string {
|
|||
return mode
|
||||
}
|
||||
|
||||
func checkArg() []byte {
|
||||
var data []byte
|
||||
arg := os.Args[1:]
|
||||
if len(arg) != 1 {
|
||||
fmt.Println("hns only takes file name as arg")
|
||||
os.Exit(1)
|
||||
}
|
||||
filename := arg[0]
|
||||
if _, err := os.Stat(filename); err == nil {
|
||||
data = file2data(filename)
|
||||
} else {
|
||||
fmt.Println("No such file")
|
||||
os.Exit(1)
|
||||
}
|
||||
return data
|
||||
}
|
||||
|
||||
func main() {
|
||||
data := checkArg()
|
||||
mode := getMode()
|
||||
fmt.Println(mode)
|
||||
fmt.Println(len(data))
|
||||
/*
|
||||
var passwd string = "ThisAnnPassphrase"
|
||||
fmt.Println(passwd)
|
||||
|
|
Loading…
Reference in New Issue