from collections.abc import AsyncIterator

from sqlalchemy import event
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine

from config import settings

engine = create_async_engine(settings.database_url, echo=False, future=True)


# В SQLite ondelete=CASCADE / SET NULL не работают, пока для каждого подключения
# не включить PRAGMA foreign_keys=ON (по умолчанию выключено). aiosqlite не делает
# это автоматически. Без этого, например, удаление документа не очищало подписки
# в `intent_documents` — обнаружено в Спринте 8b.
@event.listens_for(engine.sync_engine, "connect")
def _enable_sqlite_foreign_keys(dbapi_connection, _connection_record):
    cursor = dbapi_connection.cursor()
    try:
        cursor.execute("PRAGMA foreign_keys=ON")
    finally:
        cursor.close()


SessionLocal = async_sessionmaker(engine, expire_on_commit=False, class_=AsyncSession)


async def get_session() -> AsyncIterator[AsyncSession]:
    async with SessionLocal() as session:
        yield session