#!/usr/bin/env python3

import sys
import requests
from keyring import get_password

if len(sys.argv) != 3:
    print(f"Usage: {sys.argv[0]} <year> <day>")
    sys.exit(1)

year = sys.argv[1]
day = sys.argv[2]

session_cookie = get_password('aoc-session-cookie', 'felixm')
assert session_cookie is not None

url = f"https://adventofcode.com/{year}/day/{day}/input"
cookies = {'session': session_cookie}
response = requests.get(url, cookies=cookies)

if response.status_code == 200:
    filename = f"d{day}.txt"
    with open(filename, 'w') as file:
        file.write(response.text)
    print(f"Year {year} {filename} written.")
else:
    print(f"Error: Failed to download input (HTTP status code {response.status_code}).")