From ac5ea3268c44ea084ec3645bf81088af619c04f9 Mon Sep 17 00:00:00 2001
From: flywind <xzsflywind@gmail.com>
Date: Wed, 31 Mar 2021 21:06:38 +0800
Subject: [PATCH] hide impl

---
 lib/impure/re.nim    | 24 ++++++++++++++++++++----
 tests/stdlib/tre.nim |  9 ++++++---
 2 files changed, 26 insertions(+), 7 deletions(-)

diff --git a/lib/impure/re.nim b/lib/impure/re.nim
index 0c96876b914b4..b2283a8a1fb52 100644
--- a/lib/impure/re.nim
+++ b/lib/impure/re.nim
@@ -199,8 +199,20 @@ proc findBounds*(s: string, pattern: Regex,
   ## `(-1,0)` is returned.
   result = findBounds(cstring(s), pattern, matches, start, s.len)
 
+proc findBoundsImpl(buf: cstring, pattern: Regex,
+                    start = 0, bufSize = 0, flags = 0): tuple[first, last: int] =
+  var rtarray = initRtArray[cint](3)
+  let rawMatches = rtarray.getRawData
+  let res = pcre.exec(pattern.h, pattern.e, buf, bufSize.cint, start.cint, flags.int32,
+                cast[ptr cint](rawMatches), 3)
+
+  if res < 0'i32:
+    result = (-1, 0)
+  else:
+    result = (int(rawMatches[0]), int(rawMatches[1]-1))
+
 proc findBounds*(buf: cstring, pattern: Regex,
-                 start = 0, bufSize: int): tuple[first, last: int] =
+                 start = 0, bufSize = 0): tuple[first, last: int] =
   ## returns the `first` and `last` position of `pattern` in `buf`,
   ## where `buf` has length `bufSize` (not necessarily `'\0'` terminated).
   ## If it does not match, `(-1,0)` is returned.
@@ -209,7 +221,7 @@ proc findBounds*(buf: cstring, pattern: Regex,
     rawMatches = rtarray.getRawData
     res = pcre.exec(pattern.h, pattern.e, buf, bufSize.cint, start.cint, 0'i32,
       cast[ptr cint](rawMatches), 3)
-  if res < 0'i32: return (int(res), 0)
+  if res < 0'i32: return (-1, 0)
   return (int(rawMatches[0]), int(rawMatches[1]-1))
 
 proc findBounds*(s: string, pattern: Regex,
@@ -433,12 +445,16 @@ proc replace*(s: string, sub: Regex, by = ""): string =
     doAssert "var1=key; var2=key2".replace(re"(\w+)=(\w+)", "?") == "?; ?"
   result = ""
   var prev = 0
+  var flags = int32(0)
   while prev < s.len:
-    var match = findBounds(s, sub, prev)
+    var match = findBoundsImpl(s.cstring, sub, prev, s.len, flags)
+    flags = 0
     if match.first < 0: break
     add(result, substr(s, prev, match.first-1))
     add(result, by)
-    if match.last + 1 == prev: break
+    if match.first > match.last:
+      # 0-len match
+      flags = pcre.NOTEMPTY_ATSTART
     prev = match.last + 1
   add(result, substr(s, prev))
 
diff --git a/tests/stdlib/tre.nim b/tests/stdlib/tre.nim
index ea1b5af32cc3b..2857c6c9e5875 100644
--- a/tests/stdlib/tre.nim
+++ b/tests/stdlib/tre.nim
@@ -99,10 +99,13 @@ proc testAll() =
       accum.add($x)
     doAssert(accum == @["a","b","c"])
 
-  block:
-    # bug #9306
+  block: # bug #9306
     doAssert replace("bar", re"^", "foo") == "foobar"
-    doAssert replace("foo", re"", "-") == "-foo"
     doAssert replace("foo", re"$", "bar") == "foobar"
 
+
+  block: # bug #9437
+    doAssert replace("foo", re"", "-") == "-f-o-o-"
+    doAssert replace("ooo", re"o", "-") == "---"
+
 testAll()