Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Add checking for unnecessary delims in closure body #136906

Open
wants to merge 6 commits into
base: master
Choose a base branch
from

Conversation

chenyukang
Copy link
Member

Fixes #136741

@rustbot
Copy link
Collaborator

rustbot commented Feb 12, 2025

r? @jieyouxu

rustbot has assigned @jieyouxu.
They will have a look at your PR within the next two weeks and either review your PR or reassign to another reviewer.

Use r? to explicitly pick a reviewer

@rustbot rustbot added S-waiting-on-review Status: Awaiting review from the assignee but also interested parties. T-compiler Relevant to the compiler team, which will review and decide on the PR/issue. labels Feb 12, 2025
@chenyukang chenyukang force-pushed the yukang-fix-136741-closure-body branch 3 times, most recently from 3b739b5 to ccdf053 Compare February 12, 2025 05:27
if matches!(closure.fn_decl.output, FnRetTy::Default(_))
// skip `#[core::contracts::requires(...)]` and `#[core::contracts::ensures(...)]` which generate closure
&& !cx
.sess()
Copy link
Member Author

@chenyukang chenyukang Feb 12, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

any better check for this?
seems the AST generated by contracts are the same with normal closures.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

we could generate an #[allow] for this lint as part of the ast we generate. Alternatively you could check that it's expanded code in general and just ignore all of this. Likely some macros will hit this issue, too (add a test for that, too)

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Alternatively you could check that it's expanded code in general and just ignore all of this.

The following check here https://github.com/rust-lang/rust/pull/136906/files#diff-e25b2c6c1fa434ab8078db81003888370f06b9da5503d1590757350be8e31103L945-L946 already checked it?

I also tried to use closure.body.span.can_be_used_for_suggestions(), seems can not exclude this scenario.

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

for the solution we could generate an #[allow] for this lint as part of the ast we generate.

I tried to add it here:
428b251

which will break the parser.

seems we only add simple tokentree for the contract and parse it here:
https://github.com/chenyukang/rust/blob/428b251feb3e87e41c9e46656cc96a3b94d88ea5/compiler/rustc_parse/src/parser/generics.rs#L302-L325

we do not expect too much change just to generate an #[allow(unused_parens)], where is the proper point to inject it.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

already checked it?

ah that means contract expansion doesn't add the expansion info to the spans. I did indeed forget to check for that at all when it was introduced

cc @celinval

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I tried to mark the span with DesugaringKind::Contract at here: https://github.com/chenyukang/rust/blob/9ec13a62cafd1448b2d43d887034ae6501fca6d0/compiler/rustc_ast_lowering/src/item.rs#L1082-L1089

But found that early lint check is run before this point.

I think this solution with span check maybe enough: 64f1f52

the generated closure span are all the same, which is different from real code.

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Ah right this is due to the macro expansion from

