package agent

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

// Hilfsfunktion: temporäres Arbeitsverzeichnis anlegen
func setupWorkDir(t *testing.T) string {
	t.Helper()
	dir, err := os.MkdirTemp("", "agent-tools-test-*")
	if err != nil {
		t.Fatalf("Konnte temp dir nicht anlegen: %v", err)
	}
	t.Cleanup(func() { os.RemoveAll(dir) }) // wird nach jedem Test aufgeräumt
	return dir
}

// ─── WRITE_FILE ──────────────────────────────────────────

func TestWriteFile_CreatesFile(t *testing.T) {
	dir := setupWorkDir(t)

	toolCall := "TOOL:WRITE_FILE:hello.go:package main\n\nfunc main() {}"
	output := executeTool(toolCall, dir)

	if output != "WRITE_FILE OK: hello.go geschrieben" {
		t.Errorf("Unerwarteter Output: %q", output)
	}

	// Datei muss wirklich existieren
	path := filepath.Join(dir, "hello.go")
	if _, err := os.Stat(path); os.IsNotExist(err) {
		t.Error("Datei wurde nicht angelegt")
	}
}

func TestWriteFile_CorrectContent(t *testing.T) {
	dir := setupWorkDir(t)
	expected := "package main\n\nfunc main() {}"

	executeTool("TOOL:WRITE_FILE:hello.go:"+expected, dir)

	content, err := os.ReadFile(filepath.Join(dir, "hello.go"))
	if err != nil {
		t.Fatalf("Datei konnte nicht gelesen werden: %v", err)
	}
	if string(content) != expected {
		t.Errorf("Inhalt falsch\n  erwartet: %q\n  bekommen: %q", expected, string(content))
	}
}

func TestWriteFile_CreatesSubdirectory(t *testing.T) {
	dir := setupWorkDir(t)

	executeTool("TOOL:WRITE_FILE:subdir/nested/file.go:package sub", dir)

	path := filepath.Join(dir, "subdir", "nested", "file.go")
	if _, err := os.Stat(path); os.IsNotExist(err) {
		t.Error("Verschachtelte Datei wurde nicht angelegt")
	}
}

func TestWriteFile_MissingContent_ReturnsError(t *testing.T) {
	dir := setupWorkDir(t)

	output := executeTool("TOOL:WRITE_FILE:hello.go", dir)

	if output != "ERROR: WRITE_FILE braucht Inhalt" {
		t.Errorf("Erwartete Fehlermeldung, bekam: %q", output)
	}
}

// ─── READ_FILE ───────────────────────────────────────────

func TestReadFile_ReadsExistingFile(t *testing.T) {
	dir := setupWorkDir(t)
	expected := "Hello, World!"

	// Datei vorbereiten
	os.WriteFile(filepath.Join(dir, "test.txt"), []byte(expected), 0644)

	output := executeTool("TOOL:READ_FILE:test.txt", dir)

	if output != "READ_FILE test.txt:\n"+expected {
		t.Errorf("Unerwarteter Output: %q", output)
	}
}

func TestReadFile_NonExistentFile_ReturnsError(t *testing.T) {
	dir := setupWorkDir(t)

	output := executeTool("TOOL:READ_FILE:gibts-nicht.txt", dir)

	if output == "" || output[:15] != "READ_FILE ERROR" {
		t.Errorf("Erwartete Fehlermeldung, bekam: %q", output)
	}
}

// ─── LIST_FILES ──────────────────────────────────────────

func TestListFiles_ReturnsFileNames(t *testing.T) {
	dir := setupWorkDir(t)

	// Testdateien anlegen
	os.WriteFile(filepath.Join(dir, "a.go"), []byte(""), 0644)
	os.WriteFile(filepath.Join(dir, "b.go"), []byte(""), 0644)

	output := executeTool("TOOL:LIST_FILES:.", dir)

	if !contains(output, "a.go") || !contains(output, "b.go") {
		t.Errorf("Erwartete beide Dateien in Output: %q", output)
	}
}

func TestListFiles_NonExistentDir_ReturnsError(t *testing.T) {
	dir := setupWorkDir(t)

	output := executeTool("TOOL:LIST_FILES:gibts-nicht", dir)

	if output[:16] != "LIST_FILES ERROR" {
		t.Errorf("Erwartete Fehlermeldung, bekam: %q", output)
	}
}

// ─── UNGÜLTIGE TOOL-CALLS ────────────────────────────────

func TestExecuteTool_UnknownTool_ReturnsError(t *testing.T) {
	dir := setupWorkDir(t)

	output := executeTool("TOOL:UNKNOWN_TOOL:arg", dir)

	if output[:5] != "ERROR" {
		t.Errorf("Erwartete Fehlermeldung, bekam: %q", output)
	}
}

func TestExecuteTool_InvalidFormat_ReturnsError(t *testing.T) {
	dir := setupWorkDir(t)

	output := executeTool("TOOL:", dir)

	if output[:5] != "ERROR" {
		t.Errorf("Erwartete Fehlermeldung, bekam: %q", output)
	}
}

// ─── EXECUTE_TOOLS (Top-Level) ───────────────────────────

func TestExecuteTools_ParsesMultipleToolCalls(t *testing.T) {
	dir := setupWorkDir(t)

	response := `Ich schreibe zwei Dateien:
TOOL:WRITE_FILE:foo.go:package foo
TOOL:WRITE_FILE:bar.go:package bar`

	_, hadTools := ExecuteTools(response, dir)

	if !hadTools {
		t.Error("Hätte Tool-Calls erkennen sollen")
	}

	// Beide Dateien müssen existieren
	for _, name := range []string{"foo.go", "bar.go"} {
		if _, err := os.Stat(filepath.Join(dir, name)); os.IsNotExist(err) {
			t.Errorf("Datei %s wurde nicht angelegt", name)
		}
	}
}

func TestExecuteTools_NoToolCalls_ReturnsFalse(t *testing.T) {
	dir := setupWorkDir(t)

	_, hadTools := ExecuteTools("Keine Tools hier, nur Text.", dir)

	if hadTools {
		t.Error("Hätte keine Tool-Calls erkennen sollen")
	}
}

// ─── Hilfsfunktion ───────────────────────────────────────

func contains(s, substr string) bool {
	return len(s) >= len(substr) && (s == substr ||
		len(s) > 0 && containsHelper(s, substr))
}

func containsHelper(s, substr string) bool {
	for i := 0; i <= len(s)-len(substr); i++ {
		if s[i:i+len(substr)] == substr {
			return true
		}
	}
	return false
}