package main

import (
	"bytes"
	"fmt"
	"os"
	"strings"

	oldwrap "github.com/rwinkhart/convertroman/wrappers"
	newwrap "github.com/rwinkhart/rcw/wrappers"

	"github.com/rwinkhart/go-boilerplate/back"
	"github.com/rwinkhart/go-boilerplate/front"
	"github.com/rwinkhart/go-boilerplate/other"
	"github.com/rwinkhart/libmutton/global"
	"github.com/rwinkhart/libmutton/synccommon"
)

func main() {
	// get info from user
	var oldRCWPassphrase, newRCWPassphrase []byte
	for {
		oldRCWPassphrase = front.InputSecret("Old RCW passphrase:")
		if !bytes.Equal(oldRCWPassphrase, front.InputSecret("Confirm old RCW passphrase:")) {
			fmt.Println(back.AnsiError + "Passphrases do not match" + back.AnsiReset)
			continue
		}
		break
	}
	for {
		newRCWPassphrase = front.InputSecret("New RCW passphrase:")
		if !bytes.Equal(newRCWPassphrase, front.InputSecret("Confirm new RCW passphrase:")) {
			fmt.Println(back.AnsiError + "Passphrases do not match" + back.AnsiReset)
			continue
		}
		break
	}

	// convert the entries
	fmt.Print("\nRe-encrypting entries. Please wait; do not force close this process.\n\n")
	var outputDir = global.EntryRoot + "-new"
	entries, folders, err := synccommon.WalkEntryDir()
	if err != nil {
		other.PrintError("Failed to walk entry directory: "+err.Error(), back.ErrorRead)
	}
	for _, folder := range folders {
		err := os.MkdirAll(outputDir+strings.ReplaceAll(folder, "/", global.PathSeparator), 0700)
		if err != nil {
			other.PrintError("Failed to create directory: "+err.Error(), back.ErrorWrite)
		}
	}
	for _, entry := range entries {
		oldEncBytes, err := os.ReadFile(global.GetRealPath(entry))
		if err != nil {
			other.PrintError("Failed to read file: "+err.Error(), back.ErrorRead)
		}
		decBytes, err := oldwrap.Decrypt(oldEncBytes, oldRCWPassphrase)
		if err != nil {
			other.PrintError("Failed to decrypt file: "+err.Error(), back.ErrorWrite)
		}
		newEncBytes := newwrap.Encrypt(decBytes, newRCWPassphrase, true, false)
		err = os.WriteFile(outputDir+strings.ReplaceAll(entry, "/", global.PathSeparator), newEncBytes, 0600)
		if err != nil {
			other.PrintError("Failed to write to file: "+err.Error(), back.ErrorWrite)
		}
	}

	// swap the new directory with the old one
	err = os.Rename(global.EntryRoot, global.EntryRoot+"-old")
	if err != nil {
		other.PrintError("Failed to rename old directory: "+err.Error(), back.ErrorWrite)
	}
	err = os.Rename(outputDir, global.EntryRoot)
	if err != nil {
		other.PrintError("Failed to rename new directory: "+err.Error(), back.ErrorWrite)
	}
}