package commands

import (
	"fmt"
	"log"
	"os"

	"congo.gg/cmd/internal"
)

const connectUsage = `Connect to a deployed server via SSH

Usage:
  congo connect [name]
  congo connect [name] -- <command>

If only one instance exists, connects directly.
Otherwise, specify the instance name.

Use -- to run a command on the server instead of opening an interactive shell.

Examples:
  congo connect                      # interactive (auto-selects single instance)
  congo connect web-1                # interactive on web-1
  congo connect web-1 -- docker ps   # run command and exit
`

func Connect() {
	args := os.Args[2:]

	if len(args) > 0 && (args[0] == "--help" || args[0] == "-h") {
		fmt.Print(connectUsage)
		return
	}

	cfg, err := scaffold.LoadInfraConfig(".")
	if err != nil {
		log.Fatalf("load infra.json: %v", err)
	}

	// Split args on "--" to get instance name and remote command.
	var name string
	var remoteCmd []string
	for i, arg := range args {
		if arg == "--" {
			remoteCmd = args[i+1:]
			break
		}
		if name == "" {
			name = arg
		}
	}

	target, err := scaffold.FindInstanceOrDefault(cfg, name)
	if err != nil {
		if multi, ok := err.(*scaffold.MultipleInstancesError); ok {
			fmt.Println("Multiple instances found. Specify a name:")
			for _, inst := range multi.Instances {
				fmt.Printf("  %s (%s)\n", inst.Name, inst.IP)
			}
			os.Exit(1)
		}
		log.Fatal(err)
	}

	server := target.ToServer()

	if len(remoteCmd) > 0 {
		out, err := server.SSH(remoteCmd...)
		if err != nil {
			log.Fatalf("ssh: %v", err)
		}
		fmt.Print(out)
		return
	}

	fmt.Printf("Connecting to %s (%s)...\n", target.Name, target.IP)
	if err := server.Interactive(); err != nil {
		os.Exit(1)
	}
}