package com.doumee.api.business; 
 | 
  
 | 
import com.doumee.config.DataSyncConfig; 
 | 
import com.doumee.core.annotation.excel.ExcelExporter; 
 | 
import com.doumee.core.annotation.pr.PreventRepeat; 
 | 
import com.doumee.service.business.third.model.ApiResponse; 
 | 
import com.doumee.service.business.third.model.PageWrap; 
 | 
import com.doumee.service.business.third.model.PageData; 
 | 
import com.doumee.core.utils.Constants; 
 | 
import com.doumee.dao.admin.request.UpdateCompanySortDTO; 
 | 
import com.doumee.dao.admin.response.CompanyDTO; 
 | 
import com.doumee.dao.business.model.Company; 
 | 
import com.doumee.service.business.CompanyService; 
 | 
import com.doumee.service.business.ext.ERPSyncService; 
 | 
import com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromHKServiceImpl; 
 | 
import io.swagger.annotations.Api; 
 | 
import io.swagger.annotations.ApiImplicitParam; 
 | 
import io.swagger.annotations.ApiImplicitParams; 
 | 
import io.swagger.annotations.ApiOperation; 
 | 
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 java.util.ArrayList; 
 | 
import java.util.List; 
 | 
  
 | 
import com.doumee.api.BaseController; 
 | 
/** 
 | 
 * @author 江蹄蹄 
 | 
 * @date 2023/11/30 15:33 
 | 
 */ 
 | 
@Api(tags = "企业信息表") 
 | 
@RestController 
 | 
@RequestMapping("/business/company") 
 | 
public class CompanyController extends BaseController { 
 | 
  
 | 
    @Autowired 
 | 
    private CompanyService companyService; 
 | 
  
 | 
    @Autowired 
 | 
    private ERPSyncService erpSyncService; 
 | 
    @Autowired 
 | 
    private HkSyncOrgUserFromHKServiceImpl hkSyncOrgUserFHKService; 
 | 
    @Autowired 
 | 
    private DataSyncConfig dataSyncConfig; 
 | 
  
 | 
    @PreventRepeat 
 | 
    @ApiOperation("新建") 
 | 
    @PostMapping("/create") 
 | 
    @RequiresPermissions("business:company:create") 
 | 
    public ApiResponse create(@RequestBody Company company) { 
 | 
        return ApiResponse.success(companyService.create(company)); 
 | 
    } 
 | 
  
 | 
    @PreventRepeat 
 | 
    @ApiOperation("创建劳务公司") 
 | 
    @PostMapping("/createLaborServices") 
 | 
    @RequiresPermissions("business:company:create") 
 | 
    public ApiResponse<Integer> createLaborServices(@RequestBody Company company){ 
 | 
        return ApiResponse.success(companyService.createLaborServices(company)); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("根据ID删除") 
 | 
    @GetMapping("/delete/{id}") 
 | 
    @RequiresPermissions("business:company:delete") 
 | 
    public ApiResponse deleteById(@PathVariable Integer id) { 
 | 
        companyService.deleteById(id,this.getLoginUser(null)); 
 | 
        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,this.getLoginUser(null)); 
 | 
        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("组织树查询") 
 | 
    @GetMapping("/tree") 
 | 
    @ApiImplicitParams({ 
 | 
            @ApiImplicitParam(name = "type", dataType = "Integer", value = "类型 0劳务公司 1内部组织", required = true) 
 | 
    }) 
 | 
    @RequiresPermissions("business:company:query") 
 | 
    public ApiResponse<List<Company>> tree (@RequestParam Integer type) { 
 | 
        return ApiResponse.success(companyService.companyTree(type)); 
 | 
    } 
 | 
  
 | 
  
 | 
    @ApiOperation("排序") 
 | 
    @PostMapping("/updateSort") 
 | 
    @RequiresPermissions("business:company:query") 
 | 
    public ApiResponse updateSort (@Validated @RequestBody UpdateCompanySortDTO dto) { 
 | 
        companyService.updateSort(dto); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
  
 | 
    @ApiOperation("修改状态") 
 | 
    @PostMapping("/updateStatusById") 
 | 
    @RequiresPermissions("business:company:update") 
 | 
    public ApiResponse updateStatusById(@RequestBody Company company) { 
 | 
        companyService.updateStatusById(company); 
 | 
        return ApiResponse.success(null); 
 | 
    } 
 | 
  
 | 
    @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 findById(@PathVariable Integer id) { 
 | 
        return ApiResponse.success(companyService.findById(id)); 
 | 
    } 
 | 
  
 | 
    @ApiOperation("根据分类ID查询") 
 | 
    @PostMapping("/findLaborServicesList") 
 | 
    @RequiresPermissions("business:company:query") 
 | 
    public ApiResponse<List<Company>> findLaborServicesList(Company query){ 
 | 
        Company company = new Company(); 
 | 
        company.setIsdeleted(Constants.ZERO); 
 | 
        company.setType(query.getType()); 
 | 
        company.setCategoryId(query.getCategoryId()); 
 | 
        return ApiResponse.success(companyService.findList(company)); 
 | 
    } 
 | 
  
 | 
  
 | 
    /** 
 | 
     * 查询部门及其子部门信息 
 | 
     * @param type 0 查询所有信息 1 查询内部信息 
 | 
     * @return 
 | 
     */ 
 | 
    @ApiOperation("根据类型查询组织信息") 
 | 
    @GetMapping("/findCompanyTreePage") 
 | 
    @RequiresPermissions("business:company:query") 
 | 
    public ApiResponse<List<CompanyDTO>> findCompanyTreePage(Integer type){ 
 | 
        return ApiResponse.success(companyService.findCompanyTreePage(type)); 
 | 
    } 
 | 
  
 | 
  
 | 
  
 | 
    @ApiOperation("全量部门信息同步") 
 | 
    @PostMapping("/syncAll") 
 | 
    @RequiresPermissions("business:company:sync") 
 | 
    public ApiResponse sync(){ 
 | 
        if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.erp){ 
 | 
            return ApiResponse.success( erpSyncService.syncCompany(null)); 
 | 
        }else if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.hk){ 
 | 
             hkSyncOrgUserFHKService.syncOrgData(this.getLoginUser(null)); 
 | 
            return ApiResponse.success("同步成功"); 
 | 
        }else if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.self){ 
 | 
            //如果是自建系統,同步安防平台初始化数据 
 | 
            hkSyncOrgUserFHKService.syncOrgData(this.getLoginUser(null)); 
 | 
            return ApiResponse.success("同步成功"); 
 | 
        } 
 | 
        return ApiResponse.failed("无效操作"); 
 | 
    } 
 | 
} 
 |