package kvm

import (
	"fmt"
	"os"
	"strconv"
)

const (
	gpioBasePath  = "/sys/class/gpio"
	ledGreenPath  = "/sys/class/leds/led:g"
	ledYellowPath = "/sys/class/leds/led:y"
)

func exportGPIO(pin int) error {
	exportFile := gpioBasePath + "/export"
	return os.WriteFile(exportFile, []byte(strconv.Itoa(pin)), 0644)
}

func unexportGPIO(pin int) error {
	unexportFile := gpioBasePath + "/unexport"
	return os.WriteFile(unexportFile, []byte(strconv.Itoa(pin)), 0644)
}

func isGPIOExported(pin int) bool {
	gpioPath := fmt.Sprintf("%s/gpio%d", gpioBasePath, pin)
	_, err := os.Stat(gpioPath)
	return err == nil
}

func setGPIODirection(pin int, direction string) error {
	if !isGPIOExported(pin) {
		if err := exportGPIO(pin); err != nil {
			return fmt.Errorf("failed to export GPIO: %v", err)
		}
	}
	directionFile := fmt.Sprintf("%s/gpio%d/direction", gpioBasePath, pin)
	return os.WriteFile(directionFile, []byte(direction), 0644)
}

func setGPIOValue(pin int, status bool) error {
	var value int
	if status {
		value = 1
	} else {
		value = 0
	}
	if !isGPIOExported(pin) {
		if err := exportGPIO(pin); err != nil {
			return fmt.Errorf("failed to export GPIO: %v", err)
		}
		if err := setGPIODirection(pin, "out"); err != nil {
			return fmt.Errorf("failed to set GPIO direction: %v", err)
		}
	}
	valueFile := fmt.Sprintf("%s/gpio%d/value", gpioBasePath, pin)
	return os.WriteFile(valueFile, []byte(strconv.Itoa(value)), 0644)
}

func setLedMode(ledConfigPath string, mode string) error {
	if mode == "network-link" {
		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED trigger: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED device name: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/link", []byte("1"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED link: %v", err)
		}
	} else if mode == "network-tx" {
		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED trigger: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED device name: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/tx", []byte("1"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED tx: %v", err)
		}
	} else if mode == "network-rx" {
		err := os.WriteFile(ledConfigPath+"/trigger", []byte("netdev"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED trigger: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/device_name", []byte("eth0"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED device name: %v", err)
		}
		err = os.WriteFile(ledConfigPath+"/rx", []byte("1"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED rx: %v", err)
		}
	} else if mode == "kernel-activity" {
		err := os.WriteFile(ledConfigPath+"/trigger", []byte("activity"), 0644)
		if err != nil {
			return fmt.Errorf("failed to set LED trigger: %v", err)
		}
	} else {
		return fmt.Errorf("invalid LED mode: %s", mode)
	}
	return nil
}

func initGPIO() {
	LoadConfig()
	// IO0: GPIO58 IO1: GPIO59
	_ = setGPIOValue(58, config.IO0Status)
	_ = setGPIOValue(59, config.IO1Status)
	_ = setLedMode(ledYellowPath, config.LEDYellowMode)
	_ = setLedMode(ledGreenPath, config.LEDGreenMode)
}