package org.springblade.mdm.flow.service;

import lombok.AllArgsConstructor;
import org.flowable.engine.IdentityService;
import org.flowable.engine.RuntimeService;
import org.flowable.engine.TaskService;
import org.flowable.engine.runtime.ProcessInstance;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.oss.OssTemplate;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.secure.utils.AuthUtil;
import org.springblade.mdm.basesetting.machine.service.MachineService;
import org.springblade.mdm.flow.constants.FlowContants;
import org.springblade.mdm.flow.constants.FlowVariableContants;
import org.springblade.mdm.flow.entity.FlowProgramFile;
import org.springblade.mdm.flow.vo.ReplaceFlowStartVO;
import org.springblade.mdm.flow.vo.ReplaceUploadVO;
import org.springblade.mdm.program.entity.NcNode;
import org.springblade.mdm.program.service.NcNodeService;
import org.springblade.mdm.program.service.NodeDeptQueryService;
import org.springblade.mdm.program.service.ProgramFlowStatusQueryService;
import org.springblade.mdm.program.vo.NcNodeVO;
import org.springblade.mdm.utils.EntityUtil;
import org.springblade.mdm.utils.ProgramFileNameCheckUtil;
import org.springframework.beans.BeanUtils;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.web.multipart.MultipartFile;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

@AllArgsConstructor
@Service
public class ReplaceFlowService {
	private final NcNodeService nodeService;
	private final RuntimeService runtimeService;
	private final TaskService taskService;
	private final IdentityService identityService;
	private final FlowProgramFileService flowProgramFileService;
	private final ProgramFlowStatusQueryService flowStatusQueryService;
	private final NodeDeptQueryService nodeDeptQueryService;
	private final FlowCommonService flowCommonService;
	private final OssTemplate ossTemplate;
	private final MachineService machineService;

	/**
	 * 准备替换流程需要的数据
	 * @param nodeId 要替换的程序包名节点id
	 * @param tempInstanceId 临时流程实例id，用于新的文件
	 * @return 启动数据
	 */
	@Transactional
	public NcNodeVO pre(long nodeId, String tempInstanceId) {
		checkFlowStatus(nodeId);

		List<NcNode> fileNodes = nodeService.lambdaQuery()
			.eq(NcNode::getParentId, nodeId)
			.eq(NcNode::getIsLastEdition,1)
			.eq(NcNode::getNodeType,NcNode.TYPE_PROGRAM_FILE).list();
		List<Long> fileIds = fileNodes.stream().map(NcNode::getFlowProgramFileId).toList();
		if(fileIds.isEmpty()){
			throw new RuntimeException("未找到程序文件");
		}
		List<FlowProgramFile> programFiles = flowProgramFileService.lambdaQuery()
			.in(FlowProgramFile::getId, fileIds).list();


		//将现有文件复制到独立的表中，后续作为审批的文件，审批界面调用的接口也不同，最好单独开发
		for(FlowProgramFile programFile : programFiles) {
			FlowProgramFile newFile = new FlowProgramFile();
			BeanUtils.copyProperties(programFile, newFile);
			EntityUtil.clearBaseProperties(newFile);
			newFile.setProcessInstanceId(tempInstanceId);
			flowProgramFileService.save(newFile);
		}

		NcNode node = nodeService.getById(nodeId);
		NcNodeVO vo = new NcNodeVO();
		BeanUtils.copyProperties(node, vo);
		vo.setProcessInstanceId(tempInstanceId);

		vo.setWorkshop(nodeDeptQueryService.getWorkshopNameByMachineCode(node.getMachineCode()));

		return vo;
	}

	void checkFlowStatus(long nodeId) {
		NcNode packageNode = nodeService.getById(nodeId);
		int status = flowStatusQueryService.queryFlowStatus(packageNode.getProcessInstanceId());
		if (status == 1) {
			throw new ServiceException("程序的流程正在进行中，不能进行替换");
		}
	}

	/**
	 * 启动替换流程
	 */
	@Transactional
	public void start(ReplaceFlowStartVO startVO){
		flowProgramFileService.checkProgramFiles(startVO.getTempInstanceId(),true);
		NcNode programPackage = nodeService.getById(startVO.getNodeId());

		Map<String, Object> vars = new HashMap<>();

		flowCommonService.putFlowVariablesByNode(vars,programPackage);
		flowCommonService.putDefaultAssignees(vars,programPackage.getDrawingNo(),startVO.getAssignee()+"");

		vars.put(FlowContants.TITLE,startVO.getTitle());

		vars.put("comment", startVO.getComment());
		vars.put(FlowContants.MY_PROCESS_NAME,"替换流程");

		String userId = String.valueOf(AuthUtil.getUserId());
		identityService.setAuthenticatedUserId(userId);//设置流程发起人
		vars.put(FlowVariableContants.ACT_PROGRAMMER,userId);//发起人按照实际编程人员算

		ProcessInstance inst = runtimeService.startProcessInstanceByKey(FlowContants.REPLACE_PROCESS_KEY,startVO.getNodeId()+"",vars);
		//更新节点的流程id，避免重复发起流程
		programPackage.setProcessInstanceId(inst.getId());
		this.nodeService.updateById(programPackage);

		flowProgramFileService.lambdaUpdate()
			.eq(FlowProgramFile::getProcessInstanceId,startVO.getTempInstanceId())
			.set(FlowProgramFile::getProcessInstanceId,inst.getProcessInstanceId()).update();

	}


	/**
	 * 上传程序徐文件（编制节点调用）
	 * @param uploadVO 上传对象
	 */
	public void uploadReplaceProgramFile(ReplaceUploadVO uploadVO) {

		MultipartFile file = uploadVO.getFile();
		if(file.getSize() == 0){
			throw new ServiceException("程序文件不可为空文件");
		}
		FlowProgramProperties progProps = new FlowProgramProperties();
		progProps.setDrawingNo(uploadVO.getDrawingNo());
		progProps.setProcessNo(uploadVO.getProcessNo());
		progProps.setProcessEdition(uploadVO.getProcessEdition());
		if(FlowProgramFile.TYPE_PROGRAM.equals(uploadVO.getFileType())) {
			ProgramFileNameCheckUtil.checkFilename(file.getOriginalFilename(),progProps);
		}


		String programName = NcNodeService.genProgramName(progProps.getDrawingNo(),progProps.getProcessNo(),progProps.getProcessEdition());

		BladeFile bfile = ossTemplate.putFile(file);

		FlowProgramFile progFile = new FlowProgramFile();
		progFile.setName(file.getOriginalFilename());
		progFile.setOssName(bfile.getName());
		progFile.setProcessInstanceId(uploadVO.getProcessInstanceId());
		progFile.setProgramName(programName);
		progFile.setFileType(uploadVO.getFileType());

		flowProgramFileService.save(progFile);

	}
}