完善关系维度解析

hsweb-system/hsweb-system-workflow/hsweb-system-workflow-local/src/main/java/org/hswebframework/web/workflow/dimension/parser/RelationCandidateDimensionParserStrategy.java

@@ -1,25 +1,37 @@
 package org.hswebframework.web.workflow.dimension.parser;
 
+import lombok.SneakyThrows;
 import lombok.extern.slf4j.Slf4j;
-import org.hswebframework.web.organizational.authorization.PersonnelAuthentication;
-import org.hswebframework.web.organizational.authorization.PersonnelAuthenticationManager;
-import org.hswebframework.web.organizational.authorization.relation.Relations;
+import org.hswebframework.expands.script.engine.DynamicScriptEngine;
+import org.hswebframework.expands.script.engine.DynamicScriptEngineFactory;
+import org.hswebframework.web.Maps;
+import org.hswebframework.web.commons.entity.GenericEntity;
+import org.hswebframework.web.organizational.authorization.relation.PersonRelations;
+import org.hswebframework.web.organizational.authorization.relation.Relation;
+import org.hswebframework.web.organizational.authorization.relation.RelationsManager;
 import org.hswebframework.web.workflow.dimension.DimensionContext;
 import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.boot.autoconfigure.condition.ConditionalOnBean;
+import org.springframework.util.DigestUtils;
+import org.springframework.util.StringUtils;
 
 import java.util.Collections;
 import java.util.List;
+import java.util.Objects;
+import java.util.function.Function;
+import java.util.function.Supplier;
+import java.util.stream.Collectors;
 
 /**
  * @author zhouhao
  * @since 3.0.0-RC
  */
 @Slf4j
+@ConditionalOnBean(RelationsManager.class)
 public class RelationCandidateDimensionParserStrategy implements CandidateDimensionParserStrategy {
 
     @Autowired
-    PersonnelAuthenticationManager authenticationManager;
-
+    RelationsManager relationsManager;
 
     @Override
     public boolean support(String dimension) {
@@ -27,41 +39,72 @@ public class RelationCandidateDimensionParserStrategy implements CandidateDimens
     }
 
     @Override
+    @SneakyThrows
+    @SuppressWarnings("all")
     public List<String> parse(DimensionContext context, StrategyConfig config) {
-        String userType = config.getStringConfig("userType")
-                .orElse("creator");
-
-        String user;
-        switch (userType) {
-            case "fixed":
-                user = config.getStringConfig("userId")
-                        .orElse(context.getCreatorId());
-                break;
-            case "pre":
-                user = context.getPreTask().getOwner();
-                break;
-            default:
-                user = context.getCreatorId();
-        }
-
-        PersonnelAuthentication authentication = authenticationManager
-                .getPersonnelAuthorizationByUserId(user);
+        String expression = config.getStringConfig("expression").orElse(null);
+        String expressionLanguage = config.getStringConfig("expressionLanguage").orElse(null);
 
-        if (null == authentication) {
+        if (StringUtils.isEmpty(expression)) {
             return Collections.emptyList();
         }
 
-        Relations relations = authentication.getRelations();
+        String preId = context.getPreTask().getOwner();
+        String creatorId = context.getCreatorId();
 
-        if (relations == null || relations.getAll().isEmpty()) {
-            return Collections.emptyList();
-        }
+        Supplier<PersonRelations> pre = () -> relationsManager.getPersonRelationsByUserId(preId);
 
+        Supplier<PersonRelations> creator = () -> relationsManager.getPersonRelationsByUserId(creatorId);
 
-        return Collections.emptyList();
-    }
+        DynamicScriptEngine engine = DynamicScriptEngineFactory.getEngine(expressionLanguage);
 
-    public static class RelationInfo {
+        if (engine == null) {
+            throw new UnsupportedOperationException("不支持的脚本:" + expressionLanguage);
+        }
+
+        String id = DigestUtils.md5DigestAsHex(expression.getBytes());
+        if (!engine.compiled(id)) {
+            engine.compile(id, expression);
+        }
 
+        Object obj = engine.execute(id, Maps.<String, Object>buildMap()
+                .put("pre", pre)
+                .put("creator", creator)
+                .put("preId", preId)
+                .put("creatorId", creatorId)
+                .put("context", context)
+                .get())
+                .getIfSuccess();
+
+        Function<Object, String> userIdConverter = o -> {
+            if (o instanceof String) {
+                return (String) o;
+            } else if (o instanceof Relation) {
+                Object target = ((Relation) o).getTargetObject();
+                if (target instanceof GenericEntity) {
+                    return String.valueOf(((GenericEntity) target).getId());
+                } else {
+                    return ((Relation) o).getRelation();
+                }
+            } else {
+                log.warn("不支持的关系:{}", o);
+                return null;
+            }
+        };
+
+        if (obj instanceof List) {
+            List<Object> list = ((List) obj);
+            return list.stream()
+                    .map(userIdConverter)
+                    .filter(Objects::nonNull)
+                    .collect(Collectors.toList());
+        } else {
+            String result = userIdConverter.apply(obj);
+            if (result == null) {
+                return Collections.emptyList();
+            }
+            return Collections.singletonList(result);
+        }
     }
+
 }