Refactor main loop

lua/main.tl

@@ -72,10 +72,6 @@ local report_task = coroutine.create(function()
     end
 end)
 
-local shell_task = coroutine.create(function()
-    shell.run("shell")
-end)
-
 -- basically parallel.waitForAny
 
 local record Task
@@ -83,15 +79,39 @@ local record Task
     filter: string | nil
 end
     
-local tasks: {Task} = {
+local background_tasks: {Task} = {
-    {coro = shell_task}, -- pid 1
     {coro = ws_task},
     {coro = report_task},
 }
 
+local shell_task: Task = {
+    coro = coroutine.create(function() shell.run("shell") end)
+}
+
+local function handle_event(e: table, pid: integer)
+    if e[1] == "terminate" then return end
+
+    local task = background_tasks[pid]
+    if task.filter == nil or task.filter == e[1] then
+        local ok, param = coroutine.resume(task.coro, table.unpack(e as {any}))
+        if not ok then
+            term.native = orig_native
+            term.redirect(term.native())
+            term.clear()
+            term.setCursorPos(1,1)
+            print(("OMEGABIG OOF @ PID %d"):format(pid))
+            error(param, 0)
+        else
+            task.filter = param as string
+        end
+    end
+end
+
 local event_queue: Ringbuffer.Ringbuffer<table> = Ringbuffer.new(64)
 event_queue:push({n = 0})
 
+local shell_deaths: {any} = {}
+
 local shell_running = true
 while shell_running do
     local e: table
@@ -112,25 +132,23 @@ while shell_running do
             socket:signal_viewer_connect(false)
         end
     else
-        for pid = 1, #tasks do
+        for pid = 1, #background_tasks do
-            local task = tasks[pid]
+            handle_event(e, pid)
-            if task.filter == nil or task.filter == e[1] or e[1] == "terminate" then
+        end
-                local ok, param = coroutine.resume(task.coro, table.unpack(e as {any}))
+
-                if not ok then
+        if shell_task.filter == nil or shell_task.filter == e[1] or e[1] == "terminate" then
-                    term.redirect(orig_native())
+            local ok, param = coroutine.resume(shell_task.coro, table.unpack(e as {any}))
-                    term.clear()
+            if not ok then
-                    term.setCursorPos(1,1)
+                -- shell died i guess?
-                    print("OMEGABIG OOF")
+                table.insert(shell_deaths, param)
-                    print(("pid %d"):format(pid))
+            else
-                    error(param, 0)
+                shell_task.filter = param as string
-                else
-                    task.filter = param as string
-                end
-                if pid == 1 and coroutine.status(task.coro) == "dead" then
-                    shell_running = false
-                end
             end
         end
+
+        if coroutine.status(shell_task.coro) == "dead" then
+            shell_running = false
+        end
     end
 end
 
@@ -139,4 +157,8 @@ socket:close()
 term.native = orig_native
 term.redirect(prev_term)
 term.clear()
-term.setCursorPos(1,1)
+term.setCursorPos(1,1)
+
+for i = 1, #shell_deaths do
+    print(shell_deaths[i])
+end