package com.product.print.util;

import cn.hutool.extra.spring.SpringUtil;
import com.deepoove.poi.converter.ObjectToPictureRenderDataConverter;
import com.deepoove.poi.converter.ToRenderDataConverter;
import com.deepoove.poi.data.PictureRenderData;
import com.deepoove.poi.data.PictureType;
import com.deepoove.poi.data.Pictures;
import com.deepoove.poi.policy.PictureRenderPolicy;
import com.deepoove.poi.render.RenderContext;
import com.product.core.dao.BaseDao;
import com.product.core.entity.FieldSetEntity;
import com.product.file.service.FileManagerService;

import java.io.File;
import java.io.FileInputStream;
import java.util.Arrays;

/**
 * Copyright © 6c
 *
 * @Date 2023年11月16日 17:19
 * @Author 6c
 * @Description
 */
public class CustomPictureRenderPolicy extends PictureRenderPolicy {

    private static ToRenderDataConverter<Object, PictureRenderData> converter = new ObjectToPictureRenderDataConverter();

    @Override
    public PictureRenderData cast(Object source) throws Exception {
        FileManagerService fileManagerService = SpringUtil.getBean(FileManagerService.class);
        String templateUid = source.toString();
        BaseDao baseDao = SpringUtil.getBean(BaseDao.class);
        FieldSetEntity attachmentFse = baseDao.getFieldSetEntity("product_sys_attachments", templateUid, false);
        fileManagerService.getFile(templateUid);
        String fileName = attachmentFse.getString("file_name");
        String type = fileName.substring(fileName.lastIndexOf(".") + 1);
        File file = fileManagerService.getFile(attachmentFse);
        PictureType imgType = PictureType.PNG;
        if (Arrays.asList("jpg", "jpeg").contains(type)) {
            imgType = PictureType.JPEG;
        } else if ("bmp".equals(type)) {
            imgType = PictureType.BMP;
        } else if ("gif".equals(type)) {
            imgType = PictureType.GIF;
        } else if ("svg".equals(type)) {
            imgType = PictureType.SVG;
        }
        return converter.convert(Pictures.ofStream(new FileInputStream(file), imgType).create());
    }

    @Override
    public void doRender(RenderContext<PictureRenderData> context) throws Exception {
        Helper.renderPicture(context.getRun(), context.getData());
    }
}