de.dagere.peass.testtransformation.ParseUtil Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of dependency Show documentation

There is a newer version: 0.4.0

package de.dagere.peass.testtransformation;

import java.util.LinkedList;
import java.util.List;

import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.Node;
import com.github.javaparser.ast.body.ClassOrInterfaceDeclaration;
import com.github.javaparser.ast.body.EnumDeclaration;

/**
 * Helper class for parsing Java code with Javaparser.
 * 
 * @author reichelt
 *
 */
public final class ParseUtil {

   /**
    * Prive Constructor - Do not Initialize Helper class.
    */
   private ParseUtil() {

   }

   public static List getClasses(final CompilationUnit unit) {
      List clazzes = new LinkedList<>();
      for (final Node node : unit.getChildNodes()) {
         if (node instanceof ClassOrInterfaceDeclaration) {
            clazzes.add((ClassOrInterfaceDeclaration) node);
         }
      }
      return clazzes;
   }

   public static List getEnums(final CompilationUnit unit) {
      List enums = new LinkedList<>();
      for (final Node node : unit.getChildNodes()) {
         if (node instanceof EnumDeclaration) {
            enums.add((EnumDeclaration) node);
         }
      }
      return enums;
   }
}