package io.socket;
import java.io.IOException;
import java.net.URI;
import java.net.URL;
import java.util.regex.Pattern;
import javax.net.ssl.SSLContext;
import javax.net.ssl.SSLSocketFactory;
import org.java_websocket.client.DefaultSSLWebSocketClientFactory;
import org.java_websocket.client.WebSocketClient;
import org.java_websocket.handshake.ServerHandshake;
import java.security.NoSuchAlgorithmException;
import java.security.NoSuchProviderException;
import java.security.KeyManagementException;
class WebsocketTransport extends WebSocketClient implements IOTransport {
private final static Pattern PATTERN_HTTP = Pattern.compile("^http");
public static final String TRANSPORT_NAME = "websocket";
private IOConnection connection;
public static IOTransport create(URL url, IOConnection connection) {
URI uri = URI.create(
PATTERN_HTTP.matcher(url.toString()).replaceFirst("ws")
+ IOConnection.SOCKET_IO_1 + TRANSPORT_NAME
+ "/" + connection.getSessionId());
return new WebsocketTransport(uri, connection);
}
public WebsocketTransport(URI uri, IOConnection connection) {
super(uri);
this.connection = connection;
// SSLContext context = IOConnection.getSslContext();
SSLContext context = null;
try {
context = SSLContext.getInstance("TLS", "HarmonyJSSE");
} catch (NoSuchAlgorithmException e) {
e.printStackTrace();
} catch (NoSuchProviderException e) {
e.printStackTrace();
}
try {
context.init(null, null, null);
} catch (KeyManagementException e) {
e.printStackTrace();
}
if("wss".equals(uri.getScheme()) && context != null) {
this.setWebSocketFactory(new DefaultSSLWebSocketClientFactory(context));
}
}
/* (non-Javadoc)
* @see io.socket.IOTransport#disconnect()
*/
@Override
public void disconnect() {
try {
this.close();
} catch (Exception e) {
connection.transportError(e);
}
}
/* (non-Javadoc)
* @see io.socket.IOTransport#canSendBulk()
*/
@Override
public boolean canSendBulk() {
return false;
}
/* (non-Javadoc)
* @see io.socket.IOTransport#sendBulk(java.lang.String[])
*/
@Override
public void sendBulk(String[] texts) throws IOException {
throw new RuntimeException("Cannot send Bulk!");
}
/* (non-Javadoc)
* @see io.socket.IOTransport#invalidate()
*/
@Override
public void invalidate() {
connection = null;
}
@Override
public void onClose(int code, String reason, boolean remote) {
if(connection != null)
connection.transportDisconnected();
}
@Override
public void onMessage(String text) {
if(connection != null)
connection.transportMessage(text);
}
@Override
public void onOpen(ServerHandshake handshakedata) {
if(connection != null)
connection.transportConnected();
}
@Override
public String getName() {
return TRANSPORT_NAME;
}
@Override
public void onError(Exception ex) {
// TODO Auto-generated method stub
}
}