Refactor to avoid duplicating find_nth_next

helix-core/src/search.rs

@@ -17,30 +17,7 @@ impl<F: Fn(&char) -> bool> CharMatcher for F {
     }
 }
 
-pub fn find_nth_next(text: RopeSlice, ch: char, mut pos: usize, n: usize) -> Option<usize> {
-    if pos >= text.len_chars() || n == 0 {
-        return None;
-    }
-
-    let mut chars = text.chars_at(pos);
-
-    for _ in 0..n {
-        loop {
-            let c = chars.next()?;
-
-            pos += 1;
-
-            if c == ch {
-                break;
-            }
-        }
-    }
-
-    Some(pos - 1)
-}
-
-/// Like find_nth_next, but stops at the first newline character.
-pub fn find_nth_next_until_newline<M: CharMatcher>(
+pub fn find_nth_next<M: CharMatcher>(
     text: RopeSlice,
     char_matcher: M,
     mut pos: usize,
@@ -58,10 +35,6 @@ pub fn find_nth_next_until_newline<M: CharMatcher>(
 
             pos += 1;
 
-            if c == '\n' || c == '\r' {
-                return None;
-            }
-
             if char_matcher.char_match(c) {
                 break;
             }

helix-term/src/commands.rs

@@ -4340,10 +4340,13 @@ fn find_next_char_until_newline<M: CharMatcher>(
     text: RopeSlice,
     char_matcher: M,
     pos: usize,
-    n: usize,
+    _count: usize,
     _inclusive: bool,
 ) -> Option<usize> {
-    search::find_nth_next_until_newline(text, char_matcher, pos, n)
+    let line_index = text.char_to_line(pos);
+    let pos_delta = text.line_to_char(line_index);
+    let pos = pos - pos_delta;
+    search::find_nth_next(text.line(line_index), char_matcher, pos, 1).map(|pos| pos + pos_delta)
 }
 
 /// Decrement object under cursor by `amount`.