// Package tool 定义 Tool 接口和工具管理系统。
//
// 工具是可以由智能体或 LLM 调用的原子能力。
// 每个工具都有名称、描述、参数模式（用于 LLM 函数调用）
// 和执行实际工作的 Execute 方法。
package tool

import (
	"context"
	"fmt"
	"sort"
	"sync"
)

// Manager 是管理工具注册和执行的安全注册表。
//
// 工具按名称（区分大小写）注册，可以通过管理器发现、
// 列出和调用。重复注册将返回错误。
type Manager struct {
	mu    sync.RWMutex
	tools map[string]Tool
}

// NewManager 创建新的空工具管理器。
func NewManager() *Manager {
	return &Manager{
		tools: make(map[string]Tool),
	}
}

// Register 将工具添加到管理器。如果已注册了同名工具，则返回错误。
func (m *Manager) Register(tool Tool) error {
	m.mu.Lock()
	defer m.mu.Unlock()

	name := tool.Name()
	if _, exists := m.tools[name]; exists {
		return fmt.Errorf("tool %q is already registered", name)
	}

	m.tools[name] = tool
	return nil
}

// Unregister 按名称从管理器中移除工具。
func (m *Manager) Unregister(name string) error {
	m.mu.Lock()
	defer m.mu.Unlock()

	if _, exists := m.tools[name]; !exists {
		return fmt.Errorf("tool %q is not registered", name)
	}

	delete(m.tools, name)
	return nil
}

// Get 按名称检索工具。如果未找到则返回 false。
func (m *Manager) Get(name string) (Tool, bool) {
	m.mu.RLock()
	defer m.mu.RUnlock()

	t, ok := m.tools[name]
	return t, ok
}

// List 返回按名称排序的所有已注册工具。
func (m *Manager) List() []Tool {
	m.mu.RLock()
	defer m.mu.RUnlock()

	result := make([]Tool, 0, len(m.tools))
	for _, t := range m.tools {
		result = append(result, t)
	}

	sort.Slice(result, func(i, j int) bool {
		return result[i].Name() < result[j].Name()
	})
	return result
}

// Execute 按名称查找工具并使用给定参数调用它。
// 如果未找到工具则返回错误。
func (m *Manager) Execute(name string, ctx context.Context, args map[string]interface{}) (*Result, error) {
	tool, ok := m.Get(name)
	if !ok {
		return nil, fmt.Errorf("tool %q not found", name)
	}
	return tool.Execute(ctx, args)
}

// Count 返回已注册工具的数量。
func (m *Manager) Count() int {
	m.mu.RLock()
	defer m.mu.RUnlock()
	return len(m.tools)
}

// Names 返回按字母顺序排序的所有已注册工具名称。
func (m *Manager) Names() []string {
	m.mu.RLock()
	defer m.mu.RUnlock()

	names := make([]string, 0, len(m.tools))
	for name := range m.tools {
		names = append(names, name)
	}
	sort.Strings(names)
	return names
}