missytake/team-bot
1
0
Fork 1
Code Issues 3 Pull requests Packages Projects Releases 4 Wiki Activity

store relay mappings in pickle.db. closes #2

Browse source
This commit is contained in:
missytake 2023-10-08 10:59:49 +02:00
parent 17a3fce45b
commit 3448eec281
1 changed files with 43 additions and 26 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

69
src/teams_bot/bot.py
View file

@ -34,6 +34,8 @@ class RelayPlugin:
self.account = account
self.kvstore = kvstore
self.crew = account.get_chat_by_id(kvstore.get("crew_id"))
if not kvstore.get("relays"):
kvstore.set("relays", list())
@account_hookimpl
def ac_incoming_message(self, message: deltachat.Message):
@ -99,23 +101,12 @@ class RelayPlugin:
def forward_to_outside(self, message: deltachat.Message):
"""forward an answer to an outsider."""
bot_localpart = self.account.get_config("addr").split("@")[0]
title_prefix = f"[{bot_localpart}] "
chat_title = message.chat.get_name().split(title_prefix)[1]
logging.debug("stripped %s to %s", message.chat.get_name(), chat_title)
for chat in self.account.get_chats():
if chat_title == chat.get_name():
if message.quote.text in [msg.text for msg in chat.get_messages()]:
outside_chat = chat
break
else:
logging.debug(
"No corresponding message in chat %s with name: %s",
chat.id,
chat.get_name(),
)
else:
logging.error("Couldn't find the chat with the title: %s", chat_title)
outside_chat = self.get_outside_chat(message.chat.id)
if not outside_chat:
logging.error(
"Couldn't find the corresponding outside chat for relay group %s",
message.chat.id,
)
return
outside_chat.send_msg(message)
@ -124,15 +115,13 @@ class RelayPlugin:
outsider = message.get_sender_contact().addr
crew_members = self.crew.get_contacts()
crew_members.remove(self.account.get_self_contact())
group_name = "[%s] %s" % (
self.account.get_config("addr").split("@")[0],
message.chat.get_name(),
)
for chat in self.account.get_chats():
if chat.get_name() == group_name:
relay_group = chat
break
else:
relay_group = self.get_relay_group(message.chat.id)
if not relay_group:
group_name = "[%s] %s" % (
self.account.get_config("addr").split("@")[0],
message.chat.get_name(),
)
logging.info("creating new relay group: '%s'", group_name)
relay_group = self.account.create_group_chat(
group_name, crew_members, verified=False
@ -142,6 +131,10 @@ class RelayPlugin:
"This is the relay group for %s; I'll only forward 'direct replies' to the outside."
% (message.chat.get_name())
)
relay_mappings = self.kvstore.get("relays")
relay_mappings.append(tuple([message.chat.id, relay_group.id]))
self.kvstore.set("relays", relay_mappings)
message.set_override_sender_name(outsider)
relay_group.send_msg(message)
@ -162,3 +155,27 @@ class RelayPlugin:
if crew_member not in chat.get_contacts():
return False # all crew members have to be in any relay group
return True
def get_outside_chat(self, relay_group_id: int) -> deltachat.Chat:
"""Get the corresponding outside chat for the ID of a relay group.
:param relay_group_id: the chat.id of the relay group
:return: the outside chat
"""
relay_mappings = self.kvstore.get("relays")
for mapping in relay_mappings:
if mapping[1] == relay_group_id:
return self.account.get_chat_by_id(mapping[0])
return None
def get_relay_group(self, outside_id: int) -> deltachat.Chat:
"""Get the corresponding relay group for the ID of the outside chat.
:param outside_id: the chat.id of the outside chat
:return: the relay group
"""
relay_mappings = self.kvstore.get("relays")
for mapping in relay_mappings:
if mapping[0] == outside_id:
return self.account.get_chat_by_id(mapping[1])
return None