前端学习 node 快速入门系列 —— 项目版权格式化-526互联

其他章节请看：

前端学习 node 快速入门系列

项目版权格式化

需求

替换整个项目的版权信息，替换文件为 .c 和 .h 结尾。

分析

版权信息通常都在文件开头，通过是否有 copyright 来判断是替换版权还是新增版权

实现

通过 node 读取文件，过滤出 .c、.h 文件，然后用正则判断是替换版权还是新增。代码如下：

const fs = require('fs');
const path = require('path');
function walkSync(currentDirPath, callback) {
    fs.readdirSync(currentDirPath, { withFileTypes: true }).forEach(function (dirent) {
        var filePath = path.join(currentDirPath, dirent.name);
        if (dirent.isFile()) {
            callback(filePath, dirent);
        } else if (dirent.isDirectory()) {
            walkSync(filePath, callback);
        }
    });
}

const dir = 'c-project';
walkSync(dir, function (filePath, stat) {
    // 输出特定文件: .c
    const fileList = ['.c', '.h'];
    // 取得文件后缀
    const extname = path.extname(filePath)
    if (!fileList.includes(extname)) {
        return
    }
    formatFile(filePath)
});

// 格式化文件
function formatFile(filePath) {
    fs.readFile(filePath, 'utf8', (err, data) => {
        if (err) throw err;
        // 新的版权信息
        const copyright = 
`/*
* 1
* 2
* COPYRIGHT ph
* 4
* 5
*/
`
        if ((/^(\s|\r|\n)*(\/\*)((\s|.|\r|\n)*?)(\*\/)/im).test(data)) {
            data = data.replace(/^(\s|\r|\n)*(\/\*)((\s|.|\r|\n)*?)(\*\/)/im, copyright)
        } else {
            data = copyright + data
        }

        fs.writeFile(filePath, data, (err) => {
            if (err) throw err;
            console.log('The file has been saved!');
        });

    });
    console.log(filePath)
}

Tip：正则表达式可以通过可视化工具帮助理解，例如这个