MCP Servers & Tool Design (ACI)

#!/usr/bin/env python3
"""project-mcp-server.py — 프로젝트 특화 MCP 서버 템플릿"""
from mcp.server.fastmcp import FastMCP
from typing import Literal
import subprocess
import json
import os

mcp = FastMCP('project-tools')

@mcp.tool()
async def run_tests(
    pattern: str = '',
    verbose: bool = False
) -> dict:
    """프로젝트 테스트를 실행한다.
    
    pattern: 실행할 테스트 파일/이름 패턴 (빈 문자열 = 전체)
    verbose: True면 전체 출력, False면 요약만
    
    코드 변경 후 검증 시 사용. PR 전 필수.
    결과: passed, failed, duration, output 반환
    """
    cmd = ['npm', 'test']
    if pattern:
        cmd.extend(['--', '--testPathPattern', pattern])
    if verbose:
        cmd.append('--verbose')
    
    result = subprocess.run(cmd, capture_output=True, text=True, timeout=120)
    
    lines = result.stdout.split('\n')
    summary_lines = [l for l in lines if 'Tests:' in l or 'PASS' in l or 'FAIL' in l]
    
    return {
        'exit_code': result.returncode,
        'passed': result.returncode == 0,
        'summary': '\n'.join(summary_lines[-10:]),
        'output': result.stdout if verbose else result.stdout[-2000:],
        'error': result.stderr[-1000:] if result.returncode != 0 else None
    }

@mcp.tool()
async def search_codebase(
    query: str,
    file_pattern: str = '**/*.ts',
    context_lines: int = 3
) -> dict:
    """코드베이스에서 패턴을 검색한다.
    
    query: grep 패턴
    file_pattern: 검색할 파일 패턴 (기본: TypeScript 파일)
    context_lines: 매칭 전후 표시할 줄 수
    
    함수, 변수, import 위치 찾을 때 사용. Read 전에 먼저 이것을 써서 파일을 특정하라.
    결과: matches (파일경로, 줄번호, 내용), total_count 반환
    """
    try:
        result = subprocess.run(
            ['grep', '-rn', f'--include={file_pattern.replace("**","*")}',
             f'-A{context_lines}', f'-B{context_lines}', query, '.'],
            capture_output=True, text=True, timeout=30
        )
        matches = []
        for line in result.stdout.split('\n')[:100]:  # 최대 100줄
            if ':' in line and not line.startswith('--'):
                parts = line.split(':', 2)
                if len(parts) >= 3:
                    matches.append({
                        'file': parts[0],
                        'line': parts[1],
                        'content': parts[2].strip()
                    })
        return {'matches': matches, 'total_count': len(matches), 'truncated': len(matches) >= 100}
    except subprocess.TimeoutExpired:
        return {'error': '검색 시간 초과 — 더 구체적인 패턴을 사용하세요', 'matches': [], 'total_count': 0}

@mcp.tool()
async def get_env_status() -> dict:
    """현재 개발 환경 상태를 반환한다.
    
    Node.js 버전, 환경변수 설정 여부, 포트 사용 현황 확인.
    세션 시작 시 또는 '왜 안되지?' 디버깅 시 먼저 실행하라.
    """
    info = {}
    
    for cmd, key in [(['node', '--version'], 'node'), (['npm', '--version'], 'npm'), (['git', 'branch', '--show-current'], 'branch')]:
        try:
            r = subprocess.run(cmd, capture_output=True, text=True)
            info[key] = r.stdout.strip()
        except:
            info[key] = 'not found'
    
    required_env = ['DATABASE_URL', 'NEXTAUTH_SECRET', 'NEXTAUTH_URL']
    info['env_vars'] = {
        var: '✅ 설정됨' if os.environ.get(var) else '❌ 미설정'
        for var in required_env
    }
    
    return info

if __name__ == '__main__':
    mcp.run(transport='stdio')