mirror of
https://github.com/MODSetter/SurfSense.git
synced 2026-06-04 20:05:16 +02:00
feat(gateway): add Telegram command and stream handling
This commit is contained in:
parent
59e6475348
commit
967ec099c8
2 changed files with 241 additions and 0 deletions
91
surfsense_backend/app/gateway/telegram/commands.py
Normal file
91
surfsense_backend/app/gateway/telegram/commands.py
Normal file
|
|
@ -0,0 +1,91 @@
|
|||
"""Telegram command handlers."""
|
||||
|
||||
from __future__ import annotations
|
||||
|
||||
from app.gateway.base.adapter import ParsedInboundEvent
|
||||
from app.gateway.pairing import redeem_pairing_code
|
||||
from app.gateway.ratelimit import acquire_token
|
||||
from app.gateway.telegram.adapter import TelegramAdapter
|
||||
|
||||
HELP_TEXT = (
|
||||
"SurfSense Telegram commands:\n"
|
||||
"/start <code> - pair this chat\n"
|
||||
"/new - start a fresh conversation\n"
|
||||
"/help - show this help"
|
||||
)
|
||||
|
||||
|
||||
def command_name(text: str | None) -> str | None:
|
||||
if not text or not text.startswith("/"):
|
||||
return None
|
||||
return text.split(maxsplit=1)[0].split("@", 1)[0].lower()
|
||||
|
||||
|
||||
async def handle_start_command(
|
||||
*,
|
||||
session,
|
||||
adapter: TelegramAdapter,
|
||||
event: ParsedInboundEvent,
|
||||
) -> bool:
|
||||
text = event.text or ""
|
||||
parts = text.split(maxsplit=1)
|
||||
if len(parts) != 2 or not event.external_peer_id:
|
||||
await adapter.send_message(
|
||||
external_peer_id=event.external_peer_id or "",
|
||||
text="Generate a pairing code in SurfSense Settings > Messaging Channels, then send /start CODE here.",
|
||||
)
|
||||
return True
|
||||
|
||||
binding = await redeem_pairing_code(
|
||||
session,
|
||||
code=parts[1].strip(),
|
||||
external_peer_id=event.external_peer_id,
|
||||
external_peer_kind=event.external_peer_kind,
|
||||
external_display_name=event.display_name,
|
||||
external_username=event.username,
|
||||
external_metadata=event.metadata,
|
||||
)
|
||||
if binding is None:
|
||||
await adapter.send_message(
|
||||
external_peer_id=event.external_peer_id,
|
||||
text="That pairing code is invalid or expired. Generate a new code in SurfSense.",
|
||||
)
|
||||
return True
|
||||
|
||||
await adapter.send_message(
|
||||
external_peer_id=event.external_peer_id,
|
||||
text="SurfSense is connected. Send a message here to chat with your agent.",
|
||||
)
|
||||
return True
|
||||
|
||||
|
||||
async def handle_help_command(*, adapter: TelegramAdapter, event: ParsedInboundEvent) -> bool:
|
||||
if not event.external_peer_id:
|
||||
return True
|
||||
await adapter.send_message(external_peer_id=event.external_peer_id, text=HELP_TEXT)
|
||||
return True
|
||||
|
||||
|
||||
async def send_unbound_onboarding(
|
||||
*,
|
||||
adapter: TelegramAdapter,
|
||||
event: ParsedInboundEvent,
|
||||
dashboard_url: str,
|
||||
) -> None:
|
||||
if not event.external_peer_id:
|
||||
return
|
||||
wait_ms = await acquire_token(
|
||||
f"tg:onboarded:{event.external_peer_id}",
|
||||
capacity=1,
|
||||
refill_per_sec=1 / 3600,
|
||||
)
|
||||
if wait_ms > 0:
|
||||
return
|
||||
await adapter.send_message(
|
||||
external_peer_id=event.external_peer_id,
|
||||
text=(
|
||||
"Hi! To use SurfSense via Telegram, generate a pairing code at "
|
||||
f"{dashboard_url} and send /start CODE here."
|
||||
),
|
||||
)
|
||||
|
||||
Loading…
Add table
Add a link
Reference in a new issue