class Protocol<Request, Notification, Result> {
  // 들어오는 요청(request)을 처리하는 핸들러를 등록합니다.
  // schema는 요청의 데이터 구조를 검증하는 역할을 하고,
  // handler는 실제 요청을 처리하는 비동기 함수입니다.
  setRequestHandler<T>(
    schema: T,
    handler: (request: T, extra: RequestHandlerExtra) => Promise<Result>
  ): void

  // 들어오는 알림(notification)을 처리하는 핸들러를 등록합니다.
  // schema는 알림 데이터의 구조를 검증하며,
  // handler는 알림을 처리하는 비동기 함수입니다.
  setNotificationHandler<T>(schema: T, handler: (notification: T) => Promise<void>): void

  // 요청(request)을 서버(또는 상대방)로 보내고 응답을 기다립니다.
  // schema를 통해 응답 데이터의 구조를 검증하며,
  // options를 통해 요청 옵션을 설정할 수 있습니다.
  request<T>(request: Request, schema: T, options?: RequestOptions): Promise<T>

  // 알림(notification)을 서버(또는 상대방)로 전송합니다.
  // 응답을 기다리지 않는 일방향 메시지입니다.
  notification(notification: Notification): Promise<void>
}

{
  "dependencies": {
    "@modelcontextprotocol/sdk": "^1.10.1",
    "cheerio": "^1.0.0",
    "tsx": "^4.19.3",
    "zod": "^3.24.3",
    "zod-to-json-schema": "^3.24.5"
  },
  "devDependencies": {
    "@types/node": "^22.14.1",
    "ts-node": "^10.9.2",
    "typescript": "^5.8.3"
  }
}

const VERSION = '0.1.0'
const server = new Server(
  {
    name: 'namuwiki-mcp',
    version: VERSION,
  },
  {
    capabilities: {
      tools: {},
    },
  }
)

import * as cheerio from 'cheerio'

export async function fetchNamuWiki(title: string): Promise<{
  contentHtml: string
}> {
  const encoded = encodeURIComponent(title)
  const url = `https://namu.wiki/w/${encoded}`

  const res = await fetch(url)
  if (!res.ok) {
    throw new Error(`Failed to fetch NamuWiki page: ${res.status}`)
  }
  const html = await res.text()

  const $ = cheerio.load(html)

  const nodeTexts = $('#app')
    .find('*') // app 이하 모든 요소
    .contents() // 텍스트·요소 모두
    .filter((i, node) => node.type === 'text') // 텍스트 노드만 남기고
    .map((i, node) => $(node).text().trim()) // 각각의 텍스트 추출
    .get() // 배열로 변환
    .join(' ') // 원하는 구분자로 합치기

  if (!nodeTexts.length) {
    throw new Error('Could not locate article content')
  }

  return {
    contentHtml: nodeTexts,
  }
}

const FetchWikiSchema = z.object({
  title: z.string().describe('나무위키 문서 제목'),
})

server.setRequestHandler(ListToolsRequestSchema, async () => {
  return {
    tools: [
      {
        name: 'fetch_namuwiki_article',
        description: '나무위키 문서 내용을 불러옵니다.',
        inputSchema: zodToJsonSchema(FetchWikiSchema),
      },
    ],
  }
})

server.setRequestHandler(CallToolRequestSchema, async (request) => {
  const { name, arguments: args } = request.params
  if (name === 'fetch_namuwiki_article') {
    try {
      const parsed = FetchWikiSchema.parse(args)
      const data = await fetchNamuWiki(parsed.title)

      const contentHtml = data?.contentHtml ?? '내용을 불러올 수 없습니다.'
      return {
        content: [
          {
            type: 'text',
            text: `📘 ${contentHtml}`,
          },
        ],
      }
    } catch (error) {
      // 에러 처리
    }
  }
  throw new Error(`도구 '${name}'을(를) 찾을 수 없습니다.`)
})

npx @modelcontextprotocol/inspector node wiki-mcp/dist/server.js