mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-04 20:05:16 +02:00
feat(gateway): add Telegram adapter and formatting
This commit is contained in:
parent
c9b7d7b572
commit
59e6475348
5 changed files with 298 additions and 0 deletions
114
surfsense_backend/app/gateway/telegram/adapter.py
Normal file
114
surfsense_backend/app/gateway/telegram/adapter.py
Normal file
|
|
@ -0,0 +1,114 @@
|
|||
"""Telegram platform adapter."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from collections.abc import AsyncIterator
|
||||
from typing import Any
|
||||
|
||||
from app.gateway.base.adapter import (
|
||||
BasePlatformAdapter,
|
||||
ParsedInboundEvent,
|
||||
PlatformSendResult,
|
||||
)
|
||||
from app.gateway.telegram.client import TelegramClient
|
||||
|
||||
|
||||
class TelegramAdapter(BasePlatformAdapter):
|
||||
platform = "telegram"
|
||||
|
||||
def __init__(self, token: str) -> None:
|
||||
self.client = TelegramClient(token)
|
||||
|
||||
def parse_inbound(self, raw_payload: dict[str, Any]) -> ParsedInboundEvent:
|
||||
event_kind = "other"
|
||||
message = raw_payload.get("message")
|
||||
if message is not None:
|
||||
event_kind = "message"
|
||||
else:
|
||||
message = raw_payload.get("edited_message")
|
||||
if message is not None:
|
||||
event_kind = "edited_message"
|
||||
|
||||
if message is None:
|
||||
return ParsedInboundEvent(
|
||||
platform=self.platform,
|
||||
event_kind=event_kind,
|
||||
external_peer_id=None,
|
||||
external_peer_kind="unknown",
|
||||
external_message_id=None,
|
||||
external_user_id=None,
|
||||
text=None,
|
||||
raw_payload=raw_payload,
|
||||
)
|
||||
|
||||
chat = message.get("chat") or {}
|
||||
sender = message.get("from") or {}
|
||||
chat_type = str(chat.get("type") or "unknown")
|
||||
peer_kind = {
|
||||
"private": "direct",
|
||||
"group": "group",
|
||||
"supergroup": "group",
|
||||
"channel": "channel",
|
||||
}.get(chat_type, "unknown")
|
||||
display_name = chat.get("title") or " ".join(
|
||||
part
|
||||
for part in (sender.get("first_name"), sender.get("last_name"))
|
||||
if part
|
||||
)
|
||||
|
||||
return ParsedInboundEvent(
|
||||
platform=self.platform,
|
||||
event_kind=event_kind,
|
||||
external_peer_id=str(chat["id"]) if chat.get("id") is not None else None,
|
||||
external_peer_kind=peer_kind,
|
||||
external_message_id=(
|
||||
str(message["message_id"]) if message.get("message_id") is not None else None
|
||||
),
|
||||
external_user_id=str(sender["id"]) if sender.get("id") is not None else None,
|
||||
text=message.get("text") or message.get("caption"),
|
||||
raw_payload=raw_payload,
|
||||
display_name=display_name or None,
|
||||
username=sender.get("username") or chat.get("username"),
|
||||
metadata={"chat_type": chat_type, "update_id": raw_payload.get("update_id")},
|
||||
)
|
||||
|
||||
async def send_message(
|
||||
self,
|
||||
*,
|
||||
external_peer_id: str,
|
||||
text: str,
|
||||
parse_mode: str | None = None,
|
||||
reply_to_message_id: str | None = None,
|
||||
) -> PlatformSendResult:
|
||||
return await self.client.send_message(
|
||||
chat_id=external_peer_id,
|
||||
text=text,
|
||||
parse_mode=parse_mode,
|
||||
reply_to_message_id=reply_to_message_id,
|
||||
)
|
||||
|
||||
async def edit_message(
|
||||
self,
|
||||
*,
|
||||
external_peer_id: str,
|
||||
external_message_id: str,
|
||||
text: str,
|
||||
parse_mode: str | None = None,
|
||||
) -> PlatformSendResult:
|
||||
return await self.client.edit_message(
|
||||
chat_id=external_peer_id,
|
||||
message_id=external_message_id,
|
||||
text=text,
|
||||
parse_mode=parse_mode,
|
||||
)
|
||||
|
||||
async def validate_credentials(self) -> dict[str, Any]:
|
||||
return await self.client.validate()
|
||||
|
||||
async def leave_chat(self, *, external_peer_id: str) -> None:
|
||||
await self.client.leave_chat(chat_id=external_peer_id)
|
||||
|
||||
async def fetch_updates(self, *, offset: int | None) -> AsyncIterator[dict[str, Any]]:
|
||||
async for update in self.client.get_updates(offset=offset):
|
||||
yield update
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue