#!/usr/bin/env python3
import datetime
import requests
import json
import pytz
import sys

weather_codes = {0: "Clear Sky", 1: "Mainly Clear", 2: "Partly Cloudy", 3: "Overcast", 45: "Fog", 48: "Fog", 51: "Light Drizzle",
                 53: "Moderate Drizzle", 55: "Dense Drizzle", 56: "Light Freezing Drizzle", 57: "Freezing Drizzle",
                 61: "Light Rain", 63: "Moderate Rain", 65: "Heavy Rain", 66: "Light Freezing Rain", 67: "Freezing Rain",
                 71: "Light Snowfall", 73: "Moderate Snowfall", 75: "Heavy Snowfall", 77: "Snow Grains", 80: "Slight Rain Showers",
                 81: "Moderate Rain Showers", 82: "Heavy Rain Showers", 85: "Slight Snow Showers", 86: "Heavy Snow Showers",
                 95: "Thunderstorm", 96: "Strong Thunderstrom", 99: "Heavy Thunderstrom"}

weather_icons = {0: ('wi-day-sunny', 'wi-night-clear'), 1: ('wi-day-sunny', 'wi-night-clear'), 2: ('wi-cloudy', 'wi-night-partly-cloudy'),
                 3: ('wi-day-sunny-overcast', 'wi-night-cloudy'), 45: ('wi-day-fog', 'wi-night-fog'), 48: ('wi-day-fog', 'wi-night-fog'),
                 51: ('wi-day-rain', 'wi-night-rain'), 53: ('wi-day-rain', 'wi-night-rain'), 55: ('wi-day-rain', 'wi-night-rain'),
                 56: ('wi-day-rain', 'wi-night-rain'), 57: ('wi-day-rain', 'wi-night-rain'), 61: ('wi-day-rain', 'wi-night-rain'),
                 63: ('wi-day-rain', 'wi-night-rain'), 65: ('wi-day-rain', 'wi-night-rain'), 66: ('wi-day-rain', 'wi-night-rain'),
                 67: ('wi-day-rain', 'wi-night-rain'), 71: ('wi-day-snow', 'wi-night-snow'), 73: ('wi-day-snow', 'wi-night-snow'),
                 71: ('wi-day-snow', 'wi-night-snow'), 75: ('wi-day-snow', 'wi-night-snow'), 77: ('wi-day-snow', 'wi-night-snow'),
                 80: ('wi-day-showers', 'wi-night-showers'), 81: ('wi-day-showers', 'wi-night-showers'), 82: ('wi-day-showers', 'wi-night-showers'),
                 85: ('wi-day-snow', 'wi-night-snow'), 86: ('wi-day-snow', 'wi-night-snow'), 95: ('wi-day-storm-showers', 'wi-night-storm-showers'),
                 96: ('wi-day-storm-showers', 'wi-night-storm-showers'), 99: ('wi-day-storm-showers', 'wi-night-storm-showers')}


def get_direction_icon(degrees):
    directions = ['N', 'NNE', 'NE', 'ENE', 'E', 'ESE', 'SE',
                  'SSE', 'S', 'SSW', 'SW', 'WSW', 'W', 'NWN', 'NW', 'NNW', 'N']
    icons = {'N': 'north', 'NNE': None, 'NE': 'north-east', 'ENE': None, 'E': 'east', 'ESE': None, 'SE': 'south-east', 'SSE': None,
             'S': 'south', 'SSW': None, 'SW': 'south-west', 'WSW': None, 'W': 'west', 'NWN': None, 'NW': 'north-west', 'NNW': None, 'N': 'north'}
    points = 8
    step = 360 / points
    index = (round((degrees + step / 2) / step) * 16) / points
    direction = directions[int(index)]
    return icons[direction]


def get_lat_long(location):
    # Converts a location into latitude and longitude
    url = f"https://geocoding-api.open-meteo.com/v1/search?name={location}&count=10&language=en&format=json"
    headers = {"User-Agent": "pywttr 0.1"}
    data = requests.get(url, headers=headers).json()
    if len(data["results"]) > 1:
        return 0, 0
    else:
        latitude = data["results"][0]["latitude"]
        longitude = data["results"][0]["longitude"]
        return latitude, longitude


def get_data(latitude, longitude):
    headers = {"User-Agent": "pywttr 0.1"}
    now = datetime.datetime.now()
    now_str = now.strftime("%Y-%m-%d")
    end_str = (now + datetime.timedelta(days=7)).strftime("%Y-%m-%d")

    url_base = f"https://api.open-meteo.com/v1/forecast?latitude={latitude}&longitude={longitude}"
    url_weather_params = "&current_weather=true&hourly=temperature_2m&hourly=relativehumidity_2m&hourly=apparent_temperature&hourly=precipitation_probability&hourly=precipitation&hourly=weathercode&hourly=windspeed_10m&hourly=winddirection_10m&daily=weathercode&daily=temperature_2m_max&daily=temperature_2m_min&daily=sunrise&daily=sunset&daily=precipitation_sum&daily=precipitation_probability_max&temperature_unit=fahrenheit&windspeed_unit=mph&precipitation_unit=inch&timeformat=iso8601&past_days=0&forecast_days=7"
    url_date_time = f"&start_date={now_str}&end_date={end_str}&timezone=auto"
    url = url_base + url_weather_params + url_date_time
    data = requests.get(url, headers=headers).json()

    for i in range(len(data["hourly"]["time"])):
        data["hourly"]["time"][i] = datetime.datetime.strptime(
            data["hourly"]["time"][i], '%Y-%m-%dT%H:%M').strftime('%a %x %I:%M %p')
    for i in range(len(data["daily"]["time"])):
        data["daily"]["time"][i] = datetime.datetime.strptime(
            data["daily"]["time"][i], '%Y-%m-%d').strftime('%a %x')
    for i in range(len(data["daily"]["sunrise"])):
        data["daily"]["sunrise"][i] = datetime.datetime.strptime(
            data["daily"]["sunrise"][i], '%Y-%m-%dT%H:%M').strftime('%I:%M %p')
    for i in range(len(data["daily"]["sunset"])):
        data["daily"]["sunset"][i] = datetime.datetime.strptime(
            data["daily"]["sunset"][i], '%Y-%m-%dT%H:%M').strftime('%I:%M %p')

    data["current_weather"]["time"] = datetime.datetime.strptime(
        data["current_weather"]["time"], '%Y-%m-%dT%H:%M').strftime('%a %x %I:%M %p')

    return data


def get_current_rounded_time(tz_str):
    # Gets current time rounded down to the hour
    tz = pytz.timezone(tz_str)
    cur_time = datetime.datetime.now(tz=tz)
    cur_time_rounded = cur_time.replace(
        second=0, microsecond=0, minute=0, hour=cur_time.hour)
    return cur_time_rounded


def make_current(values):
    # Takes a list of weather data values
    # and removes items from before the current time
    # (to the nearest hour)
    ret = []
    cur_time_rounded = get_current_rounded_time()
    for val in values:
        if val["time"] >= cur_time_rounded:
            ret.append(val)
    return ret


def translate_weather_code(weather_code: int):
    return weather_codes[weather_code]