package org.python.core;

import java.lang.reflect.Constructor;

public class PyRunnableBootstrap implements CodeBootstrap {

    public static final String REFLECTION_METHOD_NAME = "getFilenameConstructorReflectionBootstrap";
    private final PyRunnable runnable;

    PyRunnableBootstrap(PyRunnable runnable) {
        this.runnable = runnable;
    }

    public PyCode loadCode(CodeLoader loader) {
        if (runnable instanceof ContainsPyBytecode) {
            try {
                BytecodeLoader.fixPyBytecode(((ContainsPyBytecode) runnable).getClass());
            } catch (NoSuchFieldException e) {
                throw Py.JavaError(e);
            } catch (java.io.IOException e) {
                throw Py.JavaError(e);
            } catch (ClassNotFoundException e) {
                throw Py.JavaError(e);
            }  catch (IllegalAccessException e) {
                throw Py.JavaError(e);
            }
        }
        return runnable.getMain();
    }

    public static CodeBootstrap getFilenameConstructorReflectionBootstrap(
            Class cls) {
        final Constructor constructor;
        try {
            constructor = cls.getConstructor(String.class);
        } catch (Exception e) {
            throw new IllegalArgumentException(
                    "PyRunnable class does not specify apropriate constructor.",
                    e);
        }
        return new CodeBootstrap() {

            public PyCode loadCode(CodeLoader loader) {
                try {
                    PyRunnable result = constructor.newInstance(loader.filename);
                    if (result instanceof ContainsPyBytecode) {
                        try {
                            BytecodeLoader.fixPyBytecode(((ContainsPyBytecode) result).getClass());
                        } catch (NoSuchFieldException e) {
                            throw Py.JavaError(e);
                        } catch (java.io.IOException e) {
                            throw Py.JavaError(e);
                        } catch (ClassNotFoundException e) {
                            throw Py.JavaError(e);
                        }  catch (IllegalAccessException e) {
                            throw Py.JavaError(e);
                        }
                    }
                    return result.getMain();
                } catch (Exception e) {
                    throw new IllegalArgumentException(
                            "PyRunnable class constructor does not support instantiation protocol.",
                            e);
                }
            }
        };
    }
}