package com.ruoyi.report.service.impl;

import cn.hutool.core.date.DateUtil;
import com.alibaba.excel.EasyExcel;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.ruoyi.report.dto.SampleRecordDto;
import com.ruoyi.report.mapper.SampleRecordMapper;
import com.ruoyi.report.service.SampleRecordService;
import com.ruoyi.report.vo.SampleRecordVo;
import lombok.AllArgsConstructor;
import org.springframework.stereotype.Service;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.*;

/**
 * 样品领样记录服务实现
 */
@Service
@AllArgsConstructor
public class SampleRecordServiceImpl implements SampleRecordService {

    private SampleRecordMapper sampleRecordMapper;

    @Override
    public Page<SampleRecordVo> pageSampleRecord(Page page, SampleRecordDto dto) {
        Page<SampleRecordVo> result = sampleRecordMapper.pageSampleRecord(page, dto);
        // 处理操作类型名称
        List<SampleRecordVo> records = result.getRecords();
        for (SampleRecordVo vo : records) {
            vo.setOperateTypeName(formatOperateType(vo.getOperateType()));
        }
        return result;
    }

    @Override
    public List<SampleRecordVo> getFlowRecord(Long sampleId) {
        List<SampleRecordVo> list = sampleRecordMapper.getFlowRecord(sampleId);
        for (SampleRecordVo vo : list) {
            vo.setOperateTypeName(formatOperateType(vo.getOperateType()));
        }
        return list;
    }

    @Override
    public void exportSampleRecord(SampleRecordDto dto, HttpServletResponse response) {
        try {
            // 查询全部数据
            Page<SampleRecordVo> page = new Page<>();
            page.setSize(Long.MAX_VALUE);
            Page<SampleRecordVo> result = sampleRecordMapper.pageSampleRecord(page, dto);

            // 处理数据
            List<SampleRecordVo> records = result.getRecords();
            for (SampleRecordVo vo : records) {
                vo.setOperateTypeName(formatOperateType(vo.getOperateType()));
            }

            // 设置响应头
            response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
            response.setCharacterEncoding("utf-8");
            String fileName = "样品领样记录_" + DateUtil.format(new Date(), "yyyyMMddHHmmss");
            response.setHeader("Content-disposition", "attachment;filename=" + fileName + ".xlsx");

            // 使用EasyExcel导出
            EasyExcel.write(response.getOutputStream(), SampleRecordVo.class)
                    .sheet("样品领样记录")
                    .doWrite(records);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    /**
     * 格式化操作类型
     */
    private String formatOperateType(String val) {
        if (val == null) return "";
        Map<String, String> map = new HashMap<>();
        map.put("in", "入库");
        map.put("out", "出库");
        map.put("move", "移库");
        map.put("receive", "领用");
        map.put("return", "归还");
        return map.getOrDefault(val, val);
    }

}