pull/39/head
FabianLars 3 years ago
parent 2c439f5a42
commit 6f5b317ee7
No known key found for this signature in database
GPG Key ID: 3B12BC1DEBF61125

@ -1,5 +1,5 @@
import { invoke } from '@tauri-apps/api/tauri'; import { invoke } from "@tauri-apps/api/tauri";
import { appWindow } from '@tauri-apps/api/window'; import { appWindow } from "@tauri-apps/api/window";
interface BaseHandlerData { interface BaseHandlerData {
id: number; id: number;
@ -20,8 +20,8 @@ interface ResponseData {
} }
enum UploadEvent { enum UploadEvent {
progress = 'upload://progress', progress = "upload://progress",
fileSize = 'upload://file-size', fileSize = "upload://file-size",
} }
type EventId = `${UploadEvent}-${number}`; type EventId = `${UploadEvent}-${number}`;
@ -31,22 +31,32 @@ type SizeHandler = (data: SizeHandlerData) => unknown;
const handlers: Map<EventId, ProgressHandler | SizeHandler> = new Map(); const handlers: Map<EventId, ProgressHandler | SizeHandler> = new Map();
const listeningMap: Map<UploadEvent, boolean> = new Map(); const listeningMap: Map<UploadEvent, boolean> = new Map();
const getIdForEvent = (event: UploadEvent, id: number): EventId => `${event}-${id}`; const getIdForEvent = (event: UploadEvent, id: number): EventId =>
`${event}-${id}`;
async function listenToEventIfNeeded(event: UploadEvent) { async function listenToEventIfNeeded(event: UploadEvent) {
if (listeningMap.get(event)) { if (listeningMap.get(event)) {
return; return;
} }
return appWindow.listen<SizeHandlerData & ProgressHandlerData>(event, ({ payload }) => { return appWindow.listen<SizeHandlerData & ProgressHandlerData>(
event,
({ payload }) => {
const eventId = getIdForEvent(event, payload.id); const eventId = getIdForEvent(event, payload.id);
const handler = handlers.get(eventId); const handler = handlers.get(eventId);
if (typeof handler === 'function') { if (typeof handler === "function") {
handler(payload); handler(payload);
} }
}); }
);
} }
export default async function upload(url: string, filePath: string, progressHandler?: ProgressHandler, fileSizeHandler?: SizeHandler, headers?: Record<string, string>) { export default async function upload(
url: string,
filePath: string,
progressHandler?: ProgressHandler,
fileSizeHandler?: SizeHandler,
headers?: Record<string, string>
) {
const ids = new Uint32Array(1); const ids = new Uint32Array(1);
window.crypto.getRandomValues(ids); window.crypto.getRandomValues(ids);
const id = ids[0]; const id = ids[0];
@ -64,7 +74,7 @@ export default async function upload(url: string, filePath: string, progressHand
await listenToEventIfNeeded(UploadEvent.progress); await listenToEventIfNeeded(UploadEvent.progress);
await listenToEventIfNeeded(UploadEvent.fileSize); await listenToEventIfNeeded(UploadEvent.fileSize);
return await invoke<ResponseData>('plugin:upload|upload', { return await invoke<ResponseData>("plugin:upload|upload", {
id, id,
url, url,
filePath, filePath,

Loading…
Cancel
Save