#!/bin/bash
set -e

# Ожидание готовности PostgreSQL
wait_for_postgres() {
    echo "Waiting for PostgreSQL..."
    python -c "
import psycopg2
import os
import sys

dbname = os.environ.get('DB_NAME', 'inventory_db')
user = os.environ.get('DB_USER', 'postgres')
password = os.environ.get('DB_PASSWORD', 'postgres')
host = os.environ.get('DB_HOST', 'db')
port = os.environ.get('DB_PORT', '5432')

print(f'Attempting connection to: host={host} port={port} dbname={dbname} user={user}')

try:
    conn = psycopg2.connect(
        dbname=dbname,
        user=user,
        password=password,
        host=host,
        port=port
    )
    conn.close()
    print('Connection successful!')
    exit(0)
except Exception as e:
    print(f'Error connecting to PostgreSQL: {e}', file=sys.stderr)
    exit(1)
"
    while [ $? -ne 0 ]; do
        echo "PostgreSQL is unavailable - sleeping"
        sleep 2
        python -c "
import psycopg2
import os
import sys
try:
    conn = psycopg2.connect(
        dbname=os.environ.get('DB_NAME', 'inventory_db'),
        user=os.environ.get('DB_USER', 'postgres'),
        password=os.environ.get('DB_PASSWORD', 'postgres'),
        host=os.environ.get('DB_HOST', 'db'),
        port=os.environ.get('DB_PORT', '5432')
    )
    conn.close()
    exit(0)
except Exception as e:
    print(f'Retry error: {e}', file=sys.stderr)
    exit(1)
"
    done
    echo "PostgreSQL is up!"
}

# Ожидание готовности Redis
wait_for_redis() {
    echo "Waiting for Redis..."
    python -c "
import redis
import os
import sys

host = os.environ.get('REDIS_HOST', 'redis')
port = int(os.environ.get('REDIS_PORT', '6379'))
db = int(os.environ.get('REDIS_DB', '0'))

print(f'Attempting connection to Redis: host={host} port={port} db={db}')

try:
    r = redis.Redis(host=host, port=port, db=db)
    r.ping()
    print('Redis connection successful!')
    exit(0)
except Exception as e:
    print(f'Error connecting to Redis: {e}', file=sys.stderr)
    exit(1)
"
    while [ $? -ne 0 ]; do
        echo "Redis is unavailable - sleeping"
        sleep 2
        python -c "
import redis
import os
import sys
try:
    r = redis.Redis(
        host=os.environ.get('REDIS_HOST', 'redis'),
        port=int(os.environ.get('REDIS_PORT', '6379')),
        db=int(os.environ.get('REDIS_DB', '0'))
    )
    r.ping()
    exit(0)
except Exception as e:
    print(f'Redis retry error: {e}', file=sys.stderr)
    exit(1)
"
    done
    echo "Redis is up!"
}

# Применение миграций и создание суперпользователя
run_migrations() {
    echo "Running migrations for shared apps..."
    python manage.py migrate_schemas --shared

    echo "Running migrations for tenant schemas..."
    python manage.py migrate_schemas --tenant

    echo "Collecting static files..."
    python manage.py collectstatic --noinput

    echo "Ensuring public tenant exists..."
    python /app/docker/create_public_tenant.py
}

# Создание суперпользователя если не существует
create_superuser() {
    echo "Creating superuser if not exists..."
    python manage.py shell << EOF
from django.contrib.auth import get_user_model
from django.db import connection
from django_tenants.utils import schema_context
import os

User = get_user_model()

# Создаём суперпользователя в public схеме из переменных окружения
with schema_context('public'):
    email = os.environ.get('TENANT_ADMIN_EMAIL', 'admin@example.com')
    password = os.environ.get('TENANT_ADMIN_PASSWORD', 'changeme')
    first_name = os.environ.get('TENANT_ADMIN_NAME', 'Admin')

    if not User.objects.filter(email=email).exists():
        user = User.objects.create_superuser(
            email=email,
            password=password,
            name=first_name
        )
        print(f'Superuser {email} created successfully!')
    else:
        print(f'Superuser {email} already exists.')
EOF
}

# Если manage.py не в текущей директории, но есть в подпапке myproject
if [ ! -f "manage.py" ] && [ -d "myproject" ]; then
    # Пытаемся войти в директорию, перенаправляя ошибки в /dev/null
    if cd myproject 2>/dev/null; then
        echo "Changing directory to myproject..."
        # Устанавливаем PYTHONPATH чтобы Python мог найти модуль myproject
        export PYTHONPATH=$(pwd):$PYTHONPATH
        echo "PYTHONPATH set to: $PYTHONPATH"
    else
        # Если не можем войти в директорию (проблема с правами), устанавливаем PYTHONPATH из текущей директории
        echo "Warning: Cannot access myproject directory (permission denied). Setting PYTHONPATH to include myproject..."
        export PYTHONPATH=/app/myproject:$PYTHONPATH
        echo "PYTHONPATH set to: $PYTHONPATH"
    fi
fi

case "$1" in
    web)
        wait_for_postgres
        wait_for_redis
        run_migrations
        create_superuser
        echo "Starting Gunicorn..."
        exec gunicorn myproject.wsgi:application \
            --bind 0.0.0.0:8000 \
            --workers 3 \
            --threads 2 \
            --timeout 120 \
            --access-logfile - \
            --error-logfile - \
            --capture-output
        ;;
    celery-worker)
        wait_for_postgres
        wait_for_redis
        echo "Starting Celery Worker..."
        exec celery -A myproject worker \
            -l info \
            -Q celery,photo_processing \
            --concurrency=2
        ;;
    celery-beat)
        wait_for_postgres
        wait_for_redis
        echo "Starting Celery Beat..."
        exec celery -A myproject beat -l info
        ;;
    migrate)
        wait_for_postgres
        run_migrations
        create_superuser
        ;;
    shell)
        exec python manage.py shell
        ;;
    *)
        exec "$@"
        ;;
esac