#!/usr/bin/env python3

"""Download Claude Code CLI binary for bundling in wheel.

This script is run during the wheel build process to fetch the Claude Code CLI

binary using the official install script and place it in the package directory.

"""

import os

import platform

import random

import shutil

import subprocess

import sys

import time

from pathlib import Path

def get_cli_version() -> str:

"""Get the CLI version to download from environment or default."""

return os.environ.get("CLAUDE_CLI_VERSION", "latest")

def find_installed_cli() -> Path | None:

"""Find the installed Claude CLI binary."""

system = platform.system()

if system == "Windows":

# Windows installation locations (matches test.yml: $USERPROFILE\.local\bin)

locations = [

Path.home() / ".local" / "bin" / "claude.exe",

Path(os.environ.get("LOCALAPPDATA", "")) / "Claude" / "claude.exe",

]

else:

# Unix installation locations

locations = [

Path.home() / ".local" / "bin" / "claude",

Path("/usr/local/bin/claude"),

Path.home() / "node_modules" / ".bin" / "claude",

]

# Also check PATH

cli_path = shutil.which("claude")

if cli_path:

return Path(cli_path)

for path in locations:

if path.exists() and path.is_file():

return path

return None

def download_cli() -> None:

"""Download Claude Code CLI using the official install script."""

version = get_cli_version()

system = platform.system()

print(f"Downloading Claude Code CLI version: {version}")

# Build install command based on platform

if system == "Windows":

# Use PowerShell installer on Windows

if version == "latest":

install_cmd = [

"powershell",

"-ExecutionPolicy",

"Bypass",

"-Command",

"irm https://claude.ai/install.ps1 | iex",

]

else:

install_cmd = [

"powershell",

"-ExecutionPolicy",

"Bypass",

"-Command",

f"& ([scriptblock]::Create((irm https://claude.ai/install.ps1))) {version}",

]

else:

# --retry-all-errors covers 429 from claude.ai when multiple matrix

# jobs fetch install.sh simultaneously. pipefail propagates curl's

# exit code through the pipe so subprocess.run's check=True catches it.

curl = "curl -fsSL --retry 5 --retry-delay 2 --retry-all-errors https://claude.ai/install.sh"

target = "" if version == "latest" else f" -s {version}"

install_cmd = ["bash", "-c", f"set -o pipefail; {curl} | bash{target}"]

# Small jitter to stagger parallel matrix builds hitting the same endpoint

time.sleep(random.uniform(0, 5))

last_err: subprocess.CalledProcessError | None = None

for attempt in range(1, 4):

try:

subprocess.run(install_cmd, check=True, capture_output=True)

return

except subprocess.CalledProcessError as e:

last_err = e

if attempt < 3:

delay = 2**attempt

print(

f"Install attempt {attempt} failed (exit {e.returncode}), "

f"retrying in {delay}s...",

file=sys.stderr,

)

time.sleep(delay)

print(f"Error downloading CLI after 3 attempts: {last_err}", file=sys.stderr)

print(f"stdout: {last_err.stdout.decode()}", file=sys.stderr)

print(f"stderr: {last_err.stderr.decode()}", file=sys.stderr)

sys.exit(1)

def copy_cli_to_bundle() -> None:

"""Copy the installed CLI to the package _bundled directory."""

# Find project root (parent of scripts directory)

script_dir = Path(__file__).parent

project_root = script_dir.parent

bundle_dir = project_root / "src" / "claude_agent_sdk" / "_bundled"

# Ensure bundle directory exists

bundle_dir.mkdir(parents=True, exist_ok=True)

# Find installed CLI

cli_path = find_installed_cli()

if not cli_path:

print("Error: Could not find installed Claude CLI binary", file=sys.stderr)

sys.exit(1)

print(f"Found CLI at: {cli_path}")

# Determine target filename based on platform

system = platform.system()

target_name = "claude.exe" if system == "Windows" else "claude"

target_path = bundle_dir / target_name

# Copy the binary

print(f"Copying CLI to: {target_path}")

shutil.copy2(cli_path, target_path)

# Make it executable (Unix-like systems)

if system != "Windows":

target_path.chmod(0o755)

print(f"Successfully bundled CLI binary: {target_path}")

# Print size info

size_mb = target_path.stat().st_size / (1024 * 1024)

print(f"Binary size: {size_mb:.2f} MB")

def main() -> None:

"""Main entry point."""

print("=" * 60)

print("Claude Code CLI Download Script")

print("=" * 60)

# Download CLI

download_cli()

# Copy to bundle directory

copy_cli_to_bundle()

print("=" * 60)

print("CLI download and bundling complete!")

print("=" * 60)

if __name__ == "__main__":

main()