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
44
45
46
47
48
49
50
51
52
53
54

use crate::{helpers, syntax_tree::SyntaxTreeCtx, uri::UrisCtx, LanguageService};
use lsp_types::{
    DocumentFormattingParams, DocumentRangeFormattingParams, FormattingOptions, TextEdit,
};
use rowan::ast::AstNode;
use wat_formatter::config::{FormatOptions, LanguageOptions, LayoutOptions};
use wat_syntax::{ast::Root, SyntaxNode};

impl LanguageService {
    /// Handler for `textDocument/formatting` request.
    pub fn formatting(&self, params: DocumentFormattingParams) -> Option<Vec<TextEdit>> {
        let uri = self.uri(params.text_document.uri);
        let line_index = self.line_index(uri);
        let root = Root::cast(SyntaxNode::new_root(self.root(uri)))?;
        let formatted = wat_formatter::format(
            &root,
            &build_options(&params.options, self.get_config(uri).format.clone()),
        );
        let text_edit = TextEdit {
            range: helpers::rowan_range_to_lsp_range(&line_index, root.syntax().text_range()),
            new_text: formatted,
        };
        Some(vec![text_edit])
    }

    /// Handler for `textDocument/rangeFormatting` request.
    pub fn range_formatting(&self, params: DocumentRangeFormattingParams) -> Option<Vec<TextEdit>> {
        let uri = self.uri(params.text_document.uri);
        let line_index = self.line_index(uri);
        let root = Root::cast(SyntaxNode::new_root(self.root(uri)))?;
        let (formatted, range) = wat_formatter::format_range(
            &root,
            &build_options(&params.options, self.get_config(uri).format.clone()),
            helpers::lsp_range_to_rowan_range(&line_index, params.range)?,
            &line_index,
        )?;
        let text_edit = TextEdit {
            range: helpers::rowan_range_to_lsp_range(&line_index, range),
            new_text: formatted,
        };
        Some(vec![text_edit])
    }
}

fn build_options(layout: &FormattingOptions, language: LanguageOptions) -> FormatOptions {
    FormatOptions {
        layout: LayoutOptions {
            indent_width: layout.tab_size as usize,
            use_tabs: !layout.insert_spaces,
            ..Default::default()
        },
        language,
    }
}