zz-marketing/src/main/java/com/zhongzheng/controller/SysCompanyController.java

package com.zhongzheng.controller;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.update.LambdaUpdateWrapper;
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.zhongzheng.bo.SysCertificateBo;
import com.zhongzheng.bo.SysDictTypeBo;
import com.zhongzheng.common.croe.BaseController;
import com.zhongzheng.common.croe.LoginUser;
import com.zhongzheng.common.filter.CustomException;
import com.zhongzheng.common.filter.UserConstants;
import com.zhongzheng.common.model.AjaxResult;
import com.zhongzheng.common.model.TableDataInfo;
import com.zhongzheng.common.model.TokenService;
import com.zhongzheng.common.util.DateUtils;
import com.zhongzheng.common.util.SecurityUtils;
import com.zhongzheng.common.util.ServletUtils;
import com.zhongzheng.common.util.ToolsUtils;
import com.zhongzheng.domian.*;
import com.zhongzheng.service.*;
import io.swagger.annotations.Api;
import io.swagger.annotations.ApiOperation;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.validation.annotation.Validated;
import org.springframework.web.bind.annotation.*;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;

import static net.sf.jsqlparser.util.validation.metadata.NamedObject.user;

/**
 * 数据字典信息
 *
 * @author zhongzheng
 */
@Api(tags ="我的客户管理")
@RestController
@RequestMapping("/system/company")
public class SysCompanyController extends BaseController
{
    @Autowired
    private ISysCompanyService companyService;
    @Autowired
    private ISysCustomerSeaService seaService;
    @Autowired
    private ISysCompanyContactsService companyContactsService;
    @Autowired
    private TokenService tokenService;
    @Autowired
    private ISysPersonService personService;

    @Autowired
    private ISysCertificateService certService;

    @Autowired
    private ICertificateMajorService majorService;

    @Autowired
    private ISysCertificateTypeService typeService;



