add clean_rust command

Author: fafhrd91

CHANGES.rst

@@ -1,11 +1,13 @@
 CHANGES
 =======
 
-0.4 (2017-03-xx)
+0.4 (2017-03-10)
 ----------------
 
 - Fixed bdist_egg and bdist_wheel support
 
+- setuptool's clean command cleans rust project as well
+
 - Use absolute path to cargo manifest
 
 - Enable debug builds for inplace builds, otherwise build release

setup.py

@@ -33,6 +33,7 @@
     ],
     entry_points="""
     [distutils.commands]
+    clean_rust=setuptools_rust:clean_rust
     build_ext=setuptools_rust:build_ext
     build_rust=setuptools_rust:build_rust
     """

setuptools_rust/__init__.py

@@ -14,7 +14,7 @@
 from . import patch  # noqa
 from .build_ext import build_ext
 
-__all__ = ('RustExtension', 'build_ext', 'build_rust')
+__all__ = ('RustExtension', 'clean_rust', 'build_ext', 'build_rust')
 
 patch.monkey_patch_dist(build_ext)
 
@@ -232,3 +232,34 @@ def run(self):
                         version, ext.rust_version))
 
             self.build_extension(ext)
+
+
+class clean_rust(Command):
+    """ Clean rust extensions. """
+
+    description = "clean rust extensions (compile/link to build directory)"
+
+    def initialize_options(self):
+        self.extensions = ()
+        self.inplace = False
+
+    def finalize_options(self):
+        self.extensions = [ext for ext in self.distribution.rust_extensions
+                           if isinstance(ext, RustExtension)]
+
+    def run(self):
+        if not self.extensions:
+            return
+
+        for ext in self.extensions:
+            # build cargo command
+            args = (["cargo", "clean", "--manifest-path", ext.path])
+
+            if not ext.quiet:
+                print(" ".join(args), file=sys.stderr)
+
+            # Execute cargo command
+            try:
+                subprocess.check_output(args)
+            except:
+                pass

setuptools_rust/patch.py

@@ -1,3 +1,4 @@
+from distutils.command.clean import clean
 from distutils.command.install import install
 from distutils.dist import Distribution as DistDistribution
 from setuptools.dist import Distribution
@@ -50,3 +51,13 @@ def finalize_options(self):
         self.distribution.ext_modules = ext_modules
 
     install.finalize_options = finalize_options
+
+    # clean rust project
+    def run_clean(self):
+        self.orig_run()
+
+        if not self.dry_run:
+            self.run_command("clean_rust")
+
+    clean.orig_run = clean.run
+    clean.run = run_clean