NodeJs获取文件扩展名

path.extname 是 Node.js 路径模块 (path) 中的一个方法，用于获取文件路径的扩展名。扩展名是指文件名中最后一个 .（点）之后的部分，包括这个 .。

const path = require('path');

const filename = 'example.txt';
const ext = path.extname(filename);

console.log(ext); // 输出: .txt

Node.js 内置的 path.extname 方法只能使用默认的 . 作为扩展名分隔符。如果你需要自定义扩展名分隔符，可以编写一个自定义函数来实现这个功能。

/**
 * 获取自定义扩展名分隔符的文件扩展名
 * @param {string} filename - 文件名
 * @param {string} separator - 自定义扩展名分隔符
 * @returns {string} 文件扩展名，包括分隔符
 */
function getCustomExtname(filename, separator) {
  const index = filename.lastIndexOf(separator);
  if (index === -1 || index === filename.length - 1) {
    return ''; // 没有找到分隔符或分隔符在最后
  }
  return filename.slice(index);
}

// 使用示例
const filename1 = 'example@abc@txt';
const customSeparator1 = '@';
const ext1 = getCustomExtname(filename1, customSeparator1);
console.log(ext1); // 输出: @txt