83 lines
2.2 KiB
Elixir
83 lines
2.2 KiB
Elixir
defmodule Pleroma.Akkoma.Translators.LibreTranslate do
|
|
@behaviour Pleroma.Akkoma.Translator
|
|
|
|
alias Pleroma.Config
|
|
alias Pleroma.HTTP
|
|
require Logger
|
|
|
|
defp api_key do
|
|
Config.get([:libre_translate, :api_key])
|
|
end
|
|
|
|
defp url do
|
|
Config.get([:libre_translate, :url])
|
|
end
|
|
|
|
@impl Pleroma.Akkoma.Translator
|
|
def languages do
|
|
with {:ok, %{status: 200} = response} <- do_languages(),
|
|
{:ok, body} <- Jason.decode(response.body) do
|
|
resp = Enum.map(body, fn %{"code" => code, "name" => name} -> %{code: code, name: name} end)
|
|
# No separate source/dest
|
|
{:ok, resp, resp}
|
|
else
|
|
{:ok, %{status: status} = response} ->
|
|
Logger.warning("LibreTranslate: Request rejected: #{inspect(response)}")
|
|
{:error, "LibreTranslate request failed (code #{status})"}
|
|
|
|
{:error, reason} ->
|
|
{:error, reason}
|
|
end
|
|
end
|
|
|
|
@impl Pleroma.Akkoma.Translator
|
|
def translate(string, from_language, to_language) do
|
|
with {:ok, %{status: 200} = response} <- do_request(string, from_language, to_language),
|
|
{:ok, body} <- Jason.decode(response.body) do
|
|
%{"translatedText" => translated} = body
|
|
|
|
detected =
|
|
if Map.has_key?(body, "detectedLanguage") do
|
|
get_in(body, ["detectedLanguage", "language"])
|
|
else
|
|
from_language || ""
|
|
end
|
|
|
|
{:ok, detected, translated}
|
|
else
|
|
{:ok, %{status: status} = response} ->
|
|
Logger.warning("libre_translate: request failed, #{inspect(response)}")
|
|
{:error, "libre_translate: request failed (code #{status})"}
|
|
|
|
{:error, reason} ->
|
|
{:error, reason}
|
|
end
|
|
end
|
|
|
|
defp do_request(string, from_language, to_language) do
|
|
url = URI.parse(url())
|
|
url = %{url | path: "/translate"}
|
|
|
|
HTTP.post(
|
|
to_string(url),
|
|
Jason.encode!(%{
|
|
q: string,
|
|
source: if(is_nil(from_language), do: "auto", else: from_language),
|
|
target: to_language,
|
|
format: "html",
|
|
api_key: api_key()
|
|
}),
|
|
[
|
|
{"content-type", "application/json"}
|
|
]
|
|
)
|
|
end
|
|
|
|
defp do_languages() do
|
|
url = URI.parse(url())
|
|
url = %{url | path: "/languages"}
|
|
|
|
HTTP.get(to_string(url))
|
|
end
|
|
end
|