launch.ps1

$ErrorActionPreference = "Stop"

function Write-ColorOutput {
    param(
        [string]$Message,
        [ValidateSet("Black", "DarkBlue", "DarkGreen", "DarkCyan", "DarkRed", "DarkMagenta", "DarkYellow", "Gray", "DarkGray", "Blue", "Green", "Cyan", "Red", "Magenta", "Yellow", "White")]
        [string]$Color = "White"
    )
    Write-Host $Message -ForegroundColor $Color
}

function Write-Header {
    param([string]$Title)
    Write-ColorOutput "=====================================" "Blue"
    Write-ColorOutput "  $Title" "Blue"
    Write-ColorOutput "=====================================" "Blue"
    Write-Host ""
}

function Show-Usage {
    Write-Header "WeatherTrack Pro - Launch Script"
    Write-Host "Usage: .\launch.ps1 [option]"
    Write-Host ""
    Write-Host "Options:"
    Write-Host "  (no option)  Launch both old and new versions"
    Write-Host "  old          Launch only the old version"
    Write-Host "  new          Launch only the new version"
    Write-Host "  stop         Stop all running services"
    Write-Host "  restart      Restart all services"
    Write-Host "  logs         Show logs for all services"
    Write-Host "  clean        Stop services and remove volumes"
    Write-Host ""
}

function Check-Service {
    param(
        [string]$Url,
        [string]$Name,
        [int]$MaxAttempts = 30
    )

    Write-ColorOutput "Waiting for $Name to be ready..." "Yellow"

    for ($i = 1; $i -le $MaxAttempts; $i++) {
        try {
            $response = Invoke-WebRequest -Uri "$Url/weather/data/Lyon" -Method Get -TimeoutSec 2 -ErrorAction Stop
            Write-Host "[" -NoNewline
            Write-Host "OK" -ForegroundColor Green -NoNewline
            Write-Host "] $Name is ready!"
            return $true
        }
        catch {
            Write-Host "." -NoNewline
            Start-Sleep -Seconds 1
        }
    }

    Write-Host ""
    Write-Host "[" -NoNewline
    Write-Host "FAIL" -ForegroundColor Red -NoNewline
    Write-Host "] $Name did not start properly"
    return $false
}

function Launch-All {
    Write-ColorOutput "Launching both versions of WeatherTrack Pro..." "Yellow"
    Write-Host ""

    try {
        docker-compose up -d
    }
    catch {
        Write-ColorOutput "Error: Failed to start Docker containers" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }

    Write-Host ""
    Write-ColorOutput "Checking services availability..." "Yellow"
    Write-Host ""

    $oldReady = Check-Service -Url "http://localhost:3000" -Name "Old API"
    $newReady = Check-Service -Url "http://localhost:3001" -Name "New API"

    Write-Host ""

    if ($oldReady -and $newReady) {
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] All services started successfully!"
        Write-Host ""
        Write-ColorOutput "Services URLs:" "Blue"
        Write-ColorOutput "  Old API: http://localhost:3000" "Green"
        Write-ColorOutput "  New API: http://localhost:3001" "Green"
        Write-ColorOutput "  Old DB:  localhost:5432" "Green"
        Write-ColorOutput "  New DB:  localhost:5433" "Green"
        Write-Host ""
        Write-ColorOutput "Use 'docker-compose logs -f' to view logs" "Yellow"
    }
    else {
        Write-Host "[" -NoNewline
        Write-Host "WARN" -ForegroundColor Yellow -NoNewline
        Write-Host "] Some services failed to start properly"
        Write-Host ""

        if ($oldReady) {
            Write-ColorOutput "Services available:" "Blue"
            Write-ColorOutput "  Old API: http://localhost:3000" "Green"
            Write-ColorOutput "  Old DB:  localhost:5432" "Green"
        }

        if ($newReady) {
            Write-ColorOutput "Services available:" "Blue"
            Write-ColorOutput "  New API: http://localhost:3001" "Green"
            Write-ColorOutput "  New DB:  localhost:5433" "Green"
        }

        Write-Host ""
        Write-ColorOutput "Check logs with: docker-compose logs" "Yellow"
        Write-ColorOutput "Troubleshooting:" "Red"
        Write-ColorOutput "  1. Check if ports 3000, 3001, 5432, 5433 are available" "Red"
        Write-ColorOutput "  2. Verify Docker containers are running: docker-compose ps" "Red"
        Write-ColorOutput "  3. Check individual service logs: docker-compose logs [service-name]" "Red"
    }
}

