import { PROTOCOL_S5, Sdk } from "@lumeweb/portal-sdk"; import { useSdk } from "~/components/lib/sdk-context.js"; import { S5Client } from "@lumeweb/s5-js"; import { S5Error } from "@lumeweb/s5-js/lib/client.js"; export interface PinningStatus { id: string; progress: number; status: "processing" | "completed"; } // biome-ignore lint/complexity/noStaticOnlyClass: export class PinningProcess { private static instances: Map = new Map(); private static sdk?: Sdk; static async pin(id: string): Promise<{ success: boolean; message: string }> { try { const s5 = PinningProcess.sdk?.protocols().get("s5")!.getSdk()!; console.log({ s5 }); if (PinningProcess.instances.has(id)) { return { success: false, message: "ID is already being processed" }; } const pinningStatus: PinningStatus = { id, progress: 0, status: "processing", }; const response = await s5.pin(id); console.log({ response }); PinningProcess.instances.set(id, pinningStatus); return { success: true, message: "Pinning process started" }; } catch (e) { console.error(e); return { success: false, message: "Pinning process failed" }; } } static async unpin( id: string, ): Promise<{ success: boolean; message: string }> { if (!PinningProcess.instances.has(id)) { return { success: false, message: "ID not found or not being processed" }; } await PinningProcess.sdk ?.protocols() .get(PROTOCOL_S5)! .getSdk() .unpin(id); return { success: true, message: "Pinning process removed" }; } static *pollAllProgress(): Generator { let allStatuses = Array.from(PinningProcess.instances.values()); let inProgress = allStatuses.some((status) => { PinningProcess.checkStatus(status.id); return status.status !== "completed"; }); while (inProgress) { yield allStatuses; allStatuses = Array.from(PinningProcess.instances.values()); inProgress = allStatuses.some((status) => { PinningProcess.checkStatus(status.id); return status.status !== "completed"; }); } yield allStatuses ?? []; // Yield the final statuses } private static async checkStatus(id: string) { const s5 = PinningProcess.sdk ?.protocols() .get(PROTOCOL_S5)! .getSdk()!; try { const ret = await s5.pinStatus(id); const status = PinningProcess.instances.get(id); if (!status) { return; } status.progress = ret.progress; status.status = ret.status as any; } catch (e) { if ((e as S5Error).statusCode == 404) { PinningProcess.instances.delete(id); } return; } } public static setupSdk(sdk: Sdk) { PinningProcess.sdk = sdk; } }