igncore/core/public_channel_service.py

from core.aochat import server_packets
from core.aochat.BaseModule import BaseModule
from core.conn import Conn
from core.decorators import instance
from core.logger import Logger
from core.setting_service import SettingService
from core.setting_types import NumberSettingType, TextSettingType


@instance()
class PublicChannelService(BaseModule):
    ORG_CHANNEL_MESSAGE_EVENT = "org_channel_message"
    ORG_MSG_EVENT = "org_msg"

    ORG_MSG_CHANNEL_ID = 42949672961

    def __init__(self):
        self.logger = Logger(__name__)
        self.name_to_id = {}
        self.id_to_name = {}
        self.org_channel_id = None
        self.org_id = None
        self.org_name = None

    def inject(self, registry):
        self.bot = registry.get_instance("bot")
        self.event_service = registry.get_instance("event_service")
        self.character_service = registry.get_instance("character_service")
        self.setting_service: SettingService = registry.get_instance("setting_service")

    def pre_start(self):
        self.bot.register_packet_handler(server_packets.PublicChannelJoined.id, self.add)
        self.bot.register_packet_handler(server_packets.PublicChannelLeft.id, self.remove)
        # priority must be above that of CommandService in order for relaying of commands to work correctly
        self.bot.register_packet_handler(server_packets.PublicChannelMessage.id, self.public_channel_message,
                                         priority=30)
        self.event_service.register_event_type(self.ORG_CHANNEL_MESSAGE_EVENT)
        self.event_service.register_event_type(self.ORG_MSG_EVENT)
        self.setting_service.register('core.system', 'org_id', 0,
                                      NumberSettingType(), 'OrgID used for roster')
        self.setting_service.register('core.system', 'org_name', "",
                                      TextSettingType(allow_empty=True), 'OrgName used for roster')

    def start(self):
        org_id_setting = self.setting_service.get("org_id")
        if org_id_setting and org_id_setting.get_value() and org_id_setting.get_value() != "0":
            self.org_id = org_id_setting.get_value()

        org_name_setting = self.setting_service.get("org_name")
        if org_name_setting and org_name_setting.get_value():
            self.org_name = org_name_setting.get_value()

    def get_channel_id(self, channel_name):
        return self.name_to_id.get(channel_name)

    def get_channel_name(self, channel_id):
        return self.id_to_name.get(channel_id, None)

    def add(self, conn: Conn, packet: server_packets.PublicChannelJoined):
        if conn.id != "main":
            return

        self.id_to_name[packet.channel_id] = packet.name
        self.name_to_id[packet.name] = packet.channel_id
        if not self.org_id and self.is_org_channel_id(packet.channel_id):
            self.org_channel_id = packet.channel_id
            self.org_id = 0x00ffffffff & packet.channel_id
            if packet.name != "Clan (name unknown)":
                self.setting_service.get("org_name").set_value(packet.name)
                self.org_name = packet.name
            else:
                data = self.event_service.db.query_single('SELECT org_name from all_orgs where org_id=?', [self.org_id])
                self.org_name = data.org_name if data else 'Unknown Org'
            self.logger.info(f"Org Id: {self.org_id:d}")
            self.logger.info(f"Org Name: {self.org_name}")

    def remove(self, conn: Conn, packet: server_packets.PublicChannelLeft):
        if conn.id != "main":
            return

        channel_name = self.get_channel_name(packet.channel_id)
        del self.id_to_name[packet.channel_id]
        del self.name_to_id[channel_name]

    def public_channel_message(self, conn: Conn, packet: server_packets.PublicChannelMessage):
        if conn.id != "main":
            return

        if self.is_org_channel_id(packet.channel_id):
            if self.setting_service.get_value("log_org") == "1" and packet.char_id == self.bot.get_char_id():
                char_name = self.character_service.get_char_name(packet.char_id)
                if packet.extended_message:
                    message = packet.extended_message.get_message()
                else:
                    message = packet.message
                self.logger.log_chat(conn.id, "Org Channel", char_name, message)
            self.event_service.fire_event(self.ORG_CHANNEL_MESSAGE_EVENT, packet)
        elif packet.channel_id == self.ORG_MSG_CHANNEL_ID:
            if self.setting_service.get_value("log_org") == "1" and packet.char_id == self.bot.get_char_id():
                char_name = self.character_service.get_char_name(packet.char_id)
                if packet.extended_message:
                    message = packet.extended_message.get_message()
                else:
                    message = packet.message
                self.logger.log_chat(conn.id, "Org Msg", char_name, message)
            self.event_service.fire_event(self.ORG_MSG_EVENT, packet)

    def is_org_channel_id(self, channel_id):
        return channel_id >> 32 == 3

    def get_org_id(self):
        return self.org_id

    def get_org_name(self):
        return self.org_name

    def get_all_public_channels(self):
        return self.id_to_name