From c594b1fca854a7b9fb73d854a9830143cd1032fc Mon Sep 17 00:00:00 2001
From: "bozo.kopic" <bozo.kopic@gmail.com>
Date: Thu, 24 Aug 2017 17:51:14 +0200
Subject: frontend

---
 src_js/hatter/main.js | 29 ++++++++++++++++++++++++++---
 1 file changed, 26 insertions(+), 3 deletions(-)

(limited to 'src_js/hatter/main.js')

diff --git a/src_js/hatter/main.js b/src_js/hatter/main.js
index 892b6af..3f64ed3 100644
--- a/src_js/hatter/main.js
+++ b/src_js/hatter/main.js
@@ -1,7 +1,7 @@
 import bean from 'bean';
-import R from 'ramda';
 
 import r from 'hatter/renderer';
+import * as util from 'hatter/util';
 import * as common from 'hatter/common';
 import * as vt from 'hatter/vt';
 
@@ -10,8 +10,31 @@ import 'style/main.scss';
 
 
 function main() {
-    let root = document.body.appendChild(document.createElement('div'));
-    r.init(root, common.defaultState, vt.main);
+    let conn = new WebSocket(wsAddress);
+
+    conn.onopen = () => {
+        let root = document.body.appendChild(document.createElement('div'));
+        let state = util.set('conn', conn, common.defaultState);
+        r.init(root, state, vt.main);
+    };
+
+    conn.onclose = () => {
+        alert("Disconnected from server");
+    };
+
+    conn.onerror = () => {
+        alert("Couldn't connect to server");
+    };
+
+    conn.onmessage = (evt) => {
+        try {
+            let msg = JSON.parse(evt.data);
+            common.processMsg(msg);
+        } catch(e) {
+            conn.close();
+            throw e;
+        }
+    };
 }
 
 
-- 
cgit v1.2.3-70-g09d2