权限初始化

haoban-manage3-service/src/main/java/com/gic/haoban/manage/service/service/out/impl/role/HaobanRoleApiServiceImpl.java

@@ -30,6 +30,7 @@ import com.gic.haoban.manage.service.service.content.adaptor.MaterialEnterpriseA
 import com.gic.haoban.manage.service.service.role.HaobanMenuService;
 import com.gic.haoban.manage.service.service.role.HaobanRoleMenuService;
 import com.gic.haoban.manage.service.service.role.HaobanRoleService;
+import com.google.common.collect.Lists;
 import org.apache.commons.collections.CollectionUtils;
 import org.apache.commons.lang3.StringUtils;
 import org.apache.logging.log4j.LogManager;
@@ -579,18 +580,18 @@ public class HaobanRoleApiServiceImpl implements HaobanRoleApiService {
     static {
         // 导购
         menuCodeMap.put("0" + ContentMaterialROleInitQDTO.Version.LOW.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab",
-                "goodsDetail_material", "goodsCenter_Count", "salesClue", "customerDetail_interact"));
+                "goodsDetail_material", "goodsCenter_Count", "salesClue", "customerDetail_interact","AICreate"));
         menuCodeMap.put("0" + ContentMaterialROleInitQDTO.Version.HIGH.getCode(), Arrays.asList("materialMain","commodityCenter", "commodityCenterTab", "materialCenterTab",
-                "goodsDetail_material", "goodsCenter_Count", "salesClue", "customerDetail_interact", "publishMaterial", "userCenter_createCenter"));
+                "goodsDetail_material", "goodsCenter_Count", "salesClue", "customerDetail_interact", "publishMaterial", "userCenter_createCenter","AICreate"));
         // 店长
-        menuCodeMap.put("1" + ContentMaterialROleInitQDTO.Version.LOW.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab", "goodsDetail_material", "goodsCenter_Count", "materialManage", "salesClue", "customerDetail_interact"));
+        menuCodeMap.put("1" + ContentMaterialROleInitQDTO.Version.LOW.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab", "goodsDetail_material", "goodsCenter_Count", "materialManage", "salesClue", "customerDetail_interact","AICreate"));
         menuCodeMap.put("1" + ContentMaterialROleInitQDTO.Version.HIGH.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab",
-                "goodsDetail_material", "goodsCenter_Count", "materialManage", "salesClue", "customerDetail_interact", "publishMaterial", "userCenter_createCenter"));
+                "goodsDetail_material", "goodsCenter_Count", "materialManage", "salesClue", "customerDetail_interact", "publishMaterial", "userCenter_createCenter","AICreate"));
 
         // 区经
-        menuCodeMap.put("2" + ContentMaterialROleInitQDTO.Version.LOW.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab", "goodsDetail_material", "goodsCenter_Count", "materialManage"));
+        menuCodeMap.put("2" + ContentMaterialROleInitQDTO.Version.LOW.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab", "goodsDetail_material", "goodsCenter_Count", "materialManage","AICreate"));
         menuCodeMap.put("2" + ContentMaterialROleInitQDTO.Version.HIGH.getCode(), Arrays.asList("materialMain", "commodityCenter", "commodityCenterTab", "materialCenterTab",
-                "goodsDetail_material", "goodsCenter_Count", "materialManage", "publishMaterial", "userCenter_createCenter"));
+                "goodsDetail_material", "goodsCenter_Count", "materialManage", "publishMaterial", "userCenter_createCenter","AICreate"));
     }
 
 
@@ -604,17 +605,20 @@ public class HaobanRoleApiServiceImpl implements HaobanRoleApiService {
     public ServiceResponse<Void> flushRightOneTime(String params) {
 
         logger.info("flushRightOneTime 刷新企业内容权限:{}", params);
+        List<String> enterpriseIds1 = materialEnterpriseAdaptor.queryHasRightEnterpriseIds(MaterialEnterpriseAdaptor.MaterialLevel.LOW.getCode());
+
         // 获取所有开通内容高级版的企业
-        List<String> enterpriseIds = materialEnterpriseAdaptor.queryHasRightEnterpriseIds(MaterialEnterpriseAdaptor.MaterialLevel.HIGH.getCode());
+        List<String> enterpriseIds2 = materialEnterpriseAdaptor.queryHasRightEnterpriseIds(MaterialEnterpriseAdaptor.MaterialLevel.HIGH.getCode());
+        List<String> enterpriseIds = new ArrayList<>();
+        enterpriseIds.addAll(enterpriseIds1);
+        enterpriseIds.addAll(enterpriseIds2);
         // 执行刷新权限
         String enterpriseId = "";
         if (StringUtils.isNotBlank(params) && !StringUtils.equals("-1", params)) {
             enterpriseId = params;
         }
-
         List<String> menuCodes = new ArrayList<>();
-        menuCodes.add("publishMaterial");
-        menuCodes.add("userCenter_createCenter");
+        menuCodes.add("AICreate");
 
         for (String tempId : enterpriseIds) {
             if (StringUtils.isNotBlank(enterpriseId) && !StringUtils.equalsIgnoreCase(enterpriseId, tempId)) {

haoban-manage3-service/src/test/java/ServiceTest.java

@@ -4,7 +4,9 @@ import com.gic.haoban.manage.api.service.StaffApiService;
 import com.gic.haoban.manage.api.service.WxEnterpriseRelatedApiService;
 import com.gic.haoban.manage.api.service.fee.HaobanQywxFeeApiService;
 import com.gic.haoban.manage.api.service.hm.HmQrcodeApiService;
+import com.gic.haoban.manage.api.service.role.HaobanRoleApiService;
 import com.gic.haoban.manage.service.config.Config;
+import com.gic.haoban.manage.service.service.out.impl.role.HaobanRoleApiServiceImpl;
 import com.gic.wechat.api.service.qywx.QywxDepartmentApiService;
 import com.gic.wechat.api.service.qywx.QywxSuiteApiService;
 import com.gic.wechat.api.service.qywx.QywxUserApiService;
@@ -47,6 +49,8 @@ public class ServiceTest {
 
     @Autowired
     private StaffApiService staffApiService;
+    @Autowired
+    private HaobanRoleApiService haobanRoleApiService;
 
 
     @Test
@@ -72,4 +76,9 @@ public class ServiceTest {
         String autoActiveStatus = qywxUserApiService.getAutoActiveStatus("wp59NLDQAAJL6DsM6YwOGqJlhCBykkeA", config.getCorpid());
         System.out.println(autoActiveStatus);
     }
+
+    @Test
+    public void testRole() {
+        haobanRoleApiService.flushRightOneTime("");
+    }
 }