refactor(agent): MCP 기반 에이전트 아키텍처 재설계 — unified.md 분류기 제거, Gemini CLI + MCP 자율 도구 호출로 전환

mcp_servers/anime_server.py

@@ -0,0 +1,224 @@
+"""MCP 서버 — 애니메이션 도구.
+
+Gemini CLI에서 MCP로 연결하여 애니 검색/다운로드/편성표 조회를 수행합니다.
+stdio 트랜스포트를 사용합니다.
+"""
+
+import asyncio
+import sys
+from pathlib import Path
+
+# 프로젝트 루트를 sys.path에 추가 (tools/, config 접근용)
+PROJECT_ROOT = Path(__file__).parent.parent
+sys.path.insert(0, str(PROJECT_ROOT))
+
+import config  # noqa: E402 — .env 로드
+
+from mcp.server.fastmcp import FastMCP  # noqa: E402
+
+mcp = FastMCP("anime")
+
+
+# ──────────────────────────────────────────
+#  애니 검색
+# ──────────────────────────────────────────
+
+@mcp.tool()
+async def anime_search(title: str) -> str:
+    """애니메이션을 검색합니다.
+
+    제목으로 Anissia에서 애니를 검색하고, 자막 제작자 정보와
+    Nyaa 토렌트 검색 결과를 함께 반환합니다.
+
+    Args:
+        title: 검색할 애니 제목 (한글 또는 일본어)
+    """
+    from tools.anime_pipeline import AnimePipeline
+
+    pipeline = AnimePipeline()
+    result = await pipeline.search(title)
+
+    if not result.success:
+        errors = "; ".join(result.errors) if result.errors else ""
+        return f"검색 실패: {result.message} {errors}"
+
+    anime = result.anime
+    lines = [
+        f"## {anime.subject} ({anime.original_subject})",
+        f"- 장르: {anime.genres}",
+    ]
+
+    week_names = ['일', '월', '화', '수', '목', '금', '토', '기타']
+    if anime.week is not None:
+        lines.append(f"- 편성: {week_names[anime.week]}요일 {anime.time}")
+    lines.append(f"- NAS 폴더: {result.nas_folder}")
+
+    if result.captions:
+        lines.append(f"\n### 자막 ({len(result.captions)}명)")
+        for c in result.captions[:5]:
+            url = f" ({c.website})" if c.website else ""
+            lines.append(f"  - {c.name} — {c.episode}화{url}")
+
+    if result.torrents:
+        lines.append(f"\n### 토렌트 ({len(result.torrents)}건)")
+        for t in result.torrents[:5]:
+            ep = f"{t.episode}화 " if t.episode else ""
+            lines.append(f"  - [{t.group}] {ep}{t.size} (시드: {t.seeders})")
+
+    if result.errors:
+        lines.append(f"\n### 오류")
+        for e in result.errors:
+            lines.append(f"  - {e}")
+
+    return "\n".join(lines)
+
+
+# ──────────────────────────────────────────
+#  애니 다운로드
+# ──────────────────────────────────────────
+
+@mcp.tool()
+async def anime_download(
+    title: str,
+    mode: str = "auto",
+    episode: int | None = None,
+) -> str:
+    """애니메이션을 다운로드합니다 (자막+영상).
+
+    Anissia에서 검색 후 자막과 토렌트를 NAS에 다운로드합니다.
+
+    Args:
+        title: 다운로드할 애니 제목
+        mode: "auto" (자막+영상), "sub_only" (자막만), "video_only" (영상만)
+        episode: 특정 에피소드 번호 (None이면 최신)
+    """
+    from tools.anime_pipeline import AnimePipeline
+
+    pipeline = AnimePipeline()
+    result = await pipeline.download(title, mode=mode, episode=episode)
+    return result.message
+
+
+# ──────────────────────────────────────────
+#  편성표 조회
+# ──────────────────────────────────────────
+
+@mcp.tool()
+async def anime_schedule(weekday: int | None = None, sub_only: bool = False) -> str:
+    """애니 편성표를 조회합니다.
+
+    Args:
+        weekday: 요일 번호 (0=일, 1=월, ..., 6=토). None이면 전체 방영 중 목록.
+        sub_only: True이면 자막 있는 애니만 표시
+    """
+    from tools.anissia_client import AnissiaClient
+
+    client = AnissiaClient()
+
+    if weekday is not None:
+        schedule = await client.get_schedule(weekday)
+        week_names = ['일', '월', '화', '수', '목', '금', '토']
+        title = f"{week_names[weekday]}요일 편성표"
+    else:
+        schedule = await client.get_all_schedule()
+        schedule = [a for a in schedule if a.status == "ON"]
+        title = "현재 방영 중인 애니"
+
+    if sub_only:
+        schedule = [a for a in schedule if a.caption_count > 0]
+        title += " (자막 있음)"
+
+    lines = [f"## {title} ({len(schedule)}개)"]
+    for a in schedule[:30]:
+        sub = f" 📝{a.caption_count}" if a.caption_count > 0 else ""
+        lines.append(f"- {a.subject} — {a.time}{sub}")
+
+    if len(schedule) > 30:
+        lines.append(f"... 외 {len(schedule) - 30}개")
+
+    return "\n".join(lines)
+
+
+# ──────────────────────────────────────────
+#  다운로드 상태
+# ──────────────────────────────────────────
+
+@mcp.tool()
+async def anime_download_status() -> str:
+    """qBittorrent의 현재 다운로드 상태를 확인합니다."""
+    from tools.anime_pipeline import AnimePipeline
+
+    pipeline = AnimePipeline()
+    conn = await pipeline.qbit.test_connection()
+
+    if not conn.get("connected"):
+        return f"qBittorrent 연결 실패: {conn.get('error', '알 수 없는 오류')}"
+
+    torrents = await pipeline.get_status()
+
+    if not torrents:
+        return "다운로드 중인 항목이 없습니다."
+
+    lines = [f"## 다운로드 큐 ({len(torrents)}건)"]
+    for t in torrents[:15]:
+        icon = "✅" if t["progress"] == "100.0%" else "⏳"
+        lines.append(
+            f"- {icon} {t['name'][:50]} — {t['progress']} "
+            f"({t['speed']}, ETA: {t['eta']})"
+        )
+
+    return "\n".join(lines)
+
+
+# ──────────────────────────────────────────
+#  NAS 애니 목록
+# ──────────────────────────────────────────
+
+@mcp.tool()
+async def anime_nas_list(current_quarter: bool = False, keyword: str = "") -> str:
+    """NAS에 다운로드된 애니 목록을 조회합니다.
+
+    Args:
+        current_quarter: True이면 이번 분기 애니만 표시
+        keyword: 검색 키워드 (빈 문자열이면 전체)
+    """
+    from tools.nas_scanner import NasScanner
+
+    scanner = NasScanner()
+    if not scanner.is_accessible():
+        return f"NAS 접근 불가: {scanner.base_path}"
+
+    if keyword:
+        folders = scanner.search(keyword)
+    elif current_quarter:
+        folders = scanner.get_current_quarter_anime()
+    else:
+        folders = scanner.list_anime_folders()
+
+    if not folders:
+        return "조건에 맞는 다운로드된 애니가 없습니다."
+
+    total_vids = sum(f.video_count for f in folders)
+    total_subs = sum(f.subtitle_count for f in folders)
+    total_size = sum(f.total_size_gb for f in folders)
+
+    lines = [f"## NAS 애니 목록 ({len(folders)}개)"]
+    for f in folders[:25]:
+        sub = f" 📝{f.subtitle_count}" if f.subtitle_count > 0 else ""
+        lines.append(
+            f"- {f.title} — 🎬{f.video_count}화{sub} ({f.total_size_gb:.1f}GB)"
+        )
+
+    if len(folders) > 25:
+        lines.append(f"... 외 {len(folders) - 25}개")
+
+    lines.append(f"\n총 {total_vids}개 영상 | {total_subs}개 자막 | {total_size:.1f}GB")
+    return "\n".join(lines)
+
+
+# ──────────────────────────────────────────
+#  실행
+# ──────────────────────────────────────────
+
+if __name__ == "__main__":
+    mcp.run(transport="stdio")