org.walkmod.sonar.visitors.ReplacePrintStacktraceForLogs Maven / Gradle / Ivy

Go to download
package org.walkmod.sonar.visitors;

import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;

import org.walkmod.javalang.ASTManager;
import org.walkmod.javalang.ParseException;
import org.walkmod.javalang.ast.CompilationUnit;
import org.walkmod.javalang.ast.Node;
import org.walkmod.javalang.ast.SymbolData;
import org.walkmod.javalang.ast.SymbolDefinition;
import org.walkmod.javalang.ast.body.BodyDeclaration;
import org.walkmod.javalang.ast.body.FieldDeclaration;
import org.walkmod.javalang.ast.body.ModifierSet;
import org.walkmod.javalang.ast.body.TypeDeclaration;
import org.walkmod.javalang.ast.body.VariableDeclarator;
import org.walkmod.javalang.ast.body.VariableDeclaratorId;
import org.walkmod.javalang.ast.expr.AnnotationExpr;
import org.walkmod.javalang.ast.expr.Expression;
import org.walkmod.javalang.ast.expr.MarkerAnnotationExpr;
import org.walkmod.javalang.ast.expr.MethodCallExpr;
import org.walkmod.javalang.ast.expr.NameExpr;
import org.walkmod.javalang.ast.type.ClassOrInterfaceType;
import org.walkmod.javalang.compiler.symbols.RequiresSemanticAnalysis;
import org.walkmod.javalang.visitors.VoidVisitorAdapter;
import org.walkmod.walkers.VisitorContext;

import com.alibaba.fastjson.JSONArray;

@RequiresSemanticAnalysis
public class ReplacePrintStacktraceForLogs extends VoidVisitorAdapter {

   private String logVar = "log";

   private String logType = "java.util.logging.Logger";

   private List annotations = new LinkedList();

   private Expression init = null;
   
   private String logMethod = "severe";
   
   private boolean attachTheException = false;
   
   public void setLogVar(String logVar){
      this.logVar = logVar;
   }
   
   public void setLogType(String logType){
      this.logType = logType;
   }
   
   public void setInitExpression(String initExpression) throws ParseException {
      if (initExpression != null) {
         init = (Expression) ASTManager.parse(Expression.class, initExpression);
      }
   }
   
   public void setLogMethod(String logMethod){
      this.logMethod = logMethod;
   }
   
   public void setAttachTheException(boolean attachTheException){
      this.attachTheException = attachTheException;
   }

   private Expression getInit(TypeDeclaration td) {
      if ("java.util.logging.Logger".equals(logType) && init == null) {
         List args = new LinkedList();
         args.add(new NameExpr(td.getName()));
         init = new MethodCallExpr(new NameExpr(logType), "getLogger", args);
      }
      return init;
   }

   public void setAnnotationExpressions(JSONArray annotations) {
      if (annotations != null) {
         Iterator