media-proxy/built/file-info.js

import fs from 'node:fs';
import { fileTypeFromFile } from 'file-type';
import isSvg from 'is-svg';
import { promisify } from 'node:util';
const TYPE_OCTET_STREAM = {
    mime: 'application/octet-stream',
    ext: null,
};
const TYPE_SVG = {
    mime: 'image/svg+xml',
    ext: 'svg',
};
async function getFileSize(path) {
    const getStat = promisify(fs.stat);
    return (await getStat(path)).size;
}
export async function detectType(path) {
    // Check 0 byte
    const fileSize = await getFileSize(path);
    if (fileSize === 0) {
        return TYPE_OCTET_STREAM;
    }
    const type = await fileTypeFromFile(path);
    if (type) {
        // XMLはSVGかもしれない
        if (type.mime === 'application/xml' && await checkSvg(path)) {
            return TYPE_SVG;
        }
        if (!isMimeImage(type.mime, 'safe-file')) {
            return TYPE_OCTET_STREAM;
        }
        return {
            mime: fixMime(type.mime),
            ext: type.ext,
        };
    }
    // 種類が不明でもSVGかもしれない
    if (await checkSvg(path)) {
        return TYPE_SVG;
    }
    // それでも種類が不明なら application/octet-stream にする
    return TYPE_OCTET_STREAM;
}
async function checkSvg(path) {
    try {
        const size = await getFileSize(path);
        if (size > 1 * 1024 * 1024)
            return false;
        return isSvg(fs.readFileSync(path));
    }
    catch {
        return false;
    }
}
import { FILE_TYPE_BROWSERSAFE } from './const.js';
const dictionary = {
    'safe-file': FILE_TYPE_BROWSERSAFE,
    'sharp-convertible-image': ['image/jpeg', 'image/png', 'image/gif', 'image/apng', 'image/vnd.mozilla.apng', 'image/webp', 'image/avif', 'image/svg+xml', 'image/x-icon', 'image/bmp'],
    'sharp-animation-convertible-image': ['image/jpeg', 'image/png', 'image/gif', 'image/webp', 'image/avif', 'image/svg+xml', 'image/x-icon', 'image/bmp'],
};
export const isMimeImage = (mime, type) => dictionary[type].includes(mime);
function fixMime(mime) {
    // see https://github.com/misskey-dev/misskey/pull/10686
    if (mime === "audio/x-flac") {
        return "audio/flac";
    }
    if (mime === "audio/vnd.wave") {
        return "audio/wav";
    }
    return mime;
}