Add RustBin

Author: messense

examples/hello-world/setup.py

@@ -1,15 +1,13 @@
 from setuptools import setup
 
-from setuptools_rust import Binding, RustExtension
+from setuptools_rust import RustBin
 
 setup(
     name="hello-world",
     version="1.0",
     rust_extensions=[
-        RustExtension(
+        RustBin(
             "hello-world",
-            binding=Binding.Exec,
-            script=True,
             args=["--profile", "release-lto"],
         )
     ],

setuptools_rust/__init__.py

@@ -1,6 +1,6 @@
 from .build import build_rust
 from .clean import clean_rust
-from .extension import Binding, RustExtension, Strip
+from .extension import Binding, RustBin, RustExtension, Strip
 from .version import version as __version__
 
-__all__ = ("Binding", "RustExtension", "Strip", "build_rust", "clean_rust")
+__all__ = ("Binding", "RustBin", "RustExtension", "Strip", "build_rust", "clean_rust")

setuptools_rust/build.py

@@ -21,7 +21,7 @@
 from typing_extensions import Literal
 
 from .command import RustCommand
-from .extension import RustExtension, Strip
+from .extension import RustBin, RustExtension, Strip
 from .utils import (
     PyLimitedApi,
     binding_features,
@@ -313,9 +313,7 @@ def install_extension(
                     ext_path += exe
 
                 os.makedirs(os.path.dirname(ext_path), exist_ok=True)
-                if "." in module_name:
-                    ext.install_script(module_name.split(".")[-1], ext_path)
-                else:
+                if isinstance(ext, RustBin):
                     executable_name = module_name
                     if exe is not None:
                         executable_name += exe
@@ -325,6 +323,8 @@ def install_extension(
                     )
                     os.makedirs(scripts_dir, exist_ok=True)
                     ext_path = os.path.join(scripts_dir, executable_name)
+                else:
+                    ext.install_script(module_name.split(".")[-1], ext_path)
             else:
                 ext_path = self.get_dylib_ext_path(ext, module_name)
                 os.makedirs(os.path.dirname(ext_path), exist_ok=True)

setuptools_rust/extension.py

@@ -195,10 +195,9 @@ def entry_points(self) -> List[str]:
         entry_points = []
         if self.script and self.binding == Binding.Exec:
             for executable, mod in self.target.items():
-                if "." in mod:
-                    base_mod, name = mod.rsplit(".")
-                    script = "%s=%s.%s:run" % (name, base_mod, _script_name(executable))
-                    entry_points.append(script)
+                base_mod, name = mod.rsplit(".")
+                script = "%s=%s.%s:run" % (name, base_mod, _script_name(executable))
+                entry_points.append(script)
 
         return entry_points
 
@@ -231,6 +230,59 @@ def _uses_exec_binding(self) -> bool:
         return self.binding == Binding.Exec
 
 
+class RustBin(RustExtension):
+    """Used to define a Rust binary and its build configuration.
+
+    Args:
+        target: Rust binary target name.
+        path: Path to the ``Cargo.toml`` manifest file.
+        args: A list of extra arguments to be passed to Cargo. For example,
+            ``args=["--no-default-features"]`` will disable the default
+            features listed in ``Cargo.toml``.
+        features: A list of Cargo features to also build.
+        rustc_flags: A list of additional flags passed to rustc.
+        rust_version: Minimum Rust compiler version required for this
+            extension.
+        quiet: Suppress Cargo's output.
+        debug: Controls whether ``--debug`` or ``--release`` is passed to
+            Cargo. If set to `None` (the default) then build type is
+            automatic: ``inplace`` build will be a debug build, ``install``
+            and ``wheel`` builds will be release.
+        strip: Strip symbols from final file. Does nothing for debug build.
+        native: Build extension or executable with ``--target-cpu=native``.
+    """
+
+    def __init__(
+        self,
+        target: str,
+        path: str = "Cargo.toml",
+        args: Optional[List[str]] = None,
+        features: Optional[List[str]] = None,
+        rustc_flags: Optional[List[str]] = None,
+        rust_version: Optional[str] = None,
+        quiet: bool = False,
+        debug: Optional[bool] = None,
+        strip: Strip = Strip.No,
+        native: bool = False,
+    ):
+        super().__init__(
+            target,
+            path,
+            args,
+            features,
+            rustc_flags,
+            rust_version,
+            quiet,
+            debug,
+            binding=Binding.Exec,
+            strip=strip,
+            native=native,
+        )
+
+    def entry_points(self) -> List[str]:
+        return []
+
+
 _CargoMetadata = NewType("_CargoMetadata", Dict[str, Any])