import logging
from typing import Union

import schemas
from chalicelib.core import metrics

logger = logging.getLogger(__name__)


def get_metric(key: Union[schemas.MetricOfWebVitals, schemas.MetricOfErrors], project_id: int, data: dict):
    supported = {schemas.MetricOfWebVitals.COUNT_SESSIONS: metrics.get_processed_sessions,
                 schemas.MetricOfWebVitals.AVG_VISITED_PAGES: metrics.get_user_activity_avg_visited_pages,
                 schemas.MetricOfWebVitals.COUNT_REQUESTS: metrics.get_top_metrics_count_requests,
                 schemas.MetricOfErrors.IMPACTED_SESSIONS_BY_JS_ERRORS: metrics.get_impacted_sessions_by_js_errors,
                 schemas.MetricOfErrors.DOMAINS_ERRORS_4XX: metrics.get_domains_errors_4xx,
                 schemas.MetricOfErrors.DOMAINS_ERRORS_5XX: metrics.get_domains_errors_5xx,
                 schemas.MetricOfErrors.ERRORS_PER_DOMAINS: metrics.get_errors_per_domains,
                 schemas.MetricOfErrors.ERRORS_PER_TYPE: metrics.get_errors_per_type,
                 schemas.MetricOfErrors.RESOURCES_BY_PARTY: metrics.get_resources_by_party,
                 schemas.MetricOfWebVitals.COUNT_USERS: metrics.get_unique_users, }

    return supported.get(key, lambda *args: None)(project_id=project_id, **data)