Trilium/apps/client/src/services/import.ts

import toastService, { type ToastOptions } from "./toast.js";
import server from "./server.js";
import ws from "./ws.js";
import utils from "./utils.js";
import appContext from "../components/app_context.js";
import { t } from "./i18n.js";
import { WebSocketMessage } from "@triliumnext/commons";

type BooleanLike = boolean | "true" | "false";

export interface UploadFilesOptions {
    safeImport?: BooleanLike;
    shrinkImages: BooleanLike;
    textImportedAsText?: BooleanLike;
    codeImportedAsCode?: BooleanLike;
    explodeArchives?: BooleanLike;
    replaceUnderscoresWithSpaces?: BooleanLike;
}

export async function uploadFiles(entityType: string, parentNoteId: string, files: string[] | File[], options: UploadFilesOptions) {
    if (!["notes", "attachments"].includes(entityType)) {
        throw new Error(`Unrecognized import entity type '${entityType}'.`);
    }

    if (files.length === 0) {
        return;
    }

    const taskId = utils.randomString(10);
    let counter = 0;

    for (const file of files) {
        counter++;

        const formData = new FormData();
        formData.append("upload", file);
        formData.append("taskId", taskId);
        formData.append("last", counter === files.length ? "true" : "false");

        for (const key in options) {
            formData.append(key, (options as any)[key]);
        }

        await $.ajax({
            url: `${window.glob.baseApiUrl}notes/${parentNoteId}/${entityType}-import`,
            headers: await server.getHeaders(),
            data: formData,
            dataType: "json",
            type: "POST",
            timeout: 60 * 60 * 1000,
            error: function (xhr) {
                toastService.showError(t("import.failed", { message: xhr.responseText }));
            },
            contentType: false, // NEEDED, DON'T REMOVE THIS
            processData: false // NEEDED, DON'T REMOVE THIS
        });
    }
}

function makeToast(id: string, message: string): ToastOptions {
    return {
        id: id,
        title: t("import.import-status"),
        message: message,
        icon: "plus"
    };
}

ws.subscribeToMessages(async (message) => {
    if (!("taskType" in message) || message.taskType !== "importNotes") {
        return;
    }

    if (message.type === "taskError") {
        toastService.closePersistent(message.taskId);
        toastService.showError(message.message);
    } else if (message.type === "taskProgressCount") {
        toastService.showPersistent(makeToast(message.taskId, t("import.in-progress", { progress: message.progressCount })));
    } else if (message.type === "taskSucceeded") {
        const toast = makeToast(message.taskId, t("import.successful"));
        toast.closeAfter = 5000;

        toastService.showPersistent(toast);

        if (message.result.importedNoteId) {
            await appContext.tabManager.getActiveContext()?.setNote(message.result.importedNoteId);
        }
    }
});

ws.subscribeToMessages(async (message: WebSocketMessage) => {
    if (!("taskType" in message) || message.taskType !== "importAttachments") {
        return;
    }

    if (message.type === "taskError") {
        toastService.closePersistent(message.taskId);
        toastService.showError(message.message);
    } else if (message.type === "taskProgressCount") {
        toastService.showPersistent(makeToast(message.taskId, t("import.in-progress", { progress: message.progressCount })));
    } else if (message.type === "taskSucceeded") {
        const toast = makeToast(message.taskId, t("import.successful"));
        toast.closeAfter = 5000;

        toastService.showPersistent(toast);

        if (message.result.parentNoteId) {
            await appContext.tabManager.getActiveContext()?.setNote(message.result.importedNoteId, {
                viewScope: {
                    viewMode: "attachments"
                }
            });
        }
    }
});

export default {
    uploadFiles
};