summaryrefslogtreecommitdiff
path: root/support/texlab/crates/texlab/src/util/line_index_ext.rs
blob: bf4d9fa422d8542f659b3e1b0814bc7999b90e3d (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
use base_db::{LineCol, LineColUtf16, LineIndex};
use lsp_types::{Position, Range};
use rowan::{TextRange, TextSize};

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

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

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

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

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

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

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

    fn line_col_lsp_range(&self, offset: TextRange) -> 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));
        }

        Range::new(start, end)
    }
}