deer-flow/scripts/pnpm.py
dong f09e824460
fix(scripts): prefer Windows .cmd shims for pnpm (#5305)
* fix(scripts): prefer Windows cmd shims for pnpm

* fix(scripts): address pnpm Windows review feedback

* docs: reduce inherited agent guidance size
2026-09-11 18:41:24 +08:00

74 lines
2.2 KiB
Python

#!/usr/bin/env python3
"""Run pnpm directly when available, otherwise run it through Corepack."""
from __future__ import annotations
import os
import shutil
import subprocess
import sys
from collections.abc import Sequence
from pathlib import Path
FRONTEND_DIR = Path(__file__).resolve().parent.parent / "frontend"
COREPACK_NOTICE = "Using pnpm via Corepack."
def find_pnpm_command() -> list[str] | None:
"""Return the preferred pnpm-compatible command for this machine."""
pnpm_names = ("pnpm.cmd", "pnpm") if os.name == "nt" else ("pnpm", "pnpm.cmd")
for name in pnpm_names:
if pnpm_path := shutil.which(name):
return [str(Path(pnpm_path))]
corepack_names = ("corepack.cmd", "corepack") if os.name == "nt" else ("corepack", "corepack.cmd")
for name in corepack_names:
if corepack_path := shutil.which(name):
return [str(Path(corepack_path)), "pnpm"]
return None
def run_pnpm(arguments: Sequence[str]) -> int:
"""Run pnpm with the supplied arguments and propagate its exit status."""
command = find_pnpm_command()
if command is None:
print(
"Error: Neither pnpm nor Corepack is available on PATH.",
file=sys.stderr,
)
print(
"Install pnpm, or install Corepack and ensure 'corepack' is on PATH.",
file=sys.stderr,
)
return 127
if Path(command[0]).stem.lower() == "corepack":
print(COREPACK_NOTICE, file=sys.stderr)
try:
result = subprocess.run(
[*command, *arguments],
check=False,
shell=False,
cwd=FRONTEND_DIR,
)
except OSError as exc:
print(f"Error: Failed to run pnpm via {command[0]}: {exc}", file=sys.stderr)
return 126
exit_code = 128 - result.returncode if result.returncode < 0 else result.returncode
if exit_code != 0:
print(
f"Error: pnpm command failed with exit status {exit_code}.",
file=sys.stderr,
)
return exit_code
def main(argv: Sequence[str] | None = None) -> int:
return run_pnpm(sys.argv[1:] if argv is None else argv)
if __name__ == "__main__":
sys.exit(main())