package org.springblade.modules.yw.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import org.apache.logging.log4j.util.Strings;
import org.springblade.core.tool.utils.BeanUtil;
import org.springblade.modules.yw.entity.FirmInfo;
import org.springblade.modules.yw.entity.RescueTeamEntity;
import org.springblade.modules.yw.excel.RescueTeamExcel;
import org.springblade.modules.yw.service.IFirmInfoService;
import org.springblade.modules.yw.vo.RescueTeamVO;
import org.springblade.modules.yw.mapper.RescueTeamMapper;
import org.springblade.modules.yw.service.IRescueTeamService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import com.baomidou.mybatisplus.core.metadata.IPage;

import java.util.List;
import java.util.Objects;

/**
 * 救援队伍表 服务实现类
 *
 * @author BladeX
 * @since 2024-10-28
 */
@Service
public class RescueTeamServiceImpl extends ServiceImpl<RescueTeamMapper, RescueTeamEntity> implements IRescueTeamService {

	@Autowired
	private IFirmInfoService firmInfoService;

	/**
	 * 自定义分页查询
	 * @param page
	 * @param rescueTeam
	 * @return
	 */
	@Override
	public IPage<RescueTeamVO> selectRescueTeamPage(IPage<RescueTeamVO> page, RescueTeamVO rescueTeam) {
		return page.setRecords(baseMapper.selectRescueTeamPage(page, rescueTeam));
	}

	/**
	 * 导入救援队伍信息
	 * @param data
	 * @param isCovered
	 * @return
	 */
	@Override
	public String importRescueTeam(List<RescueTeamExcel> data, boolean isCovered) {
		for (RescueTeamExcel rescueTeamExcel : data) {
			// 数据拷贝
			RescueTeamEntity rescueTeamEntity = Objects.requireNonNull(BeanUtil.copy(rescueTeamExcel, RescueTeamEntity.class));
			// 类型转换
			if (!Strings.isBlank(rescueTeamExcel.getType())){
				if (rescueTeamExcel.getType().equals("企业")){
					rescueTeamEntity.setType(1);
					// 设置企业id
					setFirmId(rescueTeamExcel,rescueTeamEntity);
				}
				if (rescueTeamExcel.getType().equals("园区")){
					rescueTeamEntity.setType(2);
				}
			}
			// 判断是否保存
			Long id = isSave(rescueTeamExcel);
			if (null!=id){
				if (isCovered){
					// 覆盖更新
					rescueTeamEntity.setId(id);
					updateById(rescueTeamEntity);
					continue;
				}
			}
			// 保存
			save(rescueTeamEntity);
		}
		return null;
	}

	/**
	 * 判断是否已经保存救援队伍信息
	 * @param rescueTeamExcel
	 * @return
	 */
	private Long isSave(RescueTeamExcel rescueTeamExcel) {
		QueryWrapper<RescueTeamEntity> wrapper = new QueryWrapper<>();
		wrapper.eq("per_in_cha",rescueTeamExcel.getPerInCha())
			.eq("per_in_cha_pho",rescueTeamExcel.getPerInChaPho())
			.eq("is_deleted",0);
		RescueTeamEntity one = getOne(wrapper);
		if (null!=one){
			return one.getId();
		}
		return null;
	}

	/**
	 * 设置企业id
	 * @param rescueTeamExcel
	 * @param rescueTeamEntity
	 */
	private void setFirmId(RescueTeamExcel rescueTeamExcel, RescueTeamEntity rescueTeamEntity) {
		QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
		wrapper.eq("name",rescueTeamExcel.getFirmName())
			.eq("is_deleted",0);
		FirmInfo firmInfo = firmInfoService.getOne(wrapper);
		if (null!=firmInfo){
			rescueTeamEntity.setFirmId(firmInfo.getId());
		}
	}
}