X-Git-Url: https://git.njae.me.uk/?a=blobdiff_plain;f=lib%2Ferd_handler%2Flabel.rb;fp=lib%2Ferd_handler%2Flabel.rb;h=6c197588c08157320a5fde3c4818db24005a80cc;hb=55659f547afb1ed903295784f9f4ae99624810d1;hp=ad816182ff4be9c58f62ff53819c26579862cfab;hpb=042a27abb3ac3bf9faf36406204fe1d1901123d3;p=erd-marker.git

diff --git a/lib/erd_handler/label.rb b/lib/erd_handler/label.rb
index ad81618..6c19758 100644
--- a/lib/erd_handler/label.rb
+++ b/lib/erd_handler/label.rb
@@ -18,6 +18,11 @@ class Label
       else
         split_camel_case = true
       end
+      if opts.has_key? :numbers
+        split_numbers = opts[:numbers]
+      else
+        split_numbers = true
+      end
     end
     @processed = @processed.map do |segment|
       segment.split(regexp)
@@ -25,7 +30,13 @@ class Label
     
     if split_camel_case
       @processed = @processed.map do |segment|
-        segment.split(/(?=[A-Z])/)
+        segment.split(/(?<=[a-z])(?=[A-Z])/)
+      end.flatten
+    end
+    
+    if split_numbers
+      @processed = @processed.map do |segment|
+        segment.split(/(?:(?<!\d)(?=\d))|(?:(?<=\d)(?!\d))/)
       end.flatten
     end
     self