Skip to main content

rumdl_lib/utils/
range_utils.rs

1//! Utilities for position/range conversions
2
3use crate::utils::calculate_indentation_width_default;
4use std::collections::HashSet;
5use std::ops::Range;
6
7/// Find the nearest valid UTF-8 character boundary at or before the given byte index.
8/// This is critical for safely slicing strings that may contain multi-byte UTF-8 characters.
9///
10/// # Safety
11/// Returns a byte index that is guaranteed to be a valid character boundary,
12/// or the string length if the index is beyond the string.
13fn find_char_boundary(s: &str, byte_idx: usize) -> usize {
14    if byte_idx >= s.len() {
15        return s.len();
16    }
17
18    // If the index is already at a character boundary, return it
19    if s.is_char_boundary(byte_idx) {
20        return byte_idx;
21    }
22
23    // Find the nearest character boundary by scanning backwards
24    // This is safe because we know byte_idx < s.len()
25    let mut pos = byte_idx;
26    while pos > 0 && !s.is_char_boundary(pos) {
27        pos -= 1;
28    }
29    pos
30}
31
32/// Convert a byte index within a line into a 1-indexed character column.
33///
34/// rumdl reports diagnostic columns as character offsets, not byte offsets, so
35/// any position derived from a byte index (regex match, `str::find`, parser byte
36/// offset) must pass through this before being stored in a `LintWarning`.
37/// Multi-byte UTF-8 characters are handled by snapping to the nearest character
38/// boundary at or before `byte_idx`.
39pub(crate) fn byte_to_char_count(s: &str, byte_idx: usize) -> usize {
40    let safe_byte_idx = find_char_boundary(s, byte_idx);
41    s[..safe_byte_idx].chars().count() + 1 // 1-indexed
42}
43
44#[derive(Debug)]
45pub struct LineIndex<'a> {
46    line_starts: Vec<usize>,
47    content: &'a str,
48    code_block_lines: Option<HashSet<usize>>,
49}
50
51impl<'a> LineIndex<'a> {
52    pub fn new(content: &'a str) -> Self {
53        let mut line_starts = vec![0];
54        let mut pos = 0;
55
56        for c in content.chars() {
57            pos += c.len_utf8();
58            if c == '\n' {
59                line_starts.push(pos);
60            }
61        }
62
63        let mut index = Self {
64            line_starts,
65            content,
66            code_block_lines: None,
67        };
68
69        // Pre-compute code block lines for better performance
70        index.compute_code_block_lines();
71
72        index
73    }
74
75    /// Create a `LineIndex` from pre-computed line start byte offsets.
76    /// Each entry is the byte offset of the first character on that line.
77    /// The first entry must be 0 (start of content).
78    pub fn with_line_starts(content: &'a str, line_starts: Vec<usize>) -> Self {
79        let mut index = Self {
80            line_starts,
81            content,
82            code_block_lines: None,
83        };
84
85        // Pre-compute code block lines for better performance
86        index.compute_code_block_lines();
87
88        index
89    }
90
91    /// Create a `LineIndex` from pre-computed line starts and code block byte ranges.
92    ///
93    /// Instead of re-scanning content to find code blocks, this converts
94    /// the already-detected byte ranges into line-level information.
95    pub fn with_line_starts_and_code_blocks(
96        content: &'a str,
97        line_starts: Vec<usize>,
98        code_block_byte_ranges: &[(usize, usize)],
99    ) -> Self {
100        let mut code_block_lines = HashSet::new();
101
102        for &(block_start, block_end) in code_block_byte_ranges {
103            let start_line = match line_starts.binary_search(&block_start) {
104                Ok(idx) => idx,
105                Err(idx) => idx.saturating_sub(1),
106            };
107            let end_line = if block_end == 0 {
108                0
109            } else {
110                match line_starts.binary_search(&block_end) {
111                    // block_end exactly at a line start means the block ended on the previous line
112                    Ok(idx) => idx.saturating_sub(1),
113                    Err(idx) => idx.saturating_sub(1),
114                }
115            };
116            for line_idx in start_line..=end_line {
117                code_block_lines.insert(line_idx);
118            }
119        }
120
121        Self {
122            line_starts,
123            content,
124            code_block_lines: Some(code_block_lines),
125        }
126    }
127
128    /// Get the content of a line by 0-based index using pre-computed byte offsets.
129    /// Returns the line content without the trailing newline character.
130    fn get_line(&self, line_idx: usize) -> Option<&'a str> {
131        let start = *self.line_starts.get(line_idx)?;
132        let end = self
133            .line_starts
134            .get(line_idx + 1)
135            .copied()
136            .unwrap_or(self.content.len());
137        let line = &self.content[start..end];
138        // Strip trailing newline (and optional \r before it)
139        let line = line.strip_suffix('\n').unwrap_or(line);
140        let line = line.strip_suffix('\r').unwrap_or(line);
141        Some(line)
142    }
143
144    pub fn line_col_to_byte_range(&self, line: usize, column: usize) -> Range<usize> {
145        let line = line.saturating_sub(1);
146        let line_start = *self.line_starts.get(line).unwrap_or(&self.content.len());
147
148        let current_line = self.get_line(line).unwrap_or("");
149        // Column is 1-indexed character position, not byte position
150        let char_col = column.saturating_sub(1);
151        let char_count = current_line.chars().count();
152        let safe_char_col = char_col.min(char_count);
153
154        // Convert character position to byte position
155        let byte_offset = current_line
156            .char_indices()
157            .nth(safe_char_col)
158            .map_or(current_line.len(), |(idx, _)| idx);
159
160        let start = line_start + byte_offset;
161        start..start
162    }
163
164    /// Calculate a proper byte range for replacing text with a specific length
165    /// This is the correct function to use for LSP fixes
166    ///
167    /// # Safety
168    /// This function correctly handles multi-byte UTF-8 characters by converting
169    /// character positions (columns) to byte positions.
170    pub fn line_col_to_byte_range_with_length(&self, line: usize, column: usize, length: usize) -> Range<usize> {
171        let line = line.saturating_sub(1);
172        let line_start = *self.line_starts.get(line).unwrap_or(&self.content.len());
173        let line_end = self.line_starts.get(line + 1).copied().unwrap_or(self.content.len());
174        let mut current_line = &self.content[line_start..line_end];
175        if let Some(stripped) = current_line.strip_suffix('\n') {
176            current_line = stripped.strip_suffix('\r').unwrap_or(stripped);
177        }
178        if current_line.is_ascii() {
179            let line_len = current_line.len();
180            let start_byte = column.saturating_sub(1).min(line_len);
181            let end_byte = start_byte.saturating_add(length).min(line_len);
182            let start = line_start + start_byte;
183            let end = line_start + end_byte;
184            return start..end;
185        }
186        // Column is 1-indexed character position, not byte position
187        let char_col = column.saturating_sub(1);
188        let char_count = current_line.chars().count();
189        let safe_char_col = char_col.min(char_count);
190
191        // Convert character positions to byte positions
192        let mut char_indices = current_line.char_indices();
193        let start_byte = char_indices
194            .nth(safe_char_col)
195            .map_or(current_line.len(), |(idx, _)| idx);
196
197        // Calculate end position (start + length in characters)
198        let end_char_col = (safe_char_col + length).min(char_count);
199        let end_byte = current_line
200            .char_indices()
201            .nth(end_char_col)
202            .map_or(current_line.len(), |(idx, _)| idx);
203
204        let start = line_start + start_byte;
205        let end = line_start + end_byte;
206        start..end
207    }
208
209    /// Calculate byte range for entire line replacement (including newline)
210    /// This is ideal for rules that need to replace complete lines
211    pub fn whole_line_range(&self, line: usize) -> Range<usize> {
212        let line_idx = line.saturating_sub(1);
213        let start = *self.line_starts.get(line_idx).unwrap_or(&self.content.len());
214        let end = self
215            .line_starts
216            .get(line_idx + 1)
217            .copied()
218            .unwrap_or(self.content.len());
219        start..end
220    }
221
222    /// Calculate byte range spanning multiple lines (from start_line to end_line inclusive)
223    /// Both lines are 1-indexed. This is useful for replacing entire blocks like tables.
224    pub fn multi_line_range(&self, start_line: usize, end_line: usize) -> Range<usize> {
225        let start_idx = start_line.saturating_sub(1);
226        let end_idx = end_line.saturating_sub(1);
227
228        let start = *self.line_starts.get(start_idx).unwrap_or(&self.content.len());
229        let end = self.line_starts.get(end_idx + 1).copied().unwrap_or(self.content.len());
230        start..end
231    }
232
233    /// Calculate byte range for text within a line (excluding newline)
234    /// Useful for replacing specific parts of a line
235    ///
236    /// # Safety
237    /// This function correctly handles multi-byte UTF-8 characters by converting
238    /// character positions (columns) to byte positions.
239    pub fn line_text_range(&self, line: usize, start_col: usize, end_col: usize) -> Range<usize> {
240        let line_idx = line.saturating_sub(1);
241        let line_start = *self.line_starts.get(line_idx).unwrap_or(&self.content.len());
242
243        // Get the actual line content to ensure we don't exceed bounds
244        let current_line = self.get_line(line_idx).unwrap_or("");
245        let char_count = current_line.chars().count();
246
247        // Convert character positions to byte positions
248        let start_char_col = start_col.saturating_sub(1).min(char_count);
249        let end_char_col = end_col.saturating_sub(1).min(char_count);
250
251        let mut char_indices = current_line.char_indices();
252        let start_byte = char_indices
253            .nth(start_char_col)
254            .map_or(current_line.len(), |(idx, _)| idx);
255
256        let end_byte = current_line
257            .char_indices()
258            .nth(end_char_col)
259            .map_or(current_line.len(), |(idx, _)| idx);
260
261        let start = line_start + start_byte;
262        let end = line_start + end_byte.max(start_byte);
263        start..end
264    }
265
266    /// Calculate byte range from start of line to end of line content (excluding newline)
267    /// Useful for replacing line content while preserving line structure
268    pub fn line_content_range(&self, line: usize) -> Range<usize> {
269        let line_idx = line.saturating_sub(1);
270        let line_start = *self.line_starts.get(line_idx).unwrap_or(&self.content.len());
271
272        let current_line = self.get_line(line_idx).unwrap_or("");
273        let line_end = line_start + current_line.len();
274        line_start..line_end
275    }
276
277    /// Get the global start byte offset for a given 1-based line number.
278    pub fn get_line_start_byte(&self, line_num: usize) -> Option<usize> {
279        if line_num == 0 {
280            return None; // Lines are 1-based
281        }
282        // line_num is 1-based, line_starts index is 0-based
283        self.line_starts.get(line_num - 1).copied()
284    }
285
286    /// Check if the line at the given index is within a code block
287    pub fn is_code_block(&self, line: usize) -> bool {
288        if let Some(ref code_block_lines) = self.code_block_lines {
289            code_block_lines.contains(&line)
290        } else {
291            // Fallback to a simpler check if pre-computation wasn't done
292            self.is_code_fence(line)
293        }
294    }
295
296    /// Check if the line is a code fence marker (``` or ~~~)
297    pub fn is_code_fence(&self, line: usize) -> bool {
298        self.get_line(line).is_some_and(|l| {
299            let trimmed = l.trim();
300            trimmed.starts_with("```") || trimmed.starts_with("~~~")
301        })
302    }
303
304    /// Check if the line is a tilde code fence marker (~~~)
305    pub fn is_tilde_code_block(&self, line: usize) -> bool {
306        self.get_line(line).is_some_and(|l| l.trim().starts_with("~~~"))
307    }
308
309    /// Get a reference to the content
310    pub fn get_content(&self) -> &str {
311        self.content
312    }
313
314    /// Pre-compute which lines are within code blocks for faster lookup
315    fn compute_code_block_lines(&mut self) {
316        let mut code_block_lines = HashSet::new();
317        let lines: Vec<&str> = self.content.lines().collect();
318
319        // Initialize block tracking
320        let mut in_block = false;
321        let mut active_fence_type = ' '; // '`' or '~'
322        let mut block_indent = 0;
323        let mut block_fence_length = 0;
324        let mut in_markdown_block = false;
325        let mut nested_fence_start = None;
326        let mut nested_fence_end = None;
327
328        // Process each line
329        for (i, line) in lines.iter().enumerate() {
330            let trimmed = line.trim();
331            let indent = line.len() - trimmed.len();
332
333            // 1. Detect indented code blocks (4+ columns accounting for tab expansion)
334            if calculate_indentation_width_default(line) >= 4 {
335                code_block_lines.insert(i);
336                continue; // Skip further processing for indented code blocks
337            }
338
339            // 2. Handle fenced code blocks (backticks and tildes)
340            if !in_block {
341                // Check for opening fences
342                if trimmed.starts_with("```") || trimmed.starts_with("~~~") {
343                    let char_type = if trimmed.starts_with("```") { '`' } else { '~' };
344                    let count = trimmed.chars().take_while(|&c| c == char_type).count();
345                    let info_string = if trimmed.len() > count {
346                        trimmed[count..].trim()
347                    } else {
348                        ""
349                    };
350
351                    // Mark the start of a new code block
352                    in_block = true;
353                    active_fence_type = char_type;
354                    block_indent = indent;
355                    block_fence_length = count;
356                    in_markdown_block = info_string == "markdown";
357                    nested_fence_start = None;
358                    nested_fence_end = None;
359
360                    code_block_lines.insert(i);
361                }
362            } else {
363                // We're inside a code block
364                code_block_lines.insert(i);
365
366                // Detection of nested fences in markdown blocks
367                if in_markdown_block && nested_fence_start.is_none() && trimmed.starts_with("```") {
368                    // Check if this looks like a nested fence opening (has content after the backticks)
369                    let count = trimmed.chars().take_while(|&c| c == '`').count();
370                    let remaining = if trimmed.len() > count {
371                        trimmed[count..].trim()
372                    } else {
373                        ""
374                    };
375
376                    if !remaining.is_empty() {
377                        nested_fence_start = Some(i);
378                    }
379                }
380
381                // Check if we've found a nested fence end (only if we have a start)
382                if in_markdown_block
383                    && nested_fence_start.is_some()
384                    && nested_fence_end.is_none()
385                    && trimmed.starts_with("```")
386                    && trimmed.trim_start_matches('`').trim().is_empty()
387                {
388                    nested_fence_end = Some(i);
389                }
390
391                // Check if this line matches the closing fence pattern for the outer block
392                if trimmed.starts_with(&active_fence_type.to_string().repeat(3)) {
393                    let count = trimmed.chars().take_while(|&c| c == active_fence_type).count();
394                    let remaining = if trimmed.len() > count {
395                        trimmed[count..].trim()
396                    } else {
397                        ""
398                    };
399
400                    // A line is a closing fence if:
401                    // 1. It uses the same fence character as the opening fence
402                    // 2. It has at least as many fence characters as the opening fence
403                    // 3. It has no content after the fence characters (except for whitespace)
404                    // 4. Its indentation level is less than or equal to the opening fence
405                    let is_valid_closing_fence =
406                        count >= block_fence_length && remaining.is_empty() && indent <= block_indent;
407
408                    // For nested code blocks in markdown, the first backtick fence after the nested content
409                    // should be recognized as the closing fence for the outer block
410                    let is_nested_closing = nested_fence_end.is_some() && i == nested_fence_end.unwrap();
411
412                    // Skip nested closing fences
413                    if is_valid_closing_fence && !is_nested_closing {
414                        in_block = false;
415                        in_markdown_block = false;
416                    }
417                }
418            }
419        }
420
421        self.code_block_lines = Some(code_block_lines);
422    }
423}
424
425/// Calculate end position for a single-line range
426pub fn calculate_single_line_range(line: usize, start_col: usize, length: usize) -> (usize, usize, usize, usize) {
427    (line, start_col, line, start_col + length)
428}
429
430/// Calculate range for entire line.
431///
432/// The end column is a character count (rumdl's diagnostic convention), not a
433/// byte length, so the range is correct on lines containing multi-byte UTF-8.
434pub fn calculate_line_range(line: usize, line_content: &str) -> (usize, usize, usize, usize) {
435    let trimmed_char_len = line_content.trim_end().chars().count();
436    (line, 1, line, trimmed_char_len + 1)
437}
438
439/// Calculate range from regex match on a line
440///
441/// # Safety
442/// This function safely handles multi-byte UTF-8 characters by ensuring all
443/// string slicing operations occur at valid character boundaries.
444pub fn calculate_match_range(
445    line: usize,
446    line_content: &str,
447    match_start: usize,
448    match_len: usize,
449) -> (usize, usize, usize, usize) {
450    // Bounds check to prevent panic
451    let line_len = line_content.len();
452    if match_start > line_len {
453        // If match_start is beyond line bounds, return a safe range at end of line
454        let char_count = line_content.chars().count();
455        return (line, char_count + 1, line, char_count + 1);
456    }
457
458    // Find safe character boundaries for the match range
459    let safe_match_start = find_char_boundary(line_content, match_start);
460    let safe_match_end_byte = find_char_boundary(line_content, (match_start + match_len).min(line_len));
461
462    // Convert byte positions to character positions safely
463    let char_start = byte_to_char_count(line_content, safe_match_start);
464    let char_len = if safe_match_end_byte > safe_match_start {
465        // Count characters in the safe range
466        line_content[safe_match_start..safe_match_end_byte].chars().count()
467    } else {
468        0
469    };
470    (line, char_start, line, char_start + char_len)
471}
472
473/// Calculate range for trailing content (like trailing spaces)
474///
475/// # Safety
476/// This function safely handles multi-byte UTF-8 characters by ensuring all
477/// string slicing operations occur at valid character boundaries.
478pub fn calculate_trailing_range(line: usize, line_content: &str, content_end: usize) -> (usize, usize, usize, usize) {
479    // Find safe character boundary for content_end
480    let safe_content_end = find_char_boundary(line_content, content_end);
481    let char_content_end = byte_to_char_count(line_content, safe_content_end);
482    let line_char_len = line_content.chars().count() + 1;
483    (line, char_content_end, line, line_char_len)
484}
485
486/// Calculate range for a heading, from the start of the line its text begins on
487/// to the end of the text on the line it ends on.
488///
489/// An ATX heading holds its text on one line, so both are the same line. The
490/// text of a setext heading is the whole paragraph its underline ends, which can
491/// span several lines, and `last_line_content` is the last of them.
492pub fn calculate_heading_range(
493    first_line: usize,
494    last_line: usize,
495    last_line_content: &str,
496) -> (usize, usize, usize, usize) {
497    let trimmed_char_len = last_line_content.trim_end().chars().count();
498    (first_line, 1, last_line, trimmed_char_len + 1)
499}
500
501/// Calculate range for emphasis markers and content
502///
503/// # Safety
504/// This function safely handles multi-byte UTF-8 characters by ensuring all
505/// string slicing operations occur at valid character boundaries.
506pub fn calculate_emphasis_range(
507    line: usize,
508    line_content: &str,
509    start_pos: usize,
510    end_pos: usize,
511) -> (usize, usize, usize, usize) {
512    // Find safe character boundaries for start and end positions
513    let safe_start_pos = find_char_boundary(line_content, start_pos);
514    let safe_end_pos = find_char_boundary(line_content, end_pos);
515    let char_start = byte_to_char_count(line_content, safe_start_pos);
516    let char_end = byte_to_char_count(line_content, safe_end_pos);
517    (line, char_start, line, char_end)
518}
519
520/// Calculate range for HTML tags
521pub fn calculate_html_tag_range(
522    line: usize,
523    line_content: &str,
524    tag_start: usize,
525    tag_len: usize,
526) -> (usize, usize, usize, usize) {
527    calculate_match_range(line, line_content, tag_start, tag_len)
528}
529
530/// Calculate range for URLs
531pub fn calculate_url_range(
532    line: usize,
533    line_content: &str,
534    url_start: usize,
535    url_len: usize,
536) -> (usize, usize, usize, usize) {
537    calculate_match_range(line, line_content, url_start, url_len)
538}
539
540/// Calculate range for list markers
541pub fn calculate_list_marker_range(
542    line: usize,
543    line_content: &str,
544    marker_start: usize,
545    marker_len: usize,
546) -> (usize, usize, usize, usize) {
547    calculate_match_range(line, line_content, marker_start, marker_len)
548}
549
550/// Calculate range that exceeds a limit (like line length)
551pub fn calculate_excess_range(line: usize, line_content: &str, limit: usize) -> (usize, usize, usize, usize) {
552    let char_limit = std::cmp::min(limit, line_content.chars().count());
553    let line_char_len = line_content.chars().count() + 1;
554    (line, char_limit + 1, line, line_char_len)
555}
556
557#[cfg(test)]
558mod tests {
559    use super::*;
560
561    #[test]
562    fn test_single_line_range() {
563        let (start_line, start_col, end_line, end_col) = calculate_single_line_range(5, 10, 3);
564        assert_eq!(start_line, 5);
565        assert_eq!(start_col, 10);
566        assert_eq!(end_line, 5);
567        assert_eq!(end_col, 13);
568    }
569
570    #[test]
571    fn test_heading_range_over_one_line() {
572        let (start_line, start_col, end_line, end_col) = calculate_heading_range(4, 4, "# A heading  ");
573        assert_eq!((start_line, start_col, end_line, end_col), (4, 1, 4, 12));
574    }
575
576    #[test]
577    fn test_heading_range_over_several_lines() {
578        // The text of a setext heading is the whole paragraph its underline
579        // ends, so the range runs from the first of those lines to the end of
580        // the text on the last one.
581        let (start_line, start_col, end_line, end_col) = calculate_heading_range(4, 6, "second line  ");
582        assert_eq!((start_line, start_col, end_line, end_col), (4, 1, 6, 12));
583    }
584
585    #[test]
586    fn test_line_range() {
587        let content = "# This is a heading  ";
588        let (start_line, start_col, end_line, end_col) = calculate_line_range(1, content);
589        assert_eq!(start_line, 1);
590        assert_eq!(start_col, 1);
591        assert_eq!(end_line, 1);
592        assert_eq!(end_col, 20); // Trimmed length + 1
593    }
594
595    #[test]
596    fn test_line_range_non_ascii() {
597        // Issue #670: end column is a character count, not a byte length.
598        // "你好 heading" is 10 characters, so end_column is 11 (the byte length
599        // would be 16).
600        let content = "你好 heading  ";
601        let (_, start_col, _, end_col) = calculate_line_range(1, content);
602        assert_eq!(start_col, 1);
603        assert_eq!(end_col, 11);
604    }
605
606    #[test]
607    fn test_match_range() {
608        let content = "Text <div>content</div> more";
609        let tag_start = 5; // Position of '<'
610        let tag_len = 5; // Length of "<div>"
611        let (start_line, start_col, end_line, end_col) = calculate_match_range(1, content, tag_start, tag_len);
612        assert_eq!(start_line, 1);
613        assert_eq!(start_col, 6); // 1-indexed
614        assert_eq!(end_line, 1);
615        assert_eq!(end_col, 11); // 6 + 5
616    }
617
618    #[test]
619    fn test_trailing_range() {
620        let content = "Text content   "; // 3 trailing spaces
621        let content_end = 12; // End of "Text content"
622        let (start_line, start_col, end_line, end_col) = calculate_trailing_range(1, content, content_end);
623        assert_eq!(start_line, 1);
624        assert_eq!(start_col, 13); // content_end + 1 (1-indexed)
625        assert_eq!(end_line, 1);
626        assert_eq!(end_col, 16); // Total length + 1
627    }
628
629    #[test]
630    fn test_excess_range() {
631        let content = "This line is too long for the limit";
632        let limit = 20;
633        let (start_line, start_col, end_line, end_col) = calculate_excess_range(1, content, limit);
634        assert_eq!(start_line, 1);
635        assert_eq!(start_col, 21); // limit + 1
636        assert_eq!(end_line, 1);
637        assert_eq!(end_col, 36); // Total length + 1 (35 chars + 1 = 36)
638    }
639
640    #[test]
641    fn test_whole_line_range() {
642        let content = "Line 1\nLine 2\nLine 3";
643        let line_index = LineIndex::new(content);
644
645        // Test first line (includes newline)
646        let range = line_index.whole_line_range(1);
647        assert_eq!(range, 0..7); // "Line 1\n"
648
649        // Test middle line
650        let range = line_index.whole_line_range(2);
651        assert_eq!(range, 7..14); // "Line 2\n"
652
653        // Test last line (no newline)
654        let range = line_index.whole_line_range(3);
655        assert_eq!(range, 14..20); // "Line 3"
656    }
657
658    #[test]
659    fn test_line_content_range() {
660        let content = "Line 1\nLine 2\nLine 3";
661        let line_index = LineIndex::new(content);
662
663        // Test first line content (excludes newline)
664        let range = line_index.line_content_range(1);
665        assert_eq!(range, 0..6); // "Line 1"
666
667        // Test middle line content
668        let range = line_index.line_content_range(2);
669        assert_eq!(range, 7..13); // "Line 2"
670
671        // Test last line content
672        let range = line_index.line_content_range(3);
673        assert_eq!(range, 14..20); // "Line 3"
674    }
675
676    #[test]
677    fn test_line_text_range() {
678        let content = "Hello world\nAnother line";
679        let line_index = LineIndex::new(content);
680
681        // Test partial text in first line
682        let range = line_index.line_text_range(1, 1, 5); // "Hell"
683        assert_eq!(range, 0..4);
684
685        // Test partial text in second line
686        let range = line_index.line_text_range(2, 1, 7); // "Another"
687        assert_eq!(range, 12..18);
688
689        // Test bounds checking
690        let range = line_index.line_text_range(1, 1, 100); // Should clamp to line end
691        assert_eq!(range, 0..11); // "Hello world"
692    }
693
694    #[test]
695    fn test_calculate_match_range_bounds_checking() {
696        // Test case 1: match_start beyond line bounds
697        let line_content = "] not a link [";
698        let (line, start_col, end_line, end_col) = calculate_match_range(121, line_content, 57, 10);
699        assert_eq!(line, 121);
700        assert_eq!(start_col, 15); // line length + 1
701        assert_eq!(end_line, 121);
702        assert_eq!(end_col, 15); // same as start when out of bounds
703
704        // Test case 2: match extends beyond line end
705        let line_content = "short";
706        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 2, 10);
707        assert_eq!(line, 1);
708        assert_eq!(start_col, 3); // position 2 + 1
709        assert_eq!(end_line, 1);
710        assert_eq!(end_col, 6); // clamped to line length + 1
711
712        // Test case 3: normal case within bounds
713        let line_content = "normal text here";
714        let (line, start_col, end_line, end_col) = calculate_match_range(5, line_content, 7, 4);
715        assert_eq!(line, 5);
716        assert_eq!(start_col, 8); // position 7 + 1
717        assert_eq!(end_line, 5);
718        assert_eq!(end_col, 12); // position 7 + 4 + 1
719
720        // Test case 4: zero length match
721        let line_content = "test line";
722        let (line, start_col, end_line, end_col) = calculate_match_range(10, line_content, 5, 0);
723        assert_eq!(line, 10);
724        assert_eq!(start_col, 6); // position 5 + 1
725        assert_eq!(end_line, 10);
726        assert_eq!(end_col, 6); // same as start for zero length
727    }
728
729    // ============================================================================
730    // UTF-8 Multi-byte Character Tests (Issue #154)
731    // ============================================================================
732
733    #[test]
734    fn test_issue_154_korean_character_boundary() {
735        // Exact reproduction of issue #154: Korean character '후' (3 bytes: 18..21)
736        // The error was: "byte index 19 is not a char boundary; it is inside '후'"
737        let line_content = "- 2023 년 초 이후 주가 상승        +1,000% (10 배 상승)  ";
738
739        // Test match at byte 19 (middle of '후' character)
740        // This should not panic and should find the nearest character boundary
741        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 19, 1);
742
743        // Should successfully calculate without panicking
744        assert!(start_col > 0);
745        assert_eq!(line, 1);
746        assert_eq!(end_line, 1);
747        assert!(end_col >= start_col);
748    }
749
750    #[test]
751    fn test_calculate_match_range_korean() {
752        // Korean text: "안녕하세요" (Hello in Korean)
753        // Each character is 3 bytes
754        let line_content = "안녕하세요";
755        // Match at byte 3 (start of second character)
756        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 3, 3);
757        assert_eq!(line, 1);
758        assert_eq!(start_col, 2); // Second character (1-indexed)
759        assert_eq!(end_line, 1);
760        assert_eq!(end_col, 3); // End of second character
761
762        // Match at byte 4 (middle of second character - should round down)
763        let (line, start_col, end_line, _end_col) = calculate_match_range(1, line_content, 4, 3);
764        assert_eq!(line, 1);
765        assert_eq!(start_col, 2); // Should round to start of character
766        assert_eq!(end_line, 1);
767    }
768
769    #[test]
770    fn test_calculate_match_range_chinese() {
771        // Chinese text: "你好世界" (Hello World)
772        // Each character is 3 bytes
773        let line_content = "你好世界";
774        // Match at byte 6 (start of third character)
775        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 6, 3);
776        assert_eq!(line, 1);
777        assert_eq!(start_col, 3); // Third character (1-indexed)
778        assert_eq!(end_line, 1);
779        assert_eq!(end_col, 4); // End of third character
780    }
781
782    #[test]
783    fn test_calculate_match_range_japanese() {
784        // Japanese text: "こんにちは" (Hello)
785        // Each character is 3 bytes
786        let line_content = "こんにちは";
787        // Match at byte 9 (start of fourth character)
788        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 9, 3);
789        assert_eq!(line, 1);
790        assert_eq!(start_col, 4); // Fourth character (1-indexed)
791        assert_eq!(end_line, 1);
792        assert_eq!(end_col, 5); // End of fourth character
793    }
794
795    #[test]
796    fn test_calculate_match_range_mixed_unicode() {
797        // Mixed ASCII and CJK: "Hello 世界"
798        // "Hello " = 6 bytes (H, e, l, l, o, space)
799        // "世" = bytes 6-8 (3 bytes), character 7
800        // "界" = bytes 9-11 (3 bytes), character 8
801        let line_content = "Hello 世界";
802
803        // Match at byte 5 (space character)
804        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 5, 1);
805        assert_eq!(line, 1);
806        assert_eq!(start_col, 6); // Space character (1-indexed: H=1, e=2, l=3, l=4, o=5, space=6)
807        assert_eq!(end_line, 1);
808        assert_eq!(end_col, 7); // After space
809
810        // Match at byte 6 (start of first Chinese character "世")
811        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 6, 3);
812        assert_eq!(line, 1);
813        assert_eq!(start_col, 7); // First Chinese character (1-indexed)
814        assert_eq!(end_line, 1);
815        assert_eq!(end_col, 8); // End of first Chinese character
816    }
817
818    #[test]
819    fn test_calculate_trailing_range_korean() {
820        // Korean text with trailing spaces
821        let line_content = "안녕하세요   ";
822        // content_end at byte 15 (middle of last character + spaces)
823        let (line, start_col, end_line, end_col) = calculate_trailing_range(1, line_content, 15);
824        assert_eq!(line, 1);
825        assert!(start_col > 0);
826        assert_eq!(end_line, 1);
827        assert!(end_col > start_col);
828    }
829
830    #[test]
831    fn test_calculate_emphasis_range_chinese() {
832        // Chinese text with emphasis markers
833        let line_content = "这是**重要**的";
834        // start_pos and end_pos at byte boundaries within Chinese characters
835        let (line, start_col, end_line, end_col) = calculate_emphasis_range(1, line_content, 6, 12);
836        assert_eq!(line, 1);
837        assert!(start_col > 0);
838        assert_eq!(end_line, 1);
839        assert!(end_col > start_col);
840    }
841
842    #[test]
843    fn test_line_col_to_byte_range_korean() {
844        // Test that column positions (character positions) are correctly converted to byte positions
845        let content = "안녕하세요\nWorld";
846        let line_index = LineIndex::new(content);
847
848        // Column 1 (first character)
849        let range = line_index.line_col_to_byte_range(1, 1);
850        assert_eq!(range, 0..0);
851
852        // Column 2 (second character)
853        let range = line_index.line_col_to_byte_range(1, 2);
854        assert_eq!(range, 3..3); // 3 bytes for first character
855
856        // Column 3 (third character)
857        let range = line_index.line_col_to_byte_range(1, 3);
858        assert_eq!(range, 6..6); // 6 bytes for first two characters
859    }
860
861    #[test]
862    fn test_line_col_to_byte_range_with_length_chinese() {
863        // Test byte range calculation with length for Chinese characters
864        let content = "你好世界\nTest";
865        let line_index = LineIndex::new(content);
866
867        // Column 1, length 2 (first two Chinese characters)
868        let range = line_index.line_col_to_byte_range_with_length(1, 1, 2);
869        assert_eq!(range, 0..6); // 6 bytes for two 3-byte characters
870
871        // Column 2, length 1 (second Chinese character)
872        let range = line_index.line_col_to_byte_range_with_length(1, 2, 1);
873        assert_eq!(range, 3..6); // Bytes 3-6 for second character
874    }
875
876    #[test]
877    fn test_line_text_range_japanese() {
878        // Test text range calculation for Japanese characters
879        let content = "こんにちは\nHello";
880        let line_index = LineIndex::new(content);
881
882        // Columns 2-4 (second to fourth Japanese characters)
883        let range = line_index.line_text_range(1, 2, 4);
884        assert_eq!(range, 3..9); // Bytes 3-9 for three 3-byte characters
885    }
886
887    #[test]
888    fn test_find_char_boundary_edge_cases() {
889        // Test the helper function directly
890        let s = "안녕";
891
892        // Byte 0 (start) - should be valid
893        assert_eq!(find_char_boundary(s, 0), 0);
894
895        // Byte 1 (middle of first character) - should round down to 0
896        assert_eq!(find_char_boundary(s, 1), 0);
897
898        // Byte 2 (middle of first character) - should round down to 0
899        assert_eq!(find_char_boundary(s, 2), 0);
900
901        // Byte 3 (start of second character) - should be valid
902        assert_eq!(find_char_boundary(s, 3), 3);
903
904        // Byte 4 (middle of second character) - should round down to 3
905        assert_eq!(find_char_boundary(s, 4), 3);
906
907        // Byte beyond string length - should return string length
908        assert_eq!(find_char_boundary(s, 100), s.len());
909    }
910
911    #[test]
912    fn test_byte_to_char_count_unicode() {
913        // Test character counting with multi-byte characters
914        let s = "안녕하세요";
915
916        // Byte 0 (start) - 1 character
917        assert_eq!(byte_to_char_count(s, 0), 1);
918
919        // Byte 3 (start of second character) - 2 characters
920        assert_eq!(byte_to_char_count(s, 3), 2);
921
922        // Byte 6 (start of third character) - 3 characters
923        assert_eq!(byte_to_char_count(s, 6), 3);
924
925        // Byte 9 (start of fourth character) - 4 characters
926        assert_eq!(byte_to_char_count(s, 9), 4);
927
928        // Byte 12 (start of fifth character) - 5 characters
929        assert_eq!(byte_to_char_count(s, 12), 5);
930
931        // Byte 15 (end) - 6 characters (5 + 1 for 1-indexed)
932        assert_eq!(byte_to_char_count(s, 15), 6);
933    }
934
935    #[test]
936    fn test_all_range_functions_with_emoji() {
937        // Test with emoji (4-byte UTF-8 characters)
938        let line_content = "Hello 🎉 World 🌍";
939
940        // calculate_match_range
941        let (line, start_col, end_line, end_col) = calculate_match_range(1, line_content, 6, 4);
942        assert_eq!(line, 1);
943        assert!(start_col > 0);
944        assert_eq!(end_line, 1);
945        assert!(end_col > start_col);
946
947        // calculate_trailing_range
948        let (line, start_col, end_line, end_col) = calculate_trailing_range(1, line_content, 12);
949        assert_eq!(line, 1);
950        assert!(start_col > 0);
951        assert_eq!(end_line, 1);
952        assert!(end_col > start_col);
953
954        // calculate_emphasis_range
955        let (line, start_col, end_line, end_col) = calculate_emphasis_range(1, line_content, 0, 5);
956        assert_eq!(line, 1);
957        assert_eq!(start_col, 1);
958        assert_eq!(end_line, 1);
959        assert!(end_col > start_col);
960    }
961}