fn expand_contract_clause(
maybe the span can be annotated there?

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I tried to follow mark_with_reason https://github.com/chenyukang/rust/blob/59281e98d0af7ed5071d141235ce5ddff1f60253/compiler/rustc_span/src/hygiene.rs#L906-L919

to create a span with annotation:
59281e9#diff-0843065cebe8379cca975e505c67c393f9933c8d9019fb6a6e3b3d5654157bc1R70

The problem is how can we get a ctx which meet the trait here, seems we're at a very early stage:

error[E0277]: the trait bound `SyntaxContext: rustc_span::HashStableContext` is not satisfied
   --> compiler/rustc_builtin_macros/src/contracts.rs:70:49
    |
70  |     let expn_id = LocalExpnId::fresh(expn_data, attr_span.ctxt());
    |                   ------------------            ^^^^^^^^^^^^^^^^ the trait `rustc_span::HashStableContext` is not implemented for `SyntaxContext`
    |                   |
    |                   required by a bound introduced by this call
    |
note: required by a bound in `LocalExpnId::fresh`
   --> /Users/yukang/rust/compiler/rustc_span/src/hygiene.rs:200:53
    |
200 |     pub fn fresh(mut expn_data: ExpnData, ctx: impl HashStableContext) -> LocalExpnId {
    |                                                     ^^^^^^^^^^^^^^^^^ required by this bound in `LocalExpnId::fresh`

Copy link
Member Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I think the span checking here is enough for avoid lint for contract syntax here?
https://github.com/rust-lang/rust/pull/136906/files#diff-e25b2c6c1fa434ab8078db81003888370f06b9da5503d1590757350be8e31103R917-R918

it's a bit hacky, but user written code will always true on : e.span != closure.body.span.

And also considering the fact that contract syntax is temporary, if we can not find a perfect way to mark the span we should not introduce more change for it. #137134 (comment)

Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

You're absolutely right. Sorry for holding it up

@rust-log-analyzer

This comment has been minimized.

@rust-log-analyzer

This comment has been minimized.

@chenyukang chenyukang force-pushed the yukang-fix-136741-closure-body branch from c755f78 to c7ec66c Compare February 12, 2025 07:06
@rustbot
Copy link
Collaborator

rustbot commented Feb 12, 2025

Some changes occurred to MIR optimizations

cc @rust-lang/wg-mir-opt

Some changes occurred in coverage instrumentation.

cc @Zalathar

@rust-log-analyzer

This comment has been minimized.

@rustbot
Copy link
Collaborator

rustbot commented Feb 12, 2025

The Miri subtree was changed

cc @rust-lang/miri

Some changes occurred in src/tools/clippy

cc @rust-lang/clippy

Copy link
Contributor

@oli-obk oli-obk left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

that's a lot of changes in unrelated tests. Changing tests for such reasons isn't too great and the value of this change in tests isn't really there either. Maybe add the lint to the list of lints that compiletest allows for all tests?

@rust-log-analyzer

This comment has been minimized.

@jieyouxu
Copy link
Member

jieyouxu commented Feb 12, 2025

That's a lot of changes in unrelated tests. Changing tests for such reasons isn't too great and the value of this change in tests isn't really there either.

IMO we may want to just not warn on this case. Looking at the diffs both in the compiler and in the tests, it doesn't seem like a strict readability improvement to me (at times I find it even a bit harder to read). unused_parens is also a warn-by-default lint, so we'd be warning on a lot of existing code (judging by quite a few instances of compiler code getting newly linted on when we expand this lint).

Maybe add the lint to the list of lints that compiletest allows for all tests?

I think we should avoid adding blanket allows for all tests where feasible (many tests getting affected is itself a decent indicator, albeit rough, for if a change may impact a lot of cases).

@oli-obk
Copy link
Contributor

oli-obk commented Feb 12, 2025

Looking at the diffs both in the compiler and in the tests, it doesn't seem like a strict readability improvement to me (at times I find it even a bit harder to read)

in the compiler and tools it was always an improvement to me. What specific examples do you dislike?

@jieyouxu
Copy link
Member

jieyouxu commented Feb 12, 2025

in the compiler and tools it was always an improvement to me. What specific examples do you dislike?

This probably comes down to personal preference, but I don't really find these specific examples to be strictly better (I find the { } helps with visual separation of the body expr. I don't feel too strongly about this so 🤷

-.all(|(node, span_edge)| { span_edge.is_some() <= self.is_supernode(node) }),
+.all(|(node, span_edge)| span_edge.is_some() <= self.is_supernode(node)),

-write!(&mut counter, "{:#}", fmt::from_fn(|f| { self.inner_full_print(None, f, cx) }))
+write!(&mut counter, "{:#}", fmt::from_fn(|f| self.inner_full_print(None, f, cx)))

Removing parens on the very short exprs do look more readable to me. So I guess I feel mixed about it? lol

@jieyouxu
Copy link
Member

Since you're already looking at it... r? @oli-obk

@rustbot rustbot assigned oli-obk and unassigned jieyouxu Feb 12, 2025
@chenyukang chenyukang force-pushed the yukang-fix-136741-closure-body branch from 5e56316 to 9ec13a6 Compare February 13, 2025 01:14
@rustbot rustbot added A-compiletest Area: The compiletest test runner A-testsuite Area: The testsuite used to check the correctness of rustc T-bootstrap Relevant to the bootstrap subteam: Rust's build system (x.py and src/bootstrap) labels Feb 13, 2025
@rustbot
Copy link
Collaborator

rustbot commented Feb 13, 2025

Some changes occurred in src/tools/compiletest

cc @jieyouxu

@rust-log-analyzer

This comment has been minimized.

@chenyukang
Copy link
Member Author

that's a lot of changes in unrelated tests. Changing tests for such reasons isn't too great and the value of this change in tests isn't really there either. Maybe add the lint to the list of lints that compiletest allows for all tests?

done with commit 3a34ad7

@chenyukang
Copy link
Member Author

Yes, this seems comes down to personal preference, but I just found rustfmt will remove the braces in this closure:

 let some_predicate = |x: &mut i32| { *x == 2 || *x == 3 || *x == 6 };

@traviscross traviscross removed the T-rustdoc-frontend Relevant to the rustdoc-frontend team, which will review and decide on the web UI/UX output. label Mar 7, 2025
@traviscross
Copy link
Contributor

traviscross commented Mar 7, 2025

While this seemed OK to me, by way of double checking, I was still trying to convince myself this wasn't actually too opinionated somehow. Maybe it's OK if people want to disambiguate here? But then I recalled that these two lines

_ = || -> _ { 0 }.clone();
_ = || { 0 }.clone();

are quite different. That is, starting a closure body with an opening brace doesn't really serve to disambiguate the scope of that body in the way that one might expect.

So since the braces aren't really good for that purpose, it makes sense to treat them as we do other unnecessary braces.
That's enough to convince me this is OK.

This is going to be a meaningful lint expansion (so heads-up to @tmandry), but the fix is automated, so that seems fine too (and I'm glad to not have to come up with another lint name). Therefore, I propose:

@rfcbot fcp merge

@rfcbot
Copy link

rfcbot commented Mar 7, 2025

Team member @traviscross has proposed to merge this. The next step is review by the rest of the tagged team members:

Concerns:

Once a majority of reviewers approve (and at most 2 approvals are outstanding), this will enter its final comment period. If you spot a major issue that hasn't been raised at any point in this process, please speak up!

cc @rust-lang/lang-advisors: FCP proposed for lang, please feel free to register concerns.
See this document for info about what commands tagged team members can give me.

@rfcbot rfcbot added proposed-final-comment-period Proposed to merge/close by relevant subteam, see T-<team> label. Will enter FCP once signed off. disposition-merge This issue / PR is in PFCP or FCP with a disposition to merge it. labels Mar 7, 2025
@RalfJung
Copy link
Member

RalfJung commented Mar 7, 2025

these two lines [...] are quite different

Wait, what?^^ Could you spell this out / link to some place where it is spelled out?

@traviscross
Copy link
Contributor

traviscross commented Mar 7, 2025

E.g.:

fn main() {
    let x = &mut ();
    _ = move || { let x = x; }.clone(); //~ OK
    let x = &mut ();
    _ = move || -> _ { let x = x; }.clone();
    //~^ ERROR the trait bound `&mut (): Clone` is not satisfied
}

Playground link

The first one clones unit. The second one tries to clone the closure.

That is, if the return type is not annotated, then we parse an expression. If it is, then we require an explicit block.

@RalfJung
Copy link
Member

RalfJung commented Mar 7, 2025 via email

@traviscross
Copy link
Contributor

traviscross commented Mar 7, 2025

Certainly if we try hard enough, we can produce that. E.g.:

struct W(u8);
impl Clone for W {
    fn clone(&self) -> Self {
        W(1)
    }
}

fn main() {
    let f = move |x: W| { x }.clone();
    assert!(matches!(f(W(0)), W(1)));
    let f = move |x: W| -> _ { x }.clone();
    assert!(matches!(f(W(0)), W(0)));
}

Playground link

There are probably other ways.

@chenyukang chenyukang force-pushed the yukang-fix-136741-closure-body branch from 89c1195 to 5ea47e1 Compare March 11, 2025 02:33
@rustbot rustbot added T-bootstrap Relevant to the bootstrap subteam: Rust's build system (x.py and src/bootstrap) T-rustdoc-frontend Relevant to the rustdoc-frontend team, which will review and decide on the web UI/UX output. labels Mar 11, 2025
@rust-log-analyzer

This comment has been minimized.

@chenyukang chenyukang force-pushed the yukang-fix-136741-closure-body branch from 5ea47e1 to c089b1f Compare March 11, 2025 02:52
@bors
Copy link
Contributor

bors commented Mar 13, 2025

☔ The latest upstream changes (presumably #138459) made this pull request unmergeable. Please resolve the merge conflicts.

@traviscross
Copy link
Contributor

@rfcbot concern ambiguous-block-parsing

Thinking more about this, if someone writes an apparently redundant set of braces for the body of a closure without an ascribed return type but where a second block starts immediately inside the first but ends before the first block does, e.g.,

_ = || { { 0 }.clone() };

then I don't want to suggest removing those outer braces. In fact, I'd like to lint against their absence and suggest adding them, e.g.:

_ = || { 0 }.clone(); //~ WARN

The reason this case is such a problem is that it'd be pretty reasonable for people to expect that the opening brace at the start of the closure body would force block parsing of that body. If it did, then the closing brace would end the closure body (as it does for -> _ closures). In my view, it's worth warning anywhere this reasonable expectation would be violated -- we at least shouldn't push people away from this disambiguation -- and I'm interested in whether we might be able to improve this behavior over an edition.

@chenyukang
Copy link
Member Author

for this code:

_ = || { { 0 }.clone() };

our current rustfmt will remove the outer brace automatically.

@traviscross
Copy link
Contributor

cc @rust-lang/rustfmt @rust-lang/style

@ytmimi
Copy link
Contributor

ytmimi commented Mar 19, 2025

@traviscross is there a heuristic we can reliably use to force rustfmt not to remove the surrounding blocks? If so, then we could try to modify is_block_closure_forced_inner in rustfmt, though I think we might need to gate the change unless we're considering this a bugfix for rustfmt.

fn is_block_closure_forced(context: &RewriteContext<'_>, expr: &ast::Expr) -> bool {
// If we are inside macro, we do not want to add or remove block from closure body.
if context.inside_macro() {
false
} else {
is_block_closure_forced_inner(expr, context.config.style_edition())
}
}
fn is_block_closure_forced_inner(expr: &ast::Expr, style_edition: StyleEdition) -> bool {
match expr.kind {
ast::ExprKind::If(..) | ast::ExprKind::While(..) | ast::ExprKind::ForLoop { .. } => true,
ast::ExprKind::Loop(..) if style_edition >= StyleEdition::Edition2024 => true,
ast::ExprKind::AddrOf(_, _, ref expr)
| ast::ExprKind::Try(ref expr)
| ast::ExprKind::Unary(_, ref expr)
| ast::ExprKind::Cast(ref expr, _) => is_block_closure_forced_inner(expr, style_edition),
_ => false,
}
}

@traviscross

This comment was marked as duplicate.

@traviscross
Copy link
Contributor

traviscross commented Mar 19, 2025

Probably what we'd want is something along the lines of, "if the expression forming the closure body starts with two opening braces in sequence, and then ends with a closing brace (matching the outer starting brace) without an immediately preceding inner closing brace (matching the inner starting brace), then don't remove the outer braces."

That is, we're looking for:

_ || { { .. }/* Something here */ };

If we were to add disambiguating braces, the rule would be along the lines of, "if the expression forming the closure body starts with an opening brace and does not end with a closing brace (matching the opening one),then wrap the expression in a set of outer braces.

@ytmimi
Copy link
Contributor

ytmimi commented Mar 20, 2025

I'm pretty sure that rustfmt only removes the outer braces if the closure body contains a single expression, and doesn't contain any comments. The only way I can think to write code that starts with two opening braces and ends with a single closing brace is in this case of a method call receiver being a block expression. If you can think of other ways we might end up with a pattern like || { { .. }/* Something here */ }; let me know.

For this case, I think something like this could work:

 fn is_block_closure_forced_inner(expr: &ast::Expr, style_edition: StyleEdition) -> bool {
     match expr.kind {
+        ast::ExprKind::MethodCall(ref method_call)
+            if style_edition >= StyleEdition::Edition2027 =>
+        {
+            // Prevent rustfmt from removing outer braces in `_ = || { { 0 }.clone() };`,
+            // which changes the semantics of the code.
+            matches!(method_call.receiver.kind, ast::ExprKind::Block(..))
+        }
         ast::ExprKind::If(..) | ast::ExprKind::While(..) | ast::ExprKind::ForLoop { .. } => true,
         ast::ExprKind::Loop(..) if style_edition >= StyleEdition::Edition2024 => true,
         ast::ExprKind::AddrOf(_, _, ref expr)

With that in place the following

fn main() {
    _ = || { { 0 }.clone() };
}

is formatted like this.

fn main() {
    _ = || {
        { 0 }.clone()
    };
}

@traviscross
Copy link
Contributor

Sounds right to me.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
A-compiletest Area: The compiletest test runner A-testsuite Area: The testsuite used to check the correctness of rustc disposition-merge This issue / PR is in PFCP or FCP with a disposition to merge it. I-lang-easy-decision Issue: The decision needed by the team is conjectured to be easy; this does not imply nomination I-lang-nominated Nominated for discussion during a lang team meeting. proposed-final-comment-period Proposed to merge/close by relevant subteam, see T-<team> label. Will enter FCP once signed off. S-waiting-on-author Status: This is awaiting some action (such as code changes or more information) from the author. T-bootstrap Relevant to the bootstrap subteam: Rust's build system (x.py and src/bootstrap) T-lang Relevant to the language team, which will review and decide on the PR/issue. T-rustdoc-frontend Relevant to the rustdoc-frontend team, which will review and decide on the web UI/UX output.
Projects
None yet
Development

Successfully merging this pull request may close these issues.

false-negative unused_parens in closures consisting of just one expression