package model

import (
	"fmt"
	"strings"
)

// ChatRoute 表示 Chat 节点路由决策的目标路径。
type ChatRoute string

const (
	// ChatRouteDirectReply 简单任务：Chat 节点直接输出回复，不再调用下游节点。
	ChatRouteDirectReply ChatRoute = "direct_reply"

	// ChatRouteExecute 中等任务：需要用工具处理，直接进 Execute ReAct 循环。
	ChatRouteExecute ChatRoute = "execute"

	// ChatRouteDeepAnswer 复杂问答：需要深度思考但不需工具，Chat 节点原地开 thinking 回答。
	ChatRouteDeepAnswer ChatRoute = "deep_answer"

	// ChatRoutePlan 复杂规划：需要先制定计划，进 Plan 节点。
	ChatRoutePlan ChatRoute = "plan"
)

// ChatRoutingDecision 是 Chat 节点单次路由决策的结构化输出。
//
// 职责边界：
// 1. Route 决定后续处理路径；
// 2. Speak 始终填写：给用户看的话；
// 3. NeedsRoughBuild 仅在 route=execute 且满足粗排条件时为 true；
// 4. Reason 给后端和日志看。
type ChatRoutingDecision struct {
	Route           ChatRoute `json:"route"`
	Speak           string    `json:"speak,omitempty"`
	NeedsRoughBuild bool      `json:"needs_rough_build,omitempty"`
	Reason          string    `json:"reason,omitempty"`
}

// Normalize 统一清洗路由决策中的字符串字段。
func (d *ChatRoutingDecision) Normalize() {
	if d == nil {
		return
	}
	d.Route = ChatRoute(strings.TrimSpace(string(d.Route)))
	d.Speak = strings.TrimSpace(d.Speak)
	d.Reason = strings.TrimSpace(d.Reason)
}

// Validate 校验路由决策的最小合法性。
func (d *ChatRoutingDecision) Validate() error {
	if d == nil {
		return fmt.Errorf("chat routing decision 不能为空")
	}

	d.Normalize()

	switch d.Route {
	case ChatRouteDirectReply, ChatRouteExecute, ChatRouteDeepAnswer, ChatRoutePlan:
		// ok
	case "":
		return fmt.Errorf("chat routing decision.route 不能为空")
	default:
		return fmt.Errorf("未知 route: %s", d.Route)
	}

	// direct_reply 必须有 speak。
	if d.Route == ChatRouteDirectReply && d.Speak == "" {
		return fmt.Errorf("direct_reply 必须携带 speak")
	}

	return nil
}