Allow pluralization customization via constructor options (closes #464)

decls/i18n.js

@@ -61,7 +61,10 @@ declare type I18nOptions = {
   root?: I18n, // for internal
   fallbackRoot?: boolean,
   sync?: boolean,
-  silentTranslationWarn?: boolean
+  silentTranslationWarn?: boolean,
+  pluralizationRules?: {
+    [lang: string]: (choice: number, choicesLength: number) => number,
+  },
 };
 
 declare type IntlAvailability = {
@@ -100,7 +103,10 @@ declare interface I18n {
   getNumberFormat (locale: Locale): NumberFormat,
   setNumberFormat (locale: Locale, format: NumberFormat): void,
   mergeNumberFormat (locale: Locale, format: NumberFormat): void,
-  n (value: number, ...args: any): NumberFormatResult
+  n (value: number, ...args: any): NumberFormatResult,
+  pluralizationRules: {
+    [lang: string]: (choice: number, choicesLength: number) => number
+  }
 };
 
 declare interface Formatter {

gitbook/en/api.md

@@ -38,6 +38,8 @@
 
   Localize the locale message of `key` with pluralization. Localize in preferentially component locale messages than global locale messages. If not specified component locale messages, localize with global locale messages. If you specified `locale`, localize the locale messages of `locale`. If you will specify string value to `values`, localize the locale messages of value. If you will specify Array or Object value to `values`, you must specify with `values` of [$t](#t).
 
+  If default pluralization does not suit your needs, see [pluralization rules in constructor options](#pluralizationrules) and [custom pluralization](pluralization.md).
+
 #### getChoiceIndex
 
 - **Arguments:**
@@ -237,7 +239,7 @@ You can specify the below some options of `I18nOptions` constructor options of [
 
   If `false`, regardless of the root level locale, localize for each component locale.
 
-### silentTranslationWarn
+#### silentTranslationWarn
 
 > 6.1+
 
@@ -249,6 +251,26 @@ You can specify the below some options of `I18nOptions` constructor options of [
 
   If `true`, supress localization fail warnings.
 
+#### pluralizationRules
+
+> 8.5+
+
+- **Type:** `Object`
+
+- **Default:** `{}`
+
+  A set of rules for word pluralization in a following format:
+  ```js
+    {
+      // Key - locale for the rule to be applied to.
+      // Value - mapping function that maps a choice index from `$tc` to the actual choice of the plural word.
+
+      'ru': function (choice, choiceIndex) => Number/* index of the plural word */;
+      'en': function (choice, choiceIndex) => Number/* index of the plural word */;
+      'jp': function (choice, choiceIndex) => Number/* index of the plural word */;
+    }
+  ```
+
 ### Properties
 
 #### locale
@@ -321,6 +343,16 @@ You can specify the below some options of `I18nOptions` constructor options of [
 
   Whether suppress warnings outputted when localization fails.
 
+#### pluralizationRules
+
+> 8.5+
+
+- **Type:** `Object`
+
+- **Default:** `{}`
+
+  A set of rules for word pluralization. Key is a locale, value is the rule function for that locale.
+
 ### Methods
 
 #### getLocaleMessage( locale )

gitbook/en/pluralization.md

@@ -43,43 +43,46 @@ This will output the following HTML:
 
 Such pluralization, however, does not apply to all languages (Slavic languages, for example, have different pluralization rules).
 
-In order to implement these rules you can override the `VueI18n.prototype.getChoiceIndex` function.
+In order to implement these rules you can pass an optional `pluralizationRules` object into `VueI18n` constructor options.
 
 Very simplified example using rules for Slavic languages (Russian, Ukrainian, etc.):
 ```js
-/**
- * @param choice {number} a choice index given by the input to $tc: `$tc('path.to.rule', choiceIndex)`
- * @param choicesLength {number} an overall amount of available choices
- * @returns a final choice index to select plural word by
-**/
-VueI18n.prototype.getChoiceIndex = function (choice, choicesLength) {
-  // this === VueI18n instance, so the locale property also exists here
-  if (this.locale !== 'ru') {
-    // proceed to the default implementation
+new VueI18n({
+  pluralizationRules: {
+    /** Key - language to use the rule for, 'ru', in this case */
+    /** Value - function
+     * @param choice {number} a choice index given by the input to $tc: `$tc('path.to.rule', choiceIndex)`
+     * @param choicesLength {number} an overall amount of available choices
+     * @returns a final choice index to select plural word by
+     **/
+    'ru': function (choice, choicesLength) {
+      // this === VueI18n instance, so the locale property also exists here
+
+      if (choice === 0) {
+        return 0;
+      }
+
+      const teen = choice > 10 && choice < 20;
+      const endsWithOne = choice % 10 === 1;
+
+      if (choicesLength < 4) {
+        return (!teen && endsWithOne) ? 1 : 2;
+      }
+      if (!teen && endsWithOne) {
+        return 1;
+      }
+      if (!teen && choice % 10 >= 2 && choice % 10 <= 4) {
+        return 2;
+      }
+
+      return (choicesLength < 4) ? 2 : 3;
+    }
   }
-
-  if (choice === 0) {
-    return 0;
-  }
-
-  const teen = choice > 10 && choice < 20;
-  const endsWithOne = choice % 10 === 1;
-
-  if (!teen && endsWithOne) {
-    return 1;
-  }
-
-  if (!teen && choice % 10 >= 2 && choice % 10 <= 4) {
-    return 2;
-  }
-
-  return (choicesLength < 4) ? 2 : 3;
-}
+});
 ```
 
 This would effectively give this:
 
-
 ```javascript
 const messages = {
   ru: {
@@ -119,3 +122,7 @@ Which results in:
 <p>11 бананов</p>
 <p>31 банан</p>
 ```
+
+### Default pluralization
+
+If your current locale is not found in a pluralization map, the [default](#pluralization) rule of the english langugage will be used.

src/index.js

@@ -58,6 +58,9 @@ export default class VueI18n {
   _numberFormatters: Object
   _path: I18nPath
   _dataListeners: Array<any>
+  pluralizationRules: {
+    [lang: string]: (choice: number, choicesLength: number) => number
+  }
 
   constructor (options: I18nOptions = {}) {
     // Auto install if it is not done yet and `window` has `Vue`.
@@ -90,6 +93,8 @@ export default class VueI18n {
     this._path = new I18nPath()
     this._dataListeners = []
 
+    this.pluralizationRules = options.pluralizationRules || {}
+
     this._exist = (message: Object, key: Path): boolean => {
       if (!message || !key) { return false }
       return !isNull(this._path.getPathValue(message, key))
@@ -437,17 +442,26 @@ export default class VueI18n {
    * @returns a final choice index
   */
   getChoiceIndex (choice: number, choicesLength: number): number {
-    choice = Math.abs(choice)
-
-    if (choicesLength === 2) {
-      return choice
-        ? choice > 1
-          ? 1
-          : 0
-        : 1
+    // Default (old) getChoiceIndex implementation - english-compatible
+    const defaultImpl = (_choice: number, _choicesLength: number) => {
+      _choice = Math.abs(_choice)
+
+      if (_choicesLength === 2) {
+        return _choice
+          ? _choice > 1
+            ? 1
+            : 0
+          : 1
+      }
+
+      return _choice ? Math.min(_choice, 2) : 0
     }
 
-    return choice ? Math.min(choice, 2) : 0
+    if (this.locale in this.pluralizationRules) {
+      return this.pluralizationRules[this.locale].apply(this, [choice, choicesLength])
+    } else {
+      return defaultImpl(choice, choicesLength)
+    }
   }
 
   tc (key: Path, choice?: number, ...values: any): TranslateResult {