订货单：上传质保书按钮显示权限

@@ -338,6 +338,12 @@ public class GetPurchaseOrderInfoBo extends BaseBo<PurchaseOrderInfo> {
     @ApiModelProperty("质保书名称")
     private String warrantyCertificateFileName;
 
+    /**
+     * 是否展示上传按钮(非持久化字段)
+     */
+    @ApiModelProperty("是否展示上传按钮")
+    private boolean showUpload = false;
+
 
     public GetPurchaseOrderInfoBo() {
 

@@ -87,6 +87,9 @@ public class PurchaseOrderInfoController extends DefaultBaseController {
     private TransactorHandler transactorHandler;
     @Resource
     private ContractDistributorStandardService contractDistributorStandardService;
+    @Resource
+    private SysRoleService sysRoleService;
+
 
 
     /**
@@ -104,7 +107,8 @@ public class PurchaseOrderInfoController extends DefaultBaseController {
 
         if (!CollectionUtil.isEmpty(datas)) {
             String currentUserId = SecurityUtil.getCurrentUser().getId();
-
+            List<SysRole> roles = sysRoleService.getByUserId(currentUserId);
+            List<String> roleCodes = roles.stream().map(SysRole::getCode).collect(Collectors.toList());
             results = datas.stream().map(purchaseOrderInfo -> {
                 GetPurchaseOrderInfoBo bo = new GetPurchaseOrderInfoBo(purchaseOrderInfo);
                 // 判断 contractCreateById 是否等于当前用户 ID
@@ -113,6 +117,16 @@ public class PurchaseOrderInfoController extends DefaultBaseController {
                     isCreateBy = true;
                 }
                 bo.setContractCreateBy(isCreateBy);
+                // 最终精简版代码
+                if (CollectionUtils.isNotEmpty(roleCodes)) {
+                    String workshopCode = purchaseOrderInfo.getWorkshopCode();
+                    // 只有当车间代码在白名单内，且用户拥有对应权限时，才显示上传
+                    if (Arrays.asList("yfc", "efc", "sfc", "ztfc").contains(workshopCode)) {
+                        if (roleCodes.contains(workshopCode + "pzglkzjy")) {
+                            bo.setShowUpload(true);
+                        }
+                    }
+                }
                 return bo;
             }).collect(Collectors.toList());
         }