Files
notesnook/packages/core/utils/filename.js
2023-06-12 13:02:56 +05:00

83 lines
2.4 KiB
JavaScript

/*
This file is part of the Notesnook project (https://notesnook.com/)
Copyright (C) 2023 Streetwriters (Private) Limited
This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program. If not, see <http://www.gnu.org/licenses/>.
*/
import db from "mime-db";
// type MimeTypeInfo = {
// source: string;
// extensions?: string[];
// charset?: string;
// compressible?: boolean;
// };
/**
*
* @param {string} filename
* @param {string | undefined} mime
* @returns {string}
*/
export function getFileNameWithExtension(filename, mime) {
if (!mime || mime === "application/octet-stream") return filename;
const mimeData = db[mime];
if (!mimeData || !mimeData.extensions || mimeData.extensions.length === 0)
return filename;
const extension = mimeData.extensions[0];
if (mimeData.extensions.some((extension) => filename.endsWith(extension)))
return filename;
return `${filename}.${extension}`;
}
export const PDFMimeType = "application/pdf";
export const DocumentMimeTypes = [
PDFMimeType,
"application/msword",
"application/vnd.ms-word",
"application/vnd.oasis.opendocument.text",
"application/vnd.openxmlformats-officedocument.wordprocessingml",
"application/vnd.ms-excel",
"application/vnd.openxmlformats-officedocument.spreadsheetml",
"application/vnd.oasis.opendocument.spreadsheet",
"application/vnd.ms-powerpoint",
"application/vnd.openxmlformats-officedocument.presentationml",
"application/vnd.oasis.opendocument.presentation"
];
export const WebClipMimeType = "application/vnd.notesnook.web-clip";
export function isDocument(mime) {
return DocumentMimeTypes.some((a) => a.startsWith(mime));
}
export function isWebClip(mime) {
return mime === WebClipMimeType;
}
export function isImage(mime) {
return mime.startsWith("image/");
}
export function isVideo(mime) {
return mime.startsWith("video/");
}
export function isAudio(mime) {
return mime.startsWith("audio/");
}