akkoma/lib/pleroma/web/plugs/set_locale_plug.ex

100 lines
2.5 KiB
Elixir
Raw Normal View History

2019-07-09 11:30:15 +00:00
# Pleroma: A lightweight social networking server
# Copyright © 2017-2021 Pleroma Authors <https://pleroma.social/>
2019-07-09 11:30:15 +00:00
# SPDX-License-Identifier: AGPL-3.0-only
# NOTE: this module is based on https://github.com/smeevil/set_locale
2020-06-24 06:24:29 +00:00
defmodule Pleroma.Web.Plugs.SetLocalePlug do
2019-07-09 11:30:15 +00:00
import Plug.Conn, only: [get_req_header: 2, assign: 3]
2022-02-22 00:12:32 +00:00
def frontend_language_cookie_name, do: "userLanguage"
2019-07-09 11:30:15 +00:00
def init(_), do: nil
def call(conn, _) do
locales = get_locales_from_header(conn)
first_locale = Enum.at(locales, 0, Gettext.get_locale())
Pleroma.Web.Gettext.put_locales(locales)
conn
|> assign(:locale, first_locale)
|> assign(:locales, locales)
2019-07-09 11:30:15 +00:00
end
defp get_locales_from_header(conn) do
2019-07-09 11:30:15 +00:00
conn
|> extract_preferred_language()
|> normalize_language_codes()
|> all_supported()
end
defp all_supported(locales) do
locales
|> Enum.flat_map(&Pleroma.Web.Gettext.supported_variants_of_locale/1)
|> Enum.filter(&supported_locale?/1)
2019-07-09 11:30:15 +00:00
end
defp normalize_language_codes(codes) do
codes
|> Enum.map(fn code -> Pleroma.Web.Gettext.normalize_locale(code) end)
end
defp extract_preferred_language(conn) do
extract_frontend_language(conn) ++ extract_accept_language(conn)
end
defp extract_frontend_language(conn) do
%{req_cookies: cookies} =
conn
|> Plug.Conn.fetch_cookies()
case cookies[frontend_language_cookie_name()] do
nil ->
[]
fe_lang ->
[fe_lang]
|> ensure_language_fallbacks()
end
end
2019-07-09 11:30:15 +00:00
defp extract_accept_language(conn) do
case get_req_header(conn, "accept-language") do
[value | _] ->
value
|> String.split(",")
|> Enum.map(&parse_language_option/1)
|> Enum.sort(&(&1.quality > &2.quality))
|> Enum.map(& &1.tag)
|> Enum.reject(&is_nil/1)
|> ensure_language_fallbacks()
_ ->
[]
end
end
defp supported_locale?(locale) do
Pleroma.Web.Gettext.supports_locale?(locale)
2019-07-09 11:30:15 +00:00
end
defp parse_language_option(string) do
captures = Regex.named_captures(~r/^\s?(?<tag>[\w\-]+)(?:;q=(?<quality>[\d\.]+))?$/i, string)
quality =
case Float.parse(captures["quality"] || "1.0") do
{val, _} -> val
:error -> 1.0
end
%{tag: captures["tag"], quality: quality}
end
defp ensure_language_fallbacks(tags) do
Enum.flat_map(tags, fn tag ->
[language | _] = String.split(tag, "-")
if Enum.member?(tags, language), do: [tag], else: [tag, language]
end)
end
end