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