<script src="https://telegram.org/js/telegram-web-app.js"></script>

async function getUserID() {
    const tg = window.Telegram.WebApp;
    if (tg.initDataUnsafe?.user?.id) {
        console.log("User data from Telegram:", JSON.stringify(tg.initDataUnsafe.user));
        return tg.initDataUnsafe.user.id;
    }

    const urlParams = new URLSearchParams(window.location.search);
    const userID = urlParams.get('user_id');
    if (userID) {
        console.log("User ID from URL:", userID);
        return userID;
    }

    throw new Error("User ID not available");
}

from contextlib import asynccontextmanager
from typing import Callable, Optional
from fastapi import Depends
from loguru import logger
from sqlalchemy.ext.asyncio import async_sessionmaker, AsyncSession
from typing_extensions import AsyncGenerator
from sqlalchemy import text
from functools import wraps

from app.dao.database import async_session_maker


class DatabaseSessionManager:
    """
    Менеджер для работы с сессиями базы данных. Позволяет управлять сессиями,
    транзакциями и предоставляет зависимости FastAPI для удобного внедрения сессий.
    """

    def __init__(self, session_maker: async_sessionmaker[AsyncSession]):
        self.async_session_maker = session_maker

    @asynccontextmanager
    async def get_session(self) -> AsyncGenerator[AsyncSession, None]:
        """
        Создает новую сессию базы данных.

        В этом методе создается асинхронная сессия базы данных.
        В случае ошибки логирует ее и повторно поднимает исключение,
        а также гарантирует закрытие сессии при завершении работы.

        Выходной тип: Асинхронный генератор сессии `AsyncSession`.
        """
        try:
            async with self.async_session_maker() as session:
                yield session
        except Exception as e:
            logger.error(f"Не удалось создать сессию базы данных: {str(e)}")
            raise
        finally:
            await session.close()

    @asynccontextmanager
    async def managed_transaction(self, session: AsyncSession) -> AsyncGenerator[None, None]:
        """
        Управляет транзакцией, осуществляя коммит или откат при ошибке.

        Этот метод принимает существующую сессию и выполняет коммит по завершении транзакции.
        Если возникает исключение, происходит откат транзакции и логирование ошибки.

        Параметр:
        - `session`: Текущая асинхронная сессия базы данных `AsyncSession`.

        Выходной тип: Асинхронный генератор.
        """
        try:
            yield
            await session.commit()
        except Exception as e:
            await session.rollback()
            logger.exception(f"Транзакция не удалась: {str(e)}")
            raise

    async def get_db(self) -> AsyncGenerator[AsyncSession, None]:
        """
        Зависимость для FastAPI, предоставляющая сессию без управления транзакцией.

        Асинхронный генератор для использования с FastAPI в качестве зависимости,
        который предоставляет сессию для выполнения запросов к базе данных.

        Выходной тип: Асинхронный генератор сессии `AsyncSession`.
        """
        async with self.get_session() as session:
            yield session

    async def get_db_with_transaction(self) -> AsyncGenerator[AsyncSession, None]:
        """
        Зависимость для FastAPI, предоставляющая сессию с управлением транзакцией.

        Асинхронный генератор для использования с FastAPI в качестве зависимости,
        который предоставляет сессию, автоматически выполняя коммит по завершении транзакции
        или откат в случае ошибки.

        Выходной тип: Асинхронный генератор сессии `AsyncSession`.
        """
        async with self.get_session() as session:
            async with self.managed_transaction(session):
                yield session

    def connection(self, isolation_level: Optional[str] = None, commit: bool = True):
        """
        Декоратор для управления сессией и дополнительными настройками транзакции.

        Позволяет добавлять сессию в метод с возможностью указания уровня изоляции транзакции.
        Если `commit=True`, выполняется коммит в конце метода, иначе коммит не выполняется.

        Параметры:
        - `isolation_level` (опционально): Уровень изоляции для транзакции (например, "SERIALIZABLE").
        - `commit` (по умолчанию True): Определяет, будет ли выполняться коммит после метода.

        Возвращает:
        - Декорированный асинхронный метод с дополнительными параметрами управления транзакцией.
        """