yee88 0.1.0__py3-none-any.whl

takopi/telegram/api_models.py

@@ -0,0 +1,37 @@
+from __future__ import annotations
+
+from .api_schemas import (
+    CallbackQuery,
+    CallbackQueryMessage,
+    Chat,
+    ChatMember,
+    Document,
+    File,
+    ForumTopic,
+    Message,
+    MessageReply,
+    PhotoSize,
+    Sticker,
+    Update,
+    User,
+    Video,
+    Voice,
+)
+
+__all__ = [
+    "CallbackQuery",
+    "CallbackQueryMessage",
+    "Chat",
+    "ChatMember",
+    "Document",
+    "File",
+    "ForumTopic",
+    "Message",
+    "MessageReply",
+    "PhotoSize",
+    "Sticker",
+    "Update",
+    "User",
+    "Video",
+    "Voice",
+]

takopi/telegram/api_schemas.py

@@ -0,0 +1,152 @@
+"""Msgspec models for Telegram Bot API payloads (subset used by takopi).
+
+Derived from telegram-api.html in the repository.
+"""
+
+from __future__ import annotations
+
+import msgspec
+
+__all__ = [
+    "CallbackQuery",
+    "CallbackQueryMessage",
+    "Chat",
+    "ChatMember",
+    "Document",
+    "File",
+    "ForumTopic",
+    "Message",
+    "MessageReply",
+    "PhotoSize",
+    "Sticker",
+    "Update",
+    "User",
+    "Video",
+    "Voice",
+    "decode_update",
+    "decode_updates",
+]
+
+
+class User(msgspec.Struct, forbid_unknown_fields=False):
+    id: int
+    is_bot: bool | None = None
+    username: str | None = None
+    first_name: str | None = None
+    last_name: str | None = None
+
+
+class Chat(msgspec.Struct, forbid_unknown_fields=False):
+    id: int
+    type: str
+    title: str | None = None
+    username: str | None = None
+    first_name: str | None = None
+    last_name: str | None = None
+    is_forum: bool | None = None
+
+
+class PhotoSize(msgspec.Struct, forbid_unknown_fields=False):
+    file_id: str
+    width: int
+    height: int
+    file_size: int | None = None
+
+
+class Document(msgspec.Struct, forbid_unknown_fields=False):
+    file_id: str
+    file_name: str | None = None
+    mime_type: str | None = None
+    file_size: int | None = None
+
+
+class Video(msgspec.Struct, forbid_unknown_fields=False):
+    file_id: str
+    file_name: str | None = None
+    mime_type: str | None = None
+    file_size: int | None = None
+
+
+class Voice(msgspec.Struct, forbid_unknown_fields=False):
+    file_id: str
+    duration: int | None = None
+    mime_type: str | None = None
+    file_size: int | None = None
+
+
+class Sticker(msgspec.Struct, forbid_unknown_fields=False):
+    file_id: str
+    file_size: int | None = None
+
+
+class MessageReply(msgspec.Struct, forbid_unknown_fields=False):
+    message_id: int
+    text: str | None = None
+    from_: User | None = msgspec.field(default=None, name="from")
+
+
+class Message(msgspec.Struct, forbid_unknown_fields=False):
+    message_id: int
+    chat: Chat
+    message_thread_id: int | None = None
+    from_: User | None = msgspec.field(default=None, name="from")
+    text: str | None = None
+    caption: str | None = None
+    reply_to_message: MessageReply | None = None
+    forward_from: User | None = None
+    forward_from_chat: Chat | None = None
+    forward_from_message_id: int | None = None
+    forward_sender_name: str | None = None
+    forward_signature: str | None = None
+    forward_date: int | None = None
+    media_group_id: str | None = None
+    is_automatic_forward: bool | None = None
+    is_topic_message: bool | None = None
+    voice: Voice | None = None
+    document: Document | None = None
+    video: Video | None = None
+    photo: list[PhotoSize] | None = None
+    sticker: Sticker | None = None
+
+
+class CallbackQueryMessage(msgspec.Struct, forbid_unknown_fields=False):
+    message_id: int
+    chat: Chat
+
+
+class CallbackQuery(msgspec.Struct, forbid_unknown_fields=False):
+    id: str
+    from_: User = msgspec.field(name="from")
+    message: CallbackQueryMessage | None = None
+    data: str | None = None
+
+
+class Update(msgspec.Struct, forbid_unknown_fields=False):
+    update_id: int
+    message: Message | None = None
+    callback_query: CallbackQuery | None = None
+
+
+class File(msgspec.Struct, forbid_unknown_fields=False):
+    file_path: str
+
+
+class ChatMember(msgspec.Struct, forbid_unknown_fields=False):
+    status: str
+    can_manage_topics: bool | None = None
+
+
+class ForumTopic(msgspec.Struct, forbid_unknown_fields=False):
+    message_thread_id: int
+
+
+_UPDATE_DECODER = msgspec.json.Decoder(Update)
+_UPDATES_DECODER = msgspec.json.Decoder(list[Update])
+
+
+def decode_update(payload: str | bytes) -> Update:
+    return _UPDATE_DECODER.decode(payload)
+
+
+def decode_updates(payload: str | bytes) -> list[Update]:
+    return _UPDATES_DECODER.decode(payload)

takopi/telegram/backend.py

