package com.ruoyi.web.controller.api;


import com.alibaba.fastjson.JSON;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.ruoyi.common.annotation.Log;
import com.ruoyi.common.basic.PageInfo;
import com.ruoyi.common.core.domain.BaseModel;
import com.ruoyi.common.core.domain.R;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.common.utils.bean.BeanUtils;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.*;
import com.ruoyi.system.model.*;
import com.ruoyi.system.query.TProjectTeamQuery;
import com.ruoyi.system.query.TTrainLibraryDetailQuery;
import com.ruoyi.system.query.TTrainLibraryQuery;
import com.ruoyi.system.service.*;
import com.ruoyi.system.vo.ChoiceUserVO;
import com.ruoyi.system.vo.TProjectTeamVO;
import com.ruoyi.system.vo.TTrainLibraryDetailVO;
import com.ruoyi.system.vo.TTrainLibraryVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.web.bind.annotation.*;

import javax.annotation.Resource;
import java.time.LocalDateTime;
import java.util.ArrayList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * <p>
 * 菌种库 前端控制器
 * </p>
 *
 * @author 无关风月
 * @since 2025-04-12
 */
@RestController
@RequestMapping("")
public class TTrainLibraryController {
    @Resource
    private TTrainLibraryService tTrainLibraryService;
    @Resource
    private TTrainLibraryWarehousingService tTrainLibraryWarehousingService;
    @Resource
    private TokenService tokenService;
    @Resource
    private ISysUserService sysUserService;
    @Resource
    private TProjectTeamService projectTeamService;
    @Resource
    private TProjectTeamStaffService projectTeamStaffService;
    /**
     * 选择人员
     */
    @ApiOperation(value = "选择人员",tags = "菌种库管理",response = ChoiceStrainDTO.class)
    @PostMapping(value = "/api/t-train-library/choiceUserList")
    public R<PageInfo<ChoiceUserVO>> choiceUserList(@RequestBody String param) {
        ChoiceStrainDTO query = JSON.parseObject(param, ChoiceStrainDTO.class);
        List<String> userIds = projectTeamStaffService.lambdaQuery()
                .eq(TProjectTeamStaff::getRoleType,  query.getRoleType())
                .eq(TProjectTeamStaff::getTeamId, query.getProjectId()).list().stream().map(TProjectTeamStaff::getUserId).collect(Collectors.toList());
        if (userIds.isEmpty()){
            userIds.add("0");
        }
        query.setUserIds(userIds);

        return R.ok(sysUserService.choiceUserList(query));
    }
    /**
     * 选择项目组
     */
    @ApiOperation(value = "选择项目组",tags = "菌种库管理",response = ChoiceProjectDTO.class)
    @PostMapping(value = "/api/t-train-library/choiceProjectList")
    public R<PageInfo<TProjectTeamVO>> choiceProjectList(@RequestBody String param) {
        LoginUser loginUser = tokenService.getLoginUser();
        SysUser sysUser = sysUserService.selectUserById(loginUser.getUserId());
        List<String> collect = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, sysUser.getUserId()).list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
        if (collect.isEmpty()){
            collect.add("0");
        }
        ChoiceProjectDTO query = JSON.parseObject(param, ChoiceProjectDTO.class);
        query.setProjectIds(collect);
        return R.ok(projectTeamService.choiceProjectList(query));
    }
    // 提取为独立方法以提高可读性和复用性

    /**
     * 获取菌种库分页列表
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:list')")
    @ApiOperation(value = "获取菌种库分页列表",tags = "菌种库管理",response = TTrainLibraryQuery.class)
    @PostMapping(value = "/api/t-train-library/pageList")
    public R<PageInfo<TTrainLibraryVO>> pageList(@RequestBody String param) {
        TTrainLibraryQuery query = JSON.parseObject(param, TTrainLibraryQuery.class);
        return R.ok(tTrainLibraryService.pageList(query));
    }
    /**
     * 获取菌种库出入库时间轴列表
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:timeList')")
    @ApiOperation(value = "获取菌种库出入库时间轴列表",tags = "菌种库管理")
    @PostMapping(value = "/api/t-train-library/timeList")
    public R<List<TTrainLibraryWarehousing>> timeList(@RequestParam String id) {
        List<TTrainLibraryWarehousing> list = tTrainLibraryWarehousingService.lambdaQuery().eq(TTrainLibraryWarehousing::getTrainLibraryId,id)
                .orderByDesc(TTrainLibraryWarehousing::getCreateTime).list();
        List<SysUser> sysUsers = sysUserService.selectList();
        for (TTrainLibraryWarehousing tTrainLibraryWarehousing : list) {
            if (tTrainLibraryWarehousing.getHandleId()!=null){
                SysUser sysUser = sysUsers.stream().filter(e -> e.getUserId().equals(tTrainLibraryWarehousing.getHandleId())).findFirst().orElse(null);
                if (sysUser!=null){
                    tTrainLibraryWarehousing.setHandleName(sysUser.getNickName());
                }
            }
            if (tTrainLibraryWarehousing.getPreserveId()!=null){
                SysUser sysUser = sysUsers.stream().filter(e -> e.getUserId().equals(tTrainLibraryWarehousing.getPreserveId())).findFirst().orElse(null);
                if (sysUser!=null){
                    tTrainLibraryWarehousing.setPreserveName(sysUser.getNickName());
                }
            }
        }
        return R.ok(list);
    }
    @Resource
    private TNoticeService noticeService;
    /**
     * 新增菌种库
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:add')")
    @Log(title = "菌种库管理-新增菌种库", businessType = BusinessType.INSERT)
    @ApiOperation(value = "新增菌种库",response = TTrainLibraryDTO.class,tags = "菌种库管理")
    @PostMapping(value = "/api/t-train-library/add")
    public R<Boolean> add(@RequestBody String param) {
        if (tokenService.getLoginUser()==null){
            return R.fail(401,"登陆失效");
        }
        TTrainLibraryDTO dto = JSON.parseObject(param,TTrainLibraryDTO.class);
        dto.setStock(1);
        tTrainLibraryService.save(dto);
        if (dto.getIsDraft()==0){
            // 新增入库记录
            TTrainLibraryWarehousing tTrainLibraryWarehousing = new TTrainLibraryWarehousing();
            tTrainLibraryWarehousing.setTrainLibraryId(dto.getId());
            tTrainLibraryWarehousing.setType(2);
            tTrainLibraryWarehousing.setBoundTime(LocalDateTime.now());
            tTrainLibraryWarehousing.setHandleId(tokenService.getLoginUser().getUserId());
            tTrainLibraryWarehousing.setHandleSignature(dto.getHandleSignature());
            tTrainLibraryWarehousingService.save(tTrainLibraryWarehousing);
            // 新增消息
            TNotice tNotice = new TNotice();
            List<String> collect = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
                    .eq(TProjectTeamStaff::getRoleType, 2)
                    .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
            TProjectTeam one = projectTeamService.lambdaQuery().in(TProjectTeam::getId, collect).eq(TProjectTeam::getStatus, 1)
                    .orderByDesc(BaseModel::getCreateTime).last("limit 1").one();
            TProjectTeamStaff one1 = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getTeamId, one.getId()).eq(TProjectTeamStaff::getRoleType, 2).one();
            tNotice.setUserId(one1.getUserId());

            tNotice.setCommitId(tokenService.getLoginUser().getUserId()+"");
            tNotice.setStatus(1);
            StringBuilder stringBuilder = new StringBuilder();
            switch (dto.getType()){
                case 1:
                    stringBuilder.append("【原始细胞库】");
                    break;
                case 2:
                    stringBuilder.append("【主细胞库】");
                    break;
                case 3:
                    stringBuilder.append("【生产细胞库】");
                    break;
            }

            tNotice.setNoticeContent("您有【1】条"+stringBuilder+"入库记录等待签字确认;");
            tNotice.setDataId(dto.getId());
            tNotice.setNoticeType(3);
            noticeService.save(tNotice);
        }
        return R.ok();
    }
    /**
     * 新增菌种库
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:addBatch')")
    @Log(title = "菌种库管理-批量新增菌种库", businessType = BusinessType.INSERT)
    @ApiOperation(value = "批量新增菌种库",response = TTrainLibraryDTO.class,tags = "菌种库管理")
    @PostMapping(value = "/api/t-train-library/addBatch")
    public R addBatch(@RequestBody String param) {
        List<TTrainLibrary> addList = new ArrayList<>();
        TTrainLibraryDTO dto = JSON.parseObject(param,TTrainLibraryDTO.class);
        for (int i = 0; i < dto.getBatchCount(); i++) {
            TTrainLibrary tTrainLibrary = new TTrainLibrary();
            BeanUtils.copyProperties(dto,tTrainLibrary);
            if (i==0){
                tTrainLibrary.setStatus(3);
                tTrainLibrary.setIsDraft(0);
                tTrainLibraryService.save(tTrainLibrary);
                TTrainLibraryWarehousing tTrainLibraryWarehousing = new TTrainLibraryWarehousing();
                tTrainLibraryWarehousing.setTrainLibraryId(tTrainLibrary.getId());
                tTrainLibraryWarehousing.setType(2);
                tTrainLibraryWarehousing.setHandleId(tokenService.getLoginUser().getUserId());
                tTrainLibraryWarehousing.setBoundTime(LocalDateTime.now());
                tTrainLibraryWarehousingService.save(tTrainLibraryWarehousing);
                // 新增消息
                TNotice tNotice = new TNotice();
                List<String> collect = projectTeamStaffService.lambdaQuery().eq(TProjectTeamStaff::getUserId, tokenService.getLoginUser().getUserId())
                        .eq(TProjectTeamStaff::getRoleType, 2)
                        .list().stream().map(TProjectTeamStaff::getTeamId).collect(Collectors.toList());
                TProjectTeam one = projectTeamService.lambdaQuery().in(TProjectTeam::getId, collect)
                        .eq(TProjectTeam::getStatus, 1)
                        .orderByDesc(BaseModel::getCreateTime).last("limit 1").one();
                TProjectTeamStaff one1 = projectTeamStaffService.lambdaQuery()
                        .eq(TProjectTeamStaff::getTeamId, one.getId()).eq(TProjectTeamStaff::getRoleType, 2).one();
                tNotice.setUserId(one1.getUserId());

                tNotice.setCommitId(tokenService.getLoginUser().getUserId()+"");
                tNotice.setStatus(1);
                StringBuilder stringBuilder = new StringBuilder();
                switch (dto.getType()){
                    case 1:
                        stringBuilder.append("【原始细胞库】");
                        break;
                    case 2:
                        stringBuilder.append("【主细胞库】");
                        break;
                    case 3:
                        stringBuilder.append("【生产细胞库】");
                        break;
                }

                tNotice.setNoticeContent("您有【1】条"+stringBuilder+"入库记录等待签字确认;");
                tNotice.setDataId(dto.getId());
                tNotice.setNoticeType(3);
                noticeService.save(tNotice);
            }else{
                tTrainLibrary.setStrainCode(null);
                tTrainLibrary.setIsDraft(1);
                addList.add(tTrainLibrary);
            }
        }
        dto.setIsDraft(1);
        return R.ok(tTrainLibraryService.saveBatch(addList));
    }
    /**
     * 编辑菌种库
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:update')")
    @Log(title = "菌种库管理-编辑菌种库", businessType = BusinessType.UPDATE)
    @ApiOperation(value = "编辑菌种库",response = TTrainLibraryDTO.class,tags = "菌种库管理")
    @PostMapping(value = "/api/t-train-library/update")
    public R update(@RequestBody String param) {
        if (tokenService.getLoginUser()==null){
            return R.fail(401,"登陆失效");
        }
        TTrainLibraryDTO dto = JSON.parseObject(param,TTrainLibraryDTO.class);
        tTrainLibraryService.updateById(dto);
        if (dto.getIsDraft()==0){
            // 新增入库记录
            TTrainLibraryWarehousing tTrainLibraryWarehousing = new TTrainLibraryWarehousing();
            tTrainLibraryWarehousing.setTrainLibraryId(dto.getId());
            tTrainLibraryWarehousing.setType(2);
            tTrainLibraryWarehousing.setBoundTime(LocalDateTime.now());
            tTrainLibraryWarehousing.setHandleId(tokenService.getLoginUser().getUserId());
            tTrainLibraryWarehousing.setHandleSignature(dto.getHandleSignature());
            tTrainLibraryWarehousingService.updateById(tTrainLibraryWarehousing);
        }
        return R.ok();
    }
    /**
     * 菌种工程师确认出入库
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:confirm')")
    @Log(title = "菌种库管理-菌种工程师确认出入库", businessType = BusinessType.UPDATE)
    @ApiOperation(value = "菌种工程师确认出入库",tags = "菌种库管理",response = TTrainLibraryDetailConfirmDto.class)
    @PostMapping(value = "/api/t-train-library/confirm")
    public R confirm(@RequestBody String param) {
        if (tokenService.getLoginUser()==null){
            return R.fail(401,"登陆失效");
        }
        TTrainLibraryDetailConfirmDto query = JSON.parseObject(param, TTrainLibraryDetailConfirmDto.class);
        TTrainLibraryWarehousing tTrainLibraryWarehousing = tTrainLibraryWarehousingService.getById(query.getId());
        TTrainLibrary byId = tTrainLibraryService.getById(tTrainLibraryWarehousing.getTrainLibraryId());
        if (tTrainLibraryWarehousing.getType()==1){
            byId.setStatus(1);

        }else{
            byId.setStatus(3);

        }
        tTrainLibraryService.updateById(byId);
        tTrainLibraryWarehousing.setPreserveSignature(query.getPreserveSignature());
        tTrainLibraryWarehousing.setPreserveId(tokenService.getLoginUser().getUserId());
        tTrainLibraryWarehousing.setConfirmTime(LocalDateTime.now());
        tTrainLibraryWarehousingService.updateById(tTrainLibraryWarehousing);
        return R.ok();
    }

    /**
     * 编辑获取菌种库详情
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:detail')")
    @ApiOperation(value = "编辑获取菌种库详情",tags = "菌种库管理",response = TTrainLibraryDetailVO.class)
    @GetMapping(value = "/open/t-train-library/getDetailEditById")
    public R<TTrainLibraryDetailVO> getDetailEditById(@RequestParam String id) {
        TTrainLibrary byId = tTrainLibraryService.getById(id);
        TTrainLibraryDetailVO tTrainLibraryDetailVO = new TTrainLibraryDetailVO();
        BeanUtils.copyProperties(byId,tTrainLibraryDetailVO);
        return R.ok(tTrainLibraryDetailVO);
    }
    /**
     * 查看菌种库详情
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:detail')")
    @ApiOperation(value = "查看菌种库详情",tags = "菌种库管理",response = TTrainLibraryDetailQuery.class)
    @PostMapping(value = "/open/t-train-library/getDetailById")
    public R<TTrainLibraryDetailVO> getDetailById(@RequestBody String param) {
        TTrainLibraryDetailQuery query = JSON.parseObject(param, TTrainLibraryDetailQuery.class);
        TTrainLibraryDetailVO detailById = tTrainLibraryService.getDetailById(query);
        List<TTrainLibraryWarehousing> list = tTrainLibraryWarehousingService
                .lambdaQuery().eq(TTrainLibraryWarehousing::getTrainLibraryId,query.getId())
                .orderByDesc(TTrainLibraryWarehousing::getCreateTime).list();
        List<SysUser> sysUsers = sysUserService.selectList();
        for (TTrainLibraryWarehousing tTrainLibraryWarehousing : list) {
            if (tTrainLibraryWarehousing.getHandleId()!=null){
                SysUser sysUser = sysUsers.stream().filter(e -> e.getUserId().equals(tTrainLibraryWarehousing.getHandleId())).findFirst().orElse(null);
                if (sysUser!=null){
                    tTrainLibraryWarehousing.setHandleName(sysUser.getNickName());
                }
            }
            if (tTrainLibraryWarehousing.getPreserveId()!=null){
                SysUser sysUser = sysUsers.stream().filter(e -> e.getUserId().equals(tTrainLibraryWarehousing.getPreserveId())).findFirst().orElse(null);
                if (sysUser!=null){
                    tTrainLibraryWarehousing.setPreserveName(sysUser.getNickName());
                }
            }
        }
        detailById.setList(list);
        return R.ok(detailById);
    }
    /**
     * 删除项目组
     */
    //@PreAuthorize("@ss.hasPermi('system:projectTeam:delete')")
    @Log(title = "菌种库管理-删除菌种库", businessType = BusinessType.DELETE)
    @ApiOperation(value = "删除菌种库",tags = "菌种库管理")
    @DeleteMapping(value = "/open/t-train-library/deleteById")
    public R<Boolean> deleteById(@RequestParam String id) {
        return R.ok(tTrainLibraryService.remove(Wrappers.lambdaQuery(TTrainLibrary.class).eq(TTrainLibrary::getId, id)));
    }

    /**
     * 批量删除项目组
     */
    //@PreAuthorize("@ss.hasPermi('system:projectTeam:delete')")
    @Log(title = "菌种库管理-批量删除菌种库", businessType = BusinessType.DELETE)
    @ApiOperation(value = "批量删除菌种库",tags = "菌种库管理")
    @DeleteMapping(value = "/api/t-train-library/deleteByIds")
    public R<Boolean> deleteByIds(@RequestBody List<String> ids) {
        return R.ok(tTrainLibraryService.remove(Wrappers.lambdaQuery(TTrainLibrary.class).in(TTrainLibrary::getId, ids)));
    }
    /**
     * 新增菌种库出入记录
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:addWarehousing')")
    @ApiOperation(value = "新增菌种库出入记录",tags = "菌种库管理",response = AddTrainLibraryWarehousingDTO.class)
    @Log(title = "菌种库管理-新增菌种库出入记录", businessType = BusinessType.INSERT)
    @PostMapping(value = "/open/t-train-library/addWarehousing")
    public R addWarehousing(@RequestBody String param) {
        if (tokenService.getLoginUser()==null){
            return R.fail(401,"登陆失效");
        }
        AddTrainLibraryWarehousingDTO query = JSON.parseObject(param, AddTrainLibraryWarehousingDTO.class);
        TTrainLibrary tTrainLibrary = tTrainLibraryService.getById(query.getTrainLibraryId());
        if (query.getType()==1){
            if (tTrainLibrary.getStatus()==1||tTrainLibrary.getStatus()==2){
                return R.fail("当前菌种为出库状态,不可重复出库");
            }
        }
        if (query.getType()==2){
            if (tTrainLibrary.getStatus()==3||tTrainLibrary.getStatus()==4){
                return R.fail("当前菌种为入库状态,不可重复入库");
            }
        }
        query.setHandleId(tokenService.getLoginUser().getUserId());
        query.setBoundTime(LocalDateTime.now());
        tTrainLibraryWarehousingService.save(query);
        return R.ok();
    }
    /**
     * 编辑出入库记录
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:addWarehousing')")
    @ApiOperation(value = "编辑出入库记录",tags = "菌种库管理",response = AddTrainLibraryWarehousingDTO.class)
    @PostMapping(value = "/api/t-train-library/editWarehousing")
    public R editWarehousing(@RequestBody String param) {
        AddTrainLibraryWarehousingDTO query = JSON.parseObject(param, AddTrainLibraryWarehousingDTO.class);
        tTrainLibraryWarehousingService.updateById(query);
        return R.ok();
    }

    /**
     * 编辑出入库记录
     */
    //@PreAuthorize("@ss.hasPermi('system:trainLibrary:addWarehousing')")
    @ApiOperation(value = "编辑出入库记录-获取详情",tags = "菌种库管理",response = AddTrainLibraryWarehousingDTO.class)
    @PostMapping(value = "/open/t-train-library/editWarehousingDetail")
    public R editWarehousingDetail(@RequestParam String id) {
        return R.ok(tTrainLibraryWarehousingService.getById(id));
    }
    @Log(title = "菌种库管理-删除菌种库出入库记录", businessType = BusinessType.DELETE)
    @ApiOperation(value = "删除菌种库出入库记录",tags = "菌种库管理")
    @DeleteMapping(value = "/open/t-train-library/deleteWarehousingById")
    public R<Boolean> deleteWarehousingById(@RequestParam String id) {
        return R.ok(tTrainLibraryWarehousingService.remove(Wrappers.lambdaQuery(TTrainLibraryWarehousing.class).eq(TTrainLibraryWarehousing::getId, id)));
    }
}