diff --git a/main.go b/main.go index 7d097c5..1fa420f 100644 --- a/main.go +++ b/main.go @@ -71,28 +71,28 @@ func getMode() string { return mode } -func checkArg() []byte { - var data []byte +func readFn() string { + var filename string 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) + fn := arg[0] + if _, err := os.Stat(fn); err == nil { + filename = fn } else { fmt.Println("No such file") os.Exit(1) } - return data + return filename } func main() { - data := checkArg() + fn := readFn() + fmt.Println(fn) mode := getMode() fmt.Println(mode) - fmt.Println(len(data)) /* var passwd string = "ThisAnnPassphrase" fmt.Println(passwd)