"""

This module contains the API for fetching ols database

"""

from typing import List, Dict

import requests

def fetch_from_ols(term: str) -> str:

    """

    Fetch the label for a single term from OLS.

    Args:

        term (str): The term in the format "ONTOLOGY:TERM_ID".

    Returns:

        str: The label for the term or an error message.

    """

    try:

        ontology, _ = term.split(":")

        base_url = f"https://www.ebi.ac.uk/ols4/api/ontologies/{ontology.lower()}/terms"

        params = {"obo_id": term}

        response = requests.get(

            base_url,

            params=params,

            headers={"Accept": "application/json"},

            timeout=10

        )

        response.raise_for_status()

        data = response.json()

        label = '-'

        # Extract and return the label

        if "_embedded" in data and "terms" in data["_embedded"] \

             and len(data["_embedded"]["terms"]) > 0:

            label = data["_embedded"]["terms"][0].get("label", "Label not found")

        return label

    except (requests.exceptions.RequestException, KeyError, IndexError) as e:

        return f"Error: {str(e)}"

def fetch_ols_labels(terms: List[str]) -> Dict[str, str]:

    """

    Fetch labels for multiple terms from OLS.

    Args:

        terms (List[str]): A list of terms in the format "ONTOLOGY:TERM_ID".

    Returns:

        Dict[str, str]: A mapping of term IDs to their labels or error messages.

    """

    results = {}

    for term in terms:

        results[term] = fetch_from_ols(term)

    return results

def search_ols_labels(data: List[Dict[str, str]]) -> Dict[str, Dict[str, str]]:

    """

    Fetch OLS annotations grouped by ontology type.

    Args:

        data (List[Dict[str, str]]): A list of dictionaries containing 'Id' and 'Database'.

    Returns:

        Dict[str, Dict[str, str]]: A mapping of ontology type to term labels.

    """

    grouped_data = {}

    for entry in data:

        ontology = entry["Database"].lower()

        grouped_data.setdefault(ontology, []).append(entry["Id"])

    results = {}

    for ontology, terms in grouped_data.items():

        results[ontology] = fetch_ols_labels(terms)

    return results