package org.jeecg.modules.arj.controller;


import com.alibaba.fastjson.JSONObject;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import org.jeecg.modules.arj.config.AjaxResult;
import org.jeecg.modules.arj.entity.Head;
import org.jeecg.modules.arj.mapper.HeadMapper;
import org.jeecg.modules.arj.service.DabaoRdService;
import org.jeecg.modules.arj.service.DabaoZhiliangRdService;
import org.jeecg.modules.arj.service.HeadService;
import org.jeecg.modules.arj.vo.DabaoVo;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

@Api(tags = "A6打包接口")
@RestController
@RequestMapping("/dabaoVo")
public class DabaoVoController {

    @Autowired
    private HeadService headService;

    @Autowired
    private HeadMapper headMapper;
    @Autowired
    private DabaoRdService dabaoRdService;
    @Autowired
    private DabaoZhiliangRdService dabaoZhiliangRdService;


    /**
     * 通过ID查询单条数据
     *
     * @param id 主键
     * @return 实例对象
     */
    @ApiOperation("通过ID查询单条数据")
    @GetMapping("/id")
    public AjaxResult queryById(String id) {
        DabaoVo hanjiVo = new DabaoVo();
        Head head = headService.queryById(id);

        if (head == null) return AjaxResult.error("访问无数据");
        hanjiVo.setHead(head);

        hanjiVo.setDabaoRd(dabaoRdService.queryByHeadId(head.getId()));
        hanjiVo.setDabaoZhiliangRdList(dabaoZhiliangRdService.queryByHeadId(head.getId()));

        return AjaxResult.success(hanjiVo);
    }


    /**
     * 获取最新数据
     *
     * @return 实例对象
     */
    @ApiOperation("获取最新数据")
    @PostMapping("/last")
    public AjaxResult getlastRecord(@RequestBody  String leixing) {

        JSONObject j = JSONObject.parseObject(leixing);
        HashMap<String, String > myMap  = new HashMap<String, String>(){{
            put("0","一");
            put("2","二");
            put("3","三");
            put("4","四");
            put("5","五");
            put("6","六");
            put("7","七");
            put("8","八");
            put("9","九");
        }};

        Head head =  headService.queryLeixingChanxian("A6",myMap.get(j.getString("leixing")));

        if (head == null) return AjaxResult.error("访问无数据");
        DabaoVo hanjiVo = new DabaoVo();
        hanjiVo.setHead(head);
        hanjiVo.setDabaoRd(dabaoRdService.queryByHeadId(head.getId()));
        hanjiVo.setDabaoZhiliangRdList(dabaoZhiliangRdService.queryByHeadId(head.getId()));
        return AjaxResult.success(hanjiVo);
    }


    /**
     * 新增数据
     *
     * @param fanbianVo 实例对象
     * @return 实例对象
     */
    @ApiOperation("新增数据")
    @PostMapping
    public AjaxResult add(@RequestBody DabaoVo fanbianVo) {

        Head h = headService.insert(fanbianVo.getHead());


        if (fanbianVo.getDabaoRd() != null) {
            fanbianVo.getDabaoRd().setHeadId(h.getId());
            dabaoRdService.save(fanbianVo.getDabaoRd());
        }
        if (fanbianVo.getDabaoZhiliangRdList() != null) {
            fanbianVo.getDabaoZhiliangRdList().stream().forEach(item -> item.setHeadId(h.getId()));
            dabaoZhiliangRdService.saveBatch(fanbianVo.getDabaoZhiliangRdList());
        }


        return AjaxResult.success(h.getId());
    }

    /**
     * 新增数据
     *
     * @param fanbianVo 实例对象
     * @return 实例对象
     */
    @ApiOperation("新增数据")
    @PostMapping("/add")
    public AjaxResult add2(@RequestBody DabaoVo fanbianVo) {

        Head h = headService.insert(fanbianVo.getHead());


        if (fanbianVo.getDabaoRd() != null) {
            fanbianVo.getDabaoRd().setHeadId(h.getId());
            dabaoRdService.save(fanbianVo.getDabaoRd());
        }
        if (fanbianVo.getDabaoZhiliangRdList() != null) {
            fanbianVo.getDabaoZhiliangRdList().stream().forEach(item -> item.setHeadId(h.getId()));
            dabaoZhiliangRdService.saveBatch(fanbianVo.getDabaoZhiliangRdList());
        }


        return AjaxResult.success(h.getId());
    }

    /**
     * 更新数据
     *
     * @param dabaoVo 实例对象
     * @return 实例对象
     */
    @ApiOperation("更新数据")
    @PostMapping("/edit")
    public AjaxResult edit(@RequestBody DabaoVo dabaoVo) {
        Head h2 = dabaoVo.getHead();
        h2.setId("");
        Head hh = headService.insert(h2);

        if (dabaoVo.getDabaoRd() != null) {
            dabaoVo.getDabaoRd().setHeadId(hh.getId());
            dabaoRdService.saveOrUpdate(dabaoVo.getDabaoRd());
        }
        if (dabaoVo.getDabaoZhiliangRdList() != null) {
            dabaoVo.getDabaoZhiliangRdList().stream().forEach(item -> item.setHeadId(hh.getId()));
            dabaoZhiliangRdService.saveBatch(dabaoVo.getDabaoZhiliangRdList());
        }


        return AjaxResult.success(hh.getId());
    }

    /**
     * 通过主键删除数据
     *
     * @param id 主键
     * @return 是否成功
     */
    @ApiOperation("通过主键删除数据")
    @DeleteMapping
    public AjaxResult deleteById(String id) {
        return AjaxResult.success(headService.deleteById(id));
    }

    @PostMapping("/page")
    public AjaxResult findPage( @RequestParam(defaultValue = "1") Integer pageNum ,  @RequestParam(defaultValue = "10") Integer pageSize) {


        List<Head> headList = headMapper.queryPage2((pageNum - 1) * pageSize, pageSize);
        IPage iPage = new Page();
        List<DabaoVo> records = new ArrayList<>();
        for (Head h : headList) {
            DabaoVo dabaoVo = new DabaoVo();

            Head head = headService.queryById(h.getId());
            if (head == null) return AjaxResult.error("访问无数据");

            dabaoVo.setHead(head);
            dabaoVo.setDabaoRd(dabaoRdService.queryByHeadId(head.getId()));
            dabaoVo.setDabaoZhiliangRdList(dabaoZhiliangRdService.queryByHeadId(head.getId()));

            records.add(dabaoVo);
        }
        int n = headMapper.queryCount();
        iPage.setPages(n % pageSize == 0 ? n / pageSize : n / pageSize + 1);
        iPage.setRecords(records);
        iPage.setTotal(headMapper.queryCount());
        iPage.setSize(pageSize);
        iPage.setCurrent(pageNum);

        return AjaxResult.success(iPage);
    }
}