site11/services/news-api/backend/app/services/article_service.py

from typing import List, Optional
from datetime import datetime
from bson import ObjectId
from app.core.database import get_collection
from app.models.article import Article, ArticleList, ArticleSummary
from app.core.config import settings

SUPPORTED_LANGUAGES = ["ko", "en", "zh_cn", "zh_tw", "ja", "fr", "de", "es", "it"]

class ArticleService:

    @staticmethod
    def validate_language(language: str) -> bool:
        """언어 코드 검증"""
        return language in SUPPORTED_LANGUAGES

    @staticmethod
    async def get_articles(
        language: str,
        page: int = 1,
        page_size: int = 20,
        category: Optional[str] = None
    ) -> ArticleList:
        """기사 목록 조회"""
        collection = get_collection(language)

        # 필터 구성
        query = {}
        if category:
            query["category"] = category

        # 전체 개수
        total = await collection.count_documents(query)

        # 페이지네이션
        skip = (page - 1) * page_size
        cursor = collection.find(query).sort("created_at", -1).skip(skip).limit(page_size)

        articles = []
        async for doc in cursor:
            doc["_id"] = str(doc["_id"])
            articles.append(Article(**doc))

        total_pages = (total + page_size - 1) // page_size

        return ArticleList(
            total=total,
            page=page,
            page_size=page_size,
            total_pages=total_pages,
            articles=articles
        )

    @staticmethod
    async def get_article_by_id(language: str, article_id: str) -> Optional[Article]:
        """ID로 기사 조회"""
        collection = get_collection(language)

        try:
            doc = await collection.find_one({"_id": ObjectId(article_id)})
            if doc:
                doc["_id"] = str(doc["_id"])
                return Article(**doc)
        except Exception as e:
            print(f"Error fetching article: {e}")

        return None

    @staticmethod
    async def get_latest_articles(
        language: str,
        limit: int = 10
    ) -> List[ArticleSummary]:
        """최신 기사 조회"""
        collection = get_collection(language)

        cursor = collection.find().sort("created_at", -1).limit(limit)

        articles = []
        async for doc in cursor:
            doc["_id"] = str(doc["_id"])
            articles.append(ArticleSummary(**doc))

        return articles

    @staticmethod
    async def search_articles(
        language: str,
        keyword: str,
        page: int = 1,
        page_size: int = 20
    ) -> ArticleList:
        """기사 검색"""
        collection = get_collection(language)

        # 텍스트 검색 쿼리
        query = {
            "$or": [
                {"title": {"$regex": keyword, "$options": "i"}},
                {"content": {"$regex": keyword, "$options": "i"}},
                {"summary": {"$regex": keyword, "$options": "i"}},
                {"tags": {"$regex": keyword, "$options": "i"}}
            ]
        }

        # 전체 개수
        total = await collection.count_documents(query)

        # 페이지네이션
        skip = (page - 1) * page_size
        cursor = collection.find(query).sort("created_at", -1).skip(skip).limit(page_size)

        articles = []
        async for doc in cursor:
            doc["_id"] = str(doc["_id"])
            articles.append(Article(**doc))

        total_pages = (total + page_size - 1) // page_size

        return ArticleList(
            total=total,
            page=page,
            page_size=page_size,
            total_pages=total_pages,
            articles=articles
        )

    @staticmethod
    async def get_categories(language: str) -> List[str]:
        """카테고리 목록 조회"""
        collection = get_collection(language)

        categories = await collection.distinct("category")
        return [cat for cat in categories if cat]