/*
* @test /nodynamiccopyright/
* @bug 6406771
* @summary CompilationUnitTree needs access to a line map
*/
// WARNING: White-space and layout is important in this file, especially tab characters.
import java.io.*;
import java.util.*;
import javax.annotation.processing.*;
import javax.lang.model.*;
import javax.lang.model.element.*;
import javax.tools.*;
import com.sun.tools.javac.api.*;
import com.sun.source.tree.*;
import com.sun.source.util.*;
import com.sun.tools.javac.tree.JCTree;
@SupportedAnnotationTypes("*")
public class T6406771 extends AbstractProcessor {
String[] tests = {
"line:24",
"line:25",
"line:26", "line:26",
// 1 2 3 4 5 6
//3456789012345678901234567890123456789012345678901234567890
"col:7", "col:16", "col:26", // this line uses spaces
"col:9", "col:25", "col:41", // this line uses tabs
"col:20", "col:43" // this line uses a mixture
};
// White-space after this point does not matter
public static void main(String[] args) {
String self = T6406771.class.getName();
String testSrc = System.getProperty("test.src");
String testClasses = System.getProperty("test.classes");
JavacTool tool = JavacTool.create();
StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
JavaFileObject f = fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, self+".java"))).iterator().next();
List<String> opts = Arrays.asList("-d", ".", "-processorpath", testClasses, "-processor", self, "-proc:only");
JavacTask task = tool.getTask(null, fm, null, opts, null, Arrays.asList(f));
if (!task.call())
throw new AssertionError("failed");
}
public boolean process(Set<? extends TypeElement> elems, RoundEnvironment rEnv) {
final String LINE = "line" + ':'; // avoid matching this string
final String COLUMN = "col" + ':';
final Messager messager = processingEnv.getMessager();
final Trees trees = Trees.instance(processingEnv);
TreeScanner<Void,LineMap> s = new TreeScanner<Void,LineMap>() {
public Void visitLiteral(LiteralTree tree, LineMap lineMap) {
if (tree.getKind() == Tree.Kind.STRING_LITERAL) {
String s = (String) tree.getValue();
int pos = ((JCTree) tree).pos; // can't get through public api, bug 6412669 filed
String prefix;
long found;
if (s.startsWith(LINE)) {
prefix = LINE;
found = lineMap.getLineNumber(pos);
}
else if (s.startsWith(COLUMN)) {
prefix = COLUMN;
found = lineMap.getColumnNumber(pos);
}
else
return null;
int expect = Integer.parseInt(s.substring(prefix.length()));
if (expect != found) {
messager.printMessage(Diagnostic.Kind.ERROR,
"Error: " + prefix + " pos=" + pos
+ " expect=" + expect + " found=" + found);
}
}
return null;
}
};
for (Element e: rEnv.getRootElements()) {
System.err.println(e);
Tree t = trees.getTree(e);
TreePath p = trees.getPath(e);
s.scan(p.getLeaf(), p.getCompilationUnit().getLineMap());
}
return true;
}
@Override
public SourceVersion getSupportedSourceVersion() {
return SourceVersion.latest();
}
}