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 { appWindow } from '@tauri-apps/api/window';
import { invoke } from "@tauri-apps/api/tauri";
import { appWindow } from "@tauri-apps/api/window";
interface BaseHandlerData {
id: number;
@ -20,8 +20,8 @@ interface ResponseData {
}
enum UploadEvent {
progress = 'upload://progress',
fileSize = 'upload://file-size',
progress = "upload://progress",
fileSize = "upload://file-size",
}
type EventId = `${UploadEvent}-${number}`;
@ -31,22 +31,32 @@ type SizeHandler = (data: SizeHandlerData) => unknown;
const handlers: Map<EventId, ProgressHandler | SizeHandler> = 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) {
if (listeningMap.get(event)) {
return;
}
return appWindow.listen<SizeHandlerData & ProgressHandlerData>(event, ({ payload }) => {
return appWindow.listen<SizeHandlerData & ProgressHandlerData>(
event,
({ payload }) => {
const eventId = getIdForEvent(event, payload.id);
const handler = handlers.get(eventId);
if (typeof handler === 'function') {
if (typeof handler === "function") {
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);
window.crypto.getRandomValues(ids);
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.fileSize);
return await invoke<ResponseData>('plugin:upload|upload', {
return await invoke<ResponseData>("plugin:upload|upload", {
id,
url,
filePath,

Loading…
Cancel
Save