package graphs

import (
	"context"
	"encoding/json"
	"fmt"
	"strings"

	"code.tczkiot.com/wlw/ai-agent/internal/models"
	"code.tczkiot.com/wlw/ai-agent/internal/services"
)

type TriageServiceRequestInput struct {
	Goal              string `json:"goal"`
	ObservedIssue     string `json:"observed_issue"`
	NeedHumanHandoff  bool   `json:"need_human_handoff"`
	AdditionalContext string `json:"additional_context"`
}

type TriageServiceRequestResult struct {
	Analysis          AnalyzeConversationResult `json:"analysis"`
	RecommendedAction string                    `json:"recommended_action"`
	Ready             bool                      `json:"ready"`
}

type TriageServiceRequestGraph struct {
	conversation models.Conversation
}

func NewTriageServiceRequestGraph(conversation models.Conversation) *TriageServiceRequestGraph {
	return &TriageServiceRequestGraph{conversation: conversation}
}

func (g *TriageServiceRequestGraph) Run(_ context.Context, argumentsInJSON string) (string, error) {
	input, err := g.parseInput(argumentsInJSON)
	if err != nil {
		return "", err
	}
	messages, _, _ := services.MessageService.FindByConversationIDCursor(g.conversation.ID, 0, 8, "", "")
	analysis := buildAnalyzeConversationResult(g.conversation, messages, AnalyzeConversationInput{
		Goal:              input.Goal,
		ObservedIssue:     input.ObservedIssue,
		NeedHumanHandoff:  input.NeedHumanHandoff,
		AdditionalContext: input.AdditionalContext,
	})
	result := TriageServiceRequestResult{
		Analysis:          analysis,
		RecommendedAction: analysis.RecommendedNextAction,
		Ready:             analysis.RecommendedNextAction == "continue_answering" || analysis.RecommendedNextAction == "handoff_to_human",
	}
	buf, err := json.Marshal(result)
	if err != nil {
		return "", err
	}
	return string(buf), nil
}

func (g *TriageServiceRequestGraph) parseInput(argumentsInJSON string) (TriageServiceRequestInput, error) {
	var input TriageServiceRequestInput
	if strings.TrimSpace(argumentsInJSON) == "" {
		return input, nil
	}
	if err := json.Unmarshal([]byte(argumentsInJSON), &input); err != nil {
		return input, fmt.Errorf("invalid triage service request arguments: %w", err)
	}
	input.Goal = strings.TrimSpace(input.Goal)
	input.ObservedIssue = strings.TrimSpace(input.ObservedIssue)
	input.AdditionalContext = strings.TrimSpace(input.AdditionalContext)
	return input, nil
}