xNightR00T File Manager

#!/bin/bash
# Script Rollback Database
# Rollback database về version đã backup trước đó
#
# Usage: ./rollback_db.sh [backup_timestamp]
# Example: ./rollback_db.sh 20260119_143022
#          ./rollback_db.sh  # Hiển thị danh sách backup

set -e  # Dừng nếu có lỗi

# Màu sắc cho output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
BLUE='\033[0;34m'
NC='\033[0m' # No Color

# Load config
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$SCRIPT_DIR/db_config.sh"

# Kiểm tra thông tin database
if [ -z "$DB_NAME" ] || [ -z "$DB_USER" ] || [ -z "$DB_HOST" ]; then
    echo -e "${RED}❌ Lỗi: Thiếu thông tin database configuration${NC}"
    echo "Vui lòng kiểm tra file db_config.sh hoặc wp-config.php"
    exit 1
fi

# Nếu không có tham số, hiển thị danh sách backup
if [ $# -eq 0 ]; then
    echo -e "${BLUE}========================================${NC}"
    echo -e "${BLUE}  DANH SÁCH BACKUP${NC}"
    echo -e "${BLUE}========================================${NC}"
    echo ""
    
    BACKUP_PATTERN="$BACKUP_DIR/backup_${DB_NAME}_*.sql"
    
    if ls $BACKUP_PATTERN 1> /dev/null 2>&1; then
        echo -e "${GREEN}Các backup có sẵn:${NC}"
        echo ""
        printf "%-20s %-12s %-30s\n" "TIMESTAMP" "SIZE" "FILE"
        echo "--------------------------------------------"
        
        for backup_file in $(ls -t $BACKUP_PATTERN); do
            timestamp=$(basename "$backup_file" | sed "s/backup_${DB_NAME}_\(.*\)\.sql/\1/")
            size=$(du -h "$backup_file" | cut -f1)
            filename=$(basename "$backup_file")
            
            # Kiểm tra file info
            info_file="$BACKUP_DIR/backup_${DB_NAME}_${timestamp}.info"
            if [ -f "$info_file" ]; then
                deploy_date=$(grep "^DEPLOY_DATE=" "$info_file" | cut -d'=' -f2-)
                sql_file=$(grep "^SQL_FILE=" "$info_file" | cut -d'=' -f2-)
                printf "%-20s %-12s %-30s\n" "$timestamp" "$size" "$filename"
                echo "  └─ Deploy: $(basename "$sql_file") - $deploy_date"
            else
                printf "%-20s %-12s %-30s\n" "$timestamp" "$size" "$filename"
            fi
            echo ""
        done
        
        echo ""
        echo -e "${YELLOW}Usage:${NC}"
        echo "  $0 [timestamp]"
        echo ""
        echo -e "${YELLOW}Ví dụ:${NC}"
        echo "  $0 20260119_143022"
        echo ""
    else
        echo -e "${RED}❌ Không tìm thấy backup nào${NC}"
        echo "Thư mục backup: $BACKUP_DIR"
        exit 1
    fi
    
    exit 0
fi

BACKUP_TIMESTAMP="$1"
BACKUP_FILE="$BACKUP_DIR/backup_${DB_NAME}_${BACKUP_TIMESTAMP}.sql"
BACKUP_INFO_FILE="$BACKUP_DIR/backup_${DB_NAME}_${BACKUP_TIMESTAMP}.info"

# Kiểm tra file backup có tồn tại không
if [ ! -f "$BACKUP_FILE" ]; then
    echo -e "${RED}❌ Lỗi: Không tìm thấy backup với timestamp: $BACKUP_TIMESTAMP${NC}"
    echo ""
    echo "Chạy lệnh sau để xem danh sách backup:"
    echo "  $0"
    exit 1
fi

echo -e "${BLUE}========================================${NC}"
echo -e "${BLUE}  ROLLBACK DATABASE${NC}"
echo -e "${BLUE}========================================${NC}"
echo ""
echo -e "Database: ${GREEN}$DB_NAME${NC}"
echo -e "Host: ${GREEN}$DB_HOST${NC}"
echo -e "User: ${GREEN}$DB_USER${NC}"
echo -e "Backup file: ${GREEN}$BACKUP_FILE${NC}"

# Hiển thị thông tin backup nếu có
if [ -f "$BACKUP_INFO_FILE" ]; then
    deploy_date=$(grep "^DEPLOY_DATE=" "$BACKUP_INFO_FILE" | cut -d'=' -f2-)
    sql_file=$(grep "^SQL_FILE=" "$BACKUP_INFO_FILE" | cut -d'=' -f2-)
    echo -e "Deploy từ: ${YELLOW}$sql_file${NC} (ngày $deploy_date)"
fi

BACKUP_SIZE=$(du -h "$BACKUP_FILE" | cut -f1)
echo -e "Kích thước: ${YELLOW}$BACKUP_SIZE${NC}"
echo ""

# Xác nhận trước khi rollback
read -p "$(echo -e ${YELLOW}Bạn có chắc chắn muốn rollback? [y/N]: ${NC})" -n 1 -r
echo ""
if [[ ! $REPLY =~ ^[Yy]$ ]]; then
    echo -e "${YELLOW}Đã hủy rollback${NC}"
    exit 0
fi

# Bước 1: Backup database hiện tại (trước khi rollback)
echo ""
echo -e "${YELLOW}📦 Bước 1: Đang backup database hiện tại...${NC}"
CURRENT_TIMESTAMP=$(date +"%Y%m%d_%H%M%S")
PRE_ROLLBACK_BACKUP="$BACKUP_DIR/pre_rollback_${DB_NAME}_${CURRENT_TIMESTAMP}.sql"

# Export password
export MYSQL_PWD="$DB_PASSWORD"

# Backup database hiện tại
if $MYSQLDUMP_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" > "$PRE_ROLLBACK_BACKUP" 2>/dev/null; then
    echo -e "${GREEN}✅ Đã backup database hiện tại: $PRE_ROLLBACK_BACKUP${NC}"
    echo ""
else
    echo -e "${YELLOW}⚠️  Cảnh báo: Không thể backup database hiện tại${NC}"
    echo "Tiếp tục rollback..."
    echo ""
fi

# Bước 2: Rollback
echo -e "${YELLOW}🔄 Bước 2: Đang rollback...${NC}"

# Chạy backup file
if $MYSQL_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" < "$BACKUP_FILE" 2>/dev/null; then
    echo -e "${GREEN}✅ Rollback thành công!${NC}"
    echo ""
    echo -e "${BLUE}========================================${NC}"
    echo -e "${GREEN}  KẾT QUẢ${NC}"
    echo -e "${BLUE}========================================${NC}"
    echo -e "🔄 Rollback từ: ${GREEN}$BACKUP_FILE${NC}"
    echo -e "📦 Backup trước rollback: ${GREEN}$PRE_ROLLBACK_BACKUP${NC}"
    echo -e "📅 Thời gian: ${GREEN}$(date +"%Y-%m-%d %H:%M:%S")${NC}"
    echo ""
else
    echo -e "${RED}❌ Lỗi: Không thể rollback database${NC}"
    echo ""
    echo -e "${YELLOW}⚠️  Database có thể đã bị thay đổi một phần${NC}"
    echo -e "   Backup trước rollback: $PRE_ROLLBACK_BACKUP"
    echo ""
    
    # Hỏi có muốn restore từ pre-rollback backup không
    read -p "$(echo -e ${YELLOW}Khôi phục từ backup trước rollback? [y/N]: ${NC})" -n 1 -r
    echo ""
    if [[ $REPLY =~ ^[Yy]$ ]] && [ -f "$PRE_ROLLBACK_BACKUP" ]; then
        echo -e "${YELLOW}Đang khôi phục...${NC}"
        if $MYSQL_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" < "$PRE_ROLLBACK_BACKUP" 2>/dev/null; then
            echo -e "${GREEN}✅ Đã khôi phục về trạng thái trước rollback${NC}"
        else
            echo -e "${RED}❌ Lỗi: Không thể khôi phục${NC}"
            exit 1
        fi
    fi
    
    unset MYSQL_PWD
    exit 1
fi

# Unset password
unset MYSQL_PWD