Node.JS枚举统计当前文件夹和子目录下所有代码文件行数

使用Node.JS的大多数用记事本开发，有时侯会需要统计工程代码量，然后记事本大部分没有这个功能。其实用node.js几行代码就可以实现。

var path  = require('path')
var fs    = require('fs')

//需要统计的文件类型，可自己删减，均小写
var codesFiles = [ '.css', '.js', '.html', '.tmpl', '.part', '.json', '.md', '.txt', '.yml', '.java', '.cs', '.c', '.h', '.cpp', '.xml', '.go', '.py' ]

var LINES = 0

var findFolder = function(srcDir, cb) {
  fs.readdir(srcDir, function(err, files) {
    var count = 0
    var checkEnd = function() {
      ++count == files.length && cb && cb()
    }

    if (err) {
      checkEnd()
      return
    }

    files.forEach(function(file) {
      var extname = path.extname(file).toLowerCase()
      var srcPath = path.join(srcDir, file)

      fs.stat(srcPath, function(err, stats) {
        if (stats.isDirectory()) {
          findFolder(srcPath, checkEnd)
        } else {
          if (codesFiles.indexOf(extname) < 0) {
            checkEnd()
            return
          }

          fs.readFile(srcPath, function(err, data) {
            if (err) {
              checkEnd()
              return
            }

            var lines = data.toString().split('\n')
            LINES += lines.length

            console.log(srcPath, lines.length)
            checkEnd()
          })
        }
      })
    })

    //为空时直接回调
    files.length === 0 && cb && cb()
  })
}

findFolder('./', function() {
  console.log('LINES:', LINES)
})

使用时将此脚本文件命名为lines.js，然后复制到需要统计的文件夹下，然后执行

node lines.js

然后会统计每一个代码文件的长度，和代码总行数：

$ node lines.js
lines.js 56
package.json 6
local\en-US.js 122
local\FE.zh-CN.js 306
...
LINES: 40464

更新

下面的脚本通过检测是否含有ASC0的值来判断文件是不文本文件，然后统计代码行数，但实测统计数量明显偏多。

var path  = require('path')
var fs    = require('fs')

var LINES = 0
var FILES = 0

//https://github.com/hellosean1025/study/blob/master/function.js
function isTextFile( filepath, length ) {
  fd = fs.openSync( filepath, 'r' );
  length = length || 1000;
  for( var i = 0;i < length;i++ ) {
      buf = new Buffer( 1 );
      var bytes = fs.readSync( fd, buf, 0, 1, i );
      char = buf.toString().charCodeAt();
      if ( bytes === 0) {
          return true;
      } else if(bytes === 1 && char === 0) {
          return false;
      }
  }
  return true;
}

var findFolder = function(srcDir, cb) {
  fs.readdir(srcDir, function(err, files) {
    var count = 0
    var checkEnd = function() {
      ++count == files.length && cb && cb()
    }

    if (err) {
      checkEnd()
      return
    }

    files.forEach(function(file) {
      var extname = path.extname(file).toLowerCase()
      var srcPath = path.join(srcDir, file)

      fs.stat(srcPath, function(err, stats) {
        if (stats.isDirectory()) {
          findFolder(srcPath, checkEnd)
        } else {
          // if (codesFiles.indexOf(extname) < 0) {
          //   checkEnd()
          //   return
          // }
          if (!isTextFile(srcPath)) {
            checkEnd()
            return
          }

          fs.readFile(srcPath, function(err, data) {
            if (err) {
              checkEnd()
              return
            }

            var lines = data.toString().split('\n')
            LINES += lines.length

            if (lines.length > 5000) {
              console.trace(srcPath, lines.length)              
            } else {
              console.log(srcPath, lines.length)
            }

            FILES++
            checkEnd()
          })
        }
      })
    })

    //为空时直接回调
    files.length === 0 && cb && cb()
  })
}

findFolder('./', function() {
  console.log('LINES:', LINES)
  console.log('FILES:', FILES)
})