@dritan/mcp 0.5.0 → 0.6.0

package/README.md

@@ -39,6 +39,8 @@ npm run build && npm start
 ## Tools
 
 - `system_check_prereqs`
+- `auth_status`
+- `auth_set_api_key`
 - `wallet_create_local`
 - `wallet_get_address`
 - `wallet_get_balance`
@@ -79,9 +81,11 @@ npm run build && npm start
 - Wallets default to `~/.config/dritan-mcp/wallets`.
 - Private keys never leave local files; only public address/signature are returned.
 - `swap_sign_and_broadcast` signs locally, then broadcasts via Dritan.
+- `auth_set_api_key` activates a key for the running MCP process without restart.
 - Agent onboarding without `DRITAN_API_KEY` should present two options:
   - Option 1: paid x402 flow (`x402_get_pricing` -> `x402_create_api_key_quote` -> user funds agent wallet -> `wallet_transfer_sol` -> `x402_create_api_key`).
   - Option 2: user gets a free key at `https://dritan.dev`.
+- `x402_create_api_key` auto-activates returned keys for the current MCP session.
 - `token_get_ohlcv_chart` returns a shareable chart URL plus a ready-to-send markdown image snippet.
 - `token_get_ohlcv_chart` supports `chartType: "line-volume" | "candlestick"` (default is `line-volume`).
 - Ticker workflow for chart requests: `token_search` -> extract mint -> `token_get_ohlcv` or `token_get_ohlcv_chart`.

package/dist/index.js

