Parse a pattern with no arm

Nadrieril · Nadrieril · commit a445ba8a9d10 · 2023-12-03T12:25:46.000+01:00
diff --git a/src/matches.rs b/src/matches.rs
@@ -223,7 +223,7 @@ fn rewrite_match_arm(
 ) -> Option<String> {
     let (missing_span, attrs_str) = if !arm.attrs.is_empty() {
         if contains_skip(&arm.attrs) {
-            let (_, body) = flatten_arm_body(context, &arm.body, None);
+            let (_, body) = flatten_arm_body(context, arm.body.as_deref()?, None);
             // `arm.span()` does not include trailing comma, add it manually.
             return Some(format!(
                 "{}{}",
@@ -246,7 +246,7 @@ fn rewrite_match_arm(
     };
 
     // Patterns
-    let pat_shape = match &arm.body.kind {
+    let pat_shape = match &arm.body.as_ref()?.kind {
         ast::ExprKind::Block(_, Some(label)) => {
             // Some block with a label ` => 'label: {`
             // 7 = ` => : {`
@@ -280,10 +280,10 @@ fn rewrite_match_arm(
         false,
     )?;
 
-    let arrow_span = mk_sp(arm.pat.span.hi(), arm.body.span().lo());
+    let arrow_span = mk_sp(arm.pat.span.hi(), arm.body.as_ref()?.span().lo());
     rewrite_match_body(
         context,
-        &arm.body,
+        arm.body.as_ref()?,
         &lhs_str,
         shape,
         guard_str.contains('\n'),
diff --git a/src/spanned.rs b/src/spanned.rs
@@ -97,7 +97,12 @@ impl Spanned for ast::Arm {
         } else {
             self.attrs[0].span.lo()
         };
-        span_with_attrs_lo_hi!(self, lo, self.body.span.hi())
+        let hi = if let Some(body) = &self.body {
+            body.span.hi()
+        } else {
+            self.pat.span.hi()
+        };
+        span_with_attrs_lo_hi!(self, lo, hi)
     }
 }
 

Original file line number	Diff line number	Diff line change
`@@ -97,7 +97,12 @@ impl Spanned for ast::Arm {`
`97`	`97`	`} else {`
`98`	`98`	`self.attrs[0].span.lo()`
`99`	`99`	`};`
`100`		`- span_with_attrs_lo_hi!(self, lo, self.body.span.hi())`
	`100`	`+ let hi = if let Some(body) = &self.body {`
	`101`	`+ body.span.hi()`
	`102`	`+ } else {`
	`103`	`+ self.pat.span.hi()`
	`104`	`+ };`
	`105`	`+ span_with_attrs_lo_hi!(self, lo, hi)`
`101`	`106`	`}`
`102`	`107`	`}`
`103`	`108`