202 lines
7.6 KiB
Bash
Executable File
202 lines
7.6 KiB
Bash
Executable File
#!/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}"
|