webdav_backup.sh

#!/bin/bash
set -e

# 加载配置
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
if [ -f "$SCRIPT_DIR/.env" ]; then
    source "$SCRIPT_DIR/.env"
else
    echo "❌ 错误：找不到 .env 文件"
    exit 1
fi

# 检查必填字段
if [[ -z "$TARGET_DIR" || -z "$WEBDAV_URL" || -z "$WEBDAV_USER" || -z "$WEBDAV_PASS" ]]; then
    echo "❌ 错误：.env 中缺少必要字段"
    exit 1
fi

# 默认设置
BACKUP_NAME="${BACKUP_NAME:-Backup}"
# 添加时间格式配置，默认按天
DATE_FORMAT="${DATE_FORMAT:-%Y-%m-%d}"
DATE=$(date +"$DATE_FORMAT")
BASE_NAME="${BACKUP_NAME,,}-$DATE"
EXT=""
ZIP_PATH=""
mkdir -p "$BACKUP_DIR"

# 检查/安装指令
ensure_cmd() {
    local cmd=$1
    if ! command -v "$cmd" >/dev/null 2>&1; then
        echo "⚠️ 未找到 $cmd，尝试安装..."
        if command -v apt >/dev/null 2>&1; then
            if [ "$cmd" = "7z" ]; then
                sudo apt update && sudo apt install -y p7zip-full
            else
                sudo apt update && sudo apt install -y "$cmd"
            fi
        elif command -v yum >/dev/null 2>&1; then
            if [ "$cmd" = "7z" ]; then
                sudo yum install -y p7zip p7zip-plugins
            else
                sudo yum install -y "$cmd"
            fi
        elif command -v apk >/dev/null 2>&1; then
            if [ "$cmd" = "7z" ]; then
                sudo apk add p7zip
            else
                sudo apk add "$cmd"
            fi
        elif command -v brew >/dev/null 2>&1; then
            if [ "$cmd" = "7z" ]; then
                brew install p7zip
            else
                brew install "$cmd"
            fi
        else
            echo "❌ 无法自动安装 $cmd，请手动安装"
            exit 1
        fi
    fi
}

