diff --git a/compiler/rustc_parse_format/src/lib.rs b/compiler/rustc_parse_format/src/lib.rs index 84243e53faf..9f2aaca0acf 100644 --- a/compiler/rustc_parse_format/src/lib.rs +++ b/compiler/rustc_parse_format/src/lib.rs @@ -57,6 +57,13 @@ impl InnerWidthMapping { } } +/// Whether the input string is a literal. If yes, it contains the inner width mappings. +#[derive(Clone, PartialEq, Eq)] +enum InputStringKind { + NotALiteral, + Literal { width_mappings: Vec }, +} + /// The type of format string that we are parsing. #[derive(Copy, Clone, Debug, Eq, PartialEq)] pub enum ParseMode { @@ -307,7 +314,11 @@ impl<'a> Parser<'a> { append_newline: bool, mode: ParseMode, ) -> Parser<'a> { - let (width_map, is_literal) = find_width_map_from_snippet(s, snippet, style); + let input_string_kind = find_width_map_from_snippet(s, snippet, style); + let (width_map, is_literal) = match input_string_kind { + InputStringKind::Literal { width_mappings } => (width_mappings, true), + InputStringKind::NotALiteral => (Vec::new(), false), + }; Parser { mode, input: s, @@ -848,14 +859,14 @@ fn find_width_map_from_snippet( input: &str, snippet: Option, str_style: Option, -) -> (Vec, bool) { +) -> InputStringKind { let snippet = match snippet { Some(ref s) if s.starts_with('"') || s.starts_with("r\"") || s.starts_with("r#") => s, - _ => return (vec![], false), + _ => return InputStringKind::NotALiteral, }; if str_style.is_some() { - return (vec![], true); + return InputStringKind::Literal { width_mappings: Vec::new() }; } // Strip quotes. @@ -868,7 +879,7 @@ fn find_width_map_from_snippet( // Alternatively, we could just count the trailing newlines and only trim one from the input if they don't match up. let input_no_nl = input.trim_end_matches('\n'); let Ok(unescaped) = unescape_string(snippet) else { - return (vec![], false); + return InputStringKind::NotALiteral; }; let unescaped_no_nl = unescaped.trim_end_matches('\n'); @@ -876,7 +887,7 @@ fn find_width_map_from_snippet( if unescaped_no_nl != input_no_nl { // The source string that we're pointing at isn't our input, so spans pointing at it will be incorrect. // This can for example happen with proc macros that respan generated literals. - return (vec![], false); + return InputStringKind::NotALiteral; } let mut s = snippet.char_indices(); @@ -958,7 +969,7 @@ fn find_width_map_from_snippet( } } - (width_mappings, true) + InputStringKind::Literal { width_mappings } } fn unescape_string(string: &str) -> Result {