#!/bin/sh
set -e

# Database initialization entrypoint for PieFed
# This script runs as a Kubernetes Job before web/worker pods start

log() {
    echo "[$(date +'%Y-%m-%d %H:%M:%S')] $1"
}

log "Starting PieFed database initialization..."

# Wait for database to be available
wait_for_db() {
    log "Waiting for database connection..."
    until python -c "
import psycopg2
import os
from urllib.parse import urlparse

try:
    # Parse DATABASE_URL
    database_url = os.environ.get('DATABASE_URL', '')
    if not database_url:
        raise Exception('DATABASE_URL not set')
    
    # Parse the URL to extract connection details
    parsed = urlparse(database_url)
    conn = psycopg2.connect(
        host=parsed.hostname,
        port=parsed.port or 5432,
        database=parsed.path[1:],  # Remove leading slash
        user=parsed.username,
        password=parsed.password
    )
    conn.close()
    print('Database connection successful')
except Exception as e:
    print(f'Database connection failed: {e}')
    exit(1)
" 2>/dev/null; do
        log "Database not ready, waiting 2 seconds..."
        sleep 2
    done
    log "Database connection established"
}

# Wait for Redis to be available  
wait_for_redis() {
    log "Waiting for Redis connection..."
    until python -c "
import redis
import os

try:
    cache_redis_url = os.environ.get('CACHE_REDIS_URL', '')
    if cache_redis_url:
        r = redis.from_url(cache_redis_url)
    else:
        # Fallback to separate host/port for backwards compatibility
        r = redis.Redis(host='redis', port=6379, password=os.environ.get('REDIS_PASSWORD', ''))
    r.ping()
    print('Redis connection successful')
except Exception as e:
    print(f'Redis connection failed: {e}')
    exit(1)
" 2>/dev/null; do
        log "Redis not ready, waiting 2 seconds..."
        sleep 2
    done
    log "Redis connection established"
}

# Main initialization sequence
main() {
    # Change to application directory
    cd /app
    
    # Wait for dependencies
    wait_for_db
    wait_for_redis
    
    # Run database migrations
    log "Running database migrations..."
    export FLASK_APP=pyfedi.py
    
    # Run Flask database migrations
    flask db upgrade
    log "Database migrations completed"
    
    # Populate community search index
    log "Populating community search..."
    flask populate_community_search
    log "Community search populated"
    
    # Ensure log files have correct ownership for dual logging (file + stdout)
    if [ -f /app/logs/pyfedi.log ]; then
        chown piefed:piefed /app/logs/pyfedi.log
        chmod 664 /app/logs/pyfedi.log
        log "Fixed log file ownership for piefed user"
    fi
    
    log "Database initialization completed successfully!"
}

# Run the main function
main