package com.wms_admin.server.controller;

import com.alibaba.excel.EasyExcel;
import com.alibaba.excel.write.style.HorizontalCellStyleStrategy;
import com.alibaba.fastjson.JSON;
import com.alibaba.fastjson2.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.core.toolkit.IdWorker;
import com.baomidou.mybatisplus.core.toolkit.ObjectUtils;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.wms_admin.server.entity.Product;
import com.wms_admin.server.service.ProductService;
import com.wms_admin.utils.AddProductMessageVo;
import com.wms_admin.utils.RedisUtil;
import com.wms_admin.utils.Result;
import com.wms_admin.utils.StyleUtils;
import io.swagger.annotations.*;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.io.*;
import java.util.Arrays;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * <p>
 *  前端控制器
 * </p>
 *
 * @author 江苏鵷雏网络科技有限公司
 * @since 2023-05-24
 */
@Slf4j
@Api(tags = "入库操作")
@RestController
@RequestMapping("/product")
public class ProductController {
    @Autowired
    private ProductService service;

    @ApiOperation(value = "扫码入库添加接口：swagger json格式可以测试")
    @PostMapping("/add")
    public Result<Map<String, Object>> AddProductMessage(@RequestBody AddProductMessageVo map){
        for (String i : map.getCode()){
            Object andDelete = RedisUtil.getAndDelete(i);
            Product product = JSONObject.parseObject(JSONObject.toJSONString(andDelete), Product.class);
            product.setAddPerson(map.getAddPerson());
            product.setId(i);
            service.AddProductMessage(product);
        }
        return Result.success("添加成功");
    }

    @ApiOperation(value = "条码信息缓存：不与数据库关联所以需要传入产品名称与规格型号，不能是ID")
    @ApiImplicitParams(value = {
            @ApiImplicitParam(name = "productName", value = "产品名称", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "productModel", value = "规格型号", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "unit", value = "单位", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "incomingQuantity", value = "入库数量", dataTypeClass = String.class, required = true),
            @ApiImplicitParam(name = "productCode", value = "产品编码", dataTypeClass = String.class, required = true)
    })
    @PostMapping("/cache")
    public Result<String> AddProductCache(String productName, String productModel, String unit, String incomingQuantity, String productCode){
        Map<String, Object> map = new HashMap<>();
        map.put("productName", productName);
        map.put("productModel", productModel);
        map.put("unit", unit);
        map.put("incomingQuantity", incomingQuantity);
        map.put("productCode", productCode);
        String idStr = IdWorker.getIdStr();
        // 设置缓存七天后过期
        boolean set = RedisUtil.set(idStr, map, 168);
        if (set){
            return Result.success(idStr);
        } else {
            return Result.fail("条码生成失败，请重新尝试！");
        }
    }

    @ApiOperation(value = "根据条码查询缓存信息")
    @ApiImplicitParams(value = {
            @ApiImplicitParam(name = "code", value = "产品编码", dataTypeClass = String.class, required = true)
    })
    @GetMapping("/code_select")
    public Result<Map<String, Object>> AddProductCodeSelect(String code){
        Object o = RedisUtil.get(code);
        if (!ObjectUtils.isEmpty(o)){
            Map<String, Object> map = JSONObject.parseObject(JSONObject.toJSONString(o));
            return Result.success(map);
        } else {
            return Result.fail("条形码已过期，请重新打印！");
        }
    }

    @ApiImplicitParams(value = {
            @ApiImplicitParam(name = "startTime", value = "开始时间", dataTypeClass = String.class, required = false),
            @ApiImplicitParam(name = "endTime", value = "结束时间", dataTypeClass = String.class, required = false),
            @ApiImplicitParam(name = "productModel", value = "型号", dataTypeClass = String.class, required = false),
            @ApiImplicitParam(name = "pageNo", value = "起始页", dataTypeClass = Integer.class, required = true),
            @ApiImplicitParam(name = "pageSize", value = "页总数", dataTypeClass = Integer.class, required = true)
    })
    @ApiOperation(value = "分页查询接口")
    @GetMapping("/list")
    public Result<Map<String, Object>> ListProductMessage(String startTime, String  endTime, String productModel, Integer pageNo, Integer pageSize) {
        IPage<Product> productIPage = service.ListProductMessage(startTime, endTime, productModel, new Page(pageNo, pageSize));
        HashMap<String, Object> map = new HashMap<>();
        map.put("row", productIPage.getRecords());
        map.put("total", productIPage.getTotal());
        return Result.success(map);
    }

    @ApiOperation(value = "入库Excel导出")
    @GetMapping("/excel")
    @ApiImplicitParams(value = {
            @ApiImplicitParam(name = "startTime",value = "开始时间", dataTypeClass = String.class,required = false),
            @ApiImplicitParam(name = "endTime",value = "结束时间", dataTypeClass = String.class,required = false),
            @ApiImplicitParam(name = "productModel",value = "设备型号", dataTypeClass = String.class,required = false)
    })
    public void ExcelDerive(String startTime, String  endTime, String productModel, HttpServletResponse response) throws IOException {
        List<Product> map = service.ExcelDerive(startTime, endTime, productModel);
        // 设置单元格样式
        HorizontalCellStyleStrategy horizontalCellStyleStrategy = new HorizontalCellStyleStrategy(StyleUtils.getHeadStyle(), StyleUtils.getContentStyle());
        // 保存到第一个sheet中
        response.setContentType("application/vnd.openxmlformats-officedocument.spreadsheetml.sheet");
        response.setHeader("requestType","excel");
        response.setHeader("Access-Control-Expose-Headers", "requestType");
        EasyExcel
                .write(response.getOutputStream(), Product.class)
                .sheet()
                .needHead(true)
                .registerWriteHandler(horizontalCellStyleStrategy)
                .doWrite(map);
    }
}