summaryrefslogtreecommitdiff
path: root/support/texlab/crates/texlab/src/util/line_index_ext.rs
blob: 8e6bb6da032a0588d5797e72441e6cd28d8c20b3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
use line_index::{LineCol, LineColUtf16, LineIndex};
use lsp_types::{Position, Range};
use rowan::{TextRange, TextSize};

pub trait LineIndexExt {
    fn offset_lsp(&self, line_col: Position) -> Option<TextSize>;

    fn offset_lsp_range(&self, line_col: Range) -> Option<TextRange>;

    fn line_col_lsp(&self, offset: TextSize) -> Option<Position>;

    fn line_col_lsp_range(&self, offset: TextRange) -> Option<Range>;
}

impl LineIndexExt for LineIndex {
    fn offset_lsp(&self, line_col: Position) -> Option<TextSize> {
        let line_col = LineColUtf16 {
            line: line_col.line,
            col: line_col.character,
        };

        let line_col = self.to_utf8(line_col)?;
        self.offset(line_col)
    }

    fn offset_lsp_range(&self, line_col: Range) -> Option<TextRange> {
        let start = self.offset_lsp(line_col.start)?;
        let end = self.offset_lsp(line_col.end)?;
        Some(TextRange::new(start, end))
    }

    fn line_col_lsp(&self, offset: TextSize) -> Option<Position> {
        let line_col = self.line_col(offset);
        let line_col = self.to_utf16(line_col)?;
        Some(Position::new(line_col.line, line_col.col))
    }

    fn line_col_lsp_range(&self, offset: TextRange) -> Option<Range> {
        let start = self.line_col_lsp(offset.start())?;
        let mut end = self.line_col_lsp(offset.end())?;
        if end.line != start.line && end.character == 0 {
            // Prefer keeping multi-line ranges on the same line
            let line_end = self.offset(LineCol {
                line: end.line,
                col: 0,
            })?;

            end = self.line_col_lsp(line_end - TextSize::from(1))?;
        }

        Some(Range::new(start, end))
    }
}