more clean up of main

This commit is contained in:
Derek McQuay 2016-03-11 21:36:35 -08:00
parent eeff714ac6
commit 60d320088d

42
main.go
View File

@ -19,42 +19,9 @@ func main() {
//psyfer.ReadConfig() //psyfer.ReadConfig()
//psyfer.Substitution("hello") //psyfer.Substitution("hello")
var input string var input string
var cipher string
var cipType string
//var key string //var key string
var decrypt bool var decrypt bool
var sub = &cobra.Command{
Use: "sub mode -c [cipher] -k [key] -i [input]",
Short: "substitution cipher",
Long: `perform substitution cipher`,
Run: func(cmd *cobra.Command, args []string) {
if input != "" {
switch {
case cipher == "transposition":
switch {
case cipType == "random":
fmt.Println(psyfer.TransposeRandom(input))
case cipType == "rail":
fmt.Println(psyfer.TransposeRailFence(input))
case cipType == "split":
fmt.Println(psyfer.TransposeSplit(input))
default:
fmt.Println("Missing cipher sub type (random, rail, split)")
}
case cipher == "substitution":
fmt.Println("substitution")
case cipher == "vigenere":
fmt.Println("vigenere")
default:
fmt.Println("Must choose transposition, substitution, or vigenere")
}
} else {
fmt.Println("Missing input")
}
},
}
var trans = &cobra.Command{ var trans = &cobra.Command{
Use: "trans mode", Use: "trans mode",
Short: "transposition cipher", Short: "transposition cipher",
@ -64,6 +31,15 @@ func main() {
}, },
} }
var sub = &cobra.Command{
Use: "sub mode -c [cipher] -k [key] -i [input]",
Short: "substitution cipher",
Long: `perform substitution cipher`,
Run: func(cmd *cobra.Command, args []string) {
fmt.Println("missing input, see -h (--help) for more info")
},
}
var aes = &cobra.Command{ var aes = &cobra.Command{
Use: "aes mode -c [cipher] -k [key] -i [input]", Use: "aes mode -c [cipher] -k [key] -i [input]",
Short: "aes cipher", Short: "aes cipher",