    @ApiOperation("客户列表")
    @GetMapping("/list")
    public TableDataInfo<SysCompany> list(SysCompany model) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        model.setUserId(userId);
        return companyService.selectCompanyList(model);
    }

    @ApiOperation("新增企业")
    @PostMapping("/add")
    public AjaxResult add(@Validated @RequestBody SysCompany company) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        company.setUserId(userId);

        company.setSource("自有");
        SysCustomerSea sea = seaService.selectCustomerSeaByName(company.getCompanyName());
        if(sea!=null){
            throw new CustomException("企业已存在公海中，请直接在公海中领取！");
        }
        return toAjax(companyService.insertCompany(company)>0?1:0);
    }

    @ApiOperation("编辑企业")
    @PostMapping("/edit")
    public AjaxResult edit(@Validated @RequestBody SysCompany company) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        company.setUserId(userId);
        company.setSource("自有");
        return toAjax(companyService.updateCompany(company,company.getUserId()));
    }

    @ApiOperation("删除人")
    @PostMapping("/del")
    public AjaxResult del(@Validated @RequestBody SysCompany entity) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        return toAjax(companyService.deleteCompanyById(entity.getCompanyId(),userId));
    }

    @ApiOperation("客户联系人列表")
    @GetMapping("/contact/list")
    public TableDataInfo<SysCompanyContacts> contact_list(SysCompanyContacts model) {

        return companyContactsService.selectCompanyContactsList(model);
    }

    @ApiOperation("新增企业联系人")
    @PostMapping("/contact/add")
    public AjaxResult contact_add(@Validated @RequestBody SysCompanyContacts entity) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        entity.setUserId(userId);
        return toAjax(companyContactsService.insertSysCompanyContacts(entity)>0?1:0);
    }

    @ApiOperation("编辑企业联系人")
    @PostMapping("/contact/edit")
    public AjaxResult contact_edit(@Validated @RequestBody SysCompanyContacts entity) {
        return toAjax(companyContactsService.updateSysCompanyContacts(entity)>0?1:0);
    }

    @ApiOperation("删除企业联系人")
    @PostMapping("/contact/del")
    public AjaxResult contact_del(@Validated @RequestBody SysCompanyContacts entity) {
        return toAjax(companyContactsService.deleteSysCompanyContactsById(entity.getContactId()));
    }


    @ApiOperation("员工列表")
    @GetMapping("/person/list")
    public TableDataInfo<SysPerson> contact_list(SysPerson model) {
        return personService.selectPersonList(model);
    }

    @ApiOperation("新增企业员工")
    @PostMapping("/person/add")
    public AjaxResult person_add(@Validated @RequestBody SysPerson entity) {
        //entity.setTenantId((long)1);
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        entity.setCreator(userId);
        return toAjax(personService.insertPerson(entity)>0?1:0);
    }

    @ApiOperation("编辑企业员工")
    @PostMapping("/person/edit")
    public AjaxResult person_edit(@Validated @RequestBody SysPerson entity) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();

        return toAjax(personService.updatePerson(entity,userId)>0?1:0);
    }

    @ApiOperation("删除企业员工")
    @PostMapping("/person/del")
    public AjaxResult person_del(@Validated @RequestBody SysPerson entity) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        Long userId = loginUser.getUser().getUserId();
        return toAjax(personService.deletePersonById(entity.getPersonId(),userId));
    }

    @ApiOperation("员工证书列表")
    @GetMapping("/cert/list")
    public TableDataInfo<SysCertificate>  cert_list(SysCertificateBo model) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        //model.setUserId(1L);
        return certService.selectList(model);
    }

    @ApiOperation("企业安全生产许可证书列表")
    @GetMapping("/cert/safetyList")
    public TableDataInfo<SysCertificate>  cert_safety_license_list(SysCertificateBo model) {

        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        model.setCertificateTypeId(1);
        //model.setUserId(1L);
        return certService.selectEnterpriseCerList(model);
    }
    @ApiOperation("企业资质证书列表")
    @GetMapping("/cert/qualificationList")
    public TableDataInfo<SysCertificate>  cert_qualification_list(SysCertificateBo model) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        model.setCertificateTypeId(3);
        //model.setUserId(1L);
        return certService.selectEnterpriseCerList(model);
    }


    @ApiOperation("添加证书")
    @PostMapping("/cert/add")
    public AjaxResult cert_add(@Validated @RequestBody SysCertificateBo model) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        return toAjax(certService.insertCertificate(model)>0?1:0);
    }

    @ApiOperation("编辑证书")
    @PostMapping("/cert/edit")
    public AjaxResult cert_edit(@Validated @RequestBody SysCertificateBo model) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        return toAjax(certService.updateCertificate(model,model.getUserId()));
    }

    @ApiOperation("删除证书")
    @PostMapping("/cert/del")
    public AjaxResult cert_del(@Validated @RequestBody SysCertificateBo model) {
        LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        model.setUserId(loginUser.getUser().getUserId());
        return toAjax(certService.deleteCertificateById(model.getCertificateId(),model.getUserId()));
    }
    @ApiOperation("获取证书类型列表 parentId=2 个人证书")
    @GetMapping("/cert/certificate_type_list")
    public TableDataInfo<SysCertificateType>  certificate_type_list(Integer parentId) {

        return typeService.selectCertificateTypeListByCode(parentId);
    }



@ApiOperation("获取专业列表")
@GetMapping("/cert/major_list")
public TableDataInfo<CertificateMajor>  cert_major_list(Integer certificateTypeId) {

    return majorService.selectCertificateMajorList(certificateTypeId);
}



    @ApiOperation("证书预警列表")
    @GetMapping("/cert/warningList")
    public TableDataInfo<SysCertificate>  cert_warning_list(SysCertificateBo model) {

        //LoginUser loginUser = tokenService.getLoginUser(ServletUtils.getRequest());
        //model.setUserId(loginUser.getUser().getUserId());
        model.setUserId(1L);
        model.setWarningState(1);
        return certService.selectList(model);
    }











}