adventofcode/adventofcode.nim

import
  std/httpclient,
  std/os,
  std/strformat,
  std/strutils
  
proc getInput*(year, day: int): string =
  try:
    let input = readFile(getCurrentDir() / "input.txt")
    
    return input.strip(chars = {'\n'})
  except IOError:
    echo "adventofcode> Missing input.txt. Downloading..."
    let session = os.getEnv("SESSION")
    if session == "":
      echo "adventofcode> Unable to download without the SESSION environment variable."
      quit 1

    let client = newHttpClient()
    client.headers = newHttpHeaders({ "Cookie": fmt"session={session}" })

    let input = client.getContent fmt"https://adventofcode.com/{year}/day/{day}/input"
    
    writeFile(getCurrentDir() / "input.txt", input)
    
    return input.strip(chars = {'\n'})