save scripts

shell_runner.py

@@ -1,15 +1,27 @@
-import subprocess
-import shlex
+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}"
 
@@ -21,16 +33,16 @@ def run_powershell(script: str, base_dir: str) -> str:
             timeout=TIMEOUT_SECONDS,
             cwd=base_dir
         )
-        parts = []
+        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) if parts else f"EXIT CODE: {result.returncode}"
+        return "\n".join(parts)
     except subprocess.TimeoutExpired:
-        return f"ERROR: command timed out after {TIMEOUT_SECONDS}s"
+        return f"SAVED: {saved_path}\nERROR: command timed out after {TIMEOUT_SECONDS}s"
     except FileNotFoundError:
-        return "ERROR: powershell executable not found"
+        return f"SAVED: {saved_path}\nERROR: powershell executable not found"
     except Exception as e:
-        return f"ERROR: {e}"
+        return f"SAVED: {saved_path}\nERROR: {e}"