Add eta reduction (fixes #29)

This commit is contained in:
Manish Goregaokar 2015-05-10 10:39:04 +05:30
parent 709dfe1cea
commit 2447e1d5be
6 changed files with 88 additions and 1 deletions

3
.gitignore vendored
View file

@ -9,3 +9,6 @@
# Generated by Cargo
/target/
# We don't pin yet
Cargo.lock

View file

@ -1,6 +1,6 @@
[package]
name = "clippy"
version = "0.0.3"
version = "0.0.4"
authors = [
"Manish Goregaokar <manishsmail@gmail.com>",
"Andre Bogus <bogusandre@gmail.com>"

View file

@ -19,6 +19,7 @@ Lints included in this crate:
- `cmp_nan`: Denies comparisons to NAN (which will always return false, which is probably not intended)
- `float_cmp`: Warns on `==` or `!=` comparisons of floaty typed values. As floating-point operations usually involve rounding errors, it is always better to check for approximate equality within some small bounds
- `precedence`: Warns on expressions where precedence may trip up the unwary reader of the source and suggests adding parenthesis, e.g. `x << 2 + y` will be parsed as `x << (2 + y)`
- `redundant_closure`: Warns on usage of eta-reducible closures like `|a| foo(a)` (which can be written as just `foo`)
To use, add the following lines to your Cargo.toml:

59
src/eta_reduction.rs Normal file
View file

@ -0,0 +1,59 @@
use syntax::ast::*;
use rustc::lint::{Context, LintPass, LintArray, Lint, Level};
use syntax::codemap::{Span, Spanned};
use syntax::print::pprust::expr_to_string;
#[allow(missing_copy_implementations)]
pub struct EtaPass;
declare_lint!(pub REDUNDANT_CLOSURE, Warn,
"Warn on usage of redundant closures, i.e. `|a| foo(a)`");
impl LintPass for EtaPass {
fn get_lints(&self) -> LintArray {
lint_array!(REDUNDANT_CLOSURE)
}
fn check_expr(&mut self, cx: &Context, expr: &Expr) {
if let ExprClosure(_, ref decl, ref blk) = expr.node {
if blk.stmts.len() != 0 {
// || {foo(); bar()}; can't be reduced here
return;
}
if let Some(ref ex) = blk.expr {
if let ExprCall(ref caller, ref args) = ex.node {
if args.len() != decl.inputs.len() {
// Not the same number of arguments, there
// is no way the closure is the same as the function
return;
}
for (ref a1, ref a2) in decl.inputs.iter().zip(args) {
if let PatIdent(_, ident, _) = a1.pat.node {
// XXXManishearth Should I be checking the binding mode here?
if let ExprPath(None, ref p) = a2.node {
if p.segments.len() != 1 {
// If it's a proper path, it can't be a local variable
return;
}
if p.segments[0].identifier != ident.node {
// The two idents should be the same
return
}
} else {
return
}
} else {
return
}
}
cx.span_lint(REDUNDANT_CLOSURE, expr.span,
&format!("Redundant closure found, consider using `{}` in its place",
expr_to_string(caller))[..])
}
}
}
}
}

View file

@ -21,6 +21,7 @@ pub mod bit_mask;
pub mod ptr_arg;
pub mod needless_bool;
pub mod approx_const;
pub mod eta_reduction;
#[plugin_registrar]
pub fn plugin_registrar(reg: &mut Registry) {
@ -36,6 +37,7 @@ pub fn plugin_registrar(reg: &mut Registry) {
reg.register_lint_pass(box approx_const::ApproxConstant as LintPassObject);
reg.register_lint_pass(box misc::FloatCmp as LintPassObject);
reg.register_lint_pass(box misc::Precedence as LintPassObject);
reg.register_lint_pass(box eta_reduction::EtaPass as LintPassObject);
reg.register_lint_group("clippy", vec![types::BOX_VEC, types::LINKEDLIST,
misc::SINGLE_MATCH, misc::STR_TO_STRING,
@ -45,5 +47,6 @@ pub fn plugin_registrar(reg: &mut Registry) {
approx_const::APPROX_CONSTANT,
misc::CMP_NAN, misc::FLOAT_CMP,
misc::PRECEDENCE,
eta_reduction::REDUNDANT_CLOSURE,
]);
}

21
tests/compile-fail/eta.rs Normal file
View file

@ -0,0 +1,21 @@
#![feature(plugin)]
#![plugin(clippy)]
#![allow(unknown_lints, unused)]
#![deny(redundant_closure)]
fn main() {
let a = |a, b| foo(a, b);
//~^ ERROR Redundant closure found, consider using `foo` in its place
let c = |a, b| {1+2; foo}(a, b);
//~^ ERROR Redundant closure found, consider using `{ 1 + 2; foo }` in its place
let d = |a, b| foo((|c, d| foo2(c,d))(a,b), b);
//~^ ERROR Redundant closure found, consider using `foo2` in its place
}
fn foo(_: u8, _: u8) {
}
fn foo2(_: u8, _: u8) -> u8 {
1u8
}