package graphs

import (
	"strings"

	"agent-desk/internal/pkg/i18nx"
)

const (
	InterruptTypeTicketCreationConfirmation = "ticket_creation_confirmation"
	InterruptTypeHandoffConfirmation        = "handoff_confirmation"
)

var (
	ConfirmOrCancelPrompt          = i18nx.Get("graph.confirmOrCancel")
	NeedExplicitConfirmationPrompt = i18nx.Get("graph.needExplicitConfirmation")
	ConfirmationExpiredReply       = i18nx.Get("graph.confirmationExpired")
	CancelCreateTicketReply        = i18nx.Get("graph.cancelCreateTicket")
	CancelHandoffReply             = i18nx.Get("graph.cancelHandoff")
)

func ConfirmOrCancelPromptForLocale(locale string) string {
	if strings.TrimSpace(locale) == "" {
		locale = i18nx.DefaultLocale
	}
	return i18nx.Getf(locale, "graph.confirmOrCancel")
}

func NeedExplicitConfirmationPromptForLocale(locale string) string {
	if strings.TrimSpace(locale) == "" {
		locale = i18nx.DefaultLocale
	}
	return i18nx.Getf(locale, "graph.needExplicitConfirmation")
}

func ConfirmationExpiredReplyForLocale(locale string) string {
	if strings.TrimSpace(locale) == "" {
		locale = i18nx.DefaultLocale
	}
	return i18nx.Getf(locale, "graph.confirmationExpired")
}

func CancelCreateTicketReplyForLocale(locale string) string {
	if strings.TrimSpace(locale) == "" {
		locale = i18nx.DefaultLocale
	}
	return i18nx.Getf(locale, "graph.cancelCreateTicket")
}

func CancelHandoffReplyForLocale(locale string) string {
	if strings.TrimSpace(locale) == "" {
		locale = i18nx.DefaultLocale
	}
	return i18nx.Getf(locale, "graph.cancelHandoff")
}

type ConfirmationDecision string

const (
	ConfirmationDecisionConfirm ConfirmationDecision = "confirm"
	ConfirmationDecisionCancel  ConfirmationDecision = "cancel"
)

func ParseConfirmationDecision(value string) ConfirmationDecision {
	value = strings.ToLower(strings.TrimSpace(value))
	if value == "" {
		return ""
	}
	cancelWords := []string{
		"取消", "不用", "不需要", "算了", "no", "cancel", "stop",
		"нет", "отмена", "отменить", "не надо", "не нужно", "стоп", "хватит", "отбой",
	}
	for _, item := range cancelWords {
		if strings.Contains(value, item) {
			return ConfirmationDecisionCancel
		}
	}
	confirmWords := []string{
		"确认", "是", "好的", "可以", "ok", "yes", "continue", "confirm", "继续", "同意",
		"да", "подтверждаю", "подтвердить", "переводи", "переведи", "давай",
		"согласен", "согласна", "верно", "точно", "конечно",
	}
	for _, item := range confirmWords {
		if strings.Contains(value, item) {
			return ConfirmationDecisionConfirm
		}
	}
	return ""
}

func IsCancellationReply(replyText string) bool {
	replyText = strings.TrimSpace(replyText)
	return strings.Contains(replyText, CancelCreateTicketReply) ||
		strings.Contains(replyText, CancelHandoffReply) ||
		strings.Contains(replyText, "已取消本次工单创建。") ||
		strings.Contains(replyText, "已取消本次转人工。") ||
		strings.Contains(replyText, "Создание обращения отменено.") ||
		strings.Contains(replyText, "Передача человеку отменена.") ||
		strings.Contains(replyText, "Ticket creation cancelled.") ||
		strings.Contains(replyText, "Human handoff cancelled.")
}