|
| 1 | +use rustc_feature::{AttributeTemplate, template}; |
| 2 | +use rustc_hir::attrs::{AttributeKind, OnMoveAttr}; |
| 3 | +use rustc_hir::lints::AttributeLintKind; |
| 4 | +use rustc_parse_format::{ParseMode, Parser, Piece, Position}; |
| 5 | +use rustc_session::lint::builtin::{ |
| 6 | + MALFORMED_DIAGNOSTIC_ATTRIBUTES, MALFORMED_DIAGNOSTIC_FORMAT_LITERALS, |
| 7 | +}; |
| 8 | +use rustc_span::{InnerSpan, Span, Symbol, kw, sym}; |
| 9 | +use thin_vec::ThinVec; |
| 10 | + |
| 11 | +use crate::attributes::{AttributeOrder, OnDuplicate, SingleAttributeParser}; |
| 12 | +use crate::context::{AcceptContext, Stage}; |
| 13 | +use crate::parser::ArgParser; |
| 14 | +use crate::target_checking::{ALL_TARGETS, AllowedTargets}; |
| 15 | + |
| 16 | +pub(crate) struct OnMoveParser; |
| 17 | + |
| 18 | +impl<S: Stage> SingleAttributeParser<S> for OnMoveParser { |
| 19 | + const PATH: &[Symbol] = &[sym::diagnostic, sym::on_move]; |
| 20 | + const ATTRIBUTE_ORDER: AttributeOrder = AttributeOrder::KeepInnermost; |
| 21 | + const ON_DUPLICATE: OnDuplicate<S> = OnDuplicate::Warn; |
| 22 | + const ALLOWED_TARGETS: AllowedTargets = AllowedTargets::AllowList(ALL_TARGETS); // Check in check_attr. |
| 23 | + const TEMPLATE: AttributeTemplate = template!(List: &["message", "label"]); |
| 24 | + |
| 25 | + fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> { |
| 26 | + let get_format_ranges = |cx: &mut AcceptContext<'_, '_, S>, |
| 27 | + symbol: &Symbol, |
| 28 | + span: Span| |
| 29 | + -> ThinVec<(usize, usize)> { |
| 30 | + let mut parser = Parser::new(symbol.as_str(), None, None, false, ParseMode::Diagnostic); |
| 31 | + let pieces: Vec<_> = parser.by_ref().collect(); |
| 32 | + let mut spans = ThinVec::new(); |
| 33 | + |
| 34 | + for piece in pieces { |
| 35 | + match piece { |
| 36 | + Piece::NextArgument(arg) => match arg.position { |
| 37 | + Position::ArgumentNamed(name) if name == kw::SelfUpper.as_str() => { |
| 38 | + spans.push((arg.position_span.start - 2, arg.position_span.end)); |
| 39 | + } |
| 40 | + Position::ArgumentNamed(name) => { |
| 41 | + let span = span.from_inner(InnerSpan { |
| 42 | + start: arg.position_span.start, |
| 43 | + end: arg.position_span.end, |
| 44 | + }); |
| 45 | + cx.emit_lint( |
| 46 | + MALFORMED_DIAGNOSTIC_FORMAT_LITERALS, |
| 47 | + AttributeLintKind::OnMoveMalformedFormatLiterals { |
| 48 | + name: Symbol::intern(name), |
| 49 | + }, |
| 50 | + span, |
| 51 | + ) |
| 52 | + } |
| 53 | + _ => {} |
| 54 | + }, |
| 55 | + _ => continue, |
| 56 | + } |
| 57 | + } |
| 58 | + spans |
| 59 | + }; |
| 60 | + let Some(list) = args.list() else { |
| 61 | + cx.expected_list(cx.attr_span, args); |
| 62 | + return None; |
| 63 | + }; |
| 64 | + let mut message = None; |
| 65 | + let mut label = None; |
| 66 | + |
| 67 | + for item in list.mixed() { |
| 68 | + let Some(item) = item.meta_item() else { |
| 69 | + cx.expected_specific_argument(item.span(), &[sym::message, sym::label]); |
| 70 | + return None; |
| 71 | + }; |
| 72 | + |
| 73 | + let Some(name_value) = item.args().name_value() else { |
| 74 | + cx.expected_name_value(cx.attr_span, item.path().word_sym()); |
| 75 | + return None; |
| 76 | + }; |
| 77 | + |
| 78 | + let Some(value) = name_value.value_as_str() else { |
| 79 | + cx.expected_string_literal(name_value.value_span, None); |
| 80 | + return None; |
| 81 | + }; |
| 82 | + |
| 83 | + let value_span = name_value.value_span; |
| 84 | + if item.path().word_is(sym::message) && message.is_none() { |
| 85 | + let spans = get_format_ranges(cx, &value, value_span); |
| 86 | + message = Some(OnMoveAttr::new(value, spans)); |
| 87 | + continue; |
| 88 | + } else if item.path().word_is(sym::label) && label.is_none() { |
| 89 | + let spans = get_format_ranges(cx, &value, value_span); |
| 90 | + label = Some(OnMoveAttr::new(value, spans)); |
| 91 | + continue; |
| 92 | + } |
| 93 | + |
| 94 | + cx.emit_lint( |
| 95 | + MALFORMED_DIAGNOSTIC_ATTRIBUTES, |
| 96 | + AttributeLintKind::OnMoveMalformedAttr, |
| 97 | + item.span(), |
| 98 | + ) |
| 99 | + } |
| 100 | + Some(AttributeKind::OnMove { span: cx.attr_span, message, label }) |
| 101 | + } |
| 102 | +} |
0 commit comments