advent-of-code-2022/download-input.py

#!/usr/bin/python3

from abc import ABC, abstractmethod
import argparse
from os import getenv
from os.path import exists, join
import shutil
from tempfile import TemporaryFile
from urllib.request import Request, urlopen
import webbrowser


class Language(ABC):
    input_path: str
    src_path: str

    @abstractmethod
    def src_file_name(self, day):
        pass

    @abstractmethod
    def format_src(self, day):
        pass


class Kotlin(Language):
    src_path = 'kotlin/src'
    input_path = src_path

    def src_file_name(self, day):
        return "Day{}.kt".format(day)

    def format_src(self, day):
        return """fun main() {{
    fun part1(input: List<String>): Int {{
        return 0
    }}

    fun part2(input: List<String>): Int {{
        return 0
    }}

    val testInput = readInput("Day{day}_test")
    check(part1(testInput) == 0)
    check(part2(testInput) == 0)

    val input = readInput("Day{day}")
    println(part1(input))
    println(part2(input))
}}
""".format(day=day)


class Python(Language):
    input_path = 'python'
    src_path = 'python'

    def src_file_name(self, day):
        return "day{}.py".format(day)

    def format_src(self, day):
        return """#!/usr/bin/env python3

from util import read_input


def part1(input):
    pass


def part2(input):
    pass


test_input = read_input("Day{day}_test")
assert 0 == part1(test_input)
assert 0 == part2(test_input)

real_input = read_input('Day{day}')
print(part1(real_input))
print(part2(real_input))
""".format(day=day)


class Rust(Language):
    input_path = 'rust'
    src_path = 'rust/src/bin'

    def src_file_name(self, day):
        return "day{}.rs".format(day)

    def format_src(self, day):
        return """use rust::util::read_input;

fn part1(input: &Vec<String>) -> u32 {{
    0
}}

fn part2(input: &Vec<String>) -> u32 {{
    0
}}

fn main() {{
    let test_input = read_input("Day{day}_test").expect("Failed to open test input");
    assert_eq!(0, part1(&test_input));
    assert_eq!(0, part2(&test_input));

    let input = read_input("Day{day}").expect("Failed to open input");
    println!("{{:?}}", part1(&input));
    println!("{{:?}}", part2(&input));
}}
""".format(day=day)


session_cookie = getenv('AOC_SESSION')
if not session_cookie:
    print('AOC_SESSION environment variable not set, aborting')
    exit(1)

if not session_cookie.startswith('session='):
    session_cookie = 'session=' + session_cookie

parser = argparse.ArgumentParser()
parser.add_argument('-d', '--day', type=int, required=True)
parser.add_argument('--kotlin', action='store_true')
parser.add_argument('--python', action='store_true')
parser.add_argument('--rust', action='store_true')
args = parser.parse_args()

challenge_url = "https://adventofcode.com/2022/day/{}".format(args.day)
input_url = "{}/input".format(challenge_url)

webbrowser.open(challenge_url)

languages = []

if args.kotlin:
    languages.append(Kotlin())

if args.python:
    languages.append(Python())

if args.rust:
    languages.append(Rust())

if len(languages) == 0:
    print("error: at least one language is required")
    parser.print_usage()
    exit(1)

daystr = str(args.day).zfill(2)
day_file_base = 'Day{}'.format(daystr)
day_file_test = '{}_test.txt'.format(day_file_base)

request = Request(input_url, headers={'Cookie': session_cookie})
tmp_file = TemporaryFile()
with urlopen(request) as response:
    shutil.copyfileobj(response, tmp_file)

for language in languages:
    tmp_file.seek(0)
    with open(join(language.input_path, day_file_test), 'a'):
        pass

    with open(join(language.input_path, '{}.txt'.format(day_file_base)), 'wb') as input_file:
        shutil.copyfileobj(tmp_file, input_file)

    src_file = join(language.src_path, language.src_file_name(daystr))
    if not exists(src_file):
        with open(src_file, 'w') as src:
            src.write(language.format_src(str(args.day).zfill(2)))
tmp_file.close()