package com.ruoyi.project.tr.riskList.controller;

import com.ruoyi.common.utils.DateUtils;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.utils.security.ShiroUtils;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.framework.web.page.TableDataInfo;
import com.ruoyi.project.enumerate.TrRiskTypeEnum;
import com.ruoyi.project.system.user.domain.User;
import com.ruoyi.project.tr.riskCheckPoint.service.IRiskCheckPointService;
import com.ruoyi.project.tr.riskList.domain.RiskList;
import com.ruoyi.project.tr.riskList.service.IRiskListService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;

import java.util.List;

/**
 * 基础清单Controller
 *
 * @author wm
 * @date 2020-05-07
 */
@Controller
@RequestMapping("/tr/baseRisk")
public class BaseRiskController extends BaseController {
    private String prefix = "tr/baseRisk";

    @Autowired
    private IRiskListService riskListService;

    /**
     * 查询基础清单列表
     * @param mmap
     * @return
     */
    @GetMapping("")
    public String riskList( ModelMap mmap) {
        return prefix + "/riskList";
    }

    /**
     * 查询风险清单列表
     */
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(RiskList riskList) {
        startPage();
        riskList.setRiskType(TrRiskTypeEnum.BASE.getCode());
        riskList.setCompanyId(getSysUser().getCompanyId());
        List<RiskList> list = riskListService.selectRiskListList(riskList);

        return getDataTable(list);
    }

    /**
     * 导出风险清单列表
     */
    @Log(title = "风险清单", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ResponseBody
    public AjaxResult export(RiskList riskList) {
        riskList.setRiskType(TrRiskTypeEnum.BASE.getCode());
        riskList.setCompanyId(getSysUser().getCompanyId());
        List<RiskList> list = riskListService.selectRiskListList(riskList);
        ExcelUtil<RiskList> util = new ExcelUtil<RiskList>(RiskList.class);
        return util.exportExcel(list, "riskList");
    }

    /**
     * 新增基础清单
     */
    @GetMapping("/add")
    public String add( ModelMap mmap) {
        return prefix + "/add";
    }


    /**
     * 新增保存风险清单
     */
    @Log(title = "风险清单", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(RiskList riskList) {
        //老代码
        //保存前获取用户名以及公司信息
        User sysUser = getSysUser();
        riskList.setCreateBy(sysUser.getUserName());
        riskList.setCreateTime(DateUtils.getNowDate());
        riskList.setRiskType(TrRiskTypeEnum.BASE.getCode());
        riskList.setCompanyId(getSysUser().getCompanyId());
        return toAjax(riskListService.insertRiskList(riskList));
    }

    /**
     * 修改风险清单
     */
    @GetMapping("/edit/{riskListId}")
    public String edit(@PathVariable("riskListId") Long riskListId, ModelMap mmap) {
        RiskList riskList = riskListService.selectRiskListById(riskListId);
        mmap.put("riskList", riskList);
        //判断如果是作业活动，查询作业活动频率

        return prefix + "/edit";
    }




    /**
     * 修改保存风险清单
     */
    @Log(title = "风险清单", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(RiskList riskList) {
        User sysUser = getSysUser();
        riskList.setUpdateBy(sysUser.getUserName());
        riskList.setUpdateTime(DateUtils.getNowDate());
        return toAjax(riskListService.updateRiskList(riskList));
    }

    /**
     * 删除风险清单
     */
    @Log(title = "风险清单", businessType = BusinessType.DELETE)
    @PostMapping("/remove")
    @ResponseBody
    public AjaxResult remove(String ids) {
        return toAjax(riskListService.deleteRiskListByIds(ids));
    }


    /**
     * 跳转至风险单元选择页面
     * @param riskType
     * @param mmap
     * @return
     */
    @GetMapping("/selectRiskList/{riskType}")
    public String selectRiskList(@PathVariable("riskType") String riskType, ModelMap mmap) {
        mmap.put("riskType", riskType);
        mmap.put("companyId", ShiroUtils.getSysUser().getCompanyId());
        return prefix + "/selectRiskList";
    }




}