package com.sooka.sponest.data.emergency.controller;

import com.ruoyi.common.core.utils.poi.ExcelUtil;
import com.ruoyi.common.core.web.controller.BaseController;
import com.ruoyi.common.core.web.domain.AjaxResult;
import com.ruoyi.common.core.web.page.TableDataInfo;
import com.ruoyi.common.log.annotation.Log;
import com.ruoyi.common.log.enums.BusinessType;
import com.ruoyi.common.security.annotation.RequiresPermissions;
import com.sooka.sponest.data.emergency.domain.TBusEnterprise;
import com.sooka.sponest.data.emergency.service.ITBusEnterpriseService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;

import javax.servlet.http.HttpServletResponse;
import java.util.List;

/**
 * 企业Controller
 *
 * @author ruoyi
 * @date 2022-11-15
 */
@RestController
@RequestMapping("/enterprise")
public class TBusEnterpriseController extends BaseController
{
    @Autowired
    private ITBusEnterpriseService tBusEnterpriseService;

    /**
     * 查询企业列表
     */
    @RequiresPermissions("system:enterprise:list")
    @GetMapping("/list")
    public TableDataInfo list(TBusEnterprise tBusEnterprise)
    {
        startPage();
        List<TBusEnterprise> list = tBusEnterpriseService.selectTBusEnterpriseList(tBusEnterprise);
        return getDataTable(list);
    }

    /**
     * 导出企业列表
     */
    @RequiresPermissions("system:enterprise:export")
    @Log(title = "企业", businessType = BusinessType.EXPORT)
    @PostMapping("/export")
    public void export(HttpServletResponse response, TBusEnterprise tBusEnterprise)
    {
        List<TBusEnterprise> list = tBusEnterpriseService.selectTBusEnterpriseList(tBusEnterprise);
        ExcelUtil<TBusEnterprise> util = new ExcelUtil<TBusEnterprise>(TBusEnterprise.class);
        util.exportExcel(response, list, "企业数据");
    }

    /**
     * 获取企业详细信息
     */
    @RequiresPermissions("system:enterprise:query")
    @GetMapping(value = "/{busEnterpriseId}")
    public AjaxResult getInfo(@PathVariable("busEnterpriseId") String busEnterpriseId)
    {
        return success(tBusEnterpriseService.selectTBusEnterpriseByBusEnterpriseId(busEnterpriseId));
    }

    /**
     * 新增企业
     */
    @RequiresPermissions("system:enterprise:add")
    @Log(title = "企业", businessType = BusinessType.INSERT)
    @PostMapping
    public AjaxResult add(@RequestBody TBusEnterprise tBusEnterprise)
    {

        TBusEnterprise t = new TBusEnterprise();
        t.setEnterpriseCode(tBusEnterprise.getEnterpriseCode());
        if(tBusEnterpriseService.selectTBusEnterpriseList(t).size() >0){
            return AjaxResult.error("企业码重复，不能添加!");
        }

        return toAjax(tBusEnterpriseService.insertTBusEnterprise(tBusEnterprise));
    }

    /**
     * 修改企业
     */
    @RequiresPermissions("system:enterprise:edit")
    @Log(title = "企业", businessType = BusinessType.UPDATE)
    @PutMapping
    public AjaxResult edit(@RequestBody TBusEnterprise tBusEnterprise)
    {
        return toAjax(tBusEnterpriseService.updateTBusEnterprise(tBusEnterprise));
    }

    /**
     * 删除企业
     */
    @RequiresPermissions("system:enterprise:remove")
    @Log(title = "企业", businessType = BusinessType.DELETE)
	@DeleteMapping("/{busEnterpriseIds}")
    public AjaxResult remove(@PathVariable String[] busEnterpriseIds)
    {
        return toAjax(tBusEnterpriseService.deleteTBusEnterpriseByBusEnterpriseIds(busEnterpriseIds));
    }
}