"""
Copyright (c) Microsoft Corporation. All rights reserved.
Licensed under the MIT License.
"""

from typing import Optional

from microsoft_teams.api import ActivityBase


def extract_tenant_id(activity: ActivityBase) -> Optional[str]:
    """
    Extract tenant ID from an activity with fallback logic.

    Attempts to extract tenant ID from multiple sources in the following order:
    1. activity.conversation.tenant_id (primary source)
    2. activity.tenant.id (fallback source from channel data)

    Args:
        activity: The activity to extract tenant ID from

    Returns:
        The tenant ID if found, None otherwise
    """
    if activity.conversation and activity.conversation.tenant_id:
        return activity.conversation.tenant_id

    tenant = getattr(activity, "tenant", None)
    if tenant and hasattr(tenant, "id"):
        return tenant.id

    return None