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

import java.util.List;

import com.ruoyi.project.enumerate.TrEvaluationMethodDetailsEnum;
import com.ruoyi.project.tr.criterionDetail.domain.CriterionDetail;
import com.ruoyi.project.tr.criterionDetail.service.ICriterionDetailService;
import com.ruoyi.project.tr.determination.domain.Determination;
import com.ruoyi.project.tr.determination.service.IDeterminationService;
import com.ruoyi.project.tr.riskLevel.domain.RiskLevel;
import com.ruoyi.project.tr.riskLevel.service.IRiskLevelService;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.ui.ModelMap;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PathVariable;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;
import org.springframework.web.bind.annotation.ResponseBody;
import com.ruoyi.framework.aspectj.lang.annotation.Log;
import com.ruoyi.framework.aspectj.lang.enums.BusinessType;
import com.ruoyi.project.tr.method.domain.Method;
import com.ruoyi.project.tr.method.service.IMethodService;
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/method")
public class MethodController extends BaseController
{
    private String prefix = "tr/method";

    @Autowired
    private IMethodService methodService;

    @Autowired
    private ICriterionDetailService criterionDetailService;

    @Autowired
    private IDeterminationService determinationService;

    @Autowired
    private IRiskLevelService riskLevelService;

    @RequiresPermissions("tr:method:view")
    @GetMapping()
    public String method()
    {
        return prefix + "/method";
    }

    /**
     * 查询风险单元评价方法列表
     */
    @RequiresPermissions("tr:method:list")
    @PostMapping("/list")
    @ResponseBody
    public TableDataInfo list(Method method)
    {
        startPage();
        List<Method> list = methodService.selectMethodList(method);
        return getDataTable(list);
    }

    /**
     * 导出风险单元评价方法列表
     */
    @RequiresPermissions("tr:method:export")
    @Log(title = "风险单元评价方法", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ResponseBody
    public AjaxResult export(Method method)
    {
        List<Method> list = methodService.selectMethodList(method);
        ExcelUtil<Method> util = new ExcelUtil<Method>(Method.class);
        return util.exportExcel(list, "method");
    }

    /**
     * 新增风险单元评价方法
     */
    @GetMapping("/add")
    public String add()
    {
        return prefix + "/add";
    }

    /**
     * 新增保存风险单元评价方法
     */
    @RequiresPermissions("tr:method:add")
    @Log(title = "风险单元评价方法", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(Method method)
    {
        return toAjax(methodService.insertMethod(method));
    }

    /**
     * 修改风险单元评价方法
     */
    @GetMapping("/edit/{methodId}")
    public String edit(@PathVariable("methodId") Long methodId, ModelMap mmap)
    {
        Method method = methodService.selectMethodById(methodId);
        mmap.put("method", method);
        return prefix + "/edit";
    }

    /**
     * 修改保存风险单元评价方法
     */
    @RequiresPermissions("tr:method:edit")
    @Log(title = "风险单元评价方法", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(Method method)
    {
        return toAjax(methodService.updateMethod(method));
    }

    /**
     * 删除风险单元评价方法
     */
    @RequiresPermissions("tr:method:remove")
    @Log(title = "风险单元评价方法", businessType = BusinessType.DELETE)
    @PostMapping( "/remove")
    @ResponseBody
    public AjaxResult remove(String ids)
    {
        return toAjax(methodService.deleteMethodByIds(ids));
    }


    /**
     * 方法设置页面
     */
    @GetMapping("/methodSetting")
    public String methodSetting()
    {
        return prefix + "/methodSetting";
    }


    /**
     * 根据不同的条件,获取到不同的方法详情
     * @param methodCriterion
     * @return
     */
    @PostMapping("/criterionDetail/{methodCriterion}")
    @ResponseBody
    public TableDataInfo listBy(@PathVariable("methodCriterion") String methodCriterion)
    {

        CriterionDetail criterionDetail = new CriterionDetail();

        if (TrEvaluationMethodDetailsEnum.LS_L.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.LS_L.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.LS_L.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.LS_S.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.LS_S.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.LS_S.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.LEC_L.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.LEC_L.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.LEC_L.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.LEC_E.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.LEC_E.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.LEC_E.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.LEC_C.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.LEC_C.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.LEC_C.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.MES_M.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.MES_M.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.MES_M.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.MES_E.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.MES_E.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.MES_E.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.MES_S.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.MES_S.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.MES_S.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.MES_S.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.MES_S.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.MES_S.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.RS_R.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.RS_R.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.RS_R.getCriterionId());
        }else if (TrEvaluationMethodDetailsEnum.RS_S.getUrl().equals(methodCriterion)){
            criterionDetail.setMethodId(TrEvaluationMethodDetailsEnum.RS_S.getMethodId());
            criterionDetail.setCriterionId(TrEvaluationMethodDetailsEnum.RS_S.getCriterionId());
        }

        startPage();
        List<CriterionDetail> list = criterionDetailService.selectCriterionDetailList(criterionDetail);
        return getDataTable(list);
    }

    /**
     * 评价方法对应的判定表
     * @param method
     * @return
     */
    @PostMapping("/determinationList/{method}")
    @ResponseBody
    public TableDataInfo list(@PathVariable("method") String method)
    {
        Determination determination = new Determination();

        if (TrEvaluationMethodDetailsEnum.LS_R.getUrl().equals(method)){
            determination.setMethodId(TrEvaluationMethodDetailsEnum.LS_R.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.LEC_D.getUrl().equals(method)){
            determination.setMethodId(TrEvaluationMethodDetailsEnum.LEC_D.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.MES_D.getUrl().equals(method)){
            determination.setMethodId(TrEvaluationMethodDetailsEnum.MES_D.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.RS_D.getUrl().equals(method)){
            determination.setMethodId(TrEvaluationMethodDetailsEnum.RS_D.getMethodId());
        }

        startPage();
        List<Determination> list = determinationService.selectDeterminationList(determination);
        return getDataTable(list);
    }

    /**
     * 对应颜色等级表
     * @param method
     * @return
     */
    @PostMapping("/riskLevelList/{method}")
    @ResponseBody
    public TableDataInfo riskLevelList(@PathVariable("method") String method){
        RiskLevel riskLevel = new RiskLevel();

        if (TrEvaluationMethodDetailsEnum.LS_LEVEL.getUrl().equals(method)){
            riskLevel.setMethodId(TrEvaluationMethodDetailsEnum.LS_LEVEL.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.LEC_LEVEL.getUrl().equals(method)){
            riskLevel.setMethodId(TrEvaluationMethodDetailsEnum.LEC_LEVEL.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.MES_LEVEL.getUrl().equals(method)){
            riskLevel.setMethodId(TrEvaluationMethodDetailsEnum.MES_LEVEL.getMethodId());
        }else if (TrEvaluationMethodDetailsEnum.RS_LEVEL.getUrl().equals(method)){
            riskLevel.setMethodId(TrEvaluationMethodDetailsEnum.RS_LEVEL.getMethodId());
        }

        startPage();
        List<RiskLevel> list = riskLevelService.selectRiskLevelList(riskLevel);
        return getDataTable(list);
    }

}