@@ -11,6 +11,41 @@ import { Connection, Keypair, PublicKey, SystemProgram, Transaction, VersionedTr
 import { z } from "zod";
 const DEFAULT_WALLET_DIR = join(homedir(), ".config", "dritan-mcp", "wallets");
 const LAMPORTS_PER_SOL = 1_000_000_000;
+function normalizeApiKey(value) {
+    const trimmed = value?.trim();
+    return trimmed ? trimmed : null;
+}
+function apiKeyPreview(apiKey) {
+    if (!apiKey)
+        return null;
+    if (apiKey.length <= 12)
+        return `${apiKey.slice(0, 4)}...`;
+    return `${apiKey.slice(0, 8)}...${apiKey.slice(-4)}`;
+}
+let runtimeApiKey = normalizeApiKey(process.env.DRITAN_API_KEY);
+let runtimeApiKeySource = runtimeApiKey ? "env" : "none";
+function setRuntimeApiKey(apiKey, source) {
+    const normalized = normalizeApiKey(apiKey);
+    if (!normalized) {
+        throw new Error("apiKey is required");
+    }
+    runtimeApiKey = normalized;
+    runtimeApiKeySource = source;
+    process.env.DRITAN_API_KEY = normalized;
+    return normalized;
+}
+function getActiveApiKey() {
+    if (runtimeApiKey)
+        return runtimeApiKey;
+    const fromEnv = normalizeApiKey(process.env.DRITAN_API_KEY);
+    if (fromEnv) {
+        runtimeApiKey = fromEnv;
+        runtimeApiKeySource = "env";
+        return fromEnv;
+    }
+    runtimeApiKeySource = "none";
+    return null;
+}
 const STREAM_DEXES = [
     "pumpamm",
     "pumpfun",
@@ -35,7 +70,7 @@ const server = new Server({
         "This server supports two API-key onboarding options when DRITAN_API_KEY is missing:",
         "1) x402 pay-per-use key flow: create wallet, receive SOL, create quote, forward payment, claim key.",
         "2) Free key flow: user creates a free API key at https://dritan.dev.",
-        "After key is obtained, set DRITAN_API_KEY and continue with market/swap tools.",
+        "After key is obtained, set it with auth_set_api_key (no restart needed), or restart with DRITAN_API_KEY configured.",
         "Suggested setup command:",
         "  claude mcp add dritan-mcp -e DRITAN_API_KEY=<your-key> -- npx @dritan/mcp@latest",
     ].join("\n"),
@@ -48,10 +83,11 @@ function missingApiKeyError() {
         "Missing DRITAN_API_KEY in environment.",
         "Option 1 (paid): use x402 tools (x402_get_pricing, x402_create_api_key_quote, x402_create_api_key) and wallet tools.",
         "Option 2 (free): create a free key at https://dritan.dev and set DRITAN_API_KEY.",
+        "You can activate a key immediately with auth_set_api_key without restarting MCP.",
     ].join(" "));
 }
 function getDritanClient() {
-    const apiKey = process.env.DRITAN_API_KEY;
+    const apiKey = getActiveApiKey();
     if (!apiKey) {
         throw missingApiKeyError();
     }
@@ -65,7 +101,7 @@ function getDritanClient() {
 function getX402Client() {
     return new DritanClient({
         // x402 endpoints are public; SDK constructor still needs a string.
-        apiKey: process.env.DRITAN_API_KEY ?? "x402_public_endpoints",
+        apiKey: getActiveApiKey() ?? "x402_public_endpoints",
         baseUrl: process.env.DRITAN_BASE_URL,
         controlBaseUrl: getControlBaseUrl(),
         wsBaseUrl: process.env.DRITAN_WS_BASE_URL,
@@ -79,10 +115,10 @@ function getThsClient() {
 async function searchTokens(client, query, options) {
     const sdkSearch = client.searchTokens;
     if (typeof sdkSearch === "function") {
-        return await sdkSearch(query, options);
+        return await sdkSearch.call(client, query, options);
     }
     // Backward-compatible fallback for environments where dritan-sdk hasn't been upgraded yet.
-    const apiKey = process.env.DRITAN_API_KEY;
+    const apiKey = getActiveApiKey();
     if (!apiKey) {
         throw missingApiKeyError();
     }
@@ -412,6 +448,9 @@ const walletCreateSchema = z.object({
     name: z.string().min(1).default("agent-wallet"),
     walletDir: z.string().min(1).optional(),
 });
+const authSetApiKeySchema = z.object({
+    apiKey: z.string().min(8),
+});
 const walletPathSchema = z.object({
     walletPath: z.string().min(1),
 });
@@ -531,6 +570,25 @@ const tools = [
             properties: {},
         },
     },
+    {
+        name: "auth_status",
+        description: "Show current API key status for this MCP session (active source, preview, and onboarding options).",
+        inputSchema: {
+            type: "object",
+            properties: {},
+        },
+    },
+    {
+        name: "auth_set_api_key",
+        description: "Set the active Dritan API key for this running MCP process without restart (runtime session scope).",
+        inputSchema: {
+            type: "object",
+            required: ["apiKey"],
+            properties: {
+                apiKey: { type: "string" },
+            },
+        },
+    },
     {
         name: "wallet_create_local",
         description: "Create a local Solana wallet using solana-keygen and return path + public address.",
@@ -976,7 +1034,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
         switch (request.params.name) {
             case "system_check_prereqs": {
                 const solanaCli = checkSolanaCli();
-                const apiKeySet = !!process.env.DRITAN_API_KEY;
+                const activeApiKey = getActiveApiKey();
+                const apiKeySet = !!activeApiKey;
                 return ok({
                     ready: solanaCli.ok && apiKeySet,
                     readyForX402Onboarding: solanaCli.ok,
@@ -985,6 +1044,8 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                         {
                             ok: apiKeySet,
                             name: "DRITAN_API_KEY",
+                            source: runtimeApiKeySource,
+                            preview: apiKeyPreview(activeApiKey),
                             hint: apiKeySet
                                 ? "API key is configured."
                                 : "Missing DRITAN_API_KEY. You can either use x402 onboarding tools or get a free key at https://dritan.dev.",
@@ -997,6 +1058,29 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                             : "Environment ready.",
                 });
             }
+            case "auth_status": {
+                const activeApiKey = getActiveApiKey();
+                return ok({
+                    apiKeyConfigured: !!activeApiKey,
+                    source: runtimeApiKeySource,
+                    preview: apiKeyPreview(activeApiKey),
+                    controlBaseUrl: getControlBaseUrl(),
+                    onboardingOptions: [
+                        "Option 1 (paid x402): create wallet -> receive user SOL -> x402 quote -> transfer -> claim key.",
+                        "Option 2 (free): create key at https://dritan.dev and set it with auth_set_api_key.",
+                    ],
+                });
+            }
+            case "auth_set_api_key": {
+                const input = authSetApiKeySchema.parse(args);
+                const activated = setRuntimeApiKey(input.apiKey, "runtime");
+                return ok({
+                    ok: true,
+                    message: "API key activated for this MCP session without restart.",
+                    source: runtimeApiKeySource,
+                    preview: apiKeyPreview(activated),
+                });
+            }
             case "dritan_health": {
                 return ok(await checkDritanHealth());
             }
@@ -1090,7 +1174,7 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                         "If user picked paid flow, ensure agent has a local wallet (wallet_create_local + wallet_get_address).",
                         "User funds the agent wallet.",
                         "Transfer quoted SOL amount to receiver wallet using wallet_transfer_sol.",
-                        "Claim key with x402_create_api_key using returned tx signature.",
+                        "Claim key with x402_create_api_key using returned tx signature (MCP auto-activates returned apiKey).",
                     ],
                 });
             }
@@ -1098,7 +1182,19 @@ server.setRequestHandler(CallToolRequestSchema, async (request) => {
                 const input = x402CreateApiKeySchema.parse(args);
                 const client = getX402Client();
                 const created = await x402CreateApiKey(client, input);
-                return ok(created);
+                const payload = typeof created === "object" && created !== null
+                    ? { ...created }
+                    : { value: created };
+                if (typeof payload.apiKey === "string") {
+                    const activated = setRuntimeApiKey(payload.apiKey, "x402");
+                    payload.mcpAuth = {
+                        activated: true,
+                        source: runtimeApiKeySource,
+                        preview: apiKeyPreview(activated),
+                        message: "x402-created API key is active for this MCP session (no restart needed).",
+                    };
+                }
+                return ok(payload);
             }
             case "market_get_snapshot": {
                 const input = marketSnapshotSchema.parse(args);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@dritan/mcp",
-  "version": "0.5.0",
+  "version": "0.6.0",
   "license": "MIT",
   "type": "module",
   "description": "MCP server for Dritan SDK market data and local Solana wallet swap execution",