refactor(agent): MCP 기반 에이전트 아키텍처 재설계 — unified.md 분류기 제거, Gemini CLI + MCP 자율 도구 호출로 전환
This commit is contained in:
1
mcp_servers/__init__.py
Normal file
1
mcp_servers/__init__.py
Normal file
@@ -0,0 +1 @@
|
||||
# MCP 서버 패키지
|
||||
224
mcp_servers/anime_server.py
Normal file
224
mcp_servers/anime_server.py
Normal file
@@ -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")
|
||||
161
mcp_servers/infra_server.py
Normal file
161
mcp_servers/infra_server.py
Normal file
@@ -0,0 +1,161 @@
|
||||
"""MCP 서버 — 인프라 도구 (Gitea + Vikunja).
|
||||
|
||||
Gemini CLI에서 MCP로 연결하여 Git 저장소 관리, 태스크 관리를 수행합니다.
|
||||
stdio 트랜스포트를 사용합니다.
|
||||
"""
|
||||
|
||||
import sys
|
||||
from pathlib import Path
|
||||
|
||||
# 프로젝트 루트를 sys.path에 추가
|
||||
PROJECT_ROOT = Path(__file__).parent.parent
|
||||
sys.path.insert(0, str(PROJECT_ROOT))
|
||||
|
||||
import config # noqa: E402
|
||||
|
||||
from mcp.server.fastmcp import FastMCP # noqa: E402
|
||||
|
||||
mcp = FastMCP("infra")
|
||||
|
||||
|
||||
# ══════════════════════════════════════════
|
||||
# Gitea 도구
|
||||
# ══════════════════════════════════════════
|
||||
|
||||
@mcp.tool()
|
||||
async def gitea_commits(limit: int = 5, branch: str = "main") -> str:
|
||||
"""Gitea 저장소의 최근 커밋 목록을 조회합니다.
|
||||
|
||||
Args:
|
||||
limit: 조회할 커밋 수 (기본 5)
|
||||
branch: 브랜치 이름 (기본 main)
|
||||
"""
|
||||
from integrations.gitea_client import GiteaClient
|
||||
|
||||
client = GiteaClient()
|
||||
commits = await client.get_commits(limit=limit, branch=branch)
|
||||
|
||||
if not commits:
|
||||
return "커밋이 없습니다."
|
||||
|
||||
lines = [f"## 최근 커밋 ({branch}, {len(commits)}개)"]
|
||||
for c in commits:
|
||||
lines.append(f"- `{c['sha']}` {c['message']} — {c['author']}")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def gitea_prs(state: str = "open") -> str:
|
||||
"""Gitea 저장소의 PR 목록을 조회합니다.
|
||||
|
||||
Args:
|
||||
state: "open" 또는 "closed" (기본 open)
|
||||
"""
|
||||
from integrations.gitea_client import GiteaClient
|
||||
|
||||
client = GiteaClient()
|
||||
prs = await client.list_prs(state=state)
|
||||
|
||||
if not prs:
|
||||
return f"{state} 상태의 PR이 없습니다."
|
||||
|
||||
lines = [f"## PR 목록 ({state}, {len(prs)}개)"]
|
||||
for p in prs:
|
||||
lines.append(f"- #{p['number']} {p['title']} ({p['state']}, {p['user']})")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def gitea_issues(state: str = "open") -> str:
|
||||
"""Gitea 저장소의 이슈 목록을 조회합니다.
|
||||
|
||||
Args:
|
||||
state: "open" 또는 "closed" (기본 open)
|
||||
"""
|
||||
from integrations.gitea_client import GiteaClient
|
||||
|
||||
client = GiteaClient()
|
||||
issues = await client.list_issues(state=state)
|
||||
|
||||
if not issues:
|
||||
return f"{state} 상태의 이슈가 없습니다."
|
||||
|
||||
lines = [f"## 이슈 목록 ({state}, {len(issues)}개)"]
|
||||
for i in issues:
|
||||
lines.append(f"- #{i['number']} {i['title']} ({i['state']})")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def gitea_branches() -> str:
|
||||
"""Gitea 저장소의 브랜치 목록을 조회합니다."""
|
||||
from integrations.gitea_client import GiteaClient
|
||||
|
||||
client = GiteaClient()
|
||||
branches = await client.list_branches()
|
||||
return "## 브랜치 목록\n" + "\n".join(f"- {b}" for b in branches)
|
||||
|
||||
|
||||
# ══════════════════════════════════════════
|
||||
# Vikunja 도구
|
||||
# ══════════════════════════════════════════
|
||||
|
||||
@mcp.tool()
|
||||
async def vikunja_tasks(filter: str = "todo") -> str:
|
||||
"""Vikunja 프로젝트의 태스크 목록을 조회합니다.
|
||||
|
||||
Args:
|
||||
filter: "todo" (미완료), "done" (완료), "all" (전체)
|
||||
"""
|
||||
from integrations.vikunja_client import VikunjaClient
|
||||
|
||||
client = VikunjaClient()
|
||||
tasks = await client.list_tasks(filter_=filter)
|
||||
|
||||
if not tasks:
|
||||
return f"{filter} 상태의 태스크가 없습니다."
|
||||
|
||||
lines = [f"## 태스크 ({filter}, {len(tasks)}개)"]
|
||||
for t in tasks:
|
||||
icon = "✅" if t["done"] else "⬜"
|
||||
labels = f" [{', '.join(t['labels'])}]" if t["labels"] else ""
|
||||
desc = f" — {t['description']}" if t["description"] else ""
|
||||
lines.append(f"- {icon} #{t['id']} {t['title']}{labels}{desc}")
|
||||
return "\n".join(lines)
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def vikunja_create_task(title: str, description: str = "") -> str:
|
||||
"""Vikunja에 새 태스크를 생성합니다.
|
||||
|
||||
Args:
|
||||
title: 태스크 제목
|
||||
description: 태스크 설명 (선택)
|
||||
"""
|
||||
from integrations.vikunja_client import VikunjaClient
|
||||
|
||||
client = VikunjaClient()
|
||||
result = await client.create_task(title=title, description=description)
|
||||
return f"태스크 #{result['id']} 생성: {result['title']}"
|
||||
|
||||
|
||||
@mcp.tool()
|
||||
async def vikunja_complete_task(task_id: int) -> str:
|
||||
"""Vikunja 태스크를 완료 처리합니다.
|
||||
|
||||
Args:
|
||||
task_id: 완료할 태스크 ID
|
||||
"""
|
||||
from integrations.vikunja_client import VikunjaClient
|
||||
|
||||
client = VikunjaClient()
|
||||
result = await client.mark_done(task_id)
|
||||
return f"태스크 #{task_id} 완료: {result['title']}"
|
||||
|
||||
|
||||
# ──────────────────────────────────────────
|
||||
# 실행
|
||||
# ──────────────────────────────────────────
|
||||
|
||||
if __name__ == "__main__":
|
||||
mcp.run(transport="stdio")
|
||||
Reference in New Issue
Block a user