REPORT-59919 && REPORT-59921 && REPORT-59922

designer-base/src/main/java/com/fr/design/mainframe/authority/FormulaAuthorityChecker.java

@@ -1,81 +1,97 @@
 package com.fr.design.mainframe.authority;
 
 import com.fr.base.Formula;
-import com.fr.stable.StringUtils;
+import com.fr.general.ComparatorUtils;
+import com.fr.parser.FunctionCall;
+import com.fr.parser.StringLiteral;
+import com.fr.script.Calculator;
+import com.fr.stable.script.Node;
 import org.jetbrains.annotations.Nullable;
 
-import java.util.Arrays;
 import java.util.HashSet;
 import java.util.Set;
-import java.util.regex.Matcher;
-import java.util.regex.Pattern;
 
 public class FormulaAuthorityChecker extends ElementAuthorityChecker<Formula> {
-    private static final Set<FormulaPattern> CONNECTION_NAME_FORMULA_PATTERN = new HashSet<>();
-    private static final Set<FormulaPattern> DATASET_NAME_FORMULA_PATTERN = new HashSet<>();
+    private static final Set<FormulaParser> CONNECTION_NAME_FORMULA_PARSER = new HashSet<>();
+    private static final Set<FormulaParser> DATASET_NAME_FORMULA_PARSER = new HashSet<>();
 
     static {
-        CONNECTION_NAME_FORMULA_PATTERN.add(new FormulaPattern("^=SQL\\(\"(.+?)\",", 1));
-        DATASET_NAME_FORMULA_PATTERN.add(new FormulaPattern("^=VALUE\\(\"(.+?)\",", 1));
+        CONNECTION_NAME_FORMULA_PARSER.add(new FormulaParser("SQL", 0));
+        DATASET_NAME_FORMULA_PARSER.add(new FormulaParser("VALUE", 0));
     }
 
     @Override
     @Nullable
     public Set<String> getNoAuthConnectionNames(Formula formula, Set<String> authConnectionNames) {
-        return getNoAuthNames(formula, CONNECTION_NAME_FORMULA_PATTERN, authConnectionNames);
+        return getNoAuthNames(formula, CONNECTION_NAME_FORMULA_PARSER, authConnectionNames);
     }
 
+
     @Override
     @Nullable
     Set<String> getNoAuthDatasetNames(Formula formula, Set<String> authDatasetNames) {
-        return getNoAuthNames(formula, DATASET_NAME_FORMULA_PATTERN, authDatasetNames);
+        return getNoAuthNames(formula, DATASET_NAME_FORMULA_PARSER, authDatasetNames);
     }
 
-    private Set<String> getNoAuthNames(Formula formula, Set<FormulaPattern> formulaPatterns, Set<String> authNames) {
+    private Set<String> getNoAuthNames(Formula formula, Set<FormulaParser> formulaParsers, Set<String> authNames) {
         Set<String> noAuthNames = new HashSet<>();
-        for (FormulaPattern formulaPattern : formulaPatterns) {
-            String noAuthName = getNoAuthName(formula, formulaPattern, authNames);
+        try {
+            FunctionCall functionCall = (FunctionCall) Calculator.createCalculator().parse(formula.getContent()).getConditionalExpression();
+            handleNoAuthNames(functionCall, formulaParsers, authNames, noAuthNames);
+        } catch (Exception ignore) {
+
+        } finally {
+            return noAuthNames;
+        }
+    }
+
+    private void handleNoAuthNames(FunctionCall functionCall, Set<FormulaParser> formulaParsers, Set<String> authNames, Set<String> noAuthNames) {
+        for (FormulaParser formulaPattern : formulaParsers) {
+            String noAuthName = formulaPattern.getNoAuthName(functionCall, authNames);
             if (noAuthName != null) {
                 noAuthNames.add(noAuthName);
             }
         }
-        return noAuthNames;
+        Node[] nodes = functionCall.getArguments();
+        if (nodes != null) {
+            for (int i = 0; i < nodes.length; i++) {
+                Node node = nodes[i];
+                if (node instanceof FunctionCall) {
+                    handleNoAuthNames((FunctionCall) node, formulaParsers, authNames, noAuthNames);
                 }
-
-    private String getNoAuthName(Formula formula, FormulaPattern formulaPattern, Set<String> authNames) {
-        String parameter = getFormulaParameter(formula, formulaPattern);
-        if (!authNames.contains(parameter)) {
-            return parameter;
             }
-        return null;
         }
 
-    private String getFormulaParameter(Formula formula, FormulaPattern formulaPattern) {
-        String content = transformFormulaContent(formula.getContent());
-        Matcher matcher = formulaPattern.pattern.matcher(content);
-        if (matcher.find()) {
-            return matcher.group(formulaPattern.index);
-        }
-        return null;
     }
 
-    private String transformFormulaContent(String content) {
-        int index = content.indexOf('(');
-        if (index > 0) {
-            return content.substring(0, index).toUpperCase() + content.substring(index);
-        } else {
-            return StringUtils.EMPTY;
-        }
-    }
 
-    static class FormulaPattern {
-        public Pattern pattern;
+    static class FormulaParser {
+        //函数的名称
+        public String name;
+        //要检测的位置
         public int index;
 
-        FormulaPattern(String regex, int index) {
-            this.pattern = Pattern.compile(regex);
+
+        FormulaParser(String name, int index) {
+            this.name = name;
             this.index = index;
         }
+
+        String getNoAuthName(FunctionCall functionCall, Set<String> authNames) {
+            if (functionCall.getName() != null && ComparatorUtils.equals(functionCall.getName().toUpperCase(), name)) {
+                Node node = functionCall.getArguments()[index];
+                if (node instanceof StringLiteral) {
+                    String stringLiteral = node.toString();
+                    if (stringLiteral.length() > 2) {
+                        String value = stringLiteral.substring(1, stringLiteral.length() - 1);
+                        if (!authNames.contains(value)) {
+                            return value;
+                        }
+                    }
+                }
+            }
+            return null;
+        }
     }
 
 }