EXP: cover search via Brave API
This commit is contained in:
@@ -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
|
||||
Reference in New Issue
Block a user