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

import java.util.List;
import java.util.Map;
import java.util.Random;
import java.util.UUID;

import com.ruoyi.common.utils.security.ShiroUtils;
import com.ruoyi.doublePrevention.entity.PreventRiskEvent;
import com.ruoyi.doublePrevention.entity.dto.req.*;
import com.ruoyi.doublePrevention.entity.dto.resp.PreventRiskEventPageQueryRespDTO;
import com.ruoyi.doublePrevention.entity.dto.resp.PreventRiskEventRespDTO;
import com.ruoyi.doublePrevention.entity.dto.resp.PreventRiskUnitCodeAndNameListQueryRespDTO;
import com.ruoyi.doublePrevention.service.RiskService;
import com.ruoyi.doublePrevention.vo.ResultVO;
import com.ruoyi.framework.web.page.PageDomain;
import com.ruoyi.framework.web.page.TableSupport;
import com.ruoyi.project.system.user.domain.User;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.*;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.project.tr.riskType.domain.RiskType;
import com.ruoyi.project.tr.riskType.service.IRiskTypeService;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.framework.web.page.TableDataInfo;

/**
 * 风险单元类型Controller
 *
 * @author wm
 * @date 2020-05-05
 */
@Controller
@RequestMapping("/tr/riskEvent")
public class RiskEventController extends BaseController
{

    private String prefix = "tr/riskEvent";

    @Autowired
    private IRiskTypeService riskTypeService;

    @Autowired
    private RiskService riskService;

    @GetMapping()
    public String riskEvent(ModelMap mmap)
    {
        List<PreventRiskUnitCodeAndNameListQueryRespDTO> respDTOS = riskService.listRiskUnitCodeAndName();
        mmap.put("riskUnits", respDTOS);
        return prefix + "/riskEvent";
    }

    /**
     * 查询安全风险事件
     */
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(PreventRiskEventPageQueryReqDTO riskEventQueryReqDTO)
    {
        ResultVO<List<PreventRiskEventPageQueryRespDTO>> resultVO = riskService.listRiskEventPage(riskEventQueryReqDTO);
        List<PreventRiskEventPageQueryRespDTO> data = (List<PreventRiskEventPageQueryRespDTO>) resultVO.getData();

        //只查询属于该公司下的
//        riskType.setCompanyId(ShiroUtils.getSysUser().getCompanyId());
//        startPage();
//        List<RiskType> list = riskTypeService.selectRiskTypeList(riskType);
//        return getDataTable(list);
        TableDataInfo dataTable = getDataTable(data);
        dataTable.setTotal(resultVO.getCount());
        return dataTable;
    }

