Local SE Agent Masterclass

import subprocess
import uuid
import shutil
from pathlib import Path

class GitWorktreeSandbox:
    """
    ตัวจัดการห้องทดลองลับด้วย Git Worktree (Python Context Manager)
    รับประกันว่าทุกการแก้ไขและรันเทสของ AI Agent จะเกิดขึ้นในโฟลเดอร์แยกส่วนตัว
    โดยไม่รบกวน Workspace หลักของผู้ใช้บน MacBook M3
    """
    def __init__(self, repo_root: str, base_commit: str = "HEAD"):
        self.repo_root = Path(repo_root).resolve()
        
        # ตรวจสอบว่าพิกัดเป็น Git Repository จริง
        if not (self.repo_root / ".git").exists():
            raise ValueError(f"พิกัด {self.repo_root} ไม่ใช่ Git Repository ที่ถูกต้อง!")

        # สร้างรหัสไอดีสุ่ม 8 ตัวอักษร เพื่อป้องกันชื่อโฟลเดอร์และ branch ชนกัน
        self.unique_id = uuid.uuid4().hex[:8]
        self.branch_name = f"agent/sandbox-{self.unique_id}"
        self.sandbox_path = Path("/tmp") / f"agent-worktree-{self.unique_id}"
        self.base_commit = base_commit

    def __enter__(self) -> Path:
        """
        [Spawn Phase]: ทำงานอัตโนมัติเมื่อเข้าสู่บล็อก 'with'
        สั่งรันคำสั่ง git worktree add เพื่อแตกโฟลเดอร์แยกคู่ขนาน
        """
        cmd = [
            "git", "-C", str(self.repo_root),
            "worktree", "add",
            "-b", self.branch_name,
            str(self.sandbox_path),
            self.base_commit
        ]
        
        try:
            subprocess.run(cmd, check=True, capture_output=True, text=True)
            return self.sandbox_path
        except subprocess.CalledProcessError as e:
            raise RuntimeError(f"ไม่สามารถสร้าง Git Worktree Sandbox ได้: {e.stderr}")

    def __exit__(self, exc_type, exc_val, exc_tb):
        """
        [Cleanup Phase]: ทำงานอัตโนมัติเมื่อหลุดจากบล็อก 'with' (ปลอดภัยแม้เกิด Error)
        ถอด Worktree ลบโฟลเดอร์ใน /tmp และลบ Branch ชั่วคราวทิ้งสะอาดหมดจด
        """
        # 1. ถอด Worktree ออกจากสารบัญ Git
        subprocess.run(
            ["git", "-C", str(self.repo_root), "worktree", "remove", "--force", str(self.sandbox_path)],
            check=False,
            capture_output=True,
            text=True
        )

        # 2. บังคับลบโฟลเดอร์ชั่วคราวซ้ำเพื่อความมั่นใจ
        if self.sandbox_path.exists():
            shutil.rmtree(self.sandbox_path, ignore_errors=True)

        # 3. ลบ Branch ชั่วคราวทิ้ง
        subprocess.run(
            ["git", "-C", str(self.repo_root), "branch", "-D", self.branch_name],
            check=False,
            capture_output=True,
            text=True
        )

from pathlib import Path
from core.sandbox import GitWorktreeSandbox

def main():
    print("=" * 70)
    print("🚀 เริ่มทดสอบ Day 22: Safe Sandbox Execution with Git Worktrees")
    print("=" * 70)

    # สมมติใช้โฟลเดอร์โปรเจกต์ปัจจุบันเป็น Git Repo หลัก
    repo_root = Path.cwd().resolve()
    print(f"📁 พิกัด Repository หลัก: {repo_root}")

    try:
        # ทดลองเปิดใช้งาน Context Manager ของ Sandbox
        with GitWorktreeSandbox(str(repo_root)) as sandbox_dir:
            print(f"✨ เข้าสู่ห้องทดลองลับสำเร็จที่: {sandbox_dir}")
            
            # 1. ทดลองสร้างไฟล์ใหม่ข้างใน Sandbox
            test_file = sandbox_dir / "agent_experiment.txt"
            test_file.write_text("Hello from Isolated Sandbox!", encoding="utf-8")
            print(f"✍️ Agent เขียนไฟล์ทดลองใน Sandbox สำเร็จ: {test_file.name}")

            # 2. ตรวจสอบว่าไฟล์นี้ไม่มีอยู่ใน Repo หลัก (Zero Interference)
            main_repo_file = repo_root / "agent_experiment.txt"
            assert not main_repo_file.exists(), "❌ ผิดพลาด: ไฟล์รั่วไหลเข้ามาใน Repo หลัก!"
            print("🛡️ ยืนยันความปลอดภัย: โฟลเดอร์หลักของมนุษย์สะอาด ไม่ถูกแตะต้อง 100%!")

        # 3. เมื่อหลุดจากบล็อก with โฟลเดอร์ต้องถูกทำลายทิ้ง
        print(f"\n🧹 ตรวจสอบหลังออกจาก with-block:")
        print(f"   👉 โฟลเดอร์ Sandbox ยังอยู่ไหม? : {sandbox_dir.exists()} (ต้องเป็น False)")
        assert not sandbox_dir.exists(), "❌ ล้มเหลว: โฟลเดอร์ใน /tmp ไม่ยอมลบตัวเอง!"
        print("✅ การันตีความสะอาด: Sandbox ถูกทำลายทิ้งและคืนพื้นที่สมบูรณ์แบบ!")

        print("\n" + "=" * 70)
        print("🏆 ผ่านการทดสอบ Git Worktrees Sandbox สมบูรณ์แบบ 100%!")
        print("🚀 คุณพร้อมเข้าสู่ Day 23: Autonomous Branching & Auto-Rollback แล้ว!")
        print("=" * 70)

    except Exception as err:
        print(f"❌ เกิดข้อผิดพลาดในการทดสอบ Sandbox: {err}")

if __name__ == "__main__":
    main()