package com.ruoyi.project.dc.securityType.controller;

import java.util.List;
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.dc.securityType.domain.SecurityType;
import com.ruoyi.project.dc.securityType.service.ISecurityTypeService;
import com.ruoyi.framework.web.controller.BaseController;
import com.ruoyi.framework.web.domain.AjaxResult;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.framework.web.domain.Ztree;

/**
 * 安全体系文件分类Controller
 * 
 * @author wm
 * @date 2020-12-09
 */
@Controller
@RequestMapping("/dc/securityType")
public class SecurityTypeController extends BaseController
{
    private String prefix = "dc/securityType";

    @Autowired
    private ISecurityTypeService securityTypeService;

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

    /**
     * 查询安全体系文件分类树列表
     */
    @RequiresPermissions("dc:securityType:list")
    @PostMapping("/list")
    @ResponseBody
    public List<SecurityType> list(SecurityType securityType)
    {
        List<SecurityType> list = securityTypeService.selectSecurityTypeList(securityType);
        return list;
    }

    /**
     * 导出安全体系文件分类列表
     */
    @RequiresPermissions("dc:securityType:export")
    @Log(title = "安全体系文件分类", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    @ResponseBody
    public AjaxResult export(SecurityType securityType)
    {
        List<SecurityType> list = securityTypeService.selectSecurityTypeList(securityType);
        ExcelUtil<SecurityType> util = new ExcelUtil<SecurityType>(SecurityType.class);
        return util.exportExcel(list, "securityType");
    }

    /**
     * 新增安全体系文件分类
     */
    @GetMapping(value = { "/add/{securityTypeId}", "/add/" })
    public String add(@PathVariable(value = "securityTypeId", required = false) Long securityTypeId, ModelMap mmap)
    {
        if (StringUtils.isNotNull(securityTypeId))
        {
            mmap.put("securityType", securityTypeService.selectSecurityTypeById(securityTypeId));
        }
        return prefix + "/add";
    }

    /**
     * 新增保存安全体系文件分类
     */
    @RequiresPermissions("dc:securityType:add")
    @Log(title = "安全体系文件分类", businessType = BusinessType.INSERT)
    @PostMapping("/add")
    @ResponseBody
    public AjaxResult addSave(SecurityType securityType)
    {
        return toAjax(securityTypeService.insertSecurityType(securityType));
    }

    /**
     * 修改安全体系文件分类
     */
    @GetMapping("/edit/{securityTypeId}")
    public String edit(@PathVariable("securityTypeId") Long securityTypeId, ModelMap mmap)
    {
        SecurityType securityType = securityTypeService.selectSecurityTypeById(securityTypeId);
        mmap.put("securityType", securityType);
        return prefix + "/edit";
    }

    /**
     * 修改保存安全体系文件分类
     */
    @RequiresPermissions("dc:securityType:edit")
    @Log(title = "安全体系文件分类", businessType = BusinessType.UPDATE)
    @PostMapping("/edit")
    @ResponseBody
    public AjaxResult editSave(SecurityType securityType)
    {
        return toAjax(securityTypeService.updateSecurityType(securityType));
    }

    /**
     * 删除
     */
    @RequiresPermissions("dc:securityType:remove")
    @Log(title = "安全体系文件分类", businessType = BusinessType.DELETE)
    @GetMapping("/remove/{securityTypeId}")
    @ResponseBody
    public AjaxResult remove(@PathVariable("securityTypeId") Long securityTypeId)
    {
        return toAjax(securityTypeService.deleteSecurityTypeById(securityTypeId));
    }

    /**
     * 选择安全体系文件分类树
     */
    @GetMapping(value = { "/selectSecurityTypeTree/{securityTypeId}", "/selectSecurityTypeTree/" })
    public String selectSecurityTypeTree(@PathVariable(value = "securityTypeId", required = false) Long securityTypeId, ModelMap mmap)
    {
        if (StringUtils.isNotNull(securityTypeId))
        {
            mmap.put("securityType", securityTypeService.selectSecurityTypeById(securityTypeId));
        }
        return prefix + "/tree";
    }

    /**
     * 加载安全体系文件分类树列表
     */
    @GetMapping("/treeData")
    @ResponseBody
    public List<Ztree> treeData()
    {
        List<Ztree> ztrees = securityTypeService.selectSecurityTypeTree();
        return ztrees;
    }
}