265 lines
8.2 KiB
Bash
265 lines
8.2 KiB
Bash
#!/usr/bin/env bash
|
||
# 代码行数统计器 - 增强版 (Bash 版)
|
||
# 尽量等效还原原 PowerShell 脚本功能与输出
|
||
|
||
set -uo pipefail
|
||
|
||
# 彩色输出映射（接近 PS 的颜色名）
|
||
declare -A COLOR_MAP=(
|
||
["Cyan"]="\e[36m"
|
||
["Green"]="\e[32m"
|
||
["Yellow"]="\e[33m"
|
||
["Magenta"]="\e[35m"
|
||
["Blue"]="\e[34m"
|
||
["Red"]="\e[31m"
|
||
["White"]="\e[97m"
|
||
["Gray"]="\e[90m"
|
||
["DarkYellow"]="\e[33m"
|
||
["Reset"]="\e[0m"
|
||
)
|
||
|
||
# 配置：类型 → 扩展列表、颜色
|
||
# 说明：扩展以 "*.ext" 形式书写；内部会派生出 ".ext" 用于匹配
|
||
declare -A TYPE_COLOR
|
||
declare -A TYPE_EXTS
|
||
TYPE_COLOR["C/C++"]="Cyan"
|
||
TYPE_EXTS["C/C++"]='*.c *.cpp *.cc *.cxx *.h *.hpp *.hh *.hxx'
|
||
|
||
TYPE_COLOR["C#"]="Green"
|
||
TYPE_EXTS["C#"]='*.cs'
|
||
|
||
TYPE_COLOR["Java"]="Yellow"
|
||
TYPE_EXTS["Java"]='*.java'
|
||
|
||
TYPE_COLOR["Python"]="Magenta"
|
||
TYPE_EXTS["Python"]='*.py *.pyw'
|
||
|
||
TYPE_COLOR["JavaScript"]="Blue"
|
||
TYPE_EXTS["JavaScript"]='*.js *.jsx *.ts *.tsx'
|
||
|
||
TYPE_COLOR["Web"]="Red"
|
||
TYPE_EXTS["Web"]='*.html *.htm *.css *.scss *.less *.sass'
|
||
|
||
TYPE_COLOR["Shell"]="White"
|
||
TYPE_EXTS["Shell"]='*.sh *.bash *.ps1 *.psm1 *.psd1'
|
||
|
||
TYPE_COLOR["配置/数据"]="Gray"
|
||
TYPE_EXTS["配置/数据"]='*.xml *.json *.yaml *.yml *.ini *.cfg *.config'
|
||
|
||
TYPE_COLOR["其他代码"]="DarkYellow"
|
||
TYPE_EXTS["其他代码"]='*.go *.rs *.rb *.php *.swift *.kt *.scala'
|
||
|
||
# 构建 扩展后缀(.ext) → 类型 的快速映射
|
||
declare -A SUFFIX_TYPE
|
||
for t in "${!TYPE_EXTS[@]}"; do
|
||
for ext in ${TYPE_EXTS[$t]}; do
|
||
# 将 "*.py" 转为 ".py"；统一小写匹配
|
||
suf=".${ext#*.}"
|
||
suf="$(printf '%s' "$suf" | tr '[:upper:]' '[:lower:]')"
|
||
SUFFIX_TYPE["$suf"]="$t"
|
||
done
|
||
done
|
||
|
||
# 标题
|
||
printf "%b========================================%b\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b 代码行数统计%b\n" "${COLOR_MAP[Yellow]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b========================================%b\n\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
|
||
# 当前目录
|
||
current_dir="$(pwd)"
|
||
printf "%b扫描目录: %s%b\n\n" "${COLOR_MAP[Gray]}" "$current_dir" "${COLOR_MAP[Reset]}"
|
||
|
||
# 统计变量
|
||
total_lines=0
|
||
total_files=0
|
||
declare -A type_files
|
||
declare -A type_lines
|
||
start_time_ns=$(date +%s%N)
|
||
|
||
# 初始化类型统计
|
||
for t in "${!TYPE_EXTS[@]}"; do
|
||
type_files["$t"]=0
|
||
type_lines["$t"]=0
|
||
done
|
||
|
||
# 收集全部文件（按各扩展累积）
|
||
# 说明：尽量贴近 PS：按类型/扩展多次递归收集再合并
|
||
# 使用 -iname 实现大小写不敏感匹配
|
||
mapfile -d '' ALL_FILES < <(
|
||
{
|
||
for t in "${!TYPE_EXTS[@]}"; do
|
||
for ext in ${TYPE_EXTS[$t]}; do
|
||
find "$current_dir" -type f -iname "${ext}" -print0 2>/dev/null
|
||
done
|
||
done
|
||
} | awk -v RS='\0' '!seen[$0]++{printf "%s\0",$0}' # 去重，保持与 PS 相近的行为
|
||
)
|
||
|
||
total_to_process=${#ALL_FILES[@]}
|
||
|
||
if (( total_to_process == 0 )); then
|
||
printf "%b未找到任何源代码文件！%b\n" "${COLOR_MAP[Red]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b请检查当前目录是否包含源代码文件。%b\n" "${COLOR_MAP[Gray]}" "${COLOR_MAP[Reset]}"
|
||
exit 0
|
||
fi
|
||
|
||
printf "%b找到 %d 个文件需要处理...%b\n\n" "${COLOR_MAP[Gray]}" "$total_to_process" "${COLOR_MAP[Reset]}"
|
||
|
||
# 简易进度显示
|
||
processed=0
|
||
|
||
# 为格式对齐做准备
|
||
pad_right() { # $1=text $2=width
|
||
local s="$1" w="$2" len=${#1}
|
||
if (( len < w )); then printf "%s%*s" "$s" $((w-len)) ""; else printf "%s" "$s"; fi
|
||
}
|
||
|
||
# 统计循环
|
||
for file in "${ALL_FILES[@]}"; do
|
||
((processed++))
|
||
# 百分比
|
||
percent=$(( processed * 100 / total_to_process ))
|
||
|
||
# 进度行（覆盖式）
|
||
base="${file##*/}"
|
||
printf "\r正在统计代码行数 | 处理文件: %s | 已处理: %d / %d | %3d%%" \
|
||
"$(pad_right "$base" 30)" "$processed" "$total_to_process" "$percent"
|
||
|
||
# 读取行数
|
||
lines=0
|
||
if ! lines=$(wc -l <"$file" 2>/dev/null); then
|
||
# 模拟 PS 的警告输出（另起一行）
|
||
printf "\n%b 警告: 无法读取文件 %s%b\n" "${COLOR_MAP[Red]}" "$base" "${COLOR_MAP[Reset]}"
|
||
continue
|
||
fi
|
||
|
||
(( total_lines += lines ))
|
||
(( total_files++ ))
|
||
|
||
# 分类归属（根据扩展名后缀）
|
||
fname_lower="$(printf '%s' "$file" | tr '[:upper:]' '[:lower:]')"
|
||
# 取 ".ext"（含点），若没有点，则设为空
|
||
suffix=""
|
||
if [[ "$fname_lower" == *.* ]]; then
|
||
suffix=".${fname_lower##*.}"
|
||
fi
|
||
|
||
t="${SUFFIX_TYPE[$suffix]:-}"
|
||
if [[ -n "$t" ]]; then
|
||
(( type_files["$t"] += 1 ))
|
||
(( type_lines["$t"] += lines ))
|
||
fi
|
||
done
|
||
|
||
# 完成进度条
|
||
printf "\n"
|
||
|
||
# 耗时
|
||
end_time_ns=$(date +%s%N)
|
||
duration_ns=$(( end_time_ns - start_time_ns ))
|
||
# 转为 mm:ss.mmm
|
||
total_ms=$(( duration_ns / 1000000 ))
|
||
mm=$(( (total_ms / 60000) % 60 ))
|
||
ss=$(( (total_ms / 1000) % 60 ))
|
||
ms=$(( total_ms % 1000 ))
|
||
|
||
# 详细统计标题
|
||
printf "\n%b========================================%b\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b 统计结果详情%b\n" "${COLOR_MAP[Yellow]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b========================================%b\n\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
|
||
# 按类型按行数降序输出
|
||
# 收集有文件的类型
|
||
types_sorted=()
|
||
while IFS= read -r line; do
|
||
types_sorted+=("$line")
|
||
done < <(
|
||
for t in "${!TYPE_EXTS[@]}"; do
|
||
if (( type_files["$t"] > 0 )); then
|
||
printf "%s\t%d\n" "$t" "${type_lines["$t"]}"
|
||
fi
|
||
done | sort -k2,2nr | awk -F'\t' '{print $1}'
|
||
)
|
||
|
||
for t in "${types_sorted[@]}"; do
|
||
files=${type_files["$t"]}
|
||
lines=${type_lines["$t"]}
|
||
if (( files > 0 )); then
|
||
percent=0
|
||
if (( total_lines > 0 )); then
|
||
# 保留两位小数
|
||
percent=$(awk -v a="$lines" -v b="$total_lines" 'BEGIN{printf("%.2f", (b>0)?(a*100.0/b):0)}')
|
||
fi
|
||
color_name="${TYPE_COLOR[$t]}"
|
||
color="${COLOR_MAP[$color_name]}"
|
||
reset="${COLOR_MAP[Reset]}"
|
||
printf "%b%-15s 文件: %4d 行数: %8d (%6s%%)%b\n" "$color" "$t" "$files" "$lines" "$percent" "$reset"
|
||
fi
|
||
done
|
||
|
||
# 汇总
|
||
printf "\n%b========================================%b\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b汇总统计%b\n" "${COLOR_MAP[Yellow]}" "${COLOR_MAP[Reset]}"
|
||
printf "%b========================================%b\n" "${COLOR_MAP[White]}" "${COLOR_MAP[Reset]}"
|
||
|
||
printf "%b总文件数: %d%b\n" "${COLOR_MAP[Cyan]}" "$total_files" "${COLOR_MAP[Reset]}"
|
||
printf "%b总代码行: %d%b\n" "${COLOR_MAP[Green]}" "$total_lines" "${COLOR_MAP[Reset]}"
|
||
printf "%b处理文件: %d / %d%b\n" "${COLOR_MAP[Gray]}" "$processed" "$total_to_process" "${COLOR_MAP[Reset]}"
|
||
printf "%b统计耗时: %02d:%02d.%03d%b\n" "${COLOR_MAP[Magenta]}" "$mm" "$ss" "$ms" "${COLOR_MAP[Reset]}"
|
||
|
||
# 文件大小统计
|
||
total_size_bytes=0
|
||
# 使用 stat 读取字节数（兼容 GNU/ BSD）
|
||
stat_size() {
|
||
local f="$1"
|
||
if stat --version >/dev/null 2>&1; then
|
||
stat -c %s -- "$f"
|
||
else
|
||
stat -f %z -- "$f"
|
||
fi
|
||
}
|
||
for f in "${ALL_FILES[@]}"; do
|
||
if sz=$(stat_size "$f" 2>/dev/null); then
|
||
(( total_size_bytes += sz ))
|
||
fi
|
||
done
|
||
avg_size_bytes=0
|
||
if (( total_files > 0 )); then
|
||
avg_size_bytes=$(( total_size_bytes / total_files ))
|
||
fi
|
||
to_mb=$(awk -v b="$total_size_bytes" 'BEGIN{printf("%.2f", b/1024/1024)}')
|
||
avg_kb=$(awk -v b="$avg_size_bytes" 'BEGIN{printf("%.2f", b/1024)}')
|
||
|
||
printf "%b总文件大小: %s MB%b\n" "${COLOR_MAP[Blue]}" "$to_mb" "${COLOR_MAP[Reset]}"
|
||
printf "%b平均文件大小: %s KB%b\n" "${COLOR_MAP[Blue]}" "$avg_kb" "${COLOR_MAP[Reset]}"
|
||
|
||
# 导出 CSV
|
||
printf "\n"
|
||
read -r -p "是否导出统计结果到CSV文件? (Y/N) " exportChoice
|
||
if [[ "$exportChoice" == "Y" || "$exportChoice" == "y" ]]; then
|
||
timestamp="$(date +%Y%m%d_%H%M%S)"
|
||
csv_path="${current_dir}/code_stats_${timestamp}.csv"
|
||
{
|
||
# 头部
|
||
printf "文件类型,文件数量,代码行数,占比百分比\n"
|
||
for t in "${types_sorted[@]}"; do
|
||
files=${type_files["$t"]}
|
||
lines=${type_lines["$t"]}
|
||
if (( files > 0 )); then
|
||
percent=$(awk -v a="$lines" -v b="$total_lines" 'BEGIN{printf("%.2f", (b>0)?(a*100.0/b):0)}')
|
||
# 以 CSV 形式输出（简单场景，t 不含逗号）
|
||
printf "%s,%d,%d,%s\n" "$t" "$files" "$lines" "$percent"
|
||
fi
|
||
done
|
||
# 汇总行
|
||
printf "总计,%d,%d,100\n" "$total_files" "$total_lines"
|
||
} > "$csv_path"
|
||
printf "%b统计结果已导出到: %s%b\n" "${COLOR_MAP[Green]}" "$csv_path" "${COLOR_MAP[Reset]}"
|
||
fi
|
||
|
||
# 任意键退出
|
||
printf "\n%b统计完成！按任意键退出...%b" "${COLOR_MAP[Gray]}" "${COLOR_MAP[Reset]}"
|
||
# shellcheck disable=SC2162
|
||
read -rsn1 _ 2>/dev/null || true
|
||
printf "\n"
|