"""Logic for permission related to analytics."""

from ddtrace import tracer

from sound_recordings.connectors import redis
from sound_recordings.constants import cache as cache_constants
from sound_recordings.constants.access import ACCESS_ANALYTICS
from sound_recordings.logic.parallel import parallel
from sound_recordings.services import ows_account, ows_users
from sound_recordings.utils.cache import create_access_key


@tracer.wrap(name="has_analytics")
def has_analytics(
    account_type, account_id, user_id, profile_type=None, profile_id=None
):
    """
    Verify that user has an access to Analytics tab in Workstation.

    Check feature control on account level, and on user level. Also check
    for the feature flag.

    Return True if Analytics is enabled, otherwise render an access error.
    """
    if profile_type and profile_id:
        return True

    redis_key = create_access_key(account_type, account_id, user_id, ACCESS_ANALYTICS)
    result = redis.client.get(redis_key)

    if result and result == b"True":
        return True
    if result and result == b"False":
        return False

    if account_type == ows_account.account_constants.SUBACCOUNT_TYPE:
        vendor_id = ows_account.get_vendor_id(account_id).message
    else:
        vendor_id = account_id

    requests = {
        "is_enabled_for_vendor": {
            "func": ows_account.vendor_has_access,
            "args": (vendor_id,),
            "kwargs": {},
        },
        "is_enabled_for_user": {
            "func": ows_users.user_has_access,
            "args": (account_id, account_type, user_id),
            "kwargs": {},
        },
    }

    result = parallel(requests).message

    if result.get("error"):
        raise Exception(result.get("error"))

    if not result.get("is_enabled_for_vendor") or not result.get("is_enabled_for_user"):
        redis.client.set(redis_key, b"False", ex=cache_constants.ACCESS_CACHE_TTL)
        return False

    # TODO: check if we can reset key on update of art_relations
    # TODO: check if we can pre-warm the cache
    redis.client.set(redis_key, b"True", ex=cache_constants.ACCESS_CACHE_TTL)
    return True