xNightR00T File Manager

#!/bin/bash
# Script Deploy Database
# Backup database hiện tại và chạy file SQL mới
#
# Usage: ./deploy_db.sh [file.sql]
# Example: ./deploy_db.sh merge_db/output.sql
#          ./deploy_db.sh output.sql

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 tham số
if [ $# -eq 0 ]; then
    echo -e "${RED}❌ Lỗi: Chưa chỉ định file SQL${NC}"
    echo ""
    echo "Usage: $0 [file.sql]"
    echo ""
    echo "Ví dụ:"
    echo "  $0 merge_db/output.sql"
    echo "  $0 output.sql"
    exit 1
fi

SQL_FILE="$1"

# Kiểm tra file SQL có tồn tại không
if [ ! -f "$SQL_FILE" ]; then
    # Thử với đường dẫn trong thư mục script
    if [ -f "$SCRIPT_DIR/$SQL_FILE" ]; then
        SQL_FILE="$SCRIPT_DIR/$SQL_FILE"
    # Thử trong thư mục merge_db
    elif [ -f "$SCRIPT_DIR/merge_db/$SQL_FILE" ]; then
        SQL_FILE="$SCRIPT_DIR/merge_db/$SQL_FILE"
    else
        echo -e "${RED}❌ Lỗi: File SQL không tồn tại: $SQL_FILE${NC}"
        exit 1
    fi
fi

# 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

# Timestamp cho backup
TIMESTAMP=$(date +"%Y%m%d_%H%M%S")
BACKUP_FILE="$BACKUP_DIR/backup_${DB_NAME}_${TIMESTAMP}.sql"
BACKUP_INFO_FILE="$BACKUP_DIR/backup_${DB_NAME}_${TIMESTAMP}.info"

echo -e "${BLUE}========================================${NC}"
echo -e "${BLUE}  DEPLOY 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 "File SQL: ${GREEN}$SQL_FILE${NC}"
echo ""

# Bước 1: Backup database hiện tại
echo -e "${YELLOW}📦 Bước 1: Đang backup database...${NC}"
echo "Backup file: $BACKUP_FILE"

# Lưu thông tin backup
cat > "$BACKUP_INFO_FILE" <<EOF
TIMESTAMP=$TIMESTAMP
BACKUP_FILE=$BACKUP_FILE
SQL_FILE=$SQL_FILE
DEPLOY_DATE=$(date +"%Y-%m-%d %H:%M:%S")
DB_NAME=$DB_NAME
DB_HOST=$DB_HOST
EOF

# Export password để không bị prompt
export MYSQL_PWD="$DB_PASSWORD"

# Backup database
if $MYSQLDUMP_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" > "$BACKUP_FILE" 2>/dev/null; then
    BACKUP_SIZE=$(du -h "$BACKUP_FILE" | cut -f1)
    echo -e "${GREEN}✅ Backup thành công!${NC}"
    echo -e "   Kích thước: $BACKUP_SIZE"
    echo ""
else
    echo -e "${RED}❌ Lỗi: Không thể backup database${NC}"
    echo "Vui lòng kiểm tra:"
    echo "  - Database credentials"
    echo "  - Kết nối đến database server"
    echo "  - Quyền truy cập database"
    exit 1
fi

# Bước 2: Chạy file SQL mới
echo -e "${YELLOW}🚀 Bước 2: Đang chạy file SQL mới...${NC}"

# Kiểm tra xem file SQL có rỗng không
if [ ! -s "$SQL_FILE" ]; then
    echo -e "${YELLOW}⚠️  Cảnh báo: File SQL trống, bỏ qua${NC}"
    exit 0
fi

# Chạy SQL
if $MYSQL_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" < "$SQL_FILE" 2>/dev/null; then
    echo -e "${GREEN}✅ Deploy thành công!${NC}"
    echo ""
    echo -e "${BLUE}========================================${NC}"
    echo -e "${GREEN}  KẾT QUẢ${NC}"
    echo -e "${BLUE}========================================${NC}"
    echo -e "📦 Backup: ${GREEN}$BACKUP_FILE${NC}"
    echo -e "🚀 Deploy: ${GREEN}$SQL_FILE${NC}"
    echo -e "📅 Thời gian: ${GREEN}$(date +"%Y-%m-%d %H:%M:%S")${NC}"
    echo ""
    echo -e "${YELLOW}💡 Lưu ý: Backup đã được lưu tại $BACKUP_FILE${NC}"
    echo -e "   Bạn có thể rollback bằng: ${BLUE}./rollback_db.sh $TIMESTAMP${NC}"
    echo ""
else
    echo -e "${RED}❌ Lỗi: Không thể chạy file SQL${NC}"
    echo ""
    echo -e "${YELLOW}🔄 Đang rollback về backup trước đó...${NC}"
    
    # Tự động rollback nếu có lỗi
    if [ -f "$BACKUP_FILE" ]; then
        if $MYSQL_CMD -h "$DB_HOST" -u "$DB_USER" "$DB_NAME" < "$BACKUP_FILE" 2>/dev/null; then
            echo -e "${GREEN}✅ Đã rollback về backup trước đó${NC}"
        else
            echo -e "${RED}❌ Lỗi: Không thể rollback tự động${NC}"
            echo -e "${YELLOW}⚠️  Vui lòng rollback thủ công: ${BLUE}./rollback_db.sh $TIMESTAMP${NC}"
        fi
    fi
    exit 1
fi

# Unset password
unset MYSQL_PWD