MCP 서버 메타데이터 동기화: 개발자 생산성을 위한 컨텍스트 관리 전략

use jsonrpc_core::{Result, Params};
use std::collections::HashMap;
use serde_json::Value;

// 가상의 MCP 핸들러 구조체
pub struct BlogMcpServer {
    // 캐싱된 메타데이터
    cached_context: String,
}

impl BlogMcpServer {
    pub fn new() -> Self {
        // 서버 초기화 시 핵심 컨텍스트(아키텍처, 규칙)를 로드
        let context = r#"
        프로젝트: ZeroClaw Multi-Agent Runtime
        아키텍처: 파일 기반 이벤트 버스를 사용하는 비동기 멀티 에이전트 시스템.
        주요 규칙:
        1. 모든 에이전트는 독립된 스레드에서 실행됨.
        2. 통신은 JSON-RPC over MCP 프로토콜을 따름.
        3. 로그는 구조화된 JSON 형식으로 출력해야 함.
        "#;
        
        Self { cached_context: context.to_string() }
    }

    // MCP 리소스 조회 핸들러 (resources/read)
    pub fn handle_read_resource(&self, params: Params) -> Result<Value> {
        let params_obj: HashMap<String, Value> = params.parse()?;
        let uri = params_obj.get("uri").and_then(|v| v.as_str()).unwrap_or("");

        match uri {
            "context://architecture/guide" => {
                // 클라이언트가 요청할 때 캐싱된 컨텍스트 반환
                Ok(json!({
                    "contents": [
                        {
                            "uri": uri,
                            "mimeType": "text/plain",
                            "text": self.cached_context.clone()
                        }
                    ]
                }))
            },
            "context://trends/latest" => {
                // 외부 RSS(Hacker News 등)를 실시간으로 페칭하여 제공 (동적 리소스)
                let trends = fetch_latest_hn_trends().await.unwrap_or_default();
                Ok(json!({
                    "contents": [{
                        "uri": uri,
                        "mimeType": "application/json",
                        "text": trends
                    }]
                }))
            },
            _ => Err(Error::invalid_params("Unknown resource URI"))
        }
    }
}

async fn fetch_latest_hn_trends() -> Option<String> {
    // 실제로는 reqwest 등을 사용하여 RSS를 파싱
    // 예시를 위해 간단한 JSON 문자열 반환
    Some(r#"[{"title": "JSON-LD Explained", "link": "..."}]"#.to_string())
}

{
  "name": "generate_blog_post",
  "description": "Creates a technical blog post based on architecture context and latest trends.",
  "arguments": [
    { "name": "topic", "description": "Main topic", "required": true },
    { "name": "tone", "description": "Writing style", "required": false }
  ]
}

pub fn handle_get_prompt(&self, params: Params) -> Result<Value> {
    let name = params.get("name").and_then(|v| v.as_str()).unwrap_or("");
    let args = params.get("arguments").and_then(|v| v.as_object()).unwrap_or(&json!(null));

    if name == "generate_blog_post" {
        let topic = args.get("topic").unwrap_or(&json!("general"));
        
        // 리소스에서 가져온 아키텍처 가이드를 프롬프트에 내장
        let system_instruction = format!(
            "You are writing for a blog defined by the following architecture:\n{}\n\nPlease write a post about: {}",
            self.cached_context, // 앞서 로드한 아키텍처 정보
            topic
        );

        Ok(json!({
            "messages": [
                {
                    "role": "user",
                    "content": {
                        "type": "text",
                        "text": system_instruction
                    }
                }
            ]
        }))
    } else {
        Err(Error::invalid_params("Unknown prompt"))
    }
}