package commands

import (
	"flag"
	"fmt"
	"log"
	"os"
	"strings"
	"time"

	"congo.gg/cmd/internal"
)

const statusUsage = `Show deployment status

Usage:
  congo status [flags]

Flags:
  --server <name>  Show status for a specific instance

Shows running services and connectivity for all deployed instances.

Examples:
  congo status                  # all instances
  congo status --server web-1   # specific instance
`

func Status() {
	fs := flag.NewFlagSet("status", flag.ExitOnError)
	fs.Usage = func() { fmt.Print(statusUsage) }

	serverName := fs.String("server", "", "Show status for a specific instance")
	fs.Parse(os.Args[2:])

	cfg, err := scaffold.LoadInfraConfig(".")
	if err != nil {
		log.Fatalf("load infra.json: %v\n\nRun 'congo init <name>' to create a project first.", err)
	}

	all := cfg.AllInstances()

	if len(all) == 0 {
		fmt.Println("No instances deployed.")
		fmt.Println()
		if len(cfg.Servers) > 0 {
			fmt.Println("Server types defined in infra.json:")
			for name := range cfg.Servers {
				fmt.Printf("  %s\n", name)
			}
			fmt.Println()
			fmt.Println("Provision with: congo launch --new <server-type>")
		} else {
			fmt.Println("Add server types to infra.json, then run: congo launch --new <name>")
		}
		return
	}

	if *serverName != "" {
		inst, _, err := scaffold.FindInstance(cfg, *serverName)
		if err != nil {
			log.Fatal(err)
		}
		printInstanceStatus(inst)
		return
	}

	for i, inst := range all {
		if i > 0 {
			fmt.Println()
		}
		printInstanceStatus(&inst)
	}
}

func printInstanceStatus(inst *scaffold.Instance) {
	server := inst.ToServer()

	fmt.Printf("%s (%s)\n", inst.Name, inst.IP)

	out, err := server.SSHWithTimeout(5*time.Second, "docker", "ps", "--format", "table {{.Names}}\t{{.Status}}\t{{.Ports}}")
	if err != nil {
		fmt.Println("  Status: unreachable")
		if strings.Contains(err.Error(), "Connection refused") {
			fmt.Println("  Hint: server may be stopped or SSH is not running")
		} else if strings.Contains(err.Error(), "timed out") || strings.Contains(err.Error(), "deadline exceeded") {
			fmt.Println("  Hint: connection timed out — check firewall or server status")
		}
		return
	}

	fmt.Println("  Status: connected")

	lines := strings.Split(strings.TrimSpace(out), "\n")
	if len(lines) <= 1 {
		fmt.Println("  Services: none running")
		return
	}

	fmt.Println()
	for _, line := range lines {
		fmt.Printf("  %s\n", line)
	}
}