yet-another-cad-viewer/frontend/tools/upload-file.ts

import { encrypt } from "tanmayo7lock";

async function check(lockerName: string) {
    const fileUrl = `https://vouz-backend.onrender.com/api/check_key`;
    const response = await fetch(fileUrl, {
        method: 'POST',
        headers: {
            'Content-Type': 'application/json',
            'Accept': 'application/json'
        },
        body: JSON.stringify({name: encrypt(lockerName), key: encrypt(lockerName)}),
    });
    if (!response.ok) throw new Error(`Failed to get file URL: ${response.status} ${response.statusText} -- ${await response.text()}`);
    const status = await response.json();
    return {response, status};
}

export async function uploadFile(name: string, data: Uint8Array): Promise<string> {
    // "Free" storage, let's see how long it lasts...
    // Create a locker
    const lockerUrl = `https://vouz-backend.onrender.com/api/locker`
    const lockerName = `yacv-pg-${name}-${Date.now()}`; // Unique locker name
    let responsePromise = fetch(lockerUrl, {
        method: 'POST',
        headers: {
            'Content-Type': 'application/json',
            'Accept': 'application/json'
        },
        body: JSON.stringify({name: encrypt(lockerName), passkey: encrypt(lockerName)}),
    });

    // The previous request never answers 🤮
    responsePromise.then((response) => console.warn(`Locker creation response: ${response.status} ${response.statusText} -- ${response.headers.get('Content-Type')}`));
    // Instead, poll the check endpoint until the locker is created
    let i: number;
    for (i = 0; i < 10; i++) {
        await new Promise(resolve => setTimeout(resolve, 250)); // Wait a bit before checking
        try {
            let {status} = await check(lockerName);
            if (status && status.data && status.data.length == 0) break // Locker is created
        } catch (e) { // Ignore errors, they will be thrown later
        }
    }
    if (i >= 10) throw new Error(`Failed to create locker after 10 attempts: ${lockerName}`);

    // Upload file to the locker
    const uploadUrl = `https://vouz-backend.onrender.com/api/upload`;
    const formData = new FormData();
    formData.append('file', new Blob([data], {type: 'application/octet-stream'}), name);
    formData.append("name", encrypt(lockerName));
    formData.append("passkey", encrypt(lockerName));
    const response = await fetch(uploadUrl, {
        method: 'POST',
        body: formData,
    })
    if (!response.ok) throw new Error(`Failed to upload file: ${response.status} ${response.statusText} -- ${await response.text()}`);

    // Fake URL for retrieveFile to work
    return "https://vouz.tech#name=" + encodeURIComponent(name) + "&locker=" + encodeURIComponent(lockerName);
}

/** Given any URL, it retrieves the file, with custom code for the vouz.tech locker. */
export async function retrieveFile(url: string): Promise<Response> {
    let realUrl = url;// Normal fetch if the URL is not a vouz.tech locker URL
    if (url.indexOf("https://vouz.tech#") !== -1) { // Check if the URL is a vouz.tech locker URL
        // Parse the URL to get the locker name and file name
        const urlObj = new URL(url);
        const hashParams = new URLSearchParams(urlObj.hash.slice(1)); // Remove the leading '#'
        const lockerName = hashParams.get('locker') || (() => {
            throw new Error("Locker name not found in URL hash")
        })();
        const name = hashParams.get('name') || (() => {
            throw new Error("File name not found in URL hash")
        })();
        // Get the URL of the uploaded file
        let {status} = await check(lockerName);
        if (!status || !status.data || status.data.length == 0 || !status.data[0].url) {
            throw new Error(`No file URL found in response: ${JSON.stringify(status)}`);
        }
        console.debug("File access requested successfully, URL:", status.data[0].url);
        realUrl = "https://corsproxy.io/?url=" + status.data[0].url + "#name=" + encodeURIComponent(name) + "&locker=" + encodeURIComponent(lockerName);
    }
    return await fetch(realUrl);
}