added readyState support

Author: Animesh Kumar

assets/www/js/websocket.js

@@ -1,6 +1,7 @@
 /*
  * Copyright (c) 2010 Nathan Rajlich (https://github.com/TooTallNate)
  * Copyright (c) 2010 Animesh Kumar  (https://github.com/anismiles)
+ * Copyright (c) 2010 Strumsoft  (https://strumsoft.com)
  *  
  * Permission is hereby granted, free of charge, to any person
  * obtaining a copy of this software and associated documentation
@@ -24,7 +25,7 @@
  * OTHER DEALINGS IN THE SOFTWARE.
  *  
  */
-package com.strumsoft.websocket.phonegap;
+package com.strumsoft.phonegap.websocket;
 
 import java.io.IOException;
 import java.io.UnsupportedEncodingException;
@@ -64,7 +65,24 @@ public enum Draft {
 		DRAFT75, DRAFT76
 	}
 
-	////////////////// CONSTANT
+	// //////////////// CONSTANT
+	/**
+	 * The connection has not yet been established.
+	 */
+	public final static int WEBSOCKET_STATE_CONNECTING = 0;
+	/**
+	 * The WebSocket connection is established and communication is possible.
+	 */
+	public final static int WEBSOCKET_STATE_OPEN = 1;
+	/**
+	 * The connection is going through the closing handshake.
+	 */
+	public final static int WEBSOCKET_STATE_CLOSING = 2;
+	/**
+	 * The connection has been closed or could not be opened.
+	 */
+	public final static int WEBSOCKET_STATE_CLOSED = 3;
+
 	/**
 	 * An empty string
 	 */
@@ -110,7 +128,7 @@ public enum Draft {
 	 */
 	public static final byte DATA_END_OF_FRAME = (byte) 0xFF;
 
-	////////////////// INSTANCE Variables
+	// //////////////// INSTANCE Variables
 	/**
 	 * The WebView instance from Phonegap DroidGap
 	 */
@@ -183,6 +201,10 @@ public enum Draft {
 	 * Key3 used in handshake
 	 */
 	private byte[] key3 = null;
+	/**
+	 * The readyState attribute represents the state of the connection.
+	 */
+	private int readyState = WEBSOCKET_STATE_CONNECTING;
 
 	/**
 	 * Constructor.
@@ -243,6 +265,8 @@ public void run() {
 	 * Closes connection with server
 	 */
 	public void close() {
+		this.readyState = WebSocket.WEBSOCKET_STATE_CLOSING;
+		
 		// close socket channel
 		try {
 			this.socketChannel.close();
@@ -251,9 +275,11 @@ public void close() {
 		}
 		this.running = false;
 		selector.wakeup();
-
+		
 		// fire onClose method
 		this.onClose();
+		
+		this.readyState = WebSocket.WEBSOCKET_STATE_CLOSED;
 	}
 
 	/**
@@ -264,7 +290,11 @@ public void close() {
 	 */
 	public void send(String text) {
 		try {
-			_send(text);
+			if(this.readyState == WEBSOCKET_STATE_OPEN){
+				_send(text);
+			} else {
+				this.onError(new NotYetConnectedException());
+			}
 		} catch (IOException e) {
 			this.onError(e);
 		}
@@ -298,10 +328,20 @@ public String getId() {
 	}
 
 	/**
-	 * Builds text for javascript engine to invoke proper event method with proper data.
+	 * @return the readyState
+	 */
+	public int getReadyState() {
+		return readyState;
+	}
+
+	/**
+	 * Builds text for javascript engine to invoke proper event method with
+	 * proper data.
 	 * 
-	 * @param event	websocket event (onOpen, onMessage etc.)
-	 * @param msg	Text message received from websocket server
+	 * @param event
+	 *            websocket event (onOpen, onMessage etc.)
+	 * @param msg
+	 *            Text message received from websocket server
 	 * @return
 	 */
 	private String buildJavaScriptData(String event, String msg) {
@@ -350,7 +390,7 @@ private boolean _send(String text) throws IOException {
 
 	// actual connection logic
 	private void _connect() throws IOException {
-
+		this.readyState = WEBSOCKET_STATE_CONNECTING;
 		socketChannel = SocketChannel.open();
 		socketChannel.configureBlocking(false);
 		socketChannel.connect(new InetSocketAddress(uri.getHost(), port));
@@ -564,6 +604,7 @@ private void _readHandshake(byte[] handShakeBody) throws IOException, NoSuchAlgo
 		}
 
 		if (isConnectionReady) {
+			this.readyState = WEBSOCKET_STATE_OPEN;
 			// fire onOpen method
 			this.onOpen();
 		} else {
@@ -604,4 +645,3 @@ private String _randomKey() {
 		return key;
 	}
 }
-