import csv
import logging
from datetime import datetime, timedelta

import pytz
from requests_cache import CachedSession

from core.constants import HTTP_HEADERS
from data.spot import Spot
from spotproviders.http_spot_provider import HTTPSpotProvider


# Spot provider for Parks n Peaks
class ParksNPeaks(HTTPSpotProvider):
    POLL_INTERVAL_SEC = 120
    SPOTS_URL = "https://www.parksnpeaks.org/api/ALL"
    SIOTA_LIST_URL = "https://www.silosontheair.com/data/silos.csv"
    SIOTA_LIST_CACHE_TIME_DAYS = 30
    SIOTA_LIST_CACHE = CachedSession("cache/siota_data_cache", expire_after=timedelta(days=SIOTA_LIST_CACHE_TIME_DAYS))
    ZLOTA_LIST_URL = "https://ontheair.nz/assets/assets.json"
    ZLOTA_LIST_CACHE_TIME_DAYS = 30
    ZLOTA_LIST_CACHE = CachedSession("cache/zlota_data_cache", expire_after=timedelta(days=ZLOTA_LIST_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():
            # Convert to our spot format
            spot = Spot(source=self.name,
                        source_id=source_spot["actID"],
                        dx_call=source_spot["actCallsign"].upper(),
                        de_call=source_spot["actSpoter"].upper(),  # typo exists in API
                        freq=float(source_spot["actFreq"].replace(",", "")) * 1000000 if (
                                    source_spot["actFreq"] != "") else None,
                        # Seen PNP spots with empty frequency, and with comma-separated thousands digits
                        mode=source_spot["actMode"].upper(),
                        comment=source_spot["actComments"],
                        sig=source_spot["actClass"],
                        sig_refs=[source_spot["actSiteID"]],
                        time=datetime.strptime(source_spot["actTime"], "%Y-%m-%d %H:%M:%S").replace(
                            tzinfo=pytz.UTC).timestamp())

            # PNP supports a bunch of programs which should have different icons
            if spot.sig == "SiOTA":
                spot.icon = "wheat-awn"
            elif spot.sig == "ZLOTA":
                spot.icon = "kiwi-bird"
            elif spot.sig in ["POTA", "SOTA", "WWFF"]:
                # Don't care about an icon as this will be rejected anyway, we have better data from POTA/SOTA/WWFF direct
                spot.icon = ""
            else:
                # Unknown programme we've never seen before
                logging.warn(
                    "PNP spot found with sig " + spot.sig + ", developer needs to add support for icon and grid/lat/lon lookup!")
                spot.icon = "question"

            # SiOTA lat/lon/grid lookup
            if spot.sig == "SiOTA":
                siota_csv_data = self.SIOTA_LIST_CACHE.get(self.SIOTA_LIST_URL, headers=HTTP_HEADERS)
                siota_dr = csv.DictReader(siota_csv_data.content.decode().splitlines())
                for row in siota_dr:
                    if row["SILO_CODE"] == spot.sig_refs[0]:
                        spot.dx_latitude = float(row["LAT"])
                        spot.dx_longitude = float(row["LNG"])
                        spot.dx_grid = row["LOCATOR"]
                        break

            # ZLOTA name/lat/lon lookup
            if spot.sig == "ZLOTA":
                zlota_data = self.ZLOTA_LIST_CACHE.get(self.ZLOTA_LIST_URL, headers=HTTP_HEADERS).json()
                for asset in zlota_data:
                    if asset["code"] == spot.sig_refs[0]:
                        spot.sig_refs_names = [asset["name"]]
                        spot.dx_latitude = asset["y"]
                        spot.dx_longitude = asset["x"]
                        # Junk the "DE call", PNP always returns "ZLOTA" as the spotter for ZLOTA spots
                        spot.de_call = None
                        break

            # If this is POTA, SOTA or WWFF data we already have it through other means, so ignore. Otherwise, add to
            # the spot list.
            if spot.sig not in ["POTA", "SOTA", "WWFF"]:
                new_spots.append(spot)
        return new_spots