This commit is contained in:
2023-05-30 21:16:11 +08:00
parent 7cfb63ecbf
commit 9311bd64ea
6 changed files with 64 additions and 12 deletions

View File

@@ -6,11 +6,13 @@ export const handler: Handlers<Task[]> = {
GET(req, _ctx) {
const t = get_task_manager();
const { socket, response } = Deno.upgradeWebSocket(req);
const new_task_handle = (e: CustomEvent<Task>) => {
socket.send(JSON.stringify({ type: "new_task", detail: e.detail }));
const task_handle = (e: CustomEvent<Task>) => {
socket.send(JSON.stringify({ type: e.type, detail: e.detail }));
};
const removeListener = () => {
t.removeEventListener("new_task", new_task_handle);
t.removeEventListener("new_task", task_handle);
t.removeEventListener("task_started", task_handle);
t.removeEventListener("task_finished", task_handle);
};
socket.onclose = () => {
removeListener();
@@ -24,13 +26,22 @@ export const handler: Handlers<Task[]> = {
const d = JSON.parse(e.data);
if (d.type == "close") {
socket.close();
} else if (d.type == "new_download_task") {
const gid = d.gid;
const token = d.token;
if (typeof gid !== "number" || typeof token !== "string") {
return;
}
t.add_download_task(gid, token);
}
} catch (_) {
null;
}
};
socket.onopen = () => {
t.addEventListener("new_task", new_task_handle);
t.addEventListener("new_task", task_handle);
t.addEventListener("task_started", task_handle);
t.addEventListener("task_finished", task_handle);
};
return response;
},