import io
import logging
import pickle
import time
from datetime import datetime
from typing import List

import schedule
from PIL import Image
from redis import Redis

from lib.tecmap import plot_tec_map

logging.basicConfig(level=logging.INFO)

LAT_RANGE_MIN = -90
LAT_RANGE_MAX = 90
LON_RANGE_MIN = -180
LON_RANGE_MAX = 180


def main():
    redis = Redis(host='localhost', port=6379, db=0)

    utc_hr = datetime.utcnow().hour
    logging.info(f'Generating plot for hour {utc_hr}')

    ionex_data: List = pickle.loads(redis.get('tecmap_data'))
    while ionex_data is None:
        logging.warning('Redis has not been populated yet. Is cache.py running? Sleeping 10s...')
        time.sleep(10)

    for tecmap, epoch in ionex_data:
        if epoch.hour == utc_hr:
            plt = plot_tec_map(tecmap, [float(LON_RANGE_MIN), float(LON_RANGE_MAX)], [float(LAT_RANGE_MIN), float(LAT_RANGE_MAX)], timestamp=epoch)[1]
            buf = io.BytesIO()
            plt.savefig(buf, format='png', bbox_inches='tight', pad_inches=0, dpi=110)
            plt.close()
            del plt

            buf.seek(0)
            img = Image.open(buf)
            buf = io.BytesIO()
            img.save(buf, format='PNG')

            redis.set('global_map', buf.getvalue())
            buf.close()
    logging.info(f'Finished hour {utc_hr}')


if __name__ == '__main__':
    main()
    schedule.every().hour.at(':00').do(main)
    while True:
        schedule.run_pending()
        time.sleep(1)