internal: Add some path constructors to SyntaxFactory

This commit is contained in:
Giga Bowser 2024-12-14 15:55:27 -05:00
parent 551f1d00dd
commit f5ff966dea

View file

@ -1,6 +1,6 @@
//! Wrappers over [`make`] constructors
use crate::{
ast::{self, make, HasGenericParams, HasName, HasTypeBounds, HasVisibility},
ast::{self, make, HasGenericArgs, HasGenericParams, HasName, HasTypeBounds, HasVisibility},
syntax_editor::SyntaxMappingBuilder,
AstNode, NodeOrToken, SyntaxKind, SyntaxNode, SyntaxToken,
};
@ -12,6 +12,10 @@ impl SyntaxFactory {
make::name(name).clone_for_update()
}
pub fn name_ref(&self, name: &str) -> ast::NameRef {
make::name_ref(name).clone_for_update()
}
pub fn ty(&self, text: &str) -> ast::Type {
make::ty(text).clone_for_update()
}
@ -46,6 +50,71 @@ impl SyntaxFactory {
ast
}
pub fn path_segment(&self, name_ref: ast::NameRef) -> ast::PathSegment {
let ast = make::path_segment(name_ref.clone()).clone_for_update();
if let Some(mut mapping) = self.mappings() {
let mut builder = SyntaxMappingBuilder::new(ast.syntax().clone());
builder.map_node(name_ref.syntax().clone(), ast.name_ref().unwrap().syntax().clone());
builder.finish(&mut mapping);
}
ast
}
pub fn path_segment_generics(
&self,
name_ref: ast::NameRef,
generic_arg_list: ast::GenericArgList,
) -> ast::PathSegment {
let ast::Type::PathType(path) = make::ty(&format!("{name_ref}{generic_arg_list}")) else {
unreachable!();
};
let ast = path.path().unwrap().segment().unwrap().clone_for_update();
if let Some(mut mapping) = self.mappings() {
let mut builder = SyntaxMappingBuilder::new(ast.syntax().clone());
builder.map_node(name_ref.syntax().clone(), ast.name_ref().unwrap().syntax().clone());
builder.map_node(
generic_arg_list.syntax().clone(),
ast.generic_arg_list().unwrap().syntax().clone(),
);
builder.finish(&mut mapping);
}
ast
}
pub fn path_unqualified(&self, segment: ast::PathSegment) -> ast::Path {
let ast = make::path_unqualified(segment.clone()).clone_for_update();
if let Some(mut mapping) = self.mappings() {
let mut builder = SyntaxMappingBuilder::new(ast.syntax().clone());
builder.map_node(segment.syntax().clone(), ast.segment().unwrap().syntax().clone());
builder.finish(&mut mapping);
}
ast
}
pub fn path_from_segments(
&self,
segments: impl IntoIterator<Item = ast::PathSegment>,
is_abs: bool,
) -> ast::Path {
let (segments, input) = iterator_input(segments);
let ast = make::path_from_segments(segments, is_abs).clone_for_update();
if let Some(mut mapping) = self.mappings() {
let mut builder = SyntaxMappingBuilder::new(ast.syntax().clone());
builder.map_children(input.into_iter(), ast.segments().map(|it| it.syntax().clone()));
builder.finish(&mut mapping);
}
ast
}
pub fn ident_pat(&self, ref_: bool, mut_: bool, name: ast::Name) -> ast::IdentPat {
let ast = make::ident_pat(ref_, mut_, name.clone()).clone_for_update();
@ -508,6 +577,13 @@ impl SyntaxFactory {
}
}
// `ext` constructors
impl SyntaxFactory {
pub fn ident_path(&self, ident: &str) -> ast::Path {
self.path_unqualified(self.path_segment(self.name_ref(ident)))
}
}
// We need to collect `input` here instead of taking `impl IntoIterator + Clone`,
// because if we took `impl IntoIterator + Clone`, that could be something like an
// `Iterator::map` with a closure that also makes use of a `SyntaxFactory` constructor.