o2o-site-AEO/solution/backend/crud/site_crud.py

from abc import ABC, abstractmethod
from typing import Tuple

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

from common.database.db_session_manager import DB_SESSION_MNG
from common.database.model.models import place_photos, places, site_publish_logs, site_versions, sites
from common.enums import BuildStatus, ErrorType, MediaStatus, SiteStatus
from common.logger import LOG
from common.utils.gtime import GTime


def _primary_photo_subquery():
    """place_photos 에서 대표 사진 한 장의 url 만 고르는 상관 서브쿼리(사업장당 1행).

    site_payload.primary_media 와 같은 규칙 — 객실·메뉴 사진(unit_id 있음)이 아닌 첫 장,
    sort_order 순. `.correlate(places)` 라서 바깥 쿼리가 `places` 를 셀렉트에 들고 있어야 한다.
    sites.thumbnail_url 이 비어 있을 때(Azure 썸네일 저장소 미설정 등) 서비스 계층이 이걸로
    대신 채운다 — 여기서는 후보만 얹고, 언제 쓸지는 서비스 계층 몫이다."""
    return (
        select(place_photos.url)
        .where(
            place_photos.place_id == places.place_id,
            place_photos.deleted == False,  # noqa: E712
            place_photos.status == MediaStatus.APPROVED.value,
            place_photos.unit_id.is_(None),
        )
        .order_by(place_photos.sort_order.asc(), place_photos.created_at.asc())
        .limit(1)
        .correlate(places)
        .scalar_subquery()
    )


# 사이트/버전/발행로그 CRUD. 항상 place_id 또는 site_id 로 스코프한다.
class ISiteCRUD(ABC):
    @abstractmethod
    async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
        pass

    @abstractmethod
    async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
        pass

    @abstractmethod
    async def list_owner_sites(self, cdb: AsyncSession, owner_user_id, skip, limit) -> Tuple[ErrorType, list, int]:
        """(ErrorType, [(place, site, built_at, primary_photo_url)], 총건수)."""
        pass

    @abstractmethod
    async def taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
        pass

    @abstractmethod
    async def add_site(self, cdb: AsyncSession, site: sites) -> ErrorType:
        pass

    @abstractmethod
    async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def add_version(self, cdb: AsyncSession, version: site_versions) -> ErrorType:
        pass

    @abstractmethod
    async def get_version(self, cdb: AsyncSession, site_id, site_version_id) -> Tuple[ErrorType, site_versions]:
        pass

    @abstractmethod
    async def get_version_by_number(self, cdb: AsyncSession, site_id, version: int) -> Tuple[ErrorType, site_versions]:
        pass

    @abstractmethod
    async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
        pass

    @abstractmethod
    async def add_log(self, cdb: AsyncSession, log: site_publish_logs) -> ErrorType:
        pass

    @abstractmethod
    async def list_logs(self, cdb: AsyncSession, site_id, limit: int) -> Tuple[ErrorType, list]:
        pass

    @abstractmethod
    async def list_published(self, cdb: AsyncSession, limit: int) -> Tuple[ErrorType, list]:
        """(ErrorType, [(site, place, primary_photo_url)])."""
        pass


