Update

routes/api/config.ts

@@ -1,8 +1,13 @@
 import { Handlers } from "$fresh/server.ts";
 import { ConfigType, load_settings, save_settings } from "../../config.ts";
 import { get_cfg_path, get_task_manager } from "../../server.ts";
+import {
+    ConfigClientSocketData,
+    ConfigSeverSocketData,
+} from "../../server/config.ts";
 import { parse_bool } from "../../server/parse_form.ts";
 import { return_json } from "../../server/utils.ts";
+import { ExitTarget } from "../../signal_handler.ts";
 
 const UNSAFE_TYPE: (keyof ConfigType)[] = [
     "base",
@@ -20,7 +25,51 @@ export const handler: Handlers = {
         const u = new URL(req.url);
         const current = await parse_bool(u.searchParams.get("current"), false);
         if (current) {
+            const type = u.searchParams.get("type");
             const t = get_task_manager();
+            if (type === "ws") {
+                const { socket, response } = Deno.upgradeWebSocket(req);
+                const handle = (e: CustomEvent<ConfigType>) => {
+                    sendMessage({ type: "cfg", cfg: e.detail });
+                };
+                const sendMessage = (mes: ConfigSeverSocketData) => {
+                    if (socket.readyState === socket.OPEN) {
+                        socket.send(JSON.stringify(mes));
+                    }
+                };
+                const close_handle = () => {
+                    sendMessage({ type: "close" });
+                    socket.close();
+                };
+                const removeListener = () => {
+                    t.removeEventListener("current_cfg_updated", handle);
+                    ExitTarget.removeEventListener("close", close_handle);
+                };
+                socket.onclose = () => {
+                    removeListener();
+                };
+                socket.onerror = () => {
+                    removeListener();
+                    console.error("WebSocket error.");
+                };
+                socket.onmessage = (e) => {
+                    try {
+                        const d: ConfigClientSocketData = JSON.parse(e.data);
+                        if (d.type == "close") {
+                            sendMessage({ type: "close" });
+                            socket.close();
+                        }
+                    } catch (_) {
+                        null;
+                    }
+                };
+                socket.onopen = () => {
+                    t.addEventListener("current_cfg_updated", handle);
+                    ExitTarget.addEventListener("close", close_handle);
+                    sendMessage({ type: "cfg", cfg: t.cfg.to_json() });
+                };
+                return response;
+            }
             return return_json(t.cfg.to_json());
         }
         const path = get_cfg_path();
@@ -35,16 +84,21 @@ export const handler: Handlers = {
             const m = get_task_manager();
             let is_unsafe = false;
             const cfg = await load_settings(path);
+            let updated = false;
             Object.getOwnPropertyNames(d).forEach((k) => {
                 if (UNSAFE_TYPE2.indexOf(k) === -1) {
                     cfg._data[k] = d[k];
                     m.cfg._data[k] = d[k];
+                    updated = true;
                 } else {
                     cfg._data[k] = d[k];
                     is_unsafe = true;
                 }
             });
             await save_settings(path, cfg, m.force_aborts);
+            if (updated) {
+                m.dispatchEvent("current_cfg_updated", m.cfg.to_json());
+            }
             return return_json({ is_unsafe });
         } else {
             return new Response("Bad Request", { status: 400 });

routes/api/task.ts

@@ -32,7 +32,9 @@ export const handler: Handlers<Task[]> = {
             ExitTarget.removeEventListener("close", close_handle);
         };
         function sendMessage(mes: TaskServerSocketData) {
-            socket.send(JSON.stringify(mes));
+            if (socket.readyState === socket.OPEN) {
+                socket.send(JSON.stringify(mes));
+            }
         }
         socket.onclose = () => {
             removeListener();