"""Сервис для управления базой данных."""

from pathlib import Path

import aiosqlite
import structlog
from sqlmodel import SQLModel, create_engine

from ..models import AppConfig

logger = structlog.get_logger()


class DatabaseService:
    def __init__(self, config: AppConfig) -> None:
        self.config = config
        self.logger = structlog.get_logger().bind(service="database")

        self._sync_engine = create_engine(
            config.sync_db_url,
            echo=False,
            connect_args={"check_same_thread": False},
        )

    async def initialize_database(self) -> None:
        db_path = Path(self.config.db_path)
        db_path.parent.mkdir(parents=True, exist_ok=True)

        self.logger.info("Создание схемы базы данных", db_path=self.config.db_path)
        SQLModel.metadata.create_all(self._sync_engine)

        await self._configure_sqlite()

        self.logger.info("База данных инициализирована", db_path=self.config.db_path)

    async def _configure_sqlite(self) -> None:
        async with aiosqlite.connect(self.config.db_path) as conn:
            await conn.execute("PRAGMA journal_mode=WAL")

            await conn.execute("PRAGMA cache_size=10000")

            await conn.execute("PRAGMA synchronous=NORMAL")

            await conn.execute("PRAGMA busy_timeout=30000")

            await conn.commit()
            self.logger.info("SQLite настроен для оптимальной производительности")

    async def get_connection(self) -> aiosqlite.Connection:
        return await aiosqlite.connect(
            self.config.db_path,
            timeout=30.0,
        )

    async def health_check(self) -> bool:
        try:
            async with self.get_connection() as conn:
                await conn.execute("SELECT 1")
                return True
        except Exception as e:
            self.logger.error("Database health check failed", error=str(e))
            return False

    def close(self) -> None:
        self._sync_engine.dispose()