Pull getDirectives out into a util to reduce duplication

Author: cirras

delphi-frontend/src/main/java/au/com/integradev/delphi/antlr/ast/node/AnonymousMethodHeadingNodeImpl.java

@@ -18,17 +18,15 @@
  */
 package au.com.integradev.delphi.antlr.ast.node;
 
+import au.com.integradev.delphi.antlr.ast.node.utils.RoutineDirectiveUtils;
 import au.com.integradev.delphi.antlr.ast.visitors.DelphiParserVisitor;
-import com.google.common.collect.ImmutableSet;
 import java.util.Set;
 import org.antlr.runtime.Token;
 import org.sonar.plugins.communitydelphi.api.ast.AnonymousMethodHeadingNode;
-import org.sonar.plugins.communitydelphi.api.ast.DelphiNode;
 import org.sonar.plugins.communitydelphi.api.ast.RoutineParametersNode;
 import org.sonar.plugins.communitydelphi.api.ast.RoutineReturnTypeNode;
 import org.sonar.plugins.communitydelphi.api.symbol.declaration.RoutineDirective;
 import org.sonar.plugins.communitydelphi.api.symbol.declaration.RoutineKind;
-import org.sonar.plugins.communitydelphi.api.token.DelphiToken;
 
 public final class AnonymousMethodHeadingNodeImpl extends DelphiNodeImpl
     implements AnonymousMethodHeadingNode {
@@ -69,15 +67,7 @@ public RoutineKind getRoutineKind() {
   @Override
   public Set<RoutineDirective> getDirectives() {
     if (directives == null) {
-      var builder = new ImmutableSet.Builder<RoutineDirective>();
-      for (DelphiNode child : getChildren()) {
-        DelphiToken token = child.getToken();
-        RoutineDirective directive = RoutineDirective.fromToken(token);
-        if (directive != null) {
-          builder.add(directive);
-        }
-      }
-      directives = builder.build();
+      directives = RoutineDirectiveUtils.getDirectives(this);
     }
     return directives;
   }

delphi-frontend/src/main/java/au/com/integradev/delphi/antlr/ast/node/RoutineHeadingNodeImpl.java

@@ -18,14 +18,13 @@
  */
 package au.com.integradev.delphi.antlr.ast.node;
 
+import au.com.integradev.delphi.antlr.ast.node.utils.RoutineDirectiveUtils;
 import au.com.integradev.delphi.antlr.ast.visitors.DelphiParserVisitor;
-import com.google.common.collect.ImmutableSet;
 import java.util.Collections;
 import java.util.List;
 import java.util.Set;
 import org.antlr.runtime.Token;
 import org.sonar.plugins.communitydelphi.api.ast.AttributeListNode;
-import org.sonar.plugins.communitydelphi.api.ast.DelphiNode;
 import org.sonar.plugins.communitydelphi.api.ast.FormalParameterNode.FormalParameterData;
 import org.sonar.plugins.communitydelphi.api.ast.RoutineDeclarationNode;
 import org.sonar.plugins.communitydelphi.api.ast.RoutineHeadingNode;
@@ -36,7 +35,6 @@
 import org.sonar.plugins.communitydelphi.api.ast.TypeDeclarationNode;
 import org.sonar.plugins.communitydelphi.api.symbol.declaration.RoutineDirective;
 import org.sonar.plugins.communitydelphi.api.symbol.declaration.RoutineKind;
-import org.sonar.plugins.communitydelphi.api.token.DelphiToken;
 import org.sonar.plugins.communitydelphi.api.token.DelphiTokenType;
 import org.sonar.plugins.communitydelphi.api.type.Type;
 
@@ -194,15 +192,7 @@ public RoutineKind getRoutineKind() {
   @Override
   public Set<RoutineDirective> getDirectives() {
     if (directives == null) {
-      var builder = new ImmutableSet.Builder<RoutineDirective>();
-      for (DelphiNode child : getChildren()) {
-        DelphiToken token = child.getToken();
-        RoutineDirective directive = RoutineDirective.fromToken(token);
-        if (directive != null) {
-          builder.add(directive);
-        }
-      }
-      directives = builder.build();
+      directives = RoutineDirectiveUtils.getDirectives(this);
     }
     return directives;
   }

delphi-frontend/src/main/java/au/com/integradev/delphi/antlr/ast/node/utils/RoutineDirectiveUtils.java

@@ -0,0 +1,25 @@
+package au.com.integradev.delphi.antlr.ast.node.utils;
+
+import com.google.common.collect.ImmutableSet;
+import java.util.Set;
+import org.sonar.plugins.communitydelphi.api.ast.DelphiNode;
+import org.sonar.plugins.communitydelphi.api.symbol.declaration.RoutineDirective;
+import org.sonar.plugins.communitydelphi.api.token.DelphiToken;
+
+public final class RoutineDirectiveUtils {
+  private RoutineDirectiveUtils() {
+    // Utility class
+  }
+
+  public static Set<RoutineDirective> getDirectives(DelphiNode node) {
+    var builder = new ImmutableSet.Builder<RoutineDirective>();
+    for (DelphiNode child : node.getChildren()) {
+      DelphiToken token = child.getToken();
+      RoutineDirective directive = RoutineDirective.fromToken(token);
+      if (directive != null) {
+        builder.add(directive);
+      }
+    }
+    return builder.build();
+  }
+}