#!/bin/bash

# 检查必要的WebDAV环境变量
if [ -z "$WEBDAV_URL" ] || [ -z "$WEBDAV_USERNAME" ] || [ -z "$WEBDAV_PASSWORD" ]; then
    echo "缺少必要的WebDAV环境变量：WEBDAV_URL, WEBDAV_USERNAME 或 WEBDAV_PASSWORD"
    exit 1
fi

# 使用独立的工作目录
WORK_DIR="/tmp/mcp_sync_work"
mkdir -p "$WORK_DIR"

# 从 WebDAV 恢复配置的函数
restore_from_webdav() {
    echo "正在从 WebDAV 恢复配置文件..."
    
    # 创建临时恢复目录
    RESTORE_DIR="$WORK_DIR/webdav_restore"
    mkdir -p "$RESTORE_DIR"
    
    # 获取WebDAV上所有备份文件列表
    echo "获取 WebDAV 备份文件列表..."
    
    # 尝试使用PROPFIND方法获取目录列表
    RAW_RESPONSE=$(curl -s -X PROPFIND \
        --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" \
        -H "Depth: 1" \
        -H "Content-Type: text/xml" \
        -d '<?xml version="1.0" encoding="utf-8"?><propfind xmlns="DAV:"><prop><displayname/></prop></propfind>' \
        "$WEBDAV_URL/")
    
    # 如果PROPFIND失败，尝试简单的GET请求
    if [ -z "$RAW_RESPONSE" ] || echo "$RAW_RESPONSE" | grep -q "Method Not Allowed"; then
        RAW_RESPONSE=$(curl -s --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/")
    fi
    
    # 多种匹配方式
    BACKUP_LIST1=$(echo "$RAW_RESPONSE" | grep -oE '<D:displayname[^>]*>[^<]*mcp_backup\.tar\.gz[^<]*</D:displayname>' | sed 's/<[^>]*>//g')
    BACKUP_LIST2=$(echo "$RAW_RESPONSE" | grep -oE 'href="[^"]*mcp_backup\.tar\.gz[^"]*"' | sed 's/.*href="[^"]*\///g' | sed 's/".*//g')
    BACKUP_LIST3=$(echo "$RAW_RESPONSE" | grep -oE 'mcp_backup\.tar\.gz')
    
    # 合并所有结果
    BACKUP_LIST=$(printf "%s\n%s\n%s\n" "$BACKUP_LIST1" "$BACKUP_LIST2" "$BACKUP_LIST3" | grep -v '^$' | sort -u)
    
    if [ -z "$BACKUP_LIST" ]; then
        echo "未找到任何备份文件"
        rm -rf "$RESTORE_DIR"
        return 1
    fi
    
    # 使用固定文件名下载备份
    BACKUP_FILE="mcp_backup.tar.gz"
    echo "尝试下载最新备份: $BACKUP_FILE"
    
    if curl -f --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$BACKUP_FILE" -o "$RESTORE_DIR/$BACKUP_FILE" 2>/dev/null; then
        echo "下载备份成功: $BACKUP_FILE"
        
        # 解压缩备份文件
        cd "$RESTORE_DIR"
        tar -xzf "$BACKUP_FILE" && {
            echo "解压缩备份文件成功"
            
            # 恢复配置文件
            if [ -d "config" ] && [ "$(ls -A config 2>/dev/null)" ]; then
                echo "恢复 config 目录..."
                mkdir -p /mcp-proxy-server/config
                cp -r config/* /mcp-proxy-server/config/
                echo "config 目录恢复成功"
            else
                echo "备份中没有 config 目录或目录为空"
            fi
            
            if [ -d "tools" ] && [ "$(ls -A tools 2>/dev/null)" ]; then
                echo "恢复 tools 目录..."
                mkdir -p /tools
                cp -r tools/* /tools/
                echo "tools 目录恢复成功"
            else
                echo "备份中没有 tools 目录或目录为空"
            fi
            
            echo "从 WebDAV 恢复配置文件完成"
        } || {
            echo "解压缩备份文件失败"
        }
        cd - > /dev/null
    else
        echo "下载备份失败: $BACKUP_FILE"
    fi
    
    # 清理临时目录
    rm -rf "$RESTORE_DIR"
    return 0
}

# 处理恢复参数
if [ "$1" = "--restore-only" ] && [ "$2" = "webdav" ]; then
    restore_from_webdav
    exit 0
fi

# 启动时自动从WebDAV恢复
restore_from_webdav || echo "WebDAV恢复失败，将在同步时重新备份"

# 如果是仅恢复模式，则退出
if [ "$1" = "--restore-only" ]; then
    exit 0
fi

# 定义同步函数（只使用WebDAV）
sync_data() {
    # 使用固定备份文件名
    BACKUP_FILE="mcp_backup.tar.gz"
    
    while true; do
        echo "开始定期备份..."
        
        # 创建临时备份压缩包
        TEMP_DIR="$WORK_DIR/temp_backup"
        rm -rf "$TEMP_DIR"
        mkdir -p "$TEMP_DIR"
        
        # 复制要备份的目录到临时目录
        BACKUP_CREATED=0
        
        # 备份config目录（排除工作目录）
        if [ -d "/mcp-proxy-server/config" ] && [ "$(ls -A /mcp-proxy-server/config 2>/dev/null)" ]; then
            mkdir -p "$TEMP_DIR/config"
            for item in /mcp-proxy-server/config/*; do
                if [ -e "$item" ]; then
                    BASENAME=$(basename "$item")
                    case "$BASENAME" in
                        github_data|temp_backup|webdav_restore)
                            ;;
                        *)
                            if [ -d "$item" ]; then
                                cp -r "$item" "$TEMP_DIR/config/" 2>/dev/null && BACKUP_CREATED=1
                            else
                                cp "$item" "$TEMP_DIR/config/" 2>/dev/null && BACKUP_CREATED=1
                            fi
                            ;;
                    esac
                fi
            done
        fi
        
        # 备份tools目录
        if [ -d "/tools" ] && [ "$(ls -A /tools 2>/dev/null)" ]; then
            mkdir -p "$TEMP_DIR/tools"
            for item in /tools/*; do
                if [ -e "$item" ]; then
                    BASENAME=$(basename "$item")
                    if [ -d "$item" ]; then
                        mkdir -p "$TEMP_DIR/tools/$BASENAME"
                        if [ "$(ls -A "$item" 2>/dev/null)" ]; then
                            cp -r "$item"/* "$TEMP_DIR/tools/$BASENAME/" 2>/dev/null && BACKUP_CREATED=1
                            cp -r "$item"/.[!.]* "$TEMP_DIR/tools/$BASENAME/" 2>/dev/null || true
                        else
                            touch "$TEMP_DIR/tools/$BASENAME/.gitkeep"
                            BACKUP_CREATED=1
                        fi
                    else
                        cp "$item" "$TEMP_DIR/tools/" 2>/dev/null && BACKUP_CREATED=1
                    fi
                fi
            done
        fi
        
        # 创建压缩包并上传
        if [ "$BACKUP_CREATED" = "1" ] && [ "$(ls -A $TEMP_DIR)" ]; then
            (cd "$TEMP_DIR" && tar -czf "../$BACKUP_FILE" .)
            
            if curl -T "$WORK_DIR/$BACKUP_FILE" --user "$WEBDAV_USERNAME:$WEBDAV_PASSWORD" "$WEBDAV_URL/$BACKUP_FILE" >/dev/null 2>&1; then
                echo "WebDAV备份上传成功: $BACKUP_FILE"
            else
                echo "WebDAV备份上传失败"
            fi
            rm -f "$WORK_DIR/$BACKUP_FILE"
        else
            echo "无文件可备份"
        fi
        
        rm -rf "$TEMP_DIR"
        
        # 等待下一次备份
        SYNC_INTERVAL=${SYNC_INTERVAL:-7200}  # 默认间隔时间为 7200 秒
        echo "等待 ${SYNC_INTERVAL} 秒后进行下一次备份..."
        sleep $SYNC_INTERVAL
    done
}

# 后台启动同步进程
sync_data &

echo "同步进程已在后台启动 (PID: $!)"
echo "现在启动 MCP Proxy 主程序..."

# 启动原始的 MCP Proxy 程序
echo "准备启动 MCP Proxy，端口: ${PORT:-7860}，主机: ${HOST:-0.0.0.0}"

if [ -f build/sse.js ]; then
    echo "启动: node build/sse.js"
    PORT=${PORT:-7860} HOST=${HOST:-0.0.0.0} exec node build/sse.js
elif [ -f /app/build/sse.js ]; then
    echo "启动: node /app/build/sse.js"
    PORT=${PORT:-7860} HOST=${HOST:-0.0.0.0} exec node /app/build/sse.js
elif [ -f /mcp-proxy-server/build/sse.js ]; then
    echo "启动: node /mcp-proxy-server/build/sse.js"  
    PORT=${PORT:-7860} HOST=${HOST:-0.0.0.0} exec node /mcp-proxy-server/build/sse.js
else
    echo "错误: 找不到 build/sse.js 文件"
    echo "当前目录: $(pwd)"
    echo "查找可能的 Node.js 文件:"
    find / -name "sse.js" -type f 2>/dev/null | head -5
    
    # 如果找不到，保持容器运行以便调试
    echo "保持容器运行以便调试..."
    tail -f /dev/null
fi