1use std::str::{self, FromStr};
6
7use parser::Edition;
8
9use crate::{AstNode, SourceFile, TextRange, validation};
10
11fn check_file_invariants(file: &SourceFile) {
12 let root = file.syntax();
13 validation::validate_block_structure(root);
14}
15
16pub fn check_parser(text: &str) {
17 let file = SourceFile::parse(text, Edition::CURRENT);
18 check_file_invariants(&file.tree());
19}
20
21#[derive(Debug, Clone)]
22pub struct CheckReparse {
23 text: String,
24 delete: TextRange,
25 insert: String,
26 edited_text: String,
27}
28
29impl CheckReparse {
30 pub fn from_data(data: &[u8]) -> Option<Self> {
31 const PREFIX: &str = "fn main(){\n\t";
32 const SUFFIX: &str = "\n}";
33
34 let data = str::from_utf8(data).ok()?;
35 let mut lines = data.lines();
36 let delete_start = usize::from_str(lines.next()?).ok()? + PREFIX.len();
37 let delete_len = usize::from_str(lines.next()?).ok()?;
38 let insert = lines.next()?.to_owned();
39 let text = lines.collect::<Vec<_>>().join("\n");
40 let text = format!("{PREFIX}{text}{SUFFIX}");
41 text.get(delete_start..delete_start.checked_add(delete_len)?)?; let delete =
43 TextRange::at(delete_start.try_into().unwrap(), delete_len.try_into().unwrap());
44 let edited_text =
45 format!("{}{}{}", &text[..delete_start], &insert, &text[delete_start + delete_len..]);
46 Some(CheckReparse { text, insert, delete, edited_text })
47 }
48
49 #[allow(clippy::print_stderr)]
50 pub fn run(&self) {
51 let parse = SourceFile::parse(&self.text, Edition::CURRENT);
52 let new_parse = parse.reparse(self.delete, &self.insert, Edition::CURRENT);
53 check_file_invariants(&new_parse.tree());
54 assert_eq!(&new_parse.tree().syntax().text().to_string(), &self.edited_text);
55 let full_reparse = SourceFile::parse(&self.edited_text, Edition::CURRENT);
56 for (a, b) in
57 new_parse.tree().syntax().descendants().zip(full_reparse.tree().syntax().descendants())
58 {
59 if (a.kind(), a.text_range()) != (b.kind(), b.text_range()) {
60 eprint!("original:\n{:#?}", parse.tree().syntax());
61 eprint!("reparsed:\n{:#?}", new_parse.tree().syntax());
62 eprint!("full reparse:\n{:#?}", full_reparse.tree().syntax());
63 assert_eq!(
64 format!("{a:?}"),
65 format!("{b:?}"),
66 "different syntax tree produced by the full reparse"
67 );
68 }
69 }
70 }
73}