support JSON dump beyond AST_Toplevel (#4173)

2020-10-03 15:53:06 +01:00
parent 142bd1bd1a
commit 04017215cc
1 changed files with 12 additions and 1 deletions
--- a/bin/uglifyjs
+++ b/bin/uglifyjs
@@ -342,7 +342,18 @@ function run() {
        }
        fatal(ex);
    } else if (output == "ast") {
-        if (!options.compress && !options.mangle) result.ast.figure_out_scope({});
+        if (!options.compress && !options.mangle) {
+            var toplevel = result.ast;
+            if (!(toplevel instanceof UglifyJS.AST_Toplevel)) {
+                if (!(toplevel instanceof UglifyJS.AST_Statement)) toplevel = new UglifyJS.AST_SimpleStatement({
+                    body: toplevel,
+                });
+                toplevel = new UglifyJS.AST_Toplevel({
+                    body: [ toplevel ],
+                });
+            }
+            toplevel.figure_out_scope({});
+        }
        print(JSON.stringify(result.ast, function(key, value) {
            if (value) switch (key) {
              case "thedef":