Allow to pass locale as atom into t/t! functions

Author: dolfinus

lib/linguist/compiler.ex

@@ -42,7 +42,7 @@ defmodule Linguist.Compiler do
   @simple_interpol "%{"
 
   def compile(translations) do
-    langs = Keyword.keys(translations)
+    langs = Keyword.keys(translations) |> Enum.map(&to_string/1)
 
     translations =
       for {locale, source} <- translations do
@@ -51,6 +51,9 @@ defmodule Linguist.Compiler do
 
     quote do
       def t(locale, path, binding \\ [])
+      def t(locale, path, binding) when is_atom(locale) do
+        t(to_string(locale), path, binding)
+      end
       unquote(translations)
       def do_t(_locale, _path, _bindings), do: {:error, :no_translation}
 

lib/linguist/memorized_vocabulary.ex

@@ -37,6 +37,10 @@ defmodule Linguist.MemorizedVocabulary do
   def t(locale, path, bindings \\ [])
   def t(nil, _, _), do: raise(LocaleError, nil)
 
+  def t(locale, path, binding) when is_atom(locale) do
+    t(to_string(locale), path, binding)
+  end
+
   def t(locale, path, bindings) do
     pluralization_key = Application.fetch_env!(:linguist, :pluralization_key)
     norm_locale = normalize_locale(locale)

test/vocabulary_test.exs

@@ -26,6 +26,11 @@ defmodule VocabularyTest do
     assert ["fr", "en", "es"] == I18n.locales()
   end
 
+  test "it handles both string and atom locales" do
+    assert I18n.t!("en", "foo") == I18n.t!(:en, "foo")
+    assert I18n.t("en", "foo") == I18n.t(:en, "foo")
+  end
+
   test "it handles translations at rool level" do
     assert I18n.t!("en", "foo") == "bar"
     assert I18n.t("en", "foo") == {:ok, "bar"}