ra_syntax/
utils.rs

1use crate::{File, SyntaxKind, SyntaxNodeRef, WalkEvent};
2use std::fmt::Write;
3
4/// Parse a file and create a string representation of the resulting parse tree.
5pub fn dump_tree(syntax: SyntaxNodeRef) -> String {
6    let mut errors: Vec<_> = syntax.root_data().to_vec();
7    errors.sort_by_key(|e| e.offset);
8    let mut err_pos = 0;
9    let mut level = 0;
10    let mut buf = String::new();
11    macro_rules! indent {
12        () => {
13            for _ in 0..level {
14                buf.push_str("  ");
15            }
16        };
17    }
18
19    for event in syntax.preorder() {
20        match event {
21            WalkEvent::Enter(node) => {
22                indent!();
23                writeln!(buf, "{:?}", node).unwrap();
24                if node.first_child().is_none() {
25                    let off = node.range().end();
26                    while err_pos < errors.len() && errors[err_pos].offset <= off {
27                        indent!();
28                        writeln!(buf, "err: `{}`", errors[err_pos].msg).unwrap();
29                        err_pos += 1;
30                    }
31                }
32                level += 1;
33            }
34            WalkEvent::Leave(_) => level -= 1,
35        }
36    }
37
38    assert_eq!(level, 0);
39    for err in errors[err_pos..].iter() {
40        writeln!(buf, "err: `{}`", err.msg).unwrap();
41    }
42
43    buf
44}
45
46pub fn check_fuzz_invariants(text: &str) {
47    let file = File::parse(text);
48    let root = file.syntax();
49    validate_block_structure(root);
50    let _ = file.ast();
51    let _ = file.errors();
52}
53
54pub(crate) fn validate_block_structure(root: SyntaxNodeRef) {
55    let mut stack = Vec::new();
56    for node in root.descendants() {
57        match node.kind() {
58            SyntaxKind::L_CURLY => stack.push(node),
59            SyntaxKind::R_CURLY => {
60                if let Some(pair) = stack.pop() {
61                    assert_eq!(
62                        node.parent(),
63                        pair.parent(),
64                        "\nunpaired curleys:\n{}\n{}\n",
65                        root.text(),
66                        dump_tree(root),
67                    );
68                    assert!(
69                        node.next_sibling().is_none() && pair.prev_sibling().is_none(),
70                        "\nfloating curlys at {:?}\nfile:\n{}\nerror:\n{}\n",
71                        node,
72                        root.text(),
73                        node.text(),
74                    );
75                }
76            }
77            _ => (),
78        }
79    }
80}
ra_syntax/utils.rs

ra_syntax/
utils.rs
