Add functions to generate dominant colors

This commit is contained in:
Claudio Ortolina
2025-06-07 20:13:19 +01:00
parent c5e47cbc8c
commit 378992e632
2 changed files with 28 additions and 1 deletions
+16 -1
View File
@@ -6,7 +6,7 @@ defmodule MusicLibrary.Records do
import Ecto.Query, warn: false
alias MusicLibrary.Artists
alias MusicLibrary.Records.{ArtistRecord, Cover, Record, SearchParser}
alias MusicLibrary.Records.{ArtistRecord, Cover, DominantColors, Record, SearchParser}
alias MusicLibrary.{BackgroundRepo, Repo, Worker}
def essential_fields do
@@ -254,6 +254,21 @@ defmodule MusicLibrary.Records do
|> BackgroundRepo.insert()
end
def generate_dominant_colors(record) do
with {:ok, colors} <- DominantColors.extract_dominant_colors(record.cover_data) do
update_record(record, %{"dominant_colors" => colors})
end
end
def generate_dominant_colors_async(record) do
meta = %{title: record.title, artists: Enum.map(record.artists, & &1.name)}
params = %{"id" => record.id}
params
|> Worker.GenerateDominantColors.new(meta: meta)
|> BackgroundRepo.insert()
end
def resize_cover(record) do
{:ok, thumb_data} = Cover.resize(record.cover_data)
@@ -0,0 +1,12 @@
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