FWD » Core Development » Base Language

package spoonp2j;

import java.lang.reflect.Method;

import java.util.*;

import spoon.*;

import spoon.processing.*;

import spoon.reflect.code.*;

import spoon.reflect.cu.SourcePosition;

import spoon.reflect.declaration.*;

import spoon.reflect.reference.*;

import spoon.reflect.visitor.chain.*;

import spoon.support.reflect.code.CtVariableReadImpl;

public class CheckForEach

{

   public static void main(String[] args)

   {

      Launcher spoon = new Launcher();

      spoon.addProcessor(new TypeFactoryProcessor());

      spoon.getEnvironment().setIgnoreSyntaxErrors(true);

      spoon.getEnvironment().setSourceClasspath(new String[] { "/home/ca/workspace/p2j4sync/build/lib/p2j.jar" });

      // spoon.addInputResource("/working/workspace/p2j4sync/src/com/goldencode/ast/AnnotatedAst.java");

      spoon.addInputResource("/working/workspace/p2j4sync/src/com/goldencode/");

      spoon.run();

   }

   static class TypeFactoryProcessor

   extends AbstractProcessor<CtMethod>

   {

      @Override

      public void process(CtMethod m)

      {

         CtBlock body = m.getBody();

         if (body == null)

         {

            return;

         }

         if (m.getParent() instanceof CtClass)

         {

            CtClass parent = (CtClass) m.getParent();

            if (parent.getSuperclass() != null && parent.getSuperclass().getSimpleName().equals("ContextLocal"))

            {

               return;

            }

         }

         else if (m.getParent() instanceof CtInterface)

         {

            // System.out.println(m.getParent());

         }

         Map<String, String> params = new HashMap<>();

         for (int i = 0; i < m.getParameters().size(); i++)

         {

            CtParameter p = (CtParameter) m.getParameters().get(i);

            params.put(p.getSimpleName(), p.getType().getSimpleName());

         }

         String type = m.getDeclaringType().getQualifiedName();

         String mthd = type + ":" + m.getSimpleName();

         checkStatement(mthd, body, params);

      }

      private void checkStatement(String mthd, CtStatement st, Map<String, String> params)

      {

         Iterator<CtElement> iter = st.asIterable().iterator();

         while (iter.hasNext())

         {

            CtElement element = iter.next();

            if (element instanceof CtForEach)

            {

               CtForEach f = (CtForEach) element;

               SourcePosition pos = f.getPosition();

               CtExpression expr = f.getExpression();

               CtTypeReference exprType = expr.getType();

               CtType type = exprType.getTypeDeclaration();

               String qtype = type == null ? exprType.toString() : type.getQualifiedName();

               if (qtype.equals(ArrayList.class.getName()) || qtype.equals(List.class.getName()) || qtype.endsWith("[]"))

               {

                  System.out.println(mthd + " " + exprType + " " + pos.getLine());

               }

            }

         }

      }

   }

}