tests/org.eclipse.rap.rwt.cluster.testfixture/src/org/eclipse/rap/rwt/cluster/testfixture/client/RWTClient.java

/*******************************************************************************
 * Copyright (c) 2011, 2012 EclipseSource and others.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *    EclipseSource - initial API and implementation
 ******************************************************************************/
package org.eclipse.rap.rwt.cluster.testfixture.client;

import java.io.IOException;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.net.URLConnection;
import java.util.HashMap;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import org.eclipse.rap.rwt.cluster.testfixture.server.IServletEngine;
import org.eclipse.rap.rwt.internal.serverpush.ServerPushServiceHandler;
import org.eclipse.rap.rwt.internal.service.ServiceManagerImpl;
import org.eclipse.swt.dnd.Transfer;


@SuppressWarnings("restriction")
public class RWTClient {
  // Keep "text" in sync with TextTransfer#TYPE_NAME
  private static final String TEXT_TRANSFER_DATA_TYPE
    = String.valueOf( Transfer.registerType( "text" ) );

  private IServletEngine servletEngine;
  private final IConnectionProvider connectionProvider;
  private final long startTime;
  private String sessionId;
  private int requestCounter;

  public RWTClient( IServletEngine servletEngine ) {
    this( servletEngine, new DefaultConnectionProvider() );
  }

  RWTClient( IServletEngine servletEngine, IConnectionProvider connectionProvider ) {
    this.servletEngine = servletEngine;
    this.connectionProvider = connectionProvider;
    startTime = System.currentTimeMillis();
    sessionId = "";
    requestCounter = -1;
  }

  public void changeServletEngine( IServletEngine servletEngine ) {
    this.servletEngine = servletEngine;
  }

  public Object getServletEngine() {
    return servletEngine;
  }

  public String getSessionId() {
    return sessionId;
  }

  public Response sendStartupRequest() throws IOException {
    return sendGetRequest( new HashMap<String, String>() );
  }

  public Response sendInitializationRequest() throws IOException {
    JsonMessage message = new JsonMessage();
    message.setInitialize( true );
    message.addOperation( "[\"set\",\"w1\",{\"bounds\":[0,0,800,600],\"dpi\":[96,96],\"colorDepth\":32}]" );
    message.addOperation( "[\"set\",\"w1\",{\"cursorLocation\":[0,0]}]" );
    return sendPostRequest( message );
  }

  public Response sendDisplayResizeRequest( int width, int height ) throws IOException {
    JsonMessage message = new JsonMessage();
    message.addOperation( "[\"set\",\"w1\",{\"bounds\":[0,0," + width + "," + height + "]}]" );
    return sendPostRequest( message );
  }

  public Response sendWidgetSelectedRequest( String widgetId ) throws IOException {
    JsonMessage message = new JsonMessage();
    message.addOperation( "[\"notify\",\"" + widgetId + "\",\"Selection\",{}]" );
    return sendPostRequest( message );
  }

  public Response sendShellCloseRequest( String shellId ) throws IOException {
    JsonMessage message = new JsonMessage();
    message.addOperation( "[\"notify\",\"" + shellId + "\",\"Close\",{}]" );
    return sendPostRequest( message );
  }

  public Response sendDragStartRequest( String widgetId ) throws IOException {
    JsonMessage message = new JsonMessage();
    message.addOperation( "[\"notify\",\""
                          + widgetId
                          + "\",\"dragStart\",{\"x\":100,\"y\":100,\"time\":"
                          + createTimeParam()
                          + "}]" );
    return sendPostRequest( message );
  }

  public Response sendDragFinishedRequest( String sourceWidgetId, String targetWidgetId )
    throws IOException
  {
    JsonMessage message = new JsonMessage();
    message.addOperation( "[\"notify\",\""
                          + sourceWidgetId
                          + "\",\"dragFinished\",{\"x\":100,\"y\":100,\"time\":"
                          + createTimeParam()
                          + "}]" );
    message.addOperation( "[\"notify\",\""
                          + targetWidgetId
                          + "\",\"dropAccept\",{\"x\":100,\"y\":100,\"item\":null,"
                          + "\"operation\":\"move\",\"feedback\":0,\"dataType\":"
                          + TEXT_TRANSFER_DATA_TYPE
                          + ",\"source\":\""
                          + sourceWidgetId
                          + "\",\"time\":"
                          + createTimeParam()
                          + "}]" );
    return sendPostRequest( message );
  }

