125 lines
3.3 KiB
Elixir
125 lines
3.3 KiB
Elixir
defmodule Mix.Tasks.Aoc do
|
|
use Mix.Task
|
|
|
|
defp module(1), do: Aoc2023.Day1
|
|
# [MODULE INSERTION POINT]
|
|
|
|
defp base_dir(), do: System.get_env("AOC_BASE")
|
|
|
|
defp tests(day) do
|
|
File.ls!(tests_dir(day)) |> Enum.filter(&(not String.contains?(&1, ".")))
|
|
end
|
|
|
|
defp read(path) do
|
|
f = File.open!(path, [:utf8])
|
|
contents = IO.read(f, :eof)
|
|
:ok = File.close(f)
|
|
contents
|
|
end
|
|
|
|
defp test(day, test) do
|
|
f = "#{tests_dir(day)}#{test}"
|
|
{
|
|
test,
|
|
read(f),
|
|
if File.exists?(f <> ".1") do read(f <> ".1") else nil end,
|
|
if File.exists?(f <> ".2") do read(f <> ".2") else nil end
|
|
}
|
|
end
|
|
|
|
defp tests_dir(day), do: "#{base_dir()}/tests/day#{day}/"
|
|
|
|
defp create_file(path, contents) do
|
|
File.mkdir_p!(Path.dirname(path))
|
|
f = File.open!(path, [:exclusive, :utf8])
|
|
IO.write(f, contents)
|
|
:ok = File.close(f)
|
|
end
|
|
|
|
def run(params) do
|
|
{opts, params} = params |> OptionParser.parse!(strict: [day: :integer])
|
|
day = opts |> Keyword.get(:day, Date.utc_today().day)
|
|
run(day, params)
|
|
end
|
|
|
|
defp run(day, ["new"]) do
|
|
create_file("#{base_dir()}/lib/day#{day}.ex", """
|
|
defmodule Aoc2023.Day#{day} do
|
|
use Aoc2023
|
|
def parse(input) do
|
|
input
|
|
end
|
|
def part1(input) do
|
|
input
|
|
end
|
|
def part2(input) do
|
|
input
|
|
end
|
|
end
|
|
""")
|
|
this_file = "#{base_dir()}/lib/mix_tasks.ex"
|
|
contents = read(this_file) |> String.replace("\n # [MODULE INSERTION POINT]", "\n defp module(#{day}), do: Aoc2023.Day#{day}\n # [MODULE INSERTION POINT]")
|
|
f = File.open!(this_file, [:write, :utf8])
|
|
IO.write(f, contents)
|
|
:ok = File.close(f)
|
|
end
|
|
|
|
defp run(day, ["test", "new"]) do
|
|
last = tests(day) |> Enum.map(&String.to_integer(&1)) |> Enum.max(empty_fallback: fn -> 1 end)
|
|
create_file("#{tests_dir(day)}#{last + 1}", IO.read(:stdio, :eof))
|
|
end
|
|
defp run(day, ["test", "expected", test, part]) do
|
|
create_file("#{tests_dir(day)}#{test}.#{part}", IO.read(:stdio, :eof))
|
|
end
|
|
|
|
defp run(day, ["test"]) do
|
|
mod = module(day)
|
|
tests = tests(day)
|
|
tests |> Enum.map(&(test(day, &1))) |> Enum.map(fn ({test, input, p1e, p2e}) ->
|
|
parsed = mod.parse(input)
|
|
p1 = mod.part1(parsed)
|
|
if p1e do
|
|
if to_string(p1) != p1e do
|
|
IO.puts "Failed at #{test}.1: expected #{p1e}, got:"
|
|
dbg(p1)
|
|
end
|
|
else
|
|
IO.puts "Test #{test}.1"
|
|
dbg(p1)
|
|
end
|
|
p2 = mod.part2(parsed)
|
|
if p2e do
|
|
if to_string(p2) != p2e do
|
|
IO.puts "Failed at #{test}.2: expected #{p2e}, got:"
|
|
dbg(p2)
|
|
end
|
|
else
|
|
IO.puts "Test #{test}.2"
|
|
dbg(p2)
|
|
end
|
|
end)
|
|
end
|
|
|
|
defp run(day, ["run"]) do
|
|
run(day, ["fetch"])
|
|
run_file(day, "#{base_dir()}/inputs/day#{day}")
|
|
end
|
|
|
|
defp run(day, ["run", "-"]) do
|
|
run_file(day)
|
|
end
|
|
|
|
defp run(day, ["fetch"]) do
|
|
file = "#{base_dir()}/inputs/day#{day}"
|
|
if not File.exists? file do
|
|
HTTPoison.start()
|
|
resp = HTTPoison.get!("https://adventofcode.com/2023/day/#{day}/input", [ "user-agent": "aoc-ex by ben@soroos.net", cookie: "session=#{System.get_env("AOC_SESSION")}" ])
|
|
create_file(file, resp.body)
|
|
end
|
|
end
|
|
|
|
defp run_file(day, file \\ nil) do
|
|
module(day).main(if file, do: read(file), else: IO.read(:stdio, :eof))
|
|
end
|
|
end
|