mirror of
https://github.com/rust-lang/rust.git
synced 2025-05-01 04:27:38 +00:00
358a603f11
Instead of `ast::Lit`. Literal lowering now happens at two different times. Expression literals are lowered when HIR is crated. Attribute literals are lowered during parsing. This commit changes the language very slightly. Some programs that used to not compile now will compile. This is because some invalid literals that are removed by `cfg` or attribute macros will no longer trigger errors. See this comment for more details: https://github.com/rust-lang/rust/pull/102944#issuecomment-1277476773
69 lines
2.4 KiB
Rust
69 lines
2.4 KiB
Rust
use rustc_ast as ast;
|
|
use rustc_ast::tokenstream::TokenStream;
|
|
use rustc_expand::base::{self, DummyResult};
|
|
use rustc_session::errors::report_lit_error;
|
|
use rustc_span::symbol::Symbol;
|
|
|
|
use std::string::String;
|
|
|
|
pub fn expand_concat(
|
|
cx: &mut base::ExtCtxt<'_>,
|
|
sp: rustc_span::Span,
|
|
tts: TokenStream,
|
|
) -> Box<dyn base::MacResult + 'static> {
|
|
let Some(es) = base::get_exprs_from_tts(cx, sp, tts) else {
|
|
return DummyResult::any(sp);
|
|
};
|
|
let mut accumulator = String::new();
|
|
let mut missing_literal = vec![];
|
|
let mut has_errors = false;
|
|
for e in es {
|
|
match e.kind {
|
|
ast::ExprKind::Lit(token_lit) => match ast::LitKind::from_token_lit(token_lit) {
|
|
Ok(ast::LitKind::Str(ref s, _) | ast::LitKind::Float(ref s, _)) => {
|
|
accumulator.push_str(s.as_str());
|
|
}
|
|
Ok(ast::LitKind::Char(c)) => {
|
|
accumulator.push(c);
|
|
}
|
|
Ok(ast::LitKind::Int(i, _)) => {
|
|
accumulator.push_str(&i.to_string());
|
|
}
|
|
Ok(ast::LitKind::Bool(b)) => {
|
|
accumulator.push_str(&b.to_string());
|
|
}
|
|
Ok(ast::LitKind::Byte(..) | ast::LitKind::ByteStr(..)) => {
|
|
cx.span_err(e.span, "cannot concatenate a byte string literal");
|
|
has_errors = true;
|
|
}
|
|
Ok(ast::LitKind::Err) => {
|
|
has_errors = true;
|
|
}
|
|
Err(err) => {
|
|
report_lit_error(&cx.sess.parse_sess, err, token_lit, e.span);
|
|
has_errors = true;
|
|
}
|
|
},
|
|
ast::ExprKind::IncludedBytes(..) => {
|
|
cx.span_err(e.span, "cannot concatenate a byte string literal")
|
|
}
|
|
ast::ExprKind::Err => {
|
|
has_errors = true;
|
|
}
|
|
_ => {
|
|
missing_literal.push(e.span);
|
|
}
|
|
}
|
|
}
|
|
if !missing_literal.is_empty() {
|
|
let mut err = cx.struct_span_err(missing_literal, "expected a literal");
|
|
err.note("only literals (like `\"foo\"`, `42` and `3.14`) can be passed to `concat!()`");
|
|
err.emit();
|
|
return DummyResult::any(sp);
|
|
} else if has_errors {
|
|
return DummyResult::any(sp);
|
|
}
|
|
let sp = cx.with_def_site_ctxt(sp);
|
|
base::MacEager::expr(cx.expr_str(sp, Symbol::intern(&accumulator)))
|
|
}
|