1
1
use clippy_utils:: diagnostics:: { span_lint_and_sugg, span_lint_and_then} ;
2
2
use clippy_utils:: higher:: VecArgs ;
3
3
use clippy_utils:: source:: snippet_opt;
4
- use clippy_utils:: ty:: is_type_diagnostic_item ;
4
+ use clippy_utils:: ty:: type_diagnostic_name ;
5
5
use clippy_utils:: usage:: local_used_after_expr;
6
6
use clippy_utils:: { get_enclosing_loop_or_closure, higher, path_to_local, path_to_local_id} ;
7
- use if_chain:: if_chain;
8
7
use rustc_errors:: Applicability ;
9
8
use rustc_hir:: def_id:: DefId ;
10
- use rustc_hir:: { Expr , ExprKind , Param , PatKind , Unsafety } ;
9
+ use rustc_hir:: { BindingAnnotation , Expr , ExprKind , FnRetTy , Param , PatKind , TyKind , Unsafety } ;
11
10
use rustc_lint:: { LateContext , LateLintPass } ;
12
- use rustc_middle:: ty:: adjustment:: { Adjust , Adjustment , AutoBorrow } ;
13
- use rustc_middle:: ty:: binding:: BindingMode ;
14
- use rustc_middle:: ty:: subst:: Subst ;
15
- use rustc_middle:: ty:: { self , ClosureKind , Ty , TypeFoldable } ;
11
+ use rustc_middle:: ty:: subst:: { GenericArg , GenericArgKind } ;
12
+ use rustc_middle:: ty:: { self , ClosureKind , ClosureSubsts , FnSig , Region , RegionKind , Ty , TypeFoldable , TypeckResults } ;
16
13
use rustc_session:: { declare_lint_pass, declare_tool_lint} ;
17
14
use rustc_span:: symbol:: sym;
15
+ use rustc_target:: spec:: abi:: Abi ;
18
16
19
17
declare_clippy_lint ! {
20
18
/// ### What it does
@@ -78,7 +76,12 @@ impl<'tcx> LateLintPass<'tcx> for EtaReduction {
78
76
return ;
79
77
}
80
78
let body = match expr. kind {
81
- ExprKind :: Closure ( _, _, id, _, _) => cx. tcx . hir ( ) . body ( id) ,
79
+ ExprKind :: Closure ( _, decl, id, _, _)
80
+ if decl. inputs . iter ( ) . all ( |ty| matches ! ( ty. kind, TyKind :: Infer ) )
81
+ && matches ! ( decl. output, FnRetTy :: DefaultReturn ( _) ) =>
82
+ {
83
+ cx. tcx . hir ( ) . body ( id)
84
+ } ,
82
85
_ => return ,
83
86
} ;
84
87
if body. value . span . from_expansion ( ) {
@@ -100,122 +103,187 @@ impl<'tcx> LateLintPass<'tcx> for EtaReduction {
100
103
return ;
101
104
}
102
105
103
- let closure_ty = cx. typeck_results ( ) . expr_ty ( expr) ;
104
-
105
- if_chain ! (
106
- if let ExprKind :: Call ( callee, args) = body. value. kind;
107
- if let ExprKind :: Path ( _) = callee. kind;
108
- if check_inputs( cx, body. params, args) ;
109
- let callee_ty = cx. typeck_results( ) . expr_ty_adjusted( callee) ;
110
- let call_ty = cx. typeck_results( ) . type_dependent_def_id( body. value. hir_id)
111
- . map_or( callee_ty, |id| cx. tcx. type_of( id) ) ;
112
- if check_sig( cx, closure_ty, call_ty) ;
113
- let substs = cx. typeck_results( ) . node_substs( callee. hir_id) ;
114
- // This fixes some false positives that I don't entirely understand
115
- if substs. is_empty( ) || !cx. typeck_results( ) . expr_ty( expr) . has_late_bound_regions( ) ;
116
- // A type param function ref like `T::f` is not 'static, however
117
- // it is if cast like `T::f as fn()`. This seems like a rustc bug.
118
- if !substs. types( ) . any( |t| matches!( t. kind( ) , ty:: Param ( _) ) ) ;
119
- let callee_ty_unadjusted = cx. typeck_results( ) . expr_ty( callee) . peel_refs( ) ;
120
- if !is_type_diagnostic_item( cx, callee_ty_unadjusted, sym:: Arc ) ;
121
- if !is_type_diagnostic_item( cx, callee_ty_unadjusted, sym:: Rc ) ;
122
- then {
123
- span_lint_and_then( cx, REDUNDANT_CLOSURE , expr. span, "redundant closure" , |diag| {
124
- if let Some ( mut snippet) = snippet_opt( cx, callee. span) {
125
- if_chain! {
126
- if let ty:: Closure ( _, substs) = callee_ty. peel_refs( ) . kind( ) ;
127
- if substs. as_closure( ) . kind( ) == ClosureKind :: FnMut ;
128
- if get_enclosing_loop_or_closure( cx. tcx, expr) . is_some( )
129
- || path_to_local( callee) . map_or( false , |l| local_used_after_expr( cx, l, callee) ) ;
106
+ let typeck = cx. typeck_results ( ) ;
107
+ let closure = if let ty:: Closure ( _, closure_subs) = typeck. expr_ty ( expr) . kind ( ) {
108
+ closure_subs. as_closure ( )
109
+ } else {
110
+ return ;
111
+ } ;
130
112
131
- then {
132
- // Mutable closure is used after current expr; we cannot consume it.
133
- snippet = format!( "&mut {}" , snippet) ;
134
- }
113
+ match body. value . kind {
114
+ ExprKind :: Call ( callee, args)
115
+ if matches ! ( callee. kind, ExprKind :: Path ( ..) )
116
+ && !matches ! (
117
+ type_diagnostic_name( cx, typeck. expr_ty( callee) . peel_refs( ) ) ,
118
+ Some ( sym:: Arc | sym:: Rc )
119
+ )
120
+ && check_inputs ( typeck, body. params , args) =>
121
+ {
122
+ let sig = match * typeck. expr_ty_adjusted ( callee) . kind ( ) {
123
+ ty:: FnDef ( def, _) => cx. tcx . fn_sig ( def) . skip_binder ( ) ,
124
+ ty:: FnPtr ( sig) => sig. skip_binder ( ) ,
125
+ ty:: Closure ( _, subs) => cx
126
+ . tcx
127
+ . signature_unclosure ( subs. as_closure ( ) . sig ( ) , Unsafety :: Normal )
128
+ . skip_binder ( ) ,
129
+ _ => {
130
+ if typeck. type_dependent_def_id ( body. value . hir_id ) . is_some ( )
131
+ && let subs = typeck. node_substs ( body. value . hir_id )
132
+ && let output = typeck. expr_ty ( & body. value )
133
+ && let ty:: Tuple ( tys) = * subs. type_at ( 1 ) . kind ( )
134
+ {
135
+ cx. tcx . mk_fn_sig ( tys. into_iter ( ) , output, false , Unsafety :: Normal , Abi :: Rust )
136
+ } else {
137
+ return ;
135
138
}
136
- diag. span_suggestion(
137
- expr. span,
138
- "replace the closure with the function itself" ,
139
- snippet,
140
- Applicability :: MachineApplicable ,
141
- ) ;
142
- }
143
- } ) ;
144
- }
145
- ) ;
146
-
147
- if_chain ! (
148
- if let ExprKind :: MethodCall ( path, args, _) = body. value. kind;
149
- if check_inputs( cx, body. params, args) ;
150
- let method_def_id = cx. typeck_results( ) . type_dependent_def_id( body. value. hir_id) . unwrap( ) ;
151
- let substs = cx. typeck_results( ) . node_substs( body. value. hir_id) ;
152
- let call_ty = cx. tcx. type_of( method_def_id) . subst( cx. tcx, substs) ;
153
- if check_sig( cx, closure_ty, call_ty) ;
154
- then {
155
- span_lint_and_then( cx, REDUNDANT_CLOSURE_FOR_METHOD_CALLS , expr. span, "redundant closure" , |diag| {
156
- let name = get_ufcs_type_name( cx, method_def_id) ;
157
- diag. span_suggestion(
139
+ } ,
140
+ } ;
141
+ if check_sig ( cx, closure, sig)
142
+ // Given some trait fn `fn f() -> ()` and some type `T: Trait`, `T::f` is not
143
+ // `'static` unless `T: 'static`. The cast `T::f as fn()` will, however, result
144
+ // in a type which is `'static`.
145
+ // For now ignore all callee types which reference a type parameter.
146
+ && !typeck. node_substs ( callee. hir_id ) . types ( ) . any ( |t| matches ! ( t. kind( ) , ty:: Param ( _) ) )
147
+ {
148
+ span_lint_and_then (
149
+ cx,
150
+ REDUNDANT_CLOSURE ,
158
151
expr. span ,
159
- "replace the closure with the method itself" ,
160
- format!( "{}::{}" , name, path. ident. name) ,
161
- Applicability :: MachineApplicable ,
152
+ "redundant closure" ,
153
+ |diag| {
154
+ if let Some ( mut snippet) = snippet_opt ( cx, callee. span ) {
155
+ if let ty:: Closure ( _, substs) = * typeck. expr_ty ( callee) . peel_refs ( ) . kind ( )
156
+ && substs. as_closure ( ) . kind ( ) == ClosureKind :: FnMut
157
+ && (
158
+ get_enclosing_loop_or_closure ( cx. tcx , expr) . is_some ( )
159
+ || path_to_local ( callee)
160
+ . map_or ( false , |l| local_used_after_expr ( cx, l, callee) )
161
+ )
162
+ {
163
+ // Mutable closure is used after current expr; we cannot consume it.
164
+ snippet = format ! ( "&mut {}" , snippet) ;
165
+ }
166
+ diag. span_suggestion (
167
+ expr. span ,
168
+ "replace the closure with the function itself" ,
169
+ snippet,
170
+ Applicability :: MachineApplicable ,
171
+ ) ;
172
+ }
173
+ }
162
174
) ;
163
- } )
164
- }
165
- ) ;
175
+ }
176
+ } ,
177
+ ExprKind :: MethodCall ( path, args, _) if check_inputs ( typeck, body. params , args) => {
178
+ if let Some ( method_def_id) = typeck. type_dependent_def_id ( body. value . hir_id )
179
+ && check_sig ( cx, closure, cx. tcx . fn_sig ( method_def_id) . skip_binder ( ) )
180
+ {
181
+ span_lint_and_then (
182
+ cx,
183
+ REDUNDANT_CLOSURE_FOR_METHOD_CALLS ,
184
+ expr. span ,
185
+ "redundant closure" ,
186
+ |diag| {
187
+ let name = get_ufcs_type_name ( cx, method_def_id) ;
188
+ diag. span_suggestion (
189
+ expr. span ,
190
+ "replace the closure with the method itself" ,
191
+ format ! ( "{}::{}" , name, path. ident. name) ,
192
+ Applicability :: MachineApplicable ,
193
+ ) ;
194
+ } ,
195
+ )
196
+ }
197
+ } ,
198
+ _ => return ,
199
+ }
166
200
}
167
201
}
168
202
169
- fn check_inputs ( cx : & LateContext < ' _ > , params : & [ Param < ' _ > ] , call_args : & [ Expr < ' _ > ] ) -> bool {
170
- if params. len ( ) != call_args. len ( ) {
171
- return false ;
203
+ fn check_inputs ( typeck : & TypeckResults < ' _ > , params : & [ Param < ' _ > ] , call_args : & [ Expr < ' _ > ] ) -> bool {
204
+ params. len ( ) == call_args. len ( )
205
+ && params. iter ( ) . zip ( call_args) . all ( |( p, arg) | {
206
+ matches ! (
207
+ p. pat. kind, PatKind :: Binding ( BindingAnnotation :: Unannotated , id, _, None )
208
+ if path_to_local_id( arg, id)
209
+ )
210
+ // Only allow adjustments which change regions (i.e. re-borrowing).
211
+ && typeck
212
+ . expr_adjustments ( arg)
213
+ . last ( )
214
+ . map_or ( true , |a| a. target == typeck. expr_ty ( arg) )
215
+ } )
216
+ }
217
+
218
+ fn check_sig < ' tcx > ( cx : & LateContext < ' tcx > , closure : ClosureSubsts < ' tcx > , call_sig : FnSig < ' _ > ) -> bool {
219
+ call_sig. unsafety == Unsafety :: Normal
220
+ && !has_late_bound_to_non_late_bound_regions (
221
+ cx. tcx
222
+ . signature_unclosure ( closure. sig ( ) , Unsafety :: Normal )
223
+ . skip_binder ( ) ,
224
+ call_sig,
225
+ )
226
+ }
227
+
228
+ /// This walks through both signatures and checks for any time a late-bound region is expected by an
229
+ /// `impl Fn` type, but the target signature does not have a late-bound region in the same position.
230
+ ///
231
+ /// This is needed because rustc is unable to late bind early-bound regions in a function signature.
232
+ fn has_late_bound_to_non_late_bound_regions ( from_sig : FnSig < ' _ > , to_sig : FnSig < ' _ > ) -> bool {
233
+ fn check_region ( from_region : Region < ' _ > , to_region : Region < ' _ > ) -> bool {
234
+ matches ! ( from_region. kind( ) , RegionKind :: ReLateBound ( ..) )
235
+ && !matches ! ( to_region. kind( ) , RegionKind :: ReLateBound ( ..) )
172
236
}
173
- let binding_modes = cx. typeck_results ( ) . pat_binding_modes ( ) ;
174
- std:: iter:: zip ( params, call_args) . all ( |( param, arg) | {
175
- match param. pat . kind {
176
- PatKind :: Binding ( _, id, ..) if path_to_local_id ( arg, id) => { } ,
177
- _ => return false ,
178
- }
179
- // checks that parameters are not bound as `ref` or `ref mut`
180
- if let Some ( BindingMode :: BindByReference ( _) ) = binding_modes. get ( param. pat . hir_id ) {
181
- return false ;
182
- }
183
237
184
- match * cx. typeck_results ( ) . expr_adjustments ( arg) {
185
- [ ] => true ,
186
- [
187
- Adjustment {
188
- kind : Adjust :: Deref ( None ) ,
189
- ..
238
+ fn check_subs ( from_subs : & [ GenericArg < ' _ > ] , to_subs : & [ GenericArg < ' _ > ] ) -> bool {
239
+ if from_subs. len ( ) != to_subs. len ( ) {
240
+ return true ;
241
+ }
242
+ for ( from_arg, to_arg) in to_subs. iter ( ) . zip ( from_subs) {
243
+ match ( from_arg. unpack ( ) , to_arg. unpack ( ) ) {
244
+ ( GenericArgKind :: Lifetime ( from_region) , GenericArgKind :: Lifetime ( to_region) ) => {
245
+ if check_region ( from_region, to_region) {
246
+ return true ;
247
+ }
190
248
} ,
191
- Adjustment {
192
- kind : Adjust :: Borrow ( AutoBorrow :: Ref ( _, mu2) ) ,
193
- ..
249
+ ( GenericArgKind :: Type ( from_ty) , GenericArgKind :: Type ( to_ty) ) => {
250
+ if check_ty ( from_ty, to_ty) {
251
+ return true ;
252
+ }
194
253
} ,
195
- ] => {
196
- // re-borrow with the same mutability is allowed
197
- let ty = cx. typeck_results ( ) . expr_ty ( arg) ;
198
- matches ! ( * ty. kind( ) , ty:: Ref ( .., mu1) if mu1 == mu2. into( ) )
199
- } ,
200
- _ => false ,
254
+ ( GenericArgKind :: Const ( _) , GenericArgKind :: Const ( _) ) => ( ) ,
255
+ _ => return true ,
256
+ }
201
257
}
202
- } )
203
- }
204
-
205
- fn check_sig < ' tcx > ( cx : & LateContext < ' tcx > , closure_ty : Ty < ' tcx > , call_ty : Ty < ' tcx > ) -> bool {
206
- let call_sig = call_ty. fn_sig ( cx. tcx ) ;
207
- if call_sig. unsafety ( ) == Unsafety :: Unsafe {
208
- return false ;
258
+ false
209
259
}
210
- if !closure_ty. has_late_bound_regions ( ) {
211
- return true ;
260
+
261
+ fn check_ty ( from_ty : Ty < ' _ > , to_ty : Ty < ' _ > ) -> bool {
262
+ match ( from_ty. kind ( ) , to_ty. kind ( ) ) {
263
+ ( & ty:: Adt ( _, from_subs) , & ty:: Adt ( _, to_subs) ) => check_subs ( from_subs, to_subs) ,
264
+ ( & ty:: Array ( from_ty, _) , & ty:: Array ( to_ty, _) ) | ( & ty:: Slice ( from_ty) , & ty:: Slice ( to_ty) ) => {
265
+ check_ty ( from_ty, to_ty)
266
+ } ,
267
+ ( & ty:: Ref ( from_region, from_ty, _) , & ty:: Ref ( to_region, to_ty, _) ) => {
268
+ check_region ( from_region, to_region) || check_ty ( from_ty, to_ty)
269
+ } ,
270
+ ( & ty:: Tuple ( from_tys) , & ty:: Tuple ( to_tys) ) => {
271
+ from_tys. len ( ) != to_tys. len ( )
272
+ || from_tys
273
+ . iter ( )
274
+ . zip ( to_tys)
275
+ . any ( |( from_ty, to_ty) | check_ty ( from_ty, to_ty) )
276
+ } ,
277
+ _ => from_ty. has_late_bound_regions ( ) ,
278
+ }
212
279
}
213
- let substs = match closure_ty. kind ( ) {
214
- ty:: Closure ( _, substs) => substs,
215
- _ => return false ,
216
- } ;
217
- let closure_sig = cx. tcx . signature_unclosure ( substs. as_closure ( ) . sig ( ) , Unsafety :: Normal ) ;
218
- cx. tcx . erase_late_bound_regions ( closure_sig) == cx. tcx . erase_late_bound_regions ( call_sig)
280
+
281
+ assert ! ( from_sig. inputs_and_output. len( ) == to_sig. inputs_and_output. len( ) ) ;
282
+ from_sig
283
+ . inputs_and_output
284
+ . iter ( )
285
+ . zip ( to_sig. inputs_and_output )
286
+ . any ( |( from_ty, to_ty) | check_ty ( from_ty, to_ty) )
219
287
}
220
288
221
289
fn get_ufcs_type_name ( cx : & LateContext < ' _ > , method_def_id : DefId ) -> String {
0 commit comments