defmodule CanneryWeb.ExportController do
  use CanneryWeb, :controller
  alias Cannery.{ActivityLog, Ammo, Containers}

  def export(%{assigns: %{current_user: current_user}} = conn, %{"mode" => "json"}) do
    ammo_types = Ammo.list_ammo_types(current_user)
    used_counts = ammo_types |> ActivityLog.get_used_count_for_ammo_types(current_user)
    round_counts = ammo_types |> Ammo.get_round_count_for_ammo_types(current_user)
    ammo_group_counts = ammo_types |> Ammo.get_ammo_groups_count_for_types(current_user)

    total_ammo_group_counts =
      ammo_types |> Ammo.get_ammo_groups_count_for_types(current_user, true)

    average_costs = ammo_types |> Ammo.get_average_cost_for_ammo_types(current_user)

    ammo_types =
      ammo_types
      |> Enum.map(fn %{id: ammo_type_id} = ammo_type ->
        ammo_type
        |> Jason.encode!()
        |> Jason.decode!()
        |> Map.merge(%{
          "average_cost" => Map.get(average_costs, ammo_type_id),
          "round_count" => Map.get(round_counts, ammo_type_id, 0),
          "used_count" => Map.get(used_counts, ammo_type_id, 0),
          "ammo_group_count" => Map.get(ammo_group_counts, ammo_type_id, 0),
          "total_ammo_group_count" => Map.get(total_ammo_group_counts, ammo_type_id, 0)
        })
      end)

    ammo_groups = Ammo.list_ammo_groups(nil, true, current_user)
    used_counts = ammo_groups |> ActivityLog.get_used_counts(current_user)
    original_counts = ammo_groups |> Ammo.get_original_counts(current_user)
    cprs = ammo_groups |> Ammo.get_cprs(current_user)
    percentages_remaining = ammo_groups |> Ammo.get_percentages_remaining(current_user)

    ammo_groups =
      ammo_groups
      |> Enum.map(fn %{id: ammo_group_id} = ammo_group ->
        ammo_group
        |> Jason.encode!()
        |> Jason.decode!()
        |> Map.merge(%{
          "used_count" => Map.get(used_counts, ammo_group_id),
          "percentage_remaining" => Map.fetch!(percentages_remaining, ammo_group_id),
          "original_count" => Map.get(original_counts, ammo_group_id),
          "cpr" => Map.get(cprs, ammo_group_id)
        })
      end)

    shot_groups = ActivityLog.list_shot_groups(current_user)

    containers =
      Containers.list_containers(current_user)
      |> Enum.map(fn container ->
        ammo_group_count = container |> Ammo.get_ammo_groups_count_for_container!(current_user)
        round_count = container |> Ammo.get_round_count_for_container!(current_user)

        container
        |> Jason.encode!()
        |> Jason.decode!()
        |> Map.merge(%{
          "ammo_group_count" => ammo_group_count,
          "round_count" => round_count
        })
      end)

    json(conn, %{
      user: current_user,
      ammo_types: ammo_types,
      ammo_groups: ammo_groups,
      shot_groups: shot_groups,
      containers: containers
    })
  end
end