package jnpf.base.controller;

import cn.hutool.core.collection.CollUtil;
import io.swagger.v3.oas.annotations.Operation;
import io.swagger.v3.oas.annotations.Parameter;
import io.swagger.v3.oas.annotations.tags.Tag;
import jakarta.validation.Valid;
import jnpf.base.ActionResult;
import jnpf.base.AiApi;
import jnpf.base.entity.AiEntity;
import jnpf.base.model.ai.*;
import jnpf.base.model.print.PrintDevFormDTO;
import jnpf.base.service.AiService;
import jnpf.base.vo.PageListVO;
import jnpf.base.vo.PaginationVO;
import jnpf.constant.MsgCode;
import jnpf.exception.WorkFlowException;
import jnpf.model.ai.*;
import jnpf.permission.UserApi;
import jnpf.permission.entity.UserEntity;
import jnpf.util.AiLimitUtil;
import jnpf.util.JsonUtil;
import jnpf.util.UserProvider;
import lombok.RequiredArgsConstructor;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

@Tag(name = "AI列表", description = "ai")
@RestController
@RequestMapping("/Ai")
@RequiredArgsConstructor
public class AiController implements AiApi {


    private final AiService aiService;

    private final UserApi userService;

    @Operation(summary = "列表")
    @GetMapping
    public ActionResult<PageListVO<AiListVO>> list(AiPagination paginationPrint) {
        List<AiEntity> list = aiService.getList(paginationPrint);
        List<String> userId = list.stream().map(AiEntity::getCreatorUserId).collect(Collectors.toList());
        List<String> lastUserId = list.stream().map(AiEntity::getLastModifyUserId).collect(Collectors.toList());
        lastUserId.removeAll(Collections.singleton(null));
        List<UserEntity> userEntities = userService.getUserName(userId);
        List<UserEntity> lastUserIdEntities = userService.getUserName(lastUserId);
        List<AiListVO> listVOS = new ArrayList<>();
        for (AiEntity entity : list) {
            AiListVO vo = JsonUtil.getJsonToBean(entity, AiListVO.class);
            //创建者
            UserEntity creatorUser = userEntities.stream().filter(t -> t.getId().equals(entity.getCreatorUserId())).findFirst().orElse(null);
            vo.setCreatorUser(creatorUser != null ? creatorUser.getRealName() + "/" + creatorUser.getAccount() : entity.getCreatorUserId());
            //修改人
            UserEntity lastModifyUser = lastUserIdEntities.stream().filter(t -> t.getId().equals(entity.getLastModifyUserId())).findFirst().orElse(null);
            vo.setLastModifyUser(lastModifyUser != null ? lastModifyUser.getRealName() + "/" + lastModifyUser.getAccount() : entity.getLastModifyUserId());
            listVOS.add(vo);
        }
        PaginationVO paginationVO = JsonUtil.getJsonToBean(paginationPrint, PaginationVO.class);
        return ActionResult.page(listVOS, paginationVO);
    }

    @Operation(summary = "新增")
    @Parameter(name = "aiCrForm", description = "对象")
    @PostMapping
    public ActionResult<Object> create(@RequestBody @Valid AICrForm aiCrForm) {
        AiEntity entity = JsonUtil.getJsonToBean(aiCrForm, AiEntity.class);
        aiService.create(entity);
        return ActionResult.success(MsgCode.SU001.get());
    }

    @Operation(summary = "详情")
    @Parameter(name = "id", description = "id")
    @GetMapping("/{id}")
    public ActionResult<AiInfoVO> info(@PathVariable("id") String id) {
        AiEntity byId = aiService.getInfo(id);
        AiInfoVO vo = JsonUtil.getJsonToBean(byId, AiInfoVO.class);
        return ActionResult.success(vo);
    }

    @Operation(summary = "更新")
    @PutMapping("/{id}")
    @Parameter(name = "id", description = "主键", required = true)
    @Parameter(name = "form", description = "模型", required = true)
    public ActionResult<Object> update(@PathVariable("id") String id, @RequestBody @Valid AiUpForm aiForm) throws WorkFlowException {
        AiEntity entity = JsonUtil.getJsonToBean(aiForm, AiEntity.class);
        AiEntity aiEntity = aiService.getInfo(id);
        if (aiEntity == null) {
            return ActionResult.fail(MsgCode.FA003.get());
        }
        aiService.update(id, entity);
        return ActionResult.success(MsgCode.SU004.get());
    }

    @Operation(summary = "删除")
    @Parameter(name = "id", description = "打印模板id")
    @DeleteMapping("/{id}")
    public ActionResult<PrintDevFormDTO> delete(@PathVariable("id") String id) {
        AiEntity entity = aiService.getById(id);
        if (entity == null) {
            return ActionResult.fail(MsgCode.FA003.get());
        }
        if (Objects.equals(entity.getEnabledMark(), 1)) {
            return ActionResult.fail(MsgCode.SYS184.get());
        }
        aiService.delete(entity);
        return ActionResult.success(MsgCode.SU003.get());
    }

    @Operation(summary = "测试")
    @PostMapping("/checkAi")
    public ActionResult<PrintDevFormDTO> checkAi(@RequestBody AICrForm aiCrForm) {
        if (!AiLimitUtil.tryAcquire(UserProvider.getLoginId())) {
            return ActionResult.fail(MsgCode.SYS182.get());
        }
        AiModel aiModel = JsonUtil.getJsonToBean(aiCrForm, AiModel.class);
        AiSendModel sendModel = new AiSendModel();
        sendModel.setModel(aiCrForm.getModel());

        List<Message> list = new ArrayList<>();
        Message userMessage = new Message();
        userMessage.setRole(SendAiMessage.USER);
        userMessage.setContent("Hello");
        list.add(userMessage);
        sendModel.setMessages(list);

        AiReceiveModel receiveModel = SendAiMessage.sendMessageReceiver(sendModel, aiModel);
        if (receiveModel == null) {
            return ActionResult.fail(MsgCode.SYS185.get());
        }
        if (CollUtil.isEmpty(receiveModel.getChoices())) {
            return ActionResult.fail(MsgCode.SYS187.get());
        }
        return ActionResult.success(MsgCode.SU017.get());
    }

    @Override
    @GetMapping("/getDefault")
    public AiEntity getDefault() {
        return aiService.getDefault();
    }
}