117 lines
2.3 KiB
Go
117 lines
2.3 KiB
Go
package main
|
|
|
|
import (
|
|
"flag"
|
|
"log"
|
|
"os"
|
|
"path"
|
|
|
|
"github.com/otakukaze/go-pgp-tool/pgpcrypt"
|
|
|
|
"github.com/otakukaze/go-pgp-tool/libs"
|
|
"github.com/otakukaze/go-pgp-tool/tools"
|
|
)
|
|
|
|
var (
|
|
flags *libs.Flags
|
|
)
|
|
|
|
func init() {
|
|
flags = new(libs.Flags)
|
|
libs.RegFlag(flags)
|
|
flag.Parse()
|
|
}
|
|
|
|
// args [0] is this
|
|
func main() {
|
|
// check flags value
|
|
if !flags.Encrypt && !flags.Decrypt {
|
|
showUsage()
|
|
return
|
|
}
|
|
if flags.Decrypt && flags.Encrypt {
|
|
showUsage()
|
|
return
|
|
}
|
|
if len(flags.KeyFile) == 0 {
|
|
log.Fatal("please input KeyFile path")
|
|
}
|
|
if len(flags.SrcFile) == 0 {
|
|
log.Fatal("please input SrcFile path")
|
|
}
|
|
if len(flags.DstFile) == 0 {
|
|
log.Fatal("please input DstFile path")
|
|
}
|
|
|
|
// check file exists
|
|
if !tools.CheckExists(flags.KeyFile, false) {
|
|
log.Fatal("KeyFile not exists")
|
|
}
|
|
if !tools.CheckExists(flags.SrcFile, false) {
|
|
log.Fatal("SrcFile not exists")
|
|
}
|
|
if flags.DstFile != "-" {
|
|
dir := path.Dir(flags.DstFile)
|
|
if !tools.CheckExists(dir, true) {
|
|
log.Fatal("DstFile parent directory not exists")
|
|
}
|
|
if !flags.Override && tools.CheckExists(flags.DstFile, false) {
|
|
log.Fatal("DstFile has Exists if override add flag -y ")
|
|
}
|
|
flags.DstFile = tools.ParsePath(flags.DstFile)
|
|
}
|
|
|
|
flags.KeyFile = tools.ParsePath(flags.KeyFile)
|
|
flags.SrcFile = tools.ParsePath(flags.SrcFile)
|
|
|
|
// fmt.Println(flags.ToMap())
|
|
|
|
if flags.Encrypt {
|
|
// encrypt()
|
|
keyIO, err := os.Open(flags.KeyFile)
|
|
handleError(err)
|
|
|
|
srcIO, err := os.Open(flags.SrcFile)
|
|
handleError(err)
|
|
|
|
dstIO := os.Stdout
|
|
if flags.DstFile != "-" {
|
|
dstIO, err = os.Create(flags.DstFile)
|
|
handleError(err)
|
|
}
|
|
|
|
err = pgpcrypt.Encrypt2(keyIO, srcIO, dstIO)
|
|
|
|
// err := pgpcrypt.Encrypt(flags.KeyFile, flags.SrcFile, flags.DstFile)
|
|
handleError(err)
|
|
}
|
|
if flags.Decrypt {
|
|
// decrypt()
|
|
keyIO, err := os.Open(flags.KeyFile)
|
|
handleError(err)
|
|
|
|
srcIO, err := os.Open(flags.SrcFile)
|
|
handleError(err)
|
|
|
|
dstIO := os.Stdout
|
|
if flags.DstFile != "-" {
|
|
dstIO, err = os.Create(flags.DstFile)
|
|
handleError(err)
|
|
}
|
|
|
|
err = pgpcrypt.Decrypt2(keyIO, srcIO, dstIO, flags.Password)
|
|
// err := pgpcrypt.Decrypt(flags.KeyFile, flags.SrcFile, flags.DstFile, flags.Password)
|
|
handleError(err)
|
|
}
|
|
}
|
|
|
|
func handleError(err error) {
|
|
if err != nil {
|
|
log.Fatal(err)
|
|
}
|
|
}
|
|
|
|
func showUsage() {
|
|
flag.Usage()
|
|
}
|