package org.springblade.mdm.flow.excution.cure;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.oss.OssTemplate;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.tool.api.R;
import org.springblade.core.tool.utils.DateUtil;
import org.springblade.mdm.basesetting.machine.service.MachineService;
import org.springblade.mdm.basesetting.machine.entity.Machine;
import org.springblade.mdm.commons.contants.ParamConstants;
import org.springblade.mdm.flow.entity.ApproveRecord;
import org.springblade.mdm.flow.entity.FlowProgramFile;
import org.springblade.mdm.flow.excution.dispatch.FinishDataHandler;
import org.springblade.mdm.flow.service.ApproveRecordService;
import org.springblade.mdm.flow.service.FlowProgramFileService;
import org.springblade.mdm.flow.service.FlowProgramProperties;
import org.springblade.mdm.program.entity.NcNode;
import org.springblade.mdm.program.entity.NcProgramApproved;
import org.springblade.mdm.program.service.NcNodeAutoCreateService;
import org.springblade.mdm.program.service.NcNodeService;
import org.springblade.mdm.program.service.NcProgramApprovedService;
import org.springblade.mdm.program.service.ProgramAnnotationService;
import org.springblade.mdm.utils.EntityUtil;
import org.springblade.system.feign.ISysClient;
import org.springblade.system.pojo.entity.DictBiz;
import org.springframework.beans.BeanUtils;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.time.LocalDate;
import java.util.Arrays;
import java.util.Date;
import java.util.List;
import java.util.Optional;

/**
 * 普通派工完成数据处理器，功能：插入审批表数据，DispatchFinishListener调用
 */
@Slf4j
@Component("normalCureFinishDataHandler")
public class NormalCureFinishDataHandler implements FinishDataHandler {
	@Autowired
	private NcProgramApprovedService approvedService;
	@Autowired
	private ApproveRecordService approveRecordService;
	@Autowired
	private FlowProgramFileService flowProgramFileService;
	@Autowired
	private NcNodeService nodeService;
	@Autowired
	private NcNodeAutoCreateService ncNodeAutoCreateService;
	@Autowired
	private MachineService machineService;
	@Autowired
	private OssTemplate ossTemplate;
	@Autowired
	private ISysClient sysClient;
	@Autowired
	private ProgramAnnotationService programAnnotationService;
	/**
	 * 默认有效期间（月数）,2年
	 */
	private static final int DEFAULT_VALID_MONTH = 24;

	/**
	 * DispatchFinishListener中调用
	 * @param flowProps 流程属性
	 */
	public void handleData(FlowProgramProperties flowProps) {

		//需要固化的节点
		NcNode pkgNode = nodeService.lambdaQuery().eq(NcNode::getProcessInstanceId,  flowProps.getProcessInstanceId()).one();

		List<NcNode> historyProgramPackageNodes = nodeService.getNodeHistory(pkgNode);
		try {
			moveNodeToCuredTree(pkgNode, historyProgramPackageNodes, flowProps);
		}catch(Exception e) {
			log.error("固化流程完成异常",e);
			throw new ServiceException("普通固化执行异常"+e.getMessage());
		}

	}


	/**
	 * 将节点和历史节点挪动到固化的同级节点下
	 * @param pkgNode 要固化的节点
	 * @param historyProgramPackageNodes 程序包名 历史节点
	 */
	void moveNodeToCuredTree(NcNode pkgNode1,List<NcNode> historyProgramPackageNodes, FlowProgramProperties programProperties) throws IOException {

		//创建节点到机床级别.(固化树)
		NcNode machineNode = ncNodeAutoCreateService.createNodeTreeToMachine(programProperties);

		pkgNode1.setParentId(machineNode.getParentId());
		pkgNode1.setParentIds(machineNode.getParentIds());//更新上级节点，下面还要用应为在historyProgramPackageNodes中的和这个不是一个实例

		//新建一个节点，作为固化节点，老节点保留不动,置为老版本

		//机床下现在没有文件了，程序包名升级与机床同级了，改为找到机床同级程序包名更新了
		this.nodeService.lambdaUpdate().likeRight(NcNode::getParentIds, machineNode.getParentIds())
			.in(NcNode::getNodeType, Arrays.asList(NcNode.TYPE_PROGRAM_PACKAGE,NcNode.TYPE_PROGRAM_FILE))
			.set(NcNode::getIsLastEdition,0).set(NcNode::getIsLocked,1).update();

		List<NcNode> programNodes = nodeService.getProgramFilesByPackageId(pkgNode1.getId());
		boolean tryPkdNodeHasChild = !programNodes.isEmpty();
		NcNode cureNode = createNewCureNodeByTargetNode(pkgNode1,tryPkdNodeHasChild);

		for(NcNode hisPackageNode : historyProgramPackageNodes){
			hisPackageNode.setIsLastEdition(0);
			hisPackageNode.setParentId(machineNode.getParentId());
			hisPackageNode.setParentIds(machineNode.getParentIds());//程序包与机床节点同级

			//处理程序包下层的程序节点，这里：老的程序节点不用更新为历史版本，因为会建立新节点。只修改parentId 和 parentIds
			List<NcNode> oriProgramNodes = nodeService.lambdaQuery().eq(NcNode::getParentId,hisPackageNode.getId()).list();
			FlowProgramFile pfile;
			for(NcNode oldProgramNode : oriProgramNodes){
				oldProgramNode.setParentIds(hisPackageNode.getParentIds()+","+hisPackageNode.getId());

				pfile = flowProgramFileService.getById(oldProgramNode.getFlowProgramFileId());
				if(oldProgramNode.getParentId().equals(pkgNode1.getId()) && !pfile.isProgram()){

					//当前固化程序包名下的“其他文件”非程序文件。不设置为老版本，直接挪到固化树下,两个网络之间只交换程序文件，其他文件只能保留
					//其他文件，需要复制到新建的固化节点下
					NcNode newOtherFileNode = new NcNode();
					BeanUtils.copyProperties(oldProgramNode,newOtherFileNode);
					EntityUtil.clearBaseProperties(newOtherFileNode);
					newOtherFileNode.setParentId(cureNode.getId());
					newOtherFileNode.setParentIds(cureNode.getParentIds()+","+cureNode.getId());
					newOtherFileNode.setIsLastEdition(1);
					nodeService.save(newOtherFileNode);
				}

			}
			nodeService.updateBatchById(oriProgramNodes);
		}
		nodeService.updateBatchById(historyProgramPackageNodes);

		Machine machine = machineService.getByCode(cureNode.getMachineCode());
		List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();
		//新的流程文件，需要在包节点下新建，不能用老的
		List<FlowProgramFile> files = flowProgramFileService.lambdaQuery().eq(FlowProgramFile::getProcessInstanceId,pkgNode1.getProcessInstanceId()).list();
		for(FlowProgramFile flowProgramFile : files){
			NcNode newProgNode = new NcNode();

			BeanUtils.copyProperties(cureNode,newProgNode);
			EntityUtil.clearBaseProperties(newProgNode);

			newProgNode.setName(flowProgramFile.getName());
			newProgNode.setNodeType(NcNode.TYPE_PROGRAM_FILE);
			newProgNode.setParentId(cureNode.getId());
			newProgNode.setParentIds(cureNode.getParentIds()+","+cureNode.getId());
			newProgNode.setIsLastEdition(1);
			newProgNode.setFlowProgramFileId(flowProgramFile.getId());
			newProgNode.setIsCured(1);

			setGhAnnotation(flowProgramFile,machine.getMachineGroupCode(),annoDicts);

			nodeService.save(newProgNode);
		}


	}

