1 /*
   2  * Copyright 2006 Sun Microsystems, Inc.  All Rights Reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Sun Microsystems, Inc., 4150 Network Circle, Santa Clara,
  20  * CA 95054 USA or visit www.sun.com if you need additional information or
  21  * have any questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @bug 6412669
  27  * @summary Should be able to get SourcePositions from 269 world
  28  */
  29 
  30 import java.io.*;
  31 import java.util.*;
  32 import javax.annotation.*;
  33 import javax.annotation.processing.*;
  34 import javax.lang.model.*;
  35 import javax.lang.model.element.*;
  36 import javax.tools.*;
  37 import com.sun.source.util.*;
  38 import com.sun.tools.javac.api.*;
  39 
  40 @SupportedAnnotationTypes("*")
  41 public class T6412669 extends AbstractProcessor {
  42     public static void main(String... args) throws IOException {
  43         String testSrc = System.getProperty("test.src", ".");
  44         String testClasses = System.getProperty("test.classes", ".");
  45 
  46         JavacTool tool = JavacTool.create();
  47         StandardJavaFileManager fm = tool.getStandardFileManager(null, null, null);
  48         fm.setLocation(StandardLocation.CLASS_PATH, Arrays.asList(new File(testClasses)));
  49         Iterable<? extends JavaFileObject> files =
  50             fm.getJavaFileObjectsFromFiles(Arrays.asList(new File(testSrc, T6412669.class.getName()+".java")));
  51         String[] opts = { "-proc:only", "-processor", T6412669.class.getName(),
  52                           "-classpath", new File(testClasses).getPath() };
  53         JavacTask task = tool.getTask(null, fm, null, Arrays.asList(opts), null, files);
  54         if (!task.call())
  55             throw new AssertionError("test failed");
  56     }
  57 
  58     public boolean process(Set<? extends TypeElement> annotations, RoundEnvironment roundEnv) {
  59         Trees trees = Trees.instance(processingEnv);
  60         SourcePositions sp = trees.getSourcePositions();
  61         Messager m = processingEnv.getMessager();
  62         for (TypeElement anno: annotations) {
  63             for (Element e: roundEnv.getElementsAnnotatedWith(anno)) {
  64                 TreePath p = trees.getPath(e);
  65                 long start = sp.getStartPosition(p.getCompilationUnit(), p.getLeaf());
  66                 long end = sp.getEndPosition(p.getCompilationUnit(), p.getLeaf());
  67                 Diagnostic.Kind k = (start > 0 && end > 0 && start < end
  68                                      ? Diagnostic.Kind.NOTE : Diagnostic.Kind.ERROR);
  69                 m.printMessage(k, "test [" + start + "," + end + "]", e);
  70             }
  71         }
  72         return true;
  73     }
  74 
  75     @Override
  76     public SourceVersion getSupportedSourceVersion() {
  77         return SourceVersion.latest();
  78     }
  79 }