Local SE Agent Masterclass

import subprocess
import shutil
from pathlib import Path
from typing import Optional, Dict, Any

class BuildExecutionError(Exception):
    """Custom exception สำหรับข้อผิดพลาดที่ไม่สามารถรัน Build Process ได้"""
    pass

def run_maven_test(project_dir: str = ".", test_name: Optional[str] = None, timeout_seconds: int = 120) -> str:
    """
    เครื่องมือสำหรับสั่งรัน Unit Tests ของโปรเจกต์ Java ผ่าน Maven
    
    :param project_dir: ไดเรกทอรีของโปรเจกต์ที่มีไฟล์ pom.xml
    :param test_name: (ทางเลือก) ระบุชื่อคลาสเทสเฉพาะเจาะจง เช่น 'CalculatorTest'
    :param timeout_seconds: เวลาสูงสุดที่อนุญาตให้คำสั่งรัน ป้องกันการติด Infinite Loop
    :return: รายงานผลลัพธ์การรันเทส (BUILD SUCCESS หรือ BUILD FAILURE พร้อม Stack Trace)
    """
    # 1. แปลง project_dir เป็น Absolute Path และตรวจสอบไฟล์ pom.xml
    target_path = Path(project_dir).resolve()
    pom_file = target_path / "pom.xml"

    if not pom_file.exists():
        raise BuildExecutionError(f"ไม่พบไฟล์ 'pom.xml' ในโฟลเดอร์: {target_path}")

    # 2. ตรวจสอบว่าในเครื่อง Mac มีคำสั่ง mvn ติดตั้งอยู่หรือไม่
    mvn_bin = shutil.which("mvn")
    if not mvn_bin:
        # Fallback: ตรวจสอบว่ามี Maven Wrapper (mvnw) อยู่ในโปรเจกต์หรือไม่
        mvnw = target_path / "mvnw"
        if mvnw.exists():
            mvn_cmd = [str(mvnw)]
        else:
            raise BuildExecutionError("ไม่พบคำสั่ง 'mvn' หรือ 'mvnw' ในระบบ โปรดติดตั้ง Maven ก่อนใช้งาน")
    else:
        mvn_cmd = [mvn_bin]

    # 3. จัดเตรียมคำสั่ง Command ในรูปแบบ List
    # -B: Batch mode (ไม่แสดงสีหรือข้อความ interactive)
    # -Dtest=... : ระบุชื่อเทสเจาะจงถ้ามี
    cmd = mvn_cmd + ["test", "-B"]
    if test_name:
        cmd.append(f"-Dtest={test_name}")

    try:
        # 4. ยิง Process ผ่าน subprocess.run
        process = subprocess.run(
            cmd,
            cwd=str(target_path),
            capture_output=True,
            text=True,
            timeout=timeout_seconds,
            encoding="utf-8",
            errors="replace"
        )

        stdout = process.stdout or ""
        stderr = process.stderr or ""
        full_output = stdout + "\n" + stderr

        # 5. วิเคราะห์ Exit Code (0 = สำเร็จ, ไม่ใช่ 0 = ล้มเหลว)
        if process.returncode == 0:
            return (
                "✅ BUILD SUCCESS: การรันชุดทดสอบทั้งหมดผ่านเรียบร้อย 100%!\n"
                f"รายละเอียดผลลัพธ์:\n{_extract_summary_lines(stdout, 15)}"
            )
        else:
            # สกัดเฉพาะบรรทัดล่างสุดที่สรุป Failure เพื่อไม่ให้กิน Context Window
            failure_summary = _extract_summary_lines(full_output, 35)
            return (
                f"❌ BUILD FAILURE: ตรวจพบข้อผิดพลาดในชุดทดสอบ (Exit Code {process.returncode})\n"
                "--------------------------------------------------\n"
                f"{failure_summary}\n"
                "--------------------------------------------------"
            )

    except subprocess.TimeoutExpired:
        raise BuildExecutionError(f"Timeout Error: คำสั่งรันเทสใช้เวลานานเกินกว่า {timeout_seconds} วินาที")
    except Exception as err:
        raise BuildExecutionError(f"เกิดข้อผิดพลาดขณะรันกระบวนการ Maven: {str(err)}")

def _extract_summary_lines(output: str, max_lines: int = 30) -> str:
    """ฟังก์ชันช่วยตัดทอน Log ยาวๆ ให้เหลือเฉพาะส่วนท้ายที่สำคัญ"""
    lines = [line.strip() for line in output.splitlines() if line.strip()]
    if len(lines) <= max_lines:
        return "\n".join(lines)
    return "[...ตัดทอนบันทึกตอนต้นออก...]\n" + "\n".join(lines[-max_lines:])

<project xmlns="http://maven.apache.org/POM/4.0.0"
         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
    <modelVersion>4.0.0</modelVersion>
    <groupId>com.example</groupId>
    <artifactId>sample-maven-project</artifactId>
    <version>1.0.0</version>
    <properties>
        <maven.compiler.source>17</maven.compiler.source>
        <maven.compiler.target>17</maven.compiler.target>
        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
    </properties>
    <dependencies>
        <dependency>
            <groupId>org.junit.jupiter</groupId>
            <artifactId>junit-jupiter</artifactId>
            <version>5.10.0</version>
            <scope>test</scope>
        </dependency>
    </dependencies>