  public Response sendResourceRequest( String resourceLocation ) throws IOException {
    URL url = createUrl( resourceLocation );
    HttpURLConnection connection = createGetConnection( url, 0 );
    return new Response( connection );
  }

  public Response sendUICallBackRequest( int timeout ) throws IOException {
    Map<String, String> parameters = new HashMap<String, String>();
    parameters.put( ServiceManagerImpl.REQUEST_PARAM, ServerPushServiceHandler.HANDLER_ID );
    URL url = createUrl( IServletEngine.SERVLET_NAME, parameters );
    HttpURLConnection connection = createGetConnection( url, timeout );
    return new Response( connection );
  }

  Response sendPostRequest( JsonMessage message ) throws IOException {
    if( requestCounter >= 0 ) {
      message.setRequestCounter( requestCounter );
    }
    URL url = createUrl( IServletEngine.SERVLET_NAME );
    HttpURLConnection connection = createPostConnection( url, message.toString(), 0 );
    parseSessionId( connection );
    Response response = new Response( connection );
    requestCounter = parseRequestCounter( response.getContentText() );
    return response;
  }

  Response sendGetRequest( Map<String, String> parameters ) throws IOException {
    URL url = createUrl( IServletEngine.SERVLET_NAME, parameters );
    HttpURLConnection connection = createGetConnection( url, 0 );
    parseSessionId( connection );
    return new Response( connection );
  }

  private void addSessionCookie( HttpURLConnection connection ) {
    if( sessionId != null && sessionId.length() > 0 ) {
      connection.setRequestProperty( "Cookie", "JSESSIONID=" + sessionId );
    }
  }

  private URL createUrl( String path ) {
    return createUrl( path, new HashMap<String, String>() );
  }

  private URL createUrl( String path, Map<String, String> parameters ) {
    int port = servletEngine.getPort();
    HttpUrlBuilder urlBuilder = new HttpUrlBuilder( "localhost", port, path );
    urlBuilder.addParameters( parameters );
    return urlBuilder.toUrl();
  }

  private HttpURLConnection createGetConnection( URL url, int timeout )
    throws IOException
  {
    HttpURLConnection connection = ( HttpURLConnection )connectionProvider.createConnection( url );
    connection.setInstanceFollowRedirects( false );
    connection.setAllowUserInteraction( false );
    connection.setRequestMethod( "GET" );
    connection.setConnectTimeout( timeout );
    connection.setReadTimeout( timeout );
    addSessionCookie( connection );
    connection.connect();
    return connection;
  }

  private HttpURLConnection createPostConnection( URL url, String content, int timeout )
    throws IOException
  {
    byte[] bytes = content.getBytes();
    HttpURLConnection connection = ( HttpURLConnection )connectionProvider.createConnection( url );
    connection.setRequestMethod( "POST" );
    connection.setRequestProperty( "Content-Type", "application/json" );
    connection.setRequestProperty( "Content-Length", Integer.toString( bytes.length ) );
    connection.setUseCaches( false );
    connection.setDoInput( true );
    connection.setDoOutput( true );
    connection.setConnectTimeout( timeout );
    connection.setReadTimeout( timeout );
    addSessionCookie( connection );
    OutputStream outputStream = connection.getOutputStream();
    outputStream.write( content.getBytes() );
    outputStream.flush();
    outputStream.close();
    return connection;
  }

  private String createTimeParam() {
    return String.valueOf( System.currentTimeMillis() - startTime );
  }

  private void parseSessionId( HttpURLConnection connection ) {
    String cookieField = connection.getHeaderField( "Set-Cookie" );
    if( cookieField != null ) {
      String[] parts = cookieField.split( ";" );
      sessionId = parts[ 0 ].split( "=" )[ 1 ];
    }
  }

  static int parseRequestCounter( String content ) {
    Pattern pattern = Pattern.compile( "\"requestCounter\":\\s*(\\d+)" );
    Matcher matcher = pattern.matcher( content );
    if( matcher.find() ) {
      return Integer.parseInt( matcher.group( 1 ) );
    }
    return -1;
  }

  private static class DefaultConnectionProvider implements IConnectionProvider {
    public URLConnection createConnection( URL url ) throws IOException {
      return url.openConnection();
    }
  }
}