package com.ruoyi.utils.excel;

import com.alibaba.excel.EasyExcel;
import com.ruoyi.common.config.ARDConfig;
import com.ruoyi.common.constant.Constants;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

/**
 * @ClassName: TestEasyExcel
 * @Description:
 * @Author: 刘苏义
 * @Date: 2023年04月28日14:09
 * @Version: 1.0
 **/
public class EasyExcelUtil {


    private static void main(String[] args) throws Exception {
        //实现excel写的操作
        // 设置写入文件夹地址和excel文件名
        String fileName = "G:\\java.xlsx";
        // 这里 需要指定写用哪个class去写，然后写到第一个sheet，名字为模板 然后文件流会自动关闭
        //write方法两个参数 ，第一个参数文件路径名称，第二个参数实体类class
        EasyExcel.write(fileName, ExcelData.class).sheet("第一个sheet").doWrite(data("11"));
    }

    //循环设置要添加的数据，最终封装到list集合中
    private static List<ExcelData> data(String data) {
        List<ExcelData> list = new ArrayList<ExcelData>();
        ExcelData excelData = new ExcelData();
        excelData.setName(data);
        list.add(excelData);
        return list;
    }

    public static void writeAppend(String fileName, String msg) {
        File file = new File(fileName);
        File parentDir = file.getParentFile();
        if (!parentDir.exists()) {
            parentDir.mkdirs(); // 创建文件所在的目录，包括父目录
        }
        // 本地临时录像地址
        String tempExcel = ARDConfig.getProfile() + Constants.LOCAL_EXCEL_TEMP_PREFIX + "/temp.xlsx";
        File tempFile = new File(tempExcel);
        List<ExcelData> existingData = new ArrayList<>();
        if (file.exists()) {
            // 读取已有数据
            existingData = EasyExcel.read(file).head(ExcelData.class).sheet().doReadSync();
            // 去除重复数据
            boolean exist = existingData.stream().anyMatch(excelData -> excelData.getName().equals(msg));
            if (exist) {
                return;
            }
        }
        if (!existingData.isEmpty()) {
            // 第二次按照原有格式，不需要表头，追加写入
            EasyExcel.write(file, ExcelData.class).needHead(false)
                    .withTemplate(file).file(tempFile).sheet().doWrite(data(msg));
        } else {
            // 第一次写入需要表头
            EasyExcel.write(file, ExcelData.class).sheet("井不存在").doWrite(data(msg));
        }

        if (tempFile.exists()) {
            file.delete();
            tempFile.renameTo(file);
        }
    }

}