boot/src/test/java/org/apidesign/html/boot/impl/JsClassLoaderTest.java
author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
Wed, 06 Nov 2013 15:15:54 +0100
changeset 323 86aabecda7a3
parent 309 7025177bd67e
child 358 80702021b851
permissions -rw-r--r--
Introducing Agent-Class to allow java.lang.instrument-like transforms of the @JavaScriptBody annotation
jaroslav@123
     1
/**
jaroslav@123
     2
 * HTML via Java(tm) Language Bindings
jaroslav@123
     3
 * Copyright (C) 2013 Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@123
     4
 *
jaroslav@123
     5
 * This program is free software: you can redistribute it and/or modify
jaroslav@123
     6
 * it under the terms of the GNU General Public License as published by
jaroslav@123
     7
 * the Free Software Foundation, version 2 of the License.
jaroslav@123
     8
 *
jaroslav@123
     9
 * This program is distributed in the hope that it will be useful,
jaroslav@123
    10
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
jaroslav@123
    11
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
jaroslav@123
    12
 * GNU General Public License for more details. apidesign.org
jaroslav@123
    13
 * designates this particular file as subject to the
jaroslav@123
    14
 * "Classpath" exception as provided by apidesign.org
jaroslav@123
    15
 * in the License file that accompanied this code.
jaroslav@123
    16
 *
jaroslav@123
    17
 * You should have received a copy of the GNU General Public License
jaroslav@123
    18
 * along with this program. Look for COPYING file in the top folder.
jaroslav@123
    19
 * If not, see http://wiki.apidesign.org/wiki/GPLwithClassPathException
jaroslav@123
    20
 */
jaroslav@123
    21
package org.apidesign.html.boot.impl;
jaroslav@123
    22
jaroslav@323
    23
import java.io.Closeable;
jaroslav@163
    24
import java.io.Reader;
jaroslav@123
    25
import org.apidesign.html.boot.spi.Fn;
jaroslav@123
    26
import java.net.URL;
jaroslav@123
    27
import java.net.URLClassLoader;
jaroslav@123
    28
import java.util.ArrayList;
jaroslav@123
    29
import java.util.Arrays;
jaroslav@123
    30
import java.util.Enumeration;
jaroslav@123
    31
import java.util.List;
jaroslav@123
    32
import javax.script.Invocable;
jaroslav@123
    33
import javax.script.ScriptEngine;
jaroslav@123
    34
import javax.script.ScriptEngineManager;
jaroslav@123
    35
import javax.script.ScriptException;
jaroslav@123
    36
import org.testng.annotations.AfterClass;
jaroslav@123
    37
import org.testng.annotations.BeforeClass;
jaroslav@288
    38
import org.testng.annotations.BeforeMethod;
jaroslav@123
    39
jaroslav@123
    40
/**
jaroslav@123
    41
 *
jaroslav@123
    42
 * @author Jaroslav Tulach <jaroslav.tulach@apidesign.org>
jaroslav@123
    43
 */
jaroslav@123
    44
