fn flatten_arm_body<'a>( context: &'a RewriteContext<'_>, body: &'a Expr, opt_shape: Option<Shape>, ) -> (bool, &'a Expr)