package main

import (
	"fmt"
	"os"

	serveractions "manage-servers/server-actions"
	"manage-servers/webserver"

	"github.com/spf13/viper"
)

func main() {
	servers, err := loadConfig()
	if err != nil {
		fmt.Printf("Error loading servers: %v\n", err)
		return
	}

	if len(os.Args) < 2 {
		printUsage()
		return
	}

	command := os.Args[1]
	switch command {
	case "list":
		listServers(servers)
	case "wake":
		if len(os.Args) < 3 {
			fmt.Println("Please specify a server name to wake.")
			printUsage()
			return
		}
		serverName := os.Args[2]
		serveractions.WakeServer(serverName, servers)
	case "wakeall":
		serveractions.WakeAllServers(servers)
	case "status":
		serveractions.CheckServersStatus(servers)
	case "shutdown":
		if len(os.Args) < 3 {
			fmt.Println("Please specify a server name to shutdown.")
			printUsage()
			return
		}
		serverName := os.Args[2]
		serveractions.ShutdownServer(serverName, servers)
	case "reboot":
		if len(os.Args) < 3 {
			fmt.Println("Please specify a server name to reboot.")
			printUsage()
			return
		}
		serverName := os.Args[2]
		serveractions.RebootServer(serverName, servers)
	case "serve":
		webserver.StartWebServer(servers)
	default:
		fmt.Printf("Unknown command: %s\n", command)
		printUsage()
	}
}

func printUsage() {
	fmt.Println("Usage: go run . <command>")
	fmt.Println("Commands:")
	fmt.Println("  list                - List all configured servers")
	fmt.Println("  wake <server_name>  - Wake a specific server")
	fmt.Println("  wakeall             - Wake all configured servers")
	fmt.Println("  status              - Check the status of all servers")
	fmt.Println("  shutdown <server_name> - Shutdown a specific server")
	fmt.Println("  reboot <server_name>   - Reboot a specific server")
	fmt.Println("  serve               - Start a web server to manage servers")
}

func loadConfig() ([]serveractions.Server, error) {
	viper.SetConfigName("servers")    // name of config file (without extension)
	viper.SetConfigType("json")       // or viper.SetConfigType("YAML")
	viper.AddConfigPath("./config")   // path to look for the config file in
	viper.AddConfigPath(".")          // optionally look for config in the working directory
	err := viper.ReadInConfig()       // Find and read the config file
	if err != nil {                   // Handle errors reading the config file
		// Check if the error is that the file doesn't exist
		if _, ok := err.(viper.ConfigFileNotFoundError); ok {
			// Config file not found; ignore error and return empty server list
			return []serveractions.Server{}, nil
		}
		return nil, fmt.Errorf("fatal error config file: %w", err)
	}

	var servers []serveractions.Server
	err = viper.UnmarshalKey("servers", &servers)
	if err != nil {
		return nil, fmt.Errorf("unable to decode into struct, %v", err)
	}

	return servers, nil
}

func listServers(servers []serveractions.Server) {
	fmt.Println("Configured Servers:")
	for _, s := range servers {
		fmt.Printf("  - %s (%s) - %s - User: %s\n", s.Name, s.IP, s.Mac, s.SSHUser)
	}
}