public class JsClassLoaderTest extends JsClassLoaderBase{
jaroslav@288
    45
    private static Fn.Presenter loader;
jaroslav@123
    46
jaroslav@123
    47
    @BeforeClass
jaroslav@123
    48
    public static void setUpClass() throws Exception {
jaroslav@123
    49
        ScriptEngineManager sem = new ScriptEngineManager();
jaroslav@123
    50
        final ScriptEngine eng = sem.getEngineByMimeType("text/javascript");
jaroslav@123
    51
        
jaroslav@123
    52
        final URL my = JsClassLoaderTest.class.getProtectionDomain().getCodeSource().getLocation();
jaroslav@123
    53
        ClassLoader parent = JsClassLoaderTest.class.getClassLoader().getParent();
jaroslav@123
    54
        final URLClassLoader ul = new URLClassLoader(new URL[] { my }, parent);
jaroslav@288
    55
        class MyCL extends JsClassLoader implements Fn.Presenter {
jaroslav@288
    56
jaroslav@288
    57
            public MyCL(ClassLoader parent) {
jaroslav@288
    58
                super(parent);
jaroslav@288
    59
            }
jaroslav@288
    60
            
jaroslav@123
    61
            @Override
jaroslav@123
    62
            protected URL findResource(String name) {
jaroslav@123
    63
                return ul.getResource(name);
jaroslav@123
    64
            }
jaroslav@123
    65
            @Override
jaroslav@288
    66
            public Fn defineFn(String code, String... names) {
jaroslav@123
    67
                StringBuilder sb = new StringBuilder();
jaroslav@123
    68
                sb.append("(function() {");
jaroslav@123
    69
                sb.append("return function(");
jaroslav@123
    70
                String sep = "";
jaroslav@123
    71
                for (String n : names) {
jaroslav@123
    72
                    sb.append(sep);
jaroslav@123
    73
                    sb.append(n);
jaroslav@123
    74
                    sep = ", ";
jaroslav@123
    75
                }
jaroslav@123
    76
                sb.append(") {");
jaroslav@123
    77
                sb.append(code);
jaroslav@123
    78
                sb.append("};");
jaroslav@123
    79
                sb.append("})()");
jaroslav@123
    80
                try {
jaroslav@123
    81
                    final Object val = eng.eval(sb.toString());
jaroslav@288
    82
                    return new Fn(this) {
jaroslav@123
    83
                        @Override
jaroslav@289
    84
                        public Object invoke(Object thiz, Object... args) throws Exception {
jaroslav@123
    85
                            List<Object> all = new ArrayList<Object>(args.length + 1);
jaroslav@123
    86
                            all.add(thiz == null ? val : thiz);
jaroslav@123
    87
                            all.addAll(Arrays.asList(args));
jaroslav@123
    88
                            Invocable inv = (Invocable)eng;
jaroslav@161
    89
                            try {
jaroslav@161
    90
                                Object ret = inv.invokeMethod(val, "call", all.toArray());
jaroslav@214
    91
                                return val.equals(ret) ? null : ret;
jaroslav@161
    92
                            } catch (Exception ex) {
jaroslav@161
    93
                                throw ex;
jaroslav@161
    94
                            }
jaroslav@123
    95
                        }
jaroslav@123
    96
                    };
jaroslav@123
    97
                } catch (ScriptException ex) {
jaroslav@123
    98
                    throw new LinkageError("Can't parse: " + sb, ex);
jaroslav@123
    99
                }
jaroslav@123
   100
            }
jaroslav@123
   101
jaroslav@123
   102
            @Override
jaroslav@123
   103
            protected Enumeration<URL> findResources(String name) {
jaroslav@288
   104
                throw new UnsupportedOperationException();
jaroslav@123
   105
            }
jaroslav@163
   106
jaroslav@163
   107
            @Override
jaroslav@288
   108
            public void loadScript(Reader code) throws ScriptException {
jaroslav@163
   109
                eng.eval(code);
jaroslav@163
   110
            }
jaroslav@288
   111
jaroslav@288
   112
            @Override
jaroslav@288
   113
            public void displayPage(URL page, Runnable onPageLoad) {
jaroslav@288
   114
                throw new UnsupportedOperationException();
jaroslav@288
   115
            }
jaroslav@123
   116
        };
jaroslav@123
   117
        
jaroslav@288
   118
        MyCL l = new MyCL(parent);
jaroslav@323
   119
        Closeable close = FnContext.activate(l);
jaroslav@288
   120
        methodClass = l.loadClass(JsMethods.class.getName());
jaroslav@323
   121
        close.close();
jaroslav@288
   122
        loader = l;
jaroslav@288
   123
    }
jaroslav@288
   124
    
jaroslav@288
   125
    @BeforeMethod public void initPresenter() {
jaroslav@309
   126
        FnContext.currentPresenter(loader);
jaroslav@123
   127
    }
jaroslav@123
   128
jaroslav@123
   129
    @AfterClass
jaroslav@123
   130
    public static void cleanUp() {
jaroslav@123
   131
        methodClass = null;
jaroslav@123
   132
    }
jaroslav@123
   133
}