package org.springblade.mdm.flow.service;

import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.flowable.engine.RuntimeService;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.core.mp.base.BizServiceImpl;
import org.springblade.core.oss.OssTemplate;
import org.springblade.core.oss.model.BladeFile;
import org.springblade.core.tool.api.IResultCode;
import org.springblade.core.tool.api.ResultCode;
import org.springblade.core.tool.utils.Func;
import org.springblade.core.tool.utils.NumberUtil;
import org.springblade.mdm.flow.entity.FlowProgramFile;
import org.springblade.mdm.flow.mapper.FlowProgramFileMapper;
import org.springblade.mdm.flow.vo.ProgramUploadVO;
import org.springblade.mdm.program.entity.NcProgram;
import org.springblade.mdm.program.service.ProcessProgRefService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.IOException;
import java.io.InputStream;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.Map;

/**
 * 流程程序处理，包括上传，查询展示
 *
 * @author yangys
 */
@Slf4j
@Service
public class FlowProgramFileService extends BizServiceImpl<FlowProgramFileMapper, FlowProgramFile> {
	@Autowired
	private OssTemplate ossTemplate;
	@Autowired
	private RuntimeService runtimeService;
	@Autowired
	private FlowCommonService flowCommonService;

	public void uploadFlowProgramFile(ProgramUploadVO uploadVO) {

		//Map<String, Object> vars = runtimeService.getVariables(uploadVO.getProcessInstanceId());

		FlowProgramProperties progProps= flowCommonService.getProgramProperties(uploadVO.getProcessInstanceId());
		checkFilenames(uploadVO.getFiles(),progProps);

		String programName = getProgramName(progProps);
		for(MultipartFile file : uploadVO.getFiles()) {
			BladeFile bfile = ossTemplate.putFile(file);

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

			progFile.setProgramName(programName);

			save(progFile);
		}
		//
		//BladeFile bfile = ossTemplate.putFile(uploadVO.getFile());
		//String link = bfile.getLink();
		//保存profame
		/*
		NcProgram prog = new NcProgram();
		prog.setCode(generageCode());
		prog.setName(file.getOriginalFilename());
		prog.setNcNodeId(uploadVO.getNodeId());
		prog.setOssName(bfile.getName());
		prog.setDrawingNo(uploadVO.getDrawingNo());
		prog.setProcessEdition(uploadVO.getProcessEdition());
		prog.setIsLastEdition(1);

		 */
	}

	String getProgramName(FlowProgramProperties progProps){
		return progProps.getDrawingNo()+"-"+progProps.getProcessNo();
	}

	public void checkFilenames(MultipartFile[] files,FlowProgramProperties programProperties){
		for (MultipartFile file : files){
			checkFilename(file.getOriginalFilename(),programProperties);
		}
	}

	/**
	 * 检查文件名合法性
	 * @param filename
	 * @param programProperties
	 */
	void checkFilename(String filename,FlowProgramProperties programProperties){
		//程序名称：零件号加工序号，文件名应该以此开头
		String expectedProgramName = programProperties.getDrawingNo()+"-"+programProperties.getProcessNo();
		if(!StringUtils.startsWith(filename,expectedProgramName)){

			IResultCode rc = new IResultCode() {
				@Override
				public String getMessage() {
					return "程序文件名不合法，应为："+expectedProgramName+"-"+programProperties.getCraftEdition()+"-[段数]-[段号].[文件扩展名]";
				}

				@Override
				public int getCode() {
					return 1;
				}
			};
			throw new ServiceException(rc);
			//throw new ServiceException("程序文件名不合法，应为："+expectedProgramName+"-"+programProperties.getCraftEdition()+"-[段数]-[段号].[文件扩展名]");
		}

		//截取后面的段数和第几段
		String endPart = StringUtils.removeStart(filename,expectedProgramName+"-"+programProperties.getCraftEdition()+"-");

		//去掉扩展名
		if(endPart.contains(".")){
			endPart = endPart.substring(0,endPart.indexOf("."));
		}

		int sepCount = StringUtils.countMatches(endPart,"-");
		if(sepCount != 1){//- 号应该是1个
			IResultCode rc = new IResultCode() {
				@Override
				public String getMessage() {
					return "程序文件名不合法，应为："+expectedProgramName+"-"+programProperties.getCraftEdition()+"-[段数]-[段号].[文件扩展名]";
				}

				@Override
				public int getCode() {
					return 2;
				}
			};
			throw new ServiceException(rc);
			//throw new ServiceException("程序文件名不合法，应为："+expectedProgramName+"-"+programProperties.getCraftEdition()+"-[段数]-[段号].[文件扩展名]");
		}

		String[] arr = StringUtils.split(endPart,"-");
		if(!StringUtils.isNumeric(arr[0]) || Func.toInt(arr[0]) >99 || Func.toInt(arr[0]) < 1){
			IResultCode rc = new IResultCode() {
				@Override
				public String getMessage() {
					return "程序段数不合法，应为两位以内整数";
				}

				@Override
				public int getCode() {
					return 3;
				}
			};
			throw new ServiceException(rc);
		}
		int segCount = Func.toInt(arr[0]);

		if(!StringUtils.isNumeric(arr[1]) || Func.toInt(arr[1]) < 1 || Func.toInt(arr[1]) > segCount){
			IResultCode rc = new IResultCode() {
				@Override
				public String getMessage() {
					return "程序段号不合法，应为两位以内整数且小于等于段数。";
				}

				@Override
				public int getCode() {
					return 4;
				}
			};
			throw new ServiceException(rc);
		}
	}

	/**
	 * 获取文件内容
	 * @param id
	 * @return
	 */
	public String getFileContent(Long id) {
		String result  ="";

		FlowProgramFile programFile = this.getById(id);
		//if(prog.getIsTextFile()!=null && programFile.getIsTextFile()){
			String fileName = programFile.getOssName();
			try (InputStream inputStream = ossTemplate.statFileStream(fileName)) {
				result = IOUtils.toString(inputStream, StandardCharsets.UTF_8);
			} catch (IOException e) {
				throw new RuntimeException(e);
			}
		//}else{
		//	result = "非文本格式文件";
		//}
		return result;
	}


}