linux 代码统计工具,Shell脚本实现的基于SVN的代码提交量统计工具

最近没啥事，就用bash写了一个基于svn的代码统计小工具。 可以指定统计的目录，默认递归统计子目录。

目前还没有屏蔽指定目录的功能。哈 代码比较粗糙。不过先晒出来。

#!/bin/bash -

#""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""

# FILE: lines.sh

#

# USAGE: ./lines.sh [dir]

# AUTHOR: william

#

# DESCRIPTION: 基于SVN的代码提交量统计工具

# OPTIONS: ---

# CREATED: 06/05/2012 12:49:20 PM CST

#"""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""""

set -o nounset # Treat unset variables as an error

# 关注的文件类型 后罪名

FILES_TYPE="*.cpp *.h *.lua"

# 需要统计的人员,在这里写入需要统计的人,用空格隔开。哈还不智能

declare -r CODER_LIST="coder1 coder2"

declare -i coder1

declare -i coder2

declare -r USAGE="Usage: $0 [dir]. default dir is current dir.\n"

# ERROR CODES;

declare -r E_BAD_PATH=1

declare -r E_INVAILED_ARGU=2

declare -r E_NOT_SVN_DIR=3

#TODO 屏蔽一些dir 还没写哈

# TODO other way get path not with / end

getpath()

{

#debug

#echo dir_name: ${dir_name}

#echo base_name: ${base_name}

if [ $dir_name == "/" ] || [ $base_name == "/" ]; then

work_path="/"

else

work_path=${dir_name}/${base_name}

fi

}

statistic_codelines()

{

if [ -z "$1" ]; then

echo "ERROR statistic_codelines not argument"

return

fi

local pwd_length=${#PWD}

echo "--------------------------"

echo "${PWD}"

for coder in $CODER_LIST; do

local num=$(echo "$1" | grep ${coder} | wc -l)

(( ${coder} += num ))

if [ $num -ne 0 ]; then

printf "%10s | %-7d\n" ${coder} $num

fi

done

echo "--------------------------"

}

# init check argument set work_path

init_work_path()

{

if [ $# -eq 1 ]; then

if [ $1 == "-h" ]; then # is help

echo -e "$USAGE"

elif [ -d $1 ]; then

dir_name=$(dirname ${1})

base_name=$(basename ${1})

getpath;

else

echo -e "An invailed argument"

echo -e "Use -h get help."

exit $E_INVAILED_ARGU

fi

fi

}

# check work_path

check_work_path()

{

if [ -z $work_path ] || [ ! -d $work_path ]; then

exit $E_BADPATH;

fi

}

# enter work_path

enter_work_path()

{

cd ${work_path}

if [ ! $? ]; then

echo "Can not enter ${work_path} "

fi

}

# check work_pat is a svn dir

is_svn_dir()

{

(

# check if current dir is asvn dir

svn info &> /dev/null

exit $?

)

return $?

}

action()

{

local dir_name=.

local base_name=

local work_path=$dir_name

init_work_path $1

check_work_path

enter_work_path #todo can't enter

#echo "NOW DIR: $PWD, OLD DIR $OLDPWD"

is_svn_dir

#todo to next dir

local ret=$?

if [ $ret -ne 0 ]

then

echo -e "Current dir \"${work_path}\" not a svn dir."

exit $E_NOT_SVN_DIR

fi

# get source files

local files=$(ls ${FILES_TYPE} 2> /dev/null)

if [ -n "$files" ]; then

local namelist=$(echo -n ${files} | xargs -n 1 svn blame | awk '{print $2}')

#svn blame $files #| grep $1 | wc -l

statistic_codelines "$namelist"

fi

local sub_dirs=$(find -maxdepth 1 -type d -name "[^.]*" 2>/dev/null)

if [ -n "$sub_dirs" ]; then

for dir in $sub_dirs ; do

action "$dir"

done

fi

cd ..

}

total()

{

echo "-------- TOTOAL ----------"

echo " NAME | lines "

echo "--------------------------"

for coder in $CODER_LIST; do

if [ ${!coder} -ne 0 ]; then

printf "%10s | %-7d\n" ${coder} ${!coder}

fi

done

echo "--------------------------"

}

# main

echo "-----开始统计,请耐心等待.... :) "

action $1

total

exit 0