1 # Pleroma: A lightweight social networking server
2 # Copyright © 2017-2019 Pleroma Authors <https://pleroma.social/>
3 # SPDX-License-Identifier: AGPL-3.0-only
5 # NOTE: this module is based on https://github.com/smeevil/set_locale
6 defmodule Pleroma.Plugs.SetLocalePlug do
7 import Plug.Conn, only: [get_req_header: 2, assign: 3]
12 locale = get_locale_from_header(conn) || Gettext.get_locale()
13 Gettext.put_locale(locale)
14 assign(conn, :locale, locale)
17 defp get_locale_from_header(conn) do
19 |> extract_accept_language()
20 |> Enum.find(&supported_locale?/1)
23 defp extract_accept_language(conn) do
24 case get_req_header(conn, "accept-language") do
28 |> Enum.map(&parse_language_option/1)
29 |> Enum.sort(&(&1.quality > &2.quality))
31 |> Enum.reject(&is_nil/1)
32 |> ensure_language_fallbacks()
39 defp supported_locale?(locale) do
41 |> Gettext.known_locales()
42 |> Enum.member?(locale)
45 defp parse_language_option(string) do
46 captures = Regex.named_captures(~r/^\s?(?<tag>[\w\-]+)(?:;q=(?<quality>[\d\.]+))?$/i, string)
49 case Float.parse(captures["quality"] || "1.0") do
54 %{tag: captures["tag"], quality: quality}
57 defp ensure_language_fallbacks(tags) do
58 Enum.flat_map(tags, fn tag ->
59 [language | _] = String.split(tag, "-")
60 if Enum.member?(tags, language), do: [tag], else: [tag, language]