Add lang to trans helper

hoymultimedia · May 20, 2016 · 5feb57e · 5feb57e
1 parent e00f25c
commit 5feb57e
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,9 @@
 # CHANGELOG
 
+## 1.0.1 (released 2016-05-20)
+
+- Add lang to trans helper
+
 ## 1.0.0 (released 2016-05-11)
 
 - First stable release
diff --git a/src/helpers.php b/src/helpers.php
@@ -17,7 +17,7 @@
      *
      * @param array $groups
      *
-     * @throws BadFunctionCallException
+     * @throws \BadFunctionCallException
      *
      * @return void
      */
@@ -40,17 +40,22 @@ function register_translations(array $groups)
      * Get translations by their strings.
      *
      * @param string $key
+     * @param string|null $lang
      *
-     * @throws BadFunctionCallException
+     * @throws \BadFunctionCallException
      *
      * @return string
      */
-    function trans($key)
+    function trans($key, $lang = null)
     {
         if (!function_exists('pll__')) {
             throw new BadFunctionCallException('Please active the Polylang plugin.');
         }
 
+        if ($lang) {
+            return pll_translate_string($key, $lang);
+        }
+
         return pll__($key);
     }
 }