package cc.mrbird.febs.yinhe.controller;

import cc.mrbird.febs.ai.req.product.ApiProductInfoDto;
import cc.mrbird.febs.ai.req.product.ApiProductPageDto;
import cc.mrbird.febs.ai.res.product.ApiProductInfoVo;
import cc.mrbird.febs.ai.res.product.ApiProductVo;
import cc.mrbird.febs.ai.res.productPoint.ApiProductPointVo;
import cc.mrbird.febs.ai.service.AiProductService;
import cc.mrbird.febs.common.entity.FebsResponse;
import cc.mrbird.febs.mall.vo.MallMemberVo;
import cc.mrbird.febs.yinhe.req.*;
import cc.mrbird.febs.yinhe.res.*;
import cc.mrbird.febs.yinhe.service.YhAiService;
import cn.hutool.core.util.StrUtil;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import io.swagger.annotations.ApiResponse;
import io.swagger.annotations.ApiResponses;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import reactor.core.publisher.Flux;

import javax.servlet.http.HttpServletRequest;

/**
 * @author Administrator
 */
@Slf4j
@Validated
@RestController
@RequiredArgsConstructor
@RequestMapping(value = "/api/yh/agent")
@Api(value = "YHApiAgentController", tags = "AI-银和-智能体相关接口")
public class YHApiAgentController {

    private final YhAiService service;
    @ApiOperation(value = "获取用户信息", notes = "获取用户信息")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHMemberVo.class)
    })
    @GetMapping(value = "/memberInfo")
    public FebsResponse memberInfo() {

        return service.memberInfo();
    }

    @ApiOperation(value = "公司-智能体分类", notes = "公司-智能体分类")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHCategoryVo.class)
    })
    @PostMapping(value = "/categoryList")
    public FebsResponse categoryList(@RequestBody @Validated YHCategoryDto dto) {

        return service.categoryList(dto);
    }

    @ApiOperation(value = "公司-智能体", notes = "公司-智能体")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHAgentVo.class)
    })
    @PostMapping(value = "/agentList")
    public FebsResponse agentList(@RequestBody @Validated YHAgentDto dto) {

        return service.agentList(dto);
    }

    @ApiOperation(value = "公司-智能体-查看详情", notes = "公司-智能体-查看详情")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHAgentInitVo.class)
    })
    @PostMapping(value = "/agentInfo")
    public FebsResponse agentInfo(@RequestBody @Validated YHAgentInitDto dto) {

        return service.agentInfo(dto);
    }

    @ApiOperation(value = "首页-智能体分类", notes = "首页-智能体分类")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHSysCategoryVo.class)
    })
    @PostMapping(value = "/categorySysList")
    public FebsResponse categoryList(@RequestBody @Validated YHSysCategoryDto dto) {

        return service.categorySysList(dto);
    }

    @ApiOperation(value = "首页-智能体", notes = "首页-智能体")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHSysAgentVo.class)
    })
    @PostMapping(value = "/agentSysList")
    public FebsResponse agentList(@RequestBody @Validated YHSysAgentDto dto) {

        return service.agentSysList(dto);
    }

    @ApiOperation(value = "首页-智能体-查看详情", notes = "首页-智能体-查看详情")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHSysAgentInitVo.class)
    })
    @PostMapping(value = "/agentSysInfo")
    public FebsResponse agentSysInfo(@RequestBody @Validated YHSysAgentInitDto dto) {

        return service.agentSysInfo(dto);
    }

    @ApiOperation(value = "退出", notes = "退出")
    @PostMapping(value = "/logout")
    public FebsResponse logout() {

        return service.logout();
    }


    @ApiOperation(value = "初始化对话", notes = "初始化对话")
    @ApiResponses({
            @ApiResponse(code = 200, message = "success", response = YHSendInitVo.class)
    })
    @PostMapping(value = "/initSend", produces = "application/json")
    public FebsResponse initSend(@RequestBody @Validated YHSendInitDto dto) {

        return service.initSend(dto);
    }

    @ApiOperation(value = "保存对话消息", notes = "保存对话消息")
    @PostMapping(value = "/saveContext")
    public FebsResponse saveContext(@RequestBody @Validated YHSaveContextDto dto) {

        return service.saveContext(dto);
    }


    @ApiOperation("AI回答(流式)")
    @ApiResponses({
            @ApiResponse(code = 200, message = "流式响应", response = YHAitalkItemStreamVo.class),
    })
    @PostMapping("/aiAnswer")
    public Flux<FebsResponse> aiAnswer(@RequestBody @Validated YHAitalkItemStreamDto dto) {
        if (StrUtil.isEmpty(dto.getTalkId()) || StrUtil.isEmpty(dto.getReqContext())|| StrUtil.isEmpty(dto.getReqContext())){
            return Flux.just(new FebsResponse().fail().message("参数异常"));
        }
        return service.aiAnswer(dto);
    }

}