Add helper method to get a macro name from passed string

This commit is contained in:
btwotwo 2022-10-06 16:26:49 +02:00
parent 07621ce096
commit 08b0c92105
No known key found for this signature in database
GPG key ID: D456A0349D73A0A3

View file

@ -2,8 +2,8 @@
use itertools::Itertools;
use parser::T;
use syntax::{
ast::{self, HasLoopBody, PathSegmentKind, VisibilityKind},
AstNode, Preorder, RustLanguage, WalkEvent,
ast::{self, HasLoopBody, PathSegmentKind, VisibilityKind, NameRef},
AstNode, Preorder, RustLanguage, WalkEvent, AstToken,
};
pub fn expr_as_name_ref(expr: &ast::Expr) -> Option<ast::NameRef> {
@ -457,3 +457,10 @@ pub fn parse_tt_as_comma_sep_paths(input: ast::TokenTree) -> Option<Vec<ast::Pat
.collect();
Some(paths)
}
pub fn get_outer_macro_name(string: &ast::String) -> Option<NameRef> {
let macro_call = string.syntax().parent_ancestors().find_map(ast::MacroCall::cast)?;
let name = macro_call.path()?.segment()?.name_ref()?;
Some(name)
}