claude cooked

Author: jmelahman

tools/gonyx/hatch_build.py

@@ -2,32 +2,45 @@
 
 import os
 import subprocess
+from typing import Any
 
 import manygo
-from hatchling.builders.hooks.plugin import interface
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
 
 
-class GoBinaryBuildHook(interface.BuildHookInterface):
-    def initialize(self, version, build_data) -> None:  # noqa: ANN001, ARG002
+class CustomBuildHook(BuildHookInterface):
+    """Build hook to compile the Go binary and include it in the wheel."""
+
+    def initialize(self, version: Any, build_data: Any) -> None:
+        """Build the Go binary before packaging."""
         build_data["pure_python"] = False
+
+        # Set platform tag for cross-compilation
         goos = os.getenv("GOOS")
         goarch = os.getenv("GOARCH")
         if goos and goarch:
-            build_data["tag"] = "py3-none-" + manygo.get_platform_tag(goos=goos, goarch=goarch)  # type: ignore[invalid-argument-type]
-        binary_name = self.config["binary_name"]
-        tag = os.getenv("GITHUB_REF_NAME", "dev").lstrip("gonyx/")
+            build_data["tag"] = "py3-none-" + manygo.get_platform_tag(
+                goos=goos, goarch=goarch
+            )
+        else:
+            build_data["infer_tag"] = True
+
+        # Get config and environment
+        binary_name = self.config.get("binary_name", "gonyx")
         commit = os.getenv("GITHUB_SHA", "none")
 
+        # Build the Go binary if it doesn't exist
         if not os.path.exists(binary_name):
             print(f"Building Go binary '{binary_name}'...")
             subprocess.check_call(  # noqa: S603
                 [
                     "go",
                     "build",
-                    f"-ldflags=-X main.version={tag} -X main.commit={commit} -s -w",
+                    f"-ldflags=-X main.version={version} -X main.commit={commit} -s -w",
                     "-o",
                     binary_name,
                 ],
             )
 
-        build_data["shared_scripts"] = {binary_name: binary_name}
+        # Include the binary in the wheel
+        build_data["force_include"][binary_name] = binary_name

tools/gonyx/pyproject.toml

@@ -24,11 +24,16 @@ dynamic = ["version"]
 Repository = "https://github.com/onyx-dot-app/onyx"
 
 [tool.hatch.build]
-include = ["go.mod", "go.sum", "main.go"]
+include = ["go.mod", "go.sum", "main.go", "cmd/**/*.go"]
+artifacts = ["gonyx"]
 
 [tool.hatch.version]
 source = "vcs"
-tag-pattern = "gonyx/v(?P<version>\\d+\\.\\d+\\.\\d+)"
 
-[tool.hatch.build.hooks.custom]
+[tool.hatch.version.raw-options]
+root = "../.."
+tag_regex = "^gonyx/v(?P<version>[vV]?\\d+(?:\\.\\d+){0,2}[^\\+]*)(?:\\+.*)?$"
+
+[tool.hatch.build.targets.wheel.hooks.custom]
+path = "hatch_build.py"
 binary_name = "gonyx"