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;
|
import java.util.regex.Pattern;
|
|
@Component("fanucProcessor")
|
public class FanucProcessor extends AbstractProcessor{
|
@Autowired
|
private ProgramAnnotationService programAnnotationService;
|
private String controlSystem;
|
static final Pattern PATTERN = Pattern.compile("^[oO]\\d{4}.*");
|
//private final int O_LINE = 1;
|
@Override
|
public InputStream putAnnotation(AnnotationData annoData, InputStream inputStream) throws IOException {
|
//在O0001 这种文字后面增加文件名（不带扩展名）
|
ByteArrayInputStream byteStream = new ByteArrayInputStream(IOUtils.toByteArray(inputStream));
|
//String line = FileContentUtil.readLineAt(byteStream,O_LINE);
|
String line = FileContentUtil.readLineAt(byteStream,getAnnotationProperties().getProgramNameLineIndex());
|
|
byteStream.reset();
|
boolean oMatched = PATTERN.matcher(line).matches();
|
InputStream ins1;
|
if(oMatched){
|
List<DictBiz> annoDicts = programAnnotationService.getAnnotionDictList();
|
|
String programAnno = AnnotationUtil.generateAnnotation(FilenameUtils.removeExtension(annoData.getFilename()),getControlSystem(),annoDicts);
|
|
ins1 = FileContentUtil.replaceAtLine(byteStream,getAnnotationProperties().getProgramNameLineIndex(),genProgramNameLine(line,programAnno));
|
}else{
|
ins1 = byteStream;
|
}
|
return super.putAnnotation(annoData, ins1);
|
}
|
|
/**
|
* 生成程序名称注释行，O[4位数字](程序名) ，需要将用户自己的程序名去或其他注释掉放后面
|
* @param oriLine 原始行文本
|
* @param programNameAnno 程序注释文本：(程序名)
|
* @return Oxxxx+程序注释文本
|
*/
|
String genProgramNameLine(String oriLine,String programNameAnno){
|
return oriLine.substring(0,5)+programNameAnno;
|
}
|
|
@Override
|
public void setControlSystem(String controlSystemDictVal) {
|
this.controlSystem = controlSystemDictVal;
|
}
|
|
@Override
|
public String getControlSystem() {
|
return this.controlSystem;
|
}
|
}
|
