#!/bin/bash

# OpenCand Database Deployment Script
# This script deploys the database schema changes to the PostgreSQL container

set -e  # Exit on any error

# Configuration
DB_CONTAINER="opencand_db"
DB_NAME="opencand"
DB_USER="root"
SQL_URL="https://git.ivanch.me/ivanch/opencand/raw/branch/main/db/db.sql"
SQL_FILE="./db.sql"

# Colors for output
RED='\033[0;31m'
GREEN='\033[0;32m'
YELLOW='\033[1;33m'
NC='\033[0m' # No Color

echo -e "${YELLOW}🚀 OpenCand Database Deployment Script${NC}"
echo "=================================="

# Check if Docker is running
if ! docker info >/dev/null 2>&1; then
    echo -e "${RED}❌ Error: Docker is not running or not accessible${NC}"
    exit 1
fi

# Check if the database container is running
if ! docker ps --format "table {{.Names}}" | grep -q "^${DB_CONTAINER}$"; then
    echo -e "${RED}❌ Error: Database container '${DB_CONTAINER}' is not running${NC}"
    echo "Please start the services with: docker-compose up -d"
    exit 1
fi

# Download the SQL file
echo -e "${YELLOW}📥 Downloading SQL file...${NC}"
if command -v curl &> /dev/null; then
    if curl -L -o "$SQL_FILE" "$SQL_URL"; then
        echo "✅ SQL file downloaded successfully"
    else
        echo -e "${RED}❌ Error: Failed to download SQL file from ${SQL_URL}${NC}"
        exit 1
    fi
elif command -v wget &> /dev/null; then
    if wget -O "$SQL_FILE" "$SQL_URL"; then
        echo "✅ SQL file downloaded successfully"
    else
        echo -e "${RED}❌ Error: Failed to download SQL file from ${SQL_URL}${NC}"
        exit 1
    fi
else
    echo -e "${RED}❌ Error: Neither curl nor wget is available for downloading${NC}"
    exit 1
fi

# Check if SQL file exists (after download)
if [ ! -f "$SQL_FILE" ]; then
    echo -e "${RED}❌ Error: SQL file '${SQL_FILE}' not found after download${NC}"
    exit 1
fi

echo -e "${YELLOW}📋 Pre-deployment checks:${NC}"
echo "✅ Docker is running"
echo "✅ Database container is running"
echo "✅ SQL file downloaded"
echo ""

# Wait for database to be ready
echo -e "${YELLOW}⏳ Waiting for database to be ready...${NC}"
timeout=30
counter=0
while ! docker exec $DB_CONTAINER pg_isready -U $DB_USER -d $DB_NAME >/dev/null 2>&1; do
    sleep 1
    counter=$((counter + 1))
    if [ $counter -ge $timeout ]; then
        echo -e "${RED}❌ Error: Database failed to become ready within ${timeout} seconds${NC}"
        exit 1
    fi
done
echo "✅ Database is ready"
echo ""

# Deploy the SQL file
echo -e "${YELLOW}🔧 Deploying database changes...${NC}"
echo "Executing: $SQL_FILE"

if docker exec -i $DB_CONTAINER psql -U $DB_USER -d $DB_NAME < $SQL_FILE; then
    echo -e "${GREEN}✅ Database deployment completed successfully!${NC}"
    echo ""
    
    # Show table information
    echo -e "${YELLOW}📊 Current database tables:${NC}"
    docker exec $DB_CONTAINER psql -U $DB_USER -d $DB_NAME -c "\dt"
    
else
    echo -e "${RED}❌ Error: Database deployment failed${NC}"
    echo -e "${YELLOW}💡 You can restore from backup using:${NC}"
    echo "docker exec -i $DB_CONTAINER psql -U $DB_USER -d $DB_NAME < $BACKUP_FILE"
    exit 1
fi

# Clean up downloaded SQL file
echo -e "${YELLOW}🧹 Cleaning up...${NC}"
rm -f "$SQL_FILE"
echo "✅ Temporary SQL file removed"

echo ""
echo -e "${GREEN}🎉 Deployment completed successfully!${NC}"
echo "=================================="