Node.js中处理带有EXIF方向信息的图片并转换为Base64的实践指南

npm install piexif jimp

const piexif = require('piexifjs');
const fs = require('fs');
const Jimp = require('jimp'); // Jimp will be used in later steps

/**
 * 根据EXIF Orientation值获取图片应旋转的角度
 * @param {number} orientation - EXIF Orientation值 (1-8)
 * @returns {number} 旋转角度 (0, 90, 180, 270)
 */
function getImageAngle(orientation) {
  switch (orientation) {
    case 1: return 0;   // 正常
    case 2: return 0;   // 水平翻转 (通常不直接旋转，由Jimp处理)
    case 3: return 180;  // 旋转180度
    case 4: return 180;  // 垂直翻转 (通常不直接旋转，由Jimp处理)
    case 5: return 270;  // 水平翻转后逆时针旋转90度
    case 6: return 90;   // 顺时针旋转90度
    case 7: return 90;   // 水平翻转后顺时针旋转90度
    case 8: return 270;  // 顺时针旋转270度
    default: return 0;  // 默认不旋转
  }
}

async function processImageWithExifOrientation(imageBuffer, originalPath) {
  // 1. 从图片Buffer中读取EXIF数据
  const exifData = piexif.load(imageBuffer.toString("binary"));

  // 获取EXIF方向值，如果不存在则默认为1 (正常)
  const orientation = exifData["0th"] && exifData["0th"][piexif.ImageIFD.Orientation] ?
                      exifData["0th"][piexif.ImageIFD.Orientation] : 1;

  // 根据EXIF方向获取需要旋转的角度
  const angleToBeRotated = getImageAngle(orientation);

  console.log(`图片EXIF方向: ${orientation}, 需旋转角度: ${angleToBeRotated}度`);

  // ... 后续步骤
  return { angleToBeRotated, orientation };
}

// ... (接上一步代码)

async function processImageWithExifOrientation(imageBuffer, originalPath) {
  // ... (获取 angleToBeRotated 和 orientation)

  // 2. 移除图片中的EXIF数据
  // piexif.remove 返回一个不含EXIF数据的二进制字符串
  const bakedImageBinary = piexif.remove(imageBuffer.toString("binary"));

  // 将处理后的二进制数据写回一个临时文件，供Jimp读取
  const newPath = originalPath.replace(".jpg", "-rotated.jpg").replace(".png", "-rotated.png"); // 示例路径处理
  fs.writeFileSync(newPath, bakedImageBinary, { encoding: "binary" });

  console.log(`EXIF数据已移除，临时文件保存至: ${newPath}`);

  // ... 后续步骤
  return { angleToBeRotated, orientation, newPath };
}

// ... (接上一步代码)

async function processImageWithExifOrientation(imageBuffer, originalPath) {
  // ... (获取 angleToBeRotated, orientation, newPath)

  // 3. 使用Jimp读取临时文件，旋转图片并生成Base64
  const image = await Jimp.read(newPath);

  // 根据计算出的角度旋转图片
  image.rotate(angleToBeRotated);

  // 可选：设置图片质量
  image.quality(90);

  // 将处理后的图片转换为Base64字符串
  const base64Data = await image.getBase64Async(Jimp.AUTO);

  console.log("图片已成功旋转并转换为Base64。");

  // 清理临时文件 (可选)
  fs.unlinkSync(newPath);

  return base64Data;
}

// 示例调用
(async () => {
  const imageFilePath = './your_image.jpg'; // 替换为你的图片路径
  const imageBuffer = fs.readFileSync(imageFilePath);

  try {
    const finalBase64 = await processImageWithExifOrientation(imageBuffer, imageFilePath);
    // console.log(finalBase64.substring(0, 100) + '...'); // 打印Base64的前100字符
    // 现在你可以使用 finalBase64 进行API调用或在Web页面中显示
    console.log("最终Base64数据已生成，长度:", finalBase64.length);
  } catch (error) {
    console.error("处理图片时发生错误:", error);
  }
})();