Implement viewer_{dis,}connect events

lua/main.tl

@@ -13,6 +13,7 @@ local socket = Socket.new(ENDPOINT)
 
 print("[MAIN] Setup framebuffer")
 
+local prev_term = term.current()
 local orig_native = term.native
 local buffer = Framebuffer.wrap(orig_native())
 term.native = function(): term.Redirect
@@ -100,23 +101,34 @@ while shell_running do
     else
         e = table.pack(os.pullEventRaw())
     end
-
-    for pid = 1, #tasks do
-        local task = tasks[pid]
-        if task.filter == nil or task.filter == e[1] or e[1] == "terminate" then
-            local ok, param = coroutine.resume(task.coro, table.unpack(e as {any}))
-            if not ok then
-                term.redirect(orig_native())
-                term.clear()
-                term.setCursorPos(1,1)
-                print("OMEGABIG OOF")
-                print(("pid %d"):format(pid))
-                error(param, 0)
-            else
-                task.filter = param as string
-            end
-            if pid == 1 and coroutine.status(task.coro) == "dead" then
-                shell_running = false
+    
+    if e[1] == "websocket_message" and e[2] == ENDPOINT then
+        local payload = json.decode(e[3] as string) as table
+        if payload["type"] == "push_event" then
+            event_queue:push(payload["event"] as table)
+        elseif payload["type"] == "viewer_connect" then
+            socket:signal_viewer_connect(true)
+        elseif payload["type"] == "viewer_disconnect" then
+            socket:signal_viewer_connect(false)
+        end
+    else
+        for pid = 1, #tasks do
+            local task = tasks[pid]
+            if task.filter == nil or task.filter == e[1] or e[1] == "terminate" then
+                local ok, param = coroutine.resume(task.coro, table.unpack(e as {any}))
+                if not ok then
+                    term.redirect(orig_native())
+                    term.clear()
+                    term.setCursorPos(1,1)
+                    print("OMEGABIG OOF")
+                    print(("pid %d"):format(pid))
+                    error(param, 0)
+                else
+                    task.filter = param as string
+                end
+                if pid == 1 and coroutine.status(task.coro) == "dead" then
+                    shell_running = false
+                end
             end
         end
     end
@@ -125,6 +137,6 @@ end
 socket:close()
 
 term.native = orig_native
-term.redirect(term.native())
+term.redirect(prev_term)
 term.clear()
 term.setCursorPos(1,1)

lua/socket.tl

@@ -20,6 +20,7 @@ local record Socket
     send: function(self: Socket, message: string)
     reconnect: function(self: Socket)
     close: function(self: Socket)
+    signal_viewer_connect: function(self: Socket, connected: boolean)
     _endpoint: string
     _callback: StateCallback
     _ws: http.Websocket
@@ -73,6 +74,12 @@ impl.close = function(self: Socket)
     self._ws.close()
 end
 
+impl.signal_viewer_connect = function(self: Socket, connected: boolean)
+    if self:is_bad_state() then return end --how?
+    local new_state: State = connected and "viewer_connected" or "ok"
+    self:_set_state(new_state)
+end
+
 local function new(endpoint: string): Socket
     return setmetatable({
         state = "reset",