From 2426d421b41c396f106c554ca7d95455a26b78ba Mon Sep 17 00:00:00 2001 From: Young-Flash <871946895@qq.com> Date: Fri, 22 Dec 2023 21:02:13 +0800 Subject: [PATCH] fix: remove wrong comma after remove unnecessary braces --- crates/syntax/src/ast/node_ext.rs | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/crates/syntax/src/ast/node_ext.rs b/crates/syntax/src/ast/node_ext.rs index cfd919ad6f..a253827495 100644 --- a/crates/syntax/src/ast/node_ext.rs +++ b/crates/syntax/src/ast/node_ext.rs @@ -345,6 +345,12 @@ impl ast::UseTreeList { .is_some() } + pub fn comma(&self) -> impl Iterator { + self.syntax() + .children_with_tokens() + .filter_map(|it| it.into_token().filter(|it| it.kind() == T![,])) + } + /// Remove the unnecessary braces in current `UseTreeList` pub fn remove_unnecessary_braces(mut self) { let remove_brace_in_use_tree_list = |u: &ast::UseTreeList| { @@ -352,6 +358,7 @@ impl ast::UseTreeList { if use_tree_count == 1 { u.l_curly_token().map(ted::remove); u.r_curly_token().map(ted::remove); + u.comma().for_each(ted::remove); } };