/**

 * Back 2 Browser Bytecode Translator

 * Copyright (C) 2012-2015 Jaroslav Tulach <jaroslav.tulach@apidesign.org>

 *

 * This program is free software: you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation, version 2 of the License.

 *

 * This program is distributed in the hope that it will be useful,

 * but WITHOUT ANY WARRANTY; without even the implied warranty of

 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

 * GNU General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program. Look for COPYING file in the top folder.

 * If not, see http://opensource.org/licenses/GPL-2.0.

 */

package org.apidesign.vm4brwsr;

import java.io.ByteArrayInputStream;

import java.io.File;

import java.io.FileOutputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.OutputStreamWriter;

import java.io.Writer;

import java.net.URL;

import java.util.ArrayList;

import java.util.Arrays;

import java.util.Enumeration;

import java.util.HashSet;

import java.util.List;

import java.util.Set;

import javax.script.Invocable;

import javax.script.ScriptContext;

import javax.script.ScriptEngine;

import javax.script.ScriptEngineManager;

import javax.script.ScriptException;

import static org.testng.Assert.*;

public final class TestVM {

    private final Invocable code;

    private final CharSequence codeSeq;

    private final Object bck2brwsr;

    private BytesLoader resources;

    private TestVM(Invocable code, CharSequence codeSeq) throws ScriptException, NoSuchMethodException {

        this.code = code;

        this.codeSeq = codeSeq;

        this.bck2brwsr = ((ScriptEngine)code).eval("bck2brwsr(function(n) { return loader.get(n); })");

        ((ScriptEngine)code).getContext().setAttribute("loader", this, ScriptContext.ENGINE_SCOPE);

    }

    public void register(BytesLoader res) {

        this.resources = res;

    }

    public byte[] get(String res) throws IOException {

        return resources != null ? resources.get(res) : null;

    }

    public Object execCode(

        String msg, Class<?> clazz, String method, 

        Object expRes, Object... args

    ) throws Exception {

        Object ret = null;

        try {

            ret = code.invokeMethod(bck2brwsr, "loadClass", clazz.getName());

            List<Object> ma = new ArrayList<>();

            ma.add(method);

            ma.addAll(Arrays.asList(args));

            ret = code.invokeMethod(ret, "invoke", ma.toArray());

        } catch (ScriptException ex) {

            fail("Execution failed in " + dumpJS(codeSeq) + ": " + ex.getMessage(), ex);

        } catch (NoSuchMethodException ex) {

            fail("Cannot find method in " + dumpJS(codeSeq), ex);

        }

        if (ret == null && expRes == null) {

            return null;

        }

        if (expRes != null && expRes.equals(ret)) {

            return null;

        }

        if (expRes instanceof Number) {

            // in case of Long it is necessary convert it to number

            // since the Long is represented by two numbers in JavaScript

            try {

                final Object toFP = ((ScriptEngine)code).eval("Number.prototype.toFP");

                if (ret instanceof Long) {

                    ret = code.invokeMethod(toFP, "call", ret);

                }

                ret = code.invokeFunction("Number", ret);

            } catch (ScriptException ex) {

                fail("Conversion to number failed in " + dumpJS(codeSeq) + ": " + ex.getMessage(), ex);

            } catch (NoSuchMethodException ex) {

                fail("Cannot find global Number(x) function in " + dumpJS(codeSeq) + ": " + ex.getMessage(), ex);

            }

        }

        return ret;

    }

    void assertExec(

        String msg, Class clazz, String method, Object expRes, Object... args

    ) throws Exception {

        Object ret = execCode(msg, clazz, method, expRes, args);

        if (ret == null) {

            return;

        }

        if (expRes instanceof Integer && ret instanceof Double) {

            expRes = ((Integer)expRes).doubleValue();

        }

        if (expRes != null && expRes.equals(ret)) {

            return;

        }

        assertEquals(ret, expRes, msg + "was: " + ret + "\n" + dumpJS(codeSeq));

    }    

    static TestVM compileClass(String... names) throws ScriptException, IOException {

        return compileClass(null, names);

    }

    static TestVM compileClass(StringBuilder sb, String... names) throws ScriptException, IOException {

        return compileClass(sb, null, names);

    }

    static TestVM compileClass(StringBuilder sb, ScriptEngine[] eng, String... names) throws ScriptException, IOException {

        return compileClass(sb, eng, new EmulationResources(), names);

    }

    static TestVM compileClass(

        StringBuilder sb, 

        ScriptEngine[] eng, 

        Bck2Brwsr.Resources resources, 

        String... names

    ) throws ScriptException, IOException {

        if (sb == null) {

            sb = new StringBuilder();

        }

        Bck2Brwsr.generate(sb, resources, names);

        ScriptEngineManager sem = new ScriptEngineManager();

        ScriptEngine js = sem.getEngineByExtension("js");

        if (eng != null) {

            eng[0] = js;

        }

        try {

            Object res = js.eval(sb.toString());

            assertTrue(js instanceof Invocable, "It is invocable object: " + res);

            return new TestVM((Invocable) js, sb);

        } catch (Exception ex) {

            if (sb.length() > 2000) {

                sb = dumpJS(sb);

            }

            fail("Could not evaluate:" + ex.getClass() + ":" + ex.getMessage() + "\n" + sb, ex);

            return null;

        }

    }

