package com.doumee.api.business;

import com.alibaba.fastjson.JSONObject;
import com.doumee.api.BaseController;
import com.doumee.core.annotation.excel.ExcelExporter;
import com.doumee.core.annotation.pr.PreventRepeat;
import com.doumee.core.constants.ResponseStatus;
import com.doumee.core.exception.BusinessException;
import com.doumee.core.model.ApiResponse;
import com.doumee.core.model.LoginUserInfo;
import com.doumee.core.model.PageWrap;
import com.doumee.core.model.PageData;
import com.doumee.core.utils.Constants;
import com.doumee.dao.business.model.Company;
import com.doumee.dao.business.model.ServiceCompanyDTO;
import com.doumee.dao.system.dto.CompanyDTO;
import com.doumee.service.business.CompanyService;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.apache.shiro.SecurityUtils;
import org.apache.shiro.authz.annotation.RequiresPermissions;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import javax.servlet.http.HttpServletResponse;
import javax.validation.Valid;

import java.util.ArrayList;
import java.util.List;

/**
 * @author 江蹄蹄
 * @date 2023/02/15 08:55
 */
@Api(tags = "企业信息表")
@RestController
@RequestMapping("/business/company")
public class CompanyController extends BaseController {

    @Autowired
    private CompanyService companyService;

    @PreventRepeat
    @ApiOperation("新建")
    @PostMapping("/create")
    @RequiresPermissions("business:company:create")
    public ApiResponse create(@RequestBody Company company) {
        return ApiResponse.success(companyService.create(company));
    }

    @ApiOperation("根据ID删除")
    @GetMapping("/delete/{id}")
    @RequiresPermissions("business:company:delete")
    public ApiResponse deleteById(@PathVariable Integer id) {
        companyService.deleteById(id);
        return ApiResponse.success(null);
    }

    @ApiOperation("批量删除")
    @GetMapping("/delete/batch")
    @RequiresPermissions("business:company:delete")
    public ApiResponse deleteByIdInBatch(@RequestParam String ids) {
        String [] idArray = ids.split(",");
        List<Integer> idList = new ArrayList<>();
        for (String id : idArray) {
            idList.add(Integer.valueOf(id));
        }
        companyService.deleteByIdInBatch(idList);
        return ApiResponse.success(null);
    }

    @ApiOperation("根据ID修改")
    @PostMapping("/updateById")
    @RequiresPermissions("business:company:update")
    public ApiResponse updateById(@RequestBody Company company) {
        companyService.updateById(company);
        return ApiResponse.success(null);
    }

    @ApiOperation("分页查询")
    @PostMapping("/page")
    @RequiresPermissions("business:company:query")
    public ApiResponse<PageData<Company>> findPage (@RequestBody PageWrap<Company> pageWrap) {
        return ApiResponse.success(companyService.findPage(pageWrap));
    }

    @ApiOperation("查询全部企业")
    @PostMapping("/list")
    @RequiresPermissions("business:company:query")
    public ApiResponse<List<Company>> findList (@RequestBody Company pageWrap) {
        return ApiResponse.success(companyService.findList(pageWrap));
    }

    @ApiOperation("导出Excel")
    @PostMapping("/exportExcel")
    @RequiresPermissions("business:company:exportExcel")
    public void exportExcel (@RequestBody PageWrap<Company> pageWrap, HttpServletResponse response) {
        ExcelExporter.build(Company.class).export(companyService.findPage(pageWrap).getRecords(), "企业信息表", response);
    }

    @ApiOperation("根据ID查询")
    @GetMapping("/{id}")
    @RequiresPermissions("business:company:query")
    public ApiResponse<Company> findById(@PathVariable Integer id) {
        return ApiResponse.success(companyService.findById(id));
    }


    @PreventRepeat
    @ApiOperation("企业注册")
    @PostMapping("/register")
//    @RequiresPermissions("business:company:register")
    public ApiResponse register(@RequestBody Company company) {
       companyService.register(company);
        return ApiResponse.success(null);
    }

    @PreventRepeat
    @ApiOperation("注册服务机构注册")
    @PostMapping("/registerServiceCompany")
    public ApiResponse<Integer> registerServiceCompany(@Valid @RequestBody ServiceCompanyDTO company){
        return ApiResponse.success(companyService.registerServiceCompany(company));
    }

