mirror of
https://github.com/lifegpc/eh-downloader.git
synced 2026-07-08 01:31:00 +08:00
Update
This commit is contained in:
@@ -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;
|
||||
},
|
||||
|
||||
Reference in New Issue
Block a user