package cipher

import (
	"crypto/rc4"
	"log"
)

var theKey = []byte("8pUsXuZw4z6B9EhGdKgNjQnjmVsYv2x5")

func GenerateKey(key string) {
	theKey = []byte(key)
}

func Encrypt(data []byte) []byte {
	c, err := rc4.NewCipher(theKey)
	if err != nil {
		log.Fatalln(err)
	}
	dst := make([]byte, len(data))
	c.XORKeyStream(dst, data)
	return dst
}

func Decrypt(data []byte) []byte {
	c, err := rc4.NewCipher(theKey)
	if err != nil {
		log.Fatalln(err)
	}
	dst := make([]byte, len(data))
	c.XORKeyStream(dst, data)
	return dst
}