wat_service 0.10.1

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
55
56
57

use crate::{LanguageService, helpers::LineIndexExt, uri::InternUri};
use lspt::{DocumentFormattingParams, DocumentRangeFormattingParams, FormattingOptions, TextEdit};
use wat_formatter::config::{FormatOptions, LanguageOptions, LayoutOptions};
use wat_syntax::{
    TextRange,
    ast::{AstNode, Root},
};

impl LanguageService {
    /// Handler for `textDocument/formatting` request.
    pub fn formatting(&self, params: DocumentFormattingParams) -> Option<Vec<TextEdit>> {
        let uri = InternUri::new(self, params.text_document.uri);
        let document = self.get_document(uri)?;
        let configs = self.configs.read();
        let config = configs.get(&uri)?.unwrap_or_global(self);
        let line_index = document.line_index(self);
        let root = document.root(self);
        let formatted = wat_formatter::format(&root, &build_options(&params.options, config.format.clone()));
        let text_edit = TextEdit {
            range: line_index.convert(TextRange::new(0.into(), root.text_len())),
            new_text: formatted,
        };
        Some(vec![text_edit])
    }

    /// Handler for `textDocument/rangeFormatting` request.
    pub fn range_formatting(&self, params: DocumentRangeFormattingParams) -> Option<Vec<TextEdit>> {
        let uri = InternUri::new(self, params.text_document.uri);
        let document = self.get_document(uri)?;
        let configs = self.configs.read();
        let config = configs.get(&uri)?.unwrap_or_global(self);
        let line_index = document.line_index(self);
        let root = Root::cast(document.root_tree(self))?;
        let (formatted, range) = wat_formatter::format_range(
            &root,
            &build_options(&params.options, config.format.clone()),
            line_index.convert(params.range)?,
            line_index,
        )?;
        let text_edit = TextEdit {
            range: line_index.convert(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,
    }
}