dm/psyfer
1
0
Fork 0
Code Issues Pull Requests Activity

added cobra subcommands and functionality

Browse Source 
can now encrypt and decrypt for transposition ciphers.
This commit is contained in:
Derek McQuay 2016-02-22 18:01:19 -08:00
parent 13375bd8c2
commit f357a75682
1 changed files with 146 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

148
main.go
View File

@ -6,6 +6,8 @@ import (
"time"
"s.mcquay.me/dm/psyfer/psyfer"
"github.com/spf13/cobra"
)
func init() {
@ -13,6 +15,148 @@ func init() {
}
func main() {
fmt.Println(psyfer.Transpose("hello world"))
fmt.Println(psyfer.Transpose("hello"))
var input string
var cipher string
var cipType string
var key string
var encrypt = &cobra.Command{
Use: "encrypt [cipher] [key] [input]",
Short: "encrypt string",
Long: `encrypt given string`,
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 decrypt = &cobra.Command{
Use: "decrypt [cipher] [key] [input]",
Short: "decrypt string",
Long: `decrypt given string`,
Run: func(cmd *cobra.Command, args []string) {
if input != "" {
switch {
case cipher == "transposition":
switch {
case cipType == "random":
fmt.Println("random doesn't have a decryptable solution")
case cipType == "rail":
fmt.Println(psyfer.DeTransposeRailFence(input))
case cipType == "split":
fmt.Println(psyfer.DeTransposeSplit(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 crack = &cobra.Command{
Use: "times [# times] [string to echo]",
Short: "Echo anything to the screen more times",
Long: `echo things multiple times back to the user by providing a count and a string.`,
Run: func(cmd *cobra.Command, args []string) {
fmt.Println("crack")
},
}
encrypt.Flags().StringVarP(
&input,
"input",
"i",
"",
"string to be encrypted",
)
encrypt.Flags().StringVarP(
&cipher,
"cipher",
"c",
"",
"cipher to be used (transposition, substitution, vigenere)",
)
encrypt.Flags().StringVarP(
&key,
"key",
"k",
"",
"key to be used",
)
encrypt.Flags().StringVarP(
&cipType,
"type",
"t",
"",
"sub type of cipher",
)
decrypt.Flags().StringVarP(
&input,
"input",
"i",
"",
"string to be encrypted",
)
decrypt.Flags().StringVarP(
&cipher,
"cipher",
"c",
"",
"cipher to be used (transposition, substitution, vigenere)",
)
decrypt.Flags().StringVarP(
&key,
"key",
"k",
"",
"key to be used",
)
decrypt.Flags().StringVarP(
&cipType,
"type",
"t",
"",
"sub type of cipher",
)
var rootCmd = &cobra.Command{Use: "app"}
rootCmd.AddCommand(encrypt, decrypt, crack)
rootCmd.Execute()
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("helloworld")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("1")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("12")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("123")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("1234")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("12345")))
//fmt.Println(psyfer.DeTransposeRailFence(psyfer.TransposeRailFence("1234567")))
}