squawk_wasm/
lib.rs

1use line_index::LineIndex;
2use log::info;
3use serde::Serialize;
4use wasm_bindgen::prelude::*;
5use web_sys::js_sys::Error;
6
7#[wasm_bindgen(start)]
8pub fn run() {
9    use log::Level;
10
11    // When the `console_error_panic_hook` feature is enabled, we can call the
12    // `set_panic_hook` function at least once during initialization, and then
13    // we will get better error messages if our code ever panics.
14    //
15    // For more details see
16    // https://github.com/rustwasm/console_error_panic_hook#readme
17    #[cfg(feature = "console_error_panic_hook")]
18    console_error_panic_hook::set_once();
19    console_log::init_with_level(Level::Debug).expect("Initializing logger went wrong.");
20    info!("init!");
21}
22
23#[wasm_bindgen]
24pub fn dump_cst(text: String) -> String {
25    let parse = squawk_syntax::SourceFile::parse(&text);
26    format!("{:#?}", parse.syntax_node())
27}
28
29#[wasm_bindgen]
30pub fn dump_tokens(text: String) -> String {
31    let tokens = squawk_lexer::tokenize(&text);
32    let mut start = 0;
33    let mut out = String::new();
34    for token in tokens {
35        let end = start + token.len;
36        let content = &text[start as usize..(end) as usize];
37        out += &format!("{:?}@{start}..{end} {:?}\n", token.kind, content);
38        start += token.len;
39    }
40    out
41}
42
43#[allow(dead_code)]
44#[derive(Serialize)]
45enum Severity {
46    Hint,
47    Info,
48    Warning,
49    Error,
50}
51
52#[derive(Serialize)]
53struct LintError {
54    severity: Severity,
55    code: String,
56    message: String,
57    start_line_number: u32,
58    start_column: u32,
59    end_line_number: u32,
60    end_column: u32,
61    // used for the linter tab
62    range_start: usize,
63    // used for the linter tab
64    range_end: usize,
65    // used for the linter tab
66    messages: Vec<String>,
67    fix: Option<Fix>,
68}
69
70#[derive(Serialize)]
71struct Fix {
72    title: String,
73    edits: Vec<TextEdit>,
74}
75
76#[derive(Serialize)]
77struct TextEdit {
78    start_line_number: u32,
79    start_column: u32,
80    end_line_number: u32,
81    end_column: u32,
82    text: String,
83}
84
85#[wasm_bindgen]
86pub fn lint(text: String) -> Result<JsValue, Error> {
87    let mut linter = squawk_linter::Linter::with_all_rules();
88    let parse = squawk_syntax::SourceFile::parse(&text);
89    let parse_errors = parse.errors();
90
91    let line_index = LineIndex::new(&text);
92
93    // TODO: chain these with other stuff
94    let parse_errors = parse_errors.iter().map(|x| {
95        let range_start = x.range().start();
96        let range_end = x.range().end();
97        let start = line_index.line_col(range_start);
98        let end = line_index.line_col(range_end);
99        let start = line_index
100            .to_wide(line_index::WideEncoding::Utf16, start)
101            .unwrap();
102        let end = line_index
103            .to_wide(line_index::WideEncoding::Utf16, end)
104            .unwrap();
105        LintError {
106            severity: Severity::Error,
107            code: "syntax-error".to_string(),
108            message: x.message().to_string(),
109            start_line_number: start.line,
110            start_column: start.col,
111            end_line_number: end.line,
112            end_column: end.col,
113            range_start: range_start.into(),
114            range_end: range_end.into(),
115            messages: vec![],
116            fix: None,
117        }
118    });
119
120    let lint_errors = linter.lint(&parse, &text);
121    let errors = lint_errors.into_iter().map(|x| {
122        let start = line_index.line_col(x.text_range.start());
123        let end = line_index.line_col(x.text_range.end());
124        let start = line_index
125            .to_wide(line_index::WideEncoding::Utf16, start)
126            .unwrap();
127        let end = line_index
128            .to_wide(line_index::WideEncoding::Utf16, end)
129            .unwrap();
130
131        let messages = x.help.into_iter().collect();
132
133        let fix = x.fix.map(|fix| {
134            let edits = fix
135                .edits
136                .into_iter()
137                .map(|edit| {
138                    let start_pos = line_index.line_col(edit.text_range.start());
139                    let end_pos = line_index.line_col(edit.text_range.end());
140                    let start_wide = line_index
141                        .to_wide(line_index::WideEncoding::Utf16, start_pos)
142                        .unwrap();
143                    let end_wide = line_index
144                        .to_wide(line_index::WideEncoding::Utf16, end_pos)
145                        .unwrap();
146
147                    TextEdit {
148                        start_line_number: start_wide.line,
149                        start_column: start_wide.col,
150                        end_line_number: end_wide.line,
151                        end_column: end_wide.col,
152                        text: edit.text.unwrap_or_default(),
153                    }
154                })
155                .collect();
156
157            Fix {
158                title: fix.title,
159                edits,
160            }
161        });
162
163        LintError {
164            code: x.code.to_string(),
165            range_start: x.text_range.start().into(),
166            range_end: x.text_range.end().into(),
167            message: x.message.clone(),
168            messages,
169            // parser errors should be error
170            severity: Severity::Warning,
171            start_line_number: start.line,
172            start_column: start.col,
173            end_line_number: end.line,
174            end_column: end.col,
175            fix,
176        }
177    });
178
179    let mut errors_to_dump = errors.chain(parse_errors).collect::<Vec<_>>();
180    errors_to_dump.sort_by_key(|k| (k.start_line_number, k.start_column));
181
182    serde_wasm_bindgen::to_value(&errors_to_dump).map_err(into_error)
183}
184
185fn into_error<E: std::fmt::Display>(err: E) -> Error {
186    Error::new(&err.to_string())
187}
squawk_wasm/lib.rs

squawk_wasm/
lib.rs
