package com.fr.plugin.excel;

import com.fr.decision.webservice.bean.security.WatermarkBean;
import com.fr.io.exporter.AbstractAppExporter;
import com.fr.io.exporter.AppExporter;
import com.fr.log.FineLoggerFactory;
import com.fr.main.workbook.ResultWorkBook;
import com.fr.plugin.ws.wapper.WsUtils;
import com.fr.third.v2.org.apache.poi.POIXMLDocumentPart;
import com.fr.third.v2.org.apache.poi.openxml4j.opc.PackagePartName;
import com.fr.third.v2.org.apache.poi.openxml4j.opc.PackageRelationship;
import com.fr.third.v2.org.apache.poi.openxml4j.opc.TargetMode;
import com.fr.third.v2.org.apache.poi.ss.usermodel.Workbook;
import com.fr.third.v2.org.apache.poi.xssf.streaming.SXSSFWorkbook;
import com.fr.third.v2.org.apache.poi.xssf.usermodel.XSSFRelation;
import com.fr.third.v2.org.apache.poi.xssf.usermodel.XSSFSheet;
import com.fr.third.v2.org.apache.poi.xssf.usermodel.XSSFWorkbook;


import java.io.*;

public class ExAppExporter extends AbstractAppExporter {
    AppExporter realExporter;
    WatermarkBean parsedWatermarkBean;


    public ExAppExporter(AppExporter realExporter, WatermarkBean parsedWatermarkBean) {
        this.realExporter = realExporter;
        this.parsedWatermarkBean = parsedWatermarkBean;
    }

    //    @Override
//    public void export(OutputStream orginOutputStream, ResultWorkBook resultWorkBook) throws Exception {
//        //先用真的导出器把resultWorkBook 导出成excel 我们获取二进制再加工
//        ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream(4096);
//        File tempFile = File.createTempFile("ExportExcel", ".xlsx");
//        FileOutputStream fileOutputStream = new FileOutputStream(tempFile);
//        realExporter.export(fileOutputStream, resultWorkBook);
//        byte[] bytes = byteArrayOutputStream.toByteArray();
//        ByteArrayInputStream in = new ByteArrayInputStream(bytes);
//        //加水印
//        XSSFWorkbook workbook = new XSSFWorkbook(in);
//        byte[] watterMarker = WsUtils.getWatterMarker(parsedWatermarkBean);
//        int pictureIdx = workbook.addPicture(watterMarker, Workbook.PICTURE_TYPE_PNG);
//        POIXMLDocumentPart poixmlDocumentPart = workbook.getAllPictures().get(pictureIdx);
//        for (int i = 0; i < workbook.getNumberOfSheets(); i++) {//获取每个Sheet表
//            XSSFSheet sheet = workbook.getSheetAt(i);
//            PackagePartName ppn = poixmlDocumentPart.getPackagePart().getPartName();
//            String relType = XSSFRelation.IMAGES.getRelation();
//            PackageRelationship pr = sheet.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
//            sheet.getCTWorksheet().addNewPicture().setId(pr.getId());
//        }
//        workbook.write(orginOutputStream);
//        FineLoggerFactory.getLogger().info("excel导出加水印成功");
//    }

    @Override
    public void export(OutputStream orginOutputStream, ResultWorkBook resultWorkBook) throws Exception {
        //先存到临时文件
        File tempFile = File.createTempFile("ExportExcel", ".xlsx");
        FileOutputStream fileOutputStream = new FileOutputStream(tempFile);
        realExporter.export(fileOutputStream, resultWorkBook);
        //用SXSSFWorkbook 读取出来处理
        SXSSFWorkbook workbook = new   SXSSFWorkbook(new XSSFWorkbook(tempFile));
        //获取XSSFWorkbook来加水印
        XSSFWorkbook xssfWorkbook = workbook.getXSSFWorkbook();
        byte[] watterMarker = WsUtils.getWatterMarker(parsedWatermarkBean);
        int pictureIdx = workbook.addPicture(watterMarker, Workbook.PICTURE_TYPE_PNG);
        POIXMLDocumentPart poixmlDocumentPart = xssfWorkbook.getAllPictures().get(pictureIdx);
        for (int i = 0; i < workbook.getNumberOfSheets(); i++) {//获取每个Sheet表
            XSSFSheet sheet = xssfWorkbook.getSheetAt(i);
            PackagePartName ppn = poixmlDocumentPart.getPackagePart().getPartName();
            String relType = XSSFRelation.IMAGES.getRelation();
            PackageRelationship pr = sheet.getPackagePart().addRelationship(ppn, TargetMode.INTERNAL, relType, null);
            sheet.getCTWorksheet().addNewPicture().setId(pr.getId());
        }
        //写回原始导出流
        workbook.write(orginOutputStream);
        workbook.dispose();
        workbook.close();
        FineLoggerFactory.getLogger().info("--excel导出加水印成功");
        //删除临时文件
        if (tempFile.exists() &&tempFile.canWrite()) {
            tempFile.delete();
        }
    }


}