Function discoverTools

packages/plugins/mcp/src/sdk/discover.ts:102–143

(
  connector: McpConnector,
  timeoutMs: number = Duration.toMillis(DEFAULT_DISCOVER_TIMEOUT),
)

 * or the timeout above can still cancel them promptly.
 */
export const discoverTools = (
  connector: McpConnector,
  timeoutMs: number = Duration.toMillis(DEFAULT_DISCOVER_TIMEOUT),
): Effect.Effect<McpToolManifest, McpToolDiscoveryError> =>
  Effect.uninterruptibleMask((restore) =>
    Effect.gen(function* () {
      // Acquire connection
      const connection = yield* restore(
        connector.pipe(
          Effect.mapError((failure) => {
            // Preserve the handshake HTTP status (401/403 = auth wall) so the
            // liveness health check can classify structurally.
            const httpStatus = Predicate.isTagged(failure, "McpConnectionError")
              ? failure.httpStatus
              : undefined;
            return new McpToolDiscoveryError({
              stage: "connect",
              message: `Failed connecting to MCP server: ${failure.message}`,
              ...(httpStatus !== undefined ? { httpStatus } : {}),
            });
          }),
        ),
      );

      const manifest = yield* restore(listAllTools(connection)).pipe(
        Effect.onExit(() => closeConnection(connection)),
      );

      return manifest;
    }),
  ).pipe(
    Effect.timeoutOrElse({
      duration: Duration.millis(timeoutMs),
      orElse: () =>
        Effect.fail(
          new McpToolDiscoveryError({
            stage: "connect",
            message: `MCP discovery timed out after ${timeoutMs}ms`,
          }),
        ),
    }),
  );

const closeConnection = (connection: {
  readonly close: () => Promise<void>;