#!/bin/bash
# WordPress和MySQL数据库从MEGA云存储还原脚本
# 此脚本会还原最新的备份文件

# 设置变量
RESTORE_DIR="/tmp/wp_restore"
MYSQL_ROOT_PASSWORD="test123456789"
REMOTE_NAME="mega"
REMOTE_PATH="WordPress_Backups"
WORDPRESS_DIR="$HOME/wordpress"
LOG_FILE="/tmp/wp_restore_$(date +%Y%m%d).log"

# 启用日志记录
exec > >(tee -a "$LOG_FILE") 2>&1
echo "===== 开始WordPress备份还原 $(date) ====="

# 检测环境和容器
echo "检测环境..."

DOCKER_INSTALLED=false
WORDPRESS_CONTAINER_EXISTS=false
MYSQL_CONTAINER_EXISTS=false

# 检查Docker是否安装
if command -v docker &> /dev/null; then
    DOCKER_INSTALLED=true
    echo "Docker已安装"
    
    # 检查容器
    if docker container inspect wordpress &>/dev/null; then
        WORDPRESS_CONTAINER_EXISTS=true
        echo "WordPress容器存在"
    else
        echo "WordPress容器不存在"
    fi
    
    if docker container inspect wordpressdb &>/dev/null; then
        MYSQL_CONTAINER_EXISTS=true
        echo "MySQL容器存在"
    else
        echo "MySQL容器不存在"
    fi
else
    echo "Docker未安装，"
    exit 1
fi

# 创建临时还原目录
mkdir -p $RESTORE_DIR
echo "创建临时还原目录: $RESTORE_DIR"

# 列出备份文件并找到最新的
echo "查找最新的WordPress文件备份..."
LATEST_WP_BACKUP=$(rclone lsl "$REMOTE_NAME:$REMOTE_PATH" | grep "wordpress_files_" | sort -k2,3 -r | head -n 1 | awk '{print $NF}')
if [ -z "$LATEST_WP_BACKUP" ]; then
    echo "错误: 未找到WordPress文件备份"
    exit 1
fi
echo "找到最新的WordPress文件备份: $LATEST_WP_BACKUP"

echo "查找最新的MySQL数据库备份..."
LATEST_DB_BACKUP=$(rclone lsl "$REMOTE_NAME:$REMOTE_PATH" | grep "wordpress_db_" | sort -k2,3 -r | head -n 1 | awk '{print $NF}')
echo " 查找命令为 rclone lsl "$REMOTE_NAME:$REMOTE_PATH" | grep "wordpress_db_" | sort -k2,3 -r | head -n 1 | awk '{print $NF}'' "
if [ -z "$LATEST_DB_BACKUP" ]; then
    echo "错误: 未找到MySQL数据库备份"
    exit 1
fi
echo "找到最新的MySQL数据库备份: $LATEST_DB_BACKUP"

# 下载最新的备份文件
echo "下载WordPress文件备份..."
rclone copy "$REMOTE_NAME:$REMOTE_PATH/$LATEST_WP_BACKUP" "$RESTORE_DIR"
if [ $? -ne 0 ]; then
    echo "错误: 下载WordPress文件备份失败"
    exit 1
fi
echo "WordPress文件备份下载完成"

echo "下载MySQL数据库备份..."
rclone copy "$REMOTE_NAME:$REMOTE_PATH/$LATEST_DB_BACKUP" "$RESTORE_DIR"
if [ $? -ne 0 ]; then
    echo "错误: 下载MySQL数据库备份失败"
    exit 1
fi
echo "MySQL数据库备份下载完成"

# 解压WordPress文件备份
echo "解压WordPress文件备份..."
mkdir -p "$WORDPRESS_DIR"
tar -xzf "$RESTORE_DIR/$LATEST_WP_BACKUP" -C "$WORDPRESS_DIR"

# 设置适当的权限
echo "设置WordPress目录权限755..."
chmod -R 755 "$WORDPRESS_DIR"
if [ $? -ne 0 ]; then
    echo "警告: 设置WordPress目录权限失败，可能需要手动设置"
fi

# 解压MySQL数据库备份
echo "解压MySQL数据库备份..."
tar -xzf "$RESTORE_DIR/$LATEST_DB_BACKUP" -C "$RESTORE_DIR"
if [ $? -ne 0 ]; then
    echo "错误: 解压MySQL数据库备份失败"
    exit 1
fi
echo "MySQL数据库备份解压完成"

# 还原MySQL数据库
echo "还原MySQL数据库..."

# 获取SQL文件名（去掉.tar.gz后缀）
SQL_FILE=$(basename "$LATEST_DB_BACKUP" .tar.gz).sql


