js判断文件类型详解

js判断文件类型详解

通过file的type属性判断

<input type="file" onchange="onchangecb(this)" />
<script>
    function onchangecb(e) {
        const file = e.files[0];
        console.log(file.type);
    }
</script>

像html中input标签，就是根据选择文件的后缀来生成一个File对象。

像下面的几种文件：

txt文件：

jpg文件：

mp4文件：

使用文件后缀来判断

function onchangecb(e) {
    const file = e.files[0];
    //获取最后一个.的位置
    const index = file.name.lastIndexOf('.');
    //获取后缀
    const ext = file.name.substr(index + 1);
    console.log(ext);
}

在得到文件后缀名后，根据后缀即可判断文件的类型（文件格式）。比如需要判断一个文件是否是图片格式，首先定义一个判断函数：

function isImage(ext) {
    return (
        ['png', 'jpg', 'jpeg', 'bmp', 'gif', 'webp', 'psd', 'svg', 'tiff'].indexOf(
            ext.toLowerCase()
        ) !== -1
    );
}

虽然可以直接通过对象的type属性或者文件后缀来获取文件类型，但是如果强行修改文件后缀，同样可以将其他类型的文件上传至服务器，或者文件压根就没有后缀，那又要怎么判断呢？因此前端需要使用一个更加合理的方式。

根据二进制流及文件头来判断

虽然文件后缀可以手动改，因此可以直接通过读取文件的二进制来判断。

通常来说固定类型的文件头都是相同的，比如说jpeg的文件头是FF D8 FF E0。

这里提供一些常用的文件头：

const fileMap = {
    JPEG: 'FF D8 FF E0',
    JPG: 'FF D8 FF E1',
    PNG: '89 50 4E 47',
    GIF: '47 49 46 38',
    TIFF: '49 49 2A 00',
    BMP: '42 4D',
    DWG: '41 43 31 30',
    PSD: '38 42 50 53',
    RTF: '7B 5C 72 74 66',
    XML: '3C 3F 78 6D 6C',
    HTML: '68 74 6D 6C 3E',
    EML: '44 65 6C 69 76 65 72 79 2D 64 61 74 65 3A',
    DBX: 'CF AD 12 FE C5 FD 74 6F',
    PST: '21 42 44 4E',
    XLS: 'D0 CF 11 E0',
    DOC: 'D0 CF 11 E0',
    MDB: '53 74 61 6E 64 61 72 64 20 4A',
    WPD: 'FF 57 50 43',
    PDF: '25 50 44 46 2D 31 2E',
    QDF: 'AC 9E BD 8F',
    PWL: 'E3 82 85 96',
    ZIP: '50 4B 03 04',
    RAR: '52 61 72 21',
    WAV: '57 41 56 45',
    AVI: '41 56 49 20',
    RAM: '2E 72 61 FD',
    RM: '2E 52 4D 46',
    MPG: '00 00 01 BA',
    MPG: '00 00 01 B3',
    MOV: '6D 6F 6F 76',
    ASF: '30 26 B2 75 8E 66 CF 11',
    MID: '4D 54 68 64',
    MP3: '49 44 33',
};

通过onchange方法获取到选择的文件后，使用FileReader读取文件的二进制，之后判断二进制的前几位是否跟符合相应类型文件的文件头。

以下是一些使用文件头来判断文件类型的简单代码

// 判断文件后缀与该文件内容是否相同类型
async function isSameFileType(file) {
    const suffix = getFileSuffix(file.name).toUpperCase();
    const fileBlobString = await blobToString(file.slice(0, 14));
    // 如果文件类型中没有该类型，默认为false
    if (!(suffix in fileMap)) {
        return false;
    }

    return fileBlobString.includes(fileMap[suffix]);
}

async function blobToString(blob) {
    return new Promise((resolve) => {
        const reader = new FileReader();
        reader.onload = function () {
            const ret = reader.result
                .split('') // 分隔开
                .map((v) => v.charCodeAt()) // 循环返回指定位置的字符的 Unicode 编码
                .map((v) => v.toString(16).toUpperCase()) // 返回十六进制格式
                .map((v) => v.padStart(2, '0')) // 给空的那个填充 00 ，防止空缺
                .join(' '); // 每个子节之间空格隔开
            resolve(ret);
        };
        reader.readAsBinaryString(blob); // 调用之后触发onload事件
    });
    // 二进制=》ascii码=》转成16进制字符串
}

// 获取文件后缀
function getFileSuffix(filename) {
    return filename.substring(filename.lastIndexOf('.') + 1, filename.length);
}

判断是否是png类型：

async function isPng(file) {
    // 同理
    const ret = await blobToString(file.slice(0, 4));
    const ispng = ret === fileMap.PNG;
    return ispng;
}

当然也可以针对图片高宽进行限制：
在png二进制中，第18-20位是图片的宽，22-24是高。

获取图片宽高：

function getRectByOffset(file, widthOffset, heightOffset, reverse) {
  let width = await blobToString(file.slice(...widthOffset));
  let height = await blobToString(file.slice(...heightOffset));

  const w = parseInt(width.replace(" ", ""), 16);
  const h = parseInt(height.replace(" ", ""), 16);
  return { w, h };
}

修改一下之前判断isPng的方法，加上高宽限制

const IMG_WIDTH_LIMIT = 1000;
const IMG_HEIGHT_LIMIT = 800
function isPng(file) {
  // 同理
  const ret = await this.blobToString(file.slice(0, 4));
  const ispng = ret === fileMap.PNG;
  if (ispng) {
    const { w, h } = await this.getRectByOffset(file, [18, 20], [22, 24]);
    console.log(`png 宽高 ${w},${h}`);
    if (w > IMG_WIDTH_LIMIT || h > IMG_HEIGHT_LIMIT) {
      this.$message.error(
        "png图片宽高不得超过 " + IMG_WIDTH_LIMIT + "和" + IMG_HEIGHT_LIMIT
      );
      return false;
    }
  }
  return ispng;
}