//  ========================================================================

//  Copyright (c) 1995-2014 Mort Bay Consulting Pty. Ltd.

//  ------------------------------------------------------------------------

//  All rights reserved. This program and the accompanying materials

//  are made available under the terms of the Eclipse Public License v1.0

//  and Apache License v2.0 which accompanies this distribution.

//

//      The Eclipse Public License is available at

//      http://www.eclipse.org/legal/epl-v10.html

//

//      The Apache License v2.0 is available at

//      http://www.opensource.org/licenses/apache2.0.php

//

//  You may elect to redistribute this code under either of these licenses.

//  ========================================================================

//

* Utility to build out exploded directory WebApps, in the /target/tests/ directory, for testing out servers that use javax.websocket endpoints.

* <p>

* This is particularly useful when the WebSocket endpoints are discovered via the javax.websocket annotation scanning.

*/

    private static final Logger LOG = Log.getLogger(WSServer.class);

    private final File contextDir;

    private final String contextPath;

    private Server server;

    private URI serverUri;

    private ContextHandlerCollection contexts;

    private File webinf;

    private File classesDir;

    public WSServer(TestingDir testdir, String contextName)

    {

        this(testdir.getDir(),contextName);

    }

    public WSServer(File testdir, String contextName)

    {

        this.contextDir = new File(testdir,contextName);

        this.contextPath = "/" + contextName;

        FS.ensureEmpty(contextDir);

    }

    public void copyClass(Class<?> clazz) throws Exception

    {

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

        String endpointPath = clazz.getName().replace('.','/') + ".class";

        URL classUrl = cl.getResource(endpointPath);

        Assert.assertThat("Class URL for: " + clazz,classUrl,notNullValue());

        File destFile = new File(classesDir,OS.separators(endpointPath));

        FS.ensureDirExists(destFile.getParentFile());

        File srcFile = new File(classUrl.toURI());

        IO.copy(srcFile,destFile);

    }

    public void copyEndpoint(Class<?> endpointClass) throws Exception

    {

        copyClass(endpointClass);

    }

    public void copyWebInf(String testResourceName) throws IOException

    {

        webinf = new File(contextDir,"WEB-INF");

        FS.ensureDirExists(webinf);

        classesDir = new File(webinf,"classes");

        FS.ensureDirExists(classesDir);

        File webxml = new File(webinf,"web.xml");

        File testWebXml = MavenTestingUtils.getTestResourceFile(testResourceName);

        IO.copy(testWebXml,webxml);

    }

    public WebAppContext createWebAppContext() throws MalformedURLException, IOException

    {

        WebAppContext context = new WebAppContext();

        context.setContextPath(this.contextPath);

        context.setBaseResource(Resource.newResource(this.contextDir));

        context.setAttribute("org.eclipse.jetty.websocket.jsr356",Boolean.TRUE);

        // @formatter:off

        context.setConfigurations(new Configuration[] {

                new AnnotationConfiguration(),

                new WebXmlConfiguration(),

                new WebInfConfiguration(),

                new PlusConfiguration(),

                new MetaInfConfiguration(),

                new FragmentConfiguration(),

                new EnvConfiguration()});

        // @formatter:on

        return context;

    }

    public void createWebInf() throws IOException

    {

        copyWebInf("empty-web.xml");

    }

    public void deployWebapp(WebAppContext webapp) throws Exception

    {

        contexts.addHandler(webapp);

        contexts.manage(webapp);

        webapp.start();

        if (LOG.isDebugEnabled())

        {

            webapp.dump(System.err);

        }

    }

    public void dump()

    {

        server.dumpStdErr();

    }

    public URI getServerBaseURI()

    {

        return serverUri;

    }

    public File getWebAppDir()

    {

        return this.contextDir;

    }

    public void start() throws Exception

    {

        server = new Server();

        ServerConnector connector = new ServerConnector(server);

        connector.setPort(0);

        server.addConnector(connector);

        HandlerCollection handlers = new HandlerCollection();

        contexts = new ContextHandlerCollection();

        handlers.addHandler(contexts);

        server.setHandler(handlers);

        server.start();

        String host = connector.getHost();

        if (host == null)

        {

            host = "localhost";

        }

        int port = connector.getLocalPort();

        serverUri = new URI(String.format("ws://%s:%d%s/",host,port,contextPath));

        if (LOG.isDebugEnabled())

            LOG.debug("Server started on {}",serverUri);

    }

    public void stop()

    {

        if (server != null)

        {

            try

            {

                server.stop();

            }

            catch (Exception e)

            {

                e.printStackTrace(System.err);

            }

        }

    }

}