from logging import log
import sys
import traceback
import json
import time
import datetime
import requests
import urllib3

from datetime import datetime, timedelta
from random import randrange

from modules.common.module import BotModule

urllib3.disable_warnings()

# API docs at: https://gitlab.com/lemoidului/ogn-flightbook/-/blob/master/doc/API.md
class FlightBook:
    def __init__(self):
        self.base_url = 'https://flightbook.glidernet.org/api'
        self.AC_TYPES = [ '?', 'Glider', 'Towplane', \
            'Helicopter', 'Parachute', 'Drop plane', 'Hang glider', \
            'Paraglider', 'Powered', 'Jet', 'UFO', 'Balloon', \
            'Airship', 'UAV', '?', 'Static object' ]
        self.logged_flights = dict() # station -> [index of flight]
        self.device_cache = dict() # Registration -> [address, CN]

    def get_flights(self, icao):
        log_url = f'{self.base_url}/logbook/{icao}'
        data = None
        with requests.Session() as session:
            response = session.get(log_url, headers={'Connection': 'close'}, verify=False)
            data = response.json()

        # print(json.dumps(data, sort_keys=True, indent=4))
        self.update_device_cache(data)
        return data

    def update_device_cache(self, data):
        devices = data['devices']
        for device in devices:
            if device["address"] and device["registration"]:
                cache_entry = [device["address"], device["competition"]]
                self.device_cache[device["registration"]] = cache_entry

    def address_for_registration(self, registration):
        for reg in self.device_cache.keys():
            if reg.lower() == registration.lower():
                return self.device_cache[reg][0]
        return None

    def address_for_cn(self, cn):
        for reg in self.device_cache.keys():
            if self.device_cache[reg][1] == cn.upper():
                return self.device_cache[reg][0]
        return None

    def format_time(self, time):
        if not time:
            return '··:··'
        time = time.replace('h', ':')
        return time

    def flight2string(self, flight, data):
        devices = data['devices']
        device = devices[flight['device']]
        start = self.format_time(flight["start"])
        end = self.format_time(flight["stop"])
        duration = '     '
        if flight["duration"]:
            duration = time.strftime('%H:%M', time.gmtime(flight["duration"]))
        maxalt = ''
        if flight["max_alt"]:
            maxalt = str(flight["max_alt"]) + 'm'
        
        identity = f'{device.get("registration") or ""} {device.get("aircraft") or ""} {device.get("competition") or ""} {maxalt}'
        identity = ' '.join(identity.split())
        return f'{start} - {end} {duration} {identity}'

    def print_flights(self, data, showtow=False):
        print(f'✈ Flights at {data["airfield"]["name"]} ({data["airfield"]["code"]}) {data["date"]}:')
        flights = data['flights']
        for flight in flights:
            if not showtow and flight["towing"]:
                continue
            print(self.flight2string(flight, data))

    def test():
        fb = FlightBook()
        data = fb.get_flights('LFMX')
        fb.print_flights(data)

class MatrixModule(BotModule):
    def __init__(self, name):
        super().__init__(name)
        self.service_name = 'FLOG'
        self.station_rooms = dict()  # Roomid -> ogn station
        self.live_rooms = []     # Roomid's with live enabled
        self.logged_flights = dict() # Station -> number of flights
        self.first_poll = True
        self.enabled = False
        self.fb = FlightBook()

    def matrix_start(self, bot):
        super().matrix_start(bot)
        self.add_module_aliases(bot, ['sar'])

    async def matrix_poll(self, bot, pollcount):
        if pollcount % (6 * 5) == 0:  # Poll every 5 min
            await self.poll_implementation(bot)

    async def poll_implementation(self, bot):
        for roomid in self.live_rooms:
            station = self.station_rooms[roomid]
            data = self.fb.get_flights(station)
            if not data:
                self.logger.warning(f"FLOG: Failed to get flights at {station}!")
                return
            flights = data['flights']

            if len(flights) == 0 or (not station in self.logged_flights):
                self.logged_flights[station] = []
                #print('Reset flight count for station ' + station)
