# shell_runner.py
import subprocess, shutil
from pathlib import Path
from typing import Callable, Optional

TIMEOUT_SECONDS: int = 60

def run_powershell(script: str, base_dir: str, qa_callback: Optional[Callable[[str], str]] = None) -> str:
 """
    Run a PowerShell script with working directory set to base_dir.
    Returns a string combining stdout, stderr, and exit code.
    If qa_callback is provided and the command fails or has stderr, 
    the callback is called with the stderr content and its result is appended.
    """
 safe_dir: str = str(base_dir).replace("'", "''")
 full_script: str = f"Set-Location -LiteralPath '{safe_dir}'\n{script}"
 # Try common executable names
 exe: Optional[str] = next((x for x in ["powershell.exe", "pwsh.exe", "powershell", "pwsh"] if shutil.which(x)), None)
 if not exe: return "ERROR: Neither powershell nor pwsh found in PATH"
 try:
  process = subprocess.Popen(
   [exe, "-NoProfile", "-NonInteractive", "-Command", full_script],
   stdout=subprocess.PIPE, stderr=subprocess.PIPE, text=True, cwd=base_dir
  )
  stdout, stderr = process.communicate(timeout=TIMEOUT_SECONDS)
  
  parts: list[str] = []
  if stdout.strip(): parts.append(f"STDOUT:\n{stdout.strip()}")
  if stderr.strip(): parts.append(f"STDERR:\n{stderr.strip()}")
  parts.append(f"EXIT CODE: {process.returncode}")
  
  if (process.returncode != 0 or stderr.strip()) and qa_callback:
   qa_analysis: Optional[str] = qa_callback(stderr.strip())
   if qa_analysis:
    parts.append(f"\nQA ANALYSIS:\n{qa_analysis}")
  return "\n".join(parts)
 except subprocess.TimeoutExpired:
  if 'process' in locals() and process:
   subprocess.run(["taskkill", "/F", "/T", "/PID", str(process.pid)], capture_output=True)
  return f"ERROR: timed out after {TIMEOUT_SECONDS}s"
 except KeyboardInterrupt:
  if 'process' in locals() and process:
   subprocess.run(["taskkill", "/F", "/T", "/PID", str(process.pid)], capture_output=True)
  raise
 except Exception as e:
  if 'process' in locals() and process:
   subprocess.run(["taskkill", "/F", "/T", "/PID", str(process.pid)], capture_output=True)
  return f"ERROR: {e}"