From f05f6cb44df000ad7cffcd79b6447c97756d15dd Mon Sep 17 00:00:00 2001
From: Kangzhi Shi <shikangzhi@gmail.com>
Date: Sun, 3 Jan 2021 19:44:40 +0200
Subject: [PATCH] impl Pattern.sub

---
 interp.rs | 24 +++++++++++++++++++++++-
 1 file changed, 23 insertions(+), 1 deletion(-)

diff --git a/interp.rs b/interp.rs
index ca5efd9e4..98844ee9c 100644
--- a/interp.rs
+++ b/interp.rs
@@ -142,6 +142,27 @@ pub(crate) fn pymatch(
     }
 }
 
+pub(crate) fn search(
+    string: PyStrRef,
+    start: usize,
+    end: usize,
+    pattern: PyRef<Pattern>,
+) -> Option<Match> {
+    // TODO: optimize by op info and skip prefix
+    let end = std::cmp::min(end, string.char_len());
+    for i in start..end {
+        if let Some(m) = pymatch(
+            string.clone(),
+            i,
+            end,
+            pattern.clone(),
+        ) {
+            return Some(m);
+        }
+    }
+    None
+}
+
 #[derive(Debug, Copy, Clone)]
 struct MatchContext {
     string_position: usize,
@@ -750,7 +771,8 @@ fn charset(set: &[u32], c: char) -> bool {
                     let (_, blockindices, _) = unsafe { set.align_to::<u8>() };
                     let blocks = &set[64..];
                     let block = blockindices[block_index as usize];
-                    if blocks[((block as u32 * 256 + (ch & 255)) / 32) as usize] & (1u32 << (ch & 31))
+                    if blocks[((block as u32 * 256 + (ch & 255)) / 32) as usize]
+                        & (1u32 << (ch & 31))
                         != 0
                     {
                         return ok;