#wordpressdb 数据库容器
if [ -f "$RESTORE_DIR/$SQL_FILE" ]; then
    # 检查Docker和MySQL容器是否存在和运行
    if command -v docker &> /dev/null; then
        echo "检测到Docker已安装..."
        # 检查wordpressdb容器是否存在
        if docker container inspect wordpressdb &>/dev/null; then
            echo "检测到wordpressdb容器..."
            docker stop wordpressdb
            docker rm wordpressdb
            docker run -d \
                --name wordpressdb \
                --network wp-net \
                -p 3306:3306 \
                -e MYSQL_ROOT_PASSWORD=test123456789 \
                -e MYSQL_DATABASE=wordpress \
                -v /data/mysql/wordpressdb:/var/lib/mysql \
                mysql:8.0
            echo "MySQL容器创建完成✅"
        else
            echo "警告: wordpressdb容器不存在，需要创建新的MySQL容器"
            echo "是否要创建新的MySQL容器? (y/n)"
            read -r answer
            
            if [[ "$answer" =~ ^[Yy]$ ]]; then
                echo "创建MySQL容器..."
               
                if ! docker network ls | grep -q wp-net; then
                    docker network create wp-net
                fi

                # 创建MySQL数据目录
                mkdir -p /data/mysql/wordpressdb
                
                # 然后创建MySQL容器
                docker run -d \
                    --name wordpressdb \
                    --network wp-net \
                    -p 3306:3306 \
                    -e MYSQL_ROOT_PASSWORD="$MYSQL_ROOT_PASSWORD" \
                    -e MYSQL_DATABASE=wordpress \
                    -v /data/mysql/wordpressdb:/var/lib/mysql \
                    mysql:8.0
                echo "MySQL容器新建完成✅"
            else
                echo "跳过数据库还原。请手动创建MySQL容器并还原数据库。"
                echo "SQL备份文件位置: $RESTORE_DIR/$SQL_FILE"
            fi
        fi
        
    else
        echo "警告: Docker未安装，无法使用Docker还原MySQL数据库"
        echo "您可以手动安装MySQL并使用以下命令还原数据库:"
        echo "mysql -uroot -p < $RESTORE_DIR/$SQL_FILE"
        echo "SQL备份文件位置: $RESTORE_DIR/$SQL_FILE"
        
        # 暂时保留SQL文件
        mkdir -p ~/mysql_backup
        cp "$RESTORE_DIR/$SQL_FILE" ~/mysql_backup/
        echo "已将SQL文件复制到 ~/mysql_backup/$SQL_FILE 以便手动还原"
    fi
else
    echo "错误: 未找到SQL文件: $RESTORE_DIR/$SQL_FILE"
    exit 1
fi

#!/bin/bash

if [ "$DOCKER_INSTALLED" = true ]; then
    if [ "$WORDPRESS_CONTAINER_EXISTS" = false ]; then
        echo "注意: WordPress容器不存在"
        echo "您可以使用以下命令创建WordPress容器:"
        
        # 进入WordPress目录再执行容器创建命令
        cd "$WORDPRESS_DIR"
        echo "已进入WordPress目录: $(pwd)"
        
        # 检查Docker网络
        if ! docker network ls | grep -q wp-net; then
            echo "创建Docker网络wp-net..."
            docker network create wp-net
        fi
        
        echo "# 创建 php.ini 提高上传限制"
        cat > "$WORDPRESS_DIR/php.ini" <<EOF
upload_max_filesize = 64M
post_max_size = 64M
EOF
        
        echo "然后创建WordPress容器"
        docker run -d \
            --name wordpress \
            --network wp-net \
            -p 8080:80 \
            -e WORDPRESS_DB_HOST=wordpressdb:3306 \
            -e WORDPRESS_DB_NAME=wordpress \
            -e WORDPRESS_DB_USER=root \
            -e WORDPRESS_DB_PASSWORD="$MYSQL_ROOT_PASSWORD" \
            -v "$WORDPRESS_DIR":/var/www/html \
            -v "$WORDPRESS_DIR/php.ini":/usr/local/etc/php/conf.d/uploads.ini \
            wordpress:latest
    fi
else
    echo "容器已存在，重启wordpress容器"
    docker stop wordpress
    docker rm wordpress
    docker run -d \
        --name wordpress \
        --network wp-net \
        -p 8080:80 \
        -e WORDPRESS_DB_HOST=wordpressdb:3306 \
        -e WORDPRESS_DB_NAME=wordpress \
        -e WORDPRESS_DB_USER=root \
        -e WORDPRESS_DB_PASSWORD="$MYSQL_ROOT_PASSWORD" \
        -v "$WORDPRESS_DIR":/var/www/html \
        -v "$WORDPRESS_DIR/php.ini":/usr/local/etc/php/conf.d/uploads.ini \
        wordpress:latest
fi

echo "开始还原数据库。。。"
sleep 10

cat "$RESTORE_DIR/$SQL_FILE" | docker exec -i wordpressdb mysql -uroot -ptest123456 wordpress

echo "还原完成。。。"

# 清理临时文件
# 检查是否已将SQL文件复制到mysql_backup目录
echo "清理临时文件..."

if [ -f ~/mysql_backup/$(basename "$SQL_FILE") ]; then
    echo "保留了SQL备份文件以供手动还原，其余临时文件将被清除"
    find "$RESTORE_DIR" -type f -not -name "$(basename "$SQL_FILE")" -delete
    rmdir "$RESTORE_DIR" 2>/dev/null || true
else
    rm -rf "$RESTORE_DIR"
fi
echo "临时文件清理完成"