package config

import (
	"fmt"
	"os"
	"regexp"
	"strings"

	"ai-agent-scaffold-go/internal/model"
	"gopkg.in/yaml.v3"
)

// agentRoot 对应 YAML 的 ai.agent.config.tables 结构
type agentRoot struct {
	AI struct {
		Agent struct {
			Config struct {
				Tables map[string]model.AiAgentConfigTable `yaml:"tables"`
			} `yaml:"config"`
		} `yaml:"agent"`
	} `yaml:"ai"`
}

// LoadAgentTables 从字节数据加载 Agent 配置表
func LoadAgentTables(data []byte) (map[string]model.AiAgentConfigTable, error) {
	expanded := expandEnvPlaceholders(string(data))
	var root agentRoot
	if err := yaml.Unmarshal([]byte(expanded), &root); err != nil {
		return nil, fmt.Errorf("parse agent config: %w", err)
	}
	tables := root.AI.Agent.Config.Tables
	if len(tables) == 0 {
		return nil, fmt.Errorf("agent config tables are required")
	}
	for name, table := range tables {
		normalizeDefaults(&table)
		if err := validateTable(name, table); err != nil {
			return nil, err
		}
		tables[name] = table
	}
	return tables, nil
}

// LoadAgentTablesFile 从文件路径加载 Agent 配置表
func LoadAgentTablesFile(path string) (map[string]model.AiAgentConfigTable, error) {
	data, err := os.ReadFile(path)
	if err != nil {
		return nil, fmt.Errorf("read agent config %s: %w", path, err)
	}
	return LoadAgentTables(data)
}

// envPlaceholderRE 匹配 ${VAR} 和 ${VAR:-default} 格式的环境变量占位符
var envPlaceholderRE = regexp.MustCompile(`\$\{([A-Za-z_][A-Za-z0-9_]*)(?::-([^}]*))?\}`)

// expandEnvPlaceholders 替换字符串中的环境变量占位符
func expandEnvPlaceholders(input string) string {
	return envPlaceholderRE.ReplaceAllStringFunc(input, func(match string) string {
		groups := envPlaceholderRE.FindStringSubmatch(match)
		name := groups[1]
		if value, ok := os.LookupEnv(name); ok && value != "" {
			return value
		}
		if len(groups) > 2 {
			return groups[2]
		}
		return ""
	})
}

// normalizeDefaults 填充配置默认值
func normalizeDefaults(table *model.AiAgentConfigTable) {
	if table.Module.AiAPI.CompletionsPath == "" {
		table.Module.AiAPI.CompletionsPath = "v1/chat/completions"
	}
	for i := range table.Module.AgentWorkflows {
		if table.Module.AgentWorkflows[i].MaxIterations == 0 {
			table.Module.AgentWorkflows[i].MaxIterations = 3
		}
	}
}

// validateTable 校验配置表的必填字段
func validateTable(name string, table model.AiAgentConfigTable) error {
	prefix := "agent table " + name
	required := map[string]string{
		"app-name":                 table.AppName,
		"agent.agent-id":           table.Agent.AgentID,
		"module.ai-api.base-url":   table.Module.AiAPI.BaseURL,
		"module.ai-api.api-key":    table.Module.AiAPI.APIKey,
		"module.chat-model.model":  table.Module.ChatModel.Model,
		"module.runner.agent-name": table.Module.Runner.AgentName,
	}
	for field, value := range required {
		if strings.TrimSpace(value) == "" {
			return fmt.Errorf("%s: %s is required", prefix, field)
		}
	}
	if len(table.Module.Agents) == 0 {
		return fmt.Errorf("%s: module.agents is required", prefix)
	}
	for i, agent := range table.Module.Agents {
		if strings.TrimSpace(agent.Name) == "" {
			return fmt.Errorf("%s: module.agents[%d].name is required", prefix, i)
		}
		if strings.TrimSpace(agent.Instruction) == "" {
			return fmt.Errorf("%s: module.agents[%d].instruction is required", prefix, i)
		}
	}
	for i, workflow := range table.Module.AgentWorkflows {
		switch workflow.Type {
		case model.WorkflowTypeLoop, model.WorkflowTypeParallel, model.WorkflowTypeSequential:
		default:
			return fmt.Errorf("%s: module.agent-workflows[%d].type is invalid: %s", prefix, i, workflow.Type)
		}
		if strings.TrimSpace(workflow.Name) == "" {
			return fmt.Errorf("%s: module.agent-workflows[%d].name is required", prefix, i)
		}
	}
	return nil
}