package com.gkhy.exam.noncoalmine.service.impl;

import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.gkhy.exam.noncoalmine.entity.TrainingInstitution;
import com.gkhy.exam.noncoalmine.mapper.TrainingInstitutionMapper;
import com.gkhy.exam.noncoalmine.model.addForm.TrainingInstitutionAddForm;
import com.gkhy.exam.noncoalmine.model.modForm.TrainingInstitutionModForm;
import com.gkhy.exam.noncoalmine.model.query.TrainingInstitutionQuery;
import com.gkhy.exam.noncoalmine.model.vo.TrainingInstitutionVO;
import com.gkhy.exam.noncoalmine.service.TrainingInstitutionService;
import com.ruoyi.common.core.domain.entity.SysUser;
import com.ruoyi.common.utils.SecurityUtils;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.system.domain.SysDistrict;
import com.ruoyi.system.service.SysDistrictService;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;
import java.util.stream.Collectors;

/**
 * (TrainingInstitution)表服务实现类
 *
 * @author makejava
 * @since 2023-09-13 16:12:31
 */
@Service("trainingInstitutionService")
public class TrainingInstitutionServiceImpl extends ServiceImpl<TrainingInstitutionMapper, TrainingInstitution> implements TrainingInstitutionService {
	@Autowired
	private TrainingInstitutionMapper trainingInstitutionMapper;
	@Autowired
	private SysDistrictService districtService;

	@Override
	public List<TrainingInstitutionVO> listByPage(TrainingInstitutionQuery query) {
		SysUser user = SecurityUtils.getLoginUser().getUser();
		if(StringUtils.isEmpty(query.getRegionCode())){
			//传入区划为空
			if(!user.isAdmin(user.getUserId())){
				//非超管
				List<SysDistrict> districtList = districtService
						.selectAll()
						.stream()
						.filter(sysDistrict -> sysDistrict.getId().equals(user.getDistrictId()))
						.collect(Collectors.toList());
				if(districtList.size() > 0){
					SysDistrict sysDistrict = districtList.get(0);
					query.setRegionCode(sysDistrict.getCode());
				}
			}
		}
		List<TrainingInstitutionVO> trainingInstitutionVOS = trainingInstitutionMapper.listByPage(query);
		return trainingInstitutionVOS;
	}

	@Override
	public int mod(TrainingInstitutionModForm modForm) {
		TrainingInstitution trainingInstitution = new TrainingInstitution();
		BeanUtils.copyProperties(modForm,trainingInstitution);
		return baseMapper.updateById(trainingInstitution);
	}

	@Override
	public int add(TrainingInstitutionAddForm addForm) {
		TrainingInstitution trainingInstitution = new TrainingInstitution();
		BeanUtils.copyProperties(addForm,trainingInstitution);
		trainingInstitution.setDelFlag((byte)0);
		return baseMapper.insert(trainingInstitution);
	}
}