import logging

from sqlalchemy import select

from fabledassistant.models import async_session
from fabledassistant.models.setting import Setting

logger = logging.getLogger(__name__)


async def get_setting(key: str, default: str = "") -> str:
    async with async_session() as session:
        result = await session.execute(select(Setting).where(Setting.key == key))
        setting = result.scalar_one_or_none()
        return setting.value if setting else default


async def set_setting(key: str, value: str) -> None:
    async with async_session() as session:
        result = await session.execute(select(Setting).where(Setting.key == key))
        setting = result.scalar_one_or_none()
        if setting:
            setting.value = value
        else:
            session.add(Setting(key=key, value=value))
        await session.commit()


async def set_settings_batch(settings: dict[str, str]) -> None:
    """Update multiple settings in a single transaction."""
    async with async_session() as session:
        for key, value in settings.items():
            result = await session.execute(select(Setting).where(Setting.key == key))
            setting = result.scalar_one_or_none()
            if setting:
                setting.value = value
            else:
                session.add(Setting(key=key, value=value))
        await session.commit()
    logger.info("Batch-updated %d settings", len(settings))


async def get_all_settings() -> dict[str, str]:
    async with async_session() as session:
        result = await session.execute(select(Setting))
        return {s.key: s.value for s in result.scalars().all()}