airbyte-agent-mailchimp 0.1.4__py3-none-any.whl

airbyte_agent_mailchimp/_vendored/connector_sdk/auth_template.py

@@ -0,0 +1,135 @@
+"""
+Template engine for auth_mapping in x-airbyte-auth-config.
+
+Handles template substitution for mapping user-provided config values to auth parameters.
+"""
+
+import re
+from typing import Dict
+
+
+class MissingVariableError(ValueError):
+    """Raised when a template variable is not found in config.
+
+    Extends ValueError for backwards compatibility with code that catches ValueError.
+    """
+
+    def __init__(self, var_name: str, available_fields: list):
+        self.var_name = var_name
+        self.available_fields = available_fields
+        super().__init__(f"Template variable '${{{var_name}}}' not found in config. Available fields: {available_fields}")
+
+
+def apply_template(template: str, values: Dict[str, str]) -> str:
+    """
+    Apply template substitution for auth_mapping.
+
+    Template syntax:
+    - ${variable}: Replaced with value from the values dict
+    - Any other text: Used as-is (constants or concatenation)
+
+    Examples:
+        >>> apply_template("${api_key}", {"api_key": "abc123"})
+        'abc123'
+
+        >>> apply_template("${email}/token", {"email": "user@example.com"})
+        'user@example.com/token'
+
+        >>> apply_template("api_token", {})
+        'api_token'
+
+        >>> apply_template("", {})
+        ''
+
+    Args:
+        template: Template string with ${variable} placeholders
+        values: Dict of variable names to values
+
+    Returns:
+        Resolved template string
+
+    Raises:
+        MissingVariableError: If template contains unresolved variables
+    """
+    if not template:
+        return ""
+
+    # Check if it's a pure constant (no variables)
+    if "${" not in template:
+        return template
+
+    # Find all variable references
+    variable_pattern = re.compile(r"\$\{([^}]+)\}")
+    matches = variable_pattern.findall(template)
+
+    # Substitute all ${var} with values
+    result = template
+    for var_name in matches:
+        if var_name not in values:
+            raise MissingVariableError(var_name, list(values.keys()))
+        # Replace the variable with its value
+        result = result.replace(f"${{{var_name}}}", values[var_name])
+
+    return result
+
+
+def apply_auth_mapping(
+    auth_mapping: Dict[str, str],
+    user_config: Dict[str, str],
+    required_fields: list | None = None,
+) -> Dict[str, str]:
+    """
+    Apply auth_mapping templates to user config.
+
+    Takes the auth_mapping from x-airbyte-auth-config and user-provided config,
+    and returns the mapped auth parameters. Optional fields (not in required_fields)
+    are skipped if their template variables are not provided.
+
+    Example:
+        >>> auth_mapping = {
+        ...     "username": "${api_key}",
+        ...     "password": ""
+        ... }
+        >>> user_config = {"api_key": "my_key_123"}
+        >>> apply_auth_mapping(auth_mapping, user_config)
+        {'username': 'my_key_123', 'password': ''}
+
+        >>> # Optional fields are skipped if not provided
+        >>> auth_mapping = {
+        ...     "access_token": "${access_token}",
+        ...     "refresh_token": "${refresh_token}",
+        ... }
+        >>> user_config = {"access_token": "abc123"}
+        >>> apply_auth_mapping(auth_mapping, user_config, required_fields=["access_token"])
+        {'access_token': 'abc123'}
+
+    Args:
+        auth_mapping: Dict mapping auth parameters to template strings
+        user_config: Dict of user-provided field values
+        required_fields: List of required field names. If a template references
+            a variable not in user_config and that variable is not required,
+            the mapping is skipped. Behavior:
+            - None: all fields are treated as required (backward compatible)
+            - []: no fields are required (all optional)
+            - ["foo"]: only "foo" is required
+
+    Returns:
+        Dict of resolved auth parameters
+
+    Raises:
+        MissingVariableError: If a required template variable is not found
+    """
+    resolved = {}
+    required_set = set(required_fields) if required_fields is not None else None
+
+    for param, template in auth_mapping.items():
+        try:
+            resolved[param] = apply_template(template, user_config)
+        except MissingVariableError as e:
+            # If the missing variable is not in required fields, skip this mapping
+            if required_set is not None and e.var_name not in required_set:
+                continue
+            # Otherwise, re-raise the error
+            raise
+
+    return resolved

airbyte_agent_mailchimp/_vendored/connector_sdk/cloud_utils/__init__.py

@@ -0,0 +1,5 @@
+"""Cloud API utilities for Airbyte Platform integration."""
+
+from .client import AirbyteCloudClient
+
+__all__ = ["AirbyteCloudClient"]

airbyte_agent_mailchimp/_vendored/connector_sdk/cloud_utils/client.py

