#!/usr/bin/python3
# Weather module for waybar using open-meteo
import requests
import json
from datetime import datetime
import os.path
import configparser

config_file = os.path.expanduser("~/.config/weather.ini")
# Check for config file and create one if it doesn't exist
if not os.path.exists(config_file):
    f = open(config_file, "a")
    f.write("[settings]\nzip = ")
    f.close()

# Get config file
config = configparser.ConfigParser()
config.read(config_file)

# Prompt user for zip code
postal_code = config["settings"]["zip"]
if not postal_code:
    print(json.dumps({"text": "Set zip code in ~/.config/weather.ini"}))
    exit()

# Icon and text for weather codes
weather_lookup = {
    "0":  [ "", "Clear" ],
    "1":  [ "", "Mostly clear" ],
    "2":  [ "", "Partly cloudy" ],
    "3":  [ "", "Overcast"],
    "45": [ "", "Fog" ],
    "48": [ "", "Depositing rime fog" ],
    "51": [ "", "Light drizzle" ],
    "53": [ "", "Moderate drizzle" ],
    "55": [ "", "Dense drizzle" ],
    "56": [ "", "Light freezing drizzle" ],
    "57": [ "", "Dense freezing drizzle" ],
    "61": [ "", "Slight rain" ],
    "63": [ "", "Moderate rain" ],
    "65": [ "", "Heavy rain" ],
    "66": [ "", "Light freezing rain" ],
    "67": [ "", "Heavy freezing rain" ],
    "71": [ "", "Slight snow" ],
    "73": [ "", "Moderate snow" ],
    "75": [ "", "Heavy snow" ],
    "77": [ "", "Snow grains" ],
    "80": [ "", "Slight rain showers" ],
    "81": [ "", "Moderate rain showers" ],
    "82": [ "", "Violent rain showers" ],
    "85": [ "", "Slight snow showers" ],
    "86": [ "", "Heavy snow showers" ],
    "95": [ "", "Thunderstorm" ],
    "96": [ "", "Slight hailing thunderstorm" ],
    "99": [ "", "Heavy hailing thunderstorm" ]
}

# Get current time
now = datetime.now()
hour = int(now.strftime('%H'))
date = now.strftime('%Y-%m-%d')

# Take url and file path for cache file, update cache file, and return json data
def updateCache(url, file):
    data = requests.get(url).json()
    with open (file, "w") as cache:
        cache.write(json.dumps(data, indent=4))
    return data

# Get latitude and longitude for given zip code
geocode_file = os.path.expanduser("~/.cache/geocode.json")
# Try to open the geocode cache file
try:
    with open(geocode_file) as cache:
        geocode = json.load(cache)
    # If the zip code in the cache file doesn't match the config, refetch data
    if geocode["results"][0]["postcodes"][0] != postal_code:
        geocode = updateCache("https://geocoding-api.open-meteo.com/v1/search?name=" + postal_code + "&count=1&language=en&format=json", geocode_file)
except:
    # If something fails, fetch the data again
    geocode = updateCache("https://geocoding-api.open-meteo.com/v1/search?name=" + postal_code + "&count=1&language=en&format=json", geocode_file)

lat = str(geocode["results"][0]['latitude'])
lon = str(geocode["results"][0]['longitude'])
tz = geocode["results"][0]['timezone']
city = geocode["results"][0]['name']

# Get weather data
# weather = requests.get("https://api.open-meteo.com/v1/forecast?latitude=" + lat + "&longitude=" + lon + "&hourly=temperature_2m,relativehumidity_2m,weathercode,windspeed_10m,winddirection_10m,sunrise,sunset&daily=weathercode,temperature_2m_max,temperature_2m_min&temperature_unit=fahrenheit&timezone=" + tz).json()
weather_file = os.path.expanduser("~/.cache/weather.json")
weather = updateCache("https://api.open-meteo.com/v1/forecast?latitude=" + lat + "&longitude=" + lon + "&hourly=temperature_2m,relativehumidity_2m,weathercode,windspeed_10m,winddirection_10m&daily=weathercode,temperature_2m_max,temperature_2m_min,sunrise,sunset&temperature_unit=fahrenheit&timezone=" + tz, weather_file)

