EXP: cover search via Brave API
This commit is contained in:
@@ -0,0 +1,13 @@
|
||||
defmodule BraveSearch do
|
||||
alias BraveSearch.API
|
||||
|
||||
def search_images(query, opts \\ []) do
|
||||
API.search_images(query, opts, config())
|
||||
end
|
||||
|
||||
def download_image(url) do
|
||||
API.download_image(url, config())
|
||||
end
|
||||
|
||||
defp config, do: BraveSearch.Config.resolve(:music_library)
|
||||
end
|
||||
@@ -0,0 +1,87 @@
|
||||
defmodule BraveSearch.API do
|
||||
@moduledoc """
|
||||
Interface to the Brave Search API.
|
||||
"""
|
||||
|
||||
require Logger
|
||||
|
||||
def search_images(query, opts, config) do
|
||||
params = [q: query, count: Keyword.get(opts, :count, 20)]
|
||||
|
||||
case config
|
||||
|> new_request()
|
||||
|> Req.merge(url: "/res/v1/images/search", params: params)
|
||||
|> get_request() do
|
||||
{:ok, body} ->
|
||||
results =
|
||||
body
|
||||
|> Map.get("results", [])
|
||||
|> Enum.map(fn result ->
|
||||
%{
|
||||
thumbnail_url: get_in(result, ["thumbnail", "src"]),
|
||||
image_url: get_in(result, ["properties", "url"]),
|
||||
title: Map.get(result, "title", ""),
|
||||
source: Map.get(result, "source", "")
|
||||
}
|
||||
end)
|
||||
|
||||
{:ok, results}
|
||||
|
||||
{:error, reason} ->
|
||||
{:error, reason}
|
||||
end
|
||||
end
|
||||
|
||||
def download_image(url, config) do
|
||||
case Req.new(url: url, max_retries: 1, user_agent: config.user_agent)
|
||||
|> Req.Request.merge_options(config.req_options)
|
||||
|> Req.Request.append_request_steps(log_attempt: &log_attempt/1)
|
||||
|> Req.Request.append_response_steps(log_error: &log_error/1)
|
||||
|> get_request() do
|
||||
{:ok, data} -> {:ok, data}
|
||||
{:error, _reason} -> {:error, :download_failed}
|
||||
end
|
||||
end
|
||||
|
||||
defp new_request(config) do
|
||||
Req.new(
|
||||
base_url: "https://api.search.brave.com",
|
||||
max_retries: 1,
|
||||
user_agent: config.user_agent,
|
||||
headers: %{"x-subscription-token" => config.api_key}
|
||||
)
|
||||
|> Req.Request.merge_options(config.req_options)
|
||||
|> Req.Request.append_request_steps(log_attempt: &log_attempt/1)
|
||||
|> Req.Request.append_response_steps(log_error: &log_error/1)
|
||||
end
|
||||
|
||||
defp get_request(request) do
|
||||
case Req.get(request) do
|
||||
{:ok, response} when response.status == 200 ->
|
||||
{:ok, response.body}
|
||||
|
||||
{:ok, response} ->
|
||||
{:error, response.body}
|
||||
|
||||
error ->
|
||||
error
|
||||
end
|
||||
end
|
||||
|
||||
defp log_attempt(request) do
|
||||
url = URI.to_string(request.url)
|
||||
Logger.debug("Fetching data from #{url}")
|
||||
request
|
||||
end
|
||||
|
||||
defp log_error({request, response}) do
|
||||
if response.status in 400..499 or response.status in 500..599 do
|
||||
Logger.error(fn ->
|
||||
url = URI.to_string(request.url)
|
||||
"Failed to fetch data from #{url}, reason: #{inspect(response.body)}"
|
||||
end)
|
||||
end
|
||||
|
||||
{request, response}
|
||||
end
|
||||
end
|
||||
@@ -0,0 +1,39 @@
|
||||
defmodule BraveSearch.Config do
|
||||
@type t :: %__MODULE__{
|
||||
api_key: String.t(),
|
||||
user_agent: String.t(),
|
||||
req_options: Keyword.t()
|
||||
}
|
||||
|
||||
@enforce_keys [:api_key]
|
||||
defstruct api_key: "",
|
||||
user_agent: "change me",
|
||||
req_options: []
|
||||
|
||||
@schema NimbleOptions.new!(
|
||||
api_key: [
|
||||
type: :string,
|
||||
required: true
|
||||
],
|
||||
user_agent: [
|
||||
type: :string,
|
||||
required: false,
|
||||
default: "change me"
|
||||
],
|
||||
req_options: [
|
||||
type: :keyword_list,
|
||||
required: false,
|
||||
default: []
|
||||
]
|
||||
)
|
||||
|
||||
@doc NimbleOptions.docs(@schema)
|
||||
@spec resolve(Application.app()) :: t | no_return
|
||||
def resolve(otp_app) do
|
||||
app_config =
|
||||
Application.get_env(otp_app, BraveSearch)
|
||||
|> NimbleOptions.validate!(@schema)
|
||||
|
||||
struct(__MODULE__, app_config)
|
||||
end
|
||||
end
|
||||
@@ -5,6 +5,7 @@ defmodule MusicLibraryWeb.Components.RecordForm do
|
||||
only: [format_label: 1, type_label: 1, release_label: 1, release_summary: 1, record_cover: 1]
|
||||
|
||||
alias MusicLibrary.{Assets, Records}
|
||||
alias MusicLibrary.Assets.Image
|
||||
alias MusicLibrary.Records.Record
|
||||
|
||||
@impl true
|
||||
@@ -297,6 +298,76 @@ defmodule MusicLibraryWeb.Components.RecordForm do
|
||||
</div>
|
||||
</div>
|
||||
</div>
|
||||
<div class="col-span-full space-y-4">
|
||||
<.label>{gettext("Search for cover art online")}</.label>
|
||||
<div class="flex gap-2">
|
||||
<div class="flex-1">
|
||||
<.input
|
||||
type="text"
|
||||
id="cover-search-query"
|
||||
name="cover_search_query"
|
||||
value={@cover_search_query}
|
||||
phx-keyup="update_cover_search_query"
|
||||
phx-target={@myself}
|
||||
/>
|
||||
</div>
|
||||
<.button
|
||||
type="button"
|
||||
size="sm"
|
||||
id="cover-search-button"
|
||||
phx-click="search_covers"
|
||||
phx-target={@myself}
|
||||
disabled={@cover_search_loading}
|
||||
>
|
||||
<.icon
|
||||
:if={@cover_search_loading}
|
||||
name="hero-arrow-path"
|
||||
class="icon animate-spin"
|
||||
aria-hidden="true"
|
||||
data-slot="icon"
|
||||
/>
|
||||
<.icon
|
||||
:if={!@cover_search_loading}
|
||||
name="hero-magnifying-glass"
|
||||
class="icon"
|
||||
aria-hidden="true"
|
||||
data-slot="icon"
|
||||
/>
|
||||
{gettext("Search")}
|
||||
</.button>
|
||||
</div>
|
||||
<p :if={@cover_search_error} class="text-sm text-red-600 dark:text-red-400">
|
||||
{@cover_search_error}
|
||||
</p>
|
||||
<div
|
||||
:if={@cover_search_results != []}
|
||||
id="cover-search-results"
|
||||
class="grid grid-cols-3 sm:grid-cols-4 gap-2"
|
||||
>
|
||||
<button
|
||||
:for={result <- @cover_search_results}
|
||||
type="button"
|
||||
phx-click="select_cover"
|
||||
phx-value-url={result.image_url}
|
||||
phx-target={@myself}
|
||||
disabled={@cover_search_loading}
|
||||
class={[
|
||||
"group relative overflow-hidden rounded-md",
|
||||
"border border-zinc-200 dark:border-zinc-700",
|
||||
"hover:ring-2 hover:ring-indigo-500",
|
||||
"focus:outline-none focus:ring-2 focus:ring-indigo-500",
|
||||
"disabled:opacity-50 disabled:cursor-not-allowed"
|
||||
]}
|
||||
>
|
||||
<img
|
||||
src={result.thumbnail_url}
|
||||
alt={result.title}
|
||||
class="aspect-square w-full object-cover"
|
||||
loading="lazy"
|
||||
/>
|
||||
</button>
|
||||
</div>
|
||||
</div>
|
||||
<:actions>
|
||||
<div class="w-full md:flex md:justify-center">
|
||||
<.button variant="solid" class="w-full md:w-auto" phx-disable-with={gettext("Saving...")}>
|
||||
@@ -333,7 +404,13 @@ defmodule MusicLibraryWeb.Components.RecordForm do
|
||||
end)
|
||||
|> assign_new(:all_genres, fn -> Records.list_genres() end)
|
||||
|> assign_new(:genre_query, fn -> "" end)
|
||||
|> assign_new(:genre_suggestions, fn -> [] end)}
|
||||
|> assign_new(:genre_suggestions, fn -> [] end)
|
||||
|> assign_new(:cover_search_query, fn ->
|
||||
"#{Record.artist_names(record)} #{record.title} album cover"
|
||||
end)
|
||||
|> assign_new(:cover_search_results, fn -> [] end)
|
||||
|> assign_new(:cover_search_loading, fn -> false end)
|
||||
|> assign_new(:cover_search_error, fn -> nil end)}
|
||||
end
|
||||
|
||||
@impl true
|
||||
@@ -412,6 +489,89 @@ defmodule MusicLibraryWeb.Components.RecordForm do
|
||||
{:noreply, assign(socket, :form, to_form(changeset, action: :validate))}
|
||||
end
|
||||
|
||||
def handle_event("update_cover_search_query", %{"value" => query}, socket) do
|
||||
{:noreply, assign(socket, :cover_search_query, query)}
|
||||
end
|
||||
|
||||
def handle_event("search_covers", _params, socket) do
|
||||
query = socket.assigns.cover_search_query
|
||||
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_loading, true)
|
||||
|> assign(:cover_search_error, nil)
|
||||
|> start_async(:cover_search, fn -> BraveSearch.search_images(query, count: 20) end)}
|
||||
end
|
||||
|
||||
def handle_event("select_cover", %{"url" => url}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_loading, true)
|
||||
|> assign(:cover_search_error, nil)
|
||||
|> start_async(:cover_download, fn ->
|
||||
with {:ok, data} <- BraveSearch.download_image(url),
|
||||
{:ok, resized} <- Image.resize(data),
|
||||
{:ok, asset} <- Assets.store_image(%{content: resized, format: "image/jpeg"}) do
|
||||
{:ok, asset.hash}
|
||||
end
|
||||
end)}
|
||||
end
|
||||
|
||||
@impl true
|
||||
def handle_async(:cover_search, {:ok, {:ok, results}}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_results, results)
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
|
||||
def handle_async(:cover_search, {:ok, {:error, reason}}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_error, "Search failed: #{inspect(reason)}")
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
|
||||
def handle_async(:cover_search, {:exit, reason}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_error, "Search failed: #{inspect(reason)}")
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
|
||||
def handle_async(:cover_download, {:ok, {:ok, cover_hash}}, socket) do
|
||||
case Records.update_record(socket.assigns.record, %{"cover_hash" => cover_hash}) do
|
||||
{:ok, record} ->
|
||||
notify_parent({:saved, record})
|
||||
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_loading, false)
|
||||
|> put_toast(:info, gettext("Cover art updated successfully"))
|
||||
|> push_patch(to: socket.assigns.patch)}
|
||||
|
||||
{:error, _changeset} ->
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_error, gettext("Failed to save cover art"))
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
end
|
||||
|
||||
def handle_async(:cover_download, {:ok, {:error, reason}}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_error, "Download failed: #{inspect(reason)}")
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
|
||||
def handle_async(:cover_download, {:exit, reason}, socket) do
|
||||
{:noreply,
|
||||
socket
|
||||
|> assign(:cover_search_error, "Download failed: #{inspect(reason)}")
|
||||
|> assign(:cover_search_loading, false)}
|
||||
end
|
||||
|
||||
defp save_record(socket, record_params, uploaded_covers) do
|
||||
params =
|
||||
case uploaded_covers do
|
||||
|
||||
Reference in New Issue
Block a user