assume show mode when file ends in .hdn

This commit is contained in:
ed 2019-12-18 12:51:38 +01:00
parent 9ebd945d4e
commit 594f1a8b6b
1 changed files with 12 additions and 1 deletions

13
main.go
View File

@ -9,11 +9,13 @@ import (
"io" "io"
"io/ioutil" "io/ioutil"
"os" "os"
"regexp"
"syscall" "syscall"
"golang.org/x/crypto/ssh/terminal" "golang.org/x/crypto/ssh/terminal"
) )
var debug bool = false var debug bool = false
var FileName string
func checkErr(err error) { func checkErr(err error) {
if err != nil { if err != nil {
@ -69,6 +71,11 @@ func data2file(filename string, data []byte) {
func getMode() string { func getMode() string {
var mode string var mode string
//fmt.Printf("from getMode, fn = %s \n",FileName)
match, _ := regexp.Match(`.*\.hdn$`,[]byte(FileName))
if match {
return "s"
}
fmt.Print("mode: ") fmt.Print("mode: ")
fmt.Fscan(os.Stdin, &mode) fmt.Fscan(os.Stdin, &mode)
if mode == "q" {os.Exit(1)} if mode == "q" {os.Exit(1)}
@ -120,7 +127,8 @@ func checkPass() string {
} }
func main() { func main() {
fn := readFn() FileName = readFn()
fn := FileName
fmt.Println(fn) fmt.Println(fn)
mode := getMode() mode := getMode()
switch mode { switch mode {
@ -130,6 +138,9 @@ func main() {
data := file2data(fn) data := file2data(fn)
secret := encrypt(data,key) secret := encrypt(data,key)
secretfn := fn + ".hdn" secretfn := fn + ".hdn"
clearfn := fn + ".origin"
err := os.Rename(fn,clearfn)
checkErr(err)
data2file(secretfn,secret) data2file(secretfn,secret)
case "s": case "s":
fmt.Print("password: ") fmt.Print("password: ")