Minidodo
9f1da9a00d
Changed the setting registration, removed the warnings. Loot roll messages are more obvious now. Superadmins are meant to stay mostily hidden, but are being exposed in !system again.
117 lines
5.0 KiB
Python
117 lines
5.0 KiB
Python
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):
|
|
# 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:
|
|
# 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
|