js 判断字符串的文件类型|极客笔记

js 判断字符串的文件类型

在web开发中，经常会遇到需要判断一个字符串代表的文件类型的情况。比如用户上传文件后，我们需要判断该文件是图片、视频、文档还是其他类型的文件。本文将介绍如何使用JavaScript来判断一个字符串代表的文件类型，以便开发人员在实际项目中能够更加灵活地处理不同类型的文件。

1. 文件类型判断的方法

在JavaScript中，我们可以使用正则表达式或者字符串方法来判断一个字符串代表的文件类型。下面分别介绍这两种方法。

1.1 使用正则表达式

正则表达式是一种强大的匹配模式的工具，在这里我们可以利用正则表达式来判断文件类型。一般来说，不同的文件类型有不同的特征码可以用来区分。例如，图片文件通常以jpg、png、gif等结尾，视频文件通常以mp4、avi、mov等结尾，文档文件通常以pdf、doc、txt等结尾。

下面是一个简单的正则表达式判断文件类型的示例代码：

function getFileType(filename) {
    const imageRegex = /\.(jpg|jpeg|png|gif) $/i; const videoRegex = /\.(mp4|avi|mov)$ /i;
    const docRegex = /\.(pdf|doc|txt)$/i;

    if (imageRegex.test(filename)) {
        return 'image';
    } else if (videoRegex.test(filename)) {
        return 'video';
    } else if (docRegex.test(filename)) {
        return 'document';
    } else {
        return 'other';
    }
}

const filename1 = 'example.jpg';
const filename2 = 'example.mp4';
const filename3 = 'example.pdf';
const filename4 = 'example.txt';

console.log(getFileType(filename1)); // output: image
console.log(getFileType(filename2)); // output: video
console.log(getFileType(filename3)); // output: document
console.log(getFileType(filename4)); // output: other

在上面的示例代码中，我们定义了三个正则表达式imageRegex、videoRegex和docRegex，分别代表图片、视频和文档文件的匹配规则。在getFileType函数中，根据传入的文件名filename使用正则表达式进行匹配，从而判断文件的类型。最后根据匹配结果返回对应的文件类型。

1.2 使用字符串方法

除了使用正则表达式，我们还可以使用JavaScript中的字符串方法来判断一个字符串代表的文件类型。一种常见的方法是使用字符串的endsWith方法，该方法可以判断一个字符串是否以指定的后缀结尾。

下面是使用字符串方法判断文件类型的示例代码：

function getFileTypeWithEndsWith(filename) {
    if (filename.toLowerCase().endsWith('.jpg') || filename.toLowerCase().endsWith('.jpeg') || filename.toLowerCase().endsWith('.png') || filename.toLowerCase().endsWith('.gif')) {
        return 'image';
    } else if (filename.toLowerCase().endsWith('.mp4') || filename.toLowerCase().endsWith('.avi') || filename.toLowerCase().endsWith('.mov')) {
        return 'video';
    } else if (filename.toLowerCase().endsWith('.pdf') || filename.toLowerCase().endsWith('.doc') || filename.toLowerCase().endsWith('.txt')) {
        return 'document';
    } else {
        return 'other';
    }
}

const filename1 = 'example.jpg';
const filename2 = 'example.mp4';
const filename3 = 'example.pdf';
const filename4 = 'example.txt';

console.log(getFileTypeWithEndsWith(filename1)); // output: image
console.log(getFileTypeWithEndsWith(filename2)); // output: video
console.log(getFileTypeWithEndsWith(filename3)); // output: document
console.log(getFileTypeWithEndsWith(filename4)); // output: other

在上面的示例代码中，我们定义了一个getFileTypeWithEndsWith函数，该函数使用字符串的endsWith方法来判断文件类型。与正则表达式方法相比，字符串方法更加直观和简单，但是需要列出所有可能的文件后缀名。