node把md文档中的图片地址换成相对地址

出生场景

我的文章之前都是用写的,然而的文章我复制到自己的博客上,图片就显示不出来。而且多篇文章,图就更多了,我不可能一张一张替换吧。然后朋友就说,那你写个脚本替换不就行了。然后就开始了。

举个例子

  • 一般md图片都长这样的:
![抽奖转盘](http://upload-images.jianshu.io/upload_images/3453108-d3d4ecbe2309e96e.png?imageMogr2/auto-orient/strip%7CimageView2/2/w/1240)
  • 替换之后它长这样的:
![抽奖转盘](./1.jpg)

实现步骤

  1. 循环文件夹,找到文件夹里所有要替换图片的md文件。
  2. 读取md文件内容。
  3. 正则匹配图片路径。
  4. 下载图片。
  5. 保存图片到本地目录。
  6. 替换成相对路径。

具体实现

  • 循环文件夹,找到文件夹里所有要替换图片的md文件:
const postDirPath = path.resolve(__dirname, "./source/_posts");

function main() {
  const files = fs.readdirSync(postDirPath, {
    withFileTypes: true
  });

  files.forEach(file => {
    if (file.isFile()) replaceFile(file);
  });
}
  • 读取md文件内容。
const filePath = path.resolve(postDirPath, file.name);
const fileData = fs.readFileSync(filePath, "utf8");
  • 正则匹配图片路径。
const regex = /\!\[.*\]\((http.*)\)/;

if (!regex.exec(fileData)) return;
const url = regex.exec(fileData)[1];
  • 下载图片。

function download(url) {
  return new Promise((resolve, reject) => {
    const HTTP = url.includes("http://") ? http : https;
    HTTP.get(url, response => {
      let imgData = "";
      response.setEncoding("binary");

      // 有些http链接的图片 需要重定向到HTTPS
      if (response.statusCode == 301) download(response.headers.location);

      response.on("data", chunk => (imgData += chunk));
      response.on("end", () => {
        resolve(imgData);
      });
    }).on("error", err => reject(err));
  });
}
  • 保存图片到本地目录。因为一篇md文章里有很多张图片,所以简单粗暴点,图片名称依次为:1.jpg、2.jpg、3.jpg...
function saveImg(dirName, imgFileName, imgData) {
  const dirPath = path.resolve(postDirPath, dirName);
  if (!fs.existsSync(dirPath)) fs.mkdirSync(dirPath);
  fs.writeFileSync(`${dirPath}/${imgFileName}.jpg`, imgData, "binary");
}
  • 替换成相对路径
function replace(filePath, imgFileName, url) {
  const fileData = fs.readFileSync(filePath, "utf8");
  const newFile = fileData.replace(url, `./${imgFileName}.jpg`);
  fs.writeFileSync(filePath, newFile);
}

最后:

详细代码请查看lingzi的github:https://github.com/lingziyb/replace-img.git

你可能感兴趣的