jdk8-mips64-public/nashorn: samples/autoimports.js@d0fba38a0705 (annotated)

samples/autoimports.js@d0fba38a0705 (annotated)

samples/autoimports.js

Tue, 22 Mar 2016 21:31:52 -0700

author: asaha
date: Tue, 22 Mar 2016 21:31:52 -0700
changeset 1810: d0fba38a0705
parent 1401: da52a33a5e93
permissions: -rw-r--r--

Added tag jdk8u92-b13 for changeset e2294411edbd

 # autoimports script requires -scripting mode
 /*
  * Copyright (c) 2015, Oracle and/or its affiliates. All rights reserved.
  *
  * Redistribution and use in source and binary forms, with or without
  * modification, are permitted provided that the following conditions
  * are met:
  *
  *   - Redistributions of source code must retain the above copyright
  *     notice, this list of conditions and the following disclaimer.
  *
  *   - Redistributions in binary form must reproduce the above copyright
  *     notice, this list of conditions and the following disclaimer in the
  *     documentation and/or other materials provided with the distribution.
  *
  *   - Neither the name of Oracle nor the names of its
  *     contributors may be used to endorse or promote products derived
  *     from this software without specific prior written permission.
  *
  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS
  * IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
  * THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE COPYRIGHT OWNER OR
  * CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
  * EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
  * PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
  * PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF
  * LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
  * NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  */
 /*
  * It is tedious to import Java classes used in a script. Sometimes it is easier
  * use simple names of java classes and have a script auto import Java classes.
  * You can load this script at the start of an interactive jjs session or at the
  * start of your script. This script defines a __noSuchProperty__ hook to auto
  * import Java classes as needed and when they are referred to for the first time
  * in your script. You can also call the "autoimports" function to print script
  * statements that you need to use in your script, i.e., have the function generate
  * a script to import Java classes used by your script so far. After running your
  * script, you can call autoimports to get the exact Java imports you need and replace
  * the autoimports load with the generated import statements (to avoid costly init of
  * the autoimports script).
  */
 (function() {
     var ArrayList = Java.type("java.util.ArrayList");
     var HashMap = Java.type("java.util.HashMap");
     var JarFile = Java.type("java.util.jar.JarFile");
     var File = Java.type("java.io.File");
     var Files = Java.type("java.nio.file.Files");
     var FileSystems = Java.type("java.nio.file.FileSystems");
     var System = Java.type("java.lang.System");
     var URI = Java.type("java.net.URI");
     // initialize a class to package map by iterating all
     // classes available in the system by walking through "jrt fs"
     var clsToPkg = new HashMap();
     // locate rt.jar from sun.boot.class.path
     function findRtJar() {
         var paths = System.getProperty("sun.boot.class.path").split(File.pathSeparator);
         for each (var p in paths) {
             if (p.endsWith("rt.jar") && new File(p).exists()) {
                 return p;
             }
         }
     }
     function addToClsToPkg(c, p) {
         if (clsToPkg.containsKey(c)) {
             var val = clsToPkg.get(c);
             if (val instanceof ArrayList) {
                 val.add(p);
             } else {
                 var al = new ArrayList();
                 al.add(val);
                 al.add(p);
                 clsToPkg.put(c, al);
             }
         } else {
             clsToPkg.put(c, p);
         }
     }
     // handle collision and allow user to choose package
     function getPkgOfCls(c) {
         var val = clsToPkg.get(c);
         if (val instanceof ArrayList) {
             var count = 1;
             print("Multiple matches for " + c + ", choose package:");
             for each (var v in val) {
                 print(count + ". " + v);
                 count++;
             }
             var choice = parseInt(readLine());
             if (isNaN(choice) || choice < 1 || choice > val.size()) {
                 print("invalid choice: " + choice);
                 return undefined;
             }
             return val.get(choice - 1);
         } else {
             return val;
         }
     }
     var rtJar = findRtJar();
     var stream = new JarFile(rtJar).stream();
     try {
         stream.forEach(
             function(entry) {
                 var str = entry.name;
                 if (str.endsWith(".class")) {
                     if (str.startsWith("java") ||
                         str.startsWith("javax") ||
                         str.startsWith("org")) {
                         var lastIdx = str.lastIndexOf('/');
                         if (lastIdx != -1) {
                             var pkg = str.substring(0, lastIdx).replaceAll('/', '.');
                             var cls = str.substring(lastIdx + 1, str.lastIndexOf(".class"));
                             addToClsToPkg(cls, pkg);
                         }
                     }
                 }
             });
     } finally {
         stream.close();
     }
     var imports = new ArrayList();
     var global = this;
     var oldNoSuchProp = global.__noSuchProperty__;
     this.__noSuchProperty__ = function(name) {
         'use strict';
         if (clsToPkg.containsKey(name)) {
             var pkg = getPkgOfCls(name);
             if (pkg) {
                 var clsName = pkg + "." + name;
                 imports.add("var " + name + " = Java.type('" + clsName + "');");
                 return global[name] = Java.type(clsName);
             }
         } else if (typeof oldNoSuchProp == 'function') {
             return oldNoSuchProp.call(this, name);
         }
         if (typeof this == 'undefined') {
             throw new ReferenceError(name);
         } else {
             return undefined;
         }
     }
     this.autoimports = function() {
         for each (var im in imports) {
             print(im);
         }
     }
 })();