ChingTech OS：多平台 Bot Adapter 重構

from typing import Any, Protocol, runtime_checkable
from dataclasses import dataclass


@dataclass
class SentMessage:
    """已發送訊息的結果"""
    message_id: str
    platform_type: str


@runtime_checkable
class BotAdapter(Protocol):
    """所有平台必須實作的標準化介面"""

    platform_type: str

    async def send_text(
        self,
        target: str,
        text: str,
        *,
        reply_to: str | None = None,
        mention_user_id: str | None = None,
    ) -> SentMessage: ...

    async def send_image(
        self,
        target: str,
        image_url: str,
        *,
        reply_to: str | None = None,
        preview_url: str | None = None,
    ) -> SentMessage: ...

    async def send_file(
        self,
        target: str,
        file_url: str,
        file_name: str,
        *,
        reply_to: str | None = None,
        file_size: str | None = None,
    ) -> SentMessage: ...

    async def send_messages(
        self,
        target: str,
        messages: list[Any],
        *,
        reply_to: str | None = None,
    ) -> list[SentMessage]: ...

@runtime_checkable
class EditableMessageAdapter(Protocol):
    """可選：支援訊息編輯/刪除的平台（如 Telegram）"""

    async def edit_message(
        self, target: str, message_id: str, new_text: str
    ) -> None: ...

    async def delete_message(
        self, target: str, message_id: str
    ) -> None: ...


@runtime_checkable
class ProgressNotifier(Protocol):
    """可選：支援即時進度更新的平台"""

    async def send_progress(self, target: str, text: str) -> SentMessage: ...
    async def update_progress(self, target: str, message_id: str, text: str) -> None: ...
    async def finish_progress(self, target: str, message_id: str) -> None: ...

if isinstance(adapter, ProgressNotifier):
    progress = await adapter.send_progress(target, "AI 處理中...")
    # ... AI 處理 ...
    await adapter.finish_progress(target, progress.message_id)

class PlatformType(str, Enum):
    LINE = "line"
    TELEGRAM = "telegram"

class ConversationType(str, Enum):
    PRIVATE = "private"
    GROUP = "group"

@dataclass
class BotMessage:
    """入站訊息的正規化格式"""
    platform_type: PlatformType
    sender_id: str
    target_id: str
    text: str | None = None
    media_url: str | None = None
    conversation_type: ConversationType = ConversationType.PRIVATE
    reply_to_message_id: str | None = None
    platform_data: dict = field(default_factory=dict)

@dataclass
class BotContext:
    """對話情境（AI 處理所需的所有資訊）"""
    platform_type: PlatformType
    conversation_type: ConversationType
    user_uuid: UUID | None = None
    group_uuid: UUID | None = None
    platform_user_id: str | None = None
    platform_group_id: str | None = None
    user_display_name: str | None = None
    ctos_user_id: int | None = None
    message_uuid: UUID | None = None
    platform_data: dict = field(default_factory=dict)

def parse_ai_response(response: str) -> tuple[str, list[dict]]:
    """解析 AI 回應，提取文字和檔案訊息"""
    pattern = r'\[FILE_MESSAGE:(\{.*?\})\]'
    files = []
    for match in re.finditer(pattern, response):
        file_info = json.loads(match.group(1))
        files.append(file_info)
    text = re.sub(pattern, '', response).strip()
    return text, files

class LineBotAdapter:
    platform_type: str = "line"

    async def send_text(self, target, text, *, reply_to=None, mention_user_id=None):
        from .messaging import push_text, create_text_message_with_mention, push_messages

        if mention_user_id:
            msg = create_text_message_with_mention(text, mention_user_id)
            sent_ids, error = await push_messages(target, [msg])
            msg_id = sent_ids[0] if sent_ids else ""
        else:
            msg_id, error = await push_text(target, text)

        if error:
            return SentMessage(message_id="", platform_type="line")
        return SentMessage(message_id=msg_id or "", platform_type="line")

    async def reply_text(self, reply_token: str, text: str) -> SentMessage:
        """使用 reply token 回覆文字（Line 專屬，非 BotAdapter 介面）"""
        ...

    async def reply_messages(self, reply_token: str, messages: list) -> list[SentMessage]:
        """使用 reply token 回覆多則訊息（Line 專屬，非 BotAdapter 介面）"""
        ...

