package org.springblade.mdm.program.service;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.io.IOUtils;
import org.apache.commons.lang3.StringUtils;
import org.jetbrains.annotations.NotNull;
import org.springblade.core.tool.api.R;
import org.springblade.mdm.utils.FileContentUtil;
import org.springblade.system.feign.IDictBizClient;
import org.springblade.system.pojo.entity.DictBiz;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

/**
 * 下发/回传程序统计
 *
 * @author yangys
 */
@Slf4j
@Service
public class ProgramAnnotationService {

	@Autowired
	private IDictBizClient bizDictClient;

	private static final String DEFAULT_ANNOTATION_SETTING = "(,)";

	public static final String ANNOTATION_DICT = "machine_annotation";
	/**
	 * 发送目录行索引（第2行） 0 based
	 */
	public static final int SENDPATH_LINE_INDEX = 1;
	/**
	 * 状态行索引（第三行） 0 based
	 */
	public static final int STATUS_LINE_INDEX = 2;

	/**
	 * 偏离单注释行索引号 0 based
	 */
	public static final int STATUS_DEVIATION_INDEX = 3;

	public static final String SQ = "SQ";
	public static final String GH = "GH";
	public static final String PL = "PL";

	public List<DictBiz> getAnnotionList(){
		List<DictBiz> annotationList;
		//R<List<Dict>> dictsResult = dictClient.getList(ANNOTATION_DICT);
		R<List<DictBiz>> dictBizResult = bizDictClient.getList(ANNOTATION_DICT);
		R<List<DictBiz>> gr = bizDictClient.getList("machine_group");
		R<List<DictBiz>> gcate = bizDictClient.getList("org_category");
		if(dictBizResult.isSuccess()) {
			annotationList = dictBizResult.getData();
		}else{
			annotationList = Collections.emptyList();
		}

		return annotationList;
	}


	/**
	 * 判断一行文本是否是注释
	 * @param line
	 * @param annotationList 注释注释字典项列表
	 * @return
	 */
	public boolean isAnnotation(String line,String machineGroupCode,List<DictBiz> annotationList){
		boolean isAnno = false;
		Optional<DictBiz> dictOpt = annotationList.stream().filter(dict -> {return dict.getDictKey().equals(machineGroupCode);}).findFirst();
		String annotationSetting = DEFAULT_ANNOTATION_SETTING;
		if(dictOpt.isPresent()) {
			annotationSetting = dictOpt.get().getRemark();
		}
		String begin = "";
		String end = "";
		String[] arr = annotationSetting.split(",");
		if(arr.length == 2){
			begin = arr[0];
			end = arr[1];
		}
		line = StringUtils.trim(line);//去首尾空格
		isAnno = StringUtils.startsWith(line,begin) && StringUtils.endsWith(line,end);
		return isAnno;
	}



	public String generateAnnotation(String oriTest,String machineGroupCode,List<DictBiz> annotationList) {
		String annotation = oriTest;
		String begin = "(";
		String end = ")";

		Optional<DictBiz> dictOpt = annotationList.stream().filter(dict -> {return dict.getDictKey().equals(machineGroupCode);}).findFirst();
		String annotationSetting = DEFAULT_ANNOTATION_SETTING;
		if(dictOpt.isPresent()) {
			annotationSetting = dictOpt.get().getRemark();
		}
		String[] arr = StringUtils.split(annotationSetting,",");
		if(arr.length == 2){
			begin = arr[0];
			end = arr[1];
		}else if(arr.length == 1){
			begin = begin = arr[0];
			end = "";
		}

		annotation = begin + oriTest + end;
		return annotation;
	}

	public String[] getAnnotationStartAndEnd(String machineGroupCode,List<DictBiz> annotationList) {

		String begin = "(";
		String end = ")";

		Optional<DictBiz> dictOpt = annotationList.stream().filter(dict -> {return dict.getDictKey().equals(machineGroupCode);}).findFirst();
		String annotationSetting = DEFAULT_ANNOTATION_SETTING;
		if(dictOpt.isPresent()) {
			annotationSetting = dictOpt.get().getRemark();
		}
		String[] arr = StringUtils.split(annotationSetting,",");
		if(arr.length == 2){
			begin = arr[0];
			end = arr[1];
		}else if(arr.length == 1){
			begin = arr[0];
			end = "";
		}


		return new String[]{begin,end};
	}

