Use same setup.py as pythonnet

Author: filmor

Diff for: setup.py

@@ -1,33 +1,64 @@
 #!/usr/bin/env python
 
-from setuptools import setup, Command, Extension
-from wheel.bdist_wheel import bdist_wheel
+import distutils
+from distutils.command.build import build as _build
+from setuptools.command.develop import develop as _develop
+from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
+from setuptools import Distribution
+from setuptools import setup, Command
 
+import os
 
-class DotnetLib(Extension):
+# Disable SourceLink during the build until it can read repo-format v1, #1613
+os.environ["EnableSourceControlManagerQueries"] = "false"
+
+
+class DotnetLib:
     def __init__(self, name, path, **kwargs):
+        self.name = name
         self.path = path
         self.args = kwargs
-        super().__init__(name, sources=[])
 
 
-class BuildDotnet(Command):
+class build_dotnet(Command):
     """Build command for dotnet-cli based builds"""
 
     description = "Build DLLs with dotnet-cli"
-    user_options = [("dotnet-config=", None, "dotnet build configuration")]
+    user_options = [
+        ("dotnet-config=", None, "dotnet build configuration"),
+        (
+            "inplace",
+            "i",
+            "ignore build-lib and put compiled extensions into the source "
+            + "directory alongside your pure Python modules",
+        ),
+    ]
 
     def initialize_options(self):
-        self.dotnet_config = "release"
+        self.dotnet_config = None
+        self.build_lib = None
+        self.inplace = False
 
     def finalize_options(self):
-        pass
+        if self.dotnet_config is None:
+            self.dotnet_config = "release"
 
-    def get_source_files(self):
-        return []
+        build = self.distribution.get_command_obj("build")
+        build.ensure_finalized()
+        if self.inplace:
+            self.build_lib = "."
+        else:
+            self.build_lib = build.build_lib
 
     def run(self):
-        for lib in self.distribution.ext_modules:
+        dotnet_modules = self.distribution.dotnet_libs
+
+        for lib in dotnet_modules:
+            output = os.path.join(
+                os.path.abspath(self.build_lib), lib.args.pop("output")
+            )
+            rename = lib.args.pop("rename", {})
+
             opts = sum(
                 [
                     ["--" + name.replace("_", "-"), value]
@@ -36,34 +67,79 @@ def run(self):
                 [],
             )
 
-            opts.append("--configuration")
-            opts.append(self.dotnet_config)
+            opts.extend(["--configuration", self.dotnet_config])
+            opts.extend(["--output", output])
 
+            self.announce("Running dotnet build...", level=distutils.log.INFO)
             self.spawn(["dotnet", "build", lib.path] + opts)
 
+            for k, v in rename.items():
+                source = os.path.join(output, k)
+                dest = os.path.join(output, v)
+
+                if os.path.isfile(source):
+                    try:
+                        os.remove(dest)
+                    except OSError:
+                        pass
+
+                    self.move_file(src=source, dst=dest, level=distutils.log.INFO)
+                else:
+                    self.warn(
+                        "Can't find file to rename: {}, current dir: {}".format(
+                            source, os.getcwd()
+                        )
+                    )
+
 
-class bdist_wheel_patched(bdist_wheel):
+# Add build_dotnet to the build tasks:
+class build(_build):
+    sub_commands = _build.sub_commands + [("build_dotnet", None)]
+
+
+class develop(_develop):
+    def install_for_development(self):
+        # Build extensions in-place
+        self.reinitialize_command("build_dotnet", inplace=1)
+        self.run_command("build_dotnet")
+
+        return super().install_for_development()
+
+
+class bdist_wheel(_bdist_wheel):
     def finalize_options(self):
         # Monkey patch bdist_wheel to think the package is pure even though we
         # include DLLs
         super().finalize_options()
         self.root_is_pure = True
 
 
+# Monkey-patch Distribution s.t. it supports the dotnet_libs attribute
+Distribution.dotnet_libs = None
+
+cmdclass = {
+    "build": build,
+    "build_dotnet": build_dotnet,
+    "develop": develop,
+    "bdist_wheel": bdist_wheel,
+}
+
+dotnet_libs = [
+    DotnetLib(
+        "netfx-loader-x86",
+        "netfx_loader/ClrLoader.csproj",
+        runtime="win-x86",
+        output="clr_loader/ffi/dlls/x86",
+    ),
+    DotnetLib(
+        "netfx-loader-amd64",
+        "netfx_loader/ClrLoader.csproj",
+        runtime="win-x64",
+        output="clr_loader/ffi/dlls/amd64",
+    ),
+]
+
 setup(
-    cmdclass={"build_ext": BuildDotnet, "bdist_wheel": bdist_wheel_patched},
-    ext_modules={
-        DotnetLib(
-            "netfx-loader-x86",
-            "netfx_loader/ClrLoader.csproj",
-            runtime="win-x86",
-            output="clr_loader/ffi/dlls/x86",
-        ),
-        DotnetLib(
-            "netfx-loader-amd64",
-            "netfx_loader/ClrLoader.csproj",
-            runtime="win-x64",
-            output="clr_loader/ffi/dlls/amd64",
-        ),
-    },
+    cmdclass=cmdclass,
+    dotnet_libs=dotnet_libs,
 )