package core

import (
	"fmt"
	"os"
	"os/exec"
	"strings"
)

// TODO GPG support is a temporary feature - it will be replaced with a different encryption scheme in the future

// DecryptGPG decrypts a GPG-encrypted file and returns the contents as a slice of (trimmed) strings
func DecryptGPG(targetLocation string) []string {
	cmd := exec.Command("gpg", "--pinentry-mode", "loopback", "-q", "-d", targetLocation)
	output, err := cmd.Output()

	// ensure ANSI escape sequences are interpreted properly on Windows
	enableVirtualTerminalProcessing()

	if err != nil {
		fmt.Println(AnsiError + "Failed to decrypt \"" + targetLocation + "\" - ensure it is a valid GPG-encrypted file and that you entered your passphrase correctly" + AnsiReset)
		os.Exit(1)
	}
	outputSlice := strings.Split(string(output), "\n")

	return outputSlice
}

// EncryptGPG encrypts a slice of strings using GPG and returns the encrypted data as a byte slice
func EncryptGPG(input []string) []byte {
	cmd := exec.Command("gpg", "-q", "-r", ParseConfig([][2]string{{"LIBMUTTON", "gpgID"}}, "")[0], "-e")
	writeToStdin(cmd, strings.Join(input, "\n"))
	encryptedBytes, err := cmd.Output()
	if err != nil {
		fmt.Println(AnsiError + "Failed to encrypt data - ensure that your GPG key is valid and that you have a valid GPG ID set in libmutton.ini" + AnsiReset)
		os.Exit(1)
	}
	return encryptedBytes
}