    /**
     *
     * 注册综合服务单位
     * @param company
     * @return
     */
    @PreventRepeat
    @ApiOperation("注册综合服务单位注册")
    @PostMapping("/registerZFServiceCompany")
    public ApiResponse<Integer> registerZFServiceCompany(@Valid @RequestBody ServiceCompanyDTO company){
        return ApiResponse.success(companyService.registerZFServiceCompany(company));
    }

    @ApiOperation("查询企业资料")
    @GetMapping("/queryCompanyDetail")
    @RequiresPermissions("business:company:queryCompanyDetail")
    public ApiResponse<Company> queryCompanyDetail(@RequestParam Integer id) {
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        Company query=new Company();
        if(id == null){
            query.setId(user.getCompanyId());
        }else{
            query.setId(id);
        }
        Company company=   companyService.queryCompanyDetail(query);
        return ApiResponse.success(company);
    }


    @ApiOperation("查询企业简要资料")
    @GetMapping("/queryCompanyInfo")
    @RequiresPermissions("business:company:queryCompanyDetail")
    public ApiResponse<CompanyDTO> queryServiceCompanyDetail(@RequestParam Integer companyId){
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        if(companyId == null){
            companyId = user.getCompanyId();
        }
        CompanyDTO serviceCompany = companyService.queryCompanyDetail(companyId);
        return ApiResponse.success(serviceCompany);
    }




    @ApiOperation("企业完善资料")
    @PostMapping("/improveData")
    @RequiresPermissions("business:company:improveData")
    public ApiResponse improveData(@RequestBody Company company) {
         companyService.improveData(company);
        return ApiResponse.success(null);

    }

    @ApiOperation("判断企业是否完善资料")
    @PostMapping("/isFinishData")
    @RequiresPermissions("business:company:isFinishData")
    public ApiResponse isFinishData(@RequestBody Company company) {
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        if(!Constants.equalsInteger(user.getType(),Constants.UserType.COMPANY.getKey())){
            return ApiResponse.failed("非企业用户，禁止操作");
        }
        JSONObject json = new JSONObject();
        company.setId(user.getCompanyId());
        if( companyService.judgeImproveIsFinish(company)){
            json.put("code",true);
            json.put("msg","已完善企业资料");
            return ApiResponse.success(json);
        }else{
            json.put("code",false);
            json.put("msg","请先完善企业资料");
            return ApiResponse.success(json);
        }


    }

    @ApiOperation("判断服务商是否完善资料")
    @PostMapping("/judgeCompanyServiceIsFinish")
    @RequiresPermissions("business:company:isFinishData")
    public ApiResponse judgeCompanyServiceIsFinish(@RequestBody Company company){
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();
        if(!Constants.equalsInteger(user.getType(),Constants.UserType.SO_ADMIN.getKey())){
            return ApiResponse.failed("非企业用户，禁止操作");
        }
        JSONObject json = new JSONObject();
        company.setId(user.getCompanyId());
        if( companyService.judgeCompanyServiceIsFinish(company)){
            json.put("code",true);
            json.put("msg","已完善企业资料");
            return ApiResponse.success(json);
        }else{
            json.put("code",false);
            json.put("msg","请先完善企业资料");
            return ApiResponse.success(json);
        }
    }

    @PreventRepeat
    @ApiOperation("综合服务单位修改")
    @PostMapping("/updateZFServiceCompany")
    public ApiResponse<Integer> updateZFServiceCompany(@RequestBody Company company){
        LoginUserInfo user = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal();

        if (Constants.equalsInteger(user.getType(), Constants.UserType.SYSTEM.getKey())){
            company.setId(company.getId());
        }else if (Constants.equalsInteger(user.getType(), Constants.UserType.SO_ADMIN.getKey())
                ||Constants.equalsInteger(user.getType(), Constants.UserType.SD_ADMIN.getKey())){
            company.setId(user.getCompanyId());
        }else {
            throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(),"无权限修改服务单位信息");
        }
        companyService.updateZFServiceCompany(company);
        return ApiResponse.success(null);
    }



}