# Get AQI if new day
pollution_file = os.path.expanduser("~/.cache/pollution.json")
if os.path.exists(pollution_file):
    date_modified = datetime.fromtimestamp(os.path.getmtime(pollution_file)).strftime('%Y-%m-%d')
    if date == date_modified:
        with open(pollution_file) as cache:
            pollution = json.load(cache)
    else:
        pollution = updateCache("https://air-quality-api.open-meteo.com/v1/air-quality?latitude=" + lat +  "&longitude=" + lon + "&hourly=us_aqi&timezone=" + tz, pollution_file)
else:
    pollution = updateCache("https://air-quality-api.open-meteo.com/v1/air-quality?latitude=" + lat +  "&longitude=" + lon + "&hourly=us_aqi&timezone=" + tz, pollution_file)

# Print header for daily weather
tooltip="<span color='#8fa1be' font_size='16pt'>Today</span>\n"

# Convert degrees to cardinal direction
def degToCompass(num):
    val=int((num/22.5)+.5)
    arr=["N","NNE","NE","ENE","E","ESE", "SE", "SSE","S","SSW","SW","WSW","W","WNW","NW","NNW"]
    return arr[(val % 16)]

# Get data for current hour
time_now = weather["hourly"]["time"][hour]
code_now = str(weather["hourly"]["weathercode"][hour])
temp_now = str(int(weather["hourly"]["temperature_2m"][hour]))
humidity_now = str(weather["hourly"]["relativehumidity_2m"][hour])
windspeed_now = str(weather["hourly"]["windspeed_10m"][hour])
winddirection_now = degToCompass(weather["hourly"]["winddirection_10m"][hour])
sunrise = int(datetime.strptime(weather["daily"]["sunrise"][0], "%Y-%m-%dT%H:%M").strftime("%H"))
sunset = int(datetime.strptime(weather["daily"]["sunset"][0], "%Y-%m-%dT%H:%M").strftime("%H"))
aqi = int(pollution["hourly"]["us_aqi"][hour])

# Air quality
if aqi <= 50:
    quality = "Good"
elif aqi <= 100:
    quality = "Moderate"
elif aqi <= 150:
    quality = "Unhealty"
elif aqi <= 200:
    quality = "Unhealthy"
elif aqi <= 300:
    quality = "Very unhealthy"
elif aqi <= 500:
    quality = "Hazardous"

# Add data to tooltip
tooltip += "City: {}\n".format(city)
tooltip += "Description: {}\n".format(weather_lookup[code_now][1])
tooltip += "Temperature: {}°F\n".format(temp_now)
tooltip += "Humidity: {}%\n".format(humidity_now)
tooltip += "Wind: {}mph {}\n".format(windspeed_now, winddirection_now)
tooltip += "Air quality: {}\n".format(quality)

# Print header for weekly forecast
tooltip+="\n<span color='#8fa1be' font_size='16pt'>7 day forecast</span>\n"

# Split weekly data
for x in range(7):
    date = datetime.strptime(weather["daily"]["time"][x], "%Y-%m-%d")
    dow = date.strftime("%A")
    code = str(weather["daily"]["weathercode"][x])
    temp_max = str(int(weather["daily"]["temperature_2m_max"][x]))
    temp_min = str(int(weather["daily"]["temperature_2m_min"][x]))
    # Add formatted line to tooltip
    tooltip += "{}: {}/{} {}\n".format(dow[:2], temp_max, temp_min, weather_lookup[code][1])

# Get boolean from config if exists, otherwise enable night icons
try:
    night_icons = config.getboolean('settings', 'night_icons')
except:
    night_icons = 1
if night_icons == 1:
    # Change icon if night time
    if hour < sunrise or hour > sunset:
        weather_lookup["0"][0] = ""

# Print data formatted for waybar
print(json.dumps({"text": "{} {}°F".format(weather_lookup[code_now][0], temp_now), "tooltip": tooltip.rstrip() }))