from typing import Set, List

from app.core.database import DatabaseManager


class LongArticleDatabaseMapper:
    class Basic:
        @staticmethod
        async def get_unsafe_articles(pool: DatabaseManager) -> Set[str]:
            query = """
                    SELECT title FROM article_unsafe_title WHERE status = 1;
                """
            response = await pool.async_fetch(query=query)
            return set([i["title"] for i in response])

    class Recommend(Basic):
        @staticmethod
        async def get_top_articles(pool: DatabaseManager, query: str, gh_id: str):
            return await pool.async_fetch(query=query, params=(gh_id,))