Add model context server

Author: mantou132

crates/zed-plugin-gem/extension.toml

@@ -11,3 +11,6 @@ snippets = "./snippets/typescript.json"
 name = "Gem language server"
 languages = ["TypeScript", "TSX", "JavaScript", "JSDoc"]
 language_ids = { "TypeScript" = "typescript", "TSX" = "typescriptreact", "JavaScript" = "javascript" }
+
+[context_servers.gem]
+name = "Gem Context Server"

crates/zed-plugin-gem/src/lib.rs

@@ -7,59 +7,62 @@
 use std::{env, fs};
 
 use zed::settings::LspSettings;
-use zed_extension_api::{self as zed, LanguageServerId, Result};
+use zed_extension_api::{self as zed, Command, ContextServerId, LanguageServerId, Project, Result};
 
-const NPM_PKG_NAME: &str = "vscode-gem-languageservice";
+const LS_PKG_NAME: &str = "vscode-gem-languageservice";
 const LS_BIN_PATH: &str = "node_modules/.bin/vscode-gem-languageservice";
 
+const MC_PKG_NAME: &str = "gem-context-server";
+const MC_BIN_PATH: &str = "node_modules/.bin/gem-context-server";
+
 #[derive(Default)]
 struct GemExtension {
-    did_find_server: bool,
+    ls_server_find: bool,
 }
 
 impl GemExtension {
-    fn server_exists(&self) -> bool {
+    fn ls_exists(&self) -> bool {
         fs::metadata(LS_BIN_PATH).is_ok_and(|stat| stat.is_file())
     }
 
-    fn server_script_path(&mut self, language_server_id: &zed::LanguageServerId) -> Result<String> {
-        let server_exists = self.server_exists();
-        if self.did_find_server && server_exists {
+    fn ls_path(&mut self, language_server_id: &LanguageServerId) -> Result<String> {
+        let server_exists = self.ls_exists();
+        if self.ls_server_find && server_exists {
             return Ok(LS_BIN_PATH.to_string());
         }
 
         zed::set_language_server_installation_status(
             language_server_id,
             &zed::LanguageServerInstallationStatus::CheckingForUpdate,
         );
-        let version = zed::npm_package_latest_version(NPM_PKG_NAME)?;
+        let version = zed::npm_package_latest_version(LS_PKG_NAME)?;
 
         if !server_exists
-            || zed::npm_package_installed_version(NPM_PKG_NAME)?.as_ref() != Some(&version)
+            || zed::npm_package_installed_version(LS_PKG_NAME)?.as_ref() != Some(&version)
         {
             zed::set_language_server_installation_status(
                 language_server_id,
                 &zed::LanguageServerInstallationStatus::Downloading,
             );
-            let result = zed::npm_install_package(NPM_PKG_NAME, &version);
+            let result = zed::npm_install_package(LS_PKG_NAME, &version);
             match result {
                 Ok(()) => {
-                    if !self.server_exists() {
+                    if !self.ls_exists() {
                         Err(format!(
-                            "installed package '{NPM_PKG_NAME}' did not contain expected path \
+                            "installed package '{LS_PKG_NAME}' did not contain expected path \
                              '{LS_BIN_PATH}'",
                         ))?;
                     }
                 }
                 Err(error) => {
-                    if !self.server_exists() {
+                    if !self.ls_exists() {
                         Err(error)?;
                     }
                 }
             }
         }
 
-        self.did_find_server = true;
+        self.ls_server_find = true;
         Ok(LS_BIN_PATH.to_string())
     }
 }
@@ -69,12 +72,33 @@ impl zed::Extension for GemExtension {
         Self::default()
     }
 
+    fn context_server_command(
+        &mut self,
+        _context_server_id: &ContextServerId,
+        _project: &Project,
+    ) -> Result<Command> {
+        let version = zed::npm_package_latest_version(MC_PKG_NAME)?;
+        if zed::npm_package_installed_version(MC_PKG_NAME)?.as_ref() != Some(&version) {
+            zed::npm_install_package(MC_PKG_NAME, &version)?;
+        }
+
+        Ok(Command {
+            command: zed::node_binary_path()?,
+            args: vec![env::current_dir()
+                .unwrap()
+                .join(MC_BIN_PATH)
+                .to_string_lossy()
+                .to_string()],
+            env: Default::default(),
+        })
+    }
+
     fn language_server_command(
         &mut self,
         language_server_id: &LanguageServerId,
         _worktree: &zed::Worktree,
     ) -> Result<zed::Command> {
-        let server_path = self.server_script_path(language_server_id)?;
+        let server_path = self.ls_path(language_server_id)?;
         Ok(zed::Command {
             command: zed::node_binary_path()?,
             args: vec![

cspell.json

@@ -84,6 +84,7 @@
     "Microtask",
     "minisearch",
     "minmax",
+    "modelcontextprotocol",
     "mooncake",
     "mtapp",
     "multiselectable",

packages/context-server/README.md

@@ -0,0 +1,3 @@
+# Gem model context server
+
+Use for Zed & Cursor

packages/context-server/package.json

@@ -0,0 +1,37 @@
+{
+  "name": "gem-context-server",
+  "version": "0.0.1",
+  "description": "Model context server for Gem",
+  "keywords": [
+    "gem",
+    "MCP"
+  ],
+  "module": "src/index.ts",
+  "main": "src/index.ts",
+  "bin": "dist/index.js",
+  "files": [
+    "/dist/"
+  ],
+  "scripts": {
+    "prepublishOnly": "esbuild ./src/index.ts --outdir=./dist --platform=node --sourcemap --bundle --packages=external"
+  },
+  "dependencies": {
+    "adm-zip": "^0.5.16",
+    "@modelcontextprotocol/sdk": "^1.6.1"
+  },
+  "devDependencies": {
+    "@gemjs/config": "^2.1.0",
+    "@types/adm-zip": "^0.5.7"
+  },
+  "author": "mantou132",
+  "license": "MIT",
+  "repository": {
+    "type": "git",
+    "url": "git+https://github.com/mantou132/gem.git",
+    "directory": "packages/context-server"
+  },
+  "bugs": {
+    "url": "https://github.com/mantou132/gem/issues"
+  },
+  "homepage": "https://github.com/mantou132/gem#readme"
+}

packages/context-server/src/index.ts

@@ -0,0 +1,32 @@
+import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
+import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js';
+import AdmZip from 'adm-zip';
+
+async function downloadDocs() {
+  const res = await fetch('https://github.com/mantou132/gem/archive/refs/heads/main.zip');
+  const buf = await res.arrayBuffer();
+  const zip = new AdmZip(Buffer.from(buf));
+  const zipEntries = zip.getEntries();
+  return zipEntries
+    .filter((entry) => entry.entryName.includes('packages/gem/docs/en') && entry.entryName.endsWith('.md'))
+    .map((e) => ({
+      entryName: e.entryName,
+      content: zip.readAsText(e),
+    }));
+}
+
+const server = new McpServer({
+  name: 'Gem',
+  version: '0.0.1',
+});
+
+let docsCache: Awaited<ReturnType<typeof downloadDocs>> | undefined;
+server.resource('docs', 'docs://all', async () => {
+  const docs = docsCache || (docsCache = await downloadDocs());
+  return {
+    contents: docs.map(({ entryName, content }) => ({ uri: `docs://${entryName}`, text: content })),
+  };
+});
+
+const transport = new StdioServerTransport();
+server.connect(transport);

packages/context-server/tsconfig.json

@@ -0,0 +1,5 @@
+{
+  "extends": "@gemjs/config/tsconfig",
+  "include": ["src"],
+  "exclude": []
+}