wat_service 0.3.0

WebAssembly Text Format language service.
Documentation 
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

use super::find_meaningful_token;
use crate::{helpers, syntax_tree::SyntaxTreeCtx, uri::UrisCtx, LanguageService};
use line_index::LineIndex;
use lsp_types::{SelectionRange, SelectionRangeParams};
use wat_syntax::SyntaxNode;

impl LanguageService {
    /// Handler for `textDocument/selectionRange` request.
    pub fn selection_range(&self, params: SelectionRangeParams) -> Option<Vec<SelectionRange>> {
        let uri = self.uri(params.text_document.uri);
        let line_index = self.line_index(uri);
        let root = SyntaxNode::new_root(self.root(uri));
        Some(
            params
                .positions
                .into_iter()
                .filter_map(|position| {
                    find_meaningful_token(self, uri, &root, position).map(|token| SelectionRange {
                        range: helpers::rowan_range_to_lsp_range(&line_index, token.text_range()),
                        parent: token.parent().map(|parent| {
                            Box::new(SelectionRange {
                                range: helpers::rowan_range_to_lsp_range(
                                    &line_index,
                                    parent.text_range(),
                                ),
                                parent: get_parent_range(parent, &line_index),
                            })
                        }),
                    })
                })
                .collect(),
        )
    }
}

fn get_parent_range(current: SyntaxNode, line_index: &LineIndex) -> Option<Box<SelectionRange>> {
    current.parent().map(|parent| {
        Box::new(SelectionRange {
            range: helpers::rowan_range_to_lsp_range(line_index, parent.text_range()),
            parent: get_parent_range(parent, line_index),
        })
    })
}