go-pgp-tool/libs/flag.go
2018-03-12 18:07:05 +08:00

44 lines
1008 B
Go

package libs
import (
"flag"
"reflect"
)
// Flags - flag values struct
type Flags struct {
Help bool
Decrypt bool
Encrypt bool
SrcFile string
DstFile string
KeyFile string
Override bool
Password string
}
// RegFlag - Register flag to main
func RegFlag(f *Flags) {
flag.BoolVar(&f.Help, "h", false, "show usage help")
flag.BoolVar(&f.Decrypt, "d", false, "decrypt file")
flag.BoolVar(&f.Encrypt, "e", false, "encrypt file")
flag.StringVar(&f.SrcFile, "i", "", "input source `file path`")
flag.StringVar(&f.DstFile, "o", "", "output `file path`")
flag.StringVar(&f.KeyFile, "k", "", "key `file path`")
flag.BoolVar(&f.Override, "y", false, "if output file exists override")
flag.StringVar(&f.Password, "p", "", "private key password")
}
func (f *Flags) ToMap() map[string]interface{} {
t := reflect.ValueOf(f).Elem()
smap := make(map[string]interface{})
for i := 0; i < t.NumField(); i++ {
f := t.Field(i)
smap[t.Type().Field(i).Name] = f.Interface()
}
return smap
}