package types

import (
	"github.com/yaoapp/yao/agent/assistant"
	searchTypes "github.com/yaoapp/yao/agent/search/types"
	store "github.com/yaoapp/yao/agent/store/types"
)

// DSL AI assistant
type DSL struct {

	// Agent Global Settings
	// ===============================
	Uses         *Uses         `json:"uses,omitempty" yaml:"uses,omitempty"` // Which assistant to use default, title, prompt
	StoreSetting store.Setting `json:"store" yaml:"store"`                   // The store setting of the assistant
	Cache        string        `json:"cache" yaml:"cache"`                   // The cache store of the assistant, if not set, default is "__yao.agent.cache"

	// System Agents Connector Settings
	// ===============================
	// System configures connectors for system agents (__yao.keyword, __yao.querydsl, __yao.title, __yao.prompt)
	// Each agent can have its own connector, or use the default
	// If not set, fallback to the first connector that supports the required capabilities
	System *System `json:"system,omitempty" yaml:"system,omitempty"`

	// Global External Settings
	// ===============================
	KB     *store.KBSetting    `json:"kb,omitempty" yaml:"kb,omitempty"`         // The knowledge base configuration loaded from agent/kb.yml
	Search *searchTypes.Config `json:"search,omitempty" yaml:"search,omitempty"` // The search configuration loaded from agent/search.yao

	// Internal
	// ===============================
	// ID            string            `json:"-" yaml:"-"` // The id of the instance
	Assistant     assistant.API  `json:"-" yaml:"-"` // The default assistant
	Store         store.Store    `json:"-" yaml:"-"` // The store of the assistant
	GlobalPrompts []store.Prompt `json:"-" yaml:"-"` // Global prompts loaded from agent/prompts.yml
}

// Uses the default assistant settings
// ===============================
type Uses struct {
	Default     string `json:"default,omitempty" yaml:"default,omitempty"`           // The default assistant to use
	Title       string `json:"title,omitempty" yaml:"title,omitempty"`               // The assistant for generating the topic title.
	Prompt      string `json:"prompt,omitempty" yaml:"prompt,omitempty"`             // The assistant for generating the prompt.
	RobotPrompt string `json:"robot_prompt,omitempty" yaml:"robot_prompt,omitempty"` // The assistant for generating Robot's system prompt (responsibilities description).
	Vision      string `json:"vision,omitempty" yaml:"vision,omitempty"`             // The assistant for generating the image/video description, if the assistant enable the vision and model not support vision, use the vision model to describe the image/video, and return the messages with the image/video's description. Format: "agent" or "mcp:mcp_server_id"
	Audio       string `json:"audio,omitempty" yaml:"audio,omitempty"`               // The assistant for processing audio (speech-to-text, text-to-speech). If the model doesn't support audio, use this to convert audio to text. Format: "agent" or "mcp:mcp_server_id"
	Search      string `json:"search,omitempty" yaml:"search,omitempty"`             // The assistant for searching the knowledge, global web search. If not set, and the assistant enable the knowledge, it will search the result from the knowledge automatically.
	Fetch       string `json:"fetch,omitempty" yaml:"fetch,omitempty"`               // The assistant for fetching the http/https/ftp/sftp/etc. file, and return the file's content. if not set, use the http process to fetch the file.

	// Search-related processing tools (NLP)
	Web      string `json:"web,omitempty" yaml:"web,omitempty"`           // Web search handler: "builtin", "<assistant-id>", "mcp:<server>.<tool>"
	Keyword  string `json:"keyword,omitempty" yaml:"keyword,omitempty"`   // Keyword extraction: "builtin", "<assistant-id>", "mcp:<server>.<tool>"
	QueryDSL string `json:"querydsl,omitempty" yaml:"querydsl,omitempty"` // QueryDSL generation: "builtin", "<assistant-id>", "mcp:<server>.<tool>"
	Rerank   string `json:"rerank,omitempty" yaml:"rerank,omitempty"`     // Result reranking: "builtin", "<assistant-id>", "mcp:<server>.<tool>"

	// Robot pipeline phase agents (application-level, not bundled as system agents)
	// Empty means no default — must be configured per-robot via resources.phases or here globally.
	Inspiration string `json:"inspiration,omitempty" yaml:"inspiration,omitempty"` // P0: Inspiration phase agent
	Goals       string `json:"goals,omitempty" yaml:"goals,omitempty"`             // P1: Goals planning agent
	Tasks       string `json:"tasks,omitempty" yaml:"tasks,omitempty"`             // P2: Task breakdown agent
	Delivery    string `json:"delivery,omitempty" yaml:"delivery,omitempty"`       // P4: Delivery composition agent
	Learning    string `json:"learning,omitempty" yaml:"learning,omitempty"`       // P5: Learning extraction agent
	Host        string `json:"host,omitempty" yaml:"host,omitempty"`               // Host: Human interaction agent
	Validation  string `json:"validation,omitempty" yaml:"validation,omitempty"`   // Validation: Task output validation agent
}

// GetPhaseAgent returns the globally configured agent ID for a robot pipeline phase.
// Returns empty string if no global default is set for the phase.
func (u *Uses) GetPhaseAgent(phase string) string {
	if u == nil {
		return ""
	}
	switch phase {
	case "inspiration":
		return u.Inspiration
	case "goals":
		return u.Goals
	case "tasks":
		return u.Tasks
	case "delivery":
		return u.Delivery
	case "learning":
		return u.Learning
	case "host":
		return u.Host
	case "validation":
		return u.Validation
	default:
		return ""
	}
}

// System configures connectors for system agents
// ===============================
type System struct {
	// Role-level defaults (written to llmprovider via SetDefaults)
	Default string `json:"default,omitempty" yaml:"default,omitempty"` // Default connector for the "default" role
	Light   string `json:"light,omitempty" yaml:"light,omitempty"`     // Default connector for the "light" role (titles, keywords, summaries)
	Vision  string `json:"vision,omitempty" yaml:"vision,omitempty"`   // Default connector for the "vision" role
	Audio   string `json:"audio,omitempty" yaml:"audio,omitempty"`     // Default connector for the "audio" role

	// Per-agent overrides (optional, highest priority — bypasses role resolution)
	Keyword     string `json:"keyword,omitempty" yaml:"keyword,omitempty"`           // Connector for __yao.keyword agent
	QueryDSL    string `json:"querydsl,omitempty" yaml:"querydsl,omitempty"`         // Connector for __yao.querydsl agent
	Title       string `json:"title,omitempty" yaml:"title,omitempty"`               // Connector for __yao.title agent
	Prompt      string `json:"prompt,omitempty" yaml:"prompt,omitempty"`             // Connector for __yao.prompt agent
	RobotPrompt string `json:"robot_prompt,omitempty" yaml:"robot_prompt,omitempty"` // Connector for __yao.robot_prompt agent
	NeedSearch  string `json:"needsearch,omitempty" yaml:"needsearch,omitempty"`     // Connector for __yao.needsearch agent
	Entity      string `json:"entity,omitempty" yaml:"entity,omitempty"`             // Connector for __yao.entity agent
}

// Mention Structure
// ===============================
type Mention struct {
	ID     string `json:"id"`
	Name   string `json:"name"`
	Avatar string `json:"avatar,omitempty"`
	Type   string `json:"type,omitempty"`
}