feat: 添加会员订阅检查脚本 check-members.sh
This commit is contained in:
201
scripts/check-members.sh
Executable file
201
scripts/check-members.sh
Executable file
@@ -0,0 +1,201 @@
|
||||
#!/bin/bash
|
||||
# ===========================================
|
||||
# 检查会员订阅情况
|
||||
# ===========================================
|
||||
# 用法:
|
||||
# ./check-members.sh # 生产数据库
|
||||
# ./check-members.sh --dev # 开发数据库 (milkydata_dev)
|
||||
# ./check-members.sh --dev --json # JSON 格式输出
|
||||
# ===========================================
|
||||
|
||||
set -euo pipefail
|
||||
|
||||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
|
||||
ENV_FILE="$PROJECT_ROOT/.env"
|
||||
|
||||
# ---- 颜色 ----
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
CYAN='\033[0;36m'
|
||||
BOLD='\033[1m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# ---- 参数 ----
|
||||
DB_NAME="milkydata"
|
||||
OUTPUT_JSON=false
|
||||
|
||||
while [[ $# -gt 0 ]]; do
|
||||
case "$1" in
|
||||
--dev) DB_NAME="milkydata_dev"; shift ;;
|
||||
--json) OUTPUT_JSON=true; shift ;;
|
||||
--help|-h)
|
||||
echo "用法: $0 [--dev] [--json]"
|
||||
echo " --dev 查询开发数据库 (milkydata_dev)"
|
||||
echo " --json JSON 格式输出"
|
||||
exit 0
|
||||
;;
|
||||
*) echo "未知参数: $1"; exit 1 ;;
|
||||
esac
|
||||
done
|
||||
|
||||
# ---- 获取 SSH 配置 ----
|
||||
SSH_HOST=$(grep '^SSH_SERVER=' "$ENV_FILE" 2>/dev/null | head -1 | cut -d'=' -f2- | tr -d '"' | tr -d "'" | xargs)
|
||||
if [ -z "$SSH_HOST" ]; then
|
||||
echo "${RED}❌ 未找到 SSH_SERVER 配置，请检查 $ENV_FILE${NC}"
|
||||
exit 1
|
||||
fi
|
||||
|
||||
echo -e "${CYAN}══════════════════════════════════════${NC}"
|
||||
echo -e "${CYAN} 会员订阅状态检查${NC}"
|
||||
echo -e "${CYAN} 数据库: ${BOLD}$DB_NAME${NC}"
|
||||
echo -e "${CYAN} 时间: $(date '+%Y-%m-%d %H:%M:%S')${NC}"
|
||||
echo -e "${CYAN}══════════════════════════════════════${NC}"
|
||||
echo ""
|
||||
|
||||
# ---- 总览 ----
|
||||
SQL_OVERVIEW=$(cat <<'SQL'
|
||||
SELECT
|
||||
COUNT(*) AS total_users,
|
||||
COUNT(*) FILTER (WHERE is_member = true) AS total_members,
|
||||
COUNT(*) FILTER (WHERE is_member = true AND membership_expires_at > NOW()) AS active_members,
|
||||
COUNT(*) FILTER (WHERE is_member = true AND membership_expires_at <= NOW()) AS expired_members,
|
||||
COUNT(*) FILTER (WHERE is_member = true AND membership_expires_at IS NULL) AS permanent_members,
|
||||
COUNT(*) FILTER (WHERE is_member = true
|
||||
AND membership_expires_at IS NOT NULL
|
||||
AND membership_expires_at > NOW()
|
||||
AND membership_expires_at <= NOW() + INTERVAL '7 days') AS expiring_7d,
|
||||
COUNT(*) FILTER (WHERE is_member = true
|
||||
AND membership_expires_at IS NOT NULL
|
||||
AND membership_expires_at > NOW()
|
||||
AND membership_expires_at <= NOW() + INTERVAL '30 days') AS expiring_30d
|
||||
FROM users;
|
||||
SQL
|
||||
)
|
||||
|
||||
OVERVIEW=$(ssh "$SSH_HOST" "psql -d $DB_NAME -At -c \"$SQL_OVERVIEW\"" 2>/dev/null) || {
|
||||
echo -e "${RED}❌ SSH 连接失败: $SSH_HOST${NC}"
|
||||
exit 1
|
||||
}
|
||||
|
||||
IFS='|' read -r total_users total_members active_members expired_members permanent_members expiring_7d expiring_30d <<< "$OVERVIEW"
|
||||
|
||||
# ---- 活跃会员列表 ----
|
||||
SQL_LIST=$(cat <<'SQL'
|
||||
SELECT
|
||||
id,
|
||||
COALESCE(openid, 'N/A') AS openid,
|
||||
COALESCE(nickname, CONCAT('用户 #', id)) AS display_name,
|
||||
membership_expires_at::date AS expires_on,
|
||||
CASE
|
||||
WHEN membership_expires_at IS NULL THEN '永久'
|
||||
WHEN membership_expires_at <= NOW() THEN '已过期'
|
||||
ELSE CONCAT('剩 ', EXTRACT(DAY FROM membership_expires_at - NOW()), ' 天')
|
||||
END AS status,
|
||||
CASE
|
||||
WHEN membership_expires_at IS NULL THEN 99999
|
||||
WHEN membership_expires_at <= NOW() THEN -1
|
||||
ELSE EXTRACT(DAY FROM membership_expires_at - NOW())::int
|
||||
END AS days_left
|
||||
FROM users
|
||||
WHERE is_member = true
|
||||
ORDER BY
|
||||
CASE WHEN membership_expires_at IS NULL THEN 0 ELSE 1 END,
|
||||
membership_expires_at DESC;
|
||||
SQL
|
||||
)
|
||||
|
||||
MEMBER_LIST=$(ssh "$SSH_HOST" "psql -d $DB_NAME -AqtF'|' -c \"$SQL_LIST\"" 2>/dev/null)
|
||||
|
||||
# ---- 近期到期的活跃会员 ----
|
||||
SQL_EXPIRING=$(cat <<'SQL'
|
||||
SELECT
|
||||
id,
|
||||
COALESCE(nickname, CONCAT('用户 #', id)) AS display_name,
|
||||
membership_expires_at::date AS expires_on,
|
||||
EXTRACT(DAY FROM membership_expires_at - NOW())::int AS days_left
|
||||
FROM users
|
||||
WHERE is_member = true
|
||||
AND membership_expires_at IS NOT NULL
|
||||
AND membership_expires_at > NOW()
|
||||
AND membership_expires_at <= NOW() + INTERVAL '7 days'
|
||||
ORDER BY membership_expires_at;
|
||||
SQL
|
||||
)
|
||||
|
||||
EXPIRING_LIST=$(ssh "$SSH_HOST" "psql -d $DB_NAME -AqtF'|' -c \"$SQL_EXPIRING\"" 2>/dev/null)
|
||||
|
||||
# ---- JSON 输出 ----
|
||||
if [ "$OUTPUT_JSON" = true ]; then
|
||||
cat <<JSON
|
||||
{
|
||||
"db": "$DB_NAME",
|
||||
"time": "$(date --iso-8601=seconds)",
|
||||
"total_users": $total_users,
|
||||
"total_members": $total_members,
|
||||
"active_members": $active_members,
|
||||
"expired_members": $expired_members,
|
||||
"permanent_members": $permanent_members,
|
||||
"expiring_within_7d": $expiring_7d,
|
||||
"expiring_within_30d": $expiring_30d,
|
||||
"members": [
|
||||
JSON
|
||||
FIRST=true
|
||||
while IFS='|' read -r id openid name expires_on status days_left; do
|
||||
[ -z "$id" ] && continue
|
||||
[ "$FIRST" = true ] || echo ","
|
||||
FIRST=false
|
||||
cat <<JSON
|
||||
{ "id": $id, "name": "$name", "expires_on": "${expires_on:-永久}", "status": "$status", "days_left": ${days_left:-0} }
|
||||
JSON
|
||||
done <<< "$MEMBER_LIST"
|
||||
echo ""
|
||||
echo " ]"
|
||||
echo "}"
|
||||
exit 0
|
||||
fi
|
||||
|
||||
# ---- 表格输出 ----
|
||||
echo -e "${BOLD}📊 总览${NC}"
|
||||
echo -e " ${GREEN}总用户数: ${BOLD}$total_users${NC}"
|
||||
echo -e " ${CYAN}会员总数: ${BOLD}$total_members${NC}（占总用户 $(( total_users > 0 ? total_members * 100 / total_users : 0 ))%）"
|
||||
echo -e " ${GREEN}活跃会员: ${BOLD}$active_members${NC}（已付费且未过期）"
|
||||
echo -e " ${YELLOW}已过期会员: ${BOLD}$expired_members${NC}"
|
||||
echo -e " ${BOLD}永久会员: ${BOLD}$permanent_members${NC}"
|
||||
echo ""
|
||||
echo -e " ${YELLOW}7 天内到期: ${BOLD}$expiring_7d${NC}"
|
||||
echo -e " ${YELLOW}30 天内到期: ${BOLD}$expiring_30d${NC}"
|
||||
echo ""
|
||||
|
||||
# ---- 活跃会员列表 ----
|
||||
if [ -n "$MEMBER_LIST" ]; then
|
||||
echo -e "${BOLD}📋 会员列表${NC}"
|
||||
printf " ${CYAN}%-4s %-22s %-12s %s${NC}\n" "ID" "名称" "到期日" "状态"
|
||||
echo " ─────────────────────────────────────────────"
|
||||
while IFS='|' read -r id openid name expires_on status days_left; do
|
||||
[ -z "$id" ] && continue
|
||||
if [ "$status" = "永久" ]; then
|
||||
printf " %-4d %-22s %-12s %s\n" "$id" "${name:0:22}" "永久" "🔒 永久"
|
||||
elif [ "$status" = "已过期" ]; then
|
||||
printf " %-4d %-22s %-12s %s\n" "$id" "${name:0:22}" "$expires_on" "❌ $status"
|
||||
elif [ "$days_left" -le 7 ]; then
|
||||
printf " %-4d %-22s %-12s %s\n" "$id" "${name:0:22}" "$expires_on" "⚠️ $status"
|
||||
else
|
||||
printf " %-4d %-22s %-12s %s\n" "$id" "${name:0:22}" "$expires_on" "✅ $status"
|
||||
fi
|
||||
done <<< "$MEMBER_LIST"
|
||||
echo ""
|
||||
fi
|
||||
|
||||
# ---- 近期到期提醒 ----
|
||||
if [ -n "$EXPIRING_LIST" ]; then
|
||||
echo -e "${BOLD}⏰ 近期到期（7 天内）${NC}"
|
||||
while IFS='|' read -r id name expires_on days_left; do
|
||||
[ -z "$id" ] && continue
|
||||
echo -e " ${YELLOW}⚠️ 用户 #$id（${name}）将于 $expires_on 到期，剩余 ${days_left} 天${NC}"
|
||||
done <<< "$EXPIRING_LIST"
|
||||
echo ""
|
||||
fi
|
||||
|
||||
echo -e "${CYAN}══════════════════════════════════════${NC}"
|
||||
Reference in New Issue
Block a user