Add missing-plural-keys

Fix glebm#308

config/locales/en.yml

@@ -45,6 +45,7 @@ en:
         health: is everything OK?
         irb: start REPL session within i18n-tasks context
         missing: show missing translations
+        missing_plural_keys: show missing pluralizations
         mv: rename/merge the keys in locale data that match the given pattern
         normalize: 'normalize translation data: sort and move to the right files'
         remove_unused: remove unused keys

config/locales/ru.yml

@@ -42,6 +42,7 @@ ru:
         health: Всё ОК?
         irb: начать REPL сессию в контексте i18n-tasks
         missing: показать недостающие переводы
+        missing_plural_keys: показать недостающие плюрализации
         mv: переименовать / объединить ключи, которые соответствуют заданному шаблону
         normalize: нормализовать файлы переводов (сортировка и распределение)
         remove_unused: удалить неиспользуемые ключи

lib/i18n/tasks/command/commands/missing.rb

@@ -33,6 +33,17 @@ def missing(opt = {})
           :exit_1 unless forest.empty?
         end
 
+        cmd :missing_plural_keys,
+            pos:  '[locale ...]',
+            desc: t('i18n_tasks.cmd.desc.missing_plural_keys'),
+            args: %i[locales out_format]
+
+        def missing_plural_keys(opt = {})
+          forest = i18n.missing_plural_keys(opt.slice(:locales))
+          print_forest forest, opt, :missing_plural_keys
+          :exit_1 unless forest.empty?
+        end
+
         cmd :translate_missing,
             pos:  '[locale ...]',
             desc: t('i18n_tasks.cmd.desc.translate_missing'),

lib/i18n/tasks/plural_keys.rb

@@ -16,6 +16,27 @@ def collapse_plural_nodes!(tree)
     tree
   end
 
+  def missing_plural_keys(locales: nil)
+    locales ||= self.locales
+
+    locales.inject(empty_forest) do |tree, locale|
+      required_keys = Set.new(I18n.t(:'i18n.plural.keys', locale: locale, resolve: false))
+
+      data[locale].leaves.map(&:parent).compact.uniq.each do |node|
+        children     = node.children
+        present_keys = Set.new(children.to_hash.keys.map(&:to_sym))
+        next if !plural_forms?(children) || present_keys >= required_keys
+        node.value    = children.to_hash
+        node.children = nil
+        node.data[:missing_keys] = (required_keys - present_keys).to_a
+        tree.merge!(node.walk_to_root.reduce(nil) { |c, p| [p.derive(children: c)] })
+      end
+
+      tree.each { |root| root.data[:type] = :missing_plural_keys }
+      tree
+    end
+  end
+
   # @param [String] key i18n key
   # @param [String] locale to pull key data from
   # @return [String] the base form if the key is a specific plural form (e.g. apple for apple.many), the key otherwise.

lib/i18n/tasks/reports/terminal.rb

@@ -25,6 +25,24 @@ def missing_keys(forest = task.missing_keys)
           end
         end
 
+        def missing_plural_keys(forest = task.missing_plural_keys)
+          forest = collapse_missing_tree! forest
+          if forest.present?
+            print_title missing_title(forest)
+            print_table headings: [Rainbow(I18n.t('i18n_tasks.common.locale')).cyan.bright,
+                                   Rainbow(I18n.t('i18n_tasks.common.key')).cyan.bright,
+                                   I18n.t('i18n_tasks.missing.details_title')] do |t|
+              t.rows = sort_by_attr!(forest_to_attr(forest)).map do |a|
+                [{ value: Rainbow(format_locale(a[:locale])).cyan, alignment: :center },
+                 format_key(a[:key], a[:data]),
+                 a[:data][:missing_keys].to_s]
+              end
+            end
+          else
+            print_success I18n.t('i18n_tasks.missing.none')
+          end
+        end
+
         def icon(type)
           glyph = missing_type_info(type)[:glyph]
           { missing_used: Rainbow(glyph).red, missing_diff: Rainbow(glyph).yellow }[type]

spec/plural_keys_spec.rb

@@ -4,22 +4,39 @@
 
 RSpec.describe 'Plural keys' do
   let(:task) { ::I18n::Tasks::BaseTask.new }
-  before do
-    TestCodebase.setup('config/locales/en.yml' => '')
-    TestCodebase.in_test_app_dir do
-      tree = ::I18n::Tasks::Data::Tree::Siblings.from_nested_hash('en' => {
-                                                                    'regular_key' => 'a',
-                                                                    'plural_key' => {
-                                                                      'one' => 'one', 'other' => '%{count}'
-                                                                    },
-                                                                    'not_really_plural' => {
-                                                                      'one' => 'a',
-                                                                      'green' => 'b'
-                                                                    }
-                                                                  })
-      task.data['en'] = tree
-      task.data['en']
-    end
+
+  let(:base_keys) do
+    {
+      regular_key: 'a',
+
+      plural_key: {
+        one: 'one',
+        other: '%{count}'
+      },
+
+      not_really_plural: {
+        one: 'a',
+        green: 'b'
+      },
+
+      nested: {
+        plural_key: {
+          zero: 'none',
+          one: 'one',
+          other: '%{count}'
+        }
+      }
+    }
+  end
+
+  around do |ex|
+    TestCodebase.setup(
+      'config/i18n-tasks.yml' => { base_locale: 'en', locales: %w[en ar] }.to_yaml,
+      'config/locales/en.yml' => { en: base_keys }.to_yaml,
+      'config/locales/ar.yml' => { ar: base_keys }.to_yaml
+    )
+    TestCodebase.in_test_app_dir { ex.call }
+    TestCodebase.teardown
   end
 
   describe '#depluralize_key' do
@@ -40,7 +57,16 @@ def depluralize(key)
     end
   end
 
-  after do
-    TestCodebase.teardown
+  describe '#missing_plural_keys' do
+    it 'returns keys with missing pluralizations' do
+      wrong  = task.missing_plural_keys
+      leaves = wrong.leaves.to_a
+
+      expect(leaves.size).to eq 2
+      expect(leaves[0].full_key).to eq 'ar.plural_key'
+      expect(leaves[0].data[:missing_keys]).to eq [:zero, :two, :few, :many]
+      expect(leaves[1].full_key).to eq 'ar.nested.plural_key'
+      expect(leaves[1].data[:missing_keys]).to eq [:two, :few, :many]
+    end
   end
 end