read filename from argv

This commit is contained in:
ed 2019-12-18 11:38:16 +01:00
parent 6506a07253
commit 0be30e3222
1 changed files with 19 additions and 0 deletions

19
main.go
View File

@ -71,9 +71,28 @@ func getMode() string {
return mode 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() { func main() {
data := checkArg()
mode := getMode() mode := getMode()
fmt.Println(mode) fmt.Println(mode)
fmt.Println(len(data))
/* /*
var passwd string = "ThisAnnPassphrase" var passwd string = "ThisAnnPassphrase"
fmt.Println(passwd) fmt.Println(passwd)