Add new file
This commit is contained in:
Markov Andrey
241
core/handlers/message_handler.py
Normal file
241
core/handlers/message_handler.py
Normal file
@@ -0,0 +1,241 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
"""
|
||||
Обработчик входящих XMPP сообщений (личных и групповых).
|
||||
Использует RAGClient для отправки запросов на RAG-сервер.
|
||||
ИСТОРИЯ НЕ ПЕРЕДАЁТСЯ – сервер хранит её самостоятельно.
|
||||
"""
|
||||
|
||||
import logging
|
||||
import re
|
||||
import asyncio
|
||||
from core.utils.layout_converter import convert_layout
|
||||
from core.commands.registry import suggest_command
|
||||
|
||||
logger = logging.getLogger(__name__)
|
||||
|
||||
|
||||
class MessageHandler:
|
||||
"""
|
||||
Хендлер, привязанный к Bot. Вызывается при событии "message".
|
||||
"""
|
||||
|
||||
def __init__(self, bot):
|
||||
self.bot = bot
|
||||
|
||||
async def _wait_for_queue(self):
|
||||
"""Ожидает создания очереди индексации (защита от гонки)."""
|
||||
while self.bot.indexing_queue is None:
|
||||
logger.debug("Очередь индексации ещё не создана, ожидание...")
|
||||
await asyncio.sleep(0.5)
|
||||
|
||||
async def handle(self, msg):
|
||||
"""
|
||||
Основная точка входа для всех сообщений.
|
||||
"""
|
||||
# Определяем тип: личный чат (chat/normal) или групповой (groupchat)
|
||||
if msg['type'] not in ('chat', 'normal', 'groupchat'):
|
||||
return
|
||||
|
||||
user_jid = msg['from'].bare
|
||||
# Не обрабатываем сообщения от самого бота
|
||||
if msg['type'] == 'groupchat':
|
||||
if msg['from'].resource == self.bot.boundjid.user:
|
||||
return
|
||||
else:
|
||||
if msg['from'] == self.bot.boundjid:
|
||||
return
|
||||
|
||||
# Извлекаем room_jid для групповых чатов (None для личных)
|
||||
room_jid = msg['from'].bare if msg['type'] == 'groupchat' else None
|
||||
|
||||
# Проверка OMEMO (сквозное шифрование не поддерживается)
|
||||
if self.bot.is_message_encrypted(msg):
|
||||
reply = msg.reply(
|
||||
"🔒 Я получил зашифрованное сообщение (OMEMO), но не могу его расшифровать. "
|
||||
"Пожалуйста, отключите сквозное шифрование."
|
||||
)
|
||||
if reply:
|
||||
reply.send()
|
||||
return
|
||||
|
||||
# Извлечение тела сообщения и URL файла (из OOB)
|
||||
raw_body = msg['body'].strip() if msg['body'] else ""
|
||||
file_url = msg['oob']['url'] if msg['oob'] else None
|
||||
|
||||
# Если OOB отсутствует, пытаемся извлечь URL из тела сообщения
|
||||
if not file_url and raw_body:
|
||||
url_match = re.search(r'https?://[^\s<>"\'\)\]]+', raw_body)
|
||||
if url_match:
|
||||
file_url = url_match.group(0)
|
||||
logger.info(f"Извлечена ссылка из тела сообщения (fallback OOB): {file_url}")
|
||||
|
||||
if not raw_body and not file_url:
|
||||
return
|
||||
|
||||
# Для групповых чатов – проверяем, упомянут ли бот
|
||||
if msg['type'] == 'groupchat':
|
||||
is_command = raw_body.startswith(('!', '/'))
|
||||
has_file = file_url is not None
|
||||
if not (is_command or has_file):
|
||||
if not self._is_mentioned(msg, raw_body):
|
||||
return
|
||||
|
||||
# Исправление раскладки клавиатуры (русская ↔ английская)
|
||||
body = convert_layout(raw_body)
|
||||
if body != raw_body:
|
||||
reply = msg.reply(f"🔄 Исправлено: {body}")
|
||||
if reply:
|
||||
reply.send()
|
||||
|
||||
# Метрика: инкремент счётчика сообщений
|
||||
self.bot.metrics_handler.increment("messages_total")
|
||||
logger.info(f"Сообщение от {user_jid} (room={room_jid}): {body[:50]}...")
|
||||
|
||||
# ---- ИСТОРИЯ БОЛЬШЕ НЕ СОХРАНЯЕТСЯ ЛОКАЛЬНО ----
|
||||
# Вся история хранится на сервере. Клиент не занимается этим.
|
||||
|
||||
# Обработка команд (начинаются с ! или /)
|
||||
if body.startswith(('!', '/')):
|
||||
await self._handle_command(msg, body, user_jid, room_jid)
|
||||
return
|
||||
|
||||
# Режим обучения (текст без файла) – только в личных чатах
|
||||
mode_info = self.bot.learning_mode.get(user_jid)
|
||||
if mode_info and not file_url and room_jid is None:
|
||||
if isinstance(mode_info, tuple):
|
||||
mode, saved_room = mode_info
|
||||
else:
|
||||
mode = mode_info
|
||||
saved_room = None
|
||||
await self._handle_learning_text(msg, user_jid, mode, body, room_jid)
|
||||
return
|
||||
|
||||
# Обработка файла (если есть URL)
|
||||
is_command = body.startswith(('!', '/'))
|
||||
is_surgical_keyword = any(kw in body.lower() for kw in self.bot.config.surgical_keywords)
|
||||
if file_url and not (is_command or is_surgical_keyword):
|
||||
await self.bot.file_handler.handle_file(msg, file_url, user_jid, room_jid)
|
||||
return
|
||||
|
||||
# ---- RAG-ЗАПРОС (БЕЗ ИСТОРИИ) ----
|
||||
# История НЕ ПЕРЕДАЁТСЯ в запросе. Сервер сам её получит из БД.
|
||||
key = (user_jid, room_jid) if room_jid is not None else user_jid
|
||||
|
||||
result = await self.bot.rag_client.query(
|
||||
query=body,
|
||||
user_jid=user_jid,
|
||||
room_jid=room_jid,
|
||||
prompts=self.bot.prompts,
|
||||
last_file_path=self.bot._last_file_path.get(key),
|
||||
last_file_text=self.bot.last_file_texts.get(user_jid)
|
||||
)
|
||||
|
||||
answer = result.get('answer', '⚠️ Не удалось получить ответ.')
|
||||
|
||||
# Если в ответе есть файловые теги, обрабатываем их
|
||||
if self.bot.config.surgical_replace and any(tag in answer for tag in ("[SURGICAL_REPLACE]", "[REWRITE]", "[FILE]")):
|
||||
await self.bot.handle_ai_file_request(msg, answer, user_jid, room_jid, body=body)
|
||||
else:
|
||||
reply = msg.reply(answer)
|
||||
if reply:
|
||||
reply.send()
|
||||
else:
|
||||
logger.error("Не удалось отправить ответ RAG (без тегов)")
|
||||
|
||||
# ---- ИСТОРИЯ БОЛЬШЕ НЕ СОХРАНЯЕТСЯ ЛОКАЛЬНО ----
|
||||
# await self.bot.db.add_history(...) # УБРАНО!
|
||||
|
||||
# ==========================================================
|
||||
# Вспомогательные методы
|
||||
# ==========================================================
|
||||
|
||||
def _is_mentioned(self, msg, body: str) -> bool:
|
||||
"""
|
||||
Проверяет, упомянут ли бот в групповом чате.
|
||||
"""
|
||||
my_nick = self.bot.plugin['xep_0045'].our_nicks.get(msg['from'].bare, self.bot.boundjid.user)
|
||||
mention_jid = self.bot.boundjid.user.lower()
|
||||
mention_word = self.bot.config.mention_keyword.lower()
|
||||
return (mention_jid in body.lower() or
|
||||
my_nick.lower() in body.lower() or
|
||||
mention_word in body.lower()) or body.startswith(('!', '/'))
|
||||
|
||||
async def _handle_command(self, msg, body: str, user_jid: str, room_jid: str):
|
||||
"""Обрабатывает команды (начинаются с ! или /)."""
|
||||
cmd_name = body.split()[0].lower()
|
||||
if cmd_name in self.bot.commands:
|
||||
await self.bot.commands[cmd_name].execute(msg, body, user_jid, room_jid)
|
||||
else:
|
||||
suggestion = suggest_command(cmd_name, self.bot._command_names)
|
||||
if suggestion:
|
||||
reply = msg.reply(
|
||||
f"❓ Неизвестная команда. Может быть, вы имели в виду `{suggestion}`? "
|
||||
f"Напишите `!help` для списка команд."
|
||||
)
|
||||
if reply:
|
||||
reply.send()
|
||||
else:
|
||||
reply = msg.reply("❓ Неизвестная команда. Напишите `!help` для списка команд.")
|
||||
if reply:
|
||||
reply.send()
|
||||
|
||||
async def _handle_learning_text(self, msg, user_jid: str, mode: str, body: str, room_jid: str):
|
||||
"""
|
||||
Обрабатывает текстовые сообщения в режиме обучения (только в личных чатах).
|
||||
Поддерживает ссылки (URL) с флагами --recursive, --depth, --max.
|
||||
"""
|
||||
is_global = (mode == 'global')
|
||||
title = None
|
||||
|
||||
# Парсим --title "Название" (если есть)
|
||||
match = re.search(r'--title "([^"]+)"', body)
|
||||
if match:
|
||||
title = match.group(1)
|
||||
body = re.sub(r'--title "[^"]+"', '', body).strip()
|
||||
|
||||
# Проверяем, является ли тело ссылкой
|
||||
if body.startswith(('http://', 'https://')):
|
||||
parts = body.split(maxsplit=1)
|
||||
url = parts[0]
|
||||
flags_str = parts[1] if len(parts) > 1 else ""
|
||||
|
||||
recursive = '--recursive' in flags_str or '-r' in flags_str
|
||||
depth = 1
|
||||
max_pages = self.bot.config.ws_max_pages
|
||||
|
||||
depth_match = re.search(r'--depth\s+(\d+)', flags_str)
|
||||
if depth_match:
|
||||
depth = int(depth_match.group(1))
|
||||
max_match = re.search(r'--max\s+(\d+)', flags_str)
|
||||
if max_match:
|
||||
max_pages = int(max_match.group(1))
|
||||
|
||||
target_room = None
|
||||
if mode == 'room' and room_jid is not None:
|
||||
target_room = room_jid
|
||||
|
||||
# Создаём задачу для фонового индексатора
|
||||
task = {
|
||||
"type": "url",
|
||||
"user_jid": user_jid,
|
||||
"room_jid": target_room,
|
||||
"url": url,
|
||||
"title": title,
|
||||
"is_global": is_global,
|
||||
"recursive": recursive,
|
||||
"depth": depth,
|
||||
"max_pages": max_pages,
|
||||
}
|
||||
await self._wait_for_queue()
|
||||
await self.bot.indexing_queue.put(task)
|
||||
|
||||
scope = "глобальную" if is_global else ("комнатную" if target_room else "личную")
|
||||
reply = msg.reply(f"🌐 Ссылка '{url}' принята в обработку для {scope} БЗ.")
|
||||
if reply:
|
||||
reply.send()
|
||||
return
|
||||
|
||||
# Если не ссылка – сообщаем, что текст не индексируется
|
||||
reply = msg.reply("В режиме обучения можно отправлять файлы или ссылки. Текст не индексируется.")
|
||||
if reply:
|
||||
reply.send()
|
||||
Reference in New Issue
Block a user