Pick color extraction strategy from dropdown

lib/music_library/colors.ex

@@ -0,0 +1,11 @@
+defmodule MusicLibrary.Colors do
+  alias MusicLibrary.Colors.{ColorFrequencyExtractor, EdgeWeightedExtractor}
+
+  def extract_colors(image_data, :fast) do
+    ColorFrequencyExtractor.extract_dominant_colors(image_data)
+  end
+
+  def extract_colors(image_data, :slow) do
+    EdgeWeightedExtractor.extract_dominant_colors(image_data)
+  end
+end

lib/music_library/records.ex

@@ -6,7 +6,6 @@ defmodule MusicLibrary.Records do
   import Ecto.Query, warn: false
 
   alias MusicLibrary.Artists
-  alias MusicLibrary.Colors.EdgeWeightedExtractor
   alias MusicLibrary.Records.{ArtistRecord, Cover, Record, SearchParser}
   alias MusicLibrary.{BackgroundRepo, Repo, Worker}
 
@@ -255,18 +254,12 @@ defmodule MusicLibrary.Records do
     |> BackgroundRepo.insert()
   end
 
-  def generate_dominant_colors(record) do
-    with {:ok, colors} <- EdgeWeightedExtractor.extract_dominant_colors(record.cover_data) do
-      update_record(record, %{"dominant_colors" => colors})
-    end
-  end
-
-  def generate_dominant_colors_async(record) do
+  def extract_colors_async(record, method) do
     meta = %{title: record.title, artists: Enum.map(record.artists, & &1.name)}
-    params = %{"id" => record.id}
+    params = %{"id" => record.id, "method" => method}
 
     params
-    |> Worker.GenerateDominantColors.new(meta: meta)
+    |> Worker.ExtractColors.new(meta: meta)
     |> BackgroundRepo.insert()
   end
 
@@ -329,7 +322,7 @@ defmodule MusicLibrary.Records do
 
   def create_record(attrs \\ %{}) do
     with {:ok, record} <- do_create_record(attrs) do
-      generate_dominant_colors_async(record)
+      extract_colors_async(record, :fast)
 
       record
       |> Record.artist_ids()

lib/music_library/worker/extract_colors.ex

@@ -0,0 +1,16 @@
+defmodule MusicLibrary.Worker.ExtractColors do
+  use Oban.Worker, queue: :heavy_writes, max_attempts: 3
+
+  alias MusicLibrary.{Colors, Records}
+
+  @impl Oban.Worker
+  def perform(%Oban.Job{args: %{"id" => record_id, "method" => method}}) do
+    record = MusicLibrary.Records.get_record!(record_id)
+    method = String.to_existing_atom(method)
+
+    with {:ok, colors} <- Colors.extract_colors(record.cover_data, method),
+         {:ok, updated_record} <- Records.update_record(record, %{dominant_colors: colors}) do
+      MusicLibrary.Records.notify_update(updated_record)
+    end
+  end
+end

lib/music_library/worker/generate_dominant_colors.ex

@@ -1,12 +0,0 @@
-defmodule MusicLibrary.Worker.GenerateDominantColors do
-  use Oban.Worker, queue: :heavy_writes, max_attempts: 3
-
-  @impl Oban.Worker
-  def perform(%Oban.Job{args: %{"id" => record_id}}) do
-    record = MusicLibrary.Records.get_record!(record_id)
-
-    with {:ok, updated_record} <- MusicLibrary.Records.generate_dominant_colors(record) do
-      MusicLibrary.Records.notify_update(updated_record)
-    end
-  end
-end