Shell script quickly counts the number of lines of project code

demand

Recently, I suddenly want to see how many lines of code there are in the project under development , But the project is relatively large , It's too troublesome to calculate manually . I think of it. shell Script , Just give it a path , Wait for the statistical results of the script .

Realize the idea

1、 Get all the files in the given path
2、 Determine whether the file is a directory or an ordinary file , If it's a catalog , Continue traversing ; If it's a regular document , next step
3、 Determine file type , Judge according to the file suffix , Such as c++ The suffixes of the source code are .h and .cpp
4、 Calculate the number of matching file lines , Add it up

demo

#!/bin/bash

#  Define global variables , Calculate the total number 
line=0

# function listFiles: Get the total number of source lines under the specified path 
# $1  The full path 
# $2  Separator , Used to display the indenting effect of multi-level directories 
function listFiles()
{
    
	echo "path:$1"
	for file in $(ls $1)
	do
	#  If it is a folder, continue to traverse 
	if [ -d $file ]; then
		echo "$file is dir"
		listFiles $1/$file " $2"
	else
		#  It can be modified here as java、py Wait for the file suffix you want to count 
		if [ ${file##*.} == "h" -o ${file##*.} == "cpp" ];then
		echo " $2" "$1/$file line:$(wc -l $1/$file | awk '{print $1}')"
		#  Add up 
		let line+=$(wc -l $1/$file | awk '{print $1}')
		echo "current line:$line"
		fi
	fi
	done
}

listFiles $1 ""

echo "total line:$line"

effect

The verification results

It can be seen that ,shell Scripted .h and .cpp The number of lines in the file is correct . You can modify the file suffix to count the number of different source lines ,coding The fun of is also here , Haha, calculate how big the source code of the project is ~