improve Ammo.list_types

lib/cannery/ammo.ex

@@ -15,31 +15,31 @@ defmodule Cannery.Ammo do
   @pack_preloads [:type]
   @type_preloads [:packs]
 
+  @type list_types_option :: {:search, String.t() | nil} | {:class, Type.class() | :all}
+  @type list_types_options :: [list_types_option()]
+
   @doc """
   Returns the list of types.
 
   ## Examples
 
-      iex> list_types(%User{id: 123}, :all)
+      iex> list_types(%User{id: 123})
       [%Type{}, ...]
 
-      iex> list_types("cool", %User{id: 123}, :shotgun)
+      iex> list_types(%User{id: 123}, search: "cool", class: :shotgun)
       [%Type{name: "My cool type", class: :shotgun}, ...]
 
   """
-  @spec list_types(User.t(), Type.class() | :all) :: [Type.t()]
-  @spec list_types(search :: nil | String.t(), User.t(), Type.class() | :all) ::
-          [Type.t()]
-  def list_types(search \\ nil, user, type)
-
-  def list_types(search, %User{id: user_id}, type) do
+  @spec list_types(User.t()) :: [Type.t()]
+  @spec list_types(User.t(), list_types_options()) :: [Type.t()]
+  def list_types(%User{id: user_id}, opts \\ []) do
     from(t in Type,
       as: :t,
       where: t.user_id == ^user_id,
       preload: ^@type_preloads
     )
-    |> list_types_filter_type(type)
-    |> list_types_filter_search(search)
+    |> list_types_filter_class(Keyword.get(opts, :class, :all))
+    |> list_types_filter_search(Keyword.get(opts, :search))
     |> Repo.all()
   end
 
@@ -72,17 +72,17 @@ defmodule Cannery.Ammo do
     )
   end
 
-  @spec list_types_filter_type(Queryable.t(), Type.class() | :all) :: Queryable.t()
-  defp list_types_filter_type(query, :rifle),
+  @spec list_types_filter_class(Queryable.t(), Type.class() | :all) :: Queryable.t()
+  defp list_types_filter_class(query, :rifle),
     do: query |> where([t: t], t.class == :rifle)
 
-  defp list_types_filter_type(query, :pistol),
+  defp list_types_filter_class(query, :pistol),
     do: query |> where([t: t], t.class == :pistol)
 
-  defp list_types_filter_type(query, :shotgun),
+  defp list_types_filter_class(query, :shotgun),
     do: query |> where([t: t], t.class == :shotgun)
 
-  defp list_types_filter_type(query, _all), do: query
+  defp list_types_filter_class(query, _all), do: query
 
   @doc """
   Returns a count of types.

lib/cannery_web/controllers/export_controller.ex

@@ -3,7 +3,7 @@ defmodule CanneryWeb.ExportController do
   alias Cannery.{ActivityLog, Ammo, Containers}
 
   def export(%{assigns: %{current_user: current_user}} = conn, %{"mode" => "json"}) do
-    types = Ammo.list_types(current_user, :all)
+    types = Ammo.list_types(current_user)
     used_counts = types |> ActivityLog.get_used_count_for_types(current_user)
     round_counts = types |> Ammo.get_round_count_for_types(current_user)
     pack_counts = types |> Ammo.get_packs_count_for_types(current_user)

lib/cannery_web/live/pack_live/form_component.ex

@@ -26,7 +26,7 @@ defmodule CanneryWeb.PackLive.FormComponent do
       socket =
       socket
       |> assign(:pack_create_limit, @pack_create_limit)
-      |> assign(:types, Ammo.list_types(current_user, :all))
+      |> assign(:types, Ammo.list_types(current_user))
       |> assign_new(:containers, fn -> Containers.list_containers(current_user) end)
 
     params =

lib/cannery_web/live/type_live/index.ex

@@ -106,7 +106,7 @@ defmodule CanneryWeb.TypeLive.Index do
        ) do
     socket
     |> assign(
-      types: Ammo.list_types(search, current_user, class),
+      types: Ammo.list_types(current_user, class: class, search: search),
       types_count: Ammo.get_types_count!(current_user)
     )
   end