142 lines
5.1 KiB
Bash
Executable File
142 lines
5.1 KiB
Bash
Executable File
#!/bin/bash
|
||
# ===========================================
|
||
# 发送通知 - 管理员工具
|
||
# 通过 API 创建系统广播或定向通知
|
||
# ===========================================
|
||
# 用法:
|
||
# ./scripts/send-notification.sh --all --type system_maintenance --title "维护通知" --content "内容"
|
||
# ./scripts/send-notification.sh --user 1 --type payment_success --title "开通成功" --link /upgrade
|
||
#
|
||
# 选项:
|
||
# --all 发送给所有用户（scope=all）
|
||
# --user <id> 发送给指定用户（scope=user）
|
||
# --type <type> 通知类型（必填）
|
||
# --title <text> 通知标题（必填）
|
||
# --content <text> 通知内容（可选）
|
||
# --priority <lvl> 优先级: high/normal/low（默认 normal）
|
||
# --link <url> 点击跳转路径（可选）
|
||
# --dev 目标: milkydata_dev（默认 milkydata）
|
||
# --dry-run 预览不执行
|
||
# ===========================================
|
||
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||
PROJECT_ROOT="$(dirname "$SCRIPT_DIR")"
|
||
ENV_FILE="$PROJECT_ROOT/.env"
|
||
|
||
# ---------- 默认值 ----------
|
||
SCOPE=""
|
||
USER_ID=""
|
||
NOTIF_TYPE=""
|
||
TITLE=""
|
||
CONTENT=""
|
||
PRIORITY="normal"
|
||
LINK=""
|
||
DB_NAME="milkydata"
|
||
DRY_RUN=false
|
||
|
||
# ---------- 获取 SSH 主机 ----------
|
||
get_ssh_host() {
|
||
grep '^SSH_SERVER=' "$ENV_FILE" 2>/dev/null | head -1 | cut -d'=' -f2- | tr -d '"' | tr -d "'" | xargs
|
||
}
|
||
|
||
# ---------- 使用说明 ----------
|
||
usage() {
|
||
echo "用法: $0 (--all | --user <id>) --type <type> --title <title> [options]"
|
||
echo ""
|
||
echo "必填:"
|
||
echo " --all 发送给所有用户"
|
||
echo " --user <id> 发送给指定用户"
|
||
echo " --type <type> 通知类型: system_maintenance / version_update /"
|
||
echo " payment_success / member_expiry_soon / member_expired"
|
||
echo " --title <text> 通知标题"
|
||
echo ""
|
||
echo "可选:"
|
||
echo " --content <text> 通知内容"
|
||
echo " --priority <lvl> high / normal / low（默认 normal）"
|
||
echo " --link <path> 点击跳转路径（如 /pkg-extra/upgrade/upgrade）"
|
||
echo " --dev 操作开发库 milkydata_dev"
|
||
echo " --dry-run 预览，不执行"
|
||
echo ""
|
||
echo "示例:"
|
||
echo " $0 --all --type system_maintenance --title '维护通知' --content '今晚维护' --priority high"
|
||
echo " $0 --user 1 --type payment_success --title '开通成功' --link /upgrade --dev"
|
||
exit 0
|
||
}
|
||
|
||
# ---------- 参数解析 ----------
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
--all) SCOPE="all" ;;
|
||
--user) SCOPE="user"; USER_ID="$2"; shift ;;
|
||
--type) NOTIF_TYPE="$2"; shift ;;
|
||
--title) TITLE="$2"; shift ;;
|
||
--content) CONTENT="$2"; shift ;;
|
||
--priority) PRIORITY="$2"; shift ;;
|
||
--link) LINK="$2"; shift ;;
|
||
--dev) DB_NAME="milkydata_dev" ;;
|
||
--dry-run) DRY_RUN=true ;;
|
||
--help|-h) usage ;;
|
||
*) echo "未知选项: $1"; usage ;;
|
||
esac
|
||
shift
|
||
done
|
||
|
||
# ---------- 必填校验 ----------
|
||
[ -z "$SCOPE" ] && { echo "错误: 必须指定 --all 或 --user"; usage; }
|
||
[ "$SCOPE" = "user" ] && [ -z "$USER_ID" ] && { echo "错误: --user 必须指定用户 ID"; usage; }
|
||
[ -z "$NOTIF_TYPE" ] && { echo "错误: --type 必填"; usage; }
|
||
[ -z "$TITLE" ] && { echo "错误: --title 必填"; usage; }
|
||
|
||
# ---------- 连接信息 ----------
|
||
SSH_HOST=$(get_ssh_host)
|
||
[ -z "$SSH_HOST" ] && { echo "错误: 未配置 SSH_SERVER"; exit 1; }
|
||
|
||
DB_CONTAINER="1Panel-postgresql-FtMo"
|
||
DB_USER="milkydata"
|
||
|
||
# ---------- 构建 SQL ----------
|
||
USER_VAL="NULL"
|
||
[ "$SCOPE" = "user" ] && USER_VAL="$USER_ID"
|
||
|
||
CONTENT_VAL="NULL"
|
||
[ -n "$CONTENT" ] && CONTENT_VAL="'${CONTENT//\'/\'\'}'"
|
||
|
||
LINK_VAL="NULL"
|
||
[ -n "$LINK" ] && LINK_VAL="'${LINK//\'/\'\'}'"
|
||
|
||
SQL="INSERT INTO notifications (scope, user_id, type, title, content, priority, link)
|
||
VALUES ('${SCOPE}', ${USER_VAL}, '${NOTIF_TYPE}', '${TITLE//\'/\'\'}', ${CONTENT_VAL}, '${PRIORITY}', ${LINK_VAL});"
|
||
|
||
# ---------- 确认信息 ----------
|
||
echo ""
|
||
echo "═══════════════════════════════════════"
|
||
echo " 操作: 发送通知"
|
||
echo " 范围: ${SCOPE}"
|
||
echo " 类型: ${NOTIF_TYPE}"
|
||
echo " 标题: ${TITLE}"
|
||
echo " 数据库: ${DB_NAME}（${SSH_HOST}）"
|
||
echo "═══════════════════════════════════════"
|
||
echo ""
|
||
echo "将要执行："
|
||
echo " ${SQL}"
|
||
echo ""
|
||
|
||
[ "$DRY_RUN" = true ] && { echo "预览模式，未执行。"; exit 0; }
|
||
|
||
read -p "确认发送? [y/N] " -n 1 -r; echo
|
||
[[ ! $REPLY =~ ^[Yy]$ ]] && { echo "已取消"; exit 0; }
|
||
|
||
echo ""
|
||
echo "执行中..."
|
||
ssh "$SSH_HOST" "docker exec ${DB_CONTAINER} psql -U ${DB_USER} -d ${DB_NAME} -c \"${SQL}\"" && echo "✅ 通知已发送" || echo "❌ 发送失败"
|
||
|
||
# 验证
|
||
echo ""
|
||
echo "最近通知："
|
||
VERIFY_SQL="SELECT id, scope, user_id, type, title, priority, created_at::date
|
||
FROM notifications ORDER BY id DESC LIMIT 5;"
|
||
ssh "$SSH_HOST" "docker exec ${DB_CONTAINER} psql -U ${DB_USER} -d ${DB_NAME} -c \"${VERIFY_SQL}\""
|
||
echo ""
|