class TelegramBotAdapter:
    """同時實作 BotAdapter / EditableMessageAdapter / ProgressNotifier"""

    platform_type: str = "telegram"

    def __init__(self, token: str):
        self.bot = Bot(token=token)

    # === BotAdapter ===
    async def send_text(self, target, text, *, reply_to=None, mention_user_id=None):
        kwargs = {"chat_id": target, "text": text}
        if reply_to:
            kwargs["reply_to_message_id"] = int(reply_to)
        msg = await self.bot.send_message(**kwargs)
        return SentMessage(message_id=str(msg.message_id), platform_type="telegram")

    async def send_file(self, target, file_url, file_name, *, reply_to=None, file_size=None):
        """先下載檔案到記憶體，再上傳給 Telegram"""
        async with httpx.AsyncClient(follow_redirects=True, timeout=60) as client:
            resp = await client.get(file_url)
        buf = BytesIO(resp.content)
        buf.name = file_name
        msg = await self.bot.send_document(
            chat_id=target, document=InputFile(buf, filename=file_name)
        )
        return SentMessage(message_id=str(msg.message_id), platform_type="telegram")

    # === EditableMessageAdapter ===
    async def edit_message(self, target, message_id, new_text):
        await self.bot.edit_message_text(
            chat_id=target, message_id=int(message_id), text=new_text
        )

    # === ProgressNotifier ===
    async def send_progress(self, target, text):
        return await self.send_text(target, text)

    async def update_progress(self, target, message_id, text):
        await self.edit_message(target, message_id, text)

    async def finish_progress(self, target, message_id):
        try:
            await self.delete_message(target, message_id)
        except Exception:
            pass  # 可能已過期

# main.py
app.include_router(linebot_router.router, prefix="/api/bot")          # 通用
app.include_router(linebot_router.line_router, prefix="/api/bot/line") # Line Webhook
app.include_router(telegram_router.router, prefix="/api/bot/telegram") # Telegram Webhook

@router.get("/groups")
async def api_list_groups(
    platform_type: str | None = Query(None, description="平台類型過濾（line, telegram）"),
    ...
):
    items, total = await list_groups(platform_type=platform_type, ...)

router = APIRouter(prefix="/api/admin/bot-settings", tags=["Bot Settings"])

@router.get("/{platform}")        # 取得設定狀態（遮罩顯示）
@router.put("/{platform}")        # 更新憑證
@router.delete("/{platform}")     # 清除憑證
@router.post("/{platform}/test")  # 測試連線

class UpdateBotSettingsRequest(BaseModel):
    """通用，允許任意欄位組合"""
    channel_secret: str | None = None
    channel_access_token: str | None = None
    bot_token: str | None = None
    webhook_secret: str | None = None
    admin_chat_id: str | None = None

async def _test_line_connection(credentials):
    resp = await client.get("https://api.line.me/v2/bot/info", ...)

async def _test_telegram_connection(credentials):
    resp = await client.get(f"https://api.telegram.org/bot{token}/getMe", ...)

from ching_tech_os.services.bot.adapter import BotAdapter
from ching_tech_os.services.bot_line.adapter import LineBotAdapter

class TestLineBotAdapterProtocol:
    def test_is_bot_adapter(self):
        """LineBotAdapter 應該符合 BotAdapter Protocol"""
        adapter = LineBotAdapter()
        assert isinstance(adapter, BotAdapter)

    def test_platform_type(self):
        adapter = LineBotAdapter()
        assert adapter.platform_type == "line"