Updated client

Fixed WebSocket location header for ws/wss (Thanks @jdub, Fixes #40)

Author: rauchg

example/client

@@ -29,29 +29,28 @@ WebSocket.prototype._onConnect = function(req, socket){
 	if ('sec-websocket-key1' in this.request.headers){
 		this.draft = 76;
 	}
-
-	if (this.draft == 76){
-		var origin = this.request.headers.origin;
+	
+	var origin = this.request.headers.origin;
 
+	if (this.draft == 76){
 		headers = [
 			'HTTP/1.1 101 WebSocket Protocol Handshake',
 			'Upgrade: WebSocket',
 			'Connection: Upgrade',
 			'Sec-WebSocket-Origin: ' + (origin || 'null'),
-			'Sec-WebSocket-Location: ws://' + this.request.headers.host + this.request.url
+			'Sec-WebSocket-Location: ' + origin.replace(/^http/, 'ws') + this.request.url
 		];
 
 		if ('sec-websocket-protocol' in this.request.headers){
 			headers.push('Sec-WebSocket-Protocol: ' + this.request.headers['sec-websocket-protocol']);
 		}
 	} else {
-		
 		headers = [
 			'HTTP/1.1 101 Web Socket Protocol Handshake',
 			'Upgrade: WebSocket',
 			'Connection: Upgrade',
-			'WebSocket-Origin: ' + this.request.headers.origin,
-			'WebSocket-Location: ws://' + this.request.headers.host + this.request.url
+			'WebSocket-Origin: ' + origin,
+			'WebSocket-Location: ' + origin.replace(/^http/, 'ws') + this.request.url
 		];
 
 		try {