1 /* 2 * Copyright (c) 2017, Oracle and/or its affiliates. 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 Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA 20 * or visit www.oracle.com if you need additional information or have any 21 * questions. 22 */ 23 24 import java.io.PrintStream; 25 import java.nio.file.Path; 26 import java.util.EnumSet; 27 import java.util.List; 28 import java.util.Set; 29 30 import javax.lang.model.element.ModuleElement; 31 import javax.lang.model.element.ModuleElement.Directive; 32 import javax.lang.model.element.ModuleElement.DirectiveKind; 33 import javax.lang.model.element.ModuleElement.DirectiveVisitor; 34 import javax.lang.model.element.ModuleElement.ExportsDirective; 35 import javax.lang.model.element.ModuleElement.OpensDirective; 36 import javax.lang.model.element.ModuleElement.ProvidesDirective; 37 import javax.lang.model.element.ModuleElement.RequiresDirective; 38 import javax.lang.model.element.ModuleElement.UsesDirective; 39 import javax.tools.JavaCompiler; 40 import javax.tools.JavaFileObject; 41 import javax.tools.StandardJavaFileManager; 42 import javax.tools.ToolProvider; 43 44 import com.sun.source.util.JavacTask; 45 46 /* 47 * @test 48 * @bug 8175118 49 * @summary Add ModuleElement.DirectiveVisitor 50 * @library /tools/lib 51 * @modules jdk.compiler/com.sun.tools.javac.api 52 * jdk.compiler/com.sun.tools.javac.main 53 * jdk.compiler/com.sun.tools.javac.util 54 * @build toolbox.ToolBox ModuleTestBase 55 * @run main DirectiveVisitorTest 56 */ 57 58 public class DirectiveVisitorTest extends ModuleTestBase { main(String... args)59 public static void main(String... args) throws Exception { 60 new DirectiveVisitorTest().runTests(); 61 } 62 63 @Test testVisitor(Path base)64 public void testVisitor(Path base) throws Exception { 65 Path src = base.resolve("src"); 66 Path src_m1 = src.resolve("m1x"); 67 tb.writeJavaFiles(src_m1, 68 "module m1x { " 69 + " requires m2x;" 70 + " exports p1;" 71 + " opens p2;" 72 + " uses p1.Service;" 73 + " provides p1.Service with p2.Impl;" 74 + "}", 75 "package p1; public interface Service { }", 76 "package p2; public class Impl implements p1.Service { }"); 77 Path src_m2 = src.resolve("m2x"); 78 tb.writeJavaFiles(src_m2, 79 "module m2x { }"); 80 81 Path modules = base.resolve("modules"); 82 tb.createDirectories(modules); 83 84 JavaCompiler javac = ToolProvider.getSystemJavaCompiler(); 85 try (StandardJavaFileManager fm = javac.getStandardFileManager(null, null, null)) { 86 Iterable<? extends JavaFileObject> files = fm.getJavaFileObjects(findJavaFiles(src)); 87 List<String> options = List.of( 88 "--module-source-path", src.toString(), 89 "-d", modules.toString() 90 ); 91 JavacTask t = (JavacTask) javac.getTask(null, fm, null, options, null, files); 92 t.analyze(); 93 ModuleElement e = t.getElements().getModuleElement("m1x"); 94 Set<DirectiveKind> kinds = EnumSet.<DirectiveKind>allOf(DirectiveKind.class); 95 Visitor v = new Visitor(); 96 v.visit(e, kinds); 97 if (!kinds.equals(EnumSet.<DirectiveKind>noneOf(DirectiveKind.class))) { 98 error("Some kinds not found: " + kinds); 99 } 100 } 101 } 102 103 static class Visitor implements DirectiveVisitor<Void,Set<DirectiveKind>> { 104 private final PrintStream out = System.err; 105 visit(ModuleElement e, Set<DirectiveKind> kinds)106 public void visit(ModuleElement e, Set<DirectiveKind> kinds) { 107 e.getDirectives().stream().forEach(d -> visit(d, kinds)); 108 } 109 110 @Override visitRequires(RequiresDirective d, Set<DirectiveKind> kinds)111 public Void visitRequires(RequiresDirective d, Set<DirectiveKind> kinds) { 112 visitAny(d, kinds); 113 return null; 114 } 115 116 @Override visitExports(ExportsDirective d, Set<DirectiveKind> kinds)117 public Void visitExports(ExportsDirective d, Set<DirectiveKind> kinds) { 118 visitAny(d, kinds); 119 return null; 120 } 121 122 @Override visitOpens(OpensDirective d, Set<DirectiveKind> kinds)123 public Void visitOpens(OpensDirective d, Set<DirectiveKind> kinds) { 124 visitAny(d, kinds); 125 return null; 126 } 127 128 @Override visitUses(UsesDirective d, Set<DirectiveKind> kinds)129 public Void visitUses(UsesDirective d, Set<DirectiveKind> kinds) { 130 visitAny(d, kinds); 131 return null; 132 } 133 134 @Override visitProvides(ProvidesDirective d, Set<DirectiveKind> kinds)135 public Void visitProvides(ProvidesDirective d, Set<DirectiveKind> kinds) { 136 visitAny(d, kinds); 137 return null; 138 } 139 visitAny(Directive d, Set<DirectiveKind> kinds)140 private void visitAny(Directive d, Set<DirectiveKind> kinds) { 141 out.println("visit: " + d); 142 kinds.remove(d.getKind()); 143 } 144 } 145 } 146 147