RemoveLogs.java

import com.github.javaparser.StaticJavaParser;
import com.github.javaparser.ast.CompilationUnit;
import com.github.javaparser.ast.expr.MethodCallExpr;
import com.github.javaparser.ast.visitor.ModifierVisitor;
import com.github.javaparser.ast.visitor.Visitable;

import java.io.FileInputStream;
import java.io.FileNotFoundException;
import java.util.HashSet;
import java.util.Set;

public class Main {

	static Set<String> logMethods = new HashSet<>();
	static Set<String> logClasses = new HashSet<>();

	static {
		logMethods.add("info");
		logMethods.add("debug");
		logMethods.add("warn");
		logMethods.add("error");
		logMethods.add("fatal");
		logMethods.add("trace");

		logClasses.add("log");

	}

	public static void main(String[] args) throws FileNotFoundException {

		CompilationUnit cu = StaticJavaParser.parse(new FileInputStream("/Users/mauricioaniche/workspace/logremoval/fixture/A.java"));

		cu.accept(new ModifierVisitor<Void>() {
			@Override
			public Visitable visit(final MethodCallExpr n, Void arg) {

				boolean isALogClass = n.getScope().isPresent() && (logClasses.stream().anyMatch(x -> n.getScope().get().toString().startsWith(x)));
				boolean isALogMethod = logMethods.contains(n.getName().asString());

				if(isALogClass && isALogMethod) {
					System.out.println("log call!");

					if(n.getParentNode().isPresent())
						n.getParentNode().get().remove();
				}

				return super.visit(n, arg);
			}
		}, null);

		System.out.println(cu.toString());

	}
}