rt/vm/src/main/java/org/apidesign/vm4brwsr/VMLazy.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Wed, 14 May 2014 21:24:24 +0200
branchclosure
changeset 1577 042d0f256e95
parent 1576 3e9bfed0d88e
child 1578 9e6b613a00ed
permissions -rw-r--r--
Read the byte array via loadBytes call
     1 /**
     2  * Back 2 Browser Bytecode Translator
     3  * Copyright (C) 2012 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
     4  *
     5  * This program is free software: you can redistribute it and/or modify
     6  * it under the terms of the GNU General Public License as published by
     7  * the Free Software Foundation, version 2 of the License.
     8  *
     9  * This program is distributed in the hope that it will be useful,
    10  * but WITHOUT ANY WARRANTY; without even the implied warranty of
    11  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
    12  * GNU General Public License for more details.
    13  *
    14  * You should have received a copy of the GNU General Public License
    15  * along with this program. Look for COPYING file in the top folder.
    16  * If not, see http://opensource.org/licenses/GPL-2.0.
    17  */
    18 package org.apidesign.vm4brwsr;
    19 
    20 import java.io.ByteArrayInputStream;
    21 import java.io.IOException;
    22 import java.io.InputStream;
    23 import org.apidesign.bck2brwsr.core.JavaScriptBody;
    24 
    25 /**
    26  *
    27  * @author Jaroslav Tulach <jtulach@netbeans.org>
    28  */
    29 final class VMLazy {
    30     private final Object loader;
    31     private final Object[] args;
    32     
    33     private VMLazy(Object loader, Object[] args) {
    34         this.loader = loader;
    35         this.args = args;
    36     }
    37     
    38     static void init() {
    39     }
    40     
    41     static Object load(Object loader, String name, Object[] arguments, byte[] arr) 
    42     throws IOException, ClassNotFoundException {
    43         if (arr == null) {
    44             throw new ClassNotFoundException(name);
    45         }
    46         return new VMLazy(loader, arguments).defineClass(arr, name, false);
    47     }
    48 
    49     static Object reload(Object loader, String name, Object[] arguments, byte[] arr) 
    50     throws IOException, ClassNotFoundException {
    51         return new VMLazy(loader, arguments).defineClass(arr, name, false);
    52     }
    53     
    54     private Object load(String name, boolean instance)
    55     throws IOException, ClassNotFoundException {
    56         String res = name.replace('.', '/') + ".class";
    57         byte[] arr = ClassPath.loadBytes(res, args, 0);
    58         if (arr == null) {
    59             throw new ClassNotFoundException(name);
    60         }
    61         
    62         return defineClass(arr, name, instance);
    63     }
    64 
    65     private Object defineClass(byte[] arr, String name, boolean instance) throws IOException {
    66         StringBuilder out = new StringBuilder(65535);
    67         out.append("var loader = arguments[0];\n");
    68         out.append("var vm = loader.vm;\n");
    69         int prelude = out.length();
    70         String initCode = new Gen(this, out).compile(new ByteArrayInputStream(arr));
    71         String code = out.toString().toString();
    72         String under = name.replace('.', '_');
    73         Object fn = applyCode(loader, under, code, instance);
    74         
    75         if (!initCode.isEmpty()) {
    76             out.setLength(prelude);
    77             out.append(initCode);
    78             code = out.toString().toString();
    79             applyCode(loader, null, code, false);
    80         }            
    81         
    82         return fn;
    83     }
    84 
    85     @JavaScriptBody(args = {"loader", "name", "script", "instance" }, body =
    86         "try {\n" +
    87         "  new Function(script)(loader, name);\n" +
    88         "} catch (ex) {\n" +
    89         "  throw 'Cannot compile ' + name + ' ' + ex + ' line: ' + ex.lineNumber + ' script:\\n' + script;\n" +
    90         "}\n" +
    91         "return name != null ? vm[name](instance) : null;\n"
    92     )
    93     private static native Object applyCode(Object loader, String name, String script, boolean instance);
    94     
    95     
    96     private static final class Gen extends ByteCodeToJavaScript {
    97         private final VMLazy lazy;
    98 
    99         public Gen(VMLazy vm, Appendable out) {
   100             super(out);
   101             this.lazy = vm;
   102         }
   103         
   104         @JavaScriptBody(args = {"n"},
   105         body =
   106         "var cls = n.replace__Ljava_lang_String_2CC('/','_').toString();"
   107         + "\nvar dot = n.replace__Ljava_lang_String_2CC('/','.').toString();"
   108         + "\nvar lazy = this._lazy();"
   109         + "\nvar loader = lazy._loader();"
   110         + "\nvar vm = loader.vm;"
   111         + "\nif (vm[cls]) return false;"
   112         + "\nvm[cls] = function() {"
   113         + "\n  var instance = arguments.length == 0 || arguments[0] === true;"
   114         + "\n  return lazy.load__Ljava_lang_Object_2Ljava_lang_String_2Z(dot, instance);"
   115         + "\n};"
   116         + "\nreturn true;")
   117         @Override
   118         protected boolean requireReference(String internalClassName) {
   119             throw new UnsupportedOperationException();
   120         }
   121 
   122         @Override
   123         protected void requireScript(String resourcePath) throws IOException {
   124             if (!resourcePath.startsWith("/")) {
   125                 resourcePath = "/" + resourcePath;
   126             }
   127             String code = readCode(resourcePath);
   128             applyCode(lazy.loader, null, code, false);
   129         }
   130 
   131         private String readCode(String resourcePath) throws IOException {
   132             InputStream is = getClass().getResourceAsStream(resourcePath);
   133             StringBuilder sb = new StringBuilder();
   134             for (;;) {
   135                 int ch = is.read();
   136                 if (ch == -1) {
   137                     break;
   138                 }
   139                 sb.append((char)ch);
   140             }
   141             return sb.toString();
   142         }
   143 
   144         @Override
   145         String assignClass(String className) {
   146             return "vm[arguments[1]]=";
   147         }
   148 
   149         @Override
   150         String accessClass(String classOperation) {
   151             return "vm." + classOperation;
   152         }
   153     }
   154 }