from abc import ABC, abstractmethod
from typing import Tuple

from sqlalchemy import select, func, and_, update
from sqlalchemy.ext.asyncio import AsyncSession

from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import quotation_settings
from common.enums import ErrorType
from common.logger import LOG
from common.utils.gtime import GTime


# 견적 설정 CRUD. 모든 조회/변경은 user_id 로 스코프된다(유저별 설정).
class IQuotationSettingCRUD(ABC):
    @abstractmethod
    async def list_by_user(self, cdb: AsyncSession, user_id) -> Tuple[ErrorType, list, int]:
        pass

    @abstractmethod
    async def get_by_id(self, cdb: AsyncSession, qt_setting_id) -> Tuple[ErrorType, quotation_settings]:
        pass

    @abstractmethod
    async def add_setting(self, cdb: AsyncSession, setting: quotation_settings) -> ErrorType:
        pass

    @abstractmethod
    async def update_setting(self, cdb: AsyncSession, qt_setting_id, data: dict) -> ErrorType:
        pass

    @abstractmethod
    async def soft_delete(self, cdb: AsyncSession, qt_setting_id) -> ErrorType:
        pass


class QuotationSettingCRUD(IQuotationSettingCRUD):
    async def list_by_user(self, cdb: AsyncSession, user_id) -> Tuple[ErrorType, list, int]:
        try:
            where = and_(quotation_settings.deleted == False, quotation_settings.user_id == user_id)  # noqa: E712

            cnt_err, cnt_rows = await DB_SESSION_MNG.execute(
                cdb, select(func.count()).select_from(quotation_settings).where(where)
            )
            if cnt_err != ErrorType.SUCCESS:
                return cnt_err, [], 0
            total = int(cnt_rows[0] or 0) if cnt_rows else 0

            list_err, rows = await DB_SESSION_MNG.execute(
                cdb,
                select(quotation_settings).where(where).order_by(quotation_settings.created_at.desc()),
            )
            if list_err != ErrorType.SUCCESS:
                return list_err, [], 0
            return ErrorType.SUCCESS, list(rows), total
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, [], 0

    async def get_by_id(self, cdb: AsyncSession, qt_setting_id) -> Tuple[ErrorType, quotation_settings]:
        try:
            query = (
                select(quotation_settings)
                .where(quotation_settings.qt_setting_id == qt_setting_id, quotation_settings.deleted == False)  # noqa: E712
                .limit(1)
            )
            err_type, row_list = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            if len(row_list) != 1:
                return ErrorType.DB_INVALID_KEY, None
            return ErrorType.SUCCESS, row_list[0]
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def add_setting(self, cdb: AsyncSession, setting: quotation_settings) -> ErrorType:
        try:
            return await DB_SESSION_MNG.insert(cdb, setting)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED

    async def update_setting(self, cdb: AsyncSession, qt_setting_id, data: dict) -> ErrorType:
        try:
            if not data:
                return ErrorType.SUCCESS
            query = update(quotation_settings).where(quotation_settings.qt_setting_id == qt_setting_id).values(**data)
            return await DB_SESSION_MNG.add(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED

    async def soft_delete(self, cdb: AsyncSession, qt_setting_id) -> ErrorType:
        try:
            query = (
                update(quotation_settings)
                .where(quotation_settings.qt_setting_id == qt_setting_id)
                .values(deleted=True, updated_at=GTime.UTC())
            )
            return await DB_SESSION_MNG.add(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED