pub struct MatchBlock {
pub line_numbers: Vec<usize>,
}
pub fn search_content(content: &str, pattern: &str, context_lines: usize) -> Vec<MatchBlock> {
let lines: Vec<&str> = content.lines().collect();
let total = lines.len();
if total == 0 || pattern.is_empty() {
return Vec::new();
}
let match_indices: Vec<usize> = lines
.iter()
.enumerate()
.filter(|(_, line)| line.contains(pattern))
.map(|(i, _)| i)
.collect();
if match_indices.is_empty() {
return Vec::new();
}
let mut ranges: Vec<(usize, usize)> = Vec::new();
for &idx in &match_indices {
let start = idx.saturating_sub(context_lines);
let end = (idx + context_lines).min(total - 1);
if let Some(last) = ranges.last_mut() {
if start <= last.1 + 1 {
last.1 = last.1.max(end);
continue;
}
}
ranges.push((start, end));
}
ranges
.into_iter()
.map(|(start, end)| MatchBlock {
line_numbers: (start..=end).map(|i| i + 1).collect(),
})
.collect()
}
#[cfg(test)]
mod tests {
use super::*;
#[test]
fn test_no_matches() {
let blocks = search_content("hello\nworld\n", "xyz", 0);
assert!(blocks.is_empty());
}
#[test]
fn test_single_match_no_context() {
let blocks = search_content("aaa\nbbb\nccc\n", "bbb", 0);
assert_eq!(blocks.len(), 1);
assert_eq!(blocks[0].line_numbers, vec![2]);
}
#[test]
fn test_single_match_with_context() {
let blocks = search_content("aaa\nbbb\nccc\nddd\neee\n", "ccc", 1);
assert_eq!(blocks.len(), 1);
assert_eq!(blocks[0].line_numbers, vec![2, 3, 4]);
}
#[test]
fn test_multiple_matches_merge() {
let blocks = search_content("a\nb\nc\nd\ne\n", "b", 1);
assert_eq!(blocks.len(), 1);
assert_eq!(blocks[0].line_numbers, vec![1, 2, 3]);
}
#[test]
fn test_multiple_matches_separate_blocks() {
let blocks = search_content("a\nmatch\nc\nd\ne\nf\nmatch\nh\n", "match", 0);
assert_eq!(blocks.len(), 2);
assert_eq!(blocks[0].line_numbers, vec![2]);
assert_eq!(blocks[1].line_numbers, vec![7]);
}
#[test]
fn test_context_merges_adjacent_matches() {
let blocks = search_content("a\nmatch\nc\nmatch\ne\n", "match", 1);
assert_eq!(blocks.len(), 1);
assert_eq!(blocks[0].line_numbers, vec![1, 2, 3, 4, 5]);
}
#[test]
fn test_context_clamps_to_bounds() {
let blocks = search_content("match\nb\nc\n", "match", 3);
assert_eq!(blocks.len(), 1);
assert_eq!(blocks[0].line_numbers, vec![1, 2, 3]);
}
#[test]
fn test_empty_pattern() {
let blocks = search_content("hello\n", "", 0);
assert!(blocks.is_empty());
}
#[test]
fn test_empty_content() {
let blocks = search_content("", "hello", 0);
assert!(blocks.is_empty());
}
}