Initial Release of IGNCore version 2.5

2021-08-09 13:18:56 +02:00
commit a83d98c47e
910 changed files with 224171 additions and 0 deletions
@@ -0,0 +1,135 @@
+import re
+
+from core.aochat import server_packets, client_packets
+from core.conn import Conn
+from core.decorators import instance
+from core.logger import Logger
+from core.lookup.character_service import CharacterService
+from core.setting_service import SettingService
+from core.setting_types import TextSettingType, BooleanSettingType
+from core.text import Text
+from core.tyrbot import Tyrbot
+
+
+# noinspection DuplicatedCode
+@instance("AllianceRelayController")
+class AllianceRelayController:
+    relay_channel_id = None
+    MESSAGE_SOURCE = "alliance"
+
+    def __init__(self):
+        self.logger = Logger(__name__)
+
+    def inject(self, registry):
+        self.bot: Tyrbot = registry.get_instance("bot")
+        self.setting_service: SettingService = registry.get_instance("setting_service")
+        self.character_service: CharacterService = registry.get_instance("character_service")
+        self.message_hub_service = registry.get_instance("message_hub_service")
+        self.public_channel_service = registry.get_instance("public_channel_service")
+        self.text: Text = registry.get_instance("text")
+
+    def pre_start(self):
+        self.message_hub_service.register_message_source(self.MESSAGE_SOURCE)
+
+    def start(self):
+        self.setting_service.register_new(self.module_name, "arelay_symbol", "#",
+                                          TextSettingType(["!", "#", "*", "@", "$", "+", "-"]),
+                                          "Symbol for external relay")
+
+        self.setting_service.register_new(self.module_name, "arelay_symbol_method", "with_symbol",
+                                          TextSettingType(["Always", "with_symbol", "unless_symbol"]),
+                                          "When to relay messages")
+
+        self.setting_service.register_new(self.module_name, "arelaybot", "",
+                                          TextSettingType(allow_empty=True),
+                                          "Bot for alliance relay")
+
+        self.setting_service.register_new(self.module_name, "arelay_enabled", False,
+                                          BooleanSettingType(),
+                                          "Enable the alliance relay")
+
+        self.setting_service.register_new(self.module_name, "arelay_guild_abbreviation", "",
+                                          TextSettingType(allow_empty=True),
+                                          "Abbreviation to use for org name")
+
+        self.message_hub_service.register_message_destination(self.MESSAGE_SOURCE,
+                                                              self.handle_relay_hub_message,
+                                                              ["org_channel"],
+                                                              [self.MESSAGE_SOURCE])
+
+        self.bot.register_packet_handler(server_packets.PrivateChannelInvited.id, self.handle_private_channel_invite,
+                                         100)
+        self.bot.register_packet_handler(server_packets.PrivateChannelMessage.id, self.handle_private_channel_message)
+
+    def handle_private_channel_invite(self, conn: Conn, packet: server_packets.PrivateChannelInvited):
+        if conn.id != "main":
+            return
+
+        if not self.setting_service.get("arelay_enabled").get_value():
+            return
+
+        channel_name = self.character_service.get_char_name(packet.private_channel_id)
+        if self.setting_service.get_value("arelaybot").lower() == channel_name.lower():
+            self.bot.send_packet(client_packets.PrivateChannelJoin(packet.private_channel_id))
+            self.logger.info("Joined private channel {channel}".format(channel=channel_name))
+            self.relay_channel_id = packet.private_channel_id
+
+    def handle_private_channel_message(self, conn: Conn, packet: server_packets.PrivateChannelMessage):
+        if conn.id != "main":
+            return
+
+        if not self.setting_service.get("arelay_enabled").get_value():
+            return
+
+        # ignore packets from the bot's own private channel and from the bot itself
+        if packet.private_channel_id == self.bot.get_char_id() or packet.char_id == self.bot.get_char_id():
+            return
+
+        message = packet.message.lstrip()
+        if message[:6] != "!agcr ":
+            return
+        formatted_message = None
+        message = re.match("!agcr \[(.+)] (.+): (.+)", message)
+        if message:
+            org, name, text = message.groups()
+            formatted_message = self.setting_service.get('alliance_base').format_text(
+                f"[{self.setting_service.get('alliance_org').format_text(org)}] "
+                f"{self.setting_service.get('alliance_sender').format_text(name)}: "
+                f"{self.setting_service.get('alliance_msg').format_text(text)}")
+            # sender is not the bot that sent it, but rather the original char that sent the message
+            # given the format of !agcr messages, it could be possible to parse the sender for the message
+            # but currently this is not done
+            sender = None
+
+            self.message_hub_service.send_message(self.MESSAGE_SOURCE,
+                                                  sender, None,
+                                                  formatted_message or packet.message.lstrip)
+
+    def handle_relay_hub_message(self, ctx):
+        if not self.setting_service.get("arelay_enabled").get_value():
+            return
+        method = self.setting_service.get_value("arelay_symbol_method")
+        symbol = self.setting_service.get_value("arelay_symbol")
+        plain_msg = ctx.message or ctx.formatted_message
+
+        if method == "unless_symbol" and len(plain_msg) > len(symbol) and plain_msg[:len(symbol)] == symbol:
+            return
+        elif method == "with_symbol":
+            if len(plain_msg) < len(symbol) or plain_msg[:len(symbol)] != symbol:
+                return
+            else:
+                # trim symbol from message
+                plain_msg = plain_msg[len(symbol):]
+
+        org = self.setting_service.get_value("arelay_guild_abbreviation") or \
+              self.public_channel_service.get_org_name() or \
+              self.bot.get_char_name()
+        msg = f"[{org + (' - Priv' if ctx.source == 'private_channel' else '')}] {ctx.sender.name}: {plain_msg}"
+
+        self.send_message_to_alliance(msg)
+
+    def send_message_to_alliance(self, msg):
+        if self.relay_channel_id:
+            packet = client_packets.PrivateChannelMessage(self.relay_channel_id,
+                                                          "!agcr " + self.text.format_message(msg, False), "\0")
+            self.bot.conns["main"].send_packet(packet)