Add option to cargo vendor crates into sdist

Author: messense

CHANGELOG.md

@@ -7,6 +7,7 @@
 ### Added
 - Support building x86-64 wheel on arm64 macOS machine. [#114](https://github.com/PyO3/setuptools-rust/pull/114)
 - Add macOS universal2 wheel building support. [#115](https://github.com/PyO3/setuptools-rust/pull/115)
+- Add option to cargo vendor crates into sdist. [#118](https://github.com/PyO3/setuptools-rust/pull/118)
 
 ### Changed
 - Respect `PYO3_PYTHON` and `PYTHON_SYS_EXECUTABLE` environment variables if set. [#96](https://github.com/PyO3/setuptools-rust/pull/96)

setuptools_rust/setuptools_ext.py

@@ -5,6 +5,8 @@
 
 from setuptools.command.build_ext import build_ext
 from setuptools.command.install import install
+from distutils.command.sdist import sdist
+import subprocess
 
 try:
     from wheel.bdist_wheel import bdist_wheel
@@ -13,6 +15,45 @@
 
 
 def add_rust_extension(dist):
+    sdist_base_class = dist.cmdclass.get("sdist", sdist)
+    sdist_options = sdist_base_class.user_options.copy()
+    sdist_boolean_options = sdist_base_class.boolean_options.copy()
+    sdist_negative_opt = sdist_base_class.negative_opt.copy()
+    sdist_options.extend([
+        ('vendor-crates', None,
+         "vendor Rust crates"),
+        ('no-vendor-crates', None,
+         "don't vendor Rust crates."
+         "[default; enable with --vendor-crates]"),
+    ])
+    sdist_boolean_options.append('vendor-crates')
+    sdist_negative_opt['no-vendor-crates'] = 'vendor-crates'
+
+    class sdist_rust_extension(sdist_base_class):
+        user_options = sdist_options
+        boolean_options = sdist_boolean_options
+        negative_opt = sdist_negative_opt
+
+        def initialize_options(self):
+            super().initialize_options()
+            self.vendor_crates = 0
+
+        def get_file_list(self):
+            if self.vendor_crates:
+                base_dir = self.distribution.get_fullname()
+                vendor_path = os.path.join(base_dir, "vendored-crates")
+                cargo_config = subprocess.check_output(["cargo", "vendor", vendor_path])
+                cargo_config = cargo_config.replace(base_dir.encode() + b'/', b'')
+                dot_cargo_path = os.path.join(base_dir, ".cargo")
+                self.mkpath(dot_cargo_path)
+                cargo_config_path = os.path.join(dot_cargo_path, "config.toml")
+                with open(cargo_config_path, "wb") as f:
+                    f.write(cargo_config)
+                self.filelist.append(vendor_path)
+                self.filelist.append(cargo_config_path)
+            super().get_file_list()
+    dist.cmdclass["sdist"] = sdist_rust_extension
+
     build_ext_base_class = dist.cmdclass.get('build_ext', build_ext)
 
     class build_ext_rust_extension(build_ext_base_class):