    /**
     * 导出风险单元类型列表
     */
    @Log(title = "风险单元类型", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ResponseBody
    public AjaxResult export(RiskType riskType)
    {
        List<RiskType> list = riskTypeService.selectRiskTypeList(riskType);
        ExcelUtil<RiskType> util = new ExcelUtil<RiskType>(RiskType.class);
        return util.exportExcel(list, "riskType");
    }

    /**
     * 新增风险单元类型
     */
    @GetMapping("/add")
    public String add(ModelMap mmap)
    {
        List<PreventRiskUnitCodeAndNameListQueryRespDTO> respDTOS = riskService.listRiskUnitCodeAndName();
        mmap.put("riskUnits", respDTOS);
        return prefix + "/add";
    }

    /**
     * 新增保存风险单元类型
     */
    @Log(title = "风险单元类型", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(PreventRiskEventSaveReqDTO riskEventSaveReqDTO)
    {
        ResultVO<PreventRiskEvent> resultVO = riskService.saveRiskEvent(riskEventSaveReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
//        User user = ShiroUtils.getSysUser();
//        riskType.setCompanyId(user.getCompanyId());
//        riskType.setCreateBy(user.getUserName());
//        return toAjax(riskTypeService.insertRiskType(riskType));
    }

    /**
     * 修改风险单元类型
     */
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Long id, ModelMap mmap)
    {
        PreventRiskEventGetReqDTO riskEventGetReqDTO = new PreventRiskEventGetReqDTO();
        riskEventGetReqDTO.setId(id);
        ResultVO<PreventRiskEventRespDTO> riskEventById = riskService.getRiskEventById(riskEventGetReqDTO);
        Object data = riskEventById.getData();
//        RiskType riskType = riskTypeService.selectRiskTypeById(id);
        mmap.put("riskType", data);
//        System.out.println(prefix + "/edit");
        List<PreventRiskUnitCodeAndNameListQueryRespDTO> respDTOS = riskService.listRiskUnitCodeAndName();
        mmap.put("riskUnits", respDTOS);
        return prefix + "/edit";
    }

    /**
     * 修改保存风险单元类型
     */
    @Log(title = "风险单元类型", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(PreventRiskEventUpdateReqDTO riskEventUpdateReqDTO)
    {
        ResultVO<PreventRiskEvent> resultVO = riskService.updateRiskEvent(riskEventUpdateReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
//        riskType.setUpdateBy(ShiroUtils.getSysUser().getUserName());
//        return toAjax(riskTypeService.updateRiskType(riskType));
    }

    /**
     * 删除风险单元类型
     */
    @Log(title = "风险单元类型", businessType = BusinessType.DELETE)
    @PostMapping( "/remove")
    @ResponseBody
    public AjaxResult remove(Long ids)
    {
        PreventRiskEventDeleteReqDTO riskEventDeleteReqDTO = new PreventRiskEventDeleteReqDTO();
        riskEventDeleteReqDTO.setId(ids);
        ResultVO<PreventRiskEvent> resultVO = riskService.deleteRiskEvent(riskEventDeleteReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
    }




    /**
     * 跳转至设备设施类型
     * @return
     */
    @GetMapping("/riskTypeByDevice")
    public String riskTypeByDevice()
    {
        return prefix + "/riskTypeByDevice";
    }

    /**
     * 新增风险单元类型
     */
    @GetMapping("/addByDevice")
    public String addByDevice()
    {
        return prefix + "/addByDevice";
    }

    @GetMapping("/editByDevice/{riskTypeId}")
    public String editByDevice(@PathVariable("riskTypeId") Long riskTypeId, ModelMap mmap)
    {
        RiskType riskType = riskTypeService.selectRiskTypeById(riskTypeId);
        mmap.put("riskType", riskType);
        return prefix + "/editByDevice";
    }


    /**
     * 跳转至作业活动类型
     * @return
     */
    @GetMapping("/riskTypeByWork")
    public String riskTypeByWork()
    {
        return prefix + "/riskTypeByWork";
    }

    /**
     * 新增风险单元类型
     */
    @GetMapping("/addByWork")
    public String addByWork()
    {
        return prefix + "/addByWork";
    }

    @GetMapping("/editByWork/{riskTypeId}")
    public String editByWork(@PathVariable("riskTypeId") Long riskTypeId, ModelMap mmap)
    {
        RiskType riskType = riskTypeService.selectRiskTypeById(riskTypeId);
        mmap.put("riskType", riskType);
        return prefix + "/editByWork";
    }


    /**
     * 跳转至工艺节点类型
     * @return
     */
    @GetMapping("/riskTypeByCrafts")
    public String riskTypeByCrafts(){
        return prefix + "/riskTypeByCrafts";
    }

    /**
     * 新增风险单元类型
     */
    @GetMapping("/addByCrafts")
    public String addByCrafts()
    {
        return prefix + "/addByCrafts";
    }

    @GetMapping("/editByCrafts/{riskTypeId}")
    public String editByCrafts(@PathVariable("riskTypeId") Long riskTypeId, ModelMap mmap)
    {
        RiskType riskType = riskTypeService.selectRiskTypeById(riskTypeId);
        mmap.put("riskType", riskType);
        return prefix + "/editByCrafts";
    }

    /**
     * 跳转至作业频率类型
     * @return
     */
    @GetMapping("/riskTypeByFrequency")
    public String riskTypeByFrequency()
    {
        return prefix + "/riskTypeByFrequency";
    }

    /**
     * 新增风险单元类型
     */
    @GetMapping("/addByFrequency")
    public String addByFrequency()
    {
        return prefix + "/addByFrequency";
    }

    @GetMapping("/editByFrequency/{riskTypeId}")
    public String editByFrequency(@PathVariable("riskTypeId") Long riskTypeId, ModelMap mmap)
    {
        RiskType riskType = riskTypeService.selectRiskTypeById(riskTypeId);
        mmap.put("riskType", riskType);
        return prefix + "/editByFrequency";
    }

}