From dabb96fdaf9a76bf7315bc0a5c8cb0a3974670ad Mon Sep 17 00:00:00 2001
From: David Tolnay <dtolnay@gmail.com>
Date: Sat, 2 Nov 2024 20:41:34 -0700
Subject: [PATCH] Use syn's real expression parser if it has full syntax
 support

---
 impl/src/fmt.rs | 30 +++++++++++++++++++++++++++---
 1 file changed, 27 insertions(+), 3 deletions(-)

diff --git a/impl/src/fmt.rs b/impl/src/fmt.rs
index 029870b..df75044 100644
--- a/impl/src/fmt.rs
+++ b/impl/src/fmt.rs
@@ -1,12 +1,12 @@
 use crate::ast::Field;
 use crate::attr::{Display, Trait};
-use crate::scan_expr::scan_expr;
+use crate::scan_expr;
 use proc_macro2::TokenTree;
-use quote::{format_ident, quote_spanned};
+use quote::{format_ident, quote, quote_spanned};
 use std::collections::{BTreeSet as Set, HashMap as Map};
 use syn::ext::IdentExt;
 use syn::parse::{ParseStream, Parser};
-use syn::{Ident, Index, LitStr, Member, Result, Token};
+use syn::{Expr, Ident, Index, LitStr, Member, Result, Token};
 
 impl Display<'_> {
     // Transform `"error {var}"` to `"error {}", var`.
@@ -119,6 +119,12 @@ impl Display<'_> {
 }
 
 fn explicit_named_args(input: ParseStream) -> Result<Set<Ident>> {
+    let scan_expr = if is_syn_full() {
+        |input: ParseStream| input.parse::<Expr>().map(drop)
+    } else {
+        scan_expr::scan_expr
+    };
+
     let mut named_args = Set::new();
 
     while !input.is_empty() {
@@ -137,6 +143,24 @@ fn explicit_named_args(input: ParseStream) -> Result<Set<Ident>> {
     Ok(named_args)
 }
 
+fn is_syn_full() -> bool {
+    // Expr::Block contains syn::Block which contains Vec<syn::Stmt>. In the
+    // current version of Syn, syn::Stmt is exhaustive and could only plausibly
+    // represent `trait Trait {}` in Stmt::Item which contains syn::Item. Most
+    // of the point of syn's non-"full" mode is to avoid compiling Item and the
+    // entire expansive syntax tree it comprises. So the following expression
+    // being parsed to Expr::Block is a reliable indication that "full" is
+    // enabled.
+    let test = quote!({
+        trait Trait {}
+    });
+    match syn::parse2(test) {
+        Ok(Expr::Verbatim(_)) | Err(_) => false,
+        Ok(Expr::Block(_)) => true,
+        Ok(_) => unreachable!(),
+    }
+}
+
 fn take_int(read: &mut &str) -> String {
     let mut int = String::new();
     for (i, ch) in read.char_indices() {