CLI for testing out our new Pure Go file encrypter/decrypter

dcrypto-cli.go

package main

import (
	"flag"
	"fmt"
	"io"
	"os"

	"github.com/odeke-em/drive/src/dcrypto"
)

func main() {
	var password string
	var decrypt bool
	flag.StringVar(&password, "password", "", "the password to encrypt/decrypt your files")
	flag.BoolVar(&decrypt, "decrypt", false, "whether to decrypt or encrypt")
	flag.Parse()

	dcryptoFn := func(r io.Reader, password []byte) (rr io.Reader, err error) {
		if decrypt {
			rr, err = dcrypto.NewDecrypter(r, password)
		} else {
			rr, err = dcrypto.NewEncrypter(r, password)
		}

		return
	}

	mode := "encrypting"
	if decrypt {
		mode = "decrypting"
	}

	paths := flag.Args()
	bPassword := []byte(password)
	for i, arg := range paths {
		f, err := os.Open(arg)
		if err != nil {
			fmt.Fprintf(os.Stderr, "#%d: err=%v", i, err)
			continue
		}

		rr, err := dcryptoFn(f, bPassword)
		if err != nil {
			_ = f.Close()
			fmt.Fprintf(os.Stderr, "#%d: mode=%s err=%v", i, mode, err)
			continue
		}

		_, _ = io.Copy(os.Stdout, rr)
		_ = f.Close()
		if rc, ok := rr.(io.ReadCloser); ok {
			_ = rc.Close()
		}
	}
}