package com.gunshi.project.hsz.service;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gunshi.core.dateformat.DateFormatString;
import com.gunshi.project.hsz.entity.dto.RotaDto;
import com.gunshi.project.hsz.entity.dto.RotaUserDto;
import com.gunshi.project.hsz.entity.vo.RotaVo;
import com.gunshi.project.hsz.mapper.RotaBMapper;
import com.gunshi.project.hsz.model.RotaB;
import com.gunshi.project.hsz.util.DateUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.text.SimpleDateFormat;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.stream.Collectors;

/**
 * Description:
 * Created by wanyan on 2024/3/25
 *
 * @author wanyan
 * @version 1.0
 */
@Service
@Transactional(rollbackFor = Exception.class)
public class RotaService extends ServiceImpl<RotaBMapper, RotaB> {


    public Map<String,List<RotaB>> query(String yearMonth) {
        List<RotaB> list = this.baseMapper.query(yearMonth);
        return list.stream().collect(Collectors.groupingBy(rota -> new SimpleDateFormat(DateFormatString.YYYY_MM_DD).format(rota.getRotaDate())));
    }

    public String editInfo(RotaDto RotaDto) {
        Date rotaDate = RotaDto.getRotaDate();
        //先删除该日期的所有信息
        LambdaQueryWrapper<RotaB> queryWrapper = Wrappers.lambdaQuery();
        queryWrapper.eq(RotaB::getRotaDate,rotaDate);
        this.remove(queryWrapper);
        if(RotaDto.getIsHoliday() != null && RotaDto.getIsHoliday() == 1){
            RotaB rotaB = new RotaB();
            BeanUtils.copyProperties(RotaDto,rotaB,RotaB.class);
            rotaB.setId(IdWorker.getId());
            this.save(rotaB);
            return "设为放假日成功";
        }
        List<RotaUserDto> userDtoList = RotaDto.getUserDtoList();
        List<RotaB> list = userDtoList.stream().map(o -> {
            RotaB rotaB = new RotaB();
            rotaB.setId(IdWorker.getId());
            rotaB.setRotaDate(rotaDate);
            rotaB.setIsHoliday(0);
            rotaB.setRotaType(o.getRotaType());
            rotaB.setUserId(o.getUserId());
            return rotaB;
        }).collect(Collectors.toList());
        this.saveBatch(list);
        return "设置值班信息成功";
    }

    public List<RotaB> dateList(String rotaDate) {
        return this.baseMapper.dateList(rotaDate);
    }

    public Long checkPhone(String value){
        return this.baseMapper.queryUser(value);
    }

    public void saveImportData(ArrayList<RotaVo> successArrayList) {
        List<RotaB> list = new ArrayList<>();
        for(RotaVo vo : successArrayList){
            Date stm = vo.getStm();
            Date etm = vo.getEtm();
            List<Date> dates = DateUtil.getDatesBetween(stm, etm);
            dates.stream().forEach(str->{
                RotaB rotaB = new RotaB();
                rotaB.setId(IdWorker.getId());
                rotaB.setRotaDate(str);
                rotaB.setIsHoliday(0);
                rotaB.setRotaType(1);
                rotaB.setUserId(vo.getLeaderUserId());
                list.add(rotaB);

                RotaB rota = new RotaB();
                rota.setId(IdWorker.getId());
                rota.setRotaDate(str);
                rota.setIsHoliday(0);
                rota.setRotaType(2);
                rota.setUserId(vo.getDutyUserId());
                list.add(rota);
            });
        }
        this.saveOrUpdateBatch(list);
    }
}