package cn.stylefeng.guns.modular.business.controller;

import cn.hutool.core.io.IoUtil;
import cn.hutool.core.io.resource.ResourceUtil;
import cn.stylefeng.roses.kernel.rule.enums.ResBizTypeEnum;
import cn.stylefeng.roses.kernel.scanner.api.annotation.ApiResource;
import cn.stylefeng.roses.kernel.scanner.api.annotation.GetResource;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.net.URLEncoder;

/**
 * 库存管理管理类
 * @author guo
 */
@RestController
@Api(tags = "导出管理类")
@ApiResource(name = "导出管理类", resBizType = ResBizTypeEnum.BUSINESS)
@RequestMapping("/business")
public class ExcelController {

    /**
     * 下载出库管理-添加货物导入模板
     */
    @ApiOperation("下载导入模板(type:1-入库管理-添加货物导入，2-出库管理-添加货物导入，3-物流费用导入，4-货物管理导入，5-出库管理-导入出库单)")
    @GetResource(name = "下载导入模板(type:1-入库管理-添加货物导入，2-出库管理-添加货物导入，3-物流费用导入，4-货物管理导入，5-出库管理-导入出库单)", path = "/importTemplate")
    public void importTemplate(HttpServletResponse response, Integer type) throws IOException {
        String resource = "";
        if (type == 1) {
            resource = "入库管理-添加货物导入";
        } else if (type == 2) {
            resource = "出库管理-添加货物导入";
        } else if (type == 3) {
            resource = "物流费用导入";
        } else if (type == 4) {
            resource = "货物管理导入";
        }  else if (type == 5) {
            resource = "出库管理-导入出库单";
        } else {
            throw new RuntimeException("类型有误！");
        }
        String fileName = new String(resource.getBytes("gb2312"),"ISO8859-1");
        response.setHeader("Content-Disposition", "attachment;filename="+fileName+ URLEncoder.encode(".xlsx", "ISO8859-1"));
        IoUtil.copy(ResourceUtil.getStream( "excel/" + resource + ".xlsx"), response.getOutputStream());
        response.getOutputStream().close();
    }
}