import os
import yaml

ROLES_DIR = "/action/app/prompts/roles"


def load_roles() -> list[dict]:
    """載入所有角色定義"""
    roles = []
    for fname in sorted(os.listdir(ROLES_DIR)):
        if fname.endswith(".yaml"):
            with open(os.path.join(ROLES_DIR, fname), "r", encoding="utf-8") as f:
                roles.append(yaml.safe_load(f))
    return roles


def get_role_intro(roles: list[dict]) -> str:
    """產生角色介紹文字（用於 comment）"""
    lines = ["## 🤖 AI Code Review 團隊", ""]
    for r in roles:
        lines.append(f"- **{r['name']}** ({r['role']}): {r['personality']}")
    return "\n".join(lines)