language-servers/rust-analyzer
1
0
Fork 0
mirror of https://github.com/rust-lang/rust-analyzer.git synced 2025-09-30 22:01:37 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 3

Expand compile_error!

Browse source
This commit is contained in:
Jeremy Kolb 2019-11-24 19:01:51 -05:00
parent f7f9757b6b
commit 67d3600f59
2 changed files with 39 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

38
crates/ra_hir_expand/src/builtin_macro.rs
View file

@ -46,6 +46,7 @@ macro_rules! register_builtin {
register_builtin! { register_builtin! {
(COLUMN_MACRO, Column) => column_expand, (COLUMN_MACRO, Column) => column_expand,
(COMPILE_ERROR_MACRO, CompileError) => compile_error_expand,
(FILE_MACRO, File) => file_expand, (FILE_MACRO, File) => file_expand,
(LINE_MACRO, Line) => line_expand, (LINE_MACRO, Line) => line_expand,
(STRINGIFY_MACRO, Stringify) => stringify_expand (STRINGIFY_MACRO, Stringify) => stringify_expand
@ -172,6 +173,26 @@ fn file_expand(
Ok(expanded) Ok(expanded)
} }
fn compile_error_expand(
_db: &dyn AstDatabase,
_id: MacroCallId,
tt: &tt::Subtree,
) -> Result<tt::Subtree, mbe::ExpandError> {
if tt.count() == 1 {
match &tt.token_trees[0] {
tt::TokenTree::Leaf(tt::Leaf::Literal(it)) => {
let s = it.text.as_str();
if s.contains(r#"""#) {
return Ok(quote! { loop { #it }});
}
}
_ => {}
};
}
Err(mbe::ExpandError::BindingError("Must be a string".into()))
}
#[cfg(test)] #[cfg(test)]
mod tests { mod tests {
use super::*; use super::*;
@ -259,4 +280,21 @@ mod tests {
assert_eq!(expanded, "\"\""); assert_eq!(expanded, "\"\"");
} }
#[test]
fn test_compile_error_expand() {
let expanded = expand_builtin_macro(
r#"
#[rustc_builtin_macro]
macro_rules! compile_error {
($msg:expr) => ({ /* compiler built-in */ });
($msg:expr,) => ({ /* compiler built-in */ })
}
compile_error!("error!");
"#,
BuiltinFnLikeExpander::CompileError,
);
assert_eq!(expanded, r#"loop{"error!"}"#);
}
} }

1
crates/ra_hir_expand/src/name.rs
View file

@ -144,5 +144,6 @@ pub const BOX_TYPE: Name = Name::new_inline_ascii(3, b"Box");
// Builtin Macros // Builtin Macros
pub const FILE_MACRO: Name = Name::new_inline_ascii(4, b"file"); pub const FILE_MACRO: Name = Name::new_inline_ascii(4, b"file");
pub const COLUMN_MACRO: Name = Name::new_inline_ascii(6, b"column"); pub const COLUMN_MACRO: Name = Name::new_inline_ascii(6, b"column");
pub const COMPILE_ERROR_MACRO: Name = Name::new_inline_ascii(13, b"compile_error");
pub const LINE_MACRO: Name = Name::new_inline_ascii(4, b"line"); pub const LINE_MACRO: Name = Name::new_inline_ascii(4, b"line");
pub const STRINGIFY_MACRO: Name = Name::new_inline_ascii(9, b"stringify"); pub const STRINGIFY_MACRO: Name = Name::new_inline_ascii(9, b"stringify");