package org.springblade.mdm.machinefile.service;

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.digest.DigestUtils;
import org.springblade.core.log.exception.ServiceException;
import org.springblade.mdm.basesetting.machine.entity.Machine;
import org.springblade.mdm.basesetting.machine.service.MachineService;
import org.springblade.mdm.commons.service.ParamService;
import org.springblade.mdm.gkw.programnode.vo.ProgramNameVO;
import org.springblade.mdm.machinefile.entity.FileSendRecord;
import org.springblade.mdm.machinefile.entity.MachineFile;
import org.springblade.mdm.program.service.ProgramAnnotationService;
import org.springblade.mdm.program.service.programannotation.AnnotationUtil;
import org.springblade.mdm.utils.FileContentUtil;
import org.springblade.mdm.utils.ProgramFileNameParser;
import org.springblade.system.pojo.entity.DictBiz;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.scheduling.annotation.Async;
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;

import java.io.BufferedInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.nio.file.attribute.BasicFileAttributes;
import java.nio.file.attribute.FileTime;
import java.util.Date;
import java.util.List;
import java.util.Optional;

@Slf4j
@Component
public class ReceiveFileCheckService {
	@Autowired
	private MachineFileService machineFileService;
	@Autowired
	private MachineService machineService;
	@Autowired
	private ParamService paramService;
	@Autowired
	private FileSendRecordService fileSendRecordService;
	/**
	 * 文件合法性检查
	 * @param machineFile
	 */
	@Transactional(rollbackFor = Exception.class)
	public void check(MachineFile machineFile,List<MachineFile> allFilesInDir,Machine machine) throws IOException {
		String srcFilepath = MachineFileService.getBasePath(machine,machineFile.getDirType())+ File.separator+machineFile.getName();
		Path source = Paths.get(srcFilepath);
		if(!source.toFile().exists()){
			return;
		}
		try(InputStream fileIns = Files.newInputStream(Paths.get(srcFilepath));) {
			if(!FileContentUtil.isTextFile(fileIns)){
				//非文本
				machineFile.setExceptionType(MachineFile.EXCEPTION_NOT_TEXT);
			}else{
				ProgramNameVO progNameVO =  ProgramFileNameParser.parseProgramName(machineFile.getName());
				if(!progNameVO.isValidFilename()){
					machineFile.setExceptionType(MachineFile.EXCEPTION_BAD_FILENAME);
				}else {
					String prefix = progNameVO.logicProgramName()+"-";
					long matchCount = allFilesInDir.stream().filter(file -> file.getName().startsWith(prefix)).count();
					if(matchCount != progNameVO.getSegmentCount()){//文件段数缺失
						machineFile.setExceptionType(MachineFile.EXCEPTION_LOST_FILES);
					}else{
						//检查是否匹配下发记录的段数
						//正负3秒作为查询时间
						Date beginTime = new Date(machineFile.getFileCreateTime().getTime()-3000);
						Date endTime = new Date(machineFile.getFileCreateTime().getTime()+3000);
						Optional<FileSendRecord> optFile = fileSendRecordService.lambdaQuery()
							.eq(FileSendRecord::getMachineCode,machineFile.getMachineCode())
							.likeRight(FileSendRecord::getName,prefix).between(FileSendRecord::getCreateTime,beginTime,endTime).oneOpt();

						if(optFile.isPresent()){
							//确实下发过，比对总段数是否相同
							FileSendRecord sendFile = optFile.get();
							ProgramNameVO sendProgNameVO =  ProgramFileNameParser.parseProgramName(sendFile.getName());
							if(progNameVO.getSegmentCount() != sendProgNameVO.getSegmentCount()){
								//段数不匹配
								machineFile.setExceptionType(MachineFile.EXCEPTION_NOT_MATCH_SEND);
							}else{
								machineFile.setExceptionType(MachineFile.EXCEPTION_OK);
							}
						}
					}
				}
			}

			machineFileService.updateById(machineFile);
		}
	}
}