# 通知函数
send_notification() {
    local message="$1"
    local success=false

    # Telegram 通知
    if [[ -n "$TELEGRAM_BOT_TOKEN" && -n "$TELEGRAM_CHAT_ID" ]]; then
        local attempt=0
        while [ $attempt -lt ${MAX_RETRIES:-3} ]; do
            curl_opts=()
            [[ -n "$PROXY_URL" ]] && curl_opts+=(-x "$PROXY_URL")
            
            # 转义Markdown特殊字符，尤其是下划线
            local escaped_message
            escaped_message=$(echo "$message" | sed 's/_/\\_/g' | sed 's/\*/\\*/g' | sed 's/`/\\`/g' | sed 's/\[/\\[/g' | sed 's/\]/\\]/g')
            
            if curl -s "${curl_opts[@]}" -X POST "https://api.telegram.org/bot$TELEGRAM_BOT_TOKEN/sendMessage" \
                -d chat_id="$TELEGRAM_CHAT_ID" -d text="$escaped_message" -d parse_mode="Markdown"; then
                success=true
                break
            fi
            echo "Telegram 发送失败，重试中..."
            sleep ${RETRY_DELAY:-10}
            attempt=$((attempt + 1))
        done
    fi

    # Discord Webhook
    if [[ -n "$DISCORD_WEBHOOK_URL" ]]; then
        if curl -s -X POST "$DISCORD_WEBHOOK_URL" \
            -H "Content-Type: application/json" \
            -d "{\"content\":\"$message\"}"; then
            success=true
        else
            echo "❌ Discord 通知失败"
        fi
    fi

    # Slack Webhook
    if [[ -n "$SLACK_WEBHOOK_URL" ]]; then
        if curl -s -X POST "$SLACK_WEBHOOK_URL" \
            -H "Content-Type: application/json" \
            -d "{\"text\":\"$message\"}"; then
            success=true
        else
            echo "❌ Slack 通知失败"
        fi
    fi

    # 飞书 Webhook
    if [[ -n "$FEISHU_WEBHOOK_URL" ]]; then
        if curl -s -X POST "$FEISHU_WEBHOOK_URL" \
            -H "Content-Type: application/json" \
            -d "{\"msg_type\":\"text\",\"content\":{\"text\":\"$message\"}}"; then
            success=true
        else
            echo "❌ 飞书通知失败"
        fi
    fi

    # 钉钉 Webhook
    if [[ -n "$DINGTALK_WEBHOOK_URL" ]]; then
        if curl -s -X POST "$DINGTALK_WEBHOOK_URL" \
            -H "Content-Type: application/json" \
            -d "{\"msgtype\":\"text\",\"text\":{\"content\":\"$message\"}}"; then
            success=true
        else
            echo "❌ 钉钉通知失败"
        fi
    fi

    # 企业微信 Webhook
    if [[ -n "$WECOM_WEBHOOK_URL" ]]; then
        if curl -s -X POST "$WECOM_WEBHOOK_URL" \
            -H "Content-Type: application/json" \
            -d "{\"msgtype\":\"text\",\"text\":{\"content\":\"$message\"}}"; then
            success=true
        else
            echo "❌ 企业微信通知失败"
        fi
    fi

    # OneBot 协议支持
    if [[ -n "$ONEBOT_API_URL" ]]; then
        # 支持群组消息
        if [[ -n "$ONEBOT_GROUP_ID" ]]; then
            if curl -s -X POST "$ONEBOT_API_URL/send_group_msg" \
                -H "Content-Type: application/json" \
                -d "{\"group_id\":$ONEBOT_GROUP_ID,\"message\":\"$message\"}"; then
                success=true
            else
                echo "❌ OneBot群组消息发送失败"
            fi
        fi
        # 支持私聊消息
        if [[ -n "$ONEBOT_USER_ID" ]]; then
            if curl -s -X POST "$ONEBOT_API_URL/send_private_msg" \
                -H "Content-Type: application/json" \
                -d "{\"user_id\":$ONEBOT_USER_ID,\"message\":\"$message\"}"; then
                success=true
            else
                echo "❌ OneBot私聊消息发送失败"
            fi
        fi
    fi

    # 通用 Webhook
    if [[ -n "$WEBHOOK_URL" ]]; then
        # 检查jq
        ensure_cmd jq
        # 设置默认值
        local method=${WEBHOOK_METHOD:-POST}
        local headers=${WEBHOOK_HEADERS:-'{"Content-Type":"application/json"}'}
        local body_template=${WEBHOOK_BODY_TEMPLATE:-'{"text":"{{message}}"}'}
        
        # 处理请求头
        local header_args=()
        if [[ -n "$headers" ]]; then
            while IFS=':' read -r key value; do
                [[ -n "$key" ]] && header_args+=(-H "$key:$value")
            done < <(echo "$headers" | jq -r 'to_entries | .[] | "\(.key):\(.value)"')
        fi

        # 处理消息模板
        local body
        body=$(echo "$body_template" | sed "s/{{message}}/$message/g" | sed "s/{{title}}/$BACKUP_NAME/g")

        # 发送请求
        if curl -s -X "$method" "${header_args[@]}" "$WEBHOOK_URL" -d "$body"; then
            success=true
        else
            echo "❌ Webhook 通知失败"
            echo "请求详情："
            echo "URL: $WEBHOOK_URL"
            echo "方法: $method"
            echo "请求头: $headers"
            echo "请求体: $body"
        fi
    fi

    if ! $success; then
        echo "ℹ️ 所有通知方式均未配置或发送失败"
    fi
}

# 开始压缩
echo "📦 开始压缩..."
# 检查TARGET_DIR是否存在
if [ ! -d "$TARGET_DIR" ]; then
    echo "❌ 目录不存在：$TARGET_DIR"
    send_notification "❌ $BACKUP_NAME 备份失败：目录不存在"
    exit 1
fi

