Improve pyscript wheel generation

Author: Archmonger

.gitignore

@@ -2,6 +2,7 @@
 src/reactpy/static/*.js*
 src/reactpy/static/morphdom/
 src/reactpy/static/pyscript/
+src/reactpy/static/wheels/
 src/js/**/*.tgz
 src/js/**/LICENSE
 

pyproject.toml

@@ -71,8 +71,8 @@ installer = "uv"
 reactpy = "reactpy._console.cli:entry_point"
 
 [[tool.hatch.build.hooks.build-scripts.scripts]]
-commands = []
-artifacts = []
+commands = ['python "src/build_scripts/build_local_wheel.py"']
+artifacts = ["src/reactpy/static/wheels/*.whl"]
 
 #############################
 # >>> Hatch Test Runner <<< #

src/build_scripts/build_local_wheel.py

@@ -0,0 +1,106 @@
+# /// script
+# requires-python = ">=3.11"
+# dependencies = []
+# ///
+
+from __future__ import annotations
+
+import logging
+import os
+import re
+import shutil
+import subprocess
+from pathlib import Path
+
+_logger = logging.getLogger(__name__)
+_SKIP_ENV_VAR = "REACTPY_SKIP_LOCAL_WHEEL_BUILD"
+
+
+def _reactpy_version(root_dir: Path) -> str:
+    init_file = root_dir / "src" / "reactpy" / "__init__.py"
+    if match := re.search(
+        r'^__version__ = "([^"]+)"$',
+        init_file.read_text(encoding="utf-8"),
+        re.MULTILINE,
+    ):
+        return match.group(1)
+    raise RuntimeError("Could not determine the current ReactPy version.")
+
+
+def _matching_reactpy_wheel(dist_dir: Path, version: str) -> Path | None:
+    matching_wheels = sorted(
+        dist_dir.glob(f"reactpy-{version}-*.whl"),
+        key=lambda path: path.stat().st_mtime,
+        reverse=True,
+    )
+    return matching_wheels[0] if matching_wheels else None
+
+
+def _hatch_build_command(root_dir: Path) -> list[str] | None:
+    for candidate in (
+        root_dir / ".venv" / "Scripts" / "hatch.exe",
+        root_dir / ".venv" / "bin" / "hatch",
+    ):
+        if candidate.exists():
+            return [str(candidate), "build", "-t", "wheel"]
+
+    if hatch_command := shutil.which("hatch"):
+        return [hatch_command, "build", "-t", "wheel"]
+
+    return None
+
+
+def main() -> int:
+    if os.environ.get(_SKIP_ENV_VAR):
+        print("Skipping local ReactPy wheel build.")  # noqa: T201
+        return 0
+
+    root_dir = Path(__file__).parent.parent.parent
+    version = _reactpy_version(root_dir)
+    static_wheels_dir = root_dir / "src" / "reactpy" / "static" / "wheels"
+    dist_dir = root_dir / "dist"
+    hatch_build_command = _hatch_build_command(root_dir)
+
+    if not hatch_build_command:
+        _logger.error("Could not locate Hatch while building the embedded wheel.")
+        return 1
+
+    static_wheels_dir.mkdir(parents=True, exist_ok=True)
+    for wheel_file in static_wheels_dir.glob("reactpy-*.whl"):
+        wheel_file.unlink()
+
+    env = os.environ.copy()
+    env[_SKIP_ENV_VAR] = "1"
+    for key in tuple(env):
+        if key.startswith("HATCH_ENV_"):
+            env.pop(key)
+
+    result = subprocess.run(  # noqa: S603
+        hatch_build_command,
+        capture_output=True,
+        text=True,
+        check=False,
+        cwd=root_dir,
+        env=env,
+    )
+
+    if result.returncode != 0:
+        _logger.error(
+            "Failed to build the embedded ReactPy wheel.\nstdout:\n%s\nstderr:\n%s",
+            result.stdout,
+            result.stderr,
+        )
+        return result.returncode
+
+    built_wheel = _matching_reactpy_wheel(dist_dir, version)
+    if not built_wheel:
+        _logger.error("Failed to locate the newly built ReactPy wheel in %s", dist_dir)
+        return 1
+
+    shutil.copy2(built_wheel, static_wheels_dir / built_wheel.name)
+    print(f"Embedded local ReactPy wheel at '{static_wheels_dir / built_wheel.name}'")  # noqa: T201
+    return 0
+
+
+if __name__ == "__main__":
+    raise SystemExit(main())

src/reactpy/executors/asgi/middleware.py

@@ -268,6 +268,7 @@ async def __call__(
                 Error404App(),
                 root=self.parent.static_dir,
                 prefix=self.parent.static_path,
+                autorefresh=True,
             )
 
         await self._static_file_server(