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.FacilityEntity;
import org.springblade.modules.yw.entity.FirmInfo;
import org.springblade.modules.yw.excel.FacilityExcel;
import org.springblade.modules.yw.service.IFirmInfoService;
import org.springblade.modules.yw.vo.FacilityVO;
import org.springblade.modules.yw.mapper.FacilityMapper;
import org.springblade.modules.yw.service.IFacilityService;
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-11-13
 */
@Service
public class FacilityServiceImpl extends ServiceImpl<FacilityMapper, FacilityEntity> implements IFacilityService {

	@Autowired
	private IFirmInfoService firmInfoService;

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

	/**
	 * 导入设施信息
	 * @param data
	 * @param isCovered
	 * @return
	 */
	@Override
	public String importFacility(List<FacilityExcel> data, boolean isCovered) {
		for (FacilityExcel facilityExcel : data) {
			// 拷贝数据
			FacilityEntity facilityEntity = Objects.requireNonNull(BeanUtil.copy(facilityExcel, FacilityEntity.class));
			// 设置防线类型
			if(!Strings.isBlank(facilityExcel.getFacLevel())){
				facilityEntity.setFacLevel(Integer.parseInt(facilityExcel.getFacLevel()));
			}
			// 设置企业id
			setFirm(facilityExcel,facilityEntity);
			// 判断是否已保存
			Long id = isSave(facilityEntity);
			// 覆盖更新
			if (null!=id){
				facilityEntity.setId(id);
				updateById(facilityEntity);
				continue;
			}
			// 保存
			save(facilityEntity);
		}
		return null;
	}


	/**
	 * 判断是否保存入库
	 * @param facilityEntity
	 * @return
	 */
	private Long isSave(FacilityEntity facilityEntity) {
		QueryWrapper<FacilityEntity> wrapper = new QueryWrapper<>();
		wrapper.eq("name",facilityEntity.getName())
			.eq("firm_id",facilityEntity.getFirmId())
			.eq("is_deleted",0);
		FacilityEntity one = getOne(wrapper);
		if (null!=one){
			return one.getId();
		}
		return null;
	}

	/**
	 * 设置企业信息
	 * @param firmExcel
	 * @param facilityEntity
	 */
	private void setFirm(FacilityExcel firmExcel,FacilityEntity facilityEntity) {
		QueryWrapper<FirmInfo> wrapper = new QueryWrapper<>();
		wrapper.eq("name",firmExcel.getFirmName())
			.eq("is_deleted",0);
		FirmInfo firmInfo = firmInfoService.getOne(wrapper);
		if (null!=firmInfo){
			facilityEntity.setFirmId(firmInfo.getId());
		}
	}

	/**
	 * 三道防线信息查询
	 * @param facility
	 * @return
	 */
	@Override
	public List<FacilityVO> getFacilityList(FacilityVO facility) {
		// 设置默认查询一道防线
		if (null==facility.getFacLevel()){
			facility.setFacLevel(1);
		}
		// 三道防线信息查询
		if (facility.getFacLevel()==1 ){
			// 查询并返回
			return baseMapper.getFacilityList(facility);
		}
		// 二道防线查询排口，阀门
		if (facility.getFacLevel()==2){
			// 查询并返回
			return baseMapper.getTreeFacilityList(facility);
		}
		// 第三道防线查询应急空间表的应急池和雨水池信息
		if (facility.getFacLevel()==3){
			return baseMapper.getTreeFacilityList(facility);
		}
		// 返回空
		return null;
	}
}