tauri-plugins-workspace/plugins/upload/guest-js/index.ts

import { invoke } from "@tauri-apps/api/tauri";
import { appWindow } from "@tauri-apps/api/window";

interface BaseHandlerData {
  id: number;
}

interface ProgressHandlerData extends BaseHandlerData {
  progress: number;
  total: number;
}

interface SizeHandlerData extends BaseHandlerData {
  size: number;
}

interface ResponseData {
  text: string;
  status: number;
}

enum UploadEvent {
  progress = "upload://progress",
  fileSize = "upload://file-size",
}

type EventId = `${UploadEvent}-${number}`;

type ProgressHandler = (data: ProgressHandlerData) => unknown;
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}`;

async function listenToEventIfNeeded(event: UploadEvent) {
  if (listeningMap.get(event)) {
    return;
  }
  return appWindow.listen<SizeHandlerData & ProgressHandlerData>(
    event,
    ({ payload }) => {
      const eventId = getIdForEvent(event, payload.id);
      const handler = handlers.get(eventId);
      if (typeof handler === "function") {
        handler(payload);
      }
    }
  );
}

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];

  if (progressHandler) {
    const eventId = getIdForEvent(UploadEvent.progress, id);
    handlers.set(eventId, progressHandler);
  }

  if (fileSizeHandler) {
    const eventId = getIdForEvent(UploadEvent.fileSize, id);
    handlers.set(eventId, fileSizeHandler);
  }

  await listenToEventIfNeeded(UploadEvent.progress);
  await listenToEventIfNeeded(UploadEvent.fileSize);

  return await invoke<ResponseData>("plugin:upload|upload", {
    id,
    url,
    filePath,
    headers: headers ?? {},
  });
}