function Launch-Old {
    Write-ColorOutput "Launching old version of WeatherTrack Pro..." "Yellow"
    Write-Host ""

    try {
        docker-compose up -d old-db old-api
    }
    catch {
        Write-ColorOutput "Error: Failed to start Docker containers" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }

    Write-Host ""
    Write-ColorOutput "Checking service availability..." "Yellow"
    Write-Host ""

    $oldReady = Check-Service -Url "http://localhost:3000" -Name "Old API"

    Write-Host ""

    if ($oldReady) {
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] Old version started successfully!"
        Write-Host ""
        Write-ColorOutput "Service URLs:" "Blue"
        Write-ColorOutput "  Old API: http://localhost:3000" "Green"
        Write-ColorOutput "  Old DB:  localhost:5432" "Green"
        Write-Host ""
        Write-ColorOutput "Use 'docker-compose logs -f old-api old-db' to view logs" "Yellow"
    }
    else {
        Write-Host "[" -NoNewline
        Write-Host "FAIL" -ForegroundColor Red -NoNewline
        Write-Host "] Old version failed to start properly"
        Write-Host ""
        Write-ColorOutput "Check logs with: docker-compose logs old-api old-db" "Yellow"
        Write-ColorOutput "Troubleshooting:" "Red"
        Write-ColorOutput "  1. Check if ports 3000, 5432 are available" "Red"
        Write-ColorOutput "  2. Verify Docker containers are running: docker-compose ps" "Red"
        Write-ColorOutput "  3. Check service logs: docker-compose logs old-api old-db" "Red"
    }
}

function Launch-New {
    Write-ColorOutput "Launching new version of WeatherTrack Pro..." "Yellow"
    Write-Host ""

    try {
        docker-compose up -d new-db new-api
    }
    catch {
        Write-ColorOutput "Error: Failed to start Docker containers" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }

    Write-Host ""
    Write-ColorOutput "Checking service availability..." "Yellow"
    Write-Host ""

    $newReady = Check-Service -Url "http://localhost:3001" -Name "New API"

    Write-Host ""

    if ($newReady) {
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] New version started successfully!"
        Write-Host ""
        Write-ColorOutput "Service URLs:" "Blue"
        Write-ColorOutput "  New API: http://localhost:3001" "Green"
        Write-ColorOutput "  New DB:  localhost:5433" "Green"
        Write-Host ""
        Write-ColorOutput "Use 'docker-compose logs -f new-api new-db' to view logs" "Yellow"
    }
    else {
        Write-Host "[" -NoNewline
        Write-Host "FAIL" -ForegroundColor Red -NoNewline
        Write-Host "] New version failed to start properly"
        Write-Host ""
        Write-ColorOutput "Check logs with: docker-compose logs new-api new-db" "Yellow"
        Write-ColorOutput "Troubleshooting:" "Red"
        Write-ColorOutput "  1. Check if ports 3001, 5433 are available" "Red"
        Write-ColorOutput "  2. Verify Docker containers are running: docker-compose ps" "Red"
        Write-ColorOutput "  3. Check service logs: docker-compose logs new-api new-db" "Red"
    }
}

function Stop-Services {
    Write-ColorOutput "Stopping all services..." "Yellow"
    Write-Host ""

    try {
        docker-compose down
        Write-Host ""
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] All services stopped successfully!"
    }
    catch {
        Write-ColorOutput "Error: Failed to stop services" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }
}

function Restart-Services {
    Write-ColorOutput "Restarting all services..." "Yellow"
    Write-Host ""

    try {
        docker-compose restart
        Write-Host ""
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] All services restarted successfully!"
        Write-Host ""
        Write-ColorOutput "Note: Services may take a few seconds to be fully operational" "Yellow"
    }
    catch {
        Write-ColorOutput "Error: Failed to restart services" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }
}

function Show-Logs {
    Write-ColorOutput "Showing logs for all services..." "Yellow"
    Write-Host ""

    try {
        docker-compose logs -f
    }
    catch {
        Write-ColorOutput "Error: Failed to retrieve logs" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }
}

function Clean-All {
    Write-ColorOutput "Stopping services and removing volumes..." "Yellow"
    Write-Host ""

    try {
        docker-compose down -v
        Write-Host ""
        Write-Host "[" -NoNewline
        Write-Host "OK" -ForegroundColor Green -NoNewline
        Write-Host "] All services stopped and volumes removed!"
    }
    catch {
        Write-ColorOutput "Error: Failed to clean services" "Red"
        Write-Host $_.Exception.Message
        exit 1
    }
}

$option = if ($args.Count -eq 0) { "all" } else { $args[0] }

switch ($option) {
    "all" {
        Launch-All
    }
    "old" {
        Launch-Old
    }
    "new" {
        Launch-New
    }
    "stop" {
        Stop-Services
    }
    "restart" {
        Restart-Services
    }
    "logs" {
        Show-Logs
    }
    "clean" {
        Clean-All
    }
    { $_ -in "help", "--help", "-h" } {
        Show-Usage
    }
    default {
        Write-ColorOutput "Error: Invalid option '$option'" "Red"
        Write-Host ""
        Show-Usage
        exit 1
    }
}