Allow define the plural function directly, to prevent CSP errors #4

Author: oscarotero

package.json

@@ -20,6 +20,6 @@
     "sprintf-js": "^1.0.3"
   },
   "devDependencies": {
-    "mocha": "^3.4.2"
+    "mocha": "^5.0.0"
   }
 }

src/translator.js

@@ -38,7 +38,9 @@
                 return this;
             }
 
-            if (translations['plural-forms']) {
+            if (translations.fn) {
+                this.plurals[domain] = { fn: translations.fn };
+            } else if (translations['plural-forms']) {
                 var plural = translations['plural-forms'].split(';', 2);
 
                 this.plurals[domain] = {

tests/test.js

@@ -27,3 +27,25 @@ describe('vsprintf functions', function() {
         assert.equal('2 arquivos', i18n.n__('one file', '%s files', 2, 2));
     });
 });
+
+describe('custom plural functions', function() {
+    var translations = require(__dirname + '/translations3.json');
+    var called = false;
+    translations.fn = function (n) {
+        called = true;
+        return 2;
+    };
+
+    i18n.loadTranslations(translations);
+
+    it('should return always the second plural', function () {
+        assert.equal('foo 3', i18n.dngettext('foo', 'foo x', 'foo y', 0));
+        assert.equal('foo 3', i18n.dngettext('foo', 'foo x', 'foo y', 1));
+        assert.equal('foo 3', i18n.dngettext('foo', 'foo x', 'foo y', 2));
+        assert.equal('foo 3', i18n.dngettext('foo', 'foo x', 'foo y', 3));
+    });
+
+    it('the custom function was called', function () {
+        assert(called, 'The custom function was not called')
+    });
+});