一个统计文件夹代码行数的工具

约定不等于承诺〃 2024-04-19 12:24 103阅读 0赞

临时起意想看一下自己到目前为止写了多少行代码了，于是用shell脚本写了一个比较方便的统计工具。代码很简单，有需求的读者可根据自己的需求做一些修改。

#!/bin/bash
    
    #######################################################
    #作用:
    # 本脚本的作用在于统计指定文件夹内，指定类型代码的行数
    #注:
    # 本脚本目前只支持Linux平台
    #选项:
    # 第一个选项为需要统计的文件夹路径
    # -如果本脚本和要统计的文件夹在统一工作目录，则可以使用相对路径
    # -否则务必使用绝对路径
    # 第二个选项为需要统计的代码类型，有以下类型：
    # -c: C语言源文件代码
    # -cc: C++语言源文件代码
    # -h: C/C++语言头文件代码
    # -p: Python语言代码
    # -g: Go语言代码
    # -j: Java语言代码
    #
    #作者:aideny
    #######################################################
    
    
    path=$1
    code_type=$2
    
    option=""
    
    if test ${code_type} = "-c"
    then 
      option="*.c"
    elif test ${code_type} = "-cc"
    then
      option="*.cc"
    elif test ${code_type} = "-h"
    then
      option="*.h"
    elif test ${code_type} = "-p"
    then
      option="*.py"
    elif test ${code_type} = "-g"
    then
      option="*.go"
    elif test ${code_type} = "-j"
    then
      option="*.java"
    else 
      echo "检查选项是否支持！"
      exit 1
    fi
    
    echo "开始统计..."
    
    # 进行统计
    if test -z ${path}
    then
      echo "Usage: ./cl.sh [filepath][option]"
    else 
      # 这个方法如果找不到指定语言的代码，会卡住
      # wc -l `find ${path} -name ${option}` | tail -n1
      find ${path} -name ${option} | xargs wc -l | tail -n1
    fi