@@ -0,0 +1,63 @@ | |||||
# Pleroma: A lightweight social networking server | |||||
# Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/> | |||||
# SPDX-License-Identifier: AGPL-3.0-only | |||||
# NOTE: this module is based on https://github.com/smeevil/set_locale | |||||
defmodule Pleroma.Plugs.SetLocalePlug do | |||||
import Plug.Conn, only: [get_req_header: 2, assign: 3] | |||||
def init(_), do: nil | |||||
def call(conn, _) do | |||||
locale = get_locale_from_header(conn) || Gettext.get_locale() | |||||
Gettext.put_locale(locale) | |||||
assign(conn, :locale, locale) | |||||
end | |||||
defp get_locale_from_header(conn) do | |||||
conn | |||||
|> extract_accept_language() | |||||
|> Enum.find(&supported_locale?/1) | |||||
end | |||||
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 | |||||
|> Gettext.known_locales() | |||||
|> Enum.member?(locale) | |||||
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 |
@@ -7,13 +7,9 @@ defmodule Pleroma.Web.Endpoint do | |||||
socket("/socket", Pleroma.Web.UserSocket) | socket("/socket", Pleroma.Web.UserSocket) | ||||
# Serve at "/" the static files from "priv/static" directory. | |||||
# | |||||
# You should set gzip to true if you are running phoenix.digest | |||||
# when deploying your static files in production. | |||||
plug(Pleroma.Plugs.SetLocalePlug) | |||||
plug(CORSPlug) | plug(CORSPlug) | ||||
plug(Pleroma.Plugs.HTTPSecurityPlug) | plug(Pleroma.Plugs.HTTPSecurityPlug) | ||||
plug(Pleroma.Plugs.UploadedMedia) | plug(Pleroma.Plugs.UploadedMedia) | ||||
@static_cache_control "public, no-cache" | @static_cache_control "public, no-cache" | ||||
@@ -30,6 +26,10 @@ defmodule Pleroma.Web.Endpoint do | |||||
} | } | ||||
) | ) | ||||
# Serve at "/" the static files from "priv/static" directory. | |||||
# | |||||
# You should set gzip to true if you are running phoenix.digest | |||||
# when deploying your static files in production. | |||||
plug( | plug( | ||||
Plug.Static, | Plug.Static, | ||||
at: "/", | at: "/", | ||||
@@ -0,0 +1,46 @@ | |||||
# Pleroma: A lightweight social networking server | |||||
# Copyright © 2017-2018 Pleroma Authors <https://pleroma.social/> | |||||
# SPDX-License-Identifier: AGPL-3.0-only | |||||
defmodule Pleroma.Plugs.SetLocalePlugTest do | |||||
use ExUnit.Case, async: true | |||||
use Plug.Test | |||||
alias Plug.Conn | |||||
alias Pleroma.Plugs.SetLocalePlug | |||||
test "default locale is `en`" do | |||||
conn = | |||||
:get | |||||
|> conn("/cofe") | |||||
|> SetLocalePlug.call([]) | |||||
assert "en" == Gettext.get_locale() | |||||
assert %{locale: "en"} == conn.assigns | |||||
end | |||||
test "use supported locale from `accept-language`" do | |||||
conn = | |||||
:get | |||||
|> conn("/cofe") | |||||
|> Conn.put_req_header( | |||||
"accept-language", | |||||
"ru, fr-CH, fr;q=0.9, en;q=0.8, *;q=0.5" | |||||
) | |||||
|> SetLocalePlug.call([]) | |||||
assert "ru" == Gettext.get_locale() | |||||
assert %{locale: "ru"} == conn.assigns | |||||
end | |||||
test "use default locale if locale from `accept-language` is not supported" do | |||||
conn = | |||||
:get | |||||
|> conn("/cofe") | |||||
|> Conn.put_req_header("accept-language", "tlh") | |||||
|> SetLocalePlug.call([]) | |||||
assert "en" == Gettext.get_locale() | |||||
assert %{locale: "en"} == conn.assigns | |||||
end | |||||
end |