sort-by-alphabet 0.1.3 → 0.1.4

data/lib/enumerable.rb

@@ -1,11 +1,19 @@
 module Enumerable
   
   def sort_by_alphabet(alphabet)
-    sort{|a, b| a.compare_by_alphabet_sensitive(b, alphabet)}
+    if block_given?
+      sort{|a, b| yield(a).to_s.compare_by_alphabet_sensitive(yield(b).to_s, alphabet)}
+    else
+      sort{|a, b| a.to_s.compare_by_alphabet_sensitive(b.to_s, alphabet)}
+    end
   end
 
   def isort_by_alphabet(alphabet)
-    sort{|a, b| a.compare_by_alphabet_insensitive(b, alphabet)}
+    if block_given?
+      sort{|a, b| yield(a).to_s.compare_by_alphabet_insensitive(yield(b).to_s, alphabet)}
+    else
+      sort{|a, b| a.to_s.compare_by_alphabet_insensitive(b.to_s, alphabet)}
+    end
   end
 
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: sort-by-alphabet
 version: !ruby/object:Gem::Version 
-  version: 0.1.3
+  version: 0.1.4
 platform: ruby
 authors: 
 - Frantisek Havluj