read filename from arg instead of directly loading data

This commit is contained in:
ed 2019-12-18 11:43:51 +01:00
parent 0be30e3222
commit c21833fcf0
1 changed files with 8 additions and 8 deletions

16
main.go
View File

@ -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)