integrate-sdk 0.9.55 → 0.9.58

package/dist/index.js

@@ -1675,6 +1675,25 @@ function toConfiguredIntegrationWithToolMetadata(integration, toolMetadata) {
   };
 }
 
+// src/database/token-store.ts
+var USABLE_ACCESS_TOKEN_BUFFER_MS = 30 * 1000;
+var MIN_MEANINGFUL_EXPIRY_MS = Date.UTC(2000, 0, 1);
+async function listConnectedProviders(configuredIntegrationIds, getProviderToken, context) {
+  if (!context.userId || configuredIntegrationIds.length === 0) {
+    return [];
+  }
+  const connected = [];
+  await Promise.all(configuredIntegrationIds.map(async (provider) => {
+    try {
+      const token = await getProviderToken(provider, undefined, context);
+      if (token?.accessToken || typeof token?.refreshToken === "string" && token.refreshToken.length > 0) {
+        connected.push(provider);
+      }
+    } catch {}
+  }));
+  return connected.sort();
+}
+
 // src/client.ts
 init_errors();
 
@@ -3824,20 +3843,32 @@ class MCPClientBase {
   getEnabledTools() {
     return Array.from(this.availableTools.values()).filter((tool) => this.enabledToolNames.has(tool.name));
   }
-  async getEnabledToolsAsync() {
-    if (this.isConnected() && this.availableTools.size > 0) {
-      return this.getEnabledTools();
+  async getEnabledToolsAsync(options) {
+    const targetIntegrationIds = await this.resolveTargetIntegrationIds(options);
+    if (targetIntegrationIds.size === 0) {
+      return [];
     }
-    if (this.availableTools.size > 0) {
-      return this.getEnabledTools();
+    const filterToTargets = (tools2) => this.filterToolsToIntegrations(tools2, targetIntegrationIds);
+    const hasCompleteCache = () => {
+      for (const integration of this.integrations) {
+        if (!targetIntegrationIds.has(integration.id))
+          continue;
+        for (const toolName of integration.tools) {
+          if (!this.enabledToolNames.has(toolName))
+            continue;
+          if (!this.availableTools.has(toolName))
+            return false;
+        }
+      }
+      return this.availableTools.size > 0;
+    };
+    if (this.availableTools.size > 0 && hasCompleteCache()) {
+      return filterToTargets(this.getEnabledTools());
     }
     const tools = [];
-    const integrationIds = new Set;
-    for (const integration of this.integrations) {
-      integrationIds.add(integration.id);
-    }
     const { parallelWithLimit: parallelWithLimit2 } = await Promise.resolve().then(() => exports_concurrency);
-    const integrationToolsResults = await parallelWithLimit2(Array.from(integrationIds), async (integrationId) => {
+    const concurrency = options?.fetchConcurrency ?? 8;
+    const integrationToolsResults = await parallelWithLimit2(Array.from(targetIntegrationIds), async (integrationId) => {
       try {
         const response = await this.callServerToolInternal("list_tools_by_integration", {
           integration: integrationId
@@ -3848,25 +3879,14 @@ class MCPClientBase {
             if (item.type === "text" && item.text) {
               try {
                 const parsed = JSON.parse(item.text);
-                if (Array.isArray(parsed)) {
-                  for (const tool of parsed) {
-                    if (tool.name && tool.inputSchema) {
-                      integrationTools.push({
-                        name: tool.name,
-                        description: tool.description,
-                        inputSchema: tool.inputSchema
-                      });
-                    }
-                  }
-                } else if (parsed.tools && Array.isArray(parsed.tools)) {
-                  for (const tool of parsed.tools) {
-                    if (tool.name && tool.inputSchema) {
-                      integrationTools.push({
-                        name: tool.name,
-                        description: tool.description,
-                        inputSchema: tool.inputSchema
-                      });
-                    }
+                const parsedTools = Array.isArray(parsed) ? parsed : parsed.tools && Array.isArray(parsed.tools) ? parsed.tools : [];
+                for (const tool of parsedTools) {
+                  if (tool.name && tool.inputSchema) {
+                    integrationTools.push({
+                      name: tool.name,
+                      description: tool.description,
+                      inputSchema: tool.inputSchema
+                    });
                   }
                 }
               } catch {}
@@ -3878,14 +3898,40 @@ class MCPClientBase {
         logger5.error(`Failed to fetch tools for integration ${integrationId}:`, error);
         return [];
       }
-    }, 3);
+    }, concurrency);
     for (const integrationTools of integrationToolsResults) {
       tools.push(...integrationTools);
     }
     for (const tool of tools) {
       this.availableTools.set(tool.name, tool);
     }
-    return tools.filter((tool) => this.enabledToolNames.has(tool.name));
+    return filterToTargets(tools.filter((tool) => this.enabledToolNames.has(tool.name)));
+  }
+  async resolveTargetIntegrationIds(options) {
+    const configuredIds = this.integrations.map((integration) => integration.id);
+    const configuredSet = new Set(configuredIds);
+    if (options?.integrationIds && options.integrationIds.length > 0) {
+      return new Set(options.integrationIds.filter((id) => configuredSet.has(id)));
+    }
+    if (options?.connectedOnly && options.context?.userId) {
+      const connected = await listConnectedProviders(configuredIds, (provider, email, context) => this.oauthManager.getProviderToken(provider, email, context), options.context);
+      return new Set(connected);
+    }
+    return configuredSet;
+  }
+  filterToolsToIntegrations(tools, integrationIds) {
+    if (integrationIds.size === 0) {
+      return [];
+    }
+    const allowedNames = new Set;
+    for (const integration of this.integrations) {
+      if (!integrationIds.has(integration.id))
+        continue;
+      for (const toolName of integration.tools) {
+        allowedNames.add(toolName);
+      }
+    }
+    return tools.filter((tool) => this.enabledToolNames.has(tool.name) && allowedNames.has(tool.name));
   }
   getOAuthConfig(integrationId) {
     const integration = this.integrations.find((p) => p.id === integrationId);

package/dist/integrations.js

@@ -1675,6 +1675,25 @@ function toConfiguredIntegrationWithToolMetadata(integration, toolMetadata) {
   };
 }
 
+// src/database/token-store.ts
+var USABLE_ACCESS_TOKEN_BUFFER_MS = 30 * 1000;
+var MIN_MEANINGFUL_EXPIRY_MS = Date.UTC(2000, 0, 1);
+async function listConnectedProviders(configuredIntegrationIds, getProviderToken, context) {
+  if (!context.userId || configuredIntegrationIds.length === 0) {
+    return [];
+  }
+  const connected = [];
+  await Promise.all(configuredIntegrationIds.map(async (provider) => {
+    try {
+      const token = await getProviderToken(provider, undefined, context);
+      if (token?.accessToken || typeof token?.refreshToken === "string" && token.refreshToken.length > 0) {
+        connected.push(provider);
+      }
+    } catch {}
+  }));
+  return connected.sort();
+}
+
 // src/client.ts
 init_errors();
 
@@ -3824,20 +3843,32 @@ class MCPClientBase {
   getEnabledTools() {
     return Array.from(this.availableTools.values()).filter((tool) => this.enabledToolNames.has(tool.name));
   }
-  async getEnabledToolsAsync() {
-    if (this.isConnected() && this.availableTools.size > 0) {
-      return this.getEnabledTools();
+  async getEnabledToolsAsync(options) {
+    const targetIntegrationIds = await this.resolveTargetIntegrationIds(options);
+    if (targetIntegrationIds.size === 0) {
+      return [];
     }
-    if (this.availableTools.size > 0) {
-      return this.getEnabledTools();
+    const filterToTargets = (tools2) => this.filterToolsToIntegrations(tools2, targetIntegrationIds);
+    const hasCompleteCache = () => {
+      for (const integration of this.integrations) {
+        if (!targetIntegrationIds.has(integration.id))
+          continue;
+        for (const toolName of integration.tools) {
+          if (!this.enabledToolNames.has(toolName))
+            continue;
+          if (!this.availableTools.has(toolName))
+            return false;
+        }
+      }
+      return this.availableTools.size > 0;
+    };
+    if (this.availableTools.size > 0 && hasCompleteCache()) {
+      return filterToTargets(this.getEnabledTools());
     }
     const tools = [];
-    const integrationIds = new Set;
-    for (const integration of this.integrations) {
-      integrationIds.add(integration.id);
-    }
     const { parallelWithLimit: parallelWithLimit2 } = await Promise.resolve().then(() => exports_concurrency);
-    const integrationToolsResults = await parallelWithLimit2(Array.from(integrationIds), async (integrationId) => {
+    const concurrency = options?.fetchConcurrency ?? 8;
+    const integrationToolsResults = await parallelWithLimit2(Array.from(targetIntegrationIds), async (integrationId) => {
       try {
         const response = await this.callServerToolInternal("list_tools_by_integration", {
           integration: integrationId
@@ -3848,25 +3879,14 @@ class MCPClientBase {
             if (item.type === "text" && item.text) {
               try {
                 const parsed = JSON.parse(item.text);
-                if (Array.isArray(parsed)) {
-                  for (const tool of parsed) {
-                    if (tool.name && tool.inputSchema) {
-                      integrationTools.push({
-                        name: tool.name,
-                        description: tool.description,
-                        inputSchema: tool.inputSchema
-                      });
-                    }
-                  }
-                } else if (parsed.tools && Array.isArray(parsed.tools)) {
-                  for (const tool of parsed.tools) {
-                    if (tool.name && tool.inputSchema) {
-                      integrationTools.push({
-                        name: tool.name,
-                        description: tool.description,
-                        inputSchema: tool.inputSchema
-                      });
-                    }
+                const parsedTools = Array.isArray(parsed) ? parsed : parsed.tools && Array.isArray(parsed.tools) ? parsed.tools : [];
+                for (const tool of parsedTools) {
+                  if (tool.name && tool.inputSchema) {
+                    integrationTools.push({
+                      name: tool.name,
+                      description: tool.description,
+                      inputSchema: tool.inputSchema
+                    });
                   }
                 }
               } catch {}
@@ -3878,14 +3898,40 @@ class MCPClientBase {
         logger5.error(`Failed to fetch tools for integration ${integrationId}:`, error);
         return [];
       }
-    }, 3);
+    }, concurrency);
     for (const integrationTools of integrationToolsResults) {
       tools.push(...integrationTools);
     }
     for (const tool of tools) {
       this.availableTools.set(tool.name, tool);
     }
-    return tools.filter((tool) => this.enabledToolNames.has(tool.name));
+    return filterToTargets(tools.filter((tool) => this.enabledToolNames.has(tool.name)));
+  }
+  async resolveTargetIntegrationIds(options) {
+    const configuredIds = this.integrations.map((integration) => integration.id);
+    const configuredSet = new Set(configuredIds);
+    if (options?.integrationIds && options.integrationIds.length > 0) {
+      return new Set(options.integrationIds.filter((id) => configuredSet.has(id)));
+    }
+    if (options?.connectedOnly && options.context?.userId) {
+      const connected = await listConnectedProviders(configuredIds, (provider, email, context) => this.oauthManager.getProviderToken(provider, email, context), options.context);
+      return new Set(connected);
+    }
+    return configuredSet;
+  }
+  filterToolsToIntegrations(tools, integrationIds) {
+    if (integrationIds.size === 0) {
+      return [];
+    }
+    const allowedNames = new Set;
+    for (const integration of this.integrations) {
+      if (!integrationIds.has(integration.id))
+        continue;
+      for (const toolName of integration.tools) {
+        allowedNames.add(toolName);
+      }
+    }
+    return tools.filter((tool) => this.enabledToolNames.has(tool.name) && allowedNames.has(tool.name));
   }
   getOAuthConfig(integrationId) {
     const integration = this.integrations.find((p) => p.id === integrationId);