import logging from threading import Event import deltachat from deltachat import account_hookimpl from deltachat.capi import lib as dclib from .commands import help_message, set_display_name, set_avatar, get_crew_id class SetupPlugin: def __init__(self, crew_id): self.member_added = Event() self.crew_id = crew_id self.message_sent = Event() self.outgoing_messages = 0 @account_hookimpl def ac_member_added(self, chat: deltachat.Chat, contact, actor, message): if chat.id == self.crew_id and chat.num_contacts() == 2: self.member_added.set() @account_hookimpl def ac_message_delivered(self, message: deltachat.Message): if not message.is_system_message(): self.outgoing_messages -= 1 if self.outgoing_messages < 1: self.message_sent.set() class RelayPlugin: def __init__(self, account: deltachat.Account): self.account = account @account_hookimpl def ac_incoming_message(self, message: deltachat.Message): """This method is called on every incoming message and decides what to do with it.""" logging.info( "New message from %s in chat %s: %s", message.get_sender_contact().addr, message.chat.get_name(), message.text, ) if message.is_system_message(): logging.debug("This is a system message") """:TODO handle chat name changes""" return if message.chat.id == get_crew_id(self.account): if message.text.startswith("/"): logging.debug( "handling command by %s: %s", message.get_sender_contact().addr, message.text, ) arguments = message.text.split(" ") if arguments[0] == "/help": self.reply(message.chat, help_message(), quote=message) if arguments[0] == "/set_name": self.reply( message.chat, set_display_name(self.account, arguments[1]), quote=message, ) if arguments[0] == "/set_avatar": result = set_avatar(self.account, message) self.reply(message.chat, result, quote=message) else: logging.debug("Ignoring message, just the crew chatting") elif self.is_relay_group(message.chat): if message.quote: if ( message.quote.get_sender_contact() == self.account.get_self_contact() ): logging.debug("Forwarding message to outsider") self.forward_to_outside(message) else: logging.debug("Ignoring message, just the crew chatting") else: logging.debug("Ignoring message, just the crew chatting") else: logging.debug("Forwarding message to relay group") self.forward_to_relay_group(message) def reply(self, chat: deltachat.Chat, text: str, quote: deltachat.Message = None): """Send a reply to a chat, with optional quote.""" msg = deltachat.Message.new_empty(self.account, view_type="text") msg.set_text(text) msg.quote = quote sent_id = dclib.dc_send_msg(self.account._dc_context, chat.id, msg._dc_msg) assert sent_id == msg.id 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) return outside_chat.send_msg(message) def forward_to_relay_group(self, message: deltachat.Message): """forward a request to a relay group; create one if it doesn't exist yet.""" outsider = message.get_sender_contact().addr crew_members = self.account.get_chat_by_id( get_crew_id(self.account) ).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: logging.info("creating new relay group: '%s'", group_name) relay_group = self.account.create_group_chat( group_name, crew_members, verified=False ) # relay_group.set_profile_image("assets/avatar.jpg") relay_group.send_text( "This is the relay group for %s; I'll only forward 'direct replies' to the outside." % (message.chat.get_name()) ) message.set_override_sender_name(outsider) relay_group.send_msg(message) def is_relay_group(self, chat: deltachat.Chat) -> bool: """Check whether a chat is a relay group.""" if not chat.get_name().startswith( "[%s] " % (self.account.get_config("addr").split("@")[0],) ): return False # all relay groups' names begin with a [tag] with the localpart of the teamsbot's address if ( chat.get_messages()[0].get_sender_contact() != self.account.get_self_contact() ): return False # all relay groups were started by the teamsbot if chat.is_protected(): return False # relay groups don't need to be protected, so they are not for crew_member in self.account.get_chat_by_id( get_crew_id(self.account) ).get_contacts(): if crew_member not in chat.get_contacts(): return False # all crew members have to be in any relay group return True