package browserbiometrics

import (
	"fmt"
	"os"
	"path/filepath"
	"strings"

	"github.com/quexten/goldwarden/agent/config"
	"github.com/quexten/goldwarden/browserbiometrics/logging"
)

var chromiumPaths = []string{
	"~/.config/google-chrome/",
	"~/.config/google-chrome-beta/",
	"~/.config/google-chrome-unstable/",
	"~/.config/chromium/",
	"~/.config/BraveSoftware/Brave-Browser/",
	"~/.config/thorium/",
	"~/.config/microsoft-edge-beta/",
	"~/.config/microsoft-edge-dev/",
}
var mozillaPaths = []string{"~/.mozilla/", "~/.librewolf/", "~/.waterfox/"}

const appID = "com.quexten.bw-bio-handler"

var transportKey []byte

func Main(rtCfg *config.RuntimeConfig) {
	logging.Debugf("Starting browserbiometrics")
	transportKey = generateTransportKey()
	logging.Debugf("Generated transport key")

	setupCommunication()
	readLoop(rtCfg)
}

func DetectAndInstallBrowsers() error {
	var err error

	// first, ensure the native messaging hosts dirs exist
	for _, path := range chromiumPaths {
		path = strings.ReplaceAll(path, "~", os.Getenv("HOME"))
		_, err = os.Stat(path)
		if err != nil {
			continue
		}

		_, err = os.Stat(path + "NativeMessagingHosts/")
		if err == nil {
			fmt.Println("Native messaging host directory already exists: " + path + "NativeMessagingHosts/")
			continue
		}
		err = os.MkdirAll(path+"NativeMessagingHosts/", 0755)
		if err != nil {
			fmt.Println("Error creating native messaging host directory: " + err.Error())
		} else {
			fmt.Println("Created native messaging host directory: " + path + "NativeMessagingHosts/")
		}
	}
	for _, path := range mozillaPaths {
		path = strings.ReplaceAll(path, "~", os.Getenv("HOME"))
		_, err = os.Stat(path)
		if err != nil {
			continue
		}

		_, err = os.Stat(path + "native-messaging-hosts/")
		if err == nil {
			fmt.Println("Native messaging host directory already exists: " + path + "native-messaging-hosts/")
			continue
		}
		err = os.MkdirAll(path+"native-messaging-hosts/", 0755)
		if err != nil {
			fmt.Println("Error creating native messaging host directory: " + err.Error())
		} else {
			fmt.Println("Created native messaging host directory: " + path + "native-messaging-hosts/")
		}
	}

	err = detectAndInstallBrowsers(".config")
	if err != nil {
		return err
	}
	for _, path := range mozillaPaths {
		path = strings.ReplaceAll(path, "~/", "")
		err = detectAndInstallBrowsers(path)
		if err != nil {
			return err
		}
	}
	return nil
}

func detectAndInstallBrowsers(startPath string) error {
	home := os.Getenv("HOME")
	err := filepath.Walk(home+"/"+startPath, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return nil
		}
		var tempPath string
		if !strings.HasPrefix(path, home) {
			return nil
		} else {
			tempPath = strings.TrimPrefix(path, home)
		}
		if strings.Count(tempPath, "/") > 3 {
			return nil
		}

		binPath, err := os.Executable()
		if err != nil {
			return err
		}

		if info.IsDir() && info.Name() == "native-messaging-hosts" {
			fmt.Printf("Found mozilla-like browser: %s\n", path)
			manifest := strings.Replace(templateMozilla, "PATH", binPath, 1)
			err = os.WriteFile(path+"/com.8bit.bitwarden.json", []byte(manifest), 0644)
		} else if info.IsDir() && info.Name() == "NativeMessagingHosts" {
			fmt.Printf("Found chrome-like browser: %s\n", path)
			manifest := strings.Replace(templateChrome, "PATH", binPath, 1)
			err = os.WriteFile(path+"/com.8bit.bitwarden.json", []byte(manifest), 0644)
		}

		return err
	})

	return err
}