import subprocess
from pathlib import Path
from datetime import datetime

TIMEOUT_SECONDS = 60
SCRIPTS_DIR = Path("./scripts/generated")

def _save_script(script: str) -> Path:
    """Save the original (pre-Set-Location) script to ./scripts/generated/ with a timestamp name."""
    SCRIPTS_DIR.mkdir(parents=True, exist_ok=True)
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S_%f")
    path = SCRIPTS_DIR / f"ai_{timestamp}.ps1"
    path.write_text(script, encoding="utf-8")
    return path

def run_powershell(script: str, base_dir: str) -> str:
    """
    Run a PowerShell script with working directory set to base_dir.
    Saves the script to ./scripts/generated/ before running.
    Returns a string combining stdout, stderr, and exit code.
    Raises nothing - all errors are captured into the return string.
    """
    saved_path = _save_script(script)

    # 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 = [f"SAVED: {saved_path}"]
        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)
    except subprocess.TimeoutExpired:
        return f"SAVED: {saved_path}\nERROR: command timed out after {TIMEOUT_SECONDS}s"
    except FileNotFoundError:
        return f"SAVED: {saved_path}\nERROR: powershell executable not found"
    except Exception as e:
        return f"SAVED: {saved_path}\nERROR: {e}"