Graphite/frontend/src/utility-functions/files.ts

export function downloadBlob(url: string, filename: string): void {
	const element = document.createElement("a");

	element.href = url;
	element.setAttribute("download", filename);
	element.style.display = "none";

	element.click();
}

export function download(filename: string, fileData: string): void {
	const type = filename.endsWith(".svg") ? "image/svg+xml;charset=utf-8" : "text/plain;charset=utf-8";
	const blob = new Blob([fileData], { type });
	const url = URL.createObjectURL(blob);

	downloadBlob(url, filename);

	URL.revokeObjectURL(url);
}

export async function upload<T extends "text" | "data">(acceptedExtensions: string, textOrData: T): Promise<UploadResult<T>> {
	return new Promise<UploadResult<T>>((resolve, _) => {
		const element = document.createElement("input");
		element.type = "file";
		element.style.display = "none";
		element.accept = acceptedExtensions;

		element.addEventListener(
			"change",
			async () => {
				if (element.files?.length) {
					const file = element.files[0];

					const filename = file.name;
					const type = file.type;
					const content = (textOrData === "text" ? await file.text() : new Uint8Array(await file.arrayBuffer())) as UploadResultType<T>;

					resolve({ filename, type, content });
				}
			},
			{ capture: false, once: true }
		);

		element.click();

		// Once `element` goes out of scope, it has no references so it gets garbage collected along with its event listener, so `removeEventListener` is not needed
	});
}
export type UploadResult<T> = { filename: string; type: string; content: UploadResultType<T> };
type UploadResultType<T> = T extends "text" ? string : T extends "data" ? Uint8Array : never;