blob: cf371a253d98fc20d56c8245c0f6fe70d1749810 (
plain)
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
|
use std::iter::IntoIterator;
use rnix::{SyntaxNode, types::{TypedNode, self, TokenWrapper}};
fn ast_from_text<N: TypedNode>(text: &str) -> N {
let parse = rnix::parse(text);
let node = match parse.node().descendants().find_map(N::cast) {
Some(it) => it,
None => {
panic!("Failed to make ast node `{}` from text `{}`", std::any::type_name::<N>(), text)
}
};
node
}
pub fn parenthesize(node: &SyntaxNode) -> types::Paren {
ast_from_text(&format!("({})", node))
}
pub fn unary_not(node: &SyntaxNode) -> types::UnaryOp {
ast_from_text(&format!("!{}", node))
}
pub fn inherit_stmt<'a>(nodes: impl IntoIterator<Item = &'a types::Ident>) -> types::Inherit {
let inherited_idents = nodes
.into_iter()
.map(|i| i.as_str().to_owned())
.collect::<Vec<_>>()
.join(" ");
ast_from_text(&format!("{{ inherit {}; }}", inherited_idents))
}
pub fn inherit_from_stmt<'a>(from: SyntaxNode, nodes: impl IntoIterator<Item = &'a types::Ident>) -> types::Inherit {
let inherited_idents = nodes
.into_iter()
.map(|i| i.as_str().to_owned())
.collect::<Vec<_>>()
.join(" ");
ast_from_text(&format!("{{ inherit ({}) {}; }}", from, inherited_idents))
}
|
