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

import com.qiniu.util.IOUtils;
import org.apache.commons.io.FilenameUtils;
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;


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

	@Autowired
	private ProgramAnnotationService programAnnotationService;
	private String controlSystem;


	@Override
	public InputStream setAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
		List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();
		ByteArrayInputStream bais = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));

		String line = FileContentUtil.readLineAt(bais, annotationProperties.getProgramNameLineIndex());
		bais.reset();
		boolean isAnnotation = AnnotationUtil.isAnnotation(line,controlSystem,annoDicts);
		InputStream insAfter;
		//加入程序名注释行
		String proNameLine =  AnnotationUtil.generateAnnotation(FilenameUtils.removeExtension(annoData.getFilename()),getControlSystem(),annoDicts);
		if(isAnnotation){
			insAfter = FileContentUtil.replaceAtLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
		}else{
			insAfter = FileContentUtil.insertLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
		}

		return super.setAnnotation(annoData, insAfter);

	}

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

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


}