init

assistant_bot/telegram_utils.py

@@ -0,0 +1,95 @@
+from html import escape
+from uuid import uuid4
+from zoneinfo import ZoneInfo
+
+from telegram import InlineQueryResultArticle, InputTextMessageContent, Message, Update
+from telegram.constants import ParseMode
+from telegram.ext import ContextTypes
+
+from .ai import AIClient
+from .config import HTML_FORMATS, MAX_TELEGRAM_MESSAGE_LENGTH
+from .speech import SpeechTranscriber
+from .storage import AssistantStorage
+
+
+def make_article(
+    title: str,
+    text: str,
+    parse_mode: str | None = None,
+) -> InlineQueryResultArticle:
+    return InlineQueryResultArticle(
+        id=f"inline_{uuid4()}",
+        title=title,
+        input_message_content=InputTextMessageContent(text, parse_mode=parse_mode),
+    )
+
+
+def build_inline_results(query: str) -> list[InlineQueryResultArticle]:
+    escaped_query = escape(query)
+
+    return [
+        make_article("Caps", query.upper()),
+        *[
+            make_article(title, f"<{tag}>{escaped_query}</{tag}>", ParseMode.HTML)
+            for title, tag in HTML_FORMATS
+        ],
+    ]
+
+
+def get_storage(context: ContextTypes.DEFAULT_TYPE) -> AssistantStorage:
+    return context.application.bot_data["storage"]
+
+
+def get_ai_client(context: ContextTypes.DEFAULT_TYPE) -> AIClient:
+    return context.application.bot_data["ai_client"]
+
+
+def get_tz(context: ContextTypes.DEFAULT_TYPE) -> ZoneInfo:
+    return context.application.bot_data["timezone"]
+
+
+def get_speech_recognizer(context: ContextTypes.DEFAULT_TYPE) -> SpeechTranscriber:
+    return context.application.bot_data["speech_recognizer"]
+
+
+def get_voice_max_duration(context: ContextTypes.DEFAULT_TYPE) -> int:
+    return int(context.application.bot_data["voice_max_duration"])
+
+
+def command_text(context: ContextTypes.DEFAULT_TYPE) -> str:
+    return " ".join(context.args).strip()
+
+
+def require_user_id(update: Update) -> int | None:
+    if update.effective_user:
+        return int(update.effective_user.id)
+    return None
+
+
+async def reply_long(message: Message, text: str) -> None:
+    chunks = [
+        text[index : index + MAX_TELEGRAM_MESSAGE_LENGTH]
+        for index in range(0, len(text), MAX_TELEGRAM_MESSAGE_LENGTH)
+    ] or [""]
+
+    for chunk in chunks:
+        await message.reply_text(chunk)
+
+
+async def edit_or_reply(message: Message, text: str) -> None:
+    chunks = [
+        text[index : index + MAX_TELEGRAM_MESSAGE_LENGTH]
+        for index in range(0, len(text), MAX_TELEGRAM_MESSAGE_LENGTH)
+    ] or [""]
+
+    await message.edit_text(chunks[0])
+    for chunk in chunks[1:]:
+        await message.reply_text(chunk)
+
+
+def parse_positive_int(value: str) -> int | None:
+    try:
+        parsed = int(value)
+    except ValueError:
+        return None
+    return parsed if parsed > 0 else None