from datetime import datetime, timedelta

import pytz
from requests_cache import CachedSession

from data.spot import Spot
from providers.http_provider import HTTPProvider


# Provider for General Mountain Activity
class GMA(HTTPProvider):
    POLL_INTERVAL_SEC = 120
    SPOTS_URL = "https://www.cqgma.org/api/spots/25/"
    # GMA spots don't contain the details of the programme they are for, we need a separate lookup for that
    REF_INFO_URL_ROOT = "https://www.cqgma.org/api/ref/?"
    REF_INFO_CACHE_TIME_DAYS = 30
    REF_INFO_CACHE = CachedSession("gma_ref_info_cache", expire_after=timedelta(days=REF_INFO_CACHE_TIME_DAYS))

    def __init__(self, provider_config):
        super().__init__(provider_config, self.SPOTS_URL, self.POLL_INTERVAL_SEC)

    def http_response_to_spots(self, http_response):
        new_spots = []
        # Iterate through source data
        for source_spot in http_response.json()["RCD"]:
            # Convert to our spot format
            spot = Spot(source=self.name,
                        dx_call=source_spot["ACTIVATOR"].upper(),
                        de_call=source_spot["SPOTTER"].upper(),
                        freq=float(source_spot["QRG"]) if (source_spot["QRG"] != "") else None, # Seen GMA spots with no frequency
                        mode=source_spot["MODE"].upper() if "<>" not in source_spot["MODE"] else None, # Filter out some weird mode strings
                        comment=source_spot["TEXT"],
                        sig_refs=[source_spot["REF"]],
                        sig_refs_names=[source_spot["NAME"]],
                        time=datetime.strptime(source_spot["DATE"] + source_spot["TIME"], "%Y%m%d%H%M").replace(tzinfo=pytz.UTC),
                        latitude=float(source_spot["LAT"]) if (source_spot["LAT"] != "") else None, # Seen GMA spots with no lat/lon
                        longitude=float(source_spot["LON"]) if (source_spot["LON"] != "") else None)

            # GMA doesn't give what programme (SIG) the reference is for until we separately look it up.
            ref_response = self.REF_INFO_CACHE.get(self.REF_INFO_URL_ROOT + source_spot["REF"], headers=self.HTTP_HEADERS)
            # Sometimes this is blank, so handle that
            if ref_response.text is not None and ref_response.text != "":
                ref_info = ref_response.json()
                spot.sig = ref_info["reftype"]

            # Add to our list. Don't worry about de-duping, removing old spots etc. at this point; other code will do
            # that for us.
            new_spots.append(spot)
        return new_spots