/*

 * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS HEADER.

 *

 * Copyright 1997-2008 Sun Microsystems, Inc. All rights reserved.

 *

 * The contents of this file are subject to the terms of either the GNU

 * General Public License Version 2 only ("GPL") or the Common

 * Development and Distribution License("CDDL") (collectively, the

 * "License"). You may not use this file except in compliance with the

 * License. You can obtain a copy of the License at

 * http://www.netbeans.org/cddl-gplv2.html

 * or nbbuild/licenses/CDDL-GPL-2-CP. See the License for the

 * specific language governing permissions and limitations under the

 * License.  When distributing the software, include this License Header

 * Notice in each file and include the License file at

 * nbbuild/licenses/CDDL-GPL-2-CP.  Sun designates this

 * particular file as subject to the "Classpath" exception as provided

 * by Sun in the GPL Version 2 section of the License file that

 * accompanied this code. If applicable, add the following below the

 * License Header, with the fields enclosed by brackets [] replaced by

 * your own identifying information:

 * "Portions Copyrighted [year] [name of copyright owner]"

 *

 * Contributor(s):

 *

 * The Original Software is NetBeans. The Initial Developer of the Original

 * Software is Sun Microsystems, Inc. Portions Copyright 1997-2006 Sun

 * Microsystems, Inc. All Rights Reserved.

 *

 * If you wish your version of this file to be governed by only the CDDL

 * or only the GPL Version 2, indicate your decision by adding

 * "[Contributor] elects to include this software in this distribution

 * under the [CDDL or GPL Version 2] license." If you do not indicate a

 * single choice of license, a recipient has the option to distribute

 * your version of this file under either the CDDL, the GPL Version 2 or

 * to extend the choice of license to its licensees as provided above.

 * However, if you add GPL Version 2 code and therefore, elected the GPL

 * Version 2 license, then the option applies only if the new code is

 * made subject to such option by the copyright holder.

 */

package org.netbeans.modules.selenium.server;

import java.beans.PropertyChangeEvent;

import java.beans.PropertyChangeListener;

import java.io.File;

import java.lang.reflect.InvocationTargetException;

import java.net.BindException;

import java.net.MalformedURLException;

import java.net.URL;

import java.net.URLClassLoader;

import java.util.logging.Level;

import java.util.logging.Logger;

import org.netbeans.api.server.properties.InstanceProperties;

import org.openide.modules.InstalledFileLocator;

import org.openide.util.RequestProcessor;

import org.openide.util.Task;

/**

 *

 * @author Jindrich Sedek

 * @author Martin Fousek

 */

class SeleniumServerRunner implements Runnable, PropertyChangeListener {

    private static final Logger LOGGER = Logger.getLogger(SeleniumServerRunner.class.getName());

    private static final SeleniumServerRunner instance = new SeleniumServerRunner();

    private static Object server = null;

    private boolean isRunning = false;

    private static Action action = null;

    private static Task latestTask = null;

    private SeleniumServerRunner() {

    }

    static Task startServer() {

        if (isRunning()) {

            return Task.EMPTY;

        }

        action = Action.START;

        return postTask();

    }

    static Task stopServer() {

        if (!isRunning()) {

            return Task.EMPTY;

        }

        action = Action.STOP;

        return postTask();

    }

    static Task restartServer() {

        if (!isRunning()) {

            return startServer();

        } else {

            action = Action.RESTART;

            return postTask();

        }

    }

    static boolean isRunning() {

        return instance.isRunning;

    }

    private static Task postTask(){

        Task t = RequestProcessor.getDefault().post(instance);

        latestTask = t;

        return t;

    }

    @Override

    public void run() {

        try {

            if (server == null) {

                initializeServer();

            }

            switch (action) {

                case START:

                    callSeleniumServerMethod("boot");

                    callSeleniumServerMethod("start");

                    break;

                case STOP:

                    callSeleniumServerMethod("stop");

                    break;

                case RESTART:

                    callSeleniumServerMethod("stop");

                    callSeleniumServerMethod("boot");

                    callSeleniumServerMethod("start");

                    break;

                case RELOAD:

                    callSeleniumServerMethod("stop");

                    server = null;

                    initializeServer();

                    callSeleniumServerMethod("boot");

                    callSeleniumServerMethod("start");

                    break;

                default:

                    assert false : "Invalid option";

            }

            if (action == null) {

                return;

            }

            isRunning = (!action.equals(Action.STOP));

            action = null;

        } catch (BindException bi) {

            LOGGER.log(Level.INFO, "Port already in use - the server is probably already running.", bi); //NOI18N

        } catch (Exception exc) {

            LOGGER.log(Level.INFO, null, exc);

        }

    }

