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

import com.ruoyi.doublePrevention.entity.PreventRiskControlMeasure;
import com.ruoyi.doublePrevention.entity.PreventRiskEvent;
import com.ruoyi.doublePrevention.entity.dto.req.*;
import com.ruoyi.doublePrevention.entity.dto.resp.*;
import com.ruoyi.doublePrevention.service.RiskService;
import com.ruoyi.doublePrevention.vo.ResultVO;
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 org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.ui.ModelMap;
import org.springframework.util.ObjectUtils;
import org.springframework.web.bind.annotation.*;

import java.util.List;

@Controller
@RequestMapping("/tr/riskMeasure")
public class riskMeasureController extends BaseController {
    private String prefix = "tr/riskMeasure";

    @Autowired
    private RiskService riskService;

    @GetMapping()
    public String riskEvent(ModelMap mmap)
    {
        ResultVO<List<PreventRiskEventListQueryRespDTO>> resultVO = riskService.listRiskEvent();
        Object data = resultVO.getData();
        mmap.put("riskEvents", data);
        return prefix + "/riskMeasure";
    }

    /**
     * 风险管控措施分页查询
     */
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(PreventRiskControlMeasureQueryReqDTO measureQueryReqDTO)
    {
        ResultVO<List<PreventRiskControlMeasurePageQueryRespDTO>> resultVO = riskService.getRiskControlMeasurePage(measureQueryReqDTO);
        List<PreventRiskControlMeasureQueryReqDTO> data = (List<PreventRiskControlMeasureQueryReqDTO>) 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;
    }

    /**
     * 新增风险管控措施
     */
    @GetMapping("/add")
    public String add(ModelMap mmap)
    {
        ResultVO<List<PreventRiskEventListQueryRespDTO>> resultVO = riskService.listRiskEvent();
        Object data = resultVO.getData();
        mmap.put("riskEvents", data);
        return prefix + "/add";
    }


    /**
     * 新增保存安全风险事件
     */
    @Log(title = "风险单元类型", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(PreventRiskControlMeasureSaveReqDTO measureSaveReqDTO)
    {

        ResultVO<PreventRiskControlMeasure> resultVO = riskService.saveRiskControlMeasure(measureSaveReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
    }


    /**
     * 修改风险管控措施
     */
    @GetMapping("/edit/{id}")
    public String edit(@PathVariable("id") Long id, ModelMap mmap)
    {
        PreventRiskControlMeasureGetReqDTO riskControlMeasureGetReqDTO = new PreventRiskControlMeasureGetReqDTO();
        riskControlMeasureGetReqDTO.setId(id);
        ResultVO<PreventRiskControlMeasureRespDTO> riskControlMeasure = riskService.getRiskControlMeasure(riskControlMeasureGetReqDTO);
        Object measureData = riskControlMeasure.getData();
        mmap.put("measureData",measureData);

        ResultVO<List<PreventRiskEventListQueryRespDTO>> resultVO = riskService.listRiskEvent();
        Object data = resultVO.getData();
        mmap.put("riskEvents", data);
        return prefix + "/edit";
    }

    /**
     * 修改保存风险单元类型
     */
    @Log(title = "风险单元类型", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(PreventRiskControlMeasureUpdateReqDTO measureUpdateReqDTO)
    {
        ResultVO<PreventRiskControlMeasure> resultVO = riskService.updateRiskControlMeasure(measureUpdateReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
    }
    
    /**
     * 删除风险管控措施
     */
    @Log(title = "风险管控措施", businessType = BusinessType.DELETE)
    @PostMapping( "/remove")
    @ResponseBody
    public AjaxResult remove(Long ids)
    {
        PreventRiskControlMeasureDeleteReqDTO measureDeleteReqDTO = new PreventRiskControlMeasureDeleteReqDTO();
        measureDeleteReqDTO.setId(ids);
        ResultVO<PreventRiskControlMeasure> resultVO = riskService.deleteRiskControlMeasure(measureDeleteReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
    }

    /**
     * 新增风险管控措施 - 最大风险源
     */
    @GetMapping("/addOne")
    public String addOne(ModelMap mmap)
    {
        ResultVO<List<PreventRiskEventListQueryRespDTO>> resultVO = riskService.listRiskEvent();
        Object data = resultVO.getData();
        mmap.put("riskEvents", data);
        return prefix + "/add";
    }

    /**
     * 新增保存
     */
    @Log(title = "风险单元类型", businessType = BusinessType.INSERT)
    @PostMapping("/addOne")
    @ResponseBody
    public AjaxResult addSaveOne(PreventRiskControlMeasureSaveReqDTO measureSaveReqDTO)
    {

        ResultVO<PreventRiskControlMeasure> resultVO = riskService.saveRiskControlMeasure(measureSaveReqDTO);
        String code = resultVO.getCode();
        if ("200".equals(code)){
            return toAjax(1);
        }else {
            return AjaxResult.error(resultVO.getMsg());
        }
    }

}