vibin
This commit is contained in:
36
shell_runner.py
Normal file
36
shell_runner.py
Normal file
@@ -0,0 +1,36 @@
|
||||
import subprocess
|
||||
import shlex
|
||||
from pathlib import Path
|
||||
|
||||
TIMEOUT_SECONDS = 60
|
||||
|
||||
def run_powershell(script: str, base_dir: str) -> str:
|
||||
"""
|
||||
Run a PowerShell script with working directory set to base_dir.
|
||||
Returns a string combining stdout, stderr, and exit code.
|
||||
Raises nothing - all errors are captured into the return string.
|
||||
"""
|
||||
# Prepend Set-Location so the AI doesn't need to worry about cwd
|
||||
full_script = f"Set-Location -LiteralPath '{base_dir}'\n{script}"
|
||||
|
||||
try:
|
||||
result = subprocess.run(
|
||||
["powershell", "-NoProfile", "-NonInteractive", "-Command", full_script],
|
||||
capture_output=True,
|
||||
text=True,
|
||||
timeout=TIMEOUT_SECONDS,
|
||||
cwd=base_dir
|
||||
)
|
||||
parts = []
|
||||
if result.stdout.strip():
|
||||
parts.append(f"STDOUT:\n{result.stdout.strip()}")
|
||||
if result.stderr.strip():
|
||||
parts.append(f"STDERR:\n{result.stderr.strip()}")
|
||||
parts.append(f"EXIT CODE: {result.returncode}")
|
||||
return "\n".join(parts) if parts else f"EXIT CODE: {result.returncode}"
|
||||
except subprocess.TimeoutExpired:
|
||||
return f"ERROR: command timed out after {TIMEOUT_SECONDS}s"
|
||||
except FileNotFoundError:
|
||||
return "ERROR: powershell executable not found"
|
||||
except Exception as e:
|
||||
return f"ERROR: {e}"
|
||||
Reference in New Issue
Block a user