WIP

Author: kazupon

docs/.vitepress/client.ts

@@ -45,10 +45,19 @@ async function main() {
   // List resources
   const resources = await client.listResources()
   console.log('list resources', resources)
+  const resourceTemplates = await client.listResourceTemplates()
+  console.log('list resource templates', resourceTemplates)
   const contents = await client.readResource({
-    uri: 'vue-i18n://contents'
+    uri: 'vue-i18n://docs'
   })
-  console.log('read resource', contents)
+  // console.log('read resource', contents)
+  // const page = await client.readResource({
+  //   uri: 'vue-i18n://docs{page}',
+  //   params: {
+  //     page: '/guide/installation.md'
+  //   }
+  // })
+  // console.log('read page', page)
 }
 
 main()

docs/.vitepress/config.mts

@@ -8,7 +8,7 @@ import type { HeadConfig } from 'vitepress'
 
 const head: HeadConfig[] = [['link', { rel: 'icon', href: '/vue-i18n-logo.png' }]]
 
-export default defineConfig({
+const config = defineConfig({
   title: 'Vue I18n',
   description: 'Internationalization plugin for Vue.js',
 
@@ -316,3 +316,5 @@ function sidebarEcosystem() {
     }
   ]
 }
+
+export default config

docs/.vitepress/mcp.ts

@@ -1,23 +1,39 @@
+import { ResourceTemplate } from '@modelcontextprotocol/sdk/server/mcp.js'
+import { existsSync } from 'node:fs'
 import fs from 'node:fs/promises'
 import path from 'node:path'
 
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js'
 import type { ViteDevServer } from 'vite'
-import type { SiteConfig } from 'vitepress'
+import type { DefaultTheme, SiteConfig, SiteData } from 'vitepress'
 
 type Awaitable<T> = T | PromiseLike<T>;
 
-export default function mcp(mcpServer: McpServer, viteServer: ViteDevServer): Awaitable<void | McpServer> {
+export default async function mcp(mcpServer: McpServer, viteServer: ViteDevServer): Awaitable<void | McpServer> {
   const vitepress = (viteServer.config as any).vitepress as SiteConfig
-  console.log('Setting up MCP server for Vue I18n docs...', vitepress.userConfig.themeConfig)
-  // mcpServer.resource('contents', 'vue-i18n://contents', async (uri) => {
-  mcpServer.resource('contents', 'vue-i18n://contents', {
-    uri: 'vue-i18n://contents',
-    name: 'Vue I18n Contents',
-    description: 'List of Contents for Vue I18n',
+  const docRootDir = vitepress.srcDir
+  const docFiles = (vitepress.pages || []).map(page => path.resolve(docRootDir, page)).filter(file => existsSync(file))
+  console.log('VitePress pages:', docFiles)
+  const themeConfig = vitepress.site.themeConfig as DefaultTheme.Config
+  console.log('VitePress site config:', vitepress)
+  // console.log('Setting up MCP server for Vue I18n docs...', themeConfig.sidebar)
+  const sidebarDirs = await getSidebarDirNames(docRootDir)
+  console.log('Sidebar:', themeConfig.sidebar, sidebarDirs)
+  // for (const [key, value] of Object.entries(themeConfig.sidebar || {})) {
+  //   console.log(`Sidebar item: ${key}`)
+  //   const items = (themeConfig.sidebar || {})[key] as DefaultTheme.SidebarItem[]
+  //   for (const item of items) {
+  //     console.log(item.text, item)
+  //   }
+  // }
+
+  // @ts-expect-error -- FIXME:
+  mcpServer.resource('contents', 'vue-i18n://docs', {
+    uri: 'vue-i18n://docs',
+    name: 'Vue I18n Documentation top',
+    description: 'Vue I18n documentation root, provides categories and links to documentation pages.',
   }, async (uri) => {
-    const filePath = path.resolve(import.meta.dirname, './dist/llms.txt')
-    const content = await fs.readFile(filePath, 'utf-8')
+    const content = renderMarkdownTop(vitepress.site, getSideBar(sidebarDirs, themeConfig.sidebar || {}))
     return {
       contents: [
         {
@@ -27,5 +43,82 @@ export default function mcp(mcpServer: McpServer, viteServer: ViteDevServer): Aw
       ]
     }
   })
+
+  const pageUri = 'vue-i18n://docs{page}'
+  const pageTempalte = new ResourceTemplate(pageUri, { list: undefined })
+  mcpServer.resource('pages', pageTempalte, {
+    uri: pageUri,
+    name: 'Vue I18n Documentation page',
+    description: 'Vue I18n documentation page content',
+  }, async (uri, params) => {
+    console.log('Fetching page:', uri, params)
+    const p = Array.isArray(params.page) ? params.page[0] : params.page
+    const pagePath = path.resolve(docRootDir, p)
+    const file = await fs.readFile(pagePath, 'utf-8')
+    return {
+      contents: [{
+        uri: uri.href,
+        text: file,
+      }]
+    }
+  })
+
   return mcpServer
 }
+
+function renderMarkdownTop(site: SiteData, sidebar: ReturnType<typeof getSideBar>) {
+  return `# ${site.title}
+
+${site.description}
+
+## Table of Contents
+
+${Object.entries(sidebar).map(([_caegory, items]) => {
+    const buf = [] as string[]
+    items.reduce((acc, item) => {
+      acc.push(`### ${item.text}`, '')
+      const items = (item.items || []) as DefaultTheme.SidebarItem[]
+      const itemBuf = [] as string[]
+      for (const subItem of items) {
+        if (subItem.link) {
+          itemBuf.push(`- [${subItem.text}](${subItem.link.endsWith('.md') ? subItem.link : `${subItem.link}.md`})`)
+        }
+      }
+      itemBuf.push('')
+
+      if (itemBuf.length) {
+        acc.push(itemBuf.join('\n'))
+      }
+      return acc
+    }, buf)
+    return buf.join('\n')
+  }).join('\n\n')
+    }
+`
+}
+
+function getSideBar(sidebarDirs: string[], sideBar: DefaultTheme.Sidebar) {
+  return Object.keys(sideBar).reduce((acc, key) => {
+    const category = key.split('/').filter(Boolean)[0]
+    if (sidebarDirs.includes(category)) {
+      acc[category] = sideBar[key] || []
+    }
+    return acc
+  }, {} as { [key: string]: DefaultTheme.SidebarItem[] })
+}
+
+const EXCLUDE_DIR_NAMES = [
+  '.vitepress',
+  'index.md',
+  'public',
+]
+
+async function getSidebarDirNames(root: string): Promise<string[]> {
+  const dirs = await fs.readdir(root, { withFileTypes: true });
+  return dirs.reduce((acc, dir) => {
+    if (dir.isDirectory() && !EXCLUDE_DIR_NAMES.includes(dir.name)) {
+      acc.push(dir.name)
+    }
+    return acc
+  }, [] as string[])
+}