added file
This commit is contained in:
Godnessli committed
1 parent
cc433bd065
commit
c07cd997c9
10 files changed
+754
-1
No files matched your search
+184
@@ -0,0 +1,184 @@
|
||||
import login
|
||||
import answers
|
||||
import sys
|
||||
|
||||
from nio import (
|
||||
MatrixRoom,
|
||||
RoomMessageText,
|
||||
InviteMemberEvent,
|
||||
MegolmEvent,
|
||||
KeyVerificationStart,
|
||||
KeyVerificationKey,
|
||||
KeyVerificationMac,
|
||||
KeyVerificationCancel,
|
||||
LocalProtocolError,
|
||||
ToDeviceError
|
||||
)
|
||||
|
||||
WELCOME_MSG = "Добро пожаловать ко мне, УЕБОТУ! АХАХА\n" \
|
||||
"И так, вот что я умею:\n" \
|
||||
"1. Сервер — Данные об использовании сервера.\n" \
|
||||
"2. Погода в <город> — Погода в городе.=(тестово).\n" \
|
||||
"3. SCP — Выдача случайного объекта фонда для чтения.\n" \
|
||||
"4. SCP классы — Описание классов.\n" \
|
||||
"5. Помощь — Это меню.\n\n" \
|
||||
"Можете писать слова(Пример: Сервер) или цифрой(Пример: 1)."
|
||||
|
||||
client = None
|
||||
|
||||
async def message_callback(room: MatrixRoom, event: RoomMessageText):
|
||||
print(f"[DEBUG] Открытое сообщение от {event.sender}.")
|
||||
if event.sender == client.user_id:
|
||||
return
|
||||
|
||||
content = None
|
||||
|
||||
if event.body.lower() == "сервер" or event.body == "1":
|
||||
content = answers.system_request()
|
||||
elif event.body.lower() == "погода" or event.body == "2":
|
||||
content = answers.weather_request()
|
||||
elif event.body.lower() == "scp" or event.body == "3":
|
||||
content = answers.scp_request()
|
||||
elif event.body.lower() == "scp классы" or event.body == "4":
|
||||
content = answers.scp_classed()
|
||||
else:
|
||||
content = answers.unknown_request()
|
||||
|
||||
content += answers.help_request()
|
||||
|
||||
await send_message(room, content)
|
||||
|
||||
|
||||
async def megolm_callback(room: MatrixRoom, event: MegolmEvent):
|
||||
try:
|
||||
decrypted = await client.decrypt_event(event)
|
||||
except Exception as e:
|
||||
print(f"[ERROR] Расшифровка не удалась: {e}")
|
||||
return
|
||||
|
||||
if isinstance(decrypted, RoomMessageText):
|
||||
print(f"[DEBUG] Расшифровано: {decrypted.body}")
|
||||
if decrypted.sender == client.user_id:
|
||||
return
|
||||
await send_message(room, decrypted.body)
|
||||
else:
|
||||
print(f"[DEBUG] Расшифровано, но не текст: {type(decrypted).__name__}")
|
||||
|
||||
|
||||
async def invite_callback(room: MatrixRoom, event: InviteMemberEvent):
|
||||
print(f"[INFO] Приглашение в комнату {room.room_id}")
|
||||
await client.join(room.room_id)
|
||||
await send_message(room, WELCOME_MSG)
|
||||
|
||||
|
||||
async def trust_user_devices(user_id: str, device_ids=None):
|
||||
"""Доверие устройствам пользователя (локально, без интерактивной верификации)."""
|
||||
if user_id == client.user_id:
|
||||
return
|
||||
if user_id not in client.device_store:
|
||||
print(f"[WARN] Устройства {user_id} пока не известны.")
|
||||
return
|
||||
|
||||
devices = client.device_store[user_id]
|
||||
trusted = 0
|
||||
for device_id, olm_device in devices.items():
|
||||
if device_ids and device_id not in device_ids:
|
||||
continue
|
||||
try:
|
||||
client.verify_device(olm_device)
|
||||
print(f"[OK] Доверено устройство {device_id} пользователя {user_id}.")
|
||||
trusted += 1
|
||||
except Exception as e:
|
||||
print(f"[ERROR] Не удалось доверить {device_id}: {e}")
|
||||
if trusted == 0:
|
||||
print(f"[WARN] Ни одно устройство {user_id} не было доверено.")
|
||||
|
||||
async def send_message(room: MatrixRoom, text: str):
|
||||
CHUNK = 5000
|
||||
if len(text) > CHUNK:
|
||||
start = 0
|
||||
end = 0
|
||||
while start < len(text):
|
||||
end = start + CHUNK
|
||||
if end < len(text):
|
||||
while text[end] != ".":
|
||||
end -= 1
|
||||
|
||||
end += 1
|
||||
chunk = text[start:end]
|
||||
await client.room_send(
|
||||
room_id = room.room_id,
|
||||
message_type = "m.room.message",
|
||||
content = {"msgtype" : "m.text", "body" : chunk},
|
||||
ignore_unverified_devices = True
|
||||
)
|
||||
|
||||
start = end
|
||||
print(start)
|
||||
else:
|
||||
await client.room_send(
|
||||
room_id = room.room_id,
|
||||
message_type = "m.room.message",
|
||||
content = {"msgtype" : "m.text", "body" : text},
|
||||
ignore_unverified_devices = True
|
||||
)
|
||||
|
||||
async def after_first_sync():
|
||||
print("[INFO] Ожидание первого синка...")
|
||||
await client.synced.wait()
|
||||
print("[INFO] Первый синк выполнен. Доверяем устройствам...")
|
||||
for user_id in login.TRUST_USERS:
|
||||
await trust_user_devices(user_id)
|
||||
|
||||
async def to_device_callback(event):
|
||||
"""Автоматически принимает и подтверждает SAS-верификацию от доверенных пользователей."""
|
||||
try:
|
||||
if isinstance(event, KeyVerificationStart):
|
||||
if event.sender not in login.TRUST_USERS:
|
||||
print(f"[WARN] Верификация от недоверенного пользователя {event.sender}, игнорируем.")
|
||||
return
|
||||
if "emoji" not in event.short_authentication_string:
|
||||
print(f"[WARN] Устройство {event.sender} не поддерживает эмодзи-верификацию.")
|
||||
return
|
||||
|
||||
print(f"[INFO] Принимаем запрос на верификацию от {event.sender}...")
|
||||
resp = await client.accept_key_verification(event.transaction_id)
|
||||
if isinstance(resp, ToDeviceError):
|
||||
print(f"[ERROR] accept_key_verification: {resp}")
|
||||
return
|
||||
|
||||
sas = client.key_verifications[event.transaction_id]
|
||||
todevice_msg = sas.share_key()
|
||||
resp = await client.to_device(todevice_msg)
|
||||
if isinstance(resp, ToDeviceError):
|
||||
print(f"[ERROR] to_device (share_key): {resp}")
|
||||
|
||||
elif isinstance(event, KeyVerificationKey):
|
||||
sas = client.key_verifications[event.transaction_id]
|
||||
print(f"[INFO] Эмодзи: {sas.get_emoji()}")
|
||||
# Автоматически подтверждаем совпадение
|
||||
resp = await client.confirm_short_auth_string(event.transaction_id)
|
||||
if isinstance(resp, ToDeviceError):
|
||||
print(f"[ERROR] confirm_short_auth_string: {resp}")
|
||||
else:
|
||||
print("[INFO] SAS подтверждён автоматически.")
|
||||
|
||||
elif isinstance(event, KeyVerificationMac):
|
||||
sas = client.key_verifications[event.transaction_id]
|
||||
try:
|
||||
todevice_msg = sas.get_mac()
|
||||
except LocalProtocolError as e:
|
||||
print(f"[WARN] Верификация прервана: {e}")
|
||||
return
|
||||
resp = await client.to_device(todevice_msg)
|
||||
if isinstance(resp, ToDeviceError):
|
||||
print(f"[ERROR] to_device (mac): {resp}")
|
||||
else:
|
||||
print(f"[OK] Верификация с {event.sender} завершена. "
|
||||
f"Доверенные устройства: {sas.verified_devices}")
|
||||
|
||||
elif isinstance(event, KeyVerificationCancel):
|
||||
print(f"[INFO] Верификация отменена: {event.reason}")
|
||||
|
||||
except Exception as e:
|
||||
print(f"[ERROR] to_device_callback: {e}")
|
||||
Reference in new issue
Block a user