package org.jeecg.modules.utils;

import lombok.extern.slf4j.Slf4j;
import org.apache.pdfbox.pdmodel.PDDocument;
import org.apache.pdfbox.rendering.PDFRenderer;
import org.jeecg.common.util.SHA256Util;
import org.jeecg.modules.ai.vo.PhotoListVo;
import org.jeecg.modules.utils.file.FileUtils;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Component;

import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Date;
import java.util.List;

/**
 * @author clown
 * * @date 2024/8/1
 */
@Slf4j
@Component
public class PdfTurnPngUtils {

    private static String filePngPath;

    @Value("${jeecg.path.upload}")
    public void setFilePngPath(String filePngPath) {
        PdfTurnPngUtils.filePngPath = filePngPath;
    }



    /**
     * 使用pdfbox将整个pdf转换成图片
     *
     * @param fileAddress 文件地址 如:C:\\Users\\user\\Desktop\\test
     * @param filename    PDF文件名不带后缀名
     * @param type        图片类型 png 和jpg
     */
    public static void pdf2png(String fileAddress, String filename, String type) {
        long startTime = System.currentTimeMillis();
        // 将文件地址和文件名拼接成路径 注意：线上环境不能使用\\拼接
        File file = new File(fileAddress + "\\" + filename + ".pdf");
        try {
            // 写入文件
            PDDocument doc = PDDocument.load(file);
            PDFRenderer renderer = new PDFRenderer(doc);
            int pageCount = doc.getNumberOfPages();
            for (int i = 0; i < pageCount ; i++) {
                // dpi为144，越高越清晰，转换越慢
                BufferedImage image = renderer.renderImageWithDPI(i, 144); // Windows native DPI
                // 将图片写出到该路径下
                ImageIO.write(image, type,
                        new File(fileAddress + "\\备份\\" + filename + "_" + (i+1) + "." + type));
            }
            long endTime = System.currentTimeMillis();
            System.out.println("共耗时：" + ((endTime - startTime) / 1000.0) + "秒");  //转化用时
        } catch (IOException e) {
            e.printStackTrace();
        }

    }

    public static List<PhotoListVo> pdf2pngList(File filePdf, String type,String pathImg){
        PDDocument doc = null;
        try {
            List<PhotoListVo> files = new ArrayList<>();
            doc = PDDocument.load(filePdf);
            String pdfName = FileUtils.getFilenameNonSuffix(filePdf.getName());
            if (doc == null ) {
                return null;
            }
            PDFRenderer renderer = new PDFRenderer(doc);
            int pageCount = doc.getNumberOfPages();
            Date currentDate = DateUtil.getNow();
            String monthStr = DateUtil.format(currentDate, DateUtil.STR_YEAR_MONTH);
            String relativePath = "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/"+ pathImg + "/";
            //绝对路径
            String absolutePath =  filePngPath + "/" + monthStr + "/" + DateUtil.getDayStr(currentDate) + "/" + pathImg + "/";
            for (int i = 0; i < pageCount ; i++) {
                PhotoListVo pVo = new PhotoListVo();
                // dpi为144，越高越清晰，转换越慢
                BufferedImage image = renderer.renderImageWithDPI(i, 144); // Windows native DPI  144
                // 将图片写出到该路径下
                //文件路径
                //相对路径
                String encodeFileName = SHA256Util.getSHA256Str(pdfName + "_" + (i+1) + System.currentTimeMillis());
                String imgPath = absolutePath + "/" + encodeFileName + "." + type;
                File targetFile = new File(imgPath);
                if(!targetFile.getParentFile().exists()){
                    targetFile.getParentFile().mkdirs();
                }
                ImageIO.write(image, type, targetFile);
                pVo.setImgEncodeName(encodeFileName + "." + type);
                pVo.setImgName(pdfName + "_" + (i+1));
                pVo.setImgSize(Long.valueOf(new File(imgPath).length()));
                pVo.setImgSuffix(type);
                pVo.setPageNumber(i+1);
                pVo.setImgPath(relativePath);
                files.add(pVo);
            }
            return files;
        } catch (IOException e) {
            e.printStackTrace();
        }

        return null;
    }


}