let fs = require('fs')
let path = require('path')
// 获取命令行参数
let parm = process.argv.splice(2)
// 第一个参数是路径
let rootPath = parm[0]
// 后面的所有参数都是文件后缀
let types = parm.splice(1)
// 需要过滤的文件夹
let filter = ['./node_modules']
// 统计结果
let num = 0
// 获取行数
async function line(path) {
let rep = await fs.readFileSync(path)
rep = rep.toString()
let lines = rep.split('\n')
console.log(path + ' ' + lines.length)
num += lines.length
}
// 递归所有文件夹统计
async function start(pt) {
let files = fs.readdirSync(pt)
files
.map(file => {
return `${pt}/${file}`
})
.forEach(file => {
let stat = fs.statSync(file)
if (stat.isDirectory()) {
if (filter.indexOf(pt) != -1) {
return
}
start(file)
return
}
let ext = path.extname(file)
if (types.indexOf(ext) != -1) {
line(file)
}
})
}
;(async () => {
await start(rootPath)
console.log(`总代码行数：${num}`)
})()
使用方法
// 统计当前文件夹下所有 .js .json 代码行数（相对路径）
node ./app.js . .js .json
// 统计 E:\pro\aliyun 目录下 .java .kt .xml .c 代码行数（绝对路径）
node ./app.js E:\pro\aliyun .java .kt .xml .c