	/**
	 * 复制待固化节点数据，创建一个新的固化节点
	 * @param waitingCureNode 待固化节点
	 * @return 固化节点
	 */
	NcNode createNewCureNodeByTargetNode(NcNode waitingCureNode,boolean tryPkdNodeHasProgramFiles){
		//新建一个节点，作为固化节点，老节点保留不动
		NcNode cureNode = new NcNode();
		BeanUtils.copyProperties(waitingCureNode, cureNode);
		EntityUtil.clearBaseProperties(cureNode);
		cureNode.setIsCured(1);
		cureNode.setIsLocked(NcNode.UNLOCK);
		cureNode.setIsLastEdition(1);
		cureNode.setExpireDate(calculateExpireDate());//设置过期日期

		Optional<NcNode> optOldCuredPackageNode = nodeService.lambdaQuery().eq(NcNode::getName,waitingCureNode.getName())
			.eq(NcNode::getProcessNo,waitingCureNode.getProcessNo())
			.eq(NcNode::getProcessEdition,waitingCureNode.getProcessEdition())
			.eq(NcNode::getIsLastEdition,1).eq(NcNode::getIsCured,1).oneOpt();

		if(optOldCuredPackageNode.isEmpty() && tryPkdNodeHasProgramFiles){
			//之前 没有已固化的程序包名，则直接更新版本号，如果有，那就使用node现在的版本号，因为不可用转为试切，审批完成已经升级version
			//tryPkdNodeHasProgramFiles 试切程序包下有程序文件才需要升级版本（没有文件的是车床，无文件，直接等待回传）
			cureNode.upgradeVersionNumber();
		}

		nodeService.save(cureNode);

		return cureNode;
	}
	/**
	 * 设置固化注释,实现方式，获取文件字节下，修改后替换文件
	 * @param flowProgramFile oss 文件
	 *
	 */
	void setGhAnnotation(FlowProgramFile flowProgramFile,String machineGroup,List<DictBiz> annoDicts) throws IOException {
		String ossName = flowProgramFile.getOssName();
		//
		try(InputStream ins = ossTemplate.statFileStream(ossName);){
			byte[] bytes = IOUtils.toByteArray(ins);
			ByteArrayInputStream byteInputStream =  new ByteArrayInputStream(bytes);

			InputStream finishedStream = programAnnotationService.setGHAnnotation(byteInputStream,machineGroup,annoDicts);

			try(finishedStream) {
				finishedStream.reset();
				BladeFile bfile = ossTemplate.putFile(flowProgramFile.getName(), finishedStream);
				//替换原有的文件地址
				flowProgramFile.setOssName(bfile.getName());

				flowProgramFileService.updateById(flowProgramFile);
			}
		}
	}

	/**
	 * 计算有效期
	 * @return
	 */
	private Date calculateExpireDate() {
		R<String> validMonResult = sysClient.getParamValue(ParamConstants.CURE_VALID_MONTH_KEY);

		int month = DEFAULT_VALID_MONTH;
		if(validMonResult.isSuccess() && validMonResult.getData()!=null){
			month = Integer.parseInt(validMonResult.getData());
		}

		LocalDate now = LocalDate.now();
		return DateUtil.toDate(now.plusMonths(month));
	}

	/*
	void updateApproveRecordNodeId(String processInstanceId,Long nodeId){
		//更新审批记录的ncNodeId
		approveRecordService.lambdaUpdate()
			.eq(ApproveRecord::getProcessInstanceId, processInstanceId)
			.set(ApproveRecord::getNcNodeId,nodeId).update();
	}
	 */
}