    static TestVM compileClassAsExtension(

        StringBuilder sb, ScriptEngine[] eng, 

        String name, final String resourceName, final String resourceContent

    ) throws ScriptException, IOException {

        return compileClassesAsExtension(sb, eng, resourceName, resourceContent, name);

    }

    static TestVM compileClassesAsExtension(

        StringBuilder sb, ScriptEngine[] eng, 

        final String resourceName, final String resourceContent, String... names

    ) throws ScriptException, IOException {

        if (sb == null) {

            sb = new StringBuilder();

        }

        if (eng[0] == null) {

            ScriptEngineManager sem = new ScriptEngineManager();

            ScriptEngine js = sem.getEngineByExtension("js");

            eng[0] = js;

            Bck2Brwsr.newCompiler().resources(new EmulationResources())

                .obfuscation(ObfuscationLevel.NONE).generate(sb);

        }

        Set<String> exp = new HashSet<String>();

        for (String n : names) {

            int last = n.lastIndexOf('/');

            exp.add(n.substring(0, last + 1));

        }

        Bck2Brwsr b2b = Bck2Brwsr.newCompiler().

            resources(new EmulationResources() {

                @Override

                public InputStream get(String name) throws IOException {

                    if (name.equals(resourceName)) {

                        return new ByteArrayInputStream(resourceContent.getBytes("UTF-8"));

                    }

                    return super.get(name);

                }

            }).

            addClasses(names).

            addResources("org/apidesign/vm4brwsr/obj.js").

            addExported(exp.toArray(new String[0])).

            obfuscation(ObfuscationLevel.FULL).

            library();

        if (resourceName != null) {

            b2b = b2b.addResources(resourceName);

        }

        b2b.generate(sb);

        try {

            defineAtoB(eng[0]);

            Object res = eng[0].eval(sb.toString());

            assertTrue(eng[0] instanceof Invocable, "It is invocable object: " + res);

            return new TestVM((Invocable) eng[0], sb);

        } catch (Exception ex) {

            if (sb.length() > 2000) {

                sb = dumpJS(sb);

            }

            fail("Could not evaluate:" + ex.getClass() + ":" + ex.getMessage() + "\n" + sb, ex);

            return null;

        }

    }

    static TestVM compileClassAndResources(StringBuilder sb, ScriptEngine[] eng, String name, String... resources) throws ScriptException, IOException {

        if (sb == null) {

            sb = new StringBuilder();

        }

        Bck2Brwsr b2b = Bck2Brwsr.newCompiler().

            resources(new EmulationResources()).

            addRootClasses(name).

            addResources(resources);

        b2b.generate(sb);

        ScriptEngineManager sem = new ScriptEngineManager();

        ScriptEngine js = sem.getEngineByExtension("js");

        if (eng != null) {

            eng[0] = js;

        }

        try {

            defineAtoB(js);

            Object res = js.eval(sb.toString());

            assertTrue(js instanceof Invocable, "It is invocable object: " + res);

            return new TestVM((Invocable) js, sb);

        } catch (Exception ex) {

            if (sb.length() > 2000) {

                sb = dumpJS(sb);

            }

            fail("Could not evaluate:" + ex.getClass() + ":" + ex.getMessage() + "\n" + sb, ex);

            return null;

        }

    }

    private static void defineAtoB(ScriptEngine js) throws ScriptException {

        js.eval("atob = function(s) { return new String(org.apidesign.vm4brwsr.ResourcesTest.parseBase64Binary(s)); }");

    }

    Object loadClass(String loadClass, String name) throws ScriptException, NoSuchMethodException {

        return code.invokeMethod(bck2brwsr, "loadClass", Exceptions.class.getName());

    }

    Object invokeMethod(Object obj, String method, Object... params) throws ScriptException, NoSuchMethodException {

        return code.invokeMethod(obj, method, params);

    }

    Object invokeFunction(String methodName, Object... args) throws ScriptException, NoSuchMethodException {

        return code.invokeFunction(methodName, args);

    }

    static StringBuilder dumpJS(CharSequence sb) throws IOException {

        File f = File.createTempFile("execution", ".js");

        Writer w = new OutputStreamWriter(new FileOutputStream(f), "UTF-8");

        w.append(sb);

        w.close();

        return new StringBuilder(f.getPath());

    }

    @Override

    public String toString() {

        try {

            return dumpJS(codeSeq).toString();

        } catch (IOException ex) {

            return ex.toString();

        }

    }

    final CharSequence codeSeq() {

        return codeSeq;

    }

    private static class EmulationResources implements Bck2Brwsr.Resources {

        @Override

        public InputStream get(String name) throws IOException {

            if ("java/net/URI.class".equals(name)) {

                // skip

                return null;

            }

            if ("java/net/URLConnection.class".equals(name)) {

                // skip

                return null;

            }

            if ("java/lang/System.class".equals(name)) {

                // skip

                return null;

            }

            Enumeration<URL> en = StaticMethodTest.class.getClassLoader().getResources(name);

            URL u = null;

            while (en.hasMoreElements()) {

                u = en.nextElement();

            }

            if (u == null) {

                throw new IOException("Can't find " + name);

            }

            if (u.toExternalForm().contains("rt.jar!")) {

                throw new IOException("No emulation for " + u);

            }

            return u.openStream();

        }

    }

}