#            else:
#                print(f'Got {len(flights)} flights at {station}')

            flightindex = 0
            for flight in flights:
                if flight["towing"]:
                    continue
                if flight["stop"]:
                    if not flightindex in self.logged_flights[station]:
                        if not self.first_poll:
                            await bot.send_text(bot.get_room_by_id(roomid), self.fb.flight2string(flight, data))
                        self.logged_flights[station].append(flightindex)
                        # print(f'Logged flights at {station} now {self.logged_flights[station]}')
                flightindex = flightindex + 1
        self.first_poll = False

    async def matrix_message(self, bot, room, event):
        args = event.body.split()
        if len(args) == 1 and args[0] == "!flog":
            if room.room_id in self.station_rooms:
                station = self.station_rooms[room.room_id]
                await self.show_flog(bot, room, station)
            else:
                await bot.send_text(room, 'No OGN station set for this room - set it first.')

        elif len(args) == 2 and args[0] == "!flog":
            if args[1] == 'rmstation':
                bot.must_be_admin(room, event)
                del self.station_rooms[room.room_id]
                self.live_rooms.remove(room.room_id)
                await bot.send_text(room, f'Cleared OGN station for this room')

            elif args[1] == 'status':
                print(self.logged_flights)
                print(self.fb.device_cache)
                bot.must_be_admin(room, event)
                await bot.send_text(room, f'OGN station for this room: {self.station_rooms.get(room.room_id)}, live updates enabled: {room.room_id in self.live_rooms}')

            elif args[1] == 'poll':
                bot.must_be_admin(room, event)
                await self.poll_implementation(bot)

            elif args[1] == 'live':
                bot.must_be_admin(room, event)
                self.live_rooms.append(room.room_id)
                bot.save_settings()
                await bot.send_text(room, f'Sending live updates for station {self.station_rooms.get(room.room_id)} to this room')

            elif args[1] == 'rmlive':
                bot.must_be_admin(room, event)
                self.live_rooms.remove(room.room_id)
                bot.save_settings()
                await bot.send_text(room, f'Not sending live updates for station {self.station_rooms.get(room.room_id)} to this room anymore')

            else:
                # Assume parameter is a station name
                station = args[1]
                await self.show_flog(bot, room, station)
        elif len(args) == 2 and args[0] == "!sar":
            registration = args[1]
            address = self.fb.address_for_registration(registration)
            if not address:
                cn = args[1]
                address = self.fb.address_for_cn(cn)

            coords = None
            if address:
                coords = self.get_coords_for_address(address)
            if coords:
                await bot.send_location(room, f'{registration} ({coords["utc"]})', coords["lat"], coords["lng"])
            else:
                await bot.send_text(room, f'No Flarm ID found for {registration}!')

        elif len(args) == 3 and args[0] == "!flog":
            if args[1] == 'station':
                bot.must_be_admin(room, event)

                station = args[2]
                self.station_rooms[room.room_id] = station
                self.logger.info(f'Station now for this room {self.station_rooms.get(room.room_id)}')

                bot.save_settings()
                await bot.send_text(room, f'Set OGN station {station} to this room')


    def get_coords_for_address(self, address):
        # https://flightbook.glidernet.org/api/live/address/~91DADF5B86
        url = f'{self.fb.base_url}/live/address/{address}'
        data = None
        with requests.Session() as session:
            response = session.get(url, headers={'Connection': 'close'}, verify=False)
            data = response.json()

        # print(json.dumps(data, sort_keys=True, indent=4))
        return data


    def text_flog(self, data, showtow):
        out = ""
        if len(data["flights"]) == 0:
            out = f'No known flights today at {data["airfield"]["name"]}'
        else:
            out = f'Flights at {data["airfield"]["name"]} ({data["airfield"]["code"]}) {data["date"]}:' + "\n"
            flights = data['flights']
            for flight in flights:
                if not showtow and flight["towing"]:
                    continue
                out = out + self.fb.flight2string(flight, data) + "\n"
        return out

    def html_flog(self, data, showtow):
        out = ""
        if len(data["flights"]) == 0:
            out = f'No known flights today at {data["airfield"]["name"]}'
        else:
            out = f'<b>✈ Flights at {data["airfield"]["name"]} ({data["airfield"]["code"]}) {data["date"]}:' + "</b>\n"
            flights = data['flights']
            out = out + "<ul>"
            for flight in flights:
                if not showtow and flight["towing"]:
                    continue
                out = out + "<li>" + self.fb.flight2string(flight, data) + "</li>\n"
            out = out + "</ul>"
        return out

    async def show_flog(self, bot, room, station):
        data = self.fb.get_flights(station)
        if data:
            await bot.send_html(room, self.html_flog(data, False), self.text_flog(data, False))
        else:
            await bot.send_text(room, f"Failed to get flight log for {station}")

    def get_settings(self):
        data = super().get_settings()
        data['station_rooms'] = self.station_rooms
        data['live_rooms'] = self.live_rooms
        return data

    def set_settings(self, data):
        super().set_settings(data)
        if data.get('station_rooms'):
            self.station_rooms = data['station_rooms']
        if data.get('live_rooms'):
            self.live_rooms = data['live_rooms']

    def help(self):
        return ('Open Glider Network Field Log')