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 java.io.ByteArrayInputStream;
|
import java.io.IOException;
|
import java.io.InputStream;
|
import java.util.List;
|
|
public abstract class AbstractProcessor implements AnnotationProcessor{
|
|
@Autowired
|
private ProgramAnnotationService programAnnotationService;
|
|
protected AnnotationProperties annotationProperties;
|
@Override
|
public InputStream putSendPathAnnotation(String sendPath, InputStream inputStream,List<DictBiz> annoDicts) throws IOException{
|
InputStream finishedStream;
|
try(inputStream){
|
ByteArrayInputStream byteInputStream = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));
|
|
//1加入发送路径的注释
|
String sendPathAnnotation = AnnotationUtil.generateAnnotation(sendPath,getControlSystem(),annoDicts);//加了注释之后的文本
|
|
String sendDirLine = FileContentUtil.readLineAt(byteInputStream,annotationProperties.getSendPathLineIndex());//第2行是发送路径
|
byteInputStream.reset();
|
|
if(AnnotationUtil.isAnnotation(sendDirLine,getControlSystem(),annoDicts)){
|
String planText = AnnotationUtil.removeAnnotation(getControlSystem(),sendDirLine,annoDicts);
|
if(!planText.equals(sendPath)) {
|
//非路径的注释，插入
|
finishedStream = FileContentUtil.insertLine(byteInputStream,annotationProperties.getSendPathLineIndex(),sendPathAnnotation);
|
}else{
|
//是路径，不处理直接返回原输入流
|
finishedStream = inputStream;
|
}
|
//finishedStream = FileContentUtil.replaceAtLine(byteInputStream,annotationProperties.getSendPathLineIndex(),sendPathAnnotation);
|
}else{
|
finishedStream = FileContentUtil.insertLine(byteInputStream,annotationProperties.getSendPathLineIndex(),sendPathAnnotation);
|
}
|
finishedStream.reset();
|
}
|
|
return finishedStream;
|
|
}
|
@Override
|
public InputStream putFilenameAnnotation(String fileName, InputStream inputStream) throws IOException {
|
List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();//TODO 这里应该不用加载，参数传过来就行
|
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 proNameAnnotation = AnnotationUtil.generateAnnotation(FilenameUtils.removeExtension(fileName),getControlSystem(),annoDicts);
|
|
if(isAnnotation){//是注释
|
//提取注释内文字
|
//String planText = AnnotationUtil.removeAnnotation(getControlSystem(),programNameline,annoDicts);
|
if(!StringUtils.equals(programNameLine,proNameAnnotation)) {
|
//注释不是程序名（工艺员自己写的注释），插入
|
insAfterProgramName = FileContentUtil.insertLine(byteStream,annotationProperties.getProgramNameLineIndex(),proNameAnnotation);
|
}else{
|
//是程序名注释，不处理，返回原stream
|
insAfterProgramName = byteStream;
|
}
|
}else{
|
//非注释，插入状态
|
insAfterProgramName = FileContentUtil.insertLine(byteStream,annotationProperties.getProgramNameLineIndex(),proNameAnnotation);
|
}
|
/*
|
if(isAnnotation){
|
insAfter = FileContentUtil.replaceAtLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
|
}else{
|
insAfter = FileContentUtil.insertLine(bais, annotationProperties.getProgramNameLineIndex(),proNameLine);
|
}*/
|
return insAfterProgramName;
|
}
|
|
@Override
|
public InputStream putAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
|
List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();
|
|
InputStream finishedStream;
|
try(inputStream){
|
ByteArrayInputStream byteInputStream = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));
|
|
//1加入发送路径的注释
|
InputStream insAfterSetSendDir = putSendPathAnnotation(annoData.getSendPath(),byteInputStream,annoDicts);
|
/*
|
String statusLine = FileContentUtil.readLineAt(insAfterSetSendDir,annotationProperties.getStatusLineIndex());//状态注释
|
insAfterSetSendDir.reset();
|
|
//加入状态注释行
|
InputStream insAfterStatus;
|
String statusAnnotation = AnnotationUtil.generateAnnotation(annoData.getProgramStatus(),getControlSystem(),annoDicts);//注释后的状态文本
|
if(AnnotationUtil.isAnnotation(statusLine,getControlSystem(),annoDicts)){
|
insAfterStatus = FileContentUtil.replaceAtLine(insAfterSetSendDir,annotationProperties.getStatusLineIndex(),statusAnnotation);
|
}else{
|
insAfterStatus = FileContentUtil.insertLine(insAfterSetSendDir,annotationProperties.getStatusLineIndex(),statusAnnotation);
|
}*/
|
InputStream insAfterStatus = putStatusAnnotation(annoData.getProgramStatus(),insAfterSetSendDir,annoDicts);
|
|
if(StringUtils.isNotBlank(annoData.getDeviation())){
|
finishedStream = putDeviationAnnotation(annoData.getDeviation(),insAfterStatus,annoDicts);
|
}else{
|
//是更改单号，不处理直接返回原输入流
|
finishedStream = insAfterStatus;
|
}
|
}
|
|
return finishedStream;
|
}
|
|
protected InputStream putStatusAnnotation(String status,InputStream inputStream,List<DictBiz> annoDicts) throws IOException {
|
InputStream insAfterStatus;
|
//加入lg单号
|
|
String statusLine = FileContentUtil.readLineAt(inputStream,annotationProperties.getStatusLineIndex());//状态注释
|
String stationAnnotation = AnnotationUtil.generateAnnotation(status,getControlSystem(),annoDicts);
|
|
inputStream.reset();
|
if(AnnotationUtil.isAnnotation(statusLine,getControlSystem(),annoDicts) ){
|
//是注释
|
//提取注释内文字
|
String planText = AnnotationUtil.removeAnnotation(getControlSystem(),statusLine,annoDicts);
|
if(!StringUtils.equalsAny(planText,AnnotationUtil.SQ,AnnotationUtil.GH,AnnotationUtil.LG)) {
|
//注释不是状态（工艺员自己写的注释），插入
|
insAfterStatus = FileContentUtil.insertLine(inputStream,annotationProperties.getStatusLineIndex(),stationAnnotation);
|
}else{
|
//是状态注释，替换
|
insAfterStatus = FileContentUtil.replaceAtLine(inputStream,annotationProperties.getStatusLineIndex(),stationAnnotation);
|
}
|
}else{
|
//非注释，插入状态
|
insAfterStatus = FileContentUtil.insertLine(inputStream,annotationProperties.getStatusLineIndex(),stationAnnotation);
|
}
|
return insAfterStatus;
|
}
|
|
protected InputStream putDeviationAnnotation(String deviation,InputStream inputStream,List<DictBiz> annoDicts) throws IOException {
|
InputStream insAfter1;
|
//加入lg单号
|
String devLine = FileContentUtil.readLineAt(inputStream,annotationProperties.getDeviationLineIndex());//状态注释
|
|
String devAnnotation = AnnotationUtil.generateAnnotation(deviation,getControlSystem(),annoDicts);
|
|
inputStream.reset();
|
if(AnnotationUtil.isAnnotation(devLine,getControlSystem(),annoDicts) ){
|
//是注释
|
//提取注释内文字
|
String planText = AnnotationUtil.removeAnnotation(getControlSystem(),devLine,annoDicts);
|
if(!planText.equals(deviation)) {
|
//注释不是临时更改单号（工艺员自己写的注释），插入
|
insAfter1 = FileContentUtil.insertLine(inputStream,annotationProperties.getDeviationLineIndex(),devAnnotation);
|
}else{
|
//是更改单号，不处理直接返回原输入流
|
insAfter1 = inputStream;
|
}
|
}else{
|
//非注释，怎插入lg号
|
insAfter1 = FileContentUtil.insertLine(inputStream,annotationProperties.getDeviationLineIndex(),devAnnotation);
|
}
|
return insAfter1;
|
}
|
|
@Override
|
public abstract void setControlSystem(String controlSystemDictVal) ;
|
|
@Override
|
public void setAnnotationProperties(AnnotationProperties annotationProperties) {
|
this.annotationProperties= annotationProperties;
|
}
|
|
@Override
|
public AnnotationProperties getAnnotationProperties(){
|
return annotationProperties;
|
}
|
public abstract String getControlSystem();
|
|
@Override
|
public AnnotationData readAnnotationData(InputStream inputStream) {
|
AnnotationData data = new AnnotationData();
|
|
int maxLineIndex = maxAnnotationLineIndex();
|
try {
|
int lineCount = maxLineIndex+1;
|
List<String> lines = FileContentUtil.readFirstNLines(inputStream,lineCount);
|
fixLine(lines,lineCount);
|
|
String programNameLine = lines.get(this.getAnnotationProperties().getProgramNameLineIndex());
|
String sendPathLine = lines.get(this.getAnnotationProperties().getSendPathLineIndex());
|
String statusLine = lines.get(this.getAnnotationProperties().getStatusLineIndex());
|
String devLine = lines.get(this.getAnnotationProperties().getDeviationLineIndex());
|
|
List<DictBiz> annoDictList= this.programAnnotationService.getAnnotionDictList();
|
data.setFilename(AnnotationUtil.removeAnnotation(this.getControlSystem(),programNameLine,annoDictList));
|
data.setSendPath(AnnotationUtil.removeAnnotation(this.getControlSystem(),sendPathLine,annoDictList));
|
|
String statusText = AnnotationUtil.removeAnnotation(this.getControlSystem(),statusLine,annoDictList);
|
if(AnnotationUtil.isStatusContent(statusText)){
|
data.setProgramStatus(statusText);
|
}
|
|
data.setDeviation(AnnotationUtil.removeAnnotation(this.getControlSystem(),devLine,annoDictList));
|
|
} catch (IOException e) {
|
throw new RuntimeException(e);
|
}
|
|
return data;
|
}
|
|
/**
|
* 没有的行填充未空串
|
* @param lines
|
* @return
|
*/
|
void fixLine(List<String> lines,int fixLineCount){
|
int needFix = fixLineCount - lines.size();
|
if(needFix > 0){
|
for(int i = 0 ; i < needFix ; i++){
|
lines.add(StringUtils.EMPTY);
|
}
|
}
|
}
|
|
private int maxAnnotationLineIndex(){
|
int max = this.getAnnotationProperties().getDeviationLineIndex();
|
if(max < this.getAnnotationProperties().getStatusLineIndex()){
|
max = this.getAnnotationProperties().getStatusLineIndex();
|
}
|
|
if(max < this.getAnnotationProperties().getSendPathLineIndex()){
|
max = this.getAnnotationProperties().getSendPathLineIndex();
|
}
|
if(max < this.getAnnotationProperties().getProgramNameLineIndex()){
|
max = this.getAnnotationProperties().getProgramNameLineIndex();
|
}
|
return max;
|
}
|
}
|
