package warehouse_search

import (
	"ai_scheduler/internal/config"
	"ai_scheduler/internal/pkg/l_request"
	"context"
	"encoding/json"
	"fmt"
)

type Client struct {
	cfg config.ToolConfig
}

func New(cfg config.ToolConfig) *Client {
	return &Client{
		cfg: cfg,
	}
}

func (c *Client) Call(ctx context.Context, name string) (int, error) {
	if name == "" {
		// 如果没有仓库名，返回0，不报错，由上层业务决定是否允许
		return 0, nil
	}

	// GET 请求参数
	params := map[string]string{
		"name":  name,
		"page":  "1",
		"limit": "1",
	}

	req := l_request.Request{
		Method: "Get",
		Url:    c.cfg.BaseURL,
		Params: params,
		Headers: map[string]string{
			"Content-Type": "application/json",
		},
	}

	res, err := req.Send()
	if err != nil {
		return 0, fmt.Errorf("请求失败，err: %v", err)
	}

	var resData SearchResponse
	if err := json.Unmarshal([]byte(res.Text), &resData); err != nil {
		return 0, fmt.Errorf("解析响应失败，err: %v", err)
	}

	if resData.Code != 200 {
		return 0, fmt.Errorf("业务错误，code: %d, msg: %s", resData.Code, resData.Msg)
	}

	if len(resData.Data.List) == 0 {
		return 0, fmt.Errorf("仓库不存在: %s", name)
	}

	return resData.Data.List[0].ID, nil
}
