Nodejs 识别图片类型的方法

在 Node.js 中，我们可以使用第三方包 file-type 来识别图片类型，它提供了一个简单的 API 来帮助我们快速判断文件类型。

安装

可以通过 npm 安装：

npm install file-type

使用

在使用 file-type 之前，需要确保你已经将图片的文件内容读取到了内存中，如果你只有图片的文件名，可以使用 fs 模块来读取文件内容。

const fileType = require('file-type');
const fs = require('fs');

// 读取文件内容
const buffer = fs.readFileSync('path/to/image');

// 判断文件类型
const type = fileType(buffer);

console.log(type);
// => {ext: 'png', mime: 'image/png'}

上述代码会输出一个对象，包括文件类型的扩展名和 MIME 类型。

除了读取文件内容，我们还可以直接传入一个 Buffer 对象：

const fileType = require('file-type');

// 判断 Buffer 对象的类型
const type = fileType(buffer);

console.log(type);
// => {ext: 'png', mime: 'image/png'}

示例

下面我们来看两个示例：

示例一

假设在 Express 中，我们已经通过 multer 中间件上传了一张图片，现在需要判断图片的类型，如果是 jpeg 类型，就拒绝上传，并返回一个错误提示。

const fileType = require('file-type');
const multer = require('multer');
const express = require('express');

const app = express();
const upload = multer();

app.post('/upload', upload.single('image'), (req, res) => {
  const buffer = req.file.buffer;
  const type = fileType(buffer);

  if (type && type.ext === 'jpg') {
    res.status(400).json({ error: 'JPEG images are not allowed.' });
  } else {
    // handle uploaded image
  }
});

在这个示例中，我们利用 multer 中间件将上传的图片存储到内存中，然后使用 file-type 判断图片类型，如果是 jpeg，则返回 400 错误。

示例二

假设我们有一个图片服务，需要根据图片的类型返回不同的处理结果。我们可以使用 file-type 来判断图片类型，并返回不同的响应。

const fileType = require('file-type');
const fs = require('fs');
const express = require('express');

const app = express();

app.get('/image/:id', (req, res) => {
  const buffer = fs.readFileSync(`path/to/image/${req.params.id}`);
  const type = fileType(buffer);

  if (type && type.ext === 'jpg') {
    res.type('image/jpeg').send(buffer);
  } else if (type && type.ext === 'png') {
    res.type('image/png').send(buffer);
  } else {
    res.status(404).send('Image not found.');
  }
});

在这个示例中，我们通过 Express 框架处理 HTTP 请求，如果请求路径包含 /image/:id，则读取对应的图片文件，使用 file-type 判断图片类型，并返回不同类型的响应。如果图片类型不支持，返回 404 错误。

结论

通过第三方包 file-type，我们可以轻松地判断图片的类型，并且通过类型信息选择性地处理或返回响应，挺方便的。同时，这个包也支持其他类型的文件，例如视频、PDF 等，需要用到时可以参考文档进行使用。