org.specrunner.jetty.JettyStringProvider Maven / Gradle / Ivy

Go to download
/*
    SpecRunner - Acceptance Test Driven Development Tool
    Copyright (C) 2011-2013  Thiago Santos

    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, either version 3 of the License, or
    (at your option) any later version.

    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.  If not, see 
 */
package org.specrunner.jetty;

import org.eclipse.jetty.server.Connector;
import org.eclipse.jetty.server.Server;
import org.eclipse.jetty.server.nio.SelectChannelConnector;
import org.specrunner.SRServices;
import org.specrunner.context.ContextException;
import org.specrunner.context.IContext;
import org.specrunner.features.IFeatureManager;
import org.specrunner.util.string.IStringProvider;

/**
 * Default implementation of a string provider which, based on a previous
 * started Jetty server, gets the base url.
 * 
 * @author Thiago Santos
 * 
 */
public class JettyStringProvider implements IStringProvider {

    /**
     * URL feature.
     */
    public static final String FEATURE_URL = JettyStringProvider.class.getName() + ".url";
    /**
     * The start url.
     */
    private String url;

    /**
     * The url.
     * 
     * @return The url.
     */
    public String getUrl() {
        return url;
    }

    /**
     * Sets the url.
     * 
     * @param url
     *            The url.
     */
    public void setUrl(String url) {
        this.url = url;
    }

    @Override
    public String newString(IContext context) throws ContextException {
        IFeatureManager fm = SRServices.getFeatureManager();
        fm.set(FEATURE_URL, this);
        if (url == null) {
            Server server = (Server) context.getByName(PluginStartJetty.SERVER_NAME);
            if (server == null) {
                throw new ContextException("Jetty server named '" + PluginStartJetty.SERVER_NAME + "' does not exists.");
            }
            final int defaultPort = 8080;
            int port = defaultPort;
            for (Connector c : server.getConnectors()) {
                if (c instanceof SelectChannelConnector) {
                    port = c.getPort();
                    break;
                }
            }
            return "http://localhost:" + port;
        }
        return url;
    }

}