|
1 #// Usage: jjs javafoovars.js -- <directory> |
|
2 |
|
3 /* |
|
4 * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved. |
|
5 * |
|
6 * Redistribution and use in source and binary forms, with or without |
|
7 * modification, are permitted provided that the following conditions |
|
8 * are met: |
|
9 * |
|
10 * - Redistributions of source code must retain the above copyright |
|
11 * notice, this list of conditions and the following disclaimer. |
|
12 * |
|
13 * - Redistributions in binary form must reproduce the above copyright |
|
14 * notice, this list of conditions and the following disclaimer in the |
|
15 * documentation and/or other materials provided with the distribution. |
|
16 * |
|
17 * - Neither the name of Oracle nor the names of its |
|
18 * contributors may be used to endorse or promote products derived |
|
19 * from this software without specific prior written permission. |
|
20 * |
|
21 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS |
|
22 * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, |
|
23 * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
|
24 * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR |
|
25 * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, |
|
26 * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, |
|
27 * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR |
|
28 * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF |
|
29 * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING |
|
30 * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS |
|
31 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. |
|
32 */ |
|
33 |
|
34 // This example demonstrates Java subclassing by Java.extend |
|
35 // and javac Compiler and Tree API. This example counts number |
|
36 // of variables called "foo" in the given java source files! |
|
37 if (arguments.length == 0) { |
|
38 print("Usage: jjs javafoovars.js -- <directory>"); |
|
39 exit(1); |
|
40 } |
|
41 |
|
42 // Java types used |
|
43 var File = Java.type("java.io.File"); |
|
44 var Files = Java.type("java.nio.file.Files"); |
|
45 var FileVisitOption = Java.type("java.nio.file.FileVisitOption"); |
|
46 var StringArray = Java.type("java.lang.String[]"); |
|
47 var ToolProvider = Java.type("javax.tools.ToolProvider"); |
|
48 var Tree = Java.type("com.sun.source.tree.Tree"); |
|
49 var TreeScanner = Java.type("com.sun.source.util.TreeScanner"); |
|
50 var VariableTree = Java.type("com.sun.source.tree.VariableTree"); |
|
51 |
|
52 // count "foo"-s in the given .java files |
|
53 function countFoo() { |
|
54 // get the system compiler tool |
|
55 var compiler = ToolProvider.systemJavaCompiler; |
|
56 // get standard file manager |
|
57 var fileMgr = compiler.getStandardFileManager(null, null, null); |
|
58 // Using Java.to convert script array (arguments) to a Java String[] |
|
59 var compUnits = fileMgr.getJavaFileObjects( |
|
60 Java.to(arguments, StringArray)); |
|
61 // create a new compilation task |
|
62 var task = compiler.getTask(null, fileMgr, null, null, null, compUnits); |
|
63 // subclass SimpleTreeVisitor - to count variables called "foo" |
|
64 var FooCounterVisitor = Java.extend(TreeScanner); |
|
65 var fooCount = 0; |
|
66 |
|
67 var visitor = new FooCounterVisitor() { |
|
68 visitVariable: function (node, p) { |
|
69 if (node.name.toString() == "foo") { |
|
70 fooCount++; |
|
71 } |
|
72 } |
|
73 } |
|
74 |
|
75 for each (var cu in task.parse()) { |
|
76 cu.accept(visitor, null); |
|
77 } |
|
78 return fooCount; |
|
79 } |
|
80 |
|
81 // for each ".java" file in directory (recursively) count "foo". |
|
82 function main(dir) { |
|
83 var totalCount = 0; |
|
84 Files.walk(dir.toPath(), FileVisitOption.FOLLOW_LINKS). |
|
85 forEach(function(p) { |
|
86 var name = p.toFile().absolutePath; |
|
87 if (name.endsWith(".java")) { |
|
88 var count = 0; |
|
89 try { |
|
90 count = countFoo(p.toFile().getAbsolutePath()); |
|
91 } catch (e) { |
|
92 print(e); |
|
93 } |
|
94 if (count != 0) { |
|
95 print(name + ": " + count); |
|
96 } |
|
97 totalCount += count; |
|
98 } |
|
99 }); |
|
100 print("Total foo count: " + totalCount); |
|
101 } |
|
102 |
|
103 main(new File(arguments[0])); |