diff --git a/main.go b/main.go index 6113906..8605409 100644 --- a/main.go +++ b/main.go @@ -59,7 +59,22 @@ func data2file(filename string, data []byte) { f.Write(data) } +func getMode() string { + var mode string + fmt.Print("mode: ") + fmt.Fscan(os.Stdin, &mode) + if mode != "h" && mode != "s" { + fmt.Println("not a mode") + fmt.Println("type 'h' for hide or 's' for show") + mode = getMode() + } + return mode +} + func main() { + mode := getMode() + fmt.Println(mode) + /* var passwd string = "ThisAnnPassphrase" fmt.Println(passwd) key := keyMkr(passwd) @@ -71,5 +86,6 @@ func main() { fmt.Println(string(dataSealed)) fmt.Println("decryption:") fmt.Println(string(decrypt(dataSealed,key))) + */ }