@@ -0,0 +1,213 @@
+"""AirbyteCloudClient for Airbyte Platform API integration."""
+
+from __future__ import annotations
+
+from datetime import datetime, timedelta
+from typing import Any
+
+import httpx
+
+
+class AirbyteCloudClient:
+    """Client for interacting with Airbyte Platform APIs.
+
+    Handles authentication, token caching, and API calls to:
+    - Get bearer tokens for authentication
+    - Look up connectors for users
+    - Execute connectors via the cloud API
+
+    Example:
+        client = AirbyteCloudClient(
+            client_id="your-client-id",
+            client_secret="your-client-secret"
+        )
+
+        # Get a connector ID
+        connector_id = await client.get_connector_id(
+            external_user_id="user-123",
+            connector_definition_id="550e8400-e29b-41d4-a716-446655440000"
+        )
+
+        # Execute the connector
+        result = await client.execute_connector(
+            connector_id=connector_id,
+            entity="customers",
+            action="list",
+            params={"limit": 10}
+        )
+    """
+
+    AUTH_BASE_URL = "https://cloud.airbyte.com"  # For token endpoint
+    API_BASE_URL = "https://api.airbyte.ai"  # For instance lookup & execution
+
+    def __init__(self, client_id: str, client_secret: str):
+        """Initialize AirbyteCloudClient.
+
+        Args:
+            client_id: Airbyte client ID for authentication
+            client_secret: Airbyte client secret for authentication
+        """
+        self._client_id = client_id
+        self._client_secret = client_secret
+
+        # Token cache (instance-level)
+        self._cached_token: str | None = None
+        self._token_expires_at: datetime | None = None
+        self._http_client = httpx.AsyncClient(
+            timeout=httpx.Timeout(300.0),  # 5 minute timeout
+            follow_redirects=True,
+        )
+
+    async def get_bearer_token(self) -> str:
+        """Get bearer token for API authentication.
+
+        Caches the token and only requests a new one when the cached token
+        is expired or missing. Adds a 60-second buffer before expiration
+        to avoid edge cases.
+
+        Returns:
+            Bearer token string
+
+        Raises:
+            httpx.HTTPStatusError: If the token request fails with 4xx/5xx
+            httpx.RequestError: If the network request fails
+
+        Example:
+            token = await client.get_bearer_token()
+            # Use token in Authorization header: f"Bearer {token}"
+        """
+        # Check if we have a cached token that hasn't expired
+        if self._cached_token and self._token_expires_at:
+            # Add 60 second buffer before expiration to avoid edge cases
+            now = datetime.now()
+            if now < self._token_expires_at:
+                # Token is still valid, return cached version
+                return self._cached_token
+
+        # Token is missing or expired, fetch a new one
+        url = f"{self.AUTH_BASE_URL}/api/v1/applications/token"
+        request_body = {
+            "client_id": self._client_id,
+            "client_secret": self._client_secret,
+        }
+
+        response = await self._http_client.post(url, json=request_body)
+        response.raise_for_status()
+
+        data = response.json()
+        access_token = data["access_token"]
+        expires_in = 15 * 60  # default 15 min expiry time * 60 seconds
+
+        # Calculate expiration time with 60 second buffer
+        expires_at = datetime.now() + timedelta(seconds=expires_in - 60)
+        self._cached_token = access_token
+        self._token_expires_at = expires_at
+
+        return access_token
+
+    async def get_connector_id(
+        self,
+        external_user_id: str,
+        connector_definition_id: str,
+    ) -> str:
+        """Get connector ID for a user.
+
+        Looks up the connector that belongs to the specified user
+        and connector definition. Validates that exactly one connector exists.
+
+        Args:
+            external_user_id: User identifier in the Airbyte system
+            connector_definition_id: UUID of the connector definition
+
+        Returns:
+            Connector ID (UUID string)
+
+        Raises:
+            ValueError: If 0 or more than 1 connector is found
+            httpx.HTTPStatusError: If API returns 4xx/5xx status code
+            httpx.RequestError: If network request fails
+
+        Example:
+            connector_id = await client.get_connector_id(
+                external_user_id="user-123",
+                connector_definition_id="550e8400-e29b-41d4-a716-446655440000"
+            )
+        """
+
+        token = await self.get_bearer_token()
+        url = f"{self.API_BASE_URL}/api/v1/connectors/connectors_for_user"
+        params = {
+            "external_user_id": external_user_id,
+            "definition_id": connector_definition_id,
+        }
+
+        headers = {"Authorization": f"Bearer {token}"}
+        response = await self._http_client.get(url, params=params, headers=headers)
+        response.raise_for_status()
+
+        data = response.json()
+        connectors = data["connectors"]
+
+        if len(connectors) == 0:
+            raise ValueError(f"No connector found for user '{external_user_id}' and connector definition '{connector_definition_id}'")
+
+        if len(connectors) > 1:
+            raise ValueError(
+                f"Multiple connectors found for user '{external_user_id}' "
+                f"and connector definition '{connector_definition_id}'. Expected exactly 1, "
+                f"found {len(connectors)}"
+            )
+
+        connector_id = connectors[0]["id"]
+        return connector_id
+
+    async def execute_connector(
+        self,
+        connector_id: str,
+        entity: str,
+        action: str,
+        params: dict[str, Any] | None,
+    ) -> dict[str, Any]:
+        """Execute a connector operation.
+
+        Args:
+            connector_id: Connector UUID (source ID)
+            entity: Entity name (e.g., "customers", "invoices")
+            action: Operation action (e.g., "list", "get", "create")
+            params: Optional parameters for the operation
+
+        Returns:
+            Raw JSON response dict from the API
+
+        Raises:
+            httpx.HTTPStatusError: If API returns 4xx/5xx status code
+            httpx.RequestError: If network request fails
+
+        Example:
+            result = await client.execute_connector(
+                connector_id="550e8400-e29b-41d4-a716-446655440000",
+                entity="customers",
+                action="list",
+                params={"limit": 10}
+            )
+        """
+        token = await self.get_bearer_token()
+        url = f"{self.API_BASE_URL}/api/v1/connectors/sources/{connector_id}/execute"
+        headers = {"Authorization": f"Bearer {token}"}
+        request_body = {
+            "entity": entity,
+            "action": action,
+            "params": params,
+        }
+
+        response = await self._http_client.post(url, json=request_body, headers=headers)
+        response.raise_for_status()
+
+        return response.json()
+
+    async def close(self):
+        """Close the HTTP client.
+
+        Call this when you're done using the client to clean up resources.
+        """
+        await self._http_client.aclose()