Mix.install([
  {:flow, "~> 1.2.4"},
  {:ex_cldr_collation, "~> 0.7.2"}
])

defmodule Brc do
  require Logger

  def run(path) do
    compute_file(path)
    |> output_results()
  end

  def compute_file(path) do
    Logger.info("computing from file #{path}")

    File.stream!(path)
    |> Flow.from_enumerable()
    |> Flow.map(&parse_line/1)
    |> Flow.partition(key: {:elem, 0})
    |> Flow.reduce(fn -> %{} end, &update_city_temp_map/2)
  end

  def update_city_temp_map({city, temp}, acc) do
    Map.update(acc, city, [temp, temp, temp],
      fn [prev_min, prev_max, mean] ->
        [
          min(prev_min, temp),
          max(prev_max, temp),
          (mean + temp) / 2
        ]
      end)
  end

  def parse_line(line) do
    [city, temp_text] = String.trim(line) |> String.split(";")
    {city, String.to_float(temp_text)}
  end

  def output_results(results) do
    for {city, [min, max, mean]} <- results do
      "#{city}=#{min}/#{max}/#{Float.round(mean, 1)}"
    end
    |> Enum.sort(Cldr.Collation)
    |> Enum.join(",")
    |> then(fn out -> "{" <> out <> "}" end)
    |> IO.puts()
  end
end

{path, _} = System.argv() |> List.pop_at(0, "./measurements.txt")
Brc.run(path)