import json
from datetime import datetime

import pytz

from data.sig_ref import SIGRef
from data.spot import Spot
from spotproviders.websocket_spot_provider import WebsocketSpotProvider


# Spot provider for servers based on the "xOTA" software at https://github.com/nischu/xOTA/
# The provider typically doesn't give us a lat/lon or SIG explicitly, so our own config provides this information. This
# functionality is implemented for TOTA events.
class XOTA(WebsocketSpotProvider):
    FIXED_LATITUDE = None
    FIXED_LONGITUDE = None
    SIG = None

    def __init__(self, provider_config):
        super().__init__(provider_config, provider_config["url"])
        self.FIXED_LATITUDE = provider_config["latitude"] if "latitude" in provider_config else None
        self.FIXED_LONGITUDE = provider_config["longitude"] if "longitude" in provider_config else None
        self.SIG = provider_config["sig"] if "sig" in provider_config else None

    def ws_message_to_spot(self, bytes):
        string = bytes.decode("utf-8")
        source_spot = json.loads(string)
        spot = Spot(source=self.name,
                    source_id=source_spot["id"],
                    dx_call=source_spot["stationCallSign"].upper(),
                    freq=float(source_spot["freq"]) * 1000,
                    mode=source_spot["mode"].upper(),
                    sig=self.SIG,
                    sig_refs=[SIGRef(id=source_spot["reference"]["title"], sig=self.SIG, url=source_spot["reference"]["website"])],
                    time=datetime.now(pytz.UTC).timestamp(),
                    dx_latitude=self.FIXED_LATITUDE,
                    dx_longitude=self.FIXED_LONGITUDE,
                    qrt=source_spot["state"] != "active")
        return spot