Refactor backup config and Telegram notifications

2026-05-08 06:37:02 +08:00 · 2026-03-16 17:30:55 +08:00
parent 51c8c37c69
commit d4b69f10f9
8 changed files with 293 additions and 254 deletions
@@ -0,0 +1,138 @@
+#!/usr/bin/env bash
+
+tg_code_block() {
+  local text="${1-}"
+  text="$(printf '%s' "$text" | sed -e 's/\\/\\\\/g' -e 's/`/\\`/g')"
+  printf '```text\n%s\n```' "$text"
+}
+
+tg_send_message() {
+  local message="${1-}"
+  local response api
+
+  if [[ -z "${TG_BOT_TOKEN:-}" || -z "${TG_USER_ID:-}" ]]; then
+    log "INFO" "未配置Telegram通知，跳过发送"
+    return 0
+  fi
+
+  api="https://api.telegram.org/bot${TG_BOT_TOKEN}/sendMessage"
+  response="$(curl -sS -X POST "$api" \
+    --data-urlencode "chat_id=${TG_USER_ID}" \
+    --data-urlencode "parse_mode=MarkdownV2" \
+    --data-urlencode "disable_web_page_preview=true" \
+    --data-urlencode "text=${message}" 2>&1 || true)"
+
+  if echo "$response" | grep -q '"ok":true'; then
+    log "INFO" "Telegram通知发送成功"
+  else
+    log "WARN" "Telegram通知发送失败: $response"
+  fi
+}
+
+tg_read_user_excludes() {
+  local file="${1-}"
+  local out="" rule
+
+  [[ -f "$file" ]] || {
+    printf '无'
+    return 0
+  }
+
+  while IFS= read -r rule; do
+    [[ -n "$rule" ]] || continue
+    [[ "$rule" =~ ^[[:space:]]*# ]] && continue
+    out+="${rule}"$'\n'
+  done < "$file"
+
+  if [[ -z "$out" ]]; then
+    printf '无'
+  else
+    printf '%s' "${out%$'\n'}"
+  fi
+}
+
+tg_backup_success() {
+  local label="$1" current_date="$2" backup_dirs="$3" user_excludes="$4" backup_size="$5" archive_name="$6" location="$7" pack_time="$8" upload_time="$9" speed="${10}"
+  local archive_info message
+  archive_info=$(
+    cat <<EOF
+文件大小: ${backup_size}
+归档名称: ${archive_name}
+存储位置: ${location}
+打包用时: ${pack_time}
+上传用时: ${upload_time}
+平均速率: ${speed}
+EOF
+  )
+  message="*备份成功*"$'\n\n'
+  message+="*主机*"$'\n'"$(tg_code_block "$label")"$'\n'
+  message+="*时间*"$'\n'"$(tg_code_block "$current_date")"$'\n'
+  message+="*备份内容*"$'\n'"$(tg_code_block "$backup_dirs")"$'\n'
+  message+="*用户排除*"$'\n'"$(tg_code_block "$user_excludes")"$'\n'
+  message+="*归档信息*"$'\n'"$(tg_code_block "$archive_info")"
+  tg_send_message "$message"
+}
+
+tg_backup_dryrun() {
+  local label="$1" current_date="$2" backup_dirs="$3" user_excludes="$4" archive_name="$5" backup_size="$6" pack_time="$7"
+  local archive_info message
+  archive_info=$(
+    cat <<EOF
+归档名称: ${archive_name}
+文件大小: ${backup_size}
+打包用时: ${pack_time}
+EOF
+  )
+  message="*备份自检*"$'\n\n'
+  message+="*主机*"$'\n'"$(tg_code_block "$label")"$'\n'
+  message+="*时间*"$'\n'"$(tg_code_block "$current_date")"$'\n'
+  message+="*备份内容*"$'\n'"$(tg_code_block "$backup_dirs")"$'\n'
+  message+="*用户排除*"$'\n'"$(tg_code_block "$user_excludes")"$'\n'
+  message+="*归档信息*"$'\n'"$(tg_code_block "$archive_info")"
+  tg_send_message "$message"
+}
+
+tg_backup_error() {
+  local label="$1" current_date="$2" reason="$3" error_log="$4"
+  local message
+  message="*备份失败*"$'\n\n'
+  message+="*主机*"$'\n'"$(tg_code_block "$label")"$'\n'
+  message+="*时间*"$'\n'"$(tg_code_block "$current_date")"$'\n'
+  message+="*原因*"$'\n'"$(tg_code_block "$reason")"
+  if [[ -n "$error_log" ]]; then
+    message+=$'\n'"*错误日志*"$'\n'"$(tg_code_block "$error_log")"
+  fi
+  tg_send_message "$message"
+}
+
+tg_clean_success() {
+  local label="$1" current_date="$2" prefix="$3" retain_days="$4" keep_count="$5" del_ok="$6" del_fail="$7"
+  local summary message
+  summary=$(
+    cat <<EOF
+前缀: ${prefix}
+保留天数: ${retain_days}
+保留对象数: ${keep_count}
+删除成功: ${del_ok}
+删除失败: ${del_fail}
+EOF
+  )
+  message="*清理完成*"$'\n\n'
+  message+="*主机*"$'\n'"$(tg_code_block "$label")"$'\n'
+  message+="*时间*"$'\n'"$(tg_code_block "$current_date")"$'\n'
+  message+="*结果*"$'\n'"$(tg_code_block "$summary")"
+  tg_send_message "$message"
+}
+
+tg_clean_error() {
+  local label="$1" current_date="$2" reason="$3" error_log="$4"
+  local message
+  message="*清理失败*"$'\n\n'
+  message+="*主机*"$'\n'"$(tg_code_block "$label")"$'\n'
+  message+="*时间*"$'\n'"$(tg_code_block "$current_date")"$'\n'
+  message+="*原因*"$'\n'"$(tg_code_block "$reason")"
+  if [[ -n "$error_log" ]]; then
+    message+=$'\n'"*错误日志*"$'\n'"$(tg_code_block "$error_log")"
+  fi
+  tg_send_message "$message"
+}