package org.dromara.demo.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import lombok.RequiredArgsConstructor;
import org.dromara.common.core.utils.MapstructUtils;
import org.dromara.common.core.utils.StringUtils;
import org.dromara.common.mybatis.core.page.PageQuery;
import org.dromara.common.mybatis.core.page.TableDataInfo;
import org.dromara.demo.domain.RsSceneryInfo;
import org.dromara.demo.domain.RsSceneryOperationData;
import org.dromara.demo.domain.bo.RsSceneryOperationDataBo;
import org.dromara.demo.domain.vo.RsSceneryOperationDataVo;
import org.dromara.demo.mapper.RsSceneryInfoMapper;
import org.dromara.demo.mapper.RsSceneryOperationDataMapper;
import org.dromara.demo.service.IRsSceneryOperationDataService;
import org.springframework.stereotype.Service;

import java.util.*;
import java.util.function.Function;
import java.util.stream.Collectors;

/**
 * 景区运行数据Service业务层处理
 *
 * @author gonghl
 * @date 2024-03-05
 */
@RequiredArgsConstructor
@Service
public class RsSceneryOperationDataServiceImpl implements IRsSceneryOperationDataService {

    private final RsSceneryOperationDataMapper baseMapper;
    private final RsSceneryInfoMapper sceneryInfoMapper;

    /**
     * 查询景区运行数据
     */
    @Override
    public RsSceneryOperationDataVo queryById(String id){
        return baseMapper.selectVoById(id);
    }

    /**
     * 查询景区运行数据列表
     */
    @Override
    public TableDataInfo<RsSceneryOperationDataVo> queryPageList(RsSceneryOperationDataBo bo, PageQuery pageQuery) {
        LambdaQueryWrapper<RsSceneryOperationData> lqw = buildQueryWrapper(bo);
        Page<RsSceneryOperationDataVo> result = baseMapper.selectVoPage(pageQuery.build(), lqw);
        return TableDataInfo.build(result);
    }

    /**
     * 查询景区运行数据列表
     */
    @Override
    public List<RsSceneryOperationDataVo> queryList(RsSceneryOperationDataBo bo) {
        LambdaQueryWrapper<RsSceneryOperationData> lqw = buildQueryWrapper(bo);
        return baseMapper.selectVoList(lqw);
    }

    private LambdaQueryWrapper<RsSceneryOperationData> buildQueryWrapper(RsSceneryOperationDataBo bo) {
        Map<String, Object> params = bo.getParams();
        LambdaQueryWrapper<RsSceneryOperationData> lqw = Wrappers.lambdaQuery();
        lqw.like(StringUtils.isNotBlank(bo.getSceneryInfoName()), RsSceneryOperationData::getSceneryInfoName, bo.getSceneryInfoName());
        lqw.eq(StringUtils.isNotBlank(bo.getPeriodDate()), RsSceneryOperationData::getPeriodDate, bo.getPeriodDate());
        lqw.eq(bo.getStatus() != null, RsSceneryOperationData::getStatus, bo.getStatus());
        lqw.orderByDesc(RsSceneryOperationData::getCreateTime);
        return lqw;
    }

    /**
     * 新增景区运行数据
     */
    @Override
    public Boolean insertByBo(RsSceneryOperationDataBo bo) {
        RsSceneryOperationData add = MapstructUtils.convert(bo, RsSceneryOperationData.class);
        validEntityBeforeSave(add);
        boolean flag = baseMapper.insert(add) > 0;
        if (flag) {
            bo.setId(add.getId());
        }
        // 启用新数据将其余同类型数据改为禁用
        if (bo.getStatus().equals(1L)) {
            LambdaUpdateWrapper<RsSceneryOperationData> updateWrapper = new LambdaUpdateWrapper<>();
            updateWrapper.ne(RsSceneryOperationData::getId, bo.getId()).eq(RsSceneryOperationData::getSceneryInfoId, bo.getSceneryInfoId()).set(RsSceneryOperationData::getStatus, 2);
            baseMapper.update(updateWrapper);
        }
        return flag;
    }

    /**
     * 修改景区运行数据
     */
    @Override
    public Boolean updateByBo(RsSceneryOperationDataBo bo) {
        RsSceneryOperationData update = MapstructUtils.convert(bo, RsSceneryOperationData.class);
        // 启用新数据将其余同类型数据改为禁用
        if (bo.getStatus().equals(1L)) {
            LambdaUpdateWrapper<RsSceneryOperationData> updateWrapper = new LambdaUpdateWrapper<>();
            updateWrapper.ne(RsSceneryOperationData::getId, bo.getId()).eq(RsSceneryOperationData::getSceneryInfoId, bo.getSceneryInfoId()).set(RsSceneryOperationData::getStatus, 2);
            baseMapper.update(updateWrapper);
        }
        return baseMapper.updateById(update) > 0;
    }

    /**
     * 保存前的数据校验
     */
    private void validEntityBeforeSave(RsSceneryOperationData entity){
        entity.setCreateTime(new Date());
    }

    /**
     * 批量删除景区运行数据
     */
    @Override
    public Boolean deleteWithValidByIds(Collection<String> ids, Boolean isValid) {
        if(isValid){
            //TODO 做一些业务上的校验,判断是否需要校验
        }
        return baseMapper.deleteBatchIds(ids) > 0;
    }

    @Override
    public String importExcel(List<RsSceneryOperationData> list) {
        Set<String> sceneryInfoNames = new HashSet<>();
        // 查询数据库是否存在导入的景区
        Map<String, RsSceneryInfo> sceneryInfoMap = sceneryInfoMapper.selectList().stream()
            .collect(Collectors.toMap(RsSceneryInfo::getSceneryName, Function.identity()));
        for (RsSceneryOperationData operationData : list) {
            String sceneryName = operationData.getSceneryInfoName();
            RsSceneryInfo sceneryInfo = sceneryInfoMap.get(sceneryName);
            // 景区已存在，正常导入，默认禁用
            if (sceneryInfo != null) {
                operationData.setSceneryInfoId(sceneryInfo.getId());
                operationData.setPeriod(operationData.getPeriodDate().length() > 4 ? 1L : 2L);
                operationData.setCreateTime(new Date());
                operationData.setStatus(2L);
            } else {
                sceneryInfoNames.add(operationData.getSceneryInfoName());
            }
        }
        if (!sceneryInfoNames.isEmpty()) {
            return "很抱歉，导入失败！" + String.join("、", "<" + sceneryInfoNames + ">") + "景区不存在，请前往【景区配置】添加后重试！";
        }
        baseMapper.insertBatch(list);
        return "恭喜您，数据已全部导入成功！共"  + list.size() + "条！";
    }
}