@@ -0,0 +1,163 @@
+from __future__ import annotations
+
+import os
+from pathlib import Path
+from typing import Literal
+
+import anyio
+
+from ..backends import EngineBackend
+from ..logging import get_logger
+from ..runner_bridge import ExecBridgeConfig
+from ..settings import TelegramTopicsSettings, TelegramTransportSettings, load_settings_if_exists
+from ..transport_runtime import TransportRuntime
+from ..transports import SetupResult, TransportBackend
+from .bridge import (
+    TelegramBridgeConfig,
+    TelegramPresenter,
+    TelegramTransport,
+    run_main_loop,
+)
+from .client import TelegramClient
+from .onboarding import check_setup, interactive_setup
+from .topics import _resolve_topics_scope_raw
+
+logger = get_logger(__name__)
+
+
+def _expect_transport_settings(transport_config: object) -> TelegramTransportSettings:
+    if isinstance(transport_config, TelegramTransportSettings):
+        return transport_config
+    raise TypeError("transport_config must be TelegramTransportSettings")
+
+
+def _build_startup_message(
+    runtime: TransportRuntime,
+    *,
+    startup_pwd: str,
+    chat_id: int,
+    session_mode: Literal["stateless", "chat"],
+    show_resume_line: bool,
+    topics: TelegramTopicsSettings,
+) -> str:
+    available_engines = list(runtime.available_engine_ids())
+    missing_engines = list(runtime.missing_engine_ids())
+    misconfigured_engines = list(runtime.engine_ids_with_status("bad_config"))
+    failed_engines = list(runtime.engine_ids_with_status("load_error"))
+
+    engine_list = ", ".join(available_engines) if available_engines else "none"
+
+    notes: list[str] = []
+    if missing_engines:
+        notes.append(f"not installed: {', '.join(missing_engines)}")
+    if misconfigured_engines:
+        notes.append(f"misconfigured: {', '.join(misconfigured_engines)}")
+    if failed_engines:
+        notes.append(f"failed to load: {', '.join(failed_engines)}")
+    if notes:
+        engine_list = f"{engine_list} ({'; '.join(notes)})"
+    project_aliases = sorted(set(runtime.project_aliases()), key=str.lower)
+    project_list = ", ".join(project_aliases) if project_aliases else "none"
+    resume_label = "shown" if show_resume_line else "hidden"
+    topics_label = "disabled"
+    if topics.enabled:
+        resolved_scope, _ = _resolve_topics_scope_raw(
+            topics.scope, chat_id, runtime.project_chat_ids()
+        )
+        scope_label = (
+            f"auto ({resolved_scope})" if topics.scope == "auto" else resolved_scope
+        )
+        topics_label = f"enabled (scope={scope_label})"
+    return (
+        f"\N{OCTOPUS} **takopi is ready**\n\n"
+        f"default: `{runtime.default_engine}`  \n"
+        f"engines: `{engine_list}`  \n"
+        f"projects: `{project_list}`  \n"
+        f"mode: `{session_mode}`  \n"
+        f"topics: `{topics_label}`  \n"
+        f"resume lines: `{resume_label}`  \n"
+        f"working in: `{startup_pwd}`"
+    )
+
+
+class TelegramBackend(TransportBackend):
+    id = "telegram"
+    description = "Telegram bot"
+
+    def check_setup(
+        self,
+        engine_backend: EngineBackend,
+        *,
+        transport_override: str | None = None,
+    ) -> SetupResult:
+        return check_setup(engine_backend, transport_override=transport_override)
+
+    async def interactive_setup(self, *, force: bool) -> bool:
+        return await interactive_setup(force=force)
+
+    def lock_token(self, *, transport_config: object, _config_path: Path) -> str | None:
+        settings = _expect_transport_settings(transport_config)
+        return settings.bot_token
+
+    def build_and_run(
+        self,
+        *,
+        transport_config: object,
+        config_path: Path,
+        runtime: TransportRuntime,
+        final_notify: bool,
+        default_engine_override: str | None,
+    ) -> None:
+        settings = _expect_transport_settings(transport_config)
+        token = settings.bot_token
+        chat_id = settings.chat_id
+        startup_msg = _build_startup_message(
+            runtime,
+            startup_pwd=os.getcwd(),
+            chat_id=chat_id,
+            session_mode=settings.session_mode,
+            show_resume_line=settings.show_resume_line,
+            topics=settings.topics,
+        )
+        bot = TelegramClient(token)
+        transport = TelegramTransport(bot)
+        presenter = TelegramPresenter(message_overflow=settings.message_overflow)
+        exec_cfg = ExecBridgeConfig(
+            transport=transport,
+            presenter=presenter,
+            final_notify=final_notify,
+        )
+        cfg = TelegramBridgeConfig(
+            bot=bot,
+            runtime=runtime,
+            chat_id=chat_id,
+            startup_msg=startup_msg,
+            exec_cfg=exec_cfg,
+            session_mode=settings.session_mode,
+            show_resume_line=settings.show_resume_line,
+            voice_transcription=settings.voice_transcription,
+            voice_max_bytes=int(settings.voice_max_bytes),
+            voice_transcription_model=settings.voice_transcription_model,
+            voice_transcription_base_url=settings.voice_transcription_base_url,
+            voice_transcription_api_key=settings.voice_transcription_api_key,
+            forward_coalesce_s=settings.forward_coalesce_s,
+            media_group_debounce_s=settings.media_group_debounce_s,
+            allowed_user_ids=tuple(settings.allowed_user_ids),
+            topics=settings.topics,
+            files=settings.files,
+        )
+
+        async def run_loop() -> None:
+            await run_main_loop(
+                cfg,
+                watch_config=runtime.watch_config,
+                default_engine_override=default_engine_override,
+                transport_id=self.id,
+                transport_config=settings,
+            )
+
+        anyio.run(run_loop)
+
+
+telegram_backend = TelegramBackend()
+BACKEND = telegram_backend