Local SE Agent Masterclass

import subprocess
from pathlib import Path
from typing import Callable, Dict, Any
from core.sandbox import GitWorktreeSandbox

class AutonomousExecutor:
    """
    เอนจินรันงานอัตโนมัติพร้อมระบบตรวจสอบ (Verification Loop) และ Auto-Rollback
    """
    def __init__(self, repo_root: str):
        self.repo_root = repo_root

    def execute_with_verification(
        self,
        task_name: str,
        mutation_callback: Callable[[Path], None],
        verification_command: list
    ) -> Dict[str, Any]:
        """
        [Verification Loop & Auto-Rollback Flow]:
        1. Spawn Sandbox Worktree
        2. ให้ Callback เข้าไปแก้ไขโค้ด
        3. รันคำสั่งตรวจสอบ (เช่น mvn test)
        4. ถ้าผ่าน คงไว้ / ถ้าพัง ทำ Auto-Rollback ลบทำลายทิ้งทันที
        """
        print(f"🔄 [Autonomous Execution] เริ่มต้นภารกิจ: '{task_name}'")

        with GitWorktreeSandbox(self.repo_root) as sandbox_dir:
            try:
                # 1. จำลองการแก้ไขโค้ดจาก Agent
                mutation_callback(sandbox_dir)

                # 2. รันคำสั่งตรวจสอบผลลัพธ์ (เช่น Maven Test)
                print(f"⚡ [Verification] กำลังรันคำสั่งตรวจสอบ: {' '.join(verification_command)}")
                result = subprocess.run(
                    verification_command,
                    cwd=str(sandbox_dir),
                    capture_output=True,
                    text=True,
                    timeout=120
                )

                if result.returncode == 0:
                    print("✅ [Verification Success] ผ่านเกณฑ์การทดสอบ 100%!")
                    return {
                        "status": "success",
                        "message": "การแก้ไขสำเร็จและผ่านการทดสอบ",
                        "sandbox_path": str(sandbox_dir)
                    }
                else:
                    print("❌ [Verification Failed] ผลการทดสอบไม่ผ่านเกณฑ์ กำลังสั่ง Auto-Rollback...")
                    return {
                        "status": "rolled_back",
                        "message": "การทดสอบไม่ผ่าน ระบบได้ทำ Auto-Rollback ทำความสะอาดเรียบร้อย",
                        "error_log": result.stdout[-500:]
                    }

            except Exception as err:
                print(f"❌ [Error] เกิดข้อผิดพลาดระหว่างรัน Sandbox: {err}")
                return {
                    "status": "rolled_back",
                    "message": f"เกิดข้อผิดพลาด: {str(err)}"
                }

from pathlib import Path
from core.verification_loop import AutonomousExecutor

def sample_mutation(sandbox_dir: Path):
    """จำลองการแก้ไขโค้ดของ Agent"""
    target = sandbox_dir / "experiment.txt"
    target.write_text("Test update from agent", encoding="utf-8")

def main():
    print("=" * 70)
    print("🚀 เริ่มทดสอบ Day 23: Autonomous Branching, Verification Loops & Rollback")
    print("=" * 70)

    repo_root = Path.cwd().resolve()
    executor = AutonomousExecutor(str(repo_root))

    # จำลองรันคำสั่งตรวจสอบ (เช่น เช็คว่าไฟล์ถูกสร้างจริงไหม)
    verification_cmd = ["python", "-c", "import pathlib; assert pathlib.Path('experiment.txt').exists()"]

    res = executor.execute_with_verification(
        task_name="Write experimental file check",
        mutation_callback=sample_mutation,
        verification_command=verification_cmd
    )

    print(f"\n📊 ผลลัพธ์สุดท้าย: {res}")
    print("=" * 70)
    print("🏆 ผ่านการทดสอบ Verification Loop & Auto-Rollback สมบูรณ์แบบ!")
    print("=" * 70)

if __name__ == "__main__":
    main()