    protected static URLClassLoader getSeleniumServerClassLoader() {

        URL url = null;

        try {

            url = InstalledFileLocator.getDefault().locate(

                        "modules/ext/selenium/selenium-server-2.16.1.jar", //NOI18N

                        null, //NOI18N

                        false).toURI().toURL();

        } catch (MalformedURLException ex) {

            LOGGER.log(Level.SEVERE, null, ex);

        }

        return URLClassLoader.newInstance(new URL[] {url}); //NOI18N

    }

    private void callSeleniumServerMethod(String method) {

        ClassLoader original = Thread.currentThread().getContextClassLoader();

        try {

            ClassLoader curr = server.getClass().getClassLoader();

            Thread.currentThread().setContextClassLoader(curr);

            server.getClass().getMethod(method).invoke(server);

        } catch (IllegalAccessException ex) {

            LOGGER.log(Level.WARNING, null, ex);

        } catch (IllegalArgumentException ex) {

            LOGGER.log(Level.WARNING, null, ex);

        } catch (NoSuchMethodException ex) {

            LOGGER.log(Level.WARNING, null, ex);

        } catch (SecurityException ex) {

            LOGGER.log(Level.WARNING, null, ex);

        } catch (InvocationTargetException ex) {

            LOGGER.log(Level.WARNING, null, ex);

        } finally {

            Thread.currentThread().setContextClassLoader(original);

        }

    }

    private static void initializeServer() throws Exception {

        URLClassLoader urlClassLoader = getSeleniumServerClassLoader();

        Class seleniumServer = urlClassLoader.loadClass("org.openqa.selenium.server.SeleniumServer"); //NOI18N

        Class remoteControlConfiguration = urlClassLoader.loadClass(

                "org.openqa.selenium.server.RemoteControlConfiguration"); //NOI18N

        InstanceProperties ip = SeleniumProperties.getInstanceProperties();

        Object remoteControlConfigurationInstance = remoteControlConfiguration.newInstance();

        int port = ip.getInt(SeleniumProperties.PORT, SeleniumProperties.getSeleniumDefaultPort());

        remoteControlConfiguration.getMethod("setPort", int.class).invoke(

            remoteControlConfigurationInstance, port); //NOI18N

        boolean runInSingleWindow = ip.getBoolean(SeleniumProperties.SINGLE_WINDOW, false);

        remoteControlConfiguration.getMethod("setSingleWindow", Boolean.TYPE).invoke( //NOI18N

                remoteControlConfigurationInstance, runInSingleWindow);

		String firefoxProfileDir = ip.getString(SeleniumProperties.FIREFOX_PROFILE, ""); //NOI18N

		if (!firefoxProfileDir.isEmpty()) {

				File ffProfileDir = new File(firefoxProfileDir);

				if (ffProfileDir.exists()) {

					remoteControlConfiguration.getMethod("setFirefoxProfileTemplate", File.class).invoke( //NOI18N

						remoteControlConfigurationInstance, ffProfileDir);

				}

		}

        String userExtensionsString = ip.getString(SeleniumProperties.USER_EXTENSIONS, ""); //NOI18N

        if (!userExtensionsString.isEmpty()) {

            File userExtensionFile = new File(userExtensionsString);

            if (userExtensionFile.exists()) {

                remoteControlConfiguration.getMethod("setUserExtensions", File.class).invoke( //NOI18N

                        remoteControlConfigurationInstance, userExtensionFile);

            }

        }

        server = seleniumServer.getConstructor(remoteControlConfiguration).

                newInstance(remoteControlConfigurationInstance);

    }

    @Override

    public void propertyChange(PropertyChangeEvent evt) {

        if (SeleniumProperties.PORT.equals(evt.getPropertyName())){

            action = Action.RELOAD;

            RequestProcessor.getDefault().post(instance);

        }

    }

    // listen on SeleniumProperties

    static PropertyChangeListener getPropertyChangeListener() {

        return instance;

    }

    private static enum Action {

        START, STOP, RESTART, RELOAD

    }

    static void waitAllTasksFinished(){

        if (latestTask == null){

            return;

        }

        while (!latestTask.isFinished()){

            latestTask.waitFinished();

        }

    }

}