Teknium
3c7069bdcb
authz_mixin, browser_control_broker, delivery, delivery_ledger, display_config, drain_control, hosted_room_links/peer/policy_checkpoint, hosted_rooms, platform_registry, relay/__init__, relay/ws_transport, run.py and slash_commands.py (comments), session_context, session_state, streaming_tts_consumer, turn_lease and small modules. - HostedRoomPolicyCheckpoint._apply_event -> per-kind handler table - WebSocketRelayTransport._handle_frame -> frame-handler table - GatewayAuthorizationMixin: unified adapter setting/flag/extra readers - dead symbols removed (verified zero references): RoomLinkProbe/select_room_link, relay_bot_username, is_restart_loop_tripped, debug_rows, DeadTargetRegistry.all_dead, BrowserControlBroker.detach_owner/_prune_tickets, StreamingTTSConsumer.started/_enqueue_done/ _iter_stream_chunks/_next_stream_chunk, RecoverableHandleCache.status_for, _auth_env, _copy_default_catalog, _parse_timestamp_prefix, _present_* helpers, _send_result_error_kind, _truthy_env, SessionFieldView/TurnLeaseTokenView dunder shims, and their orphaned tests. - lost WHY/invariant text from the earlier compaction restored compactly (541 hunks audited)
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
"""Local index of text we've sent via ``sendRichMessage`` (Bot API 10.1).
|
|
|
|
Telegram does NOT echo a rich message's content back in ``reply_to_message``
|
|
(``.text``/``.caption`` empty, ``.api_kwargs`` None), so replies to rich sends
|
|
arrive with no quotable text. We remember ``message_id -> text`` at send time
|
|
and look it up by ``reply_to_id`` on inbound. This module is the single source
|
|
of truth for that index.
|
|
|
|
Best-effort and dependency-free: every operation swallows errors and degrades
|
|
to a no-op / ``None`` so it can never break a send or an inbound message.
|
|
"""
|
|
|
|
from __future__ import annotations
|
|
|
|
import json
|
|
import os
|
|
import time
|
|
from typing import Optional
|
|
|
|
_MAX_ENTRIES = 1000
|
|
_MAX_TEXT_CHARS = 2000
|
|
|
|
|
|
def _store_path() -> str:
|
|
# get_hermes_home() honors the active profile override.
|
|
from hermes_constants import get_hermes_home
|
|
|
|
return os.path.join(str(get_hermes_home()), "state", "rich_sent_index.json")
|
|
|
|
|
|
def _key(chat_id, message_id) -> str:
|
|
return f"{chat_id}:{message_id}"
|
|
|
|
|
|
def _load(path: str) -> dict:
|
|
try:
|
|
with open(path, "r", encoding="utf-8") as fh:
|
|
data = json.load(fh)
|
|
except (FileNotFoundError, ValueError):
|
|
return {}
|
|
return data if isinstance(data, dict) else {}
|
|
|
|
|
|
def record(chat_id, message_id, text: Optional[str]) -> None:
|
|
"""Persist ``text`` for ``(chat_id, message_id)``. No-op on any failure."""
|
|
if not text or message_id is None or chat_id is None:
|
|
return
|
|
path = _store_path()
|
|
try:
|
|
os.makedirs(os.path.dirname(path), exist_ok=True)
|
|
data = _load(path)
|
|
data[_key(chat_id, message_id)] = {"t": text[:_MAX_TEXT_CHARS], "ts": int(time.time())}
|
|
if len(data) > _MAX_ENTRIES: # trim oldest by timestamp
|
|
by_age = sorted(data.items(), key=lambda kv: kv[1].get("ts", 0))
|
|
for k, _ in by_age[: len(data) - _MAX_ENTRIES]:
|
|
data.pop(k, None)
|
|
tmp = f"{path}.tmp.{os.getpid()}"
|
|
with open(tmp, "w", encoding="utf-8") as fh:
|
|
json.dump(data, fh, ensure_ascii=False)
|
|
os.replace(tmp, path) # atomic; tolerates concurrent writers racing
|
|
except Exception:
|
|
return
|
|
|
|
|
|
def lookup(chat_id, message_id) -> Optional[str]:
|
|
"""Return stored text for ``(chat_id, message_id)`` or ``None``."""
|
|
if message_id is None or chat_id is None:
|
|
return None
|
|
entry = _load(_store_path()).get(_key(chat_id, message_id))
|
|
return (entry.get("t") or None) if isinstance(entry, dict) else None
|