go-pgp-tool/main.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()
}