|
1 #// Usage: jjs -scripting -fx nashornastviewer.js -- <scriptfile> |
|
2 |
|
3 /* |
|
4 * Copyright (c) 2015, 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 # NOTE: This script requires JDK 9 build to run |
|
35 |
|
36 if (!$OPTIONS._fx) { |
|
37 print("Usage: jjs -scripting -fx nashornastviewer.js -- <.js file>"); |
|
38 exit(1); |
|
39 } |
|
40 |
|
41 // Using JavaFX from Nashorn. See also: |
|
42 // http://docs.oracle.com/javase/8/docs/technotes/guides/scripting/nashorn/javafx.html |
|
43 |
|
44 // This example shows AST of a script file as a JavaFX |
|
45 // tree view in a window. If no file is specified, AST of |
|
46 // this script file is shown. This script demonstrates |
|
47 // Nashorn Parser API too - http://openjdk.java.net/jeps/236 |
|
48 |
|
49 // JavaFX classes used |
|
50 var StackPane = Java.type("javafx.scene.layout.StackPane"); |
|
51 var Scene = Java.type("javafx.scene.Scene"); |
|
52 var TreeItem = Java.type("javafx.scene.control.TreeItem"); |
|
53 var TreeView = Java.type("javafx.scene.control.TreeView"); |
|
54 |
|
55 // Java classes used |
|
56 var Enum = Java.type("java.lang.Enum"); |
|
57 var File = Java.type("java.io.File"); |
|
58 var List = Java.type("java.util.List"); |
|
59 var Parser = Java.type("jdk.nashorn.api.tree.Parser"); |
|
60 var Tree = Java.type("jdk.nashorn.api.tree.Tree"); |
|
61 |
|
62 // Create a javafx TreeItem to view a AST node |
|
63 function treeItemForASTNode(ast, name) { |
|
64 var item = new TreeItem(name); |
|
65 // make an iteratable script object from a Tree |
|
66 ast = Object.bindProperties({}, ast); |
|
67 for (var prop in ast) { |
|
68 var node = ast[prop]; |
|
69 var type = typeof node; |
|
70 |
|
71 if (node == null || type == "function") { |
|
72 // skip nulls and Java methods |
|
73 continue; |
|
74 } |
|
75 |
|
76 var subitem = null; |
|
77 if (node instanceof Tree) { |
|
78 subitem = treeItemForASTNode(node, prop); |
|
79 } else if (node instanceof List) { |
|
80 var len = node.size(); |
|
81 subitem = new TreeItem(prop); |
|
82 for (var i = 0; i < len; i++) { |
|
83 var li = treeItemForASTNode(node.get(i), String(i)); |
|
84 subitem.children.add(li); |
|
85 } |
|
86 } else if (node instanceof Enum || type != 'object') { |
|
87 subitem = new TreeItem(prop + ": " + node); |
|
88 } |
|
89 |
|
90 if (subitem) { |
|
91 item.children.add(subitem); |
|
92 } |
|
93 } |
|
94 return item; |
|
95 } |
|
96 |
|
97 // do we have a script file passed? if not, use current script |
|
98 var sourceName = arguments.length == 0? __FILE__ : arguments[0]; |
|
99 |
|
100 var parser = Parser.create("-scripting"); |
|
101 // parse script to get CompilationUnitTree of it |
|
102 var ast = parser.parse(new File(sourceName), null); |
|
103 |
|
104 // JavaFX start method |
|
105 function start(stage) { |
|
106 stage.title = "AST Viewer"; |
|
107 var rootItem = treeItemForASTNode(ast, sourceName); |
|
108 var tree = new TreeView(rootItem); |
|
109 var root = new StackPane(); |
|
110 root.children.add(tree); |
|
111 stage.scene = new Scene(root, 300, 450); |
|
112 stage.show(); |
|
113 } |