traxys/tree-sitter
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Trim whitespace from tag source lines

Browse source
This commit is contained in:
Timothy Clem 2020-06-17 15:11:42 -07:00
parent 15202d0b38
commit 3c39b016a4
1 changed files with 19 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

20
tags/src/lib.rs
View file

@ -495,7 +495,16 @@ fn line_range(text: &[u8], index: usize, max_line_len: usize) -> Range<usize> {
let start = memrchr(b'\n', &text[0..index]).map_or(0, |i| i + 1);
let max_line_len = max_line_len.min(text.len() - start);
let end = start + memchr(b'\n', &text[start..(start + max_line_len)]).unwrap_or(max_line_len);
start..end
trim_start(text, start..end)
}
fn trim_start(text: &[u8], r: Range<usize>) -> Range<usize> {
for (index, c) in text[r.start..r.end].iter().enumerate() {
if !c.is_ascii_whitespace(){
return index..r.end
}
}
return r
}
#[cfg(test)]
@ -514,4 +523,13 @@ mod tests {
assert_eq!(line_range(text, 5, 10), 4..8);
assert_eq!(line_range(text, 11, 10), 9..14);
}
#[test]
fn test_get_line_trims() {
let text = b" foo\nbar\n";
assert_eq!(line_range(text, 0, 10), 3..6);
let text = b"\t func foo\nbar\n";
assert_eq!(line_range(text, 0, 10), 2..10);
}
}