#!/usr/bin/env python3 """Telegram inline command picker — searchable access to EVERY command/skill. The BotCommand menu is capped (100/scope, Hermes uses 60), so inline mode (``@yourbot ``, live per keystroke, 50 per page) exposes the rest. Tapping a result sends ``/cmd args`` as the user; it starts with ``/`` so it arrives even under privacy mode and dispatches through the normal command path. PTB-object-free on purpose: plain dicts keep catalog/filter/pagination unit-testable without python-telegram-bot; the adapter converts to ``InlineQueryResultArticle``. Inline mode must be enabled via BotFather ``/setinline``; until then the handler is inert. """ from __future__ import annotations import logging from typing import Any, Dict, List, Tuple logger = logging.getLogger(__name__) # Telegram hard limit: max 50 results per answerInlineQuery call. PAGE_SIZE = 50 # Results depend on caller auth + installed skills: never share across users, keep short. CACHE_TIME_SECONDS = 10 def collect_inline_catalog() -> List[Dict[str, str]]: """Every dispatchable command as ``{name, description}``, first occurrence wins. Core gateway commands first (menu gating), then plugin + skill commands via the shared collector with ``max_slots=None`` — the inline picker has no cap. """ catalog: List[Dict[str, str]] = [] seen: set[str] = set() try: from hermes_cli.commands import ( _collect_gateway_skill_entries, _sanitize_telegram_name, telegram_bot_commands, ) except Exception: # pragma: no cover - defensive logger.debug("inline picker: commands registry unavailable", exc_info=True) return catalog try: for name, desc in telegram_bot_commands(): if name and name not in seen: seen.add(name) catalog.append({"name": name, "description": desc or ""}) except Exception: logger.debug("inline picker: core command collection failed", exc_info=True) try: entries, _hidden = _collect_gateway_skill_entries( platform="telegram", max_slots=None, # inline mode has no cap — collect everything reserved_names=set(seen), desc_limit=100, sanitize_name=_sanitize_telegram_name, ) for entry in entries: # Entry shape is (name, desc, cmd_key[, raw_name]) — tolerate both. name, desc = entry[0], entry[1] if name and name not in seen: seen.add(name) catalog.append({"name": name, "description": desc or ""}) except Exception: logger.debug("inline picker: skill/plugin collection failed", exc_info=True) return catalog def filter_catalog(catalog: List[Dict[str, str]], term: str) -> List[Dict[str, str]]: """Rank *catalog* against *term*: prefix > name-substring > description. Empty term returns the full catalog in collection order (the "browse" view).""" term = (term or "").strip().lower().lstrip("/") if not term: return list(catalog) prefix: List[Dict[str, str]] = [] name_sub: List[Dict[str, str]] = [] desc_sub: List[Dict[str, str]] = [] # Treat hyphens/underscores as equivalent, mirroring command dispatch. norm_term = term.replace("-", "_") for item in catalog: norm_name = item["name"].lower().replace("-", "_") if norm_name.startswith(norm_term): prefix.append(item) elif norm_term in norm_name: name_sub.append(item) elif term in (item.get("description") or "").lower(): desc_sub.append(item) return prefix + name_sub + desc_sub def build_inline_results( query: str, offset: str = "", page_size: int = PAGE_SIZE, ) -> Tuple[List[Dict[str, Any]], str]: """One page of inline results for *query*. First token filters the catalog; the remainder becomes the command argument (``@bot plan migrate auth`` → tapping ``/plan`` sends ``/plan migrate auth``). Returns ``(results, next_offset)``; ``next_offset == ""`` means last page. """ query = (query or "").strip() parts = query.split(None, 1) term = parts[0] if parts else "" args = parts[1].strip() if len(parts) > 1 else "" matches = filter_catalog(collect_inline_catalog(), term) try: start = int(offset) if offset else 0 except (TypeError, ValueError): start = 0 page = matches[start:start + page_size] next_offset = str(start + page_size) if len(matches) > start + page_size else "" results: List[Dict[str, Any]] = [] for item in page: message_text = f"/{item['name']}" if args: message_text += f" {args}" results.append({ "id": f"{start}:{item['name']}"[:64], # offset-scoped: unique across pages "title": f"/{item['name']}", "description": (item.get("description") or "")[:100], "message_text": message_text[:4096], }) return results, next_offset