aoqi@0: #// Usage: jjs -fx javaastviewer.js -- <.java files> aoqi@0: aoqi@0: /* aoqi@0: * Copyright (c) 2014, Oracle and/or its affiliates. All rights reserved. aoqi@0: * aoqi@0: * Redistribution and use in source and binary forms, with or without aoqi@0: * modification, are permitted provided that the following conditions aoqi@0: * are met: aoqi@0: * aoqi@0: * - Redistributions of source code must retain the above copyright aoqi@0: * notice, this list of conditions and the following disclaimer. aoqi@0: * aoqi@0: * - Redistributions in binary form must reproduce the above copyright aoqi@0: * notice, this list of conditions and the following disclaimer in the aoqi@0: * documentation and/or other materials provided with the distribution. aoqi@0: * aoqi@0: * - Neither the name of Oracle nor the names of its aoqi@0: * contributors may be used to endorse or promote products derived aoqi@0: * from this software without specific prior written permission. aoqi@0: * aoqi@0: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS aoqi@0: * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, aoqi@0: * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR aoqi@0: * PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR aoqi@0: * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, aoqi@0: * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, aoqi@0: * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR aoqi@0: * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF aoqi@0: * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING aoqi@0: * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS aoqi@0: * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. aoqi@0: */ aoqi@0: aoqi@0: // This example demonstrates Java subclassing by Java.extend aoqi@0: // and javac Compiler and Tree API. This example also uses aoqi@0: // -fx and javafx TreeView to visualize Java AST as TreeView aoqi@0: aoqi@0: if (!$OPTIONS._fx || arguments.length == 0) { aoqi@0: print("Usage: jjs -fx javaastviewer.js -- <.java files>"); aoqi@0: exit(1); aoqi@0: } aoqi@0: aoqi@0: // Java types used aoqi@0: var Enum = Java.type("java.lang.Enum"); aoqi@0: var HashSet = Java.type("java.util.HashSet"); aoqi@0: var Name = Java.type("javax.lang.model.element.Name"); aoqi@0: var List = Java.type("java.util.List"); aoqi@0: var Set = Java.type("java.util.Set"); aoqi@0: var SimpleTreeVisitor = Java.type("com.sun.source.util.SimpleTreeVisitor"); aoqi@0: var StringArray = Java.type("java.lang.String[]"); aoqi@0: var ToolProvider = Java.type("javax.tools.ToolProvider"); aoqi@0: var Tree = Java.type("com.sun.source.tree.Tree"); aoqi@0: aoqi@0: function javaASTToScriptObject(args) { aoqi@0: // properties ignored (javac implementation class properties) in AST view. aoqi@0: // may not be exhaustive - any getAbc would become "abc" property or aoqi@0: // public field becomes a property of same name. aoqi@0: var ignoredProps = new HashSet(); aoqi@0: for each (var word in aoqi@0: ['extending', 'implementing', 'init', 'mods', 'clazz', 'defs', aoqi@0: 'expr', 'tag', 'preferredPosition', 'qualid', 'recvparam', aoqi@0: 'restype', 'params', 'startPosition', 'thrown', aoqi@0: 'tree', 'typarams', 'typetag', 'vartype']) { aoqi@0: ignoredProps.add(word); aoqi@0: } aoqi@0: aoqi@0: // get the system compiler tool aoqi@0: var compiler = ToolProvider.systemJavaCompiler; aoqi@0: aoqi@0: // get standard file manager aoqi@0: var fileMgr = compiler.getStandardFileManager(null, null, null); aoqi@0: aoqi@0: // make a list of compilation unit from command line argument file names aoqi@0: // Using Java.to convert script array (arguments) to a Java String[] aoqi@0: var compUnits = fileMgr.getJavaFileObjects(Java.to(args, StringArray)); aoqi@0: aoqi@0: // create a new compilation task aoqi@0: var task = compiler.getTask(null, fileMgr, null, null, null, compUnits); aoqi@0: aoqi@0: // subclass SimpleTreeVisitor - converts Java AST node to aoqi@0: // a simple script object by walking through it aoqi@0: var ConverterVisitor = Java.extend(SimpleTreeVisitor); aoqi@0: aoqi@0: var visitor = new ConverterVisitor() { aoqi@0: // convert java AST node to a friendly script object aoqi@0: // which can be viewed. Every node ends up in defaultAction aoqi@0: // method of SimpleTreeVisitor method. aoqi@0: aoqi@0: defaultAction: function (node, p) { aoqi@0: var resultObj = {}; aoqi@0: // Nashorn does not iterate properties and methods of Java objects aoqi@0: // But, we can bind properties of any object (including java objects) aoqi@0: // to a script object and iterate it! aoqi@0: var obj = {}; aoqi@0: Object.bindProperties(obj, node); aoqi@0: aoqi@0: // we don't want every property, method of java object aoqi@0: for (var prop in obj) { aoqi@0: var val = obj[prop]; aoqi@0: var type = typeof val; aoqi@0: // ignore 'method' members aoqi@0: if (type == 'function' || type == 'undefined') { aoqi@0: continue; aoqi@0: } aoqi@0: aoqi@0: // ignore properties from Javac implementation aoqi@0: // classes - hack by name!! aoqi@0: if (ignoredProps.contains(prop)) { aoqi@0: continue; aoqi@0: } aoqi@0: aoqi@0: // subtree - recurse it aoqi@0: if (val instanceof Tree) { aoqi@0: resultObj[prop] = visitor.visit(val, p); aoqi@0: } else if (val instanceof List) { aoqi@0: // List of trees - recurse each and make an array aoqi@0: var len = val.size(); aoqi@0: if (len != 0) { aoqi@0: var arr = []; aoqi@0: for (var j = 0; j < len; j++) { aoqi@0: var e = val[j]; aoqi@0: if (e instanceof Tree) { aoqi@0: arr.push(visitor.visit(e, p)); aoqi@0: } aoqi@0: } aoqi@0: resultObj[prop] = arr; aoqi@0: } aoqi@0: } else if (val instanceof Set) { aoqi@0: // Set - used for modifier flags aoqi@0: // make array aoqi@0: var len = val.size(); aoqi@0: if (len != 0) { aoqi@0: var arr = []; aoqi@0: for each (var e in val) { aoqi@0: if (e instanceof Enum || typeof e == 'string') { aoqi@0: arr.push(e.toString()); aoqi@0: } aoqi@0: } aoqi@0: resultObj[prop] = arr; aoqi@0: } aoqi@0: } else if (val instanceof Enum || val instanceof Name) { aoqi@0: // make string for any Enum or Name aoqi@0: resultObj[prop] = val.toString(); aoqi@0: } else if (type != 'object') { aoqi@0: // primitives 'as is' aoqi@0: resultObj[prop] = val; aoqi@0: } aoqi@0: } aoqi@0: return resultObj; aoqi@0: } aoqi@0: } aoqi@0: aoqi@0: // top level object with one property for each compilation unit aoqi@0: var scriptObj = {}; aoqi@0: for each (var cu in task.parse()) { aoqi@0: scriptObj[cu.sourceFile.name] = cu.accept(visitor, null); aoqi@0: } aoqi@0: aoqi@0: return scriptObj; aoqi@0: } aoqi@0: aoqi@0: // JavaFX classes used aoqi@0: var StackPane = Java.type("javafx.scene.layout.StackPane"); aoqi@0: var Scene = Java.type("javafx.scene.Scene"); aoqi@0: var TreeItem = Java.type("javafx.scene.control.TreeItem"); aoqi@0: var TreeView = Java.type("javafx.scene.control.TreeView"); aoqi@0: aoqi@0: // Create a javafx TreeItem to view a script object aoqi@0: function treeItemForObject(obj, name) { aoqi@0: var item = new TreeItem(name); aoqi@0: for (var prop in obj) { aoqi@0: var node = obj[prop]; aoqi@0: if (typeof node == 'object') { aoqi@0: if (node == null) { aoqi@0: // skip nulls aoqi@0: continue; aoqi@0: } aoqi@0: var subitem = treeItemForObject(node, prop); aoqi@0: } else { aoqi@0: var subitem = new TreeItem(prop + ": " + node); aoqi@0: } aoqi@0: item.children.add(subitem); aoqi@0: } aoqi@0: aoqi@0: item.expanded = true; aoqi@0: return item; aoqi@0: } aoqi@0: aoqi@0: var commandArgs = arguments; aoqi@0: aoqi@0: // JavaFX start method aoqi@0: function start(stage) { aoqi@0: var obj = javaASTToScriptObject(commandArgs); aoqi@0: stage.title = "Java AST Viewer" aoqi@0: var rootItem = treeItemForObject(obj, "AST"); aoqi@0: rootItem.expanded = true; aoqi@0: var tree = new TreeView(rootItem); aoqi@0: var root = new StackPane(); aoqi@0: root.children.add(tree); aoqi@0: stage.scene = new Scene(root, 300, 450); aoqi@0: stage.show(); aoqi@0: }