mikros/rust
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

use SyntaxKind instead of "Self" literal comparison

Browse Source
This commit is contained in:
Young-Flash 2023-12-27 19:49:50 +08:00
parent 67f001e5ec
commit a38a79e2ba
1 changed files with 15 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
crates/ide-assists/src/handlers/extract_struct_from_enum_variant.rs
View File

@ -250,24 +250,21 @@ fn create_struct_def(
field_list field_list
.fields() .fields()
.filter_map(|field| match field.ty()? { .filter_map(|field| match field.ty()? {
ast::Type::PathType(p) => { ast::Type::PathType(p) => Some(
let generic_arg_list = p.path()?.segment()?.generic_arg_list()?; p.syntax()
Some( .descendants_with_tokens()
generic_arg_list .filter_map(|it| {
.generic_args() if it.kind() == T![Self] {
.filter_map(|generic_arg| { let type_arg =
if generic_arg.to_string() == "Self" { make::type_arg(make::ty(&enum_.name()?.to_string()))
let type_arg = .clone_for_update();
make::type_arg(make::ty(&enum_.name()?.to_string())) Some(ted::replace(it, type_arg.syntax()))
.clone_for_update(); } else {
Some(ted::replace(generic_arg.syntax(), type_arg.syntax())) None
} else { }
None })
} .count(),
}) ),
.count(),
)
}
_ => None, _ => None,
}) })
.count(); .count();