com.sun.source.tree.TreeVisitor Java Examples

public static void parse(Path moduleInfoPath, ModuleClassVisitor moduleClassVisitor) throws IOException {
  var compiler = ToolProvider.getSystemJavaCompiler();
  try(var fileManager = compiler.getStandardFileManager(null, null, null)) {
    var compilationUnits = fileManager.getJavaFileObjects(moduleInfoPath);
    var task = compiler.getTask(null, fileManager, null, null, null, compilationUnits);
    var javacTask = (JavacTask)task;
    var units = javacTask.parse();
    var unit = units.iterator().next();

    var moduleHandler = new ModuleHandler(moduleClassVisitor);
    var visitor = (TreeVisitor<?,?>)Proxy.newProxyInstance(TreeVisitor.class.getClassLoader(), new Class<?>[]{ TreeVisitor.class},
        (proxy, method, args) -> {
          ModuleHandler.METHOD_MAP
          .getOrDefault(method.getName(), (handler, node, v) -> { 
            throw new AssertionError("invalid node " + node.getClass());
          })
          .visit(moduleHandler, (Tree)args[0], (TreeVisitor<?,?>)proxy);
          return null;
        });

    unit.accept(visitor, null);
  }
}
 

public <R, D> R accept(TreeVisitor<R, D> arg0, D arg1) {
    R ret = null;
    for (JCVariableDecl v : vars) {
        ret = v.accept(arg0, arg1);
    }
    return ret;
}
 

@SuppressWarnings("static-method")
public void visitCompilationUnit(CompilationUnitTree node, TreeVisitor<?, ?> visitor) {
  for(var directive: node.getTypeDecls()) {
    if (!(directive instanceof ModuleTree)) {  // skip unnecessary nodes: imports, etc
      continue;
    }
    accept(visitor, directive);
  }
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitSynchronized(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitUnary(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitPrimitiveType(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitAssert(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitMethod(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitUnionType(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return expression().accept(visitor, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitTypeCast(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitMemberSelect(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitConditionalExpression(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitBreak(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitCompoundAssignment(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitContinue(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitCase(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitLiteral(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitTypeCast(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitArrayAccess(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitInstanceOf(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> visitor, D data) {
  return visitor.visitAssignment(this, data);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitModifiers(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitArrayAccess(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    throw new AssertionError("TypeBoundKind is not part of a public API");
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitWildcard(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitTypeParameter(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitUnionType(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitParameterizedType(this, d);
}
 

@Override
public <R, D> R accept(TreeVisitor<R, D> v, D d) {
    return v.visitArrayType(this, d);
}