Skip to content

Commit 74532fb

Browse files
scrabshascrabsha
committed
use ArgParser::single_element_list where possible
1 parent 011f3dc commit 74532fb

15 files changed

Lines changed: 146 additions & 137 deletions

File tree

‎compiler/rustc_attr_parsing/src/attributes/codegen_attrs.rs‎

Lines changed: 3 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -23,16 +23,7 @@ impl<S: Stage> SingleAttributeParser<S> for OptimizeParser {
2323
const TEMPLATE: AttributeTemplate = template!(List: &["size", "speed", "none"]);
2424

2525
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
26-
let Some(list) = args.list() else {
27-
let attr_span = cx.attr_span;
28-
cx.adcx().expected_list(attr_span, args);
29-
return None;
30-
};
31-
32-
let Some(single) = list.single() else {
33-
cx.adcx().expected_single_argument(list.span, list.len());
34-
return None;
35-
};
26+
let single = args.single_element_list(cx.inner_span, cx)?;
3627

3728
let res = match single.meta_item().and_then(|i| i.path().word().map(|i| i.name)) {
3829
Some(sym::size) => OptimizeAttr::Size,
@@ -84,22 +75,13 @@ impl<S: Stage> SingleAttributeParser<S> for CoverageParser {
8475
const TEMPLATE: AttributeTemplate = template!(OneOf: &[sym::off, sym::on]);
8576

8677
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
87-
let Some(args) = args.list() else {
88-
let attr_span = cx.attr_span;
89-
cx.adcx().expected_specific_argument_and_list(attr_span, &[sym::on, sym::off]);
90-
return None;
91-
};
92-
93-
let Some(arg) = args.single() else {
94-
cx.adcx().expected_single_argument(args.span, args.len());
95-
return None;
96-
};
78+
let arg = args.single_element_list(cx.inner_span, cx)?;
9779

9880
let mut fail_incorrect_argument =
9981
|span| cx.adcx().expected_specific_argument(span, &[sym::on, sym::off]);
10082

10183
let Some(arg) = arg.meta_item() else {
102-
fail_incorrect_argument(args.span);
84+
fail_incorrect_argument(arg.span());
10385
return None;
10486
};
10587

‎compiler/rustc_attr_parsing/src/attributes/debugger.rs‎

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -20,15 +20,7 @@ impl<S: Stage> CombineAttributeParser<S> for DebuggerViualizerParser {
2020
cx: &mut AcceptContext<'_, '_, S>,
2121
args: &ArgParser,
2222
) -> impl IntoIterator<Item = Self::Item> {
23-
let Some(l) = args.list() else {
24-
let attr_span = cx.attr_span;
25-
cx.adcx().expected_list(attr_span, args);
26-
return None;
27-
};
28-
let Some(single) = l.single() else {
29-
cx.adcx().expected_single_argument(l.span, l.len());
30-
return None;
31-
};
23+
let single = args.single_element_list(cx.inner_span, cx)?;
3224
let Some(mi) = single.meta_item() else {
3325
cx.adcx().expected_name_value(single.span(), None);
3426
return None;

‎compiler/rustc_attr_parsing/src/attributes/instruction_set.rs‎

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -20,11 +20,7 @@ impl<S: Stage> SingleAttributeParser<S> for InstructionSetParser {
2020
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
2121
const POSSIBLE_SYMBOLS: &[Symbol] = &[sym::arm_a32, sym::arm_t32];
2222
const POSSIBLE_ARM_SYMBOLS: &[Symbol] = &[sym::a32, sym::t32];
23-
let Some(maybe_meta_item) = args.list().and_then(MetaItemListParser::single) else {
24-
let attr_span = cx.attr_span;
25-
cx.adcx().expected_specific_argument(attr_span, POSSIBLE_SYMBOLS);
26-
return None;
27-
};
23+
let maybe_meta_item = args.single_element_list(cx.inner_span, cx)?;
2824

2925
let Some(meta_item) = maybe_meta_item.meta_item() else {
3026
cx.adcx().expected_specific_argument(maybe_meta_item.span(), POSSIBLE_SYMBOLS);

‎compiler/rustc_attr_parsing/src/attributes/link_attrs.rs‎

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -388,12 +388,7 @@ impl LinkParser {
388388
cx.adcx().duplicate_key(item.span(), sym::cfg);
389389
return true;
390390
}
391-
let Some(link_cfg) = item.args().list() else {
392-
cx.adcx().expected_list(item.span(), item.args());
393-
return true;
394-
};
395-
let Some(link_cfg) = link_cfg.single() else {
396-
cx.adcx().expected_single_argument(item.span(), link_cfg.len());
391+
let Some(link_cfg) = item.args().single_element_list(item.span(), cx) else {
397392
return true;
398393
};
399394
if !features.link_cfg() {

‎compiler/rustc_attr_parsing/src/attributes/macro_attrs.rs‎

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -175,15 +175,7 @@ impl<S: Stage> SingleAttributeParser<S> for CollapseDebugInfoParser {
175175
const ALLOWED_TARGETS: AllowedTargets = AllowedTargets::AllowList(&[Allow(Target::MacroDef)]);
176176

177177
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
178-
let Some(list) = args.list() else {
179-
let attr_span = cx.attr_span;
180-
cx.adcx().expected_list(attr_span, args);
181-
return None;
182-
};
183-
let Some(single) = list.single() else {
184-
cx.adcx().expected_single_argument(list.span, list.len());
185-
return None;
186-
};
178+
let single = args.single_element_list(cx.inner_span, cx)?;
187179
let Some(mi) = single.meta_item() else {
188180
cx.adcx().unexpected_literal(single.span());
189181
return None;

‎compiler/rustc_attr_parsing/src/attributes/rustc_internal.rs‎

Lines changed: 4 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -195,11 +195,7 @@ impl<S: Stage> SingleAttributeParser<S> for RustcLintOptDenyFieldAccessParser {
195195
const ALLOWED_TARGETS: AllowedTargets = AllowedTargets::AllowList(&[Allow(Target::Field)]);
196196
const TEMPLATE: AttributeTemplate = template!(Word);
197197
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
198-
let Some(arg) = args.list().and_then(MetaItemListParser::single) else {
199-
let attr_span = cx.attr_span;
200-
cx.adcx().expected_single_argument(attr_span, 2);
201-
return None;
202-
};
198+
let arg = args.single_element_list(cx.inner_span, cx)?;
203199

204200
let MetaItemOrLitParser::Lit(MetaItemLit { kind: LitKind::Str(lint_message, _), .. }) = arg
205201
else {
@@ -375,19 +371,10 @@ impl<S: Stage> SingleAttributeParser<S> for RustcDeprecatedSafe2024Parser {
375371
const TEMPLATE: AttributeTemplate = template!(List: &[r#"audit_that = "...""#]);
376372

377373
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
378-
let Some(args) = args.list() else {
379-
let attr_span = cx.attr_span;
380-
cx.adcx().expected_list(attr_span, args);
381-
return None;
382-
};
383-
384-
let Some(single) = args.single() else {
385-
cx.adcx().expected_single_argument(args.span, args.len());
386-
return None;
387-
};
374+
let single = args.single_element_list(cx.inner_span, cx)?;
388375

389376
let Some(arg) = single.meta_item() else {
390-
cx.adcx().expected_name_value(args.span, None);
377+
cx.adcx().expected_name_value(single.span(), None);
391378
return None;
392379
};
393380

@@ -1082,11 +1069,7 @@ impl<S: Stage> CombineAttributeParser<S> for RustcThenThisWouldNeedParser {
10821069
if !cx.cx.sess.opts.unstable_opts.query_dep_graph {
10831070
cx.emit_err(AttributeRequiresOpt { span: cx.attr_span, opt: "-Z query-dep-graph" });
10841071
}
1085-
let Some(item) = args.list().and_then(|l| l.single()) else {
1086-
let inner_span = cx.inner_span;
1087-
cx.adcx().expected_single_argument(inner_span, 2);
1088-
return None;
1089-
};
1072+
let item = args.single_element_list(cx.inner_span, cx)?;
10901073
let Some(ident) = item.meta_item().and_then(|item| item.ident()) else {
10911074
cx.adcx().expected_identifier(item.span());
10921075
return None;

‎compiler/rustc_attr_parsing/src/attributes/test_attrs.rs‎

Lines changed: 1 addition & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -201,16 +201,7 @@ impl<S: Stage> SingleAttributeParser<S> for TestRunnerParser {
201201
const TEMPLATE: AttributeTemplate = template!(List: &["path"]);
202202

203203
fn convert(cx: &mut AcceptContext<'_, '_, S>, args: &ArgParser) -> Option<AttributeKind> {
204-
let Some(list) = args.list() else {
205-
let attr_span = cx.attr_span;
206-
cx.adcx().expected_list(attr_span, args);
207-
return None;
208-
};
209-
210-
let Some(single) = list.single() else {
211-
cx.adcx().expected_single_argument(list.span, list.len());
212-
return None;
213-
};
204+
let single = args.single_element_list(cx.inner_span, cx)?;
214205

215206
let Some(meta) = single.meta_item() else {
216207
cx.adcx().unexpected_literal(single.span());

‎compiler/rustc_attr_parsing/src/attributes/util.rs‎

Lines changed: 1 addition & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -41,15 +41,7 @@ pub(crate) fn parse_single_integer<S: Stage>(
4141
cx: &mut AcceptContext<'_, '_, S>,
4242
args: &ArgParser,
4343
) -> Option<u128> {
44-
let Some(list) = args.list() else {
45-
let attr_span = cx.attr_span;
46-
cx.adcx().expected_list(attr_span, args);
47-
return None;
48-
};
49-
let Some(single) = list.single() else {
50-
cx.adcx().expected_single_argument(list.span, list.len());
51-
return None;
52-
};
44+
let single = args.single_element_list(cx.inner_span, cx)?;
5345
let Some(lit) = single.lit() else {
5446
cx.adcx().expected_integer_literal(single.span());
5547
return None;

‎compiler/rustc_attr_parsing/src/context.rs‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -769,7 +769,6 @@ where
769769
self.emit_parse_error(span, reason)
770770
}
771771

772-
/// Signals that there is no argument and that we expected one or more.
773772
pub(crate) fn expected_at_least_one_argument(&mut self, span: Span) -> ErrorGuaranteed {
774773
self.emit_parse_error(span, AttributeParseErrorReason::ExpectedAtLeastOneArgument)
775774
}

‎compiler/rustc_attr_parsing/src/parser.rs‎

Lines changed: 31 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,7 @@ use rustc_span::{Ident, Span, Symbol, sym};
2222
use thin_vec::ThinVec;
2323

2424
use crate::ShouldEmit;
25+
use crate::context::{AcceptContext, Stage};
2526
use crate::session_diagnostics::{
2627
InvalidMetaItem, InvalidMetaItemQuoteIdentSugg, InvalidMetaItemRemoveNegSugg, MetaBadDelim,
2728
MetaBadDelimSugg, SuffixedLiteralInAttribute,
@@ -183,6 +184,36 @@ impl ArgParser {
183184
}
184185
}
185186

187+
/// Asserts that this MetaItem is a list that contains a single element. Emits an error and
188+
/// returns `None` if it is not the case.
189+
///
190+
/// Some examples:
191+
///
192+
/// - In `#[allow(warnings)]`, `warnings` is returned
193+
/// - In `#[cfg_attr(docsrs, doc = "foo")]`, `None` is returned, "expected a single argument
194+
/// here" is emitted.
195+
/// - In `#[cfg()]`, `None` is returned, "expected an argument here" is emitted.
196+
pub fn single_element_list<'f, 'sess, S>(
197+
&self,
198+
span: Span,
199+
cx: &mut AcceptContext<'f, 'sess, S>,
200+
) -> Option<&MetaItemOrLitParser>
201+
where
202+
S: Stage,
203+
{
204+
let Self::List(l) = self else {
205+
cx.adcx().expected_list(span, self);
206+
return None;
207+
};
208+
209+
let Some(single) = l.single() else {
210+
cx.adcx().expected_single_argument(l.span, l.len());
211+
return None;
212+
};
213+
214+
Some(single)
215+
}
216+
186217
/// Asserts that this MetaItem is a name-value pair.
187218
///
188219
/// Some examples:

0 commit comments

Comments
 (0)