# 获取TARGET_DIR的父目录和目录名
PARENT_DIR=$(dirname "$TARGET_DIR")
DIR_NAME=$(basename "$TARGET_DIR")

# 进入TARGET_DIR的上级目录进行压缩
cd "$PARENT_DIR" || { echo "❌ 无法进入父目录：$PARENT_DIR"; send_notification "❌ $BACKUP_NAME 备份失败：无法进入父目录"; exit 1; }

# 去除密码中可能的首尾空格
ZIP_PASSWORD=$(echo "$ZIP_PASSWORD" | xargs)

case "$COMPRESS_METHOD" in
    zip)
        ensure_cmd zip
        EXT=".zip"
        ZIP_PATH="$BACKUP_DIR/$BASE_NAME$EXT"
        if [[ -n "$ZIP_PASSWORD" ]]; then
            # 使用单引号避免特殊字符解析问题
            zip -rq "$ZIP_PATH" "$DIR_NAME" -P '$ZIP_PASSWORD'
            # 测试加密是否成功
            if ! unzip -t -P "$ZIP_PASSWORD" "$ZIP_PATH" &>/dev/null; then
                echo "⚠️ 使用环境变量密码无法验证，尝试直接使用密码文本..."
                zip -rq "$ZIP_PATH.new" "$DIR_NAME" -P "$ZIP_PASSWORD"
                mv "$ZIP_PATH.new" "$ZIP_PATH"
            fi
        else
            zip -rq "$ZIP_PATH" "$DIR_NAME"
        fi
        ;;
    7z)
        ensure_cmd 7z
        EXT=".7z"
        ZIP_PATH="$BACKUP_DIR/$BASE_NAME$EXT"
        if [[ -n "$ZIP_PASSWORD" ]]; then
            # 避免嵌套引号问题
            7z a -t7z "$ZIP_PATH" "$DIR_NAME" -p"$ZIP_PASSWORD" -mhe=on
            # 验证加密
            if ! 7z t "$ZIP_PATH" -p"$ZIP_PASSWORD" &>/dev/null; then
                echo "⚠️ 7z密码验证失败，请检查密码中是否有特殊字符"
            fi
        else
            7z a -t7z "$ZIP_PATH" "$DIR_NAME"
        fi
        ;;
    tar.gz)
        ensure_cmd tar
        EXT=".tar.gz"
        ZIP_PATH="$BACKUP_DIR/$BASE_NAME$EXT"
        tar -czf "$ZIP_PATH" "$DIR_NAME"
        # tar.gz不支持加密，如果需要加密请使用zip或7z
        [[ -n "$ZIP_PASSWORD" ]] && echo "⚠️ tar.gz格式不支持密码加密，已创建未加密归档"
        ;;
    *)
        echo "❌ 不支持的压缩方式：$COMPRESS_METHOD"
        send_notification "❌ $BACKUP_NAME 备份失败：不支持的压缩方式 $COMPRESS_METHOD"
        exit 1
        ;;
esac

# 上传 WebDAV
echo "📤 上传中..."
FILENAME=$(basename "$ZIP_PATH")
attempt=0
while [ $attempt -lt ${MAX_RETRIES:-3} ]; do
    curl -u "$WEBDAV_USER:$WEBDAV_PASS" -T "$ZIP_PATH" "$WEBDAV_URL$FILENAME" && break
    echo "⚠️ 上传失败，${RETRY_DELAY:-10}s 后重试..."
    sleep ${RETRY_DELAY:-10}
    attempt=$((attempt + 1))
done

if [ $attempt -eq ${MAX_RETRIES:-3} ]; then
    send_notification "❌ $BACKUP_NAME 备份失败：上传 WebDAV 失败"
    exit 1
else
    send_notification "✅ $BACKUP_NAME 备份成功：$FILENAME 已上传"
fi

# 清理旧文件
echo "🧹 清理 7 天前的旧备份..."
find "$BACKUP_DIR" -name "${BACKUP_NAME,,}-*" -type f -mtime +7 -exec rm -f {} \;

echo "✅ 备份完成：$ZIP_PATH"