package tools

import (
	"context"
	"strings"

	"github.com/cloudwego/eino/components/tool"
	"github.com/cloudwego/eino/components/tool/utils"
)

var productsMock = []*Product{
	{ID: "P001", Name: "手机 A1", Price: 1999, Description: "入门级智能手机"},
	{ID: "P002", Name: "手机 Pro X", Price: 5999, Description: "旗舰级拍照手机"},
	{ID: "P003", Name: "笔记本 M3", Price: 8999, Description: "轻薄高性能笔记本"},
	{ID: "P004", Name: "耳机 Air", Price: 1299, Description: "降噪真无线耳机"},
	{ID: "271", Name: "商品 271", Price: 2710, Description: "样例商品用于测试按ID查询"},
}

// 工具返回
type Product struct {
	ID          string  `json:"id"`
	Name        string  `json:"name"`
	Price       float64 `json:"price"`
	Description string  `json:"description"`
}

// --- Eino ADK Tool 实现：按ID查询 ---
type ProductByIDInput struct {
	ID string `json:"id" jsonschema:"description=商品ID"`
}

func NewProductByIDTool() tool.InvokableTool {
	toolImpl, err := utils.InferTool("get_product_by_id", "根据商品ID查询商品详情，返回商品对象。", productByID)
	if err != nil {
		panic(err)
	}
	return toolImpl
}

func productByID(ctx context.Context, in *ProductByIDInput) (*Product, error) {
	for _, it := range productsMock {
		if it.ID == in.ID {
			return it, nil
		}
	}
	return &Product{}, nil
}

// --- Eino ADK Tool 实现：按名称搜索 ---

type ProductSearchInput struct {
	Name string `json:"name" jsonschema:"description=商品名称关键词"`
}

func NewProductSearchByNameTool() tool.InvokableTool {
	toolImpl, err := utils.InferTool("search_product_by_name", "根据商品名称关键词搜索商品列表，返回数组。", productSearchByName)
	if err != nil {
		panic(err)
	}
	return toolImpl
}

func productSearchByName(ctx context.Context, in *ProductSearchInput) ([]*Product, error) {
	if in.Name == "" {
		return []*Product{}, nil
	}
	q := strings.ToLower(in.Name)
	var out []*Product
	for _, it := range productsMock {
		if strings.Contains(strings.ToLower(it.Name), q) {
			out = append(out, it)
		}
	}
	return out, nil
}