Refactor: Split into more packages to reduce server binary size

crypt/rcw.go

@@ -0,0 +1,79 @@
+package crypt
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"strings"
+
+	"github.com/rwinkhart/go-boilerplate/back"
+	"github.com/rwinkhart/libmutton/global"
+	"github.com/rwinkhart/rcw/daemon"
+	"github.com/rwinkhart/rcw/wrappers"
+)
+
+// RCWDArgument reads the passphrase from stdin and caches it via an RCW daemon.
+func RCWDArgument() {
+	passphrase := back.ReadFromStdin()
+	if passphrase == "" {
+		os.Exit(0)
+	}
+	daemon.Start([]byte(passphrase))
+}
+
+// DecryptFileToSlice decrypts an RCW wrapped file and returns the contents as a slice of (trimmed) strings.
+func DecryptFileToSlice(targetLocation string) []string {
+	// read encrypted file
+	encBytes, err := os.ReadFile(targetLocation)
+	if err != nil {
+		back.PrintError("Failed to open \""+targetLocation+"\" for decryption - "+err.Error(), back.ErrorRead, true)
+	}
+
+	// decrypt data using RCW daemon
+	passphrase := launchRCWDProcess()
+	if passphrase == nil {
+		// if daemon is already running, use it to decrypt the data
+		return strings.Split(string(daemon.GetDec(encBytes)), "\n")
+	}
+	// if the daemon is not already running, use wrappers.Decrypt
+	// directly to avoid waiting for socket file creation
+	decBytes, err := wrappers.Decrypt(encBytes, passphrase)
+	if err != nil {
+		back.PrintError("Failed to decrypt \""+targetLocation+"\" - "+err.Error(), global.ErrorDecryption, true)
+	}
+	return strings.Split(string(decBytes), "\n")
+}
+
+// EncryptBytes encrypts a byte slice using RCW and returns the encrypted data.
+func EncryptBytes(decBytes []byte) []byte {
+	passphrase := launchRCWDProcess()
+	if passphrase == nil {
+		// if daemon is already running, use it to encrypt the data
+		return daemon.GetEnc(decBytes)
+	}
+	// if the daemon is not already running, use wrappers.Encrypt
+	// directly to avoid waiting for socket file creation
+	return wrappers.Encrypt(decBytes, passphrase)
+}
+
+// launchRCWDProcess launches an RCW daemon to cache a passphrase.
+// If the daemon is not already running, it returns the passphrase (otherwise returns nil).
+func launchRCWDProcess() []byte {
+	if daemon.IsOpen() {
+		return nil
+	}
+	var passphrase []byte
+	for {
+		passphrase = global.GetPassphrase("RCW Passphrase:")
+		err := wrappers.RunSanityCheck(global.ConfigDir+global.PathSeparator+"sanity.rcw", passphrase)
+		if err == nil {
+			break
+		}
+		fmt.Println(back.AnsiError + "Incorrect passphrase" + back.AnsiReset)
+	}
+	cmd := exec.Command(os.Args[0], "startrcwd")
+	back.WriteToStdin(cmd, string(passphrase))
+	cmd.Start()
+
+	return passphrase
+}