	public String removeAnnotation(String machineGroupCode,String text,String begin,String end){
		List<DictBiz> annoDicts = getAnnotionList();
		String[] arr = getAnnotationStartAndEnd(machineGroupCode,annoDicts);
		return removeAnnotationInner(text,begin,end);
	}

	public static String removeAnnotationInner(String text,String begin,String end){
		String temp = StringUtils.removeEnd(StringUtils.removeStart(text,begin),end);
		return temp;
	}



	/**
	 * 设置固化注释行
	 * @param inputStream
	 * @param machineGroup
	 * @param annoDictList
	 * @return
	 * @throws IOException
	 */
	public InputStream setGHAnnotation(InputStream inputStream, String machineGroup,List<DictBiz> annoDictList) throws IOException {

		return setAnnotationAndGetInputStream(inputStream, machineGroup, annoDictList, GH, STATUS_LINE_INDEX);
	}

	/**
	 * 设置偏离单注释
	 * @param inputStream
	 * @param machineGroup
	 * @param annoDictList
	 * @return
	 * @throws IOException
	 */
	public InputStream setPldAnnotation(String deviation,InputStream inputStream,String machineGroup,List<DictBiz> annoDictList) throws IOException {
		//
		return setAnnotationAndGetInputStream(inputStream, machineGroup, annoDictList, deviation, STATUS_DEVIATION_INDEX);
	}

	@NotNull
	private InputStream setAnnotationAndGetInputStream(InputStream inputStream, String machineGroup, List<DictBiz> annoDictList, String text, int lineIndex) throws IOException {
		InputStream finishedStream;
		try(inputStream){
			ByteArrayInputStream byteInputStream =  new ByteArrayInputStream(IOUtils.toByteArray(inputStream));

			String annoTxt = generateAnnotation(text,machineGroup,annoDictList);//加了注释之后的文本

			String lineText = FileContentUtil.readLineAt(byteInputStream, lineIndex);
			byteInputStream.reset();

			if(isAnnotation(lineText,machineGroup,annoDictList)){
				finishedStream = FileContentUtil.replaceAtLine(byteInputStream, lineIndex,annoTxt);
			}else{
				finishedStream = FileContentUtil.insertLine(byteInputStream, lineIndex,annoTxt);
			}
			finishedStream.reset();
		}

		return finishedStream;
	}


	/**
	 * 为输入流设置发送目录(第二行)和状态（第三行）
	 * @param sendPath 发送路径
	 * @param statusText 状态文本
	 * @param inputStream 输入流
	 * @param machineGroup 机床组代码
	 * @return 加好发送路径和状态文本的输入流
	 * @throws IOException
	 */
	public InputStream setSendDirAndStatusAnnotation(String sendPath,String statusText,InputStream inputStream, String machineGroup) throws IOException {
		List<DictBiz> annoDicts = getAnnotionList();

		InputStream finishedStream;
		try(inputStream){
			ByteArrayInputStream byteInputStream =  new ByteArrayInputStream(IOUtils.toByteArray(inputStream));

			//1加入发送路径的注释
			String sendPathAnnotation = generateAnnotation(sendPath,machineGroup,annoDicts);//加了注释之后的文本

			InputStream insAfterSetSendDir ;
			String sendDirLine = FileContentUtil.readLineAt(byteInputStream,SENDPATH_LINE_INDEX);//第2行是发送路径
			byteInputStream.reset();
			String statusLine = FileContentUtil.readLineAt(byteInputStream,STATUS_LINE_INDEX);//第三行是状态注释
			byteInputStream.reset();

			if(isAnnotation(sendDirLine,machineGroup,annoDicts)){
				insAfterSetSendDir = FileContentUtil.replaceAtLine(byteInputStream,SENDPATH_LINE_INDEX,sendPathAnnotation);
			}else{
				insAfterSetSendDir = FileContentUtil.insertLine(byteInputStream,SENDPATH_LINE_INDEX,sendPathAnnotation);
			}

			insAfterSetSendDir.reset();
			//加入状态注释行
			String statusAnnotation = generateAnnotation(statusText,machineGroup,annoDicts);//注释后的状态文本
			if(isAnnotation(statusLine,machineGroup,annoDicts)){
				finishedStream = FileContentUtil.replaceAtLine(insAfterSetSendDir,STATUS_LINE_INDEX,statusAnnotation);
			}else{
				finishedStream = FileContentUtil.insertLine(insAfterSetSendDir,STATUS_LINE_INDEX,statusAnnotation);
			}
		}

		return finishedStream;
	}
}