Trilium/dump-db/inc/extension.ts

import path from "path";
import mimeTypes from "mime-types";

function getFileName(note: any, childTargetPath: string, safeTitle: string) {
    let existingExtension = path.extname(safeTitle).toLowerCase();
    let newExtension;

    if (note.type === 'text') {
        newExtension = 'html';
    } else if (note.mime === 'application/x-javascript' || note.mime === 'text/javascript') {
        newExtension = 'js';
    } else if (existingExtension.length > 0) { // if the page already has an extension, then we'll just keep it
        newExtension = null;
    } else {
        if (note.mime?.toLowerCase()?.trim() === "image/jpg") { // image/jpg is invalid but pretty common
            newExtension = 'jpg';
        } else {
            newExtension = mimeTypes.extension(note.mime) || "dat";
        }
    }

    let fileNameWithPath = childTargetPath;

    // if the note is already named with extension (e.g. "jquery"), then it's silly to append exact same extension again
    if (newExtension && existingExtension !== "." + newExtension.toLowerCase()) {
        fileNameWithPath += "." + newExtension;
    }

    return fileNameWithPath;
}

export default {
    getFileName
};