@infimind-next/ai-hive-mcp 0.2.1 → 0.2.4

package/dist/api-client.d.ts

@@ -1,16 +1,19 @@
 import type { BearerTokenProvider } from "./auth-token-provider.js";
+import { type McpClientInfo } from "./client-info.js";
 type FetchImpl = typeof fetch;
 type AiHiveApiClientInput = {
     endpoint: string;
     tokenProvider: BearerTokenProvider;
     timeoutMs: number;
     fetchImpl?: FetchImpl;
+    clientInfoProvider?: () => McpClientInfo | undefined;
 };
 export declare class AiHiveApiClient {
     private readonly endpoint;
     private readonly tokenProvider;
     private readonly timeoutMs;
     private readonly fetchImpl;
+    private readonly clientInfoProvider?;
     constructor(input: AiHiveApiClientInput);
     private withTimeout;
     private request;

package/dist/api-client.js

@@ -1,3 +1,4 @@
+import { encodeMcpClientInfo, MCP_CLIENT_INFO_HEADER, } from "./client-info.js";
 function formatIssue(value) {
     if (typeof value === "string") {
         return value;
@@ -46,11 +47,13 @@ export class AiHiveApiClient {
     tokenProvider;
     timeoutMs;
     fetchImpl;
+    clientInfoProvider;
     constructor(input) {
         this.endpoint = input.endpoint.replace(/\/+$/, "");
         this.tokenProvider = input.tokenProvider;
         this.timeoutMs = input.timeoutMs;
         this.fetchImpl = input.fetchImpl ?? fetch;
+        this.clientInfoProvider = input.clientInfoProvider;
     }
     async withTimeout(operation) {
         const controller = new AbortController();
@@ -99,11 +102,15 @@ export class AiHiveApiClient {
         });
     }
     send(path, init, signal, token) {
+        const encodedClientInfo = encodeMcpClientInfo(this.clientInfoProvider?.());
         return this.fetchImpl(`${this.endpoint}${path}`, {
             ...init,
             signal,
             headers: {
                 Authorization: `Bearer ${token}`,
+                ...(encodedClientInfo
+                    ? { [MCP_CLIENT_INFO_HEADER]: encodedClientInfo }
+                    : {}),
                 ...(init.body ? { "Content-Type": "application/json" } : {}),
                 ...init.headers,
             },

package/dist/cli.d.ts

@@ -2,7 +2,7 @@ import type { McpClientConfig } from "./config.js";
 import { runServer as runMcpServer } from "./server.js";
 import { WorkBuddyCredentialStore } from "./workbuddy-credentials.js";
 import { WorkBuddyOAuthClient } from "./workbuddy-oauth-client.js";
-import { listenForWorkBuddyCallback } from "./workbuddy-loopback.js";
+import { listenForWorkBuddyCallback, type WorkBuddyResultRedirects } from "./workbuddy-loopback.js";
 export type CliDependencies = {
     stdout(line: string): void;
     stderr(line: string): void;
@@ -12,3 +12,4 @@ export type CliDependencies = {
     runServer: typeof runMcpServer;
 };
 export declare function runCli(config: McpClientConfig, dependencies?: Partial<CliDependencies>): Promise<number>;
+export declare function createWorkBuddyResultRedirects(webOrigin: string): WorkBuddyResultRedirects;

package/dist/cli.js

@@ -49,6 +49,14 @@ function resolveDependencies(config, overrides) {
     };
 }
 async function runAuth(config, dependencies) {
+    let resultRedirects;
+    try {
+        resultRedirects = createWorkBuddyResultRedirects(config.webOrigin);
+    }
+    catch {
+        dependencies.stderr("连接 AI-HIVE 失败，请重试");
+        return 1;
+    }
     const store = dependencies.createStore(config.credentialsFile);
     const oauth = dependencies.createOAuthClient(config.endpoint);
     const { state, codeVerifier, codeChallenge } = createPkcePair();
@@ -58,6 +66,7 @@ async function runAuth(config, dependencies) {
         listenerPromise = Promise.resolve(dependencies.listenForCallback({
             expectedState: state,
             timeoutMs: AUTH_CALLBACK_DEADLINE_MS,
+            resultRedirects,
             onCode: async (code) => {
                 const payload = await oauth.exchangeAuthorizationCode({
                     code,
@@ -112,6 +121,26 @@ async function runAuth(config, dependencies) {
         await listener.close().catch(() => undefined);
     }
 }
+export function createWorkBuddyResultRedirects(webOrigin) {
+    const baseUrl = new URL(webOrigin);
+    if ((baseUrl.protocol !== "http:" && baseUrl.protocol !== "https:") ||
+        baseUrl.username !== "" ||
+        baseUrl.password !== "" ||
+        baseUrl.search !== "" ||
+        baseUrl.hash !== "") {
+        throw new Error("Web origin 无效");
+    }
+    const successUrl = new URL("/welcome?from=workbuddy-auth", baseUrl);
+    const failureUrl = new URL("/authorization-failed?from=workbuddy-auth&reason=callback-error", baseUrl);
+    if (successUrl.origin !== baseUrl.origin ||
+        failureUrl.origin !== baseUrl.origin) {
+        throw new Error("授权结果地址不同源");
+    }
+    return {
+        successUrl: successUrl.href,
+        failureUrl: failureUrl.href,
+    };
+}
 async function runStatus(config, dependencies) {
     const store = dependencies.createStore(config.credentialsFile);
     const deadlineAt = Date.now() + STATUS_DEADLINE_MS;

package/dist/client-info.d.ts

@@ -0,0 +1,9 @@
+import type { McpServer } from "@modelcontextprotocol/sdk/server/mcp.js";
+export declare const MCP_CLIENT_INFO_HEADER = "x-ai-hive-mcp-client-info";
+export declare const MCP_CLIENT_INFO_HEADER_MAX_BYTES = 1024;
+export type McpClientInfo = {
+    name: string;
+    version: string;
+};
+export declare function encodeMcpClientInfo(input?: McpClientInfo): string | undefined;
+export declare function readMcpClientInfo(server: McpServer): McpClientInfo | undefined;

package/dist/client-info.js

@@ -0,0 +1,16 @@
+export const MCP_CLIENT_INFO_HEADER = "x-ai-hive-mcp-client-info";
+export const MCP_CLIENT_INFO_HEADER_MAX_BYTES = 1024;
+export function encodeMcpClientInfo(input) {
+    if (!input) {
+        return undefined;
+    }
+    const encoded = Buffer.from(JSON.stringify({ name: input.name, version: input.version }), "utf8").toString("base64url");
+    return Buffer.byteLength(encoded, "ascii") <=
+        MCP_CLIENT_INFO_HEADER_MAX_BYTES
+        ? encoded
+        : undefined;
+}
+export function readMcpClientInfo(server) {
+    const info = server.server.getClientVersion();
+    return info ? { name: info.name, version: info.version } : undefined;
+}

package/dist/schemas.d.ts

@@ -10,34 +10,34 @@ export declare const mcpCreateMediaUploadOutputSchema: z.ZodObject<{
         headers: z.ZodRecord<z.ZodString, z.ZodString>;
     }, "strict", z.ZodTypeAny, {
         method: "PUT";
-        headers: Record<string, string>;
         url: string;
+        headers: Record<string, string>;
     }, {
         method: "PUT";
-        headers: Record<string, string>;
         url: string;
+        headers: Record<string, string>;
     }>;
     expiresAt: z.ZodString;
 }, "strict", z.ZodTypeAny, {
+    mimeType: string;
     mediaId: string;
     mediaType: "IMAGE" | "VIDEO" | "DOCUMENT";
-    mimeType: string;
     sizeBytes: number;
     upload: {
         method: "PUT";
-        headers: Record<string, string>;
         url: string;
+        headers: Record<string, string>;
     };
     expiresAt: string;
 }, {
+    mimeType: string;
     mediaId: string;
     mediaType: "IMAGE" | "VIDEO" | "DOCUMENT";
-    mimeType: string;
     sizeBytes: number;
     upload: {
         method: "PUT";
-        headers: Record<string, string>;
         url: string;
+        headers: Record<string, string>;
     };
     expiresAt: string;
 }>;
@@ -65,12 +65,12 @@ export declare const chatTextShape: {
         content: z.ZodEffects<z.ZodString, string, string>;
         mediaIds: z.ZodDefault<z.ZodArray<z.ZodString, "many">>;
     }, "strict", z.ZodTypeAny, {
-        role: "system" | "user" | "assistant";
         content: string;
+        role: "user" | "assistant" | "system";
         mediaIds: string[];
     }, {
-        role: "system" | "user" | "assistant";
         content: string;
+        role: "user" | "assistant" | "system";
         mediaIds?: string[] | undefined;
     }>, "many">;
     thinkingEnabled: z.ZodOptional<z.ZodBoolean>;

package/dist/server.d.ts

@@ -4,7 +4,7 @@ import { AiHiveApiClient } from "./api-client.js";
 import { type BearerTokenProvider } from "./auth-token-provider.js";
 import type { McpClientConfig } from "./config.js";
 import { WorkBuddyCredentialStore } from "./workbuddy-credentials.js";
-export declare const SERVER_VERSION = "0.2.1";
+export declare const SERVER_VERSION = "0.2.4";
 declare function content(value: unknown): {
     content: {
         type: "text";

package/dist/server.js

@@ -4,9 +4,10 @@ import { chatTextShape, generateImageShape, generateVideoShape, getGenerationTas
 import { z } from "zod";
 import { AiHiveApiClient } from "./api-client.js";
 import { normalizeApiEndpoint, StaticBearerTokenProvider, WorkBuddyTokenProvider, } from "./auth-token-provider.js";
+import { readMcpClientInfo } from "./client-info.js";
 import { uploadMediaFromPath } from "./local-media.js";
 import { WorkBuddyOAuthClient } from "./workbuddy-oauth-client.js";
-export const SERVER_VERSION = "0.2.1";
+export const SERVER_VERSION = "0.2.4";
 const DEPRECATED_ATTACHMENT_IDS_MESSAGE = "attachmentIds 已废弃，请使用 messages[].mediaIds";
 function content(value) {
     return {
@@ -103,6 +104,7 @@ export function createServer(input) {
         endpoint: input.endpoint,
         tokenProvider: input.tokenProvider,
         timeoutMs: input.timeoutMs,
+        clientInfoProvider: () => readMcpClientInfo(server),
     });
     for (const tool of createToolDefinitions(client)) {
         server.tool(tool.name, tool.schema, tool.handler);

package/dist/workbuddy-loopback.d.ts

@@ -3,8 +3,13 @@ export type LoopbackListener = {
     completion: Promise<void>;
     close(): Promise<void>;
 };
+export type WorkBuddyResultRedirects = {
+    successUrl: string;
+    failureUrl: string;
+};
 export declare function listenForWorkBuddyCallback(_input: {
     expectedState: string;
     timeoutMs: number;
     onCode(code: string): Promise<void>;
+    resultRedirects?: WorkBuddyResultRedirects;
 }): Promise<LoopbackListener>;

package/dist/workbuddy-loopback.js

@@ -102,7 +102,12 @@ export function listenForWorkBuddyCallback(_input) {
             }
             catch {
                 if (!settled) {
-                    await sendHtmlAndWait(response, 500, failureHtml());
+                    if (input.resultRedirects) {
+                        await sendRedirectAndWait(response, input.resultRedirects.failureUrl);
+                    }
+                    else {
+                        await sendHtmlAndWait(response, 500, failureHtml());
+                    }
                     await finishFailure("WorkBuddy 授权失败，请重新连接");
                 }
                 return;
@@ -111,7 +116,12 @@ export function listenForWorkBuddyCallback(_input) {
                 response.destroy();
                 return;
             }
-            await sendHtmlAndWait(response, 200, successHtml());
+            if (input.resultRedirects) {
+                await sendRedirectAndWait(response, input.resultRedirects.successUrl);
+            }
+            else {
+                await sendHtmlAndWait(response, 200, successHtml());
+            }
             await finishSuccess();
         }
         async function handleUnexpectedRequestFailure(response) {
@@ -179,6 +189,18 @@ function sendHtmlAndWait(response, status, body) {
         response.end(body, resolve);
     });
 }
+function sendRedirectAndWait(response, location) {
+    return new Promise((resolve) => {
+        response.writeHead(302, {
+            "Cache-Control": "no-store",
+            Connection: "close",
+            "Content-Length": "0",
+            Location: location,
+            "Referrer-Policy": "no-referrer",
+        });
+        response.end(resolve);
+    });
+}
 function page(title, content) {
     return `<!doctype html>
 <html lang="zh-CN">
@@ -191,7 +213,6 @@ function page(title, content) {
       body { margin: 0; min-height: 100vh; display: grid; place-items: center; font-family: sans-serif; background: #f8fafc; color: #0f172a; }
       main { width: min(420px, calc(100% - 48px)); padding: 32px; border: 1px solid #e2e8f0; border-radius: 16px; background: white; box-shadow: 0 16px 40px rgb(15 23 42 / 8%); }
       h1 { margin-top: 0; }
-      button { border: 0; border-radius: 8px; padding: 10px 16px; background: #2563eb; color: white; cursor: pointer; }
     </style>
   </head>
   <body>
@@ -202,11 +223,7 @@ function page(title, content) {
 function successHtml() {
     return page("授权成功", `<h1>授权成功</h1>
 <p>请返回 WorkBuddy 继续使用。</p>
-<p>页面将尝试自动关闭；如果没有关闭，请手动关闭。</p>
-<button type="button" onclick="window.close()">关闭页面</button>
-<script>
-  window.setTimeout(() => window.close(), 300);
-</script>`);
+<p>您可以手动关闭此页面。</p>`);
 }
 function failureHtml() {
     return page("授权失败", `<h1>授权失败</h1>

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@infimind-next/ai-hive-mcp",
-  "version": "0.2.1",
+  "version": "0.2.4",
   "private": false,
   "type": "module",
   "engines": {
@@ -12,6 +12,13 @@
   "files": [
     "dist"
   ],
+  "scripts": {
+    "build": "tsc -p tsconfig.json",
+    "dev": "tsc -p tsconfig.json --watch --preserveWatchOutput",
+    "lint": "eslint \"src/**/*.ts\"",
+    "prepack": "pnpm run build",
+    "test": "vitest run"
+  },
   "dependencies": {
     "@modelcontextprotocol/sdk": "^1.29.0",
     "file-type": "^21.3.4",
@@ -22,11 +29,5 @@
     "typescript": "^5.0.0",
     "typescript-eslint": "^8.0.0",
     "vitest": "^3.0.0"
-  },
-  "scripts": {
-    "build": "tsc -p tsconfig.json",
-    "dev": "tsc -p tsconfig.json --watch --preserveWatchOutput",
-    "lint": "eslint \"src/**/*.ts\"",
-    "test": "vitest run"
   }
-}
+}