package com.chinaztt.mes.basic.service.impl;

import cn.hutool.core.collection.CollectionUtil;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.baomidou.mybatisplus.extension.toolkit.SqlHelper;
import com.chinaztt.mes.basic.entity.Company;
import com.chinaztt.mes.basic.entity.Division;
import com.chinaztt.mes.basic.entity.Factory;
import com.chinaztt.mes.basic.entity.Post;
import com.chinaztt.mes.basic.excel.PostData;
import com.chinaztt.mes.basic.mapper.CompanyMapper;
import com.chinaztt.mes.basic.mapper.DivisionMapper;
import com.chinaztt.mes.basic.mapper.FactoryMapper;
import com.chinaztt.mes.basic.mapper.PostMapper;
import com.chinaztt.mes.basic.service.PostService;
import com.chinaztt.mes.common.numgen.NumberGenerator;
import lombok.AllArgsConstructor;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

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

/**
 * 岗位表
 *
 * @author cxf
 * @date 2021-06-23 17:17:26
 */
@Service
@AllArgsConstructor
@Transactional(rollbackFor = Exception.class)
public class PostServiceImpl extends ServiceImpl<PostMapper, Post> implements PostService {
	private NumberGenerator<Post> postNumberGenerator;
	private DivisionMapper divisionMapper;
	private PostMapper postMapper;
	private CompanyMapper companyMapper;
	private FactoryMapper factoryMapper;

	@Override
	public boolean save(Post post) {
		checkPostNo(post);
		return SqlHelper.retBool(baseMapper.insert(post));
	}

	@Override
	public boolean updateById(Post post) {
		checkPostNo(post);
		return SqlHelper.retBool(baseMapper.updateById(post));
	}

	private void checkPostNo(Post post) {
		if (StringUtils.isBlank(post.getPostNo())) {
			// 1. 自动生成编号
			post.setPostNo(postNumberGenerator.generateNumberWithPrefix(Post.DIGIT, Post.PREFIX, Post::getPostNo));
		} else {
			// 2.判断是否重复
			List<Post> repeatNos = baseMapper.selectList(Wrappers.<Post>query().lambda().eq(Post::getPostNo, post.getPostNo()).ne(Post::getId, post.getId()));
			if (CollectionUtil.isNotEmpty(repeatNos)) {
				throw new RuntimeException("编号重复");
			}
		}
	}

	//excel导入数据
	@Override
	public void importWorkstationExcel(List<PostData> list) {
		if (CollectionUtil.isEmpty(list)) {
			return;
		}
		List<Company> companies = companyMapper.selectList(null);
		List<Factory> factories = factoryMapper.selectList(null);
		List<Division> divisions = divisionMapper.selectList(null);
		for (PostData data : list) {
			QueryWrapper<Post>QueryWrapper = new QueryWrapper<>();
			QueryWrapper.lambda().eq(Post::getPostNo,data.getPostNo());
			Post post1 = baseMapper.selectOne(QueryWrapper);
			if(!Objects.isNull(post1)){
				log.error("岗位编号重复=====》"+data.getPostNo());
				continue;
			}
			Post post = new Post();
			BeanUtils.copyProperties(data, post);
			//TODO: data.getDivisionName()+"-"+   岗位组合 部门+岗位
			post.setPostName(data.getPostName());
			for (int i = 0; i <companies.size() ; i++) {
				if (companies.get(i).getCompanyName().equals(data.getCompanyName())){
					post.setCompanyId(companies.get(i).getId());
				}
			}
			if (post.getCompanyId()==null){
				log.error("公司:" + data.getCompanyName() + "不存在");
				continue ;
			}
			for (int i = 0; i <factories.size() ; i++) {
				if (factories.get(i).getFactoryName().equals(data.getFactoryName())){
					post.setFactoryId(factories.get(i).getId());
				}
			}
			if (post.getFactoryId()==null){
				log.error("工厂:" + data.getFactoryName() + "不存在");
				continue ;
			}
			for (int i = 0; i < divisions.size(); i++) {
				if (divisions.get(i).getDivisionName().equals(data.getDivisionName())) {
					post.setDivisionId(divisions.get(i).getId());
				}
			}
			if (post.getDivisionId() == null) {
				log.error("部门:" + data.getDivisionName() + "不存在");
				continue;
			}
			baseMapper.insert(post);
		}
	}
}