class SiteCRUD(ISiteCRUD):
    async def get_site_by_place(self, cdb: AsyncSession, place_id) -> Tuple[ErrorType, sites]:
        try:
            query = select(sites).where(sites.place_id == place_id, sites.deleted == False).limit(1)  # noqa: E712
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            return ErrorType.SUCCESS, (rows[0] if rows else None)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def get_site_by_domain(self, cdb: AsyncSession, domain: str) -> Tuple[ErrorType, sites]:
        """주소(도메인 라벨)의 주인을 찾는다. 없으면 (SUCCESS, None).

        uq_sites_domain(deleted=false AND domain IS NOT NULL)과 같은 조건으로 본다 —
        인덱스가 막는 것과 조회가 막는 것이 다르면 "확인은 통과, 저장은 실패"가 난다."""
        try:
            query = select(sites).where(sites.domain == domain, sites.deleted == False).limit(1)  # noqa: E712
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            return ErrorType.SUCCESS, (rows[0] if rows else None)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def list_owner_sites(self, cdb: AsyncSession, owner_user_id, skip: int, limit: int) -> Tuple[ErrorType, list, int]:
        """사장님의 사업장 + 사이트 + 마지막 빌드 시각 + 빌더 대표 사진.
        (ErrorType, [(place, site, built_at, primary_photo_url)], 총건수).

        따로 읽으면 줄마다 사이트를 다시 물어 N+1 이다. LEFT JOIN 이라 사이트가 없는 사업장
        (위저드만 걸어온 것)도 내려간다 — 빠지면 만들다 만 것을 찾을 길이 없다.

        ★ primary_photo_url 은 site_payload.primary_media 와 같은 규칙(사진 중 객실·메뉴가 아닌
          첫 장, sort_order 순)으로 고른 place_photos.url 이다 — sites.thumbnail_url 이 비어 있을 때
          (Azure 썸네일 저장소 미설정 등으로 재호스팅에 실패한 경우) 서비스 계층이 이걸로 대신 채운다.
          여기서는 후보만 얹고, "발행한 적 있는 줄에만 쓴다"는 판단은 서비스 계층 몫이다."""
        try:
            where = and_(places.deleted == False, places.owner_user_id == owner_user_id)  # noqa: E712

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

            query = (
                select(places, sites, site_versions.built_at, _primary_photo_subquery())
                .outerjoin(sites, and_(sites.place_id == places.place_id, sites.deleted == False))  # noqa: E712
                .outerjoin(site_versions, site_versions.site_version_id == sites.current_version_id)
                .where(where)
                .order_by(places.created_at.desc())
                .offset(skip)
                .limit(limit)
            )
            list_err, rows = await DB_SESSION_MNG.execute(cdb, query)
            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 taken_domains(self, cdb: AsyncSession, domains: list) -> Tuple[ErrorType, set]:
        """후보 주소들 중 이미 쓰이는 것만 추린다. 대안 제안이 후보마다 왕복하지 않게 한 번에 본다."""
        try:
            if not domains:
                return ErrorType.SUCCESS, set()
            query = select(sites.domain).where(
                sites.domain.in_(domains), sites.deleted == False  # noqa: E712
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, set()
            return ErrorType.SUCCESS, {r for r in rows if r}
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, set()

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

    async def next_version_no(self, cdb: AsyncSession, site_id) -> Tuple[ErrorType, int]:
        """다음 버전 번호. 1부터 시작한다."""
        try:
            query = select(func.max(site_versions.version)).where(
                site_versions.site_id == site_id, site_versions.deleted == False  # noqa: E712
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, 0
            current = rows[0] if rows else None
            return ErrorType.SUCCESS, int(current or 0) + 1
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0

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

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

    async def get_version_by_number(self, cdb: AsyncSession, site_id, version: int) -> Tuple[ErrorType, site_versions]:
        """롤백 대상 조회 — site_versions.version(사람이 보는 번호) 로 찾는다.

        ★ site_version_id(uuid) 가 아니다. 화면·API 는 버전 번호로 고르는 게 자연스럽고,
          그 번호가 곧 out/versions/<slug>/<version>/ 디렉토리 이름이다(prerender.ts)."""
        try:
            query = (
                select(site_versions)
                .where(
                    site_versions.site_id == site_id,
                    site_versions.version == version,
                    site_versions.deleted == False,  # noqa: E712
                )
                .limit(1)
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            if err_type != ErrorType.SUCCESS:
                return err_type, None
            if len(rows) != 1:
                return ErrorType.DB_INVALID_KEY, None
            return ErrorType.SUCCESS, rows[0]
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, None

    async def finish_version(self, cdb: AsyncSession, site_version_id, data: dict) -> Tuple[ErrorType, int]:
        """빌드 결과를 버전에 기록한다(BUILT 또는 FAILED)."""
        try:
            query = (
                update(site_versions)
                .where(site_versions.site_version_id == site_version_id, site_versions.deleted == False)  # noqa: E712
                .values(**data, updated_at=GTime.UTC())
            )
            return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0

    async def update_site(self, cdb: AsyncSession, site_id, data: dict) -> Tuple[ErrorType, int]:
        try:
            query = (
                update(sites)
                .where(sites.site_id == site_id, sites.deleted == False)  # noqa: E712
                .values(**data, updated_at=GTime.UTC())
            )
            return await DB_SESSION_MNG.add_with_rowcount(cdb, query)
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, 0

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

    async def list_logs(self, cdb: AsyncSession, site_id, limit: int = 50) -> Tuple[ErrorType, list]:
        try:
            query = (
                select(site_publish_logs)
                .where(site_publish_logs.site_id == site_id, site_publish_logs.deleted == False)  # noqa: E712
                .order_by(site_publish_logs.created_at.desc())
                .limit(limit)
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            return (err_type, list(rows) if err_type == ErrorType.SUCCESS else [])
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, []

    async def list_published(self, cdb: AsyncSession, limit: int = 12) -> Tuple[ErrorType, list]:
        """발행된 사이트 + 그 사업장 + 빌더 대표 사진을 최신순으로. 랜딩 쇼케이스가 읽는 목록이다.
        (ErrorType, [(site, place, primary_photo_url)]).

        ★ 회사 스코프가 없는 **유일한** 사이트 조회다(비로그인 API 가 쓴다). 그래서 행을 통째로
          돌려주고, 무엇이 밖으로 나갈지는 services/showcase_service 한 곳에서만 고른다 —
          여기서 열을 골라 두면 나중에 필드를 늘릴 때 공개 여부를 판단할 자리가 사라진다.

        ★ primary_photo_url 은 list_owner_sites 와 같은 서브쿼리(_primary_photo_subquery) —
          sites.thumbnail_url 이 비어 있을 때 showcase_service 가 이걸로 대신 채운다."""
        try:
            query = (
                select(sites, places, _primary_photo_subquery())
                .join(places, places.place_id == sites.place_id)
                .where(
                    sites.status == SiteStatus.PUBLISHED.value,
                    sites.deleted == False,  # noqa: E712
                    places.deleted == False,  # noqa: E712
                )
                .order_by(sites.published_at.desc().nulls_last(), sites.created_at.desc())
                .limit(limit)
            )
            err_type, rows = await DB_SESSION_MNG.execute(cdb, query)
            return (err_type, list(rows) if err_type == ErrorType.SUCCESS else [])
        except Exception as ex:
            LOG.e_no_callstack(ex)
            return ErrorType.DB_RUN_FAILED, []