package com.ruoyi.system.web;

import com.ruoyi.common.core.controller.BaseController;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.domain.model.LoginUser;
import com.ruoyi.common.utils.poi.ExcelUtil;
import com.ruoyi.system.domain.dto.CompanyInfoDto;
import com.ruoyi.system.domain.req.*;
import com.ruoyi.system.domain.resp.CompanyInfoRespDTO;
import com.ruoyi.system.domain.vo.CompanyInfoVO;
import com.ruoyi.system.service.enterpriseManage.EnterpriseManageService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;
import org.springframework.web.multipart.MultipartFile;

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


@RestController
@RequestMapping("/enterpriseManage")
public class EnterpriseManageController extends BaseController {

    @Autowired
    private EnterpriseManageService enterpriseManageService;


    /**
     * @description 新增企业信息
     */
    @PostMapping("/insert/saveEnterpriseInfo")
    public AjaxResult saveEnterpriseInfo(EnterpriseInfoAddReqDTO infoAddReqDTO, MultipartFile companyImageFile,
                                         MultipartFile businessLicenseFile, MultipartFile securityCertificateFile) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.saveEnterpriseInfo(user, infoAddReqDTO, companyImageFile, businessLicenseFile, securityCertificateFile);
    }

    /**
     * @description 删除企业信息通过id
     */
    @PostMapping("/delete/deleteEnterpriseInfoById")
    public AjaxResult deleteEnterpriseInfoById(@RequestBody EnterpriseInfoDeleteByIdReqDTO deleteByIdReqDTO) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.deleteEnterpriseInfoById(user, deleteByIdReqDTO);
    }

    /**
     * @description 修改企业信息
     */
    @PostMapping("/update/updateEnterpriseInfoById")
    public AjaxResult updateEnterpriseInfoById(EnterpriseInfoUpdateByIdReqDTO updateByIdReqDTO, MultipartFile companyImageFile,
                                               MultipartFile businessLicenseFile, MultipartFile securityCertificateFile) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.updateEnterpriseInfoById(user, updateByIdReqDTO, companyImageFile, businessLicenseFile, securityCertificateFile);
    }

    /**
     * @description 查询企业信息通过id
     */
    @PostMapping("/select/getEnterpriseInfoById")
    public AjaxResult getEnterpriseInfoById(@RequestBody EnterpriseInfoSelectByIdReqDTO selectById) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.getEnterpriseInfoById(user, selectById);
    }

    /**
     * @description 查询各地市的企业信息
     */
    @PostMapping("/select/getEnterpriseInfoByCity")
    public AjaxResult getEnterpriseInfoByCity(@RequestBody EnterpriseInfoSelectByCityReqDTO selectByCityReqDTO) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.getEnterpriseInfoByCity(user, selectByCityReqDTO);
    }

    /**
     * @description 根据条件查询企业信息
     */
    @PostMapping("/select/getEnterpriseInfoByCondition")
    public AjaxResult getEnterpriseInfoByCondition(@RequestBody EnterpriseInfoSelectByConditionReqDTO conditionReqDTO) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.getEnterpriseInfoByCondition(user, conditionReqDTO);
    }

    /**
     * @description 统计全省的数据
     */
    @PostMapping("/select/countProvinceData")
    public AjaxResult countProvinceData() {
        LoginUser user = getLoginUser();
        return enterpriseManageService.countProvinceData(user);
    }

    /**
     * @description 下载企业信息表格模板
     */
    @PostMapping("/download/downLoadTemplate")
    public void downLoadTemplate(HttpServletResponse response) {
        List<CompanyInfoRespDTO> list = new ArrayList<>();
        ExcelUtil<CompanyInfoRespDTO> util = new ExcelUtil<>(CompanyInfoRespDTO.class);
        util.exportExcel(response, list, "导入模板");
    }

    /**
     * @description 批量导入企业信息
     */
    @RequestMapping("/import/importData")
    public AjaxResult importData(MultipartFile file) {
        LoginUser user = getLoginUser();
        return enterpriseManageService.importData(user,file);
    }
}