package com.doumee.cloud.admin; import com.baomidou.mybatisplus.annotation.TableField; import com.doumee.api.BaseController; import com.doumee.config.DataSyncConfig; import com.doumee.config.annotation.CloudRequiredPermission; import com.doumee.core.annotation.excel.ExcelExporter; import com.doumee.core.annotation.pr.PreventRepeat; import com.doumee.core.model.ApiResponse; import com.doumee.core.model.PageData; import com.doumee.core.model.PageWrap; 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 com.doumee.service.business.impl.hksync.fhk.HkSyncOrgUserFromSelfServiceImpl; import io.swagger.annotations.*; 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; /** * @author 江蹄蹄 * @date 2023/11/30 15:33 */ @Api(tags = "企业信息表") @RestController @RequestMapping(Constants.CLOUD_SERVICE_URL_INDEX+"/business/company") public class CompanyCloudController extends BaseController { @Autowired private CompanyService companyService; // @Autowired private ERPSyncService erpSyncService; @Autowired private HkSyncOrgUserFromHKServiceImpl hkSyncOrgUserFHKService; @Autowired private HkSyncOrgUserFromSelfServiceImpl hkSyncOrgUserFromSelfService; @Autowired private DataSyncConfig dataSyncConfig; @PreventRepeat @ApiOperation("新建") @PostMapping("/create") @CloudRequiredPermission("business:company:create") public ApiResponse create(@RequestHeader(Constants.HEADER_USER_TOKEN) String token,@RequestBody Company company) { company.setLoginUserInfo(this.getLoginUser(token)); return ApiResponse.success(companyService.create(company)); } @PreventRepeat @ApiOperation("创建财务公司") @PostMapping("/createFinanceCompany") @CloudRequiredPermission("business:company:create") public ApiResponse createFinanceCompany(@RequestBody Company company, @RequestHeader(Constants.HEADER_USER_TOKEN) String token){ company.setLoginUserInfo(this.getLoginUser(token)); return ApiResponse.success(companyService.createFinanceCompany(company)); } @PreventRepeat @ApiOperation("更新财务公司") @PostMapping("/updateFinanceCompany") @CloudRequiredPermission("business:company:update") public ApiResponse updateFinanceCompany(@RequestBody Company company, @RequestHeader(Constants.HEADER_USER_TOKEN) String token){ company.setLoginUserInfo(this.getLoginUser(token)); return ApiResponse.success(companyService.updateFinanceCompany(company)); } @PreventRepeat @ApiOperation("创建劳务公司") @PostMapping("/createLaborServices") @CloudRequiredPermission("business:company:create") public ApiResponse createLaborServices(@RequestBody Company company, @RequestHeader(Constants.HEADER_USER_TOKEN) String token){ company.setLoginUserInfo(this.getLoginUser(token)); return ApiResponse.success(companyService.createLaborServices(company)); } @ApiOperation("根据ID删除") @GetMapping("/delete/{id}") @CloudRequiredPermission("business:company:delete") public ApiResponse deleteById(@PathVariable Integer id,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ companyService.deleteById(id,this.getLoginUser(token)); return ApiResponse.success(null); } @ApiOperation("批量删除") @GetMapping("/delete/batch") @CloudRequiredPermission("business:company:delete") public ApiResponse deleteByIdInBatch(@RequestParam String ids,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ String [] idArray = ids.split(","); List idList = new ArrayList<>(); for (String id : idArray) { idList.add(Integer.valueOf(id)); } companyService.deleteByIdInBatch(idList,this.getLoginUser(token)); return ApiResponse.success(null); } @ApiOperation("根据ID修改") @PostMapping("/updateById") @CloudRequiredPermission("business:company:update") public ApiResponse updateById(@RequestBody Company company,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ company.setLoginUserInfo(this.getLoginUser(token)); companyService.updateById(company); return ApiResponse.success(null); } @ApiOperation("分页查询") @PostMapping("/page") @CloudRequiredPermission("business:company:query") public ApiResponse> findPage (@RequestBody PageWrap pageWrap,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ return ApiResponse.success(companyService.findPage(pageWrap)); } @ApiOperation("组织列表") @PostMapping("/list") @CloudRequiredPermission("business:company:query") public ApiResponse> list (@RequestBody Company company,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ return ApiResponse.success(companyService.findList(company)); } @ApiOperation("组织树查询") @GetMapping("/tree") @ApiImplicitParams({ @ApiImplicitParam(name = "type", dataType = "Integer", value = "类型 0劳务公司 1内部组织 2指定组织", required = true), }) @CloudRequiredPermission("business:company:query") public ApiResponse> tree (@RequestParam(required=false) Integer type, @RequestHeader(Constants.HEADER_USER_TOKEN) String token){ return ApiResponse.success(companyService.companyTree(type)); } @ApiOperation("排序") @PostMapping("/updateSort") @CloudRequiredPermission("business:company:sort") public ApiResponse updateSort (@Validated @RequestBody UpdateCompanySortDTO dto,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ companyService.updateSort(dto); return ApiResponse.success(null); } @ApiOperation("修改状态") @PostMapping("/updateStatusById") @CloudRequiredPermission("business:company:update") public ApiResponse updateStatusById(@RequestBody Company company,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ company.setLoginUserInfo(getLoginUser(token)); companyService.updateStatusById(company); return ApiResponse.success(null); } @ApiOperation("导出Excel") @PostMapping("/exportExcel") @CloudRequiredPermission("business:company:exportExcel") public void exportExcel (@RequestBody PageWrap pageWrap, HttpServletResponse response,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ ExcelExporter.build(Company.class).export(companyService.findPage(pageWrap).getRecords(), "企业信息表", response); } @ApiOperation("根据ID查询") @GetMapping("/{id}") @CloudRequiredPermission("business:company:query") public ApiResponse findById(@PathVariable Integer id,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ return ApiResponse.success(companyService.findById(id)); } @ApiOperation("根据分类ID查询") @PostMapping("/findLaborServicesList") @CloudRequiredPermission("business:company:query") public ApiResponse> findLaborServicesList(Company query,@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ 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") @CloudRequiredPermission("business:company:query") public ApiResponse> findCompanyTreePage(Integer type){ return ApiResponse.success(companyService.findCompanyTreePage(type)); } @ApiOperation("全量部门信息同步") @PostMapping("/syncAll") @CloudRequiredPermission("business:company:sync") public ApiResponse sync(@RequestHeader(Constants.HEADER_USER_TOKEN) String token){ 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(token)); return ApiResponse.success("同步成功"); }else if(Constants.formatIntegerNum(dataSyncConfig.getOrgUserDataOrigin()) == DataSyncConfig.origin.self){ hkSyncOrgUserFromSelfService.syncOrgData(this.getLoginUser(token)); return ApiResponse.success("同步成功"); } return ApiResponse.failed("无效操作"); } }