@uzqw/mcp 0.1.0 → 0.1.2

package/dist/index.js

@@ -8,7 +8,7 @@ import { writeFileSync } from "fs";
 import { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
 import { StdioServerTransport } from "@modelcontextprotocol/sdk/server/stdio.js";
 import open from "open";
-import { RecolxClient, runOAuthCallback, LoginInput, LogoutInput } from "@uzqw/shared";
+import { RecolxClient, runOAuthCallback, LoginInput, LogoutInput, SetEnvInput, GetEnvInput } from "@uzqw/shared";
 
 // src/tools.ts
 import {
@@ -237,11 +237,12 @@ var CALLBACK_PORT = 8199;
 var LOGIN_TIMEOUT_MS = 12e4;
 var sub = process.argv[2];
 if (sub === "install" || sub === "uninstall") {
-  const { runInstall, runUninstall } = await import("./install-VFWABBBV.js");
+  const { runInstall, runUninstall, parseEnvFlag } = await import("./install-D7FNIM42.js");
   const args = process.argv.slice(3);
   const yes = args.includes("--yes") || args.includes("-y");
   const noLogin = args.includes("--no-login");
-  if (sub === "install") await runInstall({ yes, noLogin });
+  const env = parseEnvFlag(args);
+  if (sub === "install") await runInstall({ yes, noLogin, env: env ?? void 0 });
   else await runUninstall({ yes });
   process.exit(0);
 }
@@ -252,6 +253,7 @@ if (sub === "--help" || sub === "-h") {
   recolx-mcp                  \u4EE5 stdio \u542F\u52A8 MCP server\uFF08Claude/Codex/Cursor/Zed \u7B49\u5BA2\u6237\u7AEF\u9ED8\u8BA4\u8DEF\u5F84\uFF09
   recolx-mcp install          \u68C0\u6D4B\u5E76\u5199\u5165\u672C\u5730\u5BA2\u6237\u7AEF\u914D\u7F6E\uFF08Claude Desktop / Codex / Claude Code / Cursor / Windsurf / VS Code / Zed\uFF09
   recolx-mcp install --yes    \u5168\u81EA\u52A8\uFF1A\u4E0D\u8BE2\u95EE\uFF0C\u76F4\u63A5\u914D\u7F6E\u6240\u6709\u68C0\u6D4B\u5230\u7684\u5BA2\u6237\u7AEF
+  recolx-mcp install --env local|staging|prod  \u5207\u6362\u73AF\u5883\uFF08\u5199 ~/.recolx/config.json\uFF0C\u9ED8\u8BA4 prod\uFF09
   recolx-mcp install --no-login  \u8DF3\u8FC7 OAuth \u767B\u5F55\uFF08CI / \u8DF3\u677F\u673A\uFF09
   recolx-mcp uninstall        \u79FB\u9664\u5404\u5BA2\u6237\u7AEF\u91CC\u7684 recolx MCP \u914D\u7F6E
 `);
@@ -263,7 +265,7 @@ var READ_ONLY2 = { readOnlyHint: true, destructiveHint: false, openWorldHint: tr
 server.registerTool(
   "login",
   {
-    description: "\u767B\u5F55 Recolx\uFF08OAuth PKCE\uFF09\uFF1A\u6253\u5F00\u6D4F\u89C8\u5668\u8DF3\u8F6C\u6388\u6743\u9875\uFF0C\u540C\u610F\u540E\u5728\u672C\u5730 8199 \u56DE\u8C03\u6536\u53D6 token \u5E76\u5199\u5165 ~/.recolx/tokens-mcp.json\uFF1B\u5DF2\u767B\u5F55\u65F6\u76F4\u63A5\u8FD4\u56DE",
+    description: "\u767B\u5F55 Recolx\uFF08OAuth PKCE\uFF09\uFF1A\u6253\u5F00\u6D4F\u89C8\u5668\u8DF3\u8F6C\u6388\u6743\u9875\uFF0C\u540C\u610F\u540E\u5728\u672C\u5730 8199 \u56DE\u8C03\u6536\u53D6 token \u5E76\u5199\u5165 ~/.recolx/tokens-mcp-{env}.json\uFF08\u6BCF\u73AF\u5883\u72EC\u7ACB\uFF09\uFF1B\u5DF2\u767B\u5F55\u65F6\u76F4\u63A5\u8FD4\u56DE",
     annotations: { title: "\u767B\u5F55 Recolx", ...READ_ONLY2 },
     inputSchema: LoginInput
   },
@@ -362,6 +364,42 @@ server.registerTool(
   }
 );
 registerTools(server, client);
+var ENV_TOOL = { destructiveHint: false, openWorldHint: true };
+server.registerTool(
+  "set_env",
+  {
+    description: "\u5207\u6362 Recolx \u73AF\u5883\uFF08local \u672C\u5730\u6D4B\u8BD5 / staging \u7EBF\u4E0A\u6D4B\u8BD5 / prod \u7EBF\u4E0A\u6B63\u5F0F\uFF09\uFF1A\u5199 ~/.recolx/config.json \u5E76\u5373\u65F6\u751F\u6548\uFF1B\u8FD4\u56DE\u65B0\u73AF\u5883\u3001\u540E\u7AEF\u5730\u5740\u4E0E\u767B\u5F55\u6001",
+    annotations: { title: "\u5207\u6362\u73AF\u5883", ...ENV_TOOL },
+    inputSchema: SetEnvInput
+  },
+  async ({ env }) => {
+    const state = await client.setEnv(env);
+    const login = state.logged_in ? "\u5DF2\u767B\u5F55" : "\u672A\u767B\u5F55\uFF08\u8BF7\u8FD0\u884C login\uFF09";
+    return { content: [{ type: "text", text: `\u5DF2\u5207\u6362\u5230 ${state.env}\uFF08${state.api_base}\uFF09\u3002${login}` }] };
+  }
+);
+server.registerTool(
+  "get_env",
+  {
+    description: "\u67E5\u770B\u5F53\u524D Recolx \u73AF\u5883\uFF08local/staging/prod\uFF09\u3001\u540E\u7AEF\u5730\u5740\u3001\u9274\u6743\u6A21\u5F0F\u4E0E\u767B\u5F55\u6001",
+    annotations: { title: "\u67E5\u770B\u73AF\u5883", ...READ_ONLY2 },
+    inputSchema: GetEnvInput
+  },
+  async () => {
+    const state = await client.getEnvState();
+    return {
+      content: [
+        {
+          type: "text",
+          text: `\u73AF\u5883\uFF1A${state.env}
+\u540E\u7AEF\uFF1A${state.api_base}
+\u9274\u6743\uFF1A${state.auth_mode}
+\u767B\u5F55\uFF1A${state.logged_in ? "\u5DF2\u767B\u5F55" : "\u672A\u767B\u5F55"}`
+        }
+      ]
+    };
+  }
+);
 try {
   const skills = await loadSkills();
   for (const skill of skills) {

package/dist/{install-VFWABBBV.js → install-D7FNIM42.js}

@@ -12,10 +12,16 @@ import { dirname, join } from "path";
 import { spawnSync } from "child_process";
 import { createInterface } from "readline/promises";
 import open from "open";
-import { RecolxClient, runOAuthCallback } from "@uzqw/shared";
+import { RecolxClient, writeEnvConfig, runOAuthCallback } from "@uzqw/shared";
 var SERVER_NAME = "recolx";
 var CALLBACK_PORT = 8199;
 var LOGIN_TIMEOUT_MS = 12e4;
+function parseEnvFlag(args) {
+  const i = args.indexOf("--env");
+  if (i === -1) return null;
+  const v = args[i + 1];
+  return v === "local" || v === "staging" || v === "prod" ? v : null;
+}
 function findNodeBinCommand(command) {
   const commandName = platform() === "win32" ? `${command}.cmd` : command;
   const sibling = join(dirname(process.execPath), commandName);
@@ -30,10 +36,9 @@ function isLocalPackageSpec(spec) {
 }
 function getMcpEntry() {
   const packageSpec = process.env.RECOLX_MCP_PACKAGE_SPEC?.trim() || "@uzqw/mcp@latest";
-  if (isLocalPackageSpec(packageSpec)) {
-    return { command: findNodeBinCommand("npm"), args: ["exec", "--yes", "--package", packageSpec, "--", "recolx-mcp"] };
-  }
-  return { command: findNodeBinCommand("npx"), args: ["-y", packageSpec] };
+  const base = isLocalPackageSpec(packageSpec) ? { command: findNodeBinCommand("npm"), args: ["exec", "--yes", "--package", packageSpec, "--", "recolx-mcp"] } : { command: findNodeBinCommand("npx"), args: ["-y", packageSpec] };
+  const env = process.env.RECOLX_API_BASE?.trim() ? { RECOLX_API_BASE: process.env.RECOLX_API_BASE.trim().replace(/\/+$/, "") } : {};
+  return { ...base, env };
 }
 function commandPathIsStale(command) {
   if (!command) return false;
@@ -64,6 +69,9 @@ function configPaths() {
     claudeCodeDir: join(h, ".claude")
   };
 }
+function entryEnv(e) {
+  return e.env && Object.keys(e.env).length > 0 ? e.env : void 0;
+}
 function isJsonObject(v) {
   return typeof v === "object" && v !== null && !Array.isArray(v);
 }
@@ -75,9 +83,18 @@ function entryAutoUpdates(entry) {
   const args = Array.isArray(entry.args) ? entry.args : [];
   return args.some((a) => typeof a === "string" && a.includes("@uzqw/mcp@latest"));
 }
+function entryEnvMatches(entry, expected) {
+  if (!isJsonObject(entry)) return false;
+  const env = isJsonObject(entry.env) ? entry.env : {};
+  if (!expected) return true;
+  return Object.entries(expected).every(([k, v]) => env[k] === v);
+}
 function jsonAdapter(spec) {
   const rootKey = spec.rootKey ?? "mcpServers";
-  const toEntry = spec.toEntry ?? ((e) => ({ command: e.command, args: e.args }));
+  const toEntry = spec.toEntry ?? ((e) => {
+    const env = entryEnv(e);
+    return env ? { command: e.command, args: e.args, env } : { command: e.command, args: e.args };
+  });
   async function load() {
     try {
       const parsed = JSON.parse(await readFile(spec.configPath, "utf8"));
@@ -106,7 +123,7 @@ function jsonAdapter(spec) {
       const existing = servers[SERVER_NAME];
       if (existing && entryAutoUpdates(existing)) {
         const cmd = isJsonObject(existing) && typeof existing.command === "string" ? existing.command : void 0;
-        if (!commandPathIsStale(cmd)) {
+        if (!commandPathIsStale(cmd) && entryEnvMatches(existing, entryEnv(ctx.entry))) {
           return { status: "already-configured", message: "\u5DF2\u914D\u7F6E\uFF08npx @latest \u81EA\u52A8\u66F4\u65B0\uFF09", restartHint: spec.restartHint };
         }
       }
@@ -136,9 +153,25 @@ function codexBlockText(entry) {
   const win = platform() === "win32";
   const command = win ? normalizeWindowsPath(entry.command, "win32") : entry.command;
   const args = (win ? entry.args.map((a) => normalizeWindowsPath(a, "win32")) : entry.args).map((a) => JSON.stringify(a)).join(", ");
+  const env = entryEnv(entry);
+  const envLine = env ? `
+env = { ${Object.entries(env).map(([k, v]) => `${JSON.stringify(k)} = ${JSON.stringify(v)}`).join(", ")} }` : "";
   return `${CODEX_HEADER}
 command = ${JSON.stringify(command)}
-args = [${args}]`;
+args = [${args}]${envLine}`;
+}
+function extractTomlEnv(lines) {
+  for (const line of lines) {
+    const m = line.match(/^\s*env\s*=\s*\{(.*)\}\s*$/);
+    if (!m) continue;
+    const env = {};
+    for (const pair of m[1].split(",")) {
+      const kv = pair.match(/"([^"]+)"\s*=\s*"([^"]*)"/);
+      if (kv) env[kv[1]] = kv[2];
+    }
+    return env;
+  }
+  return void 0;
 }
 function codexBlockRange(lines) {
   const start = lines.findIndex((l) => l.trim() === CODEX_HEADER);
@@ -184,13 +217,13 @@ function codexAdapter(configPath) {
       const block = codexBlockText(ctx.entry);
       if (range) {
         const existingCommand = extractTomlCommand(lines.slice(range.start, range.end));
-        if (!commandPathIsStale(existingCommand)) {
+        if (!commandPathIsStale(existingCommand) && entryEnvMatches({ env: extractTomlEnv(lines.slice(range.start, range.end)) }, entryEnv(ctx.entry))) {
           return { status: "already-configured", message: "\u5DF2\u914D\u7F6E\uFF08npx @latest \u81EA\u52A8\u66F4\u65B0\uFF09" };
         }
         const rebuilt = [...lines.slice(0, range.start), ...block.split("\n"), ...lines.slice(range.end)].join("\n");
         await mkdir(dirname(configPath), { recursive: true });
         await writeFile(configPath, rebuilt, "utf8");
-        return { status: "configured", message: "\u5DF2\u4FEE\u590D\u5931\u6548\u542F\u52A8\u8DEF\u5F84\uFF08npx @latest\uFF09" };
+        return { status: "configured", message: "\u5DF2\u4FEE\u590D\u542F\u52A8\u8DEF\u5F84/env\uFF08npx @latest\uFF09" };
       }
       await mkdir(dirname(configPath), { recursive: true });
       await writeFile(configPath, `${content}${content && !content.endsWith("\n") ? "\n" : ""}
@@ -297,13 +330,16 @@ function zedAdapter(configPath) {
       const existing = servers[SERVER_NAME];
       if (existing && entryAutoUpdates(existing)) {
         const path = isJsonObject(existing) && isJsonObject(existing.command) && typeof existing.command.path === "string" ? existing.command.path : void 0;
-        if (!commandPathIsStale(path)) {
+        const existingCommand = isJsonObject(existing) && isJsonObject(existing.command) ? existing.command : void 0;
+        if (!commandPathIsStale(path) && entryEnvMatches(existingCommand, entryEnv(ctx.entry))) {
           return { status: "already-configured", message: "\u5DF2\u914D\u7F6E\uFF08npx @latest \u81EA\u52A8\u66F4\u65B0\uFF09" };
         }
       }
+      const env = entryEnv(ctx.entry);
+      const commandObj = env ? { path: ctx.entry.command, args: ctx.entry.args, env } : { path: ctx.entry.command, args: ctx.entry.args };
       config.context_servers = {
         ...servers,
-        [SERVER_NAME]: { command: { path: ctx.entry.command, args: ctx.entry.args } }
+        [SERVER_NAME]: { command: commandObj }
       };
       await mkdir(dirname(configPath), { recursive: true });
       await writeFile(configPath, JSON.stringify(config, null, 2) + "\n", "utf8");
@@ -370,7 +406,10 @@ function detectClients() {
       label: "VS Code",
       configPath: p.vsCode,
       rootKey: "servers",
-      toEntry: (e) => ({ type: "stdio", command: e.command, args: e.args }),
+      toEntry: (e) => {
+        const env = entryEnv(e);
+        return env ? { type: "stdio", command: e.command, args: e.args, env } : { type: "stdio", command: e.command, args: e.args };
+      },
       configuredMessage: "\u5DF2\u5199\u5165 .vscode/mcp.json",
       restartHint: RESTART_HINTS.vscode
     }),
@@ -523,6 +562,11 @@ function printNextSteps(clients, login) {
 }
 async function runInstall(opts = {}) {
   console.log("Recolx MCP installer\n");
+  if (opts.env) {
+    writeEnvConfig(opts.env);
+    console.log(`\u2192 \u73AF\u5883\u5DF2\u8BBE\u4E3A ${opts.env}\uFF08~/.recolx/config.json\uFF09
+`);
+  }
   const clients = detectClients();
   console.log("\u68C0\u6D4B\u5230\u7684 AI \u5BA2\u6237\u7AEF\uFF1A");
   for (const c of clients) {
@@ -594,7 +638,7 @@ async function runUninstall(opts = {}) {
     }
   }
   console.log();
-  console.log("\u5DF2\u6E05\u7406\u5BA2\u6237\u7AEF\u914D\u7F6E\u3002token \u6587\u4EF6 ~/.recolx/tokens-mcp.json \u4FDD\u7559\uFF08\u5982\u9700\u5F7B\u5E95\u79FB\u9664\u8BF7\u5220\u9664\u8BE5\u6587\u4EF6\uFF09\u3002");
+  console.log("\u5DF2\u6E05\u7406\u5BA2\u6237\u7AEF\u914D\u7F6E\u3002token \u6587\u4EF6 ~/.recolx/tokens-mcp-*.json \u4FDD\u7559\uFF08\u5982\u9700\u5F7B\u5E95\u79FB\u9664\u8BF7\u5220\u9664\u8BE5\u76EE\u5F55\uFF09\u3002");
 }
 export {
   CALLBACK_PORT,
@@ -604,8 +648,10 @@ export {
   codexBlockRange,
   codexBlockText,
   commandPathIsStale,
+  extractTomlEnv,
   getMcpEntry,
   normalizeWindowsPath,
+  parseEnvFlag,
   runInstall,
   runUninstall
 };

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@uzqw/mcp",
-  "version": "0.1.0",
+  "version": "0.1.2",
   "description": "Recolx read-only MCP server (stdio) — 7 read-only tools, readOnlyHint: true",
   "mcpName": "io.github.uzqw/recolx-mcp",
   "publishConfig": {
@@ -8,7 +8,7 @@
   },
   "type": "module",
   "bin": {
-    "recolx-mcp": "./dist/index.js"
+    "recolx-mcp": "dist/index.js"
   },
   "main": "./dist/index.js",
   "types": "./dist/index.d.ts",
@@ -18,7 +18,10 @@
       "import": "./dist/index.js"
     }
   },
-  "files": ["dist", "skills"],
+  "files": [
+    "dist",
+    "skills"
+  ],
   "scripts": {
     "build": "tsup src/index.ts --format esm --dts --clean",
     "test": "tsx --test test/*.test.ts"