Fix wss connection on Heroku

closes #2

1 files changed, 2 insertions, 2 deletions

diff --git a/src/client/Main.hx b/src/client/Main.hx
index c8750ca..89a6ac6 100644
--- a/src/client/Main.hx
+++ b/src/client/Main.hx
@@ -48,7 +48,6 @@ class Main {
 		if (host == "") host = "localhost";
 		this.host = host;
 		if (port == null) port = Browser.location.port;
-		if (port == "") port = "80";
 
 		final defaults:ClientSettings = {
 			version: SETTINGS_VERSION,
@@ -102,9 +101,10 @@ class Main {
 	}
 
 	function openWebSocket(host:String, port:String):Void {
+		if (port.length > 0) port = ':$port';
 		var protocol = "ws:";
 		if (Browser.location.protocol == "https:") protocol = "wss:";
-		ws = new WebSocket('$protocol//$host:$port');
+		ws = new WebSocket('$protocol//$host$port');
 		ws.onmessage = onMessage;
 		ws.onopen = () -> {
 			serverMessage(1);