ai-mini-box-telegram 0.1.0__tar.gz

ai_mini_box_telegram-0.1.0/.gitignore

@@ -0,0 +1,15 @@
+# TAUSIK
+.tausik/
+
+__pycache__/
+*.pyc
+*.pyo
+*.egg-info/
+dist/
+build/
+.env
+*.db
+migrations/versions/
+!migrations/versions/.gitkeep
+data/
+.DS_Store

ai_mini_box_telegram-0.1.0/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Kibertum
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

ai_mini_box_telegram-0.1.0/PKG-INFO

@@ -0,0 +1,94 @@
+Metadata-Version: 2.4
+Name: ai-mini-box-telegram
+Version: 0.1.0
+Summary: Telegram integration for ai-mini-box — poll/daemon commands via Bot API
+Project-URL: Homepage, https://github.com/Kibertum/ai-mini-box
+Project-URL: Repository, https://github.com/Kibertum/ai-mini-box
+License: MIT License
+        
+        Copyright (c) 2026 Kibertum
+        
+        Permission is hereby granted, free of charge, to any person obtaining a copy
+        of this software and associated documentation files (the "Software"), to deal
+        in the Software without restriction, including without limitation the rights
+        to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+        copies of the Software, and to permit persons to whom the Software is
+        furnished to do so, subject to the following conditions:
+        
+        The above copyright notice and this permission notice shall be included in all
+        copies or substantial portions of the Software.
+        
+        THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+        IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+        FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+        AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+        LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+        OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+        SOFTWARE.
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Developers
+Classifier: License :: OSI Approved :: MIT License
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Topic :: Communications :: Chat
+Requires-Python: >=3.12
+Requires-Dist: ai-mini-box-core>=5.0.0
+Requires-Dist: requests>=2.31
+Provides-Extra: dev
+Requires-Dist: pytest-mock>=3; extra == 'dev'
+Requires-Dist: pytest>=8; extra == 'dev'
+Description-Content-Type: text/markdown
+
+# ai-mini-box-telegram
+
+Telegram Bot API integration for ai-mini-box. Receives messages via long polling and saves them to the database.
+
+## Installation
+
+```bash
+pip install ai-mini-box-telegram
+```
+
+Requires `ai-mini-box-core>=5.0.0`.
+
+## Setup
+
+Set your Telegram bot token (get one from [@BotFather](https://t.me/botfather)):
+
+```bash
+ai-mini-box config set telegram_token "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
+```
+
+Optionally restrict which chats the bot processes:
+
+```bash
+ai-mini-box config set telegram_allowed_chat_ids "[123456789, 987654321]"
+```
+
+## Usage
+
+### Poll once
+
+Fetch new messages and save them:
+
+```bash
+ai-mini-box telegram poll
+# → Processed 3 new messages
+```
+
+### Run daemon
+
+Continuous polling loop (Ctrl+C to stop):
+
+```bash
+ai-mini-box telegram daemon
+```
+
+Configuration fields used:
+
+| Field | Default | Description |
+|---|---|---|
+| `telegram_token` | — | Bot token (required, encrypted) |
+| `telegram_allowed_chat_ids` | `[]` | Restrict to specific chats (empty = all) |
+| `poll_interval` | `30` | Seconds between polls in daemon mode |

ai_mini_box_telegram-0.1.0/README.md

@@ -0,0 +1,52 @@
+# ai-mini-box-telegram
+
+Telegram Bot API integration for ai-mini-box. Receives messages via long polling and saves them to the database.
+
+## Installation
+
+```bash
+pip install ai-mini-box-telegram
+```
+
+Requires `ai-mini-box-core>=5.0.0`.
+
+## Setup
+
+Set your Telegram bot token (get one from [@BotFather](https://t.me/botfather)):
+
+```bash
+ai-mini-box config set telegram_token "123456:ABC-DEF1234ghIkl-zyx57W2v1u123ew11"
+```
+
+Optionally restrict which chats the bot processes:
+
+```bash
+ai-mini-box config set telegram_allowed_chat_ids "[123456789, 987654321]"
+```
+
+## Usage
+
+### Poll once
+
+Fetch new messages and save them:
+
+```bash
+ai-mini-box telegram poll
+# → Processed 3 new messages
+```
+
+### Run daemon
+
+Continuous polling loop (Ctrl+C to stop):
+
+```bash
+ai-mini-box telegram daemon
+```
+
+Configuration fields used:
+
+| Field | Default | Description |
+|---|---|---|
+| `telegram_token` | — | Bot token (required, encrypted) |
+| `telegram_allowed_chat_ids` | `[]` | Restrict to specific chats (empty = all) |
+| `poll_interval` | `30` | Seconds between polls in daemon mode |

ai_mini_box_telegram-0.1.0/ai_mini_box_telegram/bot.py

@@ -0,0 +1,36 @@
+import requests
+
+from .exceptions import TelegramAPIError
+
+
+class TelegramBot:
+    BASE = "https://api.telegram.org/bot"
+
+    def __init__(self, token: str):
+        self.token = token
+
+    def _url(self, method: str) -> str:
+        return f"{self.BASE}{self.token}/{method}"
+
+    def get_updates(self, offset: int | None = None, timeout: int = 10) -> list[dict]:
+        url = self._url("getUpdates")
+        params: dict = {"timeout": timeout}
+        if offset is not None:
+            params["offset"] = offset
+        try:
+            resp = requests.get(url, params=params, timeout=timeout + 5)
+            resp.raise_for_status()
+            data = resp.json()
+        except requests.exceptions.RequestException as e:
+            raise TelegramAPIError(str(e), status_code=getattr(e.response, "status_code", 0)) from e
+        if not data.get("ok"):
+            raise TelegramAPIError(data.get("description", "Unknown error"), status_code=0)
+        return data["result"]
+
+    def send_message(self, chat_id: int, text: str) -> bool:
+        url = self._url("sendMessage")
+        try:
+            resp = requests.post(url, json={"chat_id": chat_id, "text": text}, timeout=10)
+            return resp.ok
+        except requests.exceptions.RequestException:
+            return False

ai_mini_box_telegram-0.1.0/ai_mini_box_telegram/commands.py

@@ -0,0 +1,92 @@
+import signal
+import sys
+import time
+
+import typer
+from loguru import logger
+
+from ai_mini_box.infrastructure.config import JsonConfigManager
+from ai_mini_box.infrastructure.database import get_db
+
+from .bot import TelegramBot
+from .exceptions import TelegramAPIError
+from .handlers import process_update
+from .state import MemoryTelegramStateRepo
+
+logger.add("logs/plugin_telegram.log", rotation="1 MB", retention=3)
+
+
+def _resolve_config() -> tuple[str, list[int], int]:
+    config = JsonConfigManager().load()
+    token = config.telegram_token
+    if not token:
+        typer.echo("Error: telegram_token not set. Run: ai-mini-box config set telegram_token <token>")
+        raise typer.Exit(1)
+    return token, config.telegram_allowed_chat_ids, config.poll_interval
+
+
+def _do_poll(bot: TelegramBot, state: MemoryTelegramStateRepo, allowed_chat_ids: list[int]) -> int:
+    offset = state.get_offset()
+    updates = bot.get_updates(offset=offset)
+    count = 0
+    for update in updates:
+        with get_db() as session:
+            if process_update(update, session, allowed_chat_ids=allowed_chat_ids):
+                count += 1
+                state.save_offset(update["update_id"] + 1)
+    return count
+
+
+def register(app: typer.Typer):
+    tg = typer.Typer(help="Telegram bot integration")
+    app.add_typer(tg, name="telegram")
+
+    @tg.command()
+    def poll():
+        """Poll Telegram for new messages once and save them."""
+        token, allowed_chat_ids, _ = _resolve_config()
+        bot = TelegramBot(token)
+        state = MemoryTelegramStateRepo()
+        try:
+            count = _do_poll(bot, state, allowed_chat_ids)
+        except TelegramAPIError as e:
+            typer.echo(f"Error: {e}")
+            raise typer.Exit(1)
+        typer.echo(f"Processed {count} new messages")
+
+    @tg.command()
+    def daemon():
+        """Run continuous polling loop until interrupted."""
+        token, allowed_chat_ids, interval = _resolve_config()
+        bot = TelegramBot(token)
+        state = MemoryTelegramStateRepo()
+        logger.info(f"Starting telegram daemon (poll interval={interval}s)")
+
+        stop = False
+
+        def _signal_handler(signum, frame):
+            nonlocal stop
+            stop = True
+            logger.info("Shutdown requested, finishing current cycle...")
+
+        signal.signal(signal.SIGINT, _signal_handler)
+        signal.signal(signal.SIGTERM, _signal_handler)
+
+        while not stop:
+            logger.info(f"Polling... (interval={interval}s)")
+            try:
+                count = _do_poll(bot, state, allowed_chat_ids)
+                if count:
+                    logger.info(f"Processed {count} new messages")
+            except TelegramAPIError as e:
+                logger.error(f"Polling error: {e}")
+            except Exception as e:
+                logger.error(f"Unexpected polling error: {e}")
+            if stop:
+                break
+            try:
+                time.sleep(interval)
+            except KeyboardInterrupt:
+                stop = True
+
+        logger.info("Telegram daemon stopped")

ai_mini_box_telegram-0.1.0/ai_mini_box_telegram/exceptions.py

@@ -0,0 +1,5 @@
+class TelegramAPIError(Exception):
+    def __init__(self, description: str, status_code: int = 0):
+        self.description = description
+        self.status_code = status_code
+        super().__init__(f"Telegram API error ({status_code}): {description}")

ai_mini_box_telegram-0.1.0/ai_mini_box_telegram/handlers.py

@@ -0,0 +1,48 @@
+from sqlalchemy.orm import Session
+
+from ai_mini_box.core.container import RepoContainer
+from ai_mini_box.core.models import Contact, Message, MessageSource, Topic
+
+
+def process_update(
+    update: dict,
+    session: Session,
+    allowed_chat_ids: list[int] | None = None,
+) -> bool:
+    if "message" not in update:
+        return False
+
+    message_data = update["message"]
+    chat_id = message_data["chat"]["id"]
+
+    if allowed_chat_ids and chat_id not in allowed_chat_ids:
+        return False
+
+    update_id = update["update_id"]
+    text = message_data.get("text") or message_data.get("caption", "")
+    repos = RepoContainer(session)
+
+    contacts = repos.contacts.list(telegram=str(chat_id), limit=1)
+    if contacts:
+        contact = contacts[0]
+    else:
+        contact = repos.contacts.add(
+            Contact(
+                name=str(chat_id),
+                telegram=str(chat_id),
+                source=MessageSource.TELEGRAM,
+            )
+        )
+
+    repos.messages.add(
+        Message(
+            source=MessageSource.TELEGRAM,
+            external_id=str(update_id),
+            chat_id=str(chat_id),
+            contact_id=contact.id,
+            text=text,
+            topic=Topic.OTHER,
+        )
+    )
+
+    return True

ai_mini_box_telegram-0.1.0/ai_mini_box_telegram/state.py

@@ -0,0 +1,22 @@
+from abc import ABC, abstractmethod
+
+
+class TelegramStateRepo(ABC):
+    @abstractmethod
+    def get_offset(self) -> int | None:
+        ...
+
+    @abstractmethod
+    def save_offset(self, offset: int) -> None:
+        ...
+
+
+class MemoryTelegramStateRepo(TelegramStateRepo):
+    def __init__(self):
+        self._offset: int | None = None
+
+    def get_offset(self) -> int | None:
+        return self._offset
+
+    def save_offset(self, offset: int) -> None:
+        self._offset = offset

ai_mini_box_telegram-0.1.0/pyproject.toml

@@ -0,0 +1,36 @@
+[build-system]
+requires = ["hatchling"]
+build-backend = "hatchling.build"
+
+[project]
+name = "ai-mini-box-telegram"
+version = "0.1.0"
+description = "Telegram integration for ai-mini-box — poll/daemon commands via Bot API"
+requires-python = ">=3.12"
+readme = "README.md"
+license = { file = "LICENSE" }
+classifiers = [
+    "Development Status :: 4 - Beta",
+    "Intended Audience :: Developers",
+    "License :: OSI Approved :: MIT License",
+    "Programming Language :: Python :: 3.12",
+    "Programming Language :: Python :: 3.13",
+    "Topic :: Communications :: Chat",
+]
+dependencies = [
+    "ai-mini-box-core>=5.0.0",
+    "requests>=2.31",
+]
+
+[project.optional-dependencies]
+dev = ["pytest>=8", "pytest-mock>=3"]
+
+[project.urls]
+Homepage = "https://github.com/Kibertum/ai-mini-box"
+Repository = "https://github.com/Kibertum/ai-mini-box"
+
+[project.entry-points."ai_mini_box.tools"]
+telegram = "ai_mini_box_telegram.commands:register"
+
+[tool.hatch.build]
+include = ["ai_mini_box_telegram/**"]