227 lines
9.6 KiB
Python
227 lines
9.6 KiB
Python
"""Update dispatch — the heart of the bot.
|
|
|
|
Pipeline per update:
|
|
callback_query → callbacks router (link-gated)
|
|
message text starting "/" → command router
|
|
voice/audio → download → STT → treated as text (source=voice)
|
|
text → pending-flow interception (linking / accept-arrive confirmations)
|
|
→ auth gate (link or admin)
|
|
→ NLU → intent router
|
|
"""
|
|
import logging
|
|
|
|
from app import drafts as draft_store
|
|
from app.config import settings
|
|
from app.nlu import understand
|
|
from app.runtime import BotContext, LinkedUser
|
|
from app.stt import transcribe, transcribe_multi
|
|
from app.texts import CANCELLED, LINK_START, NOTHING_TO_COMMIT, STT_FAILED, UNKNOWN_HINT, VOICE_EMPTY, esc
|
|
|
|
from . import commands, callbacks, quote, supplier, material, queries, decisions
|
|
|
|
logger = logging.getLogger("bot.handlers")
|
|
|
|
|
|
async def handle_update(ctx: BotContext, update: dict) -> None:
|
|
if update.get("callback_query"):
|
|
await callbacks.handle_callback(ctx, update["callback_query"])
|
|
return
|
|
message = update.get("message") or update.get("edited_message")
|
|
if not message:
|
|
return
|
|
chat_id = message["chat"]["id"]
|
|
try:
|
|
await handle_message(ctx, chat_id, message)
|
|
except Exception:
|
|
logger.exception("message handling failed for chat %s", chat_id)
|
|
await ctx.send(chat_id, "خطای غیرمنتظره رخ داد؛ دوباره تلاش کن.")
|
|
|
|
|
|
async def handle_message(ctx: BotContext, chat_id: int, message: dict) -> None:
|
|
text = (message.get("text") or "").strip()
|
|
source = "text"
|
|
voice_transcripts: list[dict] | None = None
|
|
|
|
if text.startswith("/"):
|
|
await commands.handle_command(ctx, chat_id, text)
|
|
return
|
|
|
|
if not text and (message.get("voice") or message.get("audio") or message.get("document")):
|
|
source = "voice"
|
|
await ctx.bale.send_chat_action(chat_id, "typing")
|
|
file_id = (message.get("voice") or message.get("audio") or message.get("document"))["file_id"]
|
|
try:
|
|
audio = await ctx.bale.download_file(file_id)
|
|
voice_transcripts = await transcribe_multi(
|
|
settings,
|
|
audio,
|
|
mime_type=(message.get("voice") or {}).get("mime_type"),
|
|
filename=(message.get("voice") or {}).get("file_name"),
|
|
)
|
|
except Exception as exc:
|
|
detail = "هوش مصنوعی تنظیم نشده" if "not configured" in str(exc) else str(exc)[:200]
|
|
await ctx.send(chat_id, STT_FAILED.format(detail=esc(detail)))
|
|
return
|
|
if not voice_transcripts:
|
|
await ctx.send(chat_id, VOICE_EMPTY)
|
|
return
|
|
text = voice_transcripts[0]["text"]
|
|
|
|
# Pending multi-step interactions take the message first.
|
|
flow = ctx.pending.get(chat_id)
|
|
if flow:
|
|
handled = await _route_to_flow(ctx, chat_id, flow, text)
|
|
if handled:
|
|
return
|
|
ctx.pending.pop(chat_id) # stale/irrelevant flow → fall through
|
|
|
|
user = await _gate(ctx, chat_id, message)
|
|
if user is None:
|
|
return # gate either started linking or is mid-linking
|
|
|
|
message_id = message.get("message_id")
|
|
reply_to = message.get("reply_to_message")
|
|
reply_to_msg_id = reply_to.get("message_id") if reply_to else None
|
|
|
|
target_draft = None
|
|
if reply_to_msg_id is not None:
|
|
target_draft = await draft_store.get_by_message_id(chat_id, reply_to_msg_id)
|
|
if target_draft is None:
|
|
await ctx.send(
|
|
chat_id,
|
|
"⚠️ پیشنویس فعال مربوط به این پیام یافت نشد (ممکن است ثبت نهایی، منقضی یا لغو شده باشد).\nبرای ثبت استعلام جدید، پیام خود را بدون ریپلای ارسال فرمایید.",
|
|
reply_to=message_id,
|
|
)
|
|
return
|
|
|
|
draft_summary = _draft_summary(target_draft) if target_draft else None
|
|
input_payload = voice_transcripts if voice_transcripts else text
|
|
result = await understand(settings, input_payload, draft_summary)
|
|
intent, patch = result.get("intent"), result.get("patch") or {}
|
|
consensus_text = result.get("consensus_text") or text
|
|
|
|
if source == "voice":
|
|
await ctx._log(chat_id, "in", "voice", consensus_text)
|
|
|
|
logger.info("chat %s intent=%s conf=%s consensus=%r (is_edit=%s)",
|
|
chat_id, intent, result.get("confidence"), consensus_text[:40], target_draft is not None)
|
|
|
|
if intent in ("register_quote", "update_draft"):
|
|
await quote.handle_entities(
|
|
ctx, chat_id, user, patch, source,
|
|
incoming_msg_id=message_id,
|
|
target_draft=target_draft,
|
|
)
|
|
elif intent == "commit_draft":
|
|
active = target_draft or await draft_store.get_active(chat_id)
|
|
if active is None:
|
|
await ctx.send(chat_id, NOTHING_TO_COMMIT, reply_to=message_id)
|
|
elif active.kind == draft_store.KIND_QUOTE:
|
|
await quote.commit(ctx, chat_id, user, active)
|
|
elif active.kind == draft_store.KIND_SUPPLIER:
|
|
await supplier.commit(ctx, chat_id, user, active)
|
|
elif active.kind == draft_store.KIND_MATERIAL:
|
|
await material.commit(ctx, chat_id, user, active)
|
|
elif intent == "discard_draft":
|
|
active = target_draft or await draft_store.get_active(chat_id)
|
|
if active:
|
|
await draft_store.set_status(active.id, "cancelled")
|
|
if active.review_message_id:
|
|
try:
|
|
await ctx.delete(chat_id, active.review_message_id)
|
|
except Exception:
|
|
pass
|
|
ctx.pending.pop(chat_id, None)
|
|
await ctx.send(chat_id, CANCELLED, reply_to=message_id)
|
|
elif intent == "show_draft":
|
|
active = target_draft or await draft_store.get_active(chat_id)
|
|
if active is None:
|
|
await ctx.send(chat_id, NOTHING_TO_COMMIT, reply_to=message_id)
|
|
elif active.kind == draft_store.KIND_QUOTE:
|
|
await quote.render(ctx, chat_id, active, fresh=False)
|
|
elif active.kind == draft_store.KIND_SUPPLIER:
|
|
await supplier.render(ctx, chat_id, active)
|
|
elif active.kind == draft_store.KIND_MATERIAL:
|
|
await material.render(ctx, chat_id, active)
|
|
elif intent == "add_supplier":
|
|
await supplier.handle_entities(ctx, chat_id, user, patch, source)
|
|
elif intent == "add_material":
|
|
await material.handle_entities(ctx, chat_id, user, patch, source)
|
|
elif intent == "status_query":
|
|
await queries.handle_query(ctx, chat_id, user, patch)
|
|
elif intent == "accept_quote":
|
|
await decisions.start_accept(ctx, chat_id, user, patch)
|
|
elif intent == "arrive_items":
|
|
await decisions.start_arrive(ctx, chat_id, user, patch)
|
|
else:
|
|
await ctx.send(chat_id, UNKNOWN_HINT)
|
|
|
|
|
|
# ---------------------------------------------------------------------------
|
|
# helpers
|
|
# ---------------------------------------------------------------------------
|
|
|
|
async def _gate(ctx: BotContext, chat_id: int, message: dict) -> LinkedUser | None:
|
|
"""Link or admin gate. Returns None when the message was consumed by linking."""
|
|
from . import links
|
|
|
|
link = await ctx.links.get(chat_id)
|
|
if link and link.user_id:
|
|
return LinkedUser(chat_id, link.user_id, link.display_name, link.role, is_admin=False)
|
|
if chat_id in ctx.admin_chat_ids:
|
|
return LinkedUser(chat_id, None, None, None, is_admin=True)
|
|
|
|
flow = ctx.pending.get(chat_id)
|
|
if flow and flow.kind == "link_password":
|
|
await links.finish_link(ctx, chat_id, flow, text=None, password=message.get("text") or "")
|
|
return None
|
|
if flow and flow.kind == "link_username":
|
|
await links.finish_username_step(ctx, chat_id, flow, message.get("text") or "")
|
|
return None
|
|
await ctx.send(chat_id, LINK_START)
|
|
ctx.pending.set(chat_id, "link_username", {})
|
|
return None
|
|
|
|
|
|
async def _route_to_flow(ctx: BotContext, chat_id: int, flow, text: str) -> bool:
|
|
from . import links, decisions
|
|
|
|
if flow.kind == "link_username":
|
|
await links.finish_username_step(ctx, chat_id, flow, text)
|
|
return True
|
|
if flow.kind == "link_password":
|
|
await links.finish_link(ctx, chat_id, flow, text=text, password=text)
|
|
return True
|
|
if flow.kind == "quote_pick":
|
|
await decisions.handle_quote_pick_text(ctx, chat_id, flow, text)
|
|
return True
|
|
if flow.kind in ("accept", "arrive_qty", "arrive_confirm"):
|
|
return await decisions.handle_flow_text(ctx, chat_id, flow, text)
|
|
return False
|
|
|
|
|
|
def _draft_summary(draft) -> str | None:
|
|
if draft is None:
|
|
return None
|
|
import json
|
|
|
|
payload = draft.payload or {}
|
|
if draft.kind == draft_store.KIND_QUOTE:
|
|
items = ", ".join(
|
|
f"{it.get('material_name') or '?'}"
|
|
+ (f" {it.get('quantity')}" if it.get("quantity") else "")
|
|
+ (f" {it.get('unit_name')}" if it.get("unit_name") else "")
|
|
for it in payload.get("items") or []
|
|
)
|
|
summary = (
|
|
f"kind=quote; supplier={payload.get('supplier_name') or '?'}; project={payload.get('project_name') or '?'}; "
|
|
f"items=[{items}]; terms={[t.get('label') for t in payload.get('termsheets') or []]}"
|
|
)
|
|
elif draft.kind == draft_store.KIND_SUPPLIER:
|
|
summary = f"kind=add_supplier; name={payload.get('name') or '?'}; category={payload.get('category_name') or '?'}"
|
|
elif draft.kind == draft_store.KIND_MATERIAL:
|
|
summary = f"kind=add_material; name={payload.get('name') or '?'}; unit={payload.get('unit_name') or '?'}"
|
|
else:
|
|
summary = json.dumps(payload, ensure_ascii=False)[:500]
|
|
return summary
|