package org.springblade.mdm.program.service.programannotation;

import org.apache.commons.io.FilenameUtils;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.springblade.mdm.program.service.ProgramAnnotationService;
import org.springblade.mdm.utils.FileContentUtil;
import org.springblade.system.pojo.entity.DictBiz;
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.util.List;
import java.util.regex.Matcher;
import java.util.regex.Pattern;


@Component("defaultProcessor")
public class DefaultProcessor extends AbstractProcessor{

	//@Autowired
	//private ProgramAnnotationService programAnnotationService;
	private String controlSystem;


	@Override
	public InputStream putAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
		InputStream insAfter = putFilenameAnnotation(annoData.getFilename(),inputStream);

		return super.putAnnotation(annoData, insAfter);

	}

	@Override
	public InputStream putFilenameAnnotation(String fileName, InputStream inputStream) throws IOException {
		List<DictBiz> annoDicts = getAnnotationDictList();
		ByteArrayInputStream byteStream = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));

		String programNameLine = FileContentUtil.readLineAt(byteStream, annotationProperties.getProgramNameLineIndex());
		byteStream.reset();
		boolean isAnnotation = AnnotationUtil.isAnnotation(programNameLine,this.getControlSystem(),annoDicts);
		InputStream insAfterProgramName;
		//加入程序名注释行
		String programNameStr = genProgramNameText(FilenameUtils.removeExtension(fileName));

		String proNameAnnotation =  AnnotationUtil.generateAnnotation(programNameStr,getControlSystem(),annoDicts);

		if(isAnnotation){//是注释
			//提取注释内文字
			String programName = FilenameUtils.removeExtension(fileName);
			String progNameLineWithoutAnno = AnnotationUtil.removeAnnotation(this.getControlSystem(),programNameLine,annoDicts);
			if(!isCurrentProgramNameAnnotation(progNameLineWithoutAnno,programName)) {
				//注释不是程序名（工艺员自己写的注释），插入
				insAfterProgramName = FileContentUtil.insertLine(byteStream,annotationProperties.getProgramNameLineIndex(),proNameAnnotation);
			}else{
				//是程序名注释，不处理，返回原stream
				insAfterProgramName =  byteStream;
			}
		}else{
			//非注释，插入状态
			insAfterProgramName = FileContentUtil.insertLine(byteStream,annotationProperties.getProgramNameLineIndex(),proNameAnnotation);
		}
		return insAfterProgramName;
	}

	/**
	 * 是否为当前程序的注释
	 * @param lineTextWithoutAnno 注释行的文字
	 * @param programName 程序名称
	 * @return 是否时程序注释行，如果与当前程序名匹配则true
	 */
	boolean isCurrentProgramNameAnnotation(String lineTextWithoutAnno,String programName){
		Pattern pattern = Pattern.compile("^\\((.+)\\)");
		Matcher matcher = pattern.matcher(lineTextWithoutAnno);
		if(matcher.find()){
			String findText = matcher.group(1);
			return findText.equals(programName);
		}else{
			return false;
		}
	}
	/**
	 * 生成程序名成的行 加入 (pname=[程序名称])
	 * @param proggramName
	 * @return 报装后的程序名称
	 */
	private String genProgramNameText(String proggramName) {
		return String.format("(%s)",proggramName);
	}

	@Override
	public void setControlSystem(String controlSystemDictVal) {
		this.controlSystem = controlSystemDictVal;
	}

	@Override
	public String getControlSystem() {
		return this.controlSystem;
	}


}