package agent

import (
	"context"
	"fmt"
	"strings"
	"time"

	"github.com/openai/openai-go"
	oaioption "github.com/openai/openai-go/option"

	"llm-agent/prd"
)

const (
	baseURL    = "http://127.0.0.1:12434/v1"
	maxRetries = 3
	maxTurns   = 10
)

var systemPrompt = `Du bist ein autonomer Coding-Agent. Du bekommst einen Task und erledigst ihn vollständig.

Du hast folgende Tools zur Verfügung:
- TOOL:READ_FILE:pfad        → Datei lesen
- TOOL:WRITE_FILE:pfad:inhalt → Datei schreiben
- TOOL:LIST_FILES:pfad       → Verzeichnis auflisten

Regeln:
1. Analysiere den Task zuerst
2. Nutze die Tools um Dateien zu lesen/schreiben
3. Wenn der Task vollständig erledigt ist, schreibe am Ende: TASK_COMPLETE
4. Bei Fehlern beschreibe das Problem klar`

type AgentLoop struct {
	client  *openai.Client
	model   string
	workDir string
	prdFile string
	log     *Logger
}

func NewAgentLoop(model, workDir, prdFile string, verbose bool) *AgentLoop {
	client := openai.NewClient(
		oaioption.WithBaseURL(baseURL),
		oaioption.WithAPIKey("ollama"),
	)
	return &AgentLoop{
		client:  &client,
		model:   model,
		workDir: workDir,
		prdFile: prdFile,
		log:     NewLogger(verbose),
	}
}

func (a *AgentLoop) Run() error {
	tasks, err := prd.ParseTasks(a.prdFile)
	if err != nil {
		return fmt.Errorf("PRD lesen fehlgeschlagen: %w", err)
	}

	pending := 0
	for _, t := range tasks {
		if !t.Completed {
			pending++
		}
	}
	a.log.Info("📋 %d Tasks gefunden, %d offen", len(tasks), pending)

	for _, task := range tasks {
		if task.Completed {
			a.log.Info("✅ Überspringe (bereits erledigt): %s", task.Title)
			continue
		}

		a.log.TaskStart(task.Title)

		success := false
		for attempt := 1; attempt <= maxRetries; attempt++ {
			if attempt > 1 {
				a.log.Info("🔁 Retry %d/%d...", attempt, maxRetries)
				time.Sleep(time.Duration(attempt) * 2 * time.Second)
			}

			if err := a.runTask(task); err == nil {
				success = true
				break
			} else {
				a.log.Info("⚠️  Fehler: %v", err)
			}
		}

		if success {
			prd.MarkTaskComplete(a.prdFile, task.Title)
			a.log.TaskDone(task.Title)
		} else {
			a.log.TaskFailed(task.Title, maxRetries)
		}
	}

	a.log.Info("\n🎉 Alle Tasks abgearbeitet!")
	return nil
}

func (a *AgentLoop) runTask(task prd.Task) error {
	// Frischer Kontext pro Task
	messages := []openai.ChatCompletionMessageParamUnion{
		openai.SystemMessage(systemPrompt),
		openai.UserMessage(fmt.Sprintf("Task: %s\nArbeitsverzeichnis: %s", task.Title, a.workDir)),
	}

	// System-Prompt und initialen User-Message loggen
	a.log.ChatMessage("system", systemPrompt)
	a.log.ChatMessage("user", fmt.Sprintf("Task: %s\nArbeitsverzeichnis: %s", task.Title, a.workDir))

	for turn := 0; turn < maxTurns; turn++ {
		a.log.Turn(turn + 1)

		resp, err := a.client.Chat.Completions.New(
			context.Background(),
			openai.ChatCompletionNewParams{
				Model:    a.model,
				Messages: messages,
			},
		)
		if err != nil {
			return fmt.Errorf("API-Fehler: %w", err)
		}

		response := resp.Choices[0].Message.Content
		a.log.ChatMessage("assistant", response)
		messages = append(messages, openai.AssistantMessage(response))

		// Completion Detection
		if strings.Contains(response, "TASK_COMPLETE") {
			return nil
		}

		// Tool Execution
		toolOutput, hadTools := ExecuteTools(response, a.workDir)
		if hadTools {
			a.log.ChatMessage("tool", toolOutput)
			messages = append(messages, openai.UserMessage(toolOutput))
			continue
		}

		// Kein Tool, kein TASK_COMPLETE → anstupsen
		nudge := "Bitte fahre fort. Wenn der Task erledigt ist, schreibe TASK_COMPLETE."
		a.log.ChatMessage("user", nudge)
		messages = append(messages, openai.UserMessage(nudge))
	}

	return fmt.Errorf("maximale Turns (%d) erreicht ohne TASK_COMPLETE", maxTurns)
}