package main

import (
	"os"
	"path/filepath"
	"strings"
	"testing"
)

func setMainTestHome(t *testing.T) string {
	t.Helper()
	home := t.TempDir()
	t.Setenv("HOME", home)
	t.Setenv("USERPROFILE", home)
	return home
}

func TestGuardScriptPathRejectsUnsafeAgentID(t *testing.T) {
	setMainTestHome(t)
	_, err := guardScriptPathForAgent("../escape")
	if err == nil || !strings.Contains(err.Error(), "invalid agent ID") {
		t.Fatalf("guardScriptPathForAgent() error = %v", err)
	}
}

func TestFindAgentIDByNameReturnsValidatedDirectoryIdentity(t *testing.T) {
	home := setMainTestHome(t)
	runtimeDirectory := filepath.Join(home, ".elydora", "agent-1")
	if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
		t.Fatalf("MkdirAll() error = %v", err)
	}
	config := []byte(`{"agent_name":"opencode","agent_id":"agent-1"}`)
	if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
		t.Fatalf("WriteFile() error = %v", err)
	}

	agentID, err := findAgentIDByName("opencode")
	if err != nil {
		t.Fatalf("findAgentIDByName() error = %v", err)
	}
	if agentID != "agent-1" {
		t.Fatalf("agent ID = %q, want agent-1", agentID)
	}
}

func TestFindAgentIDByNameRejectsConfigDirectoryMismatch(t *testing.T) {
	home := setMainTestHome(t)
	runtimeDirectory := filepath.Join(home, ".elydora", "stored-directory")
	if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
		t.Fatalf("MkdirAll() error = %v", err)
	}
	config := []byte(`{"agent_name":"opencode","agent_id":"different-agent"}`)
	if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
		t.Fatalf("WriteFile() error = %v", err)
	}

	_, err := findAgentIDByName("opencode")
	if err == nil || !strings.Contains(err.Error(), "crosses its runtime directory") {
		t.Fatalf("findAgentIDByName() error = %v", err)
	}
}

func TestFindAgentIDByNameRejectsAmbiguousRuntimes(t *testing.T) {
	home := setMainTestHome(t)
	for _, agentID := range []string{"agent-1", "agent-2"} {
		runtimeDirectory := filepath.Join(home, ".elydora", agentID)
		if err := os.MkdirAll(runtimeDirectory, 0700); err != nil {
			t.Fatalf("MkdirAll() error = %v", err)
		}
		config := []byte(`{"agent_name":"opencode","agent_id":"` + agentID + `"}`)
		if err := os.WriteFile(filepath.Join(runtimeDirectory, "config.json"), config, 0600); err != nil {
			t.Fatalf("WriteFile() error = %v", err)
		}
	}

	_, err := findAgentIDByName("opencode")
	if err == nil || !strings.Contains(err.Error(), "multiple") {
		t.Fatalf("findAgentIDByName() error = %v", err)
	}
}