@ai-sdk/mcp 1.0.73 → 1.0.75

package/dist/index.mjs

@@ -524,7 +524,10 @@ function checkResourceAllowed({
 }
 
 // src/tool/oauth.ts
-import { parseJSON as parseJSON2 } from "@ai-sdk/provider-utils";
+import {
+  parseJSON as parseJSON2,
+  validateDownloadUrl
+} from "@ai-sdk/provider-utils";
 var UnauthorizedError = class extends Error {
   constructor(message = "Unauthorized") {
     super(message);
@@ -534,6 +537,23 @@ var UnauthorizedError = class extends Error {
 function normalizeUrl(url) {
   return new URL(url).href;
 }
+function isOAuthLoopbackHost(hostname) {
+  const normalized = hostname.toLowerCase().replace(/\.+$/, "");
+  return normalized === "localhost" || normalized.endsWith(".localhost") || normalized === "127.0.0.1" || normalized === "[::1]" || normalized === "::1";
+}
+function assertSafeOAuthEndpoint(endpointUrl) {
+  if ((endpointUrl.protocol === "http:" || endpointUrl.protocol === "https:") && isOAuthLoopbackHost(endpointUrl.hostname)) {
+    return;
+  }
+  try {
+    validateDownloadUrl(endpointUrl.href);
+  } catch (error) {
+    throw new MCPClientOAuthError({
+      message: `OAuth endpoint URL is not allowed: ${endpointUrl.href}`,
+      cause: error
+    });
+  }
+}
 function createAuthorizationServerInformation(authorizationServerUrl, metadata) {
   return {
     authorizationServerUrl: normalizeUrl(authorizationServerUrl),
@@ -966,6 +986,7 @@ async function exchangeAuthorization(authorizationServerUrl, {
   var _a3;
   const grantType = "authorization_code";
   const tokenUrl = (metadata == null ? void 0 : metadata.token_endpoint) ? new URL(metadata.token_endpoint) : new URL("/token", authorizationServerUrl);
+  assertSafeOAuthEndpoint(tokenUrl);
   if ((metadata == null ? void 0 : metadata.grant_types_supported) && !metadata.grant_types_supported.includes(grantType)) {
     throw new Error(
       `Incompatible auth server: does not support grant type ${grantType}`
@@ -1002,7 +1023,8 @@ async function exchangeAuthorization(authorizationServerUrl, {
   const response = await (fetchFn != null ? fetchFn : fetch)(tokenUrl, {
     method: "POST",
     headers,
-    body: params
+    body: params,
+    redirect: "error"
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1030,6 +1052,7 @@ async function refreshAuthorization(authorizationServerUrl, {
   } else {
     tokenUrl = new URL("/token", authorizationServerUrl);
   }
+  assertSafeOAuthEndpoint(tokenUrl);
   const headers = new Headers({
     "Content-Type": "application/x-www-form-urlencoded",
     Accept: "application/json"
@@ -1059,7 +1082,8 @@ async function refreshAuthorization(authorizationServerUrl, {
   const response = await (fetchFn != null ? fetchFn : fetch)(tokenUrl, {
     method: "POST",
     headers,
-    body: params
+    body: params,
+    redirect: "error"
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1085,12 +1109,14 @@ async function registerClient(authorizationServerUrl, {
   } else {
     registrationUrl = new URL("/register", authorizationServerUrl);
   }
+  assertSafeOAuthEndpoint(registrationUrl);
   const response = await (fetchFn != null ? fetchFn : fetch)(registrationUrl, {
     method: "POST",
     headers: {
       "Content-Type": "application/json"
     },
-    body: JSON.stringify(clientMetadata)
+    body: JSON.stringify(clientMetadata),
+    redirect: "error"
   });
   if (!response.ok) {
     throw await parseErrorResponse(response);
@@ -1176,6 +1202,12 @@ async function authInternal(provider, {
     }
   );
   const currentAuthorizationServerInformation = createAuthorizationServerInformation(authorizationServerUrl, metadata);
+  const clientMetadata = provider.clientMetadata;
+  const selectedScope = selectScope({
+    scope,
+    resourceMetadata,
+    clientMetadata
+  });
   let clientInformation = await Promise.resolve(provider.clientInformation());
   if (!clientInformation) {
     if (authorizationCode !== void 0) {
@@ -1190,7 +1222,10 @@ async function authInternal(provider, {
     }
     const fullInformation = await registerClient(authorizationServerUrl, {
       metadata,
-      clientMetadata: provider.clientMetadata,
+      clientMetadata: {
+        ...clientMetadata,
+        scope: selectedScope
+      },
       fetchFn
     });
     clientInformation = addAuthorizationServerInformationToClientInformation(
@@ -1294,11 +1329,7 @@ async function authInternal(provider, {
       clientInformation,
       state,
       redirectUrl: provider.redirectUrl,
-      scope: selectScope({
-        scope,
-        resourceMetadata,
-        clientMetadata: provider.clientMetadata
-      }),
+      scope: selectedScope,
       resource
     }
   );
@@ -2505,8 +2536,15 @@ var DefaultMCPClient = class {
   async tools({
     schemas = "automatic"
   } = {}) {
-    const definitions = await this.listTools();
-    return this.toolsFromDefinitions(definitions, {
+    let definitions = await this.listTools();
+    const tools = [...definitions.tools];
+    while (definitions.nextCursor != null) {
+      definitions = await this.listTools({
+        params: { cursor: definitions.nextCursor }
+      });
+      tools.push(...definitions.tools);
+    }
+    return this.toolsFromDefinitions({ ...definitions, tools }, {
       schemas
     });
   }