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.R;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.enums.BusinessType;
import com.ruoyi.framework.web.service.TokenService;
import com.ruoyi.system.dto.AuditStatusDTO;
import com.ruoyi.system.dto.ExperimentDispatchSignDTO;
import com.ruoyi.system.dto.TExperimentDispatchDTO;
import com.ruoyi.system.dto.UpAndDownDTO;
import com.ruoyi.system.mapper.SysUserMapper;
import com.ruoyi.system.model.TExperimentDispatch;
import com.ruoyi.system.model.TExperimentDispatchGroup;
import com.ruoyi.system.model.TExperimentDispatchParticipants;
import com.ruoyi.system.model.TExperimentDispatchTask;
import com.ruoyi.system.query.TExperimentDispatchQuery;
import com.ruoyi.system.service.*;
import com.ruoyi.system.vo.TExperimentDispatchVO;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.security.access.prepost.PreAuthorize;
import org.springframework.util.CollectionUtils;
import org.springframework.web.bind.annotation.*;

import java.time.LocalDateTime;
import java.util.List;
import java.util.Objects;
import java.util.stream.Collectors;

/**
 * <p>
 * 实验调度管理 前端控制器
 * </p>
 *
 * @author xiaochen
 * @since 2025-04-08
 */
@Api(tags = "实验调度管理")
@RestController
@RequestMapping("/t-experiment-dispatch")
public class TExperimentDispatchController {

    private final TExperimentDispatchService experimentDispatchService;
    private final TokenService tokenService;
    private final ISysUserService sysUserService;
    private final SysUserMapper sysUserMapper;
    private final TExperimentDispatchGroupService experimentDispatchGroupService;
    private final TExperimentDispatchParticipantsService experimentDispatchParticipantsService;
    private final TExperimentDispatchTaskService experimentDispatchTaskService;
    @Autowired
    public TExperimentDispatchController(TExperimentDispatchService experimentDispatchService, TokenService tokenService, ISysUserService sysUserService, SysUserMapper sysUserMapper, TExperimentDispatchGroupService experimentDispatchGroupService, TExperimentDispatchParticipantsService experimentDispatchParticipantsService, TExperimentDispatchTaskService experimentDispatchTaskService) {
        this.experimentDispatchService = experimentDispatchService;
        this.tokenService = tokenService;
        this.sysUserService = sysUserService;
        this.sysUserMapper = sysUserMapper;
        this.experimentDispatchGroupService = experimentDispatchGroupService;
        this.experimentDispatchParticipantsService = experimentDispatchParticipantsService;
        this.experimentDispatchTaskService = experimentDispatchTaskService;
    }

