package com.yuanchu.mom.service.impl;

import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.deepoove.poi.XWPFTemplate;
import com.deepoove.poi.config.Configure;
import com.deepoove.poi.data.Pictures;
import com.yuanchu.mom.dto.FeTempHumDateDto;
import com.yuanchu.mom.dto.FeTempHumRecordDto;
import com.yuanchu.mom.pojo.FeTempHumDate;
import com.yuanchu.mom.mapper.FeTempHumDateMapper;
import com.yuanchu.mom.pojo.FeTempHumRecord;
import com.yuanchu.mom.service.FeTempHumDateService;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.yuanchu.mom.service.FeTempHumRecordService;
import com.yuanchu.mom.utils.HackLoopTableRenderPolicy;
import com.yuanchu.mom.utils.DateImageUtil;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletResponse;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URLEncoder;
import java.time.LocalDateTime;
import java.time.Month;
import java.time.format.DateTimeFormatter;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <p>
 * 设施和环境条件-设施和环境条件要求-温湿度 区域 -父 服务实现类
 * </p>
 *
 * @author
 * @since 2024-11-09 11:02:18
 */
@Service
public class FeTempHumDateServiceImpl extends ServiceImpl<FeTempHumDateMapper, FeTempHumDate> implements FeTempHumDateService {

    @Autowired
    private FeTempHumRecordService feTempHumRecordService;

    @Override
    public IPage<FeTempHumDateDto> getFeTempHumDate(Page page) {
        return baseMapper.getFeTempHumDate(page);
    }

    @Override
    public void exportTemperatureAndHumidityRecords(Integer dateId, HttpServletResponse response) {
        FeTempHumDate feTempHumDate = baseMapper.selectById(dateId);
        IPage<FeTempHumRecordDto> list = feTempHumRecordService.getFeTempHumRecordPage(new Page<>(1, -1), dateId);
        DateTimeFormatter formatter = DateTimeFormatter.ofPattern("yyyy-MM-dd HH:mm:ss");
        list.getRecords().forEach(i -> {
            i.setMonth(i.getMorningTestTime().getMonth().getValue());
            i.setMorningTestTimeStr(i.getMorningTestTime().format(formatter));
            i.setAfternoonTimeStr(i.getAfternoonTime().format(formatter));
        });
        // 获取路径
        InputStream inputStream = this.getClass().getResourceAsStream("/static/temp_hum_date.docx");
        Configure configure = Configure.builder()
                .bind("recordList", new HackLoopTableRenderPolicy())
                .build();

        HashMap<String, Object> map = new HashMap<>();
        map.put("testAreaName", feTempHumDate.getTestAreaName());
        map.put("experimentalYear", feTempHumDate.getMonthDate().getYear());
        map.put("experimentalMonth", feTempHumDate.getMonthDate().getMonth().getValue());

        XWPFTemplate template = XWPFTemplate.compile(inputStream, configure).render(
                new HashMap<String, Object>() {{
                    put("feTempHum", map);
                    put("recordList", list.getRecords());
                }});

        try {
            response.setContentType("application/msword");
            String fileName = URLEncoder.encode(
                    "温湿度记录导出", "UTF-8");
            response.setHeader("Content-disposition",
                    "attachment;filename=" + fileName + ".docx");
            OutputStream os = response.getOutputStream();
            template.write(os);
            os.flush();
            os.close();
        } catch (Exception e) {
            e.printStackTrace();
            throw new RuntimeException("导出失败");
        }
    }
}