用 ChatGPT 生成小脚本,直接复制就能用了

用 ChatGPT 生成小脚本,直接复制就能用了

大家好,我是前端西瓜哥。最近我有个个人需求,要写个小脚本。

我有一个基于 hexo 生成的个人静态博客网站,想要给博客的所有的篇文章的尾部加一段广告文案,我不想写这种没啥意思的脚本,就找 ChatGPT 帮我写一段小脚本。

初版

因为我是前端开发,所以我选择让 ChatGPT 给我生成 nodejs 脚本。其他 sh、python 其实也可以,但我真不熟,不好确认代码是否符合需求。

图片

描述很简洁,ChatGPT 给的结果看着不错。

加上细节要求

但我接着发现我有个细节没有提供,就是文件路径需要在文件里提供。我期待通过命令行参数传入路径,因为在 bash 环境下,可以通过 tab 键来补全路径,既方便又不容易出错。

得益于 ChatGPT 的上下文联想能力,我们可以基于上面的问题,加入新的要求。

图片

还提供了使用案例:

图片

完整代码

const fs = require('fs');
const path = require('path');

// 从命令行参数中获取文件夹路径
const directoryPath = process.argv[2];

if (!directoryPath) {
  console.error('请提供要处理的文件夹路径');
  process.exit(1);
}

const suffix = '\n\n这是添加的字符串。'; // 将要添加的字符串放在这里

fs.readdir(directoryPath, (err, files) => {
  if (err) {
    console.error(`无法读取目录:${directoryPath}`, err);
    return;
  }

  // 仅处理 Markdown 文件
  const markdownFiles = files.filter(file => path.extname(file).toLowerCase() === '.md');

  markdownFiles.forEach(file => {
    const filePath = path.join(directoryPath, file);

    // 读取文件内容
    fs.readFile(filePath, 'utf8', (err, data) => {
      if (err) {
        console.error(`无法读取文件:${filePath}`, err);
        return;
      }

      // 将字符串添加到文件末尾
      const content = `${data.trim()}${suffix}`;

      // 将更改保存回原始文件
      fs.writeFile(filePath, content, 'utf8', err => {
        if (err) {
          console.error(`无法写入文件:${filePath}`, err);
          return;
        }

        console.log(`已更新文件:${filePath}`);
      });
    });
  });
});

使用

将代码拷贝到我创建的 js 文件下,将里面的 suffix 替换为自己需要填充的内容。

然后执行:

node scripts/add-footer-qrcode.js source/_posts

图片

可以看到,文件夹下的 100 多个 md 文件的末尾都加上了这段文字。

结尾

我是前端西瓜哥,欢迎关注我,一起感受 AI 的日新月异的变化。

ChatGPT 用来写简单的脚本或算法,那叫一个牛逼。

你要是让我来写,还是挺花时间的,里面小细节挺多,快的话写加测试也得半小时,但 ChatGPT 它来了,它 10s 就给你写好了,我们复制一下,改改就能用,甚至不用改!

发表回复

您的邮箱地址不会被公开。 必填项已用 * 标注