1.1 --- /dev/null Thu Jan 01 00:00:00 1970 +0000
1.2 +++ b/test/tools/javac/diags/DocCommentProcessor.java Wed Apr 27 01:34:52 2016 +0800
1.3 @@ -0,0 +1,114 @@
1.4 +/*
1.5 + * Copyright (c) 2012, Oracle and/or its affiliates. All rights reserved.
1.6 + * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
1.7 + *
1.8 + * This code is free software; you can redistribute it and/or modify it
1.9 + * under the terms of the GNU General Public License version 2 only, as
1.10 + * published by the Free Software Foundation.
1.11 + *
1.12 + * This code is distributed in the hope that it will be useful, but WITHOUT
1.13 + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
1.14 + * FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
1.15 + * version 2 for more details (a copy is included in the LICENSE file that
1.16 + * accompanied this code).
1.17 + *
1.18 + * You should have received a copy of the GNU General Public License version
1.19 + * 2 along with this work; if not, write to the Free Software Foundation,
1.20 + * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
1.21 + *
1.22 + * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
1.23 + * or visit www.oracle.com if you need additional information or have any
1.24 + * questions.
1.25 + */
1.26 +
1.27 +import java.util.*;
1.28 +
1.29 +import javax.annotation.processing.*;
1.30 +import javax.lang.model.*;
1.31 +import javax.lang.model.element.*;
1.32 +
1.33 +import com.sun.source.doctree.DocCommentTree;
1.34 +import com.sun.source.doctree.ErroneousTree;
1.35 +import com.sun.source.tree.ClassTree;
1.36 +import com.sun.source.tree.MethodTree;
1.37 +import com.sun.source.tree.VariableTree;
1.38 +import com.sun.source.util.DocTreeScanner;
1.39 +import com.sun.source.util.DocTrees;
1.40 +import com.sun.source.util.TreePath;
1.41 +import com.sun.source.util.TreePathScanner;
1.42 +import com.sun.source.util.TreeScanner;
1.43 +import com.sun.tools.javac.tree.DocPretty;
1.44 +import java.io.PrintWriter;
1.45 +import javax.tools.Diagnostic;
1.46 +
1.47 +/**
1.48 + * Standard annotation processor for use by examples to
1.49 + * scan DocCommentTree nodes looking for ErroneousTree,
1.50 + * on which to call {@code getMessage}.
1.51 + */
1.52 +@SupportedAnnotationTypes("*")
1.53 +public class DocCommentProcessor extends AbstractProcessor {
1.54 + @Override
1.55 + public SourceVersion getSupportedSourceVersion() {
1.56 + return SourceVersion.latest();
1.57 + }
1.58 +
1.59 + @Override
1.60 + public void init(ProcessingEnvironment pEnv) {
1.61 + super.init(pEnv);
1.62 + trees = DocTrees.instance(pEnv);
1.63 + messager = pEnv.getMessager();
1.64 + }
1.65 +
1.66 + @Override
1.67 + public boolean process(Set<? extends TypeElement> annos, RoundEnvironment rEnv) {
1.68 + for (Element e : rEnv.getRootElements()) {
1.69 + new DocCommentScanner().scan(e);
1.70 + }
1.71 + return true;
1.72 + }
1.73 +
1.74 + class DocCommentScanner extends TreePathScanner<Void,Void> {
1.75 + public void scan(Element e) {
1.76 + scan(trees.getPath(e), null);
1.77 + }
1.78 +
1.79 + @Override
1.80 + public Void visitClass(ClassTree tree, Void ignore) {
1.81 + check();
1.82 + return super.visitClass(tree, ignore);
1.83 + }
1.84 +
1.85 + @Override
1.86 + public Void visitMethod(MethodTree tree, Void ignore) {
1.87 + check();
1.88 + return super.visitMethod(tree, ignore);
1.89 + }
1.90 +
1.91 + @Override
1.92 + public Void visitVariable(VariableTree tree, Void ignore) {
1.93 + check();
1.94 + return super.visitVariable(tree, ignore);
1.95 + }
1.96 +
1.97 + private void check() {
1.98 + DocCommentTree dc = trees.getDocCommentTree(getCurrentPath());
1.99 + if (dc == null)
1.100 + return;
1.101 +
1.102 + DocTreeScanner<Void, Void> s = new DocTreeScanner<Void, Void>() {
1.103 + @Override
1.104 + public Void visitErroneous(ErroneousTree tree, Void ignore) {
1.105 + messager.printMessage(Diagnostic.Kind.NOTE, tree.getDiagnostic().getMessage(null));
1.106 + return null;
1.107 + }
1.108 + };
1.109 +
1.110 + s.scan(dc, null);
1.111 + }
1.112 +
1.113 + }
1.114 +
1.115 + private DocTrees trees;
1.116 + private Messager messager;
1.117 +}
