chore: add one-time windows cross-check setup

2 files changed, 188 insertions, 0 deletions

diff --git a/scripts/test_windows_cross.py b/scripts/test_windows_cross.py
new file mode 100644
index 00000000..4f8f0fe4
--- /dev/null
+++ b/scripts/test_windows_cross.py
@@ -0,0 +1,82 @@
+import os
+from pathlib import Path
+import tempfile
+import unittest
+from unittest.mock import patch
+
+from scripts import windows_cross
+
+
+class WindowsCrossTests(unittest.TestCase):
+    def test_libc_configuration_matches_xwin_layout(self):
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            for relative in (
+                "sdk/include/ucrt/stdlib.h", "crt/include/vcruntime.h",
+                "sdk/lib/ucrt/x86_64/ucrt.lib", "crt/lib/x86_64/vcruntime.lib",
+                "sdk/lib/um/x86_64/kernel32.lib",
+            ):
+                path = root / relative
+                path.parent.mkdir(parents=True, exist_ok=True)
+                path.touch()
+            contents = windows_cross.libc_contents(root)
+            self.assertIn(f"include_dir={root}/sdk/include/ucrt\n", contents)
+            self.assertIn(f"msvc_lib_dir={root}/crt/lib/x86_64\n", contents)
+            self.assertTrue(contents.endswith("gcc_dir=\n"))
+            (root / "crt/include/vcruntime.h").unlink()
+            with self.assertRaisesRegex(ValueError, "missing .*vcruntime.h"):
+                windows_cross.libc_contents(root)
+
+    def test_persistent_configuration_and_explicit_override(self):
+        with tempfile.TemporaryDirectory() as directory:
+            root = Path(directory)
+            default = root / "libc.txt"
+            override = root / "custom.txt"
+            default.touch()
+            override.touch()
+            with patch.object(windows_cross, "SDK_ROOT", root), patch.dict(os.environ, {}, clear=True):
+                self.assertEqual(windows_cross.libc_path(), default)
+                with patch.dict(os.environ, {windows_cross.LIBC_ENV: str(override)}):
+                    self.assertEqual(windows_cross.libc_path(), override)
+                    override.unlink()
+                    with self.assertRaisesRegex(ValueError, "just setup-windows-cross"):
+                        windows_cross.libc_path()
+
+    def test_missing_setup_does_not_run_build_or_download(self):
+        with tempfile.TemporaryDirectory() as directory:
+            with patch.object(windows_cross, "SDK_ROOT", Path(directory)), \
+                    patch.dict(os.environ, {}, clear=True), \
+                    patch.object(windows_cross.subprocess, "run") as run:
+                with self.assertRaisesRegex(ValueError, "just setup-windows-cross"):
+                    windows_cross.lint()
+                run.assert_not_called()
+
+    def test_lint_passes_sdk_to_cargo_without_changing_parent_environment(self):
+        with patch.object(windows_cross, "libc_path", return_value=Path("/sdk/libc.txt")), \
+                patch.dict(os.environ, {"KEEP_ME": "yes"}, clear=True), \
+                patch.object(windows_cross.subprocess, "run") as run:
+            windows_cross.lint()
+            self.assertEqual(run.call_count, 2)
+            cargo = run.call_args
+            self.assertEqual(cargo.args[0][:2], ["cargo", "clippy"])
+            self.assertEqual(cargo.kwargs["env"][windows_cross.LIBC_ENV], "/sdk/libc.txt")
+            self.assertEqual(cargo.kwargs["env"]["KEEP_ME"], "yes")
+            self.assertNotIn(windows_cross.LIBC_ENV, os.environ)
+
+    def test_license_acceptance_is_only_forwarded_when_explicit(self):
+        with tempfile.TemporaryDirectory() as directory:
+            with patch.object(windows_cross, "SDK_ROOT", Path(directory)), \
+                    patch.object(windows_cross.shutil, "which", return_value="tool"), \
+                    patch.object(windows_cross, "libc_contents", return_value="configuration"), \
+                    patch.object(windows_cross.subprocess, "run") as run:
+                for accepted in (False, True):
+                    run.reset_mock()
+                    windows_cross.setup(accepted)
+                    command = run.call_args_list[0].args[0]
+                    self.assertEqual("--accept-license" in command, accepted)
+                    self.assertEqual(command[0], "xwin")
+                    self.assertIn("--copy", command)
+
+
+if __name__ == "__main__":
+    unittest.main()
diff --git a/scripts/windows_cross.py b/scripts/windows_cross.py
new file mode 100644
index 00000000..55795d35
--- /dev/null
+++ b/scripts/windows_cross.py
@@ -0,0 +1,106 @@
+"""One-time Windows SDK setup and Windows target linting from Unix hosts."""
+
+import argparse
+import os
+from pathlib import Path
+import shutil
+import subprocess
+import sys
+import tempfile
+
+
+SDK_ROOT = Path.home() / ".local/share/herdr/windows-cross"
+LIBC_ENV = "LIBGHOSTTY_VT_WINDOWS_LIBC"
+TARGET = "x86_64-pc-windows-msvc"
+
+
+def libc_contents(root: Path) -> str:
+    paths = {
+        "include_dir": root / "sdk/include/ucrt",
+        "sys_include_dir": root / "crt/include",
+        "crt_dir": root / "sdk/lib/ucrt/x86_64",
+        "msvc_lib_dir": root / "crt/lib/x86_64",
+        "kernel32_lib_dir": root / "sdk/lib/um/x86_64",
+    }
+    required = {
+        "include_dir": "stdlib.h",
+        "sys_include_dir": "vcruntime.h",
+        "crt_dir": "ucrt.lib",
+        "msvc_lib_dir": "vcruntime.lib",
+        "kernel32_lib_dir": "kernel32.lib",
+    }
+    for key, filename in required.items():
+        if not (paths[key] / filename).is_file():
+            raise ValueError(f"Windows SDK is incomplete: missing {paths[key] / filename}")
+    return "".join(f"{key}={value}\n" for key, value in paths.items()) + "gcc_dir=\n"
+
+
+def libc_path() -> Path:
+    override = os.environ.get(LIBC_ENV)
+    path = Path(override).expanduser() if override else SDK_ROOT / "libc.txt"
+    if not path.is_file():
+        raise ValueError(
+            f"Windows cross-check needs SDK configuration at {path}.\n"
+            "Run `just setup-windows-cross` once, or set "
+            f"{LIBC_ENV} to an existing Zig libc configuration."
+        )
+    return path.resolve()
+
+
+def setup(accept_license: bool) -> None:
+    if not shutil.which("xwin"):
+        raise ValueError("Install xwin first: cargo install xwin --locked")
+    if not shutil.which(os.environ.get("ZIG", "zig")):
+        raise ValueError("Install Zig 0.16.0 first, or set ZIG to its executable.")
+    SDK_ROOT.mkdir(parents=True, exist_ok=True)
+    # SDK downloads can be large; /tmp is often RAM-backed on Linux.
+    temp_parent = "/var/tmp" if sys.platform.startswith("linux") else None
+    with tempfile.TemporaryDirectory(prefix="herdr-windows-sdk-", dir=temp_parent) as cache:
+        command = ["xwin", "--arch", "x86_64", "--cache-dir", cache]
+        if accept_license:
+            command.append("--accept-license")
+        subprocess.run(command + ["splat", "--copy", "--output", str(SDK_ROOT)], check=True)
+    config = SDK_ROOT / "libc.txt"
+    config.write_text(libc_contents(SDK_ROOT))
+    subprocess.run(
+        [os.environ.get("ZIG", "zig"), "libc", "-target", "x86_64-windows-msvc", str(config)],
+        check=True,
+    )
+    print(f"Windows SDK configured at {config}. Run `just windows-lint` or `just check`.")
+
+
+def lint() -> None:
+    env = {**os.environ, LIBC_ENV: str(libc_path()), "LIBGHOSTTY_VT_SIMD": "false"}
+    subprocess.run(["rustup", "target", "add", TARGET], check=True)
+    subprocess.run(
+        ["cargo", "clippy", "--bin", "herdr", "--locked", "--target", TARGET, "--", "-D", "warnings"],
+        env=env,
+        check=True,
+    )
+
+
+def main() -> int:
+    parser = argparse.ArgumentParser(description=__doc__)
+    commands = parser.add_subparsers(dest="command", required=True)
+    setup_parser = commands.add_parser("setup", help="Download Microsoft's SDK using xwin")
+    setup_parser.add_argument(
+        "--accept-license", action="store_true",
+        help="Explicitly accept Microsoft's SDK license instead of xwin's interactive prompt",
+    )
+    commands.add_parser("lint", help="Run Windows clippy with the configured SDK")
+    args = parser.parse_args()
+    try:
+        if args.command == "setup":
+            setup(args.accept_license)
+        else:
+            lint()
+    except (ValueError, OSError) as error:
+        print(error, file=sys.stderr)
+        return 1
+    except subprocess.CalledProcessError as error:
+        return error.returncode
+    return 0
+
+
+if __name__ == "__main__":
+    sys.exit(main())