107 lines
3.8 KiB
Bash
107 lines
3.8 KiB
Bash
#!/usr/bin/env bash
|
||
# 阿里云百炼 Token Plan 余量/配额查询 — 依赖 bl CLI 的控制台登录（一次性授权）
|
||
# 用法: bash bailian-token-plan.sh [--summary]
|
||
# --summary 额外输出统一用量摘要（免费额度 + 近期用量）
|
||
set -euo pipefail
|
||
|
||
if ! command -v bl >/dev/null 2>&1; then
|
||
echo "❌ 未找到 bl CLI，请先: npm install -g bailian-cli" >&2
|
||
exit 1
|
||
fi
|
||
|
||
CFG_FILE="$HOME/.bailian/config.json"
|
||
# macOS / Linux 兼容的 mtime 秒数
|
||
cfg_mtime() {
|
||
if [ -f "$CFG_FILE" ]; then
|
||
if command -v stat >/dev/null 2>&1; then
|
||
stat -f %m "$CFG_FILE" 2>/dev/null || stat -c %Y "$CFG_FILE" 2>/dev/null || echo 0
|
||
else
|
||
echo 0
|
||
fi
|
||
else
|
||
echo 0
|
||
fi
|
||
}
|
||
|
||
# 尝试查询；失败原因写入 /tmp/bailian-token-plan.err
|
||
query_token_plan() {
|
||
echo "🔍 查询百炼 Token Plan 配额 ..." >&2
|
||
bl usage token-plan --output json > /tmp/bailian-token-plan.json 2>/tmp/bailian-token-plan.err
|
||
}
|
||
|
||
# 控制台会话过期/未登录时自动续期：拉起浏览器授权，等 config.json 刷新，然后重试。
|
||
# 前提：本机浏览器保持阿里云登录态（勾选「自动登录」），授权页会自动完成回调，无需点击。
|
||
relogin_console() {
|
||
local before after mtime_before
|
||
mtime_before=$(cfg_mtime)
|
||
echo "⚠️ 控制台会话已过期/未登录，自动发起浏览器续期（15 分钟内有效）..." >&2
|
||
bl auth login --console --console-site domestic >/tmp/bailian-relogin.log 2>&1 &
|
||
local LPID=$!
|
||
|
||
# 最多等 150s：config.json 被刷新，或登录进程退出
|
||
local i
|
||
for i in $(seq 1 30); do
|
||
sleep 5
|
||
after=$(cfg_mtime)
|
||
if [ "$after" != "$mtime_before" ] && [ "$after" != "0" ]; then
|
||
wait "$LPID" 2>/dev/null || true
|
||
echo "✅ 控制台授权完成，继续查询" >&2
|
||
return 0
|
||
fi
|
||
if ! kill -0 "$LPID" 2>/dev/null; then
|
||
wait "$LPID" 2>/dev/null || true
|
||
break
|
||
fi
|
||
done
|
||
|
||
# 仍然没刷新：杀掉可能残留的登录进程，提示手动完成
|
||
kill "$LPID" 2>/dev/null || true
|
||
wait "$LPID" 2>/dev/null || true
|
||
echo "⚠️ 未检测到授权回调完成。请检查是否弹出浏览器并完成授权（保持阿里云「自动登录」可免点击），然后重跑本脚本。" >&2
|
||
echo " 浏览器授权页若已关闭，可手动执行: bl auth login --console --console-site domestic" >&2
|
||
return 1
|
||
}
|
||
|
||
if ! query_token_plan; then
|
||
if grep -qiE "not logged in|has expired|NotLogined" /tmp/bailian-token-plan.err; then
|
||
if ! relogin_console || ! query_token_plan; then
|
||
echo "❌ bl usage token-plan 仍失败:" >&2
|
||
cat /tmp/bailian-token-plan.err >&2
|
||
exit 1
|
||
fi
|
||
else
|
||
echo "❌ bl usage token-plan 失败:" >&2
|
||
cat /tmp/bailian-token-plan.err >&2
|
||
exit 1
|
||
fi
|
||
fi
|
||
cat /tmp/bailian-token-plan.json
|
||
echo
|
||
|
||
# 友好摘要：周配额已用/剩余比例 + 重置时间（本地时区）
|
||
if command -v python3 >/dev/null 2>&1; then
|
||
python3 - <<'PY'
|
||
import json, sys, datetime, time
|
||
try:
|
||
d = json.load(open('/tmp/bailian-token-plan.json'))
|
||
except Exception as e:
|
||
print(f"⚠️ 无法解析 JSON: {e}", file=sys.stderr)
|
||
sys.exit(0)
|
||
used = d.get('per1WeekPercentage')
|
||
reset = d.get('per1WeekResetTime')
|
||
if used is not None:
|
||
used_pct = used * 100
|
||
print(f"📊 本周 Token Plan 已用: {used_pct:.2f}% | 剩余约为: {100 - used_pct:.2f}%")
|
||
else:
|
||
print("⚠️ 响应中没有 per1WeekPercentage 字段（结构可能已变化），请参考上方原始 JSON")
|
||
if reset:
|
||
dt = datetime.datetime.fromtimestamp(reset / 1000)
|
||
tzname = time.strftime('%Z')
|
||
print(f"🔄 本周配额重置时间: {dt.strftime('%Y-%m-%d %H:%M:%S')}（本地时区 {tzname or 'local'}）")
|
||
PY
|
||
fi
|
||
|
||
if [ "${1:-}" = "--summary" ]; then
|
||
echo "=== 统一用量摘要（免费额度等） ===" >&2
|
||
bl usage summary --output json
|
||
fi |