feat: add client type identification to API requests

Author: ahonn

manifest.json

@@ -36,7 +36,8 @@
       "command": "node",
       "args": ["${__dirname}/build/index.js"],
       "env": {
-        "PAGEINDEX_API_KEY": "${user_config.api_key}"
+        "PAGEINDEX_API_KEY": "${user_config.api_key}",
+        "PAGEINDEX_API_URL": "${user_config.api_url}"
       }
     }
   },
@@ -46,6 +47,13 @@
       "title": "PageIndex API Key",
       "description": "Your PageIndex API key. Get one at https://dash.pageindex.ai/api-keys",
       "required": true
+    },
+    "api_url": {
+      "type": "string",
+      "title": "PageIndex API URL",
+      "description": "PageIndex API base URL (optional, defaults to https://dash.pageindex.ai)",
+      "required": false,
+      "default": "https://dash.pageindex.ai"
     }
   }
 }

package.json

@@ -16,7 +16,7 @@
     "format": "biome format --write .",
     "clean": "rm -rf build",
     "prepublishOnly": "npm run clean && npm run build",
-    "dxt:build": "tsup",
+    "dxt:build": "CLIENT_TYPE=dxt tsup",
     "dxt:pack": "npm run dxt:build && dxt pack",
     "dxt:dev": "dxt dev",
     "postinstall": "node -e \"try { require('fs').existsSync('.git') && require('simple-git-hooks')() } catch(e) {}\"",

src/client/auth.ts

@@ -4,11 +4,19 @@ import type { FetchLike } from '@modelcontextprotocol/sdk/shared/transport.js';
  * Creates an authenticated fetch function that adds Bearer token authentication
  * This is a simplified approach compared to implementing the complex OAuth interface
  */
-export function createAuthenticatedFetch(apiKey: string): FetchLike {
+export function createAuthenticatedFetch(
+  apiKey: string,
+  additionalHeaders: Record<string, string> = {},
+): FetchLike {
   return async (input: string | URL, init?: RequestInit): Promise<Response> => {
     const headers = new Headers(init?.headers);
     headers.set('Authorization', `Bearer ${apiKey}`);
 
+    // Add additional headers (like client identification)
+    Object.entries(additionalHeaders).forEach(([key, value]) => {
+      headers.set(key, value);
+    });
+
     return fetch(input, {
       ...init,
       headers,

src/client/mcp-client.ts

@@ -20,6 +20,16 @@ export class PageIndexMcpClient {
     this.apiKey = apiKey;
   }
 
+  /**
+   * Get client identification headers
+   */
+  private getClientHeaders(): Record<string, string> {
+    return {
+      'X-Client-Type': __CLIENT_TYPE__,
+      'X-Client-Version': __VERSION__,
+    };
+  }
+
   /**
    * Connect to the remote PageIndex MCP server
    */
@@ -43,8 +53,12 @@ export class PageIndexMcpClient {
 
         // Try StreamableHTTP first, fallback to SSE for compatibility
         try {
-          // Use simplified authenticated fetch
-          const authenticatedFetch = createAuthenticatedFetch(this.apiKey);
+          // Use simplified authenticated fetch with client headers
+          const clientHeaders = this.getClientHeaders();
+          const authenticatedFetch = createAuthenticatedFetch(
+            this.apiKey,
+            clientHeaders,
+          );
 
           this.transport = new StreamableHTTPClientTransport(
             streamableHttpUrl,
@@ -61,7 +75,11 @@ export class PageIndexMcpClient {
         } catch (_error) {
           try {
             // For SSE transport, we need to pass authenticated fetch as well
-            const authenticatedFetch = createAuthenticatedFetch(this.apiKey);
+            const clientHeaders = this.getClientHeaders();
+            const authenticatedFetch = createAuthenticatedFetch(
+              this.apiKey,
+              clientHeaders,
+            );
             this.transport = new SSEClientTransport(sseUrl, {
               fetch: authenticatedFetch,
             });

src/globals.d.ts

@@ -0,0 +1,2 @@
+declare const __VERSION__: string;
+declare const __CLIENT_TYPE__: 'dxt' | 'npm';

src/tools/remote-proxy.ts

@@ -15,6 +15,21 @@ export class RemoteToolsProxy {
     this.client = client;
   }
 
+  /**
+   * Get client type information
+   */
+  getClientInfo(): {
+    type: 'dxt' | 'npm';
+    version: string;
+    isDxt: boolean;
+  } {
+    return {
+      type: __CLIENT_TYPE__,
+      version: __VERSION__,
+      isDxt: __CLIENT_TYPE__ === 'dxt',
+    };
+  }
+
   /**
    * Fetch tool definitions from remote MCP server
    */

tsup.config.ts

@@ -14,6 +14,7 @@ export default defineConfig({
   noExternal: [/.*/], // Bundle all dependencies
   define: {
     __VERSION__: `"${packageJson.version}"`,
+    __CLIENT_TYPE__: `"${process.env.CLIENT_TYPE || 'npm'}"`,
   },
   platform: 'node',
   onSuccess: async () => {