#!/bin/bash

# Print environment variables for debugging
echo "Environment variables:"
echo "POSTGRES_USER: $POSTGRES_USER"
echo "POSTGRES_PASSWORD: $POSTGRES_PASSWORD"
echo "POSTGRES_DB: $POSTGRES_DB"
echo "DATABASE_URL: $DATABASE_URL"

# Function to wait for database
wait_for_db() {
    echo "Waiting for database..."
    while ! nc -z db 5432; do
        sleep 1
    done
    echo "Database is ready!"
}

# Function to create database if it doesn't exist
create_database() {
    echo "Creating database if it doesn't exist..."
    PGPASSWORD=$POSTGRES_PASSWORD psql -h db -U $POSTGRES_USER -tc "SELECT 1 FROM pg_database WHERE datname = '$POSTGRES_DB'" | grep -q 1 || \
    PGPASSWORD=$POSTGRES_PASSWORD psql -h db -U $POSTGRES_USER -c "CREATE DATABASE $POSTGRES_DB"
    echo "Database check/creation complete!"
}

# Wait for database to be ready
wait_for_db

# Create database if it doesn't exist
create_database

# Wait for PostgreSQL to be ready to accept connections
echo "Waiting for PostgreSQL to accept connections..."
until PGPASSWORD=$POSTGRES_PASSWORD psql -h db -U $POSTGRES_USER -d $POSTGRES_DB -c '\q'; do
  echo "PostgreSQL is unavailable - sleeping"
  sleep 1
done
echo "PostgreSQL is up - executing command"

# Run migrations
echo "Running database migrations..."
flask db upgrade

# Create events table
echo "Creating events table..."
python3 -c "
from migrations.add_events_table import upgrade
from app import create_app
app = create_app()
with app.app_context():
    try:
        upgrade()
        print('Events table created successfully')
    except Exception as e:
        print(f'Error creating events table: {e}')
"

# Create notifs table
echo "Creating notifs table..."
python3 -c "
from migrations.add_notifs_table import upgrade
from app import create_app
app = create_app()
with app.app_context():
    try:
        upgrade()
        print('Notifs table created successfully')
    except Exception as e:
        print(f'Error creating notifs table: {e}')
"

# Create default site settings if they don't exist
echo "Creating default site settings..."
python3 -c "
from app import create_app
from models import SiteSettings, db
app = create_app()
with app.app_context():
    try:
        settings = SiteSettings.get_settings()
        print('Default site settings created successfully')
    except Exception as e:
        print(f'Error creating site settings: {e}')
"

# Initialize admin user
echo "Initializing admin user..."
python3 -c "
from init_admin import init_admin
init_admin()
"

# Create admin user
echo "Creating admin user..."
flask create-admin

# Create default templates
echo "Creating default templates..."
flask create-default-templates

# Start the application
echo "Starting application..."
exec gunicorn --bind 0.0.0.0:5000 app:app