</project>

// 3.1: src/main/java/Calculator.java (มีบั๊ก: คำนวณภาษีผิดตัวเลข)
public class Calculator {
    public int multiply(int a, int b) {
        return a * b;
    }
    public double applyDiscount(double price, double discount) {
        // บั๊กจำลอง: เขียนผิดเป็นลบเงินสองเท่า
        return price - (discount * 2.0);
    }
}

// 3.2: src/test/java/CalculatorTest.java
import org.junit.jupiter.api.Test;
import static org.junit.jupiter.api.Assertions.assertEquals;

public class CalculatorTest {
    @Test
    void testMultiply() {
        Calculator calc = new Calculator();
        assertEquals(20, calc.multiply(4, 5));
    }

    @Test
    void testApplyDiscount() {
        Calculator calc = new Calculator();
        // คาดหวัง: 100 - 10 = 90.0 แต่โค้ดส่งคืน 80.0 -> เกิด AssertionFailedError!
        assertEquals(90.0, calc.applyDiscount(100.0, 10.0), 0.001);
    }
}

from core.client import OllamaLLMClient
from core.parser import StructuredOutputParser
from core.dispatcher import ToolDispatcher
from core.tools.test_runner import run_maven_test, BuildExecutionError

def main():
    print("🚀 เริ่มทดสอบ Day 5: Maven Build & Test Runner Tool บน Mac M3...")
    
    # 1. ติดตั้ง Dispatcher และจดทะเบียน Tool ทั้งสองตัว (File Reader + Maven Runner)
    dispatcher = ToolDispatcher()
    dispatcher.register("run_maven_test", run_maven_test)

    client = OllamaLLMClient(model_name="qwen2.5-coder:7b")
    if not client.check_health():
        print("❌ ข้อผิดพลาด: ไม่พบ Ollama Daemon หรือโมเดล qwen2.5-coder:7b")
        return

    # 2. System Prompt ประกาศเครื่องมือรันเทส
    system_prompt = (
        "You are an Autonomous Software Engineering Agent.\n"
        "You have tools to build and run automated tests for Java Maven projects.\n"
        "When you need to test code, respond ONLY with a JSON action object.\n\n"
        "Available Tools:\n"
        "1. run_maven_test(project_dir: string, test_name: string|null): Runs Maven unit tests.\n\n"
        "JSON Response Schema:\n"
        "{\n"
        '  "tool": "run_maven_test",\n'
        '  "arguments": {\n'
        '    "project_dir": "path/to/project"\n'
        '  }\n'
        "}"
    )

    user_query = "ช่วยรันเทสในโปรเจกต์ sample_maven_project ให้หน่อยว่าผ่านไหม และเกิดปัญหาอะไรขึ้น"
    print(f"💬 User Prompt: {user_query}")

    messages = [
        {"role": "system", "content": system_prompt},
        {"role": "user", "content": user_query}
    ]

    # 3. Turn 1: ให้โมเดลขอเรียก Tool
    print("\n📡 กำลังส่งโจทย์ให้โมเดลประมวลผล...")
    raw_response = client.chat(messages, temperature=0.1, json_mode=True)
    print(f"📥 โมเดลตัดสินใจส่ง JSON Action:\n{raw_response}")

    try:
        action = StructuredOutputParser.parse(raw_response)
        tool_name = action.get("tool")
        arguments = action.get("arguments", {})

        print(f"\n⚙️ Dispatcher กำลังสั่งรัน Process '{tool_name}'...")
        # 4. Dispatcher รันคำสั่ง mvn test จริงผ่าน Terminal
        test_output = dispatcher.dispatch(tool_name, arguments)
        print(f"✅ บันทึกผลลัพธ์จาก Process:\n{test_output}")

        # 5. Turn 2: ส่งผลลัพธ์ Observation กลับไปให้โมเดลวิเคราะห์สาเหตุและแนวทางแก้ไข
        messages.append({"role": "assistant", "content": raw_response})
        messages.append({
            "role": "user",
            "content": f"Observation from run_maven_test:\n```text\n{test_output}\n```\nโปรดวิเคราะห์ว่าทำไมเทสถึงล้มเหลว คาดหวังค่าอะไรและได้ค่าอะไรกลับมา"
        })

        print("\n🧠 โมเดลกำลังอ่าน Stack Trace และสังเคราะห์คำตอบ...")
        final_answer = client.chat(messages, temperature=0.2, json_mode=False)
        print("\n🎯 สรุปผลการวิเคราะห์จาก AI Agent:")
        print("--------------------------------------------------")
        print(final_answer)
        print("--------------------------------------------------")
        print("🎉 สำเร็จ 100%! Agent ของคุณสามารถสั่งรันเทสจริงและจับบั๊กในระดับ OS Process ได้แล้ว!")

    except BuildExecutionError as err:
        print(f"❌ Build Execution Error: {err}")
    except Exception as err:
        print(f"❌ Error: {err}")

if __name__ == "__main__":
    main()