    /**
     * 获取实验调度管理列表
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:list')")
    @ApiOperation(value = "获取实验调度分页列表")
    @PostMapping(value = "/api/t-experiment-dispatch/pageList")
    public R<PageInfo<TExperimentDispatchVO>> pageList(@RequestBody String param) {
        TExperimentDispatchQuery query = JSON.parseObject(param, TExperimentDispatchQuery.class);
        return R.ok(experimentDispatchService.pageList(query));
    }

    /**
     * 添加实验调度管理
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:add')")
    @Log(title = "实验调度信息-新增实验调度", businessType = BusinessType.INSERT)
    @ApiOperation(value = "添加实验调度",response = TExperimentDispatchDTO.class)
    @PostMapping(value = "/api/t-experiment-dispatch/add")
    public R<Boolean> add(@RequestBody String param) {
        TExperimentDispatchDTO dto = JSON.parseObject(param,TExperimentDispatchDTO.class);
        // TODO 生成实验调度编号
        experimentDispatchService.save(dto);
        // 添加实验分组
        List<TExperimentDispatchGroup> experimentDispatchGroups = dto.getExperimentDispatchGroups();
        for (TExperimentDispatchGroup experimentDispatchGroup : experimentDispatchGroups) {
            experimentDispatchGroup.setDispatchId(dto.getId());
        }
        experimentDispatchGroupService.saveBatch(experimentDispatchGroups);
        // 添加参与人员
        List<TExperimentDispatchParticipants> experimentDispatchParticipants = dto.getExperimentDispatchParticipants();
        experimentDispatchParticipants.forEach(experimentDispatchParticipant -> {
            experimentDispatchParticipant.setDispatchId(dto.getId());
        });
        experimentDispatchParticipantsService.saveBatch(experimentDispatchParticipants);
        // 添加实验任务
        List<TExperimentDispatchTask> experimentDispatchTasks = dto.getExperimentDispatchTasks();
        experimentDispatchTasks.forEach(experimentDispatchTask -> {
            experimentDispatchTask.setDispatchId(dto.getId());
        });
        experimentDispatchTaskService.saveBatch(experimentDispatchTasks);
        return R.ok();
    }

    /**
     * 修改实验调度
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:edit')")
    @Log(title = "实验调度信息-修改实验调度", businessType = BusinessType.UPDATE)
    @ApiOperation(value = "修改实验调度")
    @PostMapping(value = "/api/t-experiment-dispatch/update")
    public R<Boolean> update(@RequestBody String param) {
        TExperimentDispatchDTO dto = JSON.parseObject(param,TExperimentDispatchDTO.class);
        experimentDispatchService.updateById(dto);
        // 添加实验分组
        experimentDispatchGroupService.remove(Wrappers.lambdaQuery(TExperimentDispatchGroup.class).eq(TExperimentDispatchGroup::getDispatchId, dto.getId()));
        List<TExperimentDispatchGroup> experimentDispatchGroups = dto.getExperimentDispatchGroups();
        for (TExperimentDispatchGroup experimentDispatchGroup : experimentDispatchGroups) {
            experimentDispatchGroup.setDispatchId(dto.getId());
        }
        experimentDispatchGroupService.saveBatch(experimentDispatchGroups);
        // 添加参与人员
        experimentDispatchParticipantsService.remove(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class).eq(TExperimentDispatchParticipants::getDispatchId, dto.getId()));
        List<TExperimentDispatchParticipants> experimentDispatchParticipants = dto.getExperimentDispatchParticipants();
        experimentDispatchParticipants.forEach(experimentDispatchParticipant -> {
            experimentDispatchParticipant.setDispatchId(dto.getId());
        });
        experimentDispatchParticipantsService.saveBatch(experimentDispatchParticipants);
        // 添加实验任务
        experimentDispatchTaskService.remove(Wrappers.lambdaQuery(TExperimentDispatchTask.class).eq(TExperimentDispatchTask::getDispatchId, dto.getId()));
        List<TExperimentDispatchTask> experimentDispatchTasks = dto.getExperimentDispatchTasks();
        experimentDispatchTasks.forEach(experimentDispatchTask -> {
            experimentDispatchTask.setDispatchId(dto.getId());
        });
        experimentDispatchTaskService.saveBatch(experimentDispatchTasks);
        return R.ok();
    }

    /**
     * 查看实验调度详情
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:detail')")
    @ApiOperation(value = "查看实验调度详情")
    @GetMapping(value = "/open/t-experiment-dispatch/getDetailById")
    public R<TExperimentDispatchVO> getDetailById(@RequestParam String id) {
        TExperimentDispatch experimentDispatch = experimentDispatchService.getById(id);
        TExperimentDispatchVO experimentDispatchVO = new TExperimentDispatchVO();
        BeanUtils.copyProperties(experimentDispatch, experimentDispatchVO);

        // 查询组别
        experimentDispatchVO.setExperimentDispatchGroups(experimentDispatchGroupService.list(Wrappers.lambdaQuery(TExperimentDispatchGroup.class).eq(TExperimentDispatchGroup::getDispatchId, id)));
        // 查询参与人员
        List<TExperimentDispatchParticipants> list = experimentDispatchParticipantsService.list(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class).eq(TExperimentDispatchParticipants::getDispatchId, id));
        List<Long> userIds = list.stream().map(TExperimentDispatchParticipants::getUserId).collect(Collectors.toList());
        if(!CollectionUtils.isEmpty(userIds)){
            List<SysUser> sysUsers = sysUserMapper.selectUserByIds(userIds);
            list.forEach(tExperimentDispatchParticipant -> {
                SysUser sysUser = sysUsers.stream().filter(user -> user.getUserId().equals(tExperimentDispatchParticipant.getUserId())).findFirst().orElse(null);
                if(sysUser != null){
                    tExperimentDispatchParticipant.setNickName(sysUser.getNickName());
                    tExperimentDispatchParticipant.setAvatar(sysUser.getAvatar());
                }
            });
        }
        experimentDispatchVO.setExperimentDispatchParticipants(list);
        // 查询实验任务
        experimentDispatchVO.setExperimentDispatchTasks(experimentDispatchTaskService.list(Wrappers.lambdaQuery(TExperimentDispatchTask.class).eq(TExperimentDispatchTask::getDispatchId, id)));

        return R.ok(experimentDispatchVO);
    }

    /**
     * 删除实验调度
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:delete')")
    @Log(title = "实验调度信息-删除实验调度", businessType = BusinessType.DELETE)
    @ApiOperation(value = "删除实验调度")
    @DeleteMapping(value = "/open/t-experiment-dispatch/deleteById")
    public R<Boolean> deleteById(@RequestParam String id) {
        // 删除组别
        experimentDispatchGroupService.remove(Wrappers.lambdaQuery(TExperimentDispatchGroup.class).eq(TExperimentDispatchGroup::getDispatchId, id));
        // 删除参与人员
        experimentDispatchParticipantsService.remove(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class).eq(TExperimentDispatchParticipants::getDispatchId, id));
        // 删除任务
        experimentDispatchTaskService.remove(Wrappers.lambdaQuery(TExperimentDispatchTask.class).eq(TExperimentDispatchTask::getDispatchId, id));
        return R.ok(experimentDispatchService.removeById(id));
    }

    /**
     * 批量删除实验调度
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:delete')")
    @Log(title = "实验调度信息-删除实验调度", businessType = BusinessType.DELETE)
    @ApiOperation(value = "批量删除实验调度")
    @DeleteMapping(value = "/open/t-experiment-dispatch/deleteByIds")
    public R<Boolean> deleteByIds(@RequestBody List<String> ids) {
        // 删除组别
        experimentDispatchGroupService.remove(Wrappers.lambdaQuery(TExperimentDispatchGroup.class).in(TExperimentDispatchGroup::getDispatchId, ids));
        // 删除参与人员
        experimentDispatchParticipantsService.remove(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class).in(TExperimentDispatchParticipants::getDispatchId, ids));
        // 删除任务
        experimentDispatchTaskService.remove(Wrappers.lambdaQuery(TExperimentDispatchTask.class).in(TExperimentDispatchTask::getDispatchId, ids));
        return R.ok(experimentDispatchService.removeByIds(ids));
    }

    /**
     * 批量删除实验调度
     */
    @PreAuthorize("@ss.hasPermi('system:experimentDispatch:sign')")
    @Log(title = "实验调度信息-实验调度签字", businessType = BusinessType.UPDATE)
    @ApiOperation(value = "实验调度签字")
    @PostMapping(value = "/api/t-experiment-dispatch/sign")
    public R<Boolean> sign(@RequestBody String param) {
        ExperimentDispatchSignDTO experimentDispatchSign = JSON.parseObject(param, ExperimentDispatchSignDTO.class);
        Long userId = tokenService.getLoginUser().getUserId();
        Integer roleType = tokenService.getLoginUser().getUser().getRoleType();
        TExperimentDispatchParticipants experimentDispatchParticipants = experimentDispatchParticipantsService.getOne(Wrappers.lambdaQuery(TExperimentDispatchParticipants.class)
                .eq(TExperimentDispatchParticipants::getDispatchId, experimentDispatchSign.getDispatchId())
                .eq(TExperimentDispatchParticipants::getRoleType, roleType)
                .eq(TExperimentDispatchParticipants::getUserId, userId)
                .last("LIMIT 1"));
        if(experimentDispatchParticipants != null){
            experimentDispatchParticipants.setStatus(2);
            experimentDispatchParticipants.setSignTime(LocalDateTime.now());
            experimentDispatchParticipants.setConfirmSign(experimentDispatchSign.getConfirmSign());
            experimentDispatchParticipantsService.updateById(experimentDispatchParticipants);
        }
        return R.ok();
    }

}