package com.yanzhu.module.ai.tool.function;

import com.yanzhu.module.ai.util.AiUtils;
import com.yanzhu.framework.common.util.object.BeanUtils;
import com.yanzhu.framework.security.core.LoginUser;
import com.yanzhu.framework.tenant.core.util.TenantUtils;
import com.yanzhu.module.system.api.user.AdminUserApi;
import com.yanzhu.module.system.api.user.dto.AdminUserRespDTO;
import com.fasterxml.jackson.annotation.JsonClassDescription;
import com.fasterxml.jackson.annotation.JsonProperty;
import com.fasterxml.jackson.annotation.JsonPropertyDescription;
import javax.annotation.Resource;
import lombok.AllArgsConstructor;
import lombok.Data;
import lombok.NoArgsConstructor;
import org.springframework.stereotype.Component;
import dev.langchain4j.agent.tool.ToolExecutionRequest;
import dev.langchain4j.agent.tool.ToolParameters;
import dev.langchain4j.agent.tool.ToolSpecification;
import dev.langchain4j.service.AiServices;
import dev.langchain4j.service.MemoryId;
import dev.langchain4j.service.SystemMessage;
import dev.langchain4j.service.UserMessage;
import dev.langchain4j.service.V;

import java.util.function.BiFunction;

/**
 * 工具：用户信息查询
 *
 * 同时，也是展示 ToolContext 上下文的使用
 *
 * @author Ren
 */
@Component("user_profile_query")
public class UserProfileQueryToolFunction {

    @Resource
    private AdminUserApi adminUserApi;

    @Data
    @JsonClassDescription("用户信息查询")
    public static class Request {

        /**
         * 用户编号
         */
        @JsonProperty(value = "id")
        @JsonPropertyDescription("用户编号，例如说：1。如果查询自己，则 id 为空")
        private Long id;

    }

    @Data
    @AllArgsConstructor
    @NoArgsConstructor
    public static class Response {

        /**
         * 用户ID
         */
        private Long id;
        /**
         * 用户昵称
         */
        private String nickname;

        /**
         * 手机号码
         */
        private String mobile;
        /**
         * 用户头像
         */
        private String avatar;

    }

    public Response getUserInfo(Long userId) {
        // 在LangChain4J工具中，我们无法直接访问租户上下文
        // 这里简化实现，实际应用中可能需要其他方式传递租户上下文
        if (userId == null) {
            // 如果没有传入用户ID，返回空响应
            return new Response();
        }
        
        // 假设使用默认租户或系统租户来查询用户信息
        return TenantUtils.execute(0L, () -> {
            AdminUserRespDTO user = adminUserApi.getUser(userId);
            return BeanUtils.toBean(user, Response.class);
        });
    }

}
