boot/src/test/java/org/apidesign/html/boot/impl/FnTest.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
jaroslav@128
    22
package org.apidesign.html.boot.impl;
jaroslav@123
    23
jaroslav@323
    24
import java.io.Closeable;
jaroslav@163
    25
import java.io.Reader;
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.Collection;
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@128
    36
import org.apidesign.html.boot.spi.Fn;
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 <jtulach@netbeans.org>
jaroslav@123
    43
 */
jaroslav@123
    44
public class FnTest extends JsClassLoaderBase {
jaroslav@288
    45
    private static Fn.Presenter presenter;
jaroslav@123
    46
    
jaroslav@123
    47
    public FnTest() {
jaroslav@123
    48
    }
jaroslav@123
    49
jaroslav@123
    50
    @BeforeClass
jaroslav@123
    51
    public static void createClassLoader() throws Exception {
jaroslav@123
    52
        ScriptEngineManager sem = new ScriptEngineManager();
jaroslav@123
    53
        final ScriptEngine eng = sem.getEngineByMimeType("text/javascript");
jaroslav@123
    54
        
jaroslav@123
    55
        final URL my = FnTest.class.getProtectionDomain().getCodeSource().getLocation();
jaroslav@123
    56
        ClassLoader parent = JsClassLoaderTest.class.getClassLoader().getParent();
jaroslav@123
    57
        final URLClassLoader ul = new URLClassLoader(new URL[] { my }, parent);
jaroslav@123
    58
        
jaroslav@128
    59
        class Impl implements FindResources, Fn.Presenter {
jaroslav@123
    60
            @Override
jaroslav@123
    61
            public void findResources(String path, Collection<? super URL> results, boolean oneIsEnough) {
jaroslav@123
    62
                URL u = ul.findResource(path);
jaroslav@123
    63
                if (u != null) {
jaroslav@123
    64
                    results.add(u);
jaroslav@123
    65
                }
jaroslav@123
    66
            }
jaroslav@123
    67
jaroslav@123
    68
            @Override
jaroslav@123
    69
            public Fn defineFn(String code, String... names) {
jaroslav@123
    70
                StringBuilder sb = new StringBuilder();
jaroslav@123
    71
                sb.append("(function() {");
jaroslav@123
    72
                sb.append("return function(");
jaroslav@123
    73
                String sep = "";
jaroslav@123
    74
                for (String n : names) {
jaroslav@123
    75
                    sb.append(sep);
jaroslav@123
    76
                    sb.append(n);
jaroslav@123
    77
                    sep = ", ";
jaroslav@123
    78
                }
jaroslav@123
    79
                sb.append(") {");
jaroslav@123
    80
                sb.append(code);
jaroslav@123
    81
                sb.append("};");
jaroslav@123
    82
                sb.append("})()");
jaroslav@123
    83
                try {
jaroslav@123
    84
                    final Object val = eng.eval(sb.toString());
jaroslav@288
    85
                    return new Fn(this) {
jaroslav@123
    86
                        @Override
jaroslav@289
    87
                        public Object invoke(Object thiz, Object... args) throws Exception {
jaroslav@123
    88
                            List<Object> all = new ArrayList<Object>(args.length + 1);
jaroslav@123
    89
                            all.add(thiz == null ? val : thiz);
jaroslav@123
    90
                            all.addAll(Arrays.asList(args));
jaroslav@123
    91
                            Invocable inv = (Invocable)eng;
jaroslav@161
    92
                            try {
jaroslav@161
    93
                                Object ret = inv.invokeMethod(val, "call", all.toArray());
jaroslav@214
    94
                                return val.equals(ret) ? null : ret;
jaroslav@161
    95
                            } catch (ScriptException ex) {
jaroslav@161
    96
                                throw ex;
jaroslav@161
    97
                            }
jaroslav@123
    98
                        }
jaroslav@123
    99
                    };
jaroslav@123
   100
                } catch (ScriptException ex) {
jaroslav@123
   101
                    throw new LinkageError("Can't parse: " + sb, ex);
jaroslav@123
   102
                }
jaroslav@123
   103
            }
jaroslav@127
   104
jaroslav@127
   105
            @Override
jaroslav@128
   106
            public void displayPage(URL resource, Runnable r) {
jaroslav@127
   107
                throw new UnsupportedOperationException();
jaroslav@127
   108
            }
jaroslav@163
   109
jaroslav@163
   110
            @Override
jaroslav@163
   111
            public void loadScript(Reader code) throws Exception {
jaroslav@163
   112
                eng.eval(code);
jaroslav@163
   113
            }
jaroslav@123
   114
        }
jaroslav@123
   115
        Impl impl = new Impl();
jaroslav@126
   116
        ClassLoader loader = FnUtils.newLoader(impl, impl, parent);
jaroslav@288
   117
        presenter = impl;
jaroslav@123
   118
        
jaroslav@323
   119
        Closeable close = FnContext.activate(impl);
jaroslav@123
   120
        methodClass = loader.loadClass(JsMethods.class.getName());
jaroslav@323
   121
        close.close();
jaroslav@123
   122
    }
jaroslav@288
   123
jaroslav@288
   124
    @BeforeMethod public void initPresenter() {
jaroslav@309
   125
        FnContext.currentPresenter(presenter);
jaroslav@288
   126
    }
jaroslav@123
   127
}