package com.thhy.materials.modules.biz.materials.service.impl;

import cn.hutool.core.io.IoUtil;
import cn.hutool.poi.excel.ExcelWriter;
import com.github.pagehelper.PageHelper;
import com.github.pagehelper.PageInfo;
import com.github.pagehelper.util.StringUtil;
import com.thhy.general.common.BasicResult;
import com.thhy.general.config.SysUserInfo;
import com.thhy.general.utils.UserInfoUtils;
import com.thhy.materials.modules.biz.materials.entity.SysAssistGood;
import com.thhy.materials.modules.biz.materials.entity.dto.AssistDto;
import com.thhy.materials.modules.biz.materials.mapper.SysAssistGoodMapper;
import com.thhy.materials.modules.biz.materials.service.SysAssistGoodService;
import org.springframework.stereotype.Service;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

@Service
public class SysAssistGoodServiceImpl implements SysAssistGoodService {

    @Resource
    private SysAssistGoodMapper assistGoodMapper;


    @Override
    public BasicResult assistInsert(Map<String, Object> values) {
        SysUserInfo sysUserInfo = UserInfoUtils.getInstance().getUserInfo();
        values.put("companyId",sysUserInfo.getCompanyId());

        String assistName = values.get("assistName").toString();
        String assistType = values.get("assistModel").toString();
        String id =  assistGoodMapper.assistNameAndType(assistName,assistType);
        if(!StringUtil.isEmpty(id)){
            return BasicResult.faild("11111","error","名称和规格型号不可重复");
        }
        values.put("id","");
        assistGoodMapper.assistInsert(values);
        return BasicResult.success("添加成功");
    }

    @Override
    public BasicResult assistList(Map<String, Object> values) {
        SysUserInfo sysUserInfo = UserInfoUtils.getInstance().getUserInfo();
        String companyId = sysUserInfo.getCompanyId();
        values.put("companyId",companyId);

        Integer pageSize = Integer.valueOf(values.get("pageSize").toString());
        Integer pageNum = Integer.valueOf(values.get("pageNum").toString());
        PageHelper.startPage(pageNum,pageSize);
        List<SysAssistGood> assistGoods = assistGoodMapper.assistList(values);
        PageInfo<SysAssistGood> sysAssistGoodPageInfo = new PageInfo<>(assistGoods);
        return BasicResult.success(sysAssistGoodPageInfo);
    }

    @Override
    public BasicResult assistInfo(String assistId) {
        SysAssistGood assistGoods = assistGoodMapper.assistInfo(assistId);
        return BasicResult.success(assistGoods);
    }

    @Override
    public BasicResult assistDel(String assistId) {
        List<String> ids = assistGoodMapper.assistGood(assistId);
        if(ids!=null && ids.size()>0){
            return BasicResult.faild("11111","该类型已被占用","error");
        }
        assistGoodMapper.assistDel(assistId);
        return BasicResult.success("删除成功");
    }

    @Override
    public BasicResult assistUpdate(Map<String, Object> values) {
        String assistName = values.get("assistName").toString();
        String assistType = values.get("assistModel").toString();
        String assistId = values.get("id").toString();
        String id =  assistGoodMapper.assistNameAndType(assistName,assistType);
        if(!StringUtil.isEmpty(id)&&!assistId.equals(id)){
            return BasicResult.faild("11111","error","名称和规格型号不可重复");
        }
        assistGoodMapper.assistUpdate(values);
        return BasicResult.success("修改成功");
    }

    @Override
    public void assistExport(Map<String, Object> values, HttpServletResponse response) {
        SysUserInfo sysUserInfo = UserInfoUtils.getInstance().getUserInfo();
        String companyId = sysUserInfo.getCompanyId();
        values.put("companyId",companyId);

        List<SysAssistGood> assistGoods = assistGoodMapper.assistList(values);

        List<AssistDto> assistDtos = new ArrayList<>();
        for (SysAssistGood assistGood : assistGoods){
            AssistDto assistDto = new AssistDto();
            assistDto.setAssistName(assistGood.getAssistName());
            assistDto.setAssistModel(assistGood.getAssistModel());
            assistDto.setUnit(assistGood.getUnit());
            assistDto.setStock(assistGood.getStock());
            assistDto.setAlarmCount(assistGood.getAlarmCount());
            assistDto.setPosition(assistGood.getPosition());
            assistDto.setDictName(assistGood.getDictName());
            assistDtos.add(assistDto);
        }
        ExcelWriter excel = new ExcelWriter();//定义excel
        excel.addHeaderAlias("assistName", "物品名称");
        excel.addHeaderAlias("assistModel", "规格型号");
        excel.addHeaderAlias("dictName", "类别");
        excel.addHeaderAlias("unit", "计量单位");
        excel.addHeaderAlias("stock", "库存个数");
        excel.addHeaderAlias("alarmCount", "报警数");
        excel.addHeaderAlias("position", "位置");
        response.setContentType("application/vnd.ms-excel;charset=UTF-8");
        response.setHeader("Content-Disposition", "attachment;filename=test.xls");
        OutputStream out = null;
        try {
            out = response.getOutputStream();
            excel.write(assistDtos);
            excel.flush(out, true);
        } catch (IOException e) {
            e.printStackTrace();
        } finally {
            // 关闭writer，释放内存
            excel.close();
        }
        IoUtil.close(out);
    }

}