package com.chinaztt.mes.report.service.impl;

import cn.hutool.core.io.resource.ResourceUtil;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.chinaztt.mes.report.entity.Template;
import com.chinaztt.mes.report.mapper.JasperTemplateMapper;
import com.chinaztt.mes.report.service.JasperReportService;
import lombok.AllArgsConstructor;
import lombok.SneakyThrows;
import net.sf.jasperreports.engine.*;
import net.sf.jasperreports.engine.export.JRXlsExporter;
import net.sf.jasperreports.engine.export.ooxml.JRXlsxExporter;
import net.sf.jasperreports.engine.util.JRLoader;
import net.sf.jasperreports.export.SimpleExporterInput;
import net.sf.jasperreports.export.SimpleHtmlExporterOutput;
import net.sf.jasperreports.export.SimpleOutputStreamExporterOutput;
import net.sf.jasperreports.export.SimpleXlsxReportConfiguration;
import org.apache.commons.io.output.ByteArrayOutputStream;
import org.springframework.stereotype.Service;

import javax.sql.DataSource;
import java.io.FileNotFoundException;
import java.util.Map;

/**
 * @Author: zhangxy
 * @Date: 2021-06-03 11:09
 */
@Service
@AllArgsConstructor
public class JasperReportServiceImpl extends ServiceImpl<JasperTemplateMapper, Template> implements JasperReportService {

	private static final String JASPER_XML_EXT = ".jrxml";

	private DataSource localDataSource;

	@SneakyThrows
	@Override
	public byte[] gen(String templatePath, ReportType type, Map<String, Object> params) {
		JasperReport template = getJasperReport(templatePath);
		if (template == null) {
			throw new RuntimeException(templatePath + "模板不存在");
		}

		JasperPrint jasperPrint = JasperFillManager.fillReport(template, params, localDataSource.getConnection());
		switch (type) {
			case PDF:
				return JasperExportManager.exportReportToPdf(jasperPrint);
			case XLSX:
				JRXlsxExporter exporter = new JRXlsxExporter();
				exporter.setExporterInput(new SimpleExporterInput(jasperPrint));
				ByteArrayOutputStream stream = new ByteArrayOutputStream();
				exporter.setExporterOutput(new SimpleOutputStreamExporterOutput(stream));
				SimpleXlsxReportConfiguration configuration = new SimpleXlsxReportConfiguration();
				configuration.setDetectCellType(true);
				configuration.setCollapseRowSpan(false);
				exporter.setConfiguration(configuration);
				exporter.exportReport();
				return stream.toByteArray();
			default:
				throw new RuntimeException("没有找到对应的导出类型");

		}

	}


	private static JasperReport getJasperReport(String templatePath) {
		if (StringUtils.isNotBlank(templatePath)) {
			if (templatePath.endsWith(JASPER_XML_EXT)) {
				return getJasperReportFromXml(templatePath);
			} else {
				return getJasperReportFromJasper(templatePath);
			}
		}
		return null;
	}

	/**
	 * 获取导出对象,从 xml 中
	 *
	 * @param templatePath 模板路径(base classpath)
	 * @return
	 * @throws FileNotFoundException
	 * @throws JRException
	 */
	@SneakyThrows
	private static JasperReport getJasperReportFromXml(String templatePath) {
		return JasperCompileManager.compileReport(ResourceUtil.getStream(templatePath));
	}


	/**
	 * 获取导出对象,从 jasper 中
	 * (jasper 为 jrxml 编译后生成的文件)
	 *
	 * @param templatePath 模板路径(base classpath)
	 * @return
	 * @throws FileNotFoundException
	 * @throws JRException
	 */
	@SneakyThrows
	private static JasperReport getJasperReportFromJasper(String templatePath) {
		return (JasperReport) JRLoader.loadObject(ResourceUtil.getStream(templatePath));
	}

}