package org.springblade.mdm.program.service;

import com.baomidou.mybatisplus.core.metadata.IPage;
import lombok.AllArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.compress.archivers.zip.ZipArchiveEntry;
import org.apache.commons.compress.archivers.zip.ZipFile;
import org.apache.commons.compress.utils.SeekableInMemoryByteChannel;
import org.springblade.core.mp.base.BizServiceImpl;
import org.springblade.core.mp.support.Condition;
import org.springblade.core.mp.support.Query;
import org.springblade.core.tool.utils.FileUtil;
import org.springblade.mdm.program.entity.NcProgramExchange;
import org.springblade.mdm.program.mapper.NcProgramExchangeMapper;
import org.springblade.mdm.program.vo.DncSendBackData;
import org.springblade.mdm.utils.CustomBinaryReader;
import org.springblade.mdm.utils.FileExchangeUtil;
import org.springframework.stereotype.Service;
import org.springframework.web.multipart.MultipartFile;

import java.io.*;
import java.nio.file.Files;
import java.nio.file.Path;
import java.nio.file.Paths;
import java.util.ArrayList;
import java.util.Collections;
import java.util.Enumeration;
import java.util.List;

/**
 * 程序交换（dnc导入/导出）
 *
 * @author yangys
 */
@Slf4j
@Service
@AllArgsConstructor
public class NcProgramExchangeService extends BizServiceImpl<NcProgramExchangeMapper, NcProgramExchange> {

	/**
	 * dnc回传文件上传
	 * @param file 文件
	 * @return
	 */
	public List<DncSendBackData> dncSendBackUpload(MultipartFile file) {
		List<DncSendBackData> list ;
		try {
			String fileName = file.getOriginalFilename();
			//InputStream fileInputStream = file.getInputStream();
			InputStream zipFileInputStream = FileExchangeUtil.convertFileToZip(file.getInputStream());
			byte[] bytes = FileUtil.copyToByteArray(zipFileInputStream);
			list = parseDncZipFromByteArray(bytes);
			for(DncSendBackData dncSendBackData:list){
				NcProgramExchange exchange=new NcProgramExchange();
				exchange.setName(dncSendBackData.getProgramName());
				exchange.setStatus(1);//已导入
				this.save(exchange);

			}

		} catch (IOException e) {
			log.error("上传dnc文件失败",e);
			list = Collections.emptyList();
		}
		return list;
	}

	InputStream convertFileToZip(InputStream inputStream) throws IOException {

		File tempFile = createTempFile();
		FileOutputStream fos = new FileOutputStream(tempFile);
		CustomBinaryReader.read(inputStream,fos);


		FileInputStream dInstream = new FileInputStream(tempFile);

		return dInstream;
	}

	/**
	 * 创建一个临时文件
	 * @return
	 * @throws IOException
	 */
	File createTempFile() throws IOException {
		Path tempDir = Paths.get(System.getProperty("java.io.tmpdir"));
		// 在临时目录中创建文件
		String tfilename = "t"+System.currentTimeMillis();
		Path tempFile = Files.createTempFile(tempDir, tfilename, ".tmp");
		System.out.println("创建的临时文件: " + tempFile);
		return tempFile.toFile();
	}
	public static List<DncSendBackData> parseDncZipFromByteArray(byte[] zipData) throws IOException {
		//List<DncSendBackData> datas = new ArrayList<>();
		List<DncSendBackData> datas  = ZipFileDirectoryScanner.getFilesInDirectoryRecursive(zipData, "");



		/*
		try (ByteArrayInputStream bis = new ByteArrayInputStream(zipData);
			 ZipArchiveInputStream zis = new ZipArchiveInputStream(bis)) {


			ZipArchiveEntry entry;
			while ((entry = zis.getNextZipEntry()) != null) {
				DncSendBackData prog = new DncSendBackData();
				prog.setProgramName(entry.getName());
				if (!entry.isDirectory()) {
					System.out.println("文件名: " + entry.getName());
					System.out.println("大小: " + entry.getSize());

					// 读取文件内容到字节数组
					ByteArrayOutputStream outputStream = new ByteArrayOutputStream();
					IOUtils.copy(zis, outputStream);
					byte[] fileContent = outputStream.toByteArray();

					// 处理文件内容...
					System.out.println("内容长度: " + fileContent.length);
				}else{
					//文件夹，读内部文件，获取文件列表

					System.out.println("文件夹程序："+entry.getName());
					List<String> children = new ArrayList<>();
					prog.setChildren(children);
				}

				datas.add(prog);
			}


		}*/
		return datas;
	}
	/**
	 * 程序下发统计分页查询
	 * @param query 查询参数
	 * @return
	 */

	public IPage<DncSendBackData> dncSendBackPageQuery(Query query) {

		IPage<DncSendBackData> page = this.getBaseMapper().dncSendBackpageQuery(Condition.getPage(query),query);

		return page;

	}
}


class ZipFileDirectoryScanner {

	public static List<DncSendBackData> getFilesInDirectoryRecursive(byte[] zipData, String dirPath) throws IOException {
		List<DncSendBackData> list = new ArrayList<>();

		if (!dirPath.endsWith("/")) {
			dirPath += "/";
		}

		try (SeekableInMemoryByteChannel channel = new SeekableInMemoryByteChannel(zipData);
			 ZipFile zipFile = new ZipFile(channel)) {

			Enumeration<ZipArchiveEntry> entries = zipFile.getEntries();
			while (entries.hasMoreElements()) {
				ZipArchiveEntry entry = entries.nextElement();
				String entryName = entry.getName();
				DncSendBackData d = new DncSendBackData();
				d.setProgramName(entryName);
				if(entry.isDirectory()){
					d.setHasChildren(true);
				}
				list.add(d);

				//if (entryName.startsWith(dirPath) && !entry.isDirectory()) {
				//	fileList.add(entryName);
				//}
			}
		}
		return list;
	}
}