lifegpc/eh-downloader
1
0
Fork 0
mirror of https://github.com/lifegpc/eh-downloader.git synced 2026-07-08 01:31:00 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update

Browse Source
This commit is contained in:
lifegpc
2023-05-31 09:44:14 +08:00
parent 3fab4a8fc5
commit 5d75c211a7
4 changed files with 62 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

35
routes/api/task.ts
View File

@@ -1,18 +1,27 @@
import { Handlers } from "$fresh/server.ts";
import { get_task_manager } from "../../server.ts";
import { Task } from "../../task.ts";
import { Task, TaskProgress } from "../../task.ts";
import { DiscriminatedUnion } from "../../utils.ts";
type EventMap = {
close: Record<PropertyKey, never>;
new_download_task: { gid: number; token: string };
};
type EventData = DiscriminatedUnion<"type", EventMap>;
export const handler: Handlers<Task[]> = {
GET(req, _ctx) {
const t = get_task_manager();
const { socket, response } = Deno.upgradeWebSocket(req);
const task_handle = (e: CustomEvent<Task>) => {
const handle = (e: CustomEvent<Task | TaskProgress>) => {
socket.send(JSON.stringify({ type: e.type, detail: e.detail }));
};
const removeListener = () => {
t.removeEventListener("new_task", task_handle);
t.removeEventListener("task_started", task_handle);
t.removeEventListener("task_finished", task_handle);
t.removeEventListener("new_task", handle);
t.removeEventListener("task_started", handle);
t.removeEventListener("task_finished", handle);
t.removeEventListener("task_progress", handle);
};
socket.onclose = () => {
removeListener();
@@ -23,25 +32,21 @@ export const handler: Handlers<Task[]> = {
};
socket.onmessage = (e) => {
try {
const d = JSON.parse(e.data);
const d: EventData = 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);
t.add_download_task(d.gid, d.token);
}
} catch (_) {
null;
}
};
socket.onopen = () => {
t.addEventListener("new_task", task_handle);
t.addEventListener("task_started", task_handle);
t.addEventListener("task_finished", task_handle);
t.addEventListener("new_task", handle);
t.addEventListener("task_started", handle);
t.addEventListener("task_finished", handle);
t.addEventListener("task_progress", handle);
};
return response;
},