crypt.go

package main

import (
	"flag"
	"io/ioutil"
	"log"
	"os"

	"github.com/RNCryptor/RNCryptor-go"
)

var key = flag.String("k", "", "")
var dec = flag.Bool("d", false, "")

func main() {
	flag.Parse()

	if *dec {
		cipher, err := ioutil.ReadAll(os.Stdin)
		if err != nil {
			log.Fatal(err)
		}

		plain, err := rncryptor.Decrypt(*key, cipher)
		if err != nil {
			log.Fatal(err)
		}

		os.Stdout.Write(plain)
	} else {
		plain, err := ioutil.ReadAll(os.Stdin)
		if err != nil {
			log.Fatal(err)
		}

		cipher, err := rncryptor.Encrypt(*key, plain)
		if err != nil {
			log.Fatal(err)
		}

		os.Stdout.Write(cipher)
	}
}