From 0be30e32225d00741c7fda64add727549bee7a37 Mon Sep 17 00:00:00 2001 From: ed Date: Wed, 18 Dec 2019 11:38:16 +0100 Subject: [PATCH] read filename from argv --- main.go | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) diff --git a/main.go b/main.go index 8605409..7d097c5 100644 --- a/main.go +++ b/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)