mirror of
https://github.com/juewuy/ShellCrash.git
synced 2026-03-10 23:41:22 +00:00
修复进度条+增加文件大小显示
cdn下载链接不够规范，不给content-length，导致curl原生进度条失效。 通过小巧思拿到文件大小，再手搓了一个短小精悍的进度条进行修复。 并且在下载之前增加了文件大小显示，让用户不会陷入下载焦虑~
This commit is contained in:
小小小猫仔
@@ -2,7 +2,7 @@
|
||||
|
||||
webget(){
|
||||
if pidof CrashCore >/dev/null; then
|
||||
setproxy #设置临时代理
|
||||
setproxy #设置临时代理，【$1】代表下载目录，【$2】代表在线地址
|
||||
url=$(printf '%s\n' "$2" |
|
||||
sed -e 's#https://.*jsdelivr.net/gh/juewuy/ShellCrash[@|/]#https://raw.githubusercontent.com/juewuy/ShellCrash/#' \
|
||||
-e 's#https://gh.jwsc.eu.org/#https://raw.githubusercontent.com/juewuy/ShellCrash/#')
|
||||
@@ -10,10 +10,66 @@ webget(){
|
||||
url=$(printf '%s\n' "$2" |
|
||||
sed 's#https://raw.githubusercontent.com/juewuy/ShellCrash/#https://testingcf.jsdelivr.net/gh/juewuy/ShellCrash@#')
|
||||
fi
|
||||
|
||||
# === 新增模块 1：大小探测 (仅显示用) ===
|
||||
fsize_raw=0
|
||||
if [ "$3" != "echooff" ]; then
|
||||
# 探测头信息
|
||||
header=$(curl -sIL --connect-timeout 2 "$url")
|
||||
# 代理容错
|
||||
[ -z "$header" ] && { export https_proxy=""; export http_proxy=""; header=$(curl -sIL --connect-timeout 2 "$url"); }
|
||||
|
||||
# 提取大小 (优先 Content-Length，其次 ETag)
|
||||
fsize_raw=$(echo "$header" | grep -i 'Content-Length' | tail -n 1 | awk '{print $2}' | tr -d '\r' | awk '{print int($1)}')
|
||||
if [ -z "$fsize_raw" ] || [ "$fsize_raw" -eq 0 ]; then
|
||||
etag=$(echo "$header" | grep -i 'etag' | tail -n 1 | cut -d '"' -f2 | cut -d '-' -f1)
|
||||
[ -n "$etag" ] && fsize_raw=$(printf "%d" 0x$etag 2>/dev/null)
|
||||
fi
|
||||
# 显示文件大小
|
||||
[ -n "$fsize_raw" ] && [ "$fsize_raw" -gt 0 ] && echo "文件大小: $(awk -v n=$fsize_raw 'BEGIN {printf "%.2f", n/1048576}') MB"
|
||||
fi
|
||||
|
||||
# === 新增模块 2：手搓进度条引擎 (函数定义) ===
|
||||
execute_curl(){
|
||||
local path="$1" target_url="$2" total_size="$3" extra_args="$4"
|
||||
rm -f /tmp/webget_res
|
||||
# 后台静默下载，状态码写入临时文件
|
||||
curl $extra_args -s -L -w '%{http_code}' "$target_url" -o "$path" > /tmp/webget_res &
|
||||
local pid=$!
|
||||
|
||||
# 循环监控
|
||||
while kill -0 $pid 2>/dev/null; do
|
||||
if [ -f "$path" ]; then
|
||||
local curr=$(wc -c < "$path")
|
||||
local pct=$(awk -v c=$curr -v t=$total_size 'BEGIN {p=(c*100/t); if(p>100)p=100; printf "%.1f", p}')
|
||||
|
||||
# 40字符安全宽度
|
||||
local bar_max=40
|
||||
local num=$(awk -v p=$pct -v w=$bar_max 'BEGIN {printf "%d", p*w/100}')
|
||||
local bar=$(printf "%${num}s" | tr ' ' '#'); local spc_n=$((bar_max - num))
|
||||
local spc=""; [ "$spc_n" -gt 0 ] && spc=$(printf "%${spc_n}s")
|
||||
|
||||
# \033[2K 清除整行
|
||||
printf "\r\033[2K%s%s %6s%%" "$bar" "$spc" "$pct" >&2
|
||||
fi
|
||||
usleep 200000 2>/dev/null || sleep 1
|
||||
done
|
||||
|
||||
local code=$(cat /tmp/webget_res 2>/dev/null)
|
||||
if [ "$code" = "200" ] || [ "$code" = "206" ]; then
|
||||
local full=$(printf "%40s" | tr ' ' '#')
|
||||
printf "\r\033[2K%s 100.0%%\n" "$full" >&2
|
||||
else
|
||||
printf "\r\033[2K" >&2; [ -f "$path" ] && rm -f "$path"
|
||||
fi
|
||||
echo "$code"
|
||||
}
|
||||
# ===============================================
|
||||
|
||||
#参数【$1】代表下载目录，【$2】代表在线地址
|
||||
#参数【$3】代表输出显示，【$4】不启用重定向
|
||||
#参数【$5】代表验证证书，【$6】使用自定义UA
|
||||
[ -n "$6" ] && agent="--user-agent $6"
|
||||
[ -n "$6" ] && agent="--user-agent=$6"
|
||||
if wget --help 2>&1 | grep -q 'show-progress' >/dev/null 2>&1; then
|
||||
[ "$3" = "echooff" ] && progress='-q' || progress='-q --show-progress'
|
||||
[ "$4" = "rediroff" ] && redirect='--max-redirect=0' || redirect=''
|
||||
@@ -33,16 +89,36 @@ webget(){
|
||||
else
|
||||
certificate='-k'
|
||||
fi
|
||||
|
||||
# 判断是否启用手搓进度条 (CDN 且 非静默 且 有大小)
|
||||
use_manual_bar="" # <--- 这里加上了初始化
|
||||
has_cl=$(echo "$header" | grep -iq 'Content-Length' && echo "yes")
|
||||
[ "$3" != "echooff" ] && [ "$has_cl" != "yes" ] && [ "$fsize_raw" -gt 0 ] && use_manual_bar="yes"
|
||||
|
||||
auth_arg=""
|
||||
if curl --version | grep -q '^curl 8.' && ckcmd base64; then
|
||||
auth_b64=$(printf '%s' "$authentication" | base64)
|
||||
result=$(curl $agent -w '%{http_code}' --connect-timeout 3 --proxy-header "Proxy-Authorization: Basic $auth_b64" $progress $redirect $certificate -o "$1" "$url")
|
||||
else
|
||||
result=$(curl $agent -w '%{http_code}' --connect-timeout 3 $progress $redirect $certificate -o "$1" "$url")
|
||||
[ -n "$auth_b64" ] && auth_arg="--proxy-header Proxy-Authorization:Basic $auth_b64"
|
||||
fi
|
||||
|
||||
# --- 第一次下载 ---
|
||||
if [ "$use_manual_bar" = "yes" ]; then
|
||||
result=$(execute_curl "$1" "$url" "$fsize_raw" "$agent $auth_arg $redirect $certificate")
|
||||
else
|
||||
result=$(curl $agent $auth_arg -w '%{http_code}' --connect-timeout 3 $progress $redirect $certificate -o "$1" "$url")
|
||||
fi
|
||||
|
||||
[ "$result" = "200" ] && return 0 #成功则退出否则重试
|
||||
|
||||
# --- Fallback 重试 ---
|
||||
export https_proxy=""
|
||||
export http_proxy=""
|
||||
result=$(curl $agent -w '%{http_code}' --connect-timeout 5 $progress $redirect $certificate -o "$1" "$2")
|
||||
|
||||
if [ "$use_manual_bar" = "yes" ]; then
|
||||
result=$(execute_curl "$1" "$2" "$fsize_raw" "$agent $redirect $certificate")
|
||||
else
|
||||
result=$(curl $agent -w '%{http_code}' --connect-timeout 5 $progress $redirect $certificate -o "$1" "$2")
|
||||
fi
|
||||
[ "$result" = "200" ]
|
||||
return $?
|
||||
elif ckcmd wget;then
|
||||
@@ -54,4 +130,4 @@ webget(){
|
||||
echo "No Curl or Wget！！！"
|
||||
return 1
|
||||
fi
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user