reinit with code

Author: timkpaine

Makefile

@@ -27,16 +27,16 @@ lint-py:  ## lint python with ruff
 	python -m ruff format --check hatch_go
 
 lint-docs:  ## lint docs with mdformat and codespell
-	python -m mdformat --check README.md 
-	python -m codespell_lib README.md 
+	python -m mdformat --check README.md
+	python -m codespell_lib README.md
 
 fix-py:  ## autoformat python code with ruff
 	python -m ruff check --fix hatch_go
 	python -m ruff format hatch_go
 
 fix-docs:  ## autoformat docs with mdformat and codespell
-	python -m mdformat README.md 
-	python -m codespell_lib --write README.md 
+	python -m mdformat README.md
+	python -m codespell_lib --write README.md
 
 lint: lint-py lint-docs  ## run all linters
 lints: lint
@@ -46,15 +46,12 @@ format: fix
 ################
 # Other Checks #
 ################
-.PHONY: check-dist check-types checks check
+.PHONY: check-manifest checks check
 
-check-dist:  ## check python sdist and wheel with check-dist
-	check-dist -v
+check-manifest:  ## check python sdist manifest with check-manifest
+	check-manifest -v
 
-check-types:  ## check python types with ty
-	ty check --python $$(which python)
-
-checks: check-dist
+checks: check-manifest
 
 # Alias
 check: checks

README.md

@@ -9,5 +9,14 @@ Hatch plugin for Go builds
 
 ## Overview
 
+A simple, extensible Go build plugin for [hatch](https://hatch.pypa.io/latest/). Build Python extension modules written in Go using cgo.
+
+```toml
+[tool.hatch.build.hooks.hatch-go]
+verbose = true
+path = "."
+module = "project"
+```
+
 > [!NOTE]
 > This library was generated using [copier](https://copier.readthedocs.io/en/stable/) from the [Base Python Project Template repository](https://github.com/python-project-templates/base).

hatch_go/__init__.py

@@ -1 +1,5 @@
 __version__ = "0.1.0"
+
+from .hooks import hatch_register_build_hook
+from .plugin import HatchGoBuildHook
+from .structs import *

hatch_go/hooks.py

@@ -0,0 +1,10 @@
+from typing import Type
+
+from hatchling.plugin import hookimpl
+
+from .plugin import HatchGoBuildHook
+
+
+@hookimpl
+def hatch_register_build_hook() -> Type[HatchGoBuildHook]:
+    return HatchGoBuildHook

hatch_go/plugin.py

@@ -0,0 +1,100 @@
+from __future__ import annotations
+
+from logging import getLogger
+from os import getenv
+from pathlib import Path
+from platform import machine as platform_machine
+from sys import platform as sys_platform, version_info
+from typing import Any
+
+from hatchling.builders.hooks.plugin.interface import BuildHookInterface
+
+from .structs import HatchGoBuildConfig, HatchGoBuildPlan
+from .utils import import_string
+
+__all__ = ("HatchGoBuildHook",)
+
+log = getLogger(__name__)
+
+
+class HatchGoBuildHook(BuildHookInterface[HatchGoBuildConfig]):
+    """The hatch-go build hook."""
+
+    PLUGIN_NAME = "hatch-go"
+    _logger = log
+
+    def initialize(self, version: str, build_data: dict[str, Any]) -> None:
+        """Initialize the plugin."""
+        # Log some basic information
+        project_name = self.metadata.config["project"]["name"]
+        self._logger.info("Initializing hatch-go plugin version %s", version)
+        self._logger.info(f"Running hatch-go: {project_name}")
+
+        # Only run if creating wheel
+        if self.target_name != "wheel":
+            self._logger.info("ignoring target name %s", self.target_name)
+            return
+
+        # Skip if SKIP_HATCH_GO is set
+        if getenv("SKIP_HATCH_GO"):
+            self._logger.info("Skipping the build hook since SKIP_HATCH_GO was set")
+            return
+
+        # Get build config class or use default
+        build_config_class = import_string(self.config["build-config-class"]) if "build-config-class" in self.config else HatchGoBuildConfig
+
+        # Instantiate build config
+        config = build_config_class(name=project_name, **self.config)
+
+        # Get build plan class or use default
+        build_plan_class = import_string(self.config["build-plan-class"]) if "build-plan-class" in self.config else HatchGoBuildPlan
+
+        # Instantiate builder
+        build_plan = build_plan_class(**config.model_dump())
+
+        # Generate commands
+        build_plan.generate()
+
+        # Log commands if in verbose mode
+        if build_plan.verbose:
+            for command in build_plan.commands:
+                self._logger.warning(command)
+
+        if build_plan.skip:
+            self._logger.warning("Skipping build")
+            return
+
+        # Execute build plan
+        build_plan.execute()
+
+        # Perform any cleanup actions
+        build_plan.cleanup()
+
+        if not build_plan._libraries:
+            raise ValueError("No libraries were created by the build.")
+
+        build_data["pure_python"] = False
+        machine = platform_machine()
+        version_major = version_info.major
+        version_minor = version_info.minor
+
+        if "darwin" in sys_platform:
+            os_name = "macosx_11_0"
+        elif "linux" in sys_platform:
+            os_name = "linux"
+        else:
+            os_name = "win"
+
+        build_data["tag"] = f"cp{version_major}{version_minor}-cp{version_major}{version_minor}-{os_name}_{machine}"
+
+        # Force include libraries
+        for path in Path(".").rglob("*"):
+            if path.is_dir():
+                continue
+            if str(path).startswith("dist") or not str(path).startswith(config.module):
+                continue
+            if path.suffix in (".pyd", ".dll", ".so", ".dylib"):
+                build_data["force_include"][str(path)] = str(path)
+
+        for path in build_data["force_include"]:
+            self._logger.warning(f"Force include: {path}")