util-helpers

import { isFile, isString, toString } from 'ut2';
import { isUploadFile } from './utils/file.util';
import getExtname from './getExtname';
/**
 * 检查文件是否符合 `accept` 类型说明符。
 *
 * 通过 `file.type` `file.name` `file.url` 与 `accept` 进行匹配。
 *
 * @alias module:Browser.checkFileType
 * @since 5.1.0
 * @see {@link https://developer.mozilla.org/zh-CN/docs/Web/HTML/Element/input/file#唯一文件类型说明符 唯一文件类型说明符}
 * @see {@link https://www.iana.org/assignments/media-types/media-types.xhtml Media Types}
 * @param {File} file 文件对象。支持 antd `UploadFile` 对象。
 * @param {string} [accept] 文件类型说明符。
 * @returns {boolean} 如果 `file` 符合 `accept` 返回 `true`， 否则返回 `false`。
 * @example
 *
 * const pdf = new File([], '1.pdf', { type: 'application/pdf' });
 * const jpeg = new File([], 'xx.jpeg', { type: 'image/jpeg' });
 *
 * // 文件类型
 * checkFileType(pdf, 'application/pdf'); // true
 * checkFileType(jpeg, 'image/jpeg'); // true
 *
 * // 通配符
 * checkFileType(jpeg, 'image/*'); // true
 * checkFileType(pdf, 'image/*'); // false
 * checkFileType(pdf, 'application/*'); // true
 * checkFileType(pdf, '*'); // true
 * checkFileType(jpeg, '*'); // true
 *
 * // 文件名扩展名
 * checkFileType(jpeg, '.png,.jpeg,.jpg'); // true
 * checkFileType(pdf, '.pdf'); // true
 *
 */
function checkFileType(file, accept) {
    const isFileType = isFile(file);
    if (!isFileType && !isUploadFile(file)) {
        return false;
    }
    if (!isString(accept)) {
        accept = toString(accept);
    }
    accept = accept.trim();
    if (!accept || accept === '*') {
        return true;
    }
    let ret = false;
    const types = accept.toLowerCase().split(/,(?:\s+)?/);
    const fileName = (file.name || (!isFileType && (file.fileName || file.originFileObj?.name)) || '').toLowerCase();
    const fileType = file.type || (!isFileType && file.originFileObj?.type) || '';
    const fileUrl = (!isFileType && file.url) || '';
    types.some((type) => {
        // .doc .docx .jpg .png
        if (type === '*' || fileType === type || (type.indexOf('.') === 0 && (getExtname(fileName) === type || getExtname(fileUrl) === type))) {
            ret = true;
        }
        else if (type.includes('/*') && fileType.includes('/')) {
            // image/* 匹配所有图片类型
            const match = type.match(/(.*)\/\*/);
            const fileParentType = fileType.split('/')[0];
            if (match && match[1] === fileParentType) {
                ret = true;
            }
        }
        return ret;
    });
    return ret;
}
export default checkFileType;
checkFileType.js
