base.rb in i18n-tasks-0.5.0

- old
+ new
@@ -8,30 +8,50 @@
     end
 
     attr_reader :task
     delegate :base_locale, :locales, to: :task
 
+    protected
+
     MISSING_TYPES = {
         missing_from_base:   {glyph: '✗', summary: 'missing from base locale'},
         missing_from_locale: {glyph: '∅', summary: 'missing from locale but present in base locale'},
         eq_base: {glyph: '=', summary: 'value equals base value'}
     }
 
     def missing_types
       MISSING_TYPES
     end
 
-    def missing_title(recs)
-      "Missing translations (#{recs.length})"
+    def missing_title(forest)
+      "Missing translations (#{forest.leaves.count || '∅'})"
     end
 
-    def unused_title(recs)
-      "Unused keys (#{recs.length})"
+    def unused_title(key_values)
+      "Unused keys (#{key_values.count || '∅'})"
     end
 
-    def used_title(keys)
-      filter = keys.attr[:key_filter]
-      used_n = keys.map { |k| k[:usages].size }.reduce(:+).to_i
-      "#{keys.length} key#{'s' if keys.size != 1}#{" ~ '#{filter}'" if filter}#{" (#{used_n} usage#{'s' if used_n != 1})" if used_n > 0}"
+    def used_title(used_tree)
+      leaves = used_tree.leaves.to_a
+      filter = used_tree.first.root.data[:key_filter]
+      used_n = leaves.map { |node| node.data[:source_locations].size }.reduce(:+).to_i
+      "#{leaves.length} key#{'s' if leaves.size != 1}#{" ~ filter: '#{filter}'" if filter}#{" (#{used_n} usage#{'s' if used_n != 1})" if used_n > 0}"
+    end
+
+    # Sort keys by their attributes in order
+    # @param [Hash] order e.g. {locale: :asc, type: :desc, key: :asc}
+    def sort_by_attr!(objects, order)
+      order_keys = order.keys
+      objects.sort! { |a, b|
+        by = order_keys.detect { |by| a[by] != b[by] }
+        order[by] == :desc ? b[by] <=> a[by] : a[by] <=> b[by]
+      }
+      objects
+    end
+
+    def forest_to_attr(forest)
+      forest.keys(root: false).map { |key, node|
+        {key: key, type: node.data[:type], locale: node.root.key}
+      }
     end
   end
 end