107 lines
3.9 KiB
Python
107 lines
3.9 KiB
Python
from __future__ import annotations
|
||
|
||
from typing import Any, Awaitable, Callable
|
||
|
||
from adapter.matrix.store import set_room_meta
|
||
from core.protocol import IncomingCommand, OutgoingMessage
|
||
|
||
|
||
async def _fallback_new_chat(
|
||
event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr
|
||
) -> list:
|
||
if not await auth_mgr.is_authenticated(event.user_id):
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Введите !start чтобы начать.")]
|
||
|
||
name = " ".join(event.args).strip() if event.args else ""
|
||
chats = await chat_mgr.list_active(event.user_id)
|
||
chat_id = f"C{len(chats) + 1}"
|
||
ctx = await chat_mgr.get_or_create(
|
||
user_id=event.user_id,
|
||
chat_id=chat_id,
|
||
platform=event.platform,
|
||
surface_ref=event.chat_id,
|
||
name=name or None,
|
||
)
|
||
return [
|
||
OutgoingMessage(
|
||
chat_id=event.chat_id, text=f"Создан чат: {ctx.display_name} ({ctx.chat_id})"
|
||
)
|
||
]
|
||
|
||
|
||
def make_handle_new_chat(
|
||
client: Any | None,
|
||
store: Any | None,
|
||
) -> Callable[..., Awaitable[list]]:
|
||
async def handle_new_chat(
|
||
event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr
|
||
) -> list:
|
||
if client is None or store is None:
|
||
return await _fallback_new_chat(event, auth_mgr, platform, chat_mgr, settings_mgr)
|
||
|
||
if not await auth_mgr.is_authenticated(event.user_id):
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Введите !start чтобы начать.")]
|
||
|
||
name = " ".join(event.args).strip() if event.args else ""
|
||
chats = await chat_mgr.list_active(event.user_id)
|
||
chat_id = f"C{len(chats) + 1}"
|
||
room_name = name or f"Чат {chat_id}"
|
||
|
||
response = await client.room_create(
|
||
name=room_name,
|
||
invite=[event.user_id],
|
||
is_direct=False,
|
||
)
|
||
room_id = getattr(response, "room_id", None)
|
||
if not room_id:
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Не удалось создать комнату.")]
|
||
|
||
await set_room_meta(
|
||
store,
|
||
room_id,
|
||
{
|
||
"room_type": "chat",
|
||
"chat_id": chat_id,
|
||
"display_name": room_name,
|
||
"matrix_user_id": event.user_id,
|
||
},
|
||
)
|
||
ctx = await chat_mgr.get_or_create(
|
||
user_id=event.user_id,
|
||
chat_id=chat_id,
|
||
platform=event.platform,
|
||
surface_ref=room_id,
|
||
name=room_name,
|
||
)
|
||
return [
|
||
OutgoingMessage(
|
||
chat_id=event.chat_id,
|
||
text=f"Создан чат: {ctx.display_name} ({ctx.chat_id})\nКомната: {room_id}",
|
||
)
|
||
]
|
||
|
||
return handle_new_chat
|
||
|
||
|
||
async def handle_list_chats(
|
||
event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr
|
||
) -> list:
|
||
chats = await chat_mgr.list_active(event.user_id)
|
||
if not chats:
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Нет активных чатов.")]
|
||
lines = [f"• {c.display_name} ({c.chat_id})" for c in chats]
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="\n".join(lines))]
|
||
|
||
|
||
async def handle_rename(event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr) -> list:
|
||
if not event.args:
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Укажите название: !rename Название")]
|
||
ctx = await chat_mgr.rename(event.chat_id, " ".join(event.args), user_id=event.user_id)
|
||
return [OutgoingMessage(chat_id=event.chat_id, text=f"Переименован в: {ctx.display_name}")]
|
||
|
||
|
||
async def handle_archive(
|
||
event: IncomingCommand, auth_mgr, platform, chat_mgr, settings_mgr
|
||
) -> list:
|
||
await chat_mgr.archive(event.chat_id, user_id=event.user_id)
|
||
return [OutgoingMessage(chat_id=event.chat_id, text="Чат архивирован.")]
|