From 7a74a617ea8ba8deba22a22263f4188419d8d02d Mon Sep 17 00:00:00 2001
From: ksss <co000ri@gmail.com>
Date: Sun, 19 Mar 2017 23:04:22 +0900
Subject: Callback should yield with pattern every time

---
 mrblib/string.rb | 43 ++++++++++++++++++++++++-------------------
 1 file changed, 24 insertions(+), 19 deletions(-)

(limited to 'mrblib')

diff --git a/mrblib/string.rb b/mrblib/string.rb
index 24487cb00..26dbdcb9c 100644
--- a/mrblib/string.rb
+++ b/mrblib/string.rb
@@ -59,28 +59,33 @@ class String
   # ISO 15.2.10.5.18
   def gsub(*args, &block)
     return to_enum(:gsub, *args) if args.length == 1 && !block
-    if args.size == 2
-      pattern, replace = *args
-      plen = pattern.length
+    raise ArgumentError, "wrong number of arguments" unless (1..2).include?(args.length)
+
+    pattern, replace = *args
+    plen = pattern.length
+    if args.length == 2 && block
+      block = nil
+    end
+    if !replace.nil? || !block
       replace = replace.to_str
-      offset = 0
-      result = []
-      while found = index(pattern, offset)
-        result << self[offset, found - offset]
-        offset = found + plen
-        result << replace.__sub_replace(self[0, found], pattern, self[offset..-1] || "")
-        if plen == 0
-          result << self[offset, 1]
-          offset += 1
-        end
+    end
+    offset = 0
+    result = []
+    while found = index(pattern, offset)
+      result << self[offset, found - offset]
+      offset = found + plen
+      result << if block
+        block.call(pattern).to_s
+      else
+        replace.__sub_replace(self[0, found], pattern, self[offset..-1] || "")
+      end
+      if plen == 0
+        result << self[offset, 1]
+        offset += 1
       end
-      result << self[offset..-1] if offset < length
-      result.join
-    elsif args.size == 1 && block
-      split(args[0], -1).join(block.call(args[0]))
-    else
-      raise ArgumentError, "wrong number of arguments"
     end
+    result << self[offset..-1] if offset < length
+    result.join
   end
 
   ##
-- 
cgit v1.2.3