get rid of the MessageDistributor module... & update discord, to work with API v10

Added discord commands (issue: as they're running over the event hub, they're processed on the same track as other events. => activity ingame triggers the next run; otherwise there's some delay for responses)
relay is a standard module now.
This commit is contained in:
2022-04-15 17:05:30 +02:00
parent fd84d82af2
commit a3a26f2ba4
32 changed files with 1036 additions and 729 deletions
@@ -1,137 +1,137 @@
import asyncio
import datetime
import html
# noinspection PyPackageRequirements
from discord import Embed
from core.command_service import CommandService
from core.db import DB
from core.decorators import instance
from core.logger import Logger
from core.lookup.character_service import CharacterService
from core.lookup.pork_service import PorkService
from core.message_hub_service import MessageHubService
from core.registry import Registry
from core.setting_service import SettingService
from core.setting_types import NumberSettingType
from core.text import Text
from core.igncore import IgnCore
from modules.core.accounting.services.access_service import AccessService
from modules.core.discord.discord_controller import DiscordController
# noinspection DuplicatedCode,PyAttributeOutsideInit
@instance()
class MessageDistributor:
MESSAGE_SOURCE = "discord"
def __init__(self):
self.logger = Logger(__name__)
def inject(self, registry):
self.bot: IgnCore = registry.get_instance("bot")
self.db: DB = registry.get_instance("db")
self.character_service: CharacterService = registry.get_instance("character_service")
self.access_service: AccessService = registry.get_instance("access_service")
self.pork: PorkService = registry.get_instance("pork_service")
self.relay_hub_service: MessageHubService = registry.get_instance("message_hub_service")
self.discord: DiscordController = registry.get_instance("discord_controller")
self.cmd_service: CommandService = registry.get_instance("command_service")
self.setting_service: SettingService = registry.get_instance("setting_service")
self.text: Text = registry.get_instance("text")
def pre_start(self):
self.relay_hub_service.register_message_source("public_relay")
self.relay_hub_service.register_message_source("tell_logger")
self.relay_hub_service.register_message_source("system_logger")
self.relay_hub_service.register_message_source("access_denied_logger")
self.relay_hub_service.register_message_source("member_logger")
def start(self):
self.relay_hub_service.register_message_destination("tell_log", self.handle_logging_tell, ["tell_logger"],
["tell_log"])
self.setting_service.register(self.module_name, "dc_tell_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Tell Log")
self.relay_hub_service.register_message_destination("access_denied_log", self.handle_logging_denied,
["access_denied_logger"], ["access_denied_log"])
self.setting_service.register(self.module_name, "dc_denied_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Access Denied Log")
self.relay_hub_service.register_message_destination("relay_log", self.handle_logging_relay,
["alliance", "public_relay"], ["relay_log"])
self.setting_service.register(self.module_name, "dc_relay_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Relay Log")
self.relay_hub_service.register_message_destination("public_relay", self.handle_public_relay, ["alliance"],
["public_relay"])
self.setting_service.register(self.module_name, "dc_relay_public", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Public Relay Channel")
self.relay_hub_service.register_message_destination("system", self.handle_logging_system, ["system_logger"],
["system"])
self.setting_service.register(self.module_name, "dc_system_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the System Channel")
self.relay_hub_service.register_message_destination("dc_member_log", self.handle_logging_members,
["member_logger"], ["dc_member_log"])
self.setting_service.register(self.module_name, "dc_member_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Member Log")
if Registry.get_instance("darknet", is_optional=True):
self.relay_hub_service.register_message_destination("dc_darknet_log", self.handle_darknet_log, ["darknet"],
[""])
self.setting_service.register(self.module_name, "dc_darknet_log", 0, NumberSettingType(allow_empty=True),
"ChannelID for the Darknet Log")
def handle_public_relay(self, ctx):
if self.setting_service.get_value("dc_relay_public") != "0":
message = self.format(ctx.formatted_message)
self.discord.send_message(self.setting_service.get_value("dc_relay_public"), html.unescape(message),
delete=3600) # delete after 1h
def handle_logging_tell(self, ctx):
if self.setting_service.get_value("dc_tell_log") != "0":
message = self.format(ctx.message)
self.discord.send_message(self.setting_service.get_value("dc_tell_log"),
f"[FROM] {self.character_service.get_char_name(ctx.sender)}: {message}")
def handle_logging_denied(self, ctx):
if self.setting_service.get_value("dc_denied_log") != "0":
message = self.format(ctx.message)
self.discord.send_message(self.setting_service.get_value("dc_denied_log"), message)
def handle_darknet_log(self, ctx):
if self.setting_service.get_value("dc_darknet_log") != "0":
message = self.format(ctx.message)
self.discord.send_message(self.setting_service.get_value("dc_darknet_log"), message)
def handle_logging_relay(self, ctx):
if self.setting_service.get_value("dc_relay_log") != "0":
message = self.format(ctx.formatted_message)
self.discord.send_message(self.setting_service.get_value("dc_relay_log"), f"{html.unescape(message)}")
def handle_logging_system(self, ctx):
if self.setting_service.get_value("dc_system_log") != "0":
message = self.format(ctx.message)
self.discord.send_message(self.setting_service.get_value("dc_system_log"), f"{html.unescape(message)}")
def handle_logging_members(self, ctx):
if self.setting_service.get_value("dc_member_log") != "0":
channel = self.discord.client.get_channel(int(self.setting_service.get_value("dc_member_log")))
spam = []
current = 0
for message in ctx.message:
current += 1
spam.append(Embed(title=f"Recent changes - {datetime.date.today()} - ({current}/{len(ctx.message)})",
description=message, color=3066993))
asyncio.run_coroutine_threadsafe(self.send_spam(channel, spam), self.discord.loop)
async def send_spam(self, channel, message):
if self.discord.client.is_ready():
for i in message:
await channel.send(embed=i)
def format(self, message):
return self.text.strip_html_tags(message)
# import asyncio
# import datetime
# import html
#
# # noinspection PyPackageRequirements
# from discord import Embed
#
# from core.command_service import CommandService
# from core.db import DB
# from core.decorators import instance
# from core.logger import Logger
# from core.lookup.character_service import CharacterService
# from core.lookup.pork_service import PorkService
# from core.message_hub_service import MessageHubService
# from core.registry import Registry
# from core.setting_service import SettingService
# from core.setting_types import NumberSettingType
# from core.text import Text
# from core.igncore import IgnCore
# from modules.core.accounting.services.access_service import AccessService
# from modules.core.discord.discord_controller import DiscordController
#
#
# # noinspection DuplicatedCode,PyAttributeOutsideInit
# @instance()
# class MessageDistributor:
# MESSAGE_SOURCE = "discord"
#
# def __init__(self):
# self.logger = Logger(__name__)
#
# def inject(self, registry):
# self.bot: IgnCore = registry.get_instance("bot")
# self.db: DB = registry.get_instance("db")
# self.character_service: CharacterService = registry.get_instance("character_service")
# self.access_service: AccessService = registry.get_instance("access_service")
# self.pork: PorkService = registry.get_instance("pork_service")
# self.relay_hub_service: MessageHubService = registry.get_instance("message_hub_service")
# self.discord: DiscordController = registry.get_instance("discord_controller")
# self.cmd_service: CommandService = registry.get_instance("command_service")
# self.setting_service: SettingService = registry.get_instance("setting_service")
# self.text: Text = registry.get_instance("text")
#
# def pre_start(self):
# self.relay_hub_service.register_message_source("public_relay")
# self.relay_hub_service.register_message_source("tell_logger")
# self.relay_hub_service.register_message_source("system_logger")
# self.relay_hub_service.register_message_source("access_denied_logger")
# self.relay_hub_service.register_message_source("member_logger")
#
# def start(self):
# self.relay_hub_service.register_message_destination("tell_log", self.handle_logging_tell, ["tell_logger"],
# ["tell_log"])
# self.setting_service.register(self.module_name, "dc_tell_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Tell Log")
#
# self.relay_hub_service.register_message_destination("access_denied_log", self.handle_logging_denied,
# ["access_denied_logger"], ["access_denied_log"])
# self.setting_service.register(self.module_name, "dc_denied_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Access Denied Log")
#
# self.relay_hub_service.register_message_destination("relay_log", self.handle_logging_relay,
# ["alliance", "public_relay"], ["relay_log"])
# self.setting_service.register(self.module_name, "dc_relay_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Relay Log")
#
# self.relay_hub_service.register_message_destination("public_relay", self.handle_public_relay, ["alliance"],
# ["public_relay"])
# self.setting_service.register(self.module_name, "dc_relay_public", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Public Relay Channel")
#
# self.relay_hub_service.register_message_destination("system", self.handle_logging_system, ["system_logger"],
# ["system"])
# self.setting_service.register(self.module_name, "dc_system_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the System Channel")
#
# self.relay_hub_service.register_message_destination("dc_member_log", self.handle_logging_members,
# ["member_logger"], ["dc_member_log"])
# self.setting_service.register(self.module_name, "dc_member_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Member Log")
#
# if Registry.get_instance("darknet", is_optional=True):
# self.relay_hub_service.register_message_destination("dc_darknet_log", self.handle_darknet_log, ["darknet"],
# [""])
# self.setting_service.register(self.module_name, "dc_darknet_log", 0, NumberSettingType(allow_empty=True),
# "ChannelID for the Darknet Log")
#
# def handle_public_relay(self, ctx):
# if self.setting_service.get_value("dc_relay_public") != "0":
# message = self.format(ctx.formatted_message)
# self.discord.send_message(self.setting_service.get_value("dc_relay_public"), html.unescape(message),
# delete=3600) # delete after 1h
#
# def handle_logging_tell(self, ctx):
# if self.setting_service.get_value("dc_tell_log") != "0":
# message = self.format(ctx.message)
# self.discord.send_message(self.setting_service.get_value("dc_tell_log"),
# f"[FROM] {self.character_service.get_char_name(ctx.sender)}: {message}")
#
# def handle_logging_denied(self, ctx):
# if self.setting_service.get_value("dc_denied_log") != "0":
# message = self.format(ctx.message)
# self.discord.send_message(self.setting_service.get_value("dc_denied_log"), message)
#
# def handle_darknet_log(self, ctx):
# if self.setting_service.get_value("dc_darknet_log") != "0":
# message = self.format(ctx.message)
# self.discord.send_message(self.setting_service.get_value("dc_darknet_log"), message)
#
# def handle_logging_relay(self, ctx):
# if self.setting_service.get_value("dc_relay_log") != "0":
# message = self.format(ctx.formatted_message)
# self.discord.send_message(self.setting_service.get_value("dc_relay_log"), f"{html.unescape(message)}")
#
# def handle_logging_system(self, ctx):
# if self.setting_service.get_value("dc_system_log") != "0":
# message = self.format(ctx.message)
# self.discord.send_message(self.setting_service.get_value("dc_system_log"), f"{html.unescape(message)}")
#
# def handle_logging_members(self, ctx):
# if self.setting_service.get_value("dc_member_log") != "0":
# channel = self.discord.client.get_channel(int(self.setting_service.get_value("dc_member_log")))
# spam = []
# current = 0
# for message in ctx.message:
# current += 1
# spam.append(Embed(title=f"Recent changes - {datetime.date.today()} - ({current}/{len(ctx.message)})",
# description=message, color=3066993))
# asyncio.run_coroutine_threadsafe(self.send_spam(channel, spam), self.discord.loop)
#
# async def send_spam(self, channel, message):
# if self.discord.client.is_ready():
# for i in message:
# await channel.send(embed=i)
#
# def format(self, message):
# return self.text.strip_html_tags(message)
+489
View File
@@ -0,0 +1,489 @@
import re
import typing
from core.aochat import server_packets, client_packets
from core.aochat.client_packets import PrivateChannelLeave
from core.chat_blob import ChatBlob
from core.command_param_types import Const, Character, Options, Any
from core.conn import Conn
from core.decorators import instance, command, setting
from core.dict_object import DictObject
from core.logger import Logger
from core.lookup.character_service import CharacterService
from core.setting_service import SettingService
from core.setting_types import DictionarySettingType
from core.text import Text
from core.igncore import IgnCore
# noinspection DuplicatedCode
if typing.TYPE_CHECKING:
from modules.core.discord.discord_controller import DiscordController
@instance()
class AllianceRelay:
MESSAGE_SOURCE = "alliance"
def __init__(self):
self.logger = Logger(__name__)
self.relay_channel = []
def inject(self, registry):
self.bot: IgnCore = 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.discord: DiscordController = registry.get_instance("discord_controller")
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.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)
@command(command="mrelay", params=[], access_level="member",
description="View the relay settings", sub_command="info")
def mrelay(self, request):
def display_color(color, msg=None):
if not msg:
msg = color
if color:
return f"<font color={color}>{msg}</font>"
return "UNSET"
blob = ""
for bot in self.relay_bots().get_value().keys():
enabled = self.relay_enabled().get_value().get(bot,
self.relay_enabled().get_value().get('default', 'UNSET'))
prefix = self.relay_symbols().get_value().get(bot, self.relay_symbols().get_value().get('default'))
relay_cmd = self.relay_command().get_value().get(bot, self.relay_command().get_value().get('default'))
base = self.relay_color_base().get_value().get(bot,
self.relay_color_base().get_value().get('default', None))
sender = self.relay_color_sender().get_value().get(bot, self.relay_color_sender().get_value().get('default',
None))
abbrv = self.relay_color_org().get_value().get(bot, self.relay_color_org().get_value().get('default', None))
msg = self.relay_color_msg().get_value().get(bot, self.relay_color_msg().get_value().get('default', None))
our_abbrv = self.relay_guild_abbreviations().get_value().get(bot,
self.relay_guild_abbreviations().get_value().get(
'default', 'UNSET'))
example = display_color(base,
f'[{display_color(abbrv, our_abbrv)}] {display_color(sender, request.sender.name)}: {display_color(msg, "And I sent you a message")}')
blob += f"Relay <highlight>{self.character_service.get_char_name(int(bot))}</highlight>:\n"
blob += f" Enabled: {enabled} [{self.text.make_tellcmd('Toggle', f'mrelay status {bot} {not enabled}')}]\n"
blob += f" Prefix: {prefix} [{self.text.make_tellcmd('Edit', f'mrelay prefix {bot}')}]\n"
blob += f" Relay command: {relay_cmd} [{self.text.make_tellcmd('Edit', f'mrelay rcmd {bot}')}]\n"
blob += f" Base color: {display_color(base)} [{self.text.make_tellcmd('Edit', f'mrelay color {bot} base')}]\n"
blob += f" Abbrv color: {display_color(abbrv)} [{self.text.make_tellcmd('Edit', f'mrelay color {bot} org')}]\n"
blob += f" Sender color: {display_color(sender)} [{self.text.make_tellcmd('Edit', f'mrelay color {bot} sender')}]\n"
blob += f" Message color: {display_color(msg)} [{self.text.make_tellcmd('Edit', f'mrelay color {bot} msg')}]\n"
blob += f" Abbreviation: {our_abbrv} [Use <highlight>!mrelay abbrv {bot} &lt;your Abbreviation&gt;</highlight>]\n"
blob += f" Example msg: {example}"
blob += "\n\n"
return ChatBlob("Current Relays", blob)
@command(command="mrelay", params=[Const("create"), Character("relaybot_name")], access_level="admin",
description="Create a new Relay")
def mrelay_create(self, request, _, bot):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if bot.char_id in bots.keys():
return f"There's already a relay for <highlight>{bot.name}</highlight>."
bots[bot.char_id] = ""
self.relay_bots().set_value(bots)
return f"Successfully created relay <highlight>{bot.name}</highlight>. Use <symbol>mrelay to edit it."
@command(command="mrelay", params=[Const("delete"), Character("relaybot_name")], access_level="admin",
description="Delete a Relay")
def mrelay_delete(self, request, _, bot):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight>."
bots.pop(str(bot.char_id))
self.bot.send_packet(PrivateChannelLeave(bot.char_id))
self.relay_bots().set_value(bots)
user = str(bot.char_id)
cur = self.relay_enabled().get_value()
cur.pop(user, None)
self.relay_enabled().set_value(cur)
cur = self.relay_command().get_value()
cur.pop(user, None)
self.relay_command().set_value(cur)
cur = self.relay_bots().get_value()
cur.pop(user, None)
self.relay_bots().set_value(cur)
cur = self.relay_guild_abbreviations().get_value()
cur.pop(user, None)
self.relay_guild_abbreviations().set_value(cur)
cur = self.relay_color_base().get_value()
cur.pop(user, None)
self.relay_color_base().set_value(cur)
cur = self.relay_color_msg().get_value()
cur.pop(user, None)
self.relay_color_msg().set_value(cur)
cur = self.relay_color_sender().get_value()
cur.pop(user, None)
self.relay_color_sender().set_value(cur)
cur = self.relay_color_org().get_value()
cur.pop(user, None)
self.relay_color_msg().set_value(cur)
cur = self.relay_symbols().get_value()
cur.pop(user, None)
self.relay_symbols().set_value(cur)
cur = self.relay_symbol_methods().get_value()
cur.pop(user, None)
self.relay_symbol_methods().set_value(cur)
return f"Successfully deleted relay <highlight>{bot.name}</highlight>"
@command(command="mrelay",
params=[Const("rcmd"), Character("relaybot_name"), Any("relay_command", is_optional=True)],
access_level="admin",
description="Change the relay command used in a relay")
def mrelay_rcmd(self, request, _, bot, relay_command):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight> registered. Please use <symbol>mrelay create {bot.name}"
bot = str(bot.char_id)
current = self.relay_command().get_value()
if not relay_command:
blob = f"Current relay command: {current.get(bot, current.get('default', 'UNSET'))}\n\n"
blob += "Here are some presets:\n"
for rcmd in ["!agrc", "!gcr", "agrc", "gcr"]:
blob += f" {self.text.make_tellcmd(rcmd, f'mrelay rcmd {bot} {rcmd}')}\n"
blob += "\n"
blob += f"Or you can use <highlight><symbol>mrelay rcmd {self.character_service.get_char_name(int(bot))} &lt;Your Prefix&gt;</highlight> to set a custom one."
return ChatBlob("Pick your relay command", blob)
current[bot] = relay_command
self.relay_command().set_value(current)
return f"Successfully changed the relaying command for relaying messages to " \
f"<highlight>{self.character_service.get_char_name(int(bot))}</highlight>: <highlight>{relay_command}</highlight>"
@command(command="mrelay",
params=[Const("prefix"), Character("relaybot_name"), Any("prefix", is_optional=True)],
access_level="admin",
description="Change the prefix setting of the relays")
def mrelay_prefix(self, request, _, bot, prefix):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight> registered. Please use <symbol>mrelay create {bot.name}"
bot = str(bot.char_id)
current = self.relay_symbols().get_value()
if not prefix:
blob = f"Current prefix: {current.get(bot, current.get('default', 'UNSET'))}\n\n"
blob += "Here are some presets:\n"
for prefix in ["-", "--", "+", "#", "*", "@", "$", "$$"]:
blob += f" {self.text.make_tellcmd(prefix, f'mrelay prefix {bot} {prefix}')}\n"
blob += "\n"
blob += f"Or you can use <highlight><symbol>mrelay prefix {self.character_service.get_char_name(int(bot))} &lt;Your Prefix&gt;</highlight> to set a custom one."
return ChatBlob("Pick your prefix", blob)
current[bot] = prefix
self.relay_symbols().set_value(current)
return f"Successfully changed the symbol for relaying messages to " \
f"<highlight>{self.character_service.get_char_name(int(bot))}</highlight>: <highlight>{prefix}</highlight>"
@command(command="mrelay",
params=[Const("abbrv"), Character("relaybot_name"), Any("Abbreviation")],
access_level="admin",
description="Change the abbreviation setting of the relays")
def mrelay_abbrv(self, request, _, bot, abbreviation):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight> registered. Please use <symbol>mrelay create {bot.name}"
bot = str(bot.char_id)
current = self.relay_guild_abbreviations().get_value()
current[bot] = abbreviation
self.relay_guild_abbreviations().set_value(current)
return f"Successfully changed the Abbreviation for relaying messages to " \
f"<highlight>{self.character_service.get_char_name(int(bot))}</highlight>: <highlight>{abbreviation}</highlight>"
@command(command="mrelay",
params=[Const("status"), Character("relaybot_name"), Options(["on", "off", "true", "false"])],
access_level="admin",
description="Change the enabled status of the relays")
def mrelay_status(self, request, _, bot, option):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight> registered. Please use <symbol>mrelay create {bot.name}"
bot = str(bot.char_id)
current = self.relay_enabled().get_value()
option = True if option.lower() in ["on", "true"] else False
if current.get(bot, None) == option:
return f"The relay status of {self.character_service.get_char_name(int(bot))} is already {option}"
current[bot] = option
self.relay_enabled().set_value(current)
return f"Successfully changed the status of the relay " \
f"<highlight>{self.character_service.get_char_name(int(bot))}</highlight> to: <highlight>{option}</highlight>"
@command(command="mrelay",
params=[Const("color"), Character("relaybot_name"), Options(["base", "sender", "org", "msg"]),
Any("color_code", is_optional=True)], access_level="admin",
description="Change the color settings of the relays")
def mrelay_color(self, request, _, bot, option, color):
bots: dict = self.relay_bots().get_value()
if not bot.char_id:
return f"The character <highlight>{bot.name}</highlight> does not exist."
if str(bot.char_id) not in bots.keys():
return f"There's no relay for <highlight>{bot.name}</highlight> registered. Please use <symbol>mrelay create {bot.name}"
bot = str(bot.char_id)
if option == "base":
current = self.relay_color_base().get_value()
if color is None:
return ChatBlob("Pick a color", self.color_template(bot, option, current))
if self.check_color(color):
current[bot] = color
self.relay_color_base().set_value(current)
else:
return f"Your color is not valid: {color}. Syntax: #000000 or #000"
elif option == "sender":
current = self.relay_color_sender().get_value()
if color is None:
return ChatBlob("Pick a color", self.color_template(bot, option, current))
if self.check_color(color):
current[bot] = color
self.relay_color_sender().set_value(current)
else:
return f"Your color is not valid: {color}. Syntax: #000000 or #000"
elif option == "org":
current = self.relay_color_org().get_value()
if color is None:
return ChatBlob("Pick a color", self.color_template(bot, option, current))
if self.check_color(color):
current[bot] = color
self.relay_color_org().set_value(current)
else:
return f"Your color is not valid: {color}. Syntax: #000000 or #000"
elif option == "msg":
current = self.relay_color_msg().get_value()
if color is None:
return ChatBlob("Pick a color", self.color_template(bot, option, current))
if self.check_color(color):
current[bot] = color
self.relay_color_msg().set_value(current)
else:
return f"Your color is not valid: {color}. Syntax: #000000 or #000"
return f"Successfully set the <highlight>{option}</highlight> color to <font color={color}>{color}</font>" \
f" for relay <highlight>{self.character_service.get_char_name(int(bot))}</highlight>"
def color_template(self, bot, option, current):
def display(color, msg):
return f"<font color={color}>{msg}</font> (<a href='chatcmd:///tell <myname> mrelay color {bot} {option} {color}'>Save it</a>)"
blob = f"Current color: <font color={current.get(bot, current.get('default', '#00ff00'))}>" \
f"{current.get(bot, current.get('default', '#00ff00'))}</font>\n\n"
blob += f"{display('#FF0000', 'Red')}\n"
blob += f"{display('#FFFFFF', 'White')}\n"
blob += f"{display('#808080', 'Grey')}\n"
blob += f"{display('#DDDDDD', 'Light Grey')}\n"
blob += f"{display('#9CC6E7', 'Dark Grey')}\n"
blob += f"{display('#000000', 'Black')}\n"
blob += f"{display('#FFFF00', 'Yellow')}\n"
blob += f"{display('#8CB5FF', 'Blue')}\n"
blob += f"{display('#00BFFF', 'Deep Sky Blue')}\n"
blob += f"{display('#00DE42', 'Green')}\n"
blob += f"{display('#FCA712', 'Orange')}\n"
blob += f"{display('#FFD700', 'Gold')}\n"
blob += f"{display('#FF1493', 'Deep Pink')}\n"
blob += f"{display('#EE82EE', 'Violet')}\n"
blob += f"{display('#8B7355', 'Brown')}\n"
blob += f"{display('#00FFFF', 'Cyan')}\n"
blob += f"{display('#000080', 'Navy Blue')}\n"
blob += f"{display('#FF8C00', 'Dark Orange')}\n"
return blob
def handle_private_channel_invite(self, conn: Conn, packet: server_packets.PrivateChannelInvited):
if conn.id != "main":
return
if str(packet.private_channel_id) not in self.relay_bots().get_value().keys():
return
if not self.relay_enabled().get_value().get(str(packet.private_channel_id),
self.relay_enabled().get_value().get('default', False)):
return
self.bot.send_packet(client_packets.PrivateChannelJoin(packet.private_channel_id))
self.logger.info(f"Joined private channel {self.character_service.get_char_name(packet.private_channel_id)}")
self.relay_channel.append(str(packet.private_channel_id))
def handle_private_channel_message(self, conn: Conn, packet: server_packets.PrivateChannelMessage):
if conn.id != "main":
return
if packet.private_channel_id == self.bot.get_char_id():
return
if packet.char_id == self.bot.get_char_id():
return
if not self.relay_enabled().get_value().get(str(packet.private_channel_id), self.relay_enabled().get_value().get('default', False)):
return
priv = str(packet.private_channel_id)
message = packet.message.lstrip()
rcmd = self.relay_command().get_value().get(priv, self.relay_command().get_value().get('default', "UNSET"))
if rcmd == "UNSET":
self.logger.warning("##FIX ME## No relay command set!")
if message[:len(rcmd)] != rcmd:
return
message = message.replace("\n", "<br>")
message = re.match(f"{rcmd} \[(.+?)\] (.+?): (.+)", message)
if not message:
return
org, name, text = message.groups()
org = org.strip()
name = name.strip()
text = text.strip()
plain = f"[{org}] {name}: {text}"
org = self.format_text(
self.relay_color_org().get_value().get(priv, self.relay_color_org().get_value().get("default")), org)
name = self.format_text(
self.relay_color_sender().get_value().get(priv, self.relay_color_sender().get_value().get("default")), name)
text = self.format_text(
self.relay_color_msg().get_value().get(priv, self.relay_color_msg().get_value().get("default")), text)
formatted = self.format_text(
self.relay_color_base().get_value().get(priv, self.relay_color_base().get_value().get("default")),
f"[{org}] {name}: {text}")
sender = DictObject({"char_id": packet.char_id, "name": self.character_service.get_char_name(packet.char_id)})
self.message_hub_service.send_message(self.MESSAGE_SOURCE, sender, self.text.strip_html_tags(plain), formatted)
def handle_relay_hub_message(self, ctx):
enabled = lambda x: self.relay_enabled().get_value().get(x,
self.relay_enabled().get_value().get('default', False))
if not ctx.message:
return
if not ctx.sender:
return
if type(ctx.message) == ChatBlob:
return
for key in self.relay_bots().get_value().keys():
if key not in self.relay_channel:
continue
if not enabled(key):
continue
prefix = self.relay_symbols().get_value().get(key, self.relay_symbols().get_value().get('default', False))
if not(ctx.message.startswith(prefix)):
continue
sender = ctx.sender.name
message = ctx.message[len(prefix):].strip()
if len(message) < 1:
continue
abbrv = self.relay_guild_abbreviations().get_value().get(key, self.setting_service.get_value("org_name"))
if ctx.source == "private_channel":
abbrv += " - Guest"
elif ctx.source.startswith("Discord_("):
abbrv += " - DC"
cmd = self.relay_command().get_value().get(key, self.relay_command().get_value().get("default"))
# msg = f"{cmd} [{abbrv}] {sender}: {message}"
blob = ""
if ctx.source.startswith("Discord_("):
if self.discord.alias_usage().get_value():
abbrv = self.discord.alias_controller.get_alias(ctx.sender.org_id)
invite = self.text.make_chatcmd("click here", "/tell <myname> discord invite",
style="style='text-decoration:none'")
# name = "<header2><myname></header2>"
if self.discord.guild and self.discord.bot:
name = f"<header2>DC Server: <highlight>{self.discord.guild.name}</highlight> - " \
f"Bot: <highlight><myname></highlight></header2>"
blob = self.text.format_page('Info',
f"<header>::: Information :::</header><br><br>"
f"This message has been sent to you by:<br><br>"
f"{name}<br>"
f"<tab>Sender: <notice>{ctx.sender.name}</notice><br>"
f"<tab>Tag: <highlight>{ctx.sender.discord_nick}</highlight><br><br>"
f"To reply, either respond in the relay or "
f"contact them directly at the provided handles.<br><br>"
f"<header2>Have you joined our Discord Server yet? "
f"If not <highlight>{invite}</highlight> to receive an invite.</header2>")
blob = f"<yellow>[{blob}]</yellow>"
# msg = f"{cmd} [{self.discord.alias_controller.get_alias(ctx.sender.org_id)}] {ctx.message} <yellow>[{blob}]</yellow>"
msg = f"{cmd} [{abbrv}] {sender}: {message} {blob}"
self.send_message_to_alliance(key, msg)
def send_message_to_alliance(self, alliance, msg):
packet = client_packets.PrivateChannelMessage(int(alliance), self.text.format_message(msg, False), "\0")
self.bot.conns["main"].send_packet(packet)
@setting(name="relay_symbols", value={"default": "-"}, description="Symbol for external relay")
def relay_symbols(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_symbol_methods", value={'default': "always"}, description="Relay methods for the relays")
def relay_symbol_methods(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_bots", value={}, description="Bots used for relaying")
def relay_bots(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_enabled", value={"default": True}, description="Enabled status of the relays")
def relay_enabled(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_guild_abbreviations", value={"default": "UNSET"},
description="Abbreviations used for the relays")
def relay_guild_abbreviations(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_color_base", value={"default": "#00FF00"}, description="Base color used for the relays")
def relay_color_base(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_color_org", value={"default": "#FFFF00"}, description="Abbrv color used for the relays")
def relay_color_org(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_color_sender", value={"default": "#FF8C00"}, description="Sender color used for the relays")
def relay_color_sender(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_color_msg", value={"default": "#FF8C00"}, description="Message color used for the relays")
def relay_color_msg(self) -> DictionarySettingType:
return DictionarySettingType()
@setting(name="relay_command", value={"default": "!agcr"}, description="Relay command used for the relays")
def relay_command(self) -> DictionarySettingType:
return DictionarySettingType()
def check_color(self, msg: str):
if not msg.startswith('#'):
return False
if len(msg) not in [4, 7]:
return False
return True
def format_text(self, color, message):
return f"<font color={color}>{message}</font>"
# def alias_usage(self):
# return self.setting_service.get("use_prefix")
+1 -1
View File
@@ -47,7 +47,7 @@ class TimeController:
day += "rd"
else:
day += "th"
return f"Currently its {now.hour}:{now.minute}:{now.second} " \
return f"Currently its {now.hour:02}:{now.minute:02}:{now.second:02} " \
f"{calendar.month_name[now.month]} {day}, {now.year + 27474} Rubi-Ka Universal Time."
@command(command="time", params=[Any("timezone")], access_level="member",
+9 -7
View File
@@ -1,4 +1,5 @@
import time
import typing
from datetime import datetime
from core.buddy_service import BuddyService
@@ -8,16 +9,17 @@ from core.db import DB
from core.decorators import instance, command
from core.job_scheduler import JobScheduler
from core.logger import Logger
from core.lookup.pork_service import PorkService
from core.setting_service import SettingService
from core.text import Text
from core.igncore import IgnCore
from core.util import Util
from modules.core.accounting.preference_controller import PreferenceController
from modules.core.accounting.services.account_service import AccountService
from modules.core.ban.ban_service import BanService
from modules.core.discord.discord_controller import DiscordController
from modules.onlinebot.online.org_alias_controller import OrgAliasController
if typing.TYPE_CHECKING:
from core.lookup.pork_service import PorkService
from core.setting_service import SettingService
from modules.core.accounting.preference_controller import PreferenceController
from modules.core.accounting.services.account_service import AccountService
from modules.core.ban.ban_service import BanService
from modules.core.discord.discord_controller import DiscordController
from modules.onlinebot.online.org_alias_controller import OrgAliasController
@instance()
+12 -3
View File
@@ -3,13 +3,14 @@ import time
from core.chat_blob import ChatBlob
from core.command_alias_service import CommandAliasService
from core.command_param_types import Any
from core.decorators import instance, command, event
from core.decorators import instance, command, event, setting
from core.dict_object import DictObject
from core.igncore import IgnCore
from core.job_scheduler import JobScheduler
from core.logger import Logger
from core.message_hub_service import MessageHubService
from core.setting_service import SettingService
from core.setting_types import BooleanSettingType
from core.setting_types import BooleanSettingType, TextSettingType
from core.text import Text
from core.util import Util
from modules.standard.datanet.ws_controller import WebsocketRelayController
@@ -35,8 +36,11 @@ class WorldBossController:
self.command_alias_service: CommandAliasService = registry.get_instance("command_alias_service")
self.job_scheduler: JobScheduler = registry.get_instance("job_scheduler")
self.setting_service: SettingService = registry.get_instance("setting_service")
self.relay_hub: MessageHubService = registry.get_instance("message_hub_service")
def pre_start(self):
self.relay_hub.register_message_source("timers")
self.setting_service.register(self.module_name, 'timer_spam', False, BooleanSettingType(),
"should timers be spammed")
self.command_alias_service.add_alias("tara", "wb tara")
@@ -120,7 +124,7 @@ class WorldBossController:
return DictObject({'spawn': last - immortal, 'mortal': last})
def send_warn(self, msg):
self.bot.send_private_channel_message(f"[<red>WB</red>] {msg}")
self.relay_hub.send_message("timers", None, f"[<red>WB</red>] {msg}", f"[<red>WB</red>] {msg}")
def get_next_alert(self, duration):
for alert in self.alerts:
@@ -190,3 +194,8 @@ class WorldBossController:
job['id'] = job_id
return
self.jobs.append({'name': timer.name, 'id': job_id})
# @setting('alert_times', '[480 * 60, 360 * 60, 240 * 60, 120 * 60, 60 * 60, 60 * 15, 60 * 5, 60 * 3, 60 * 2, 60, 30, 15, 5, 0]', 'Worldboss timer spam messages (ETA and actual)')
# def alert_times(self):
# return TextSettingType(['[480 * 60, 360 * 60, 240 * 60, 120 * 60, 60 * 60, 60 * 15, 60 * 5, 60 * 3, 60 * 2, 60, 30, 15, 5, 0]'])
+3 -3
View File
@@ -39,11 +39,11 @@ class OnlineDisplay:
if org > 0:
postfix.append(f"<notice>Org: {org}</notice>")
if priv > 0:
postfix.append(f"<highlight>Priv: {priv}</notice>")
postfix.append(f"<notice>Priv: {priv}</notice>")
if notify > 0:
postfix.append(f"<notice>Buddylist: {notify}</notice>")
blob = ChatBlob(title, blob, suffix=f" ({f', '.join(postfix)})")
return blob
return blob if (org+priv+notify) != 0 else "There's nobody online."
def format_by_channel_main(self, query, params):
query += "order by channel_id, main_name, p.name=main_name, p.name, r_id"
@@ -161,7 +161,7 @@ class OnlineDisplay:
if org_id != temp_blob[key]["id"]:
org_id = temp_blob[key]["id"]
blob += f"\n<pagebreak><highlight>{temp_blob[key]['name']}</highlight>" \
f"<header> ({len(temp_blob[key]['online']): >3} " \
f" <header>({len(temp_blob[key]['online']): >3} " \
f"| {len(temp_blob[key]['online']) / len(query) * 100:.2f}%)</header>\n"
blob += self.format_org(user)
return blob, 0, 0, len(query)
@@ -214,18 +214,18 @@ class SpecialsController:
blob += "<header2>Martial Artist</header2>\n"
blob += f"Speed: <highlight>{ma_info.ma_speed:.2f} / {ma_info.ma_speed:.2f} secs</highlight>\n"
blob += f"Damage: <highlight>{ma_info.ma_min_dmg:d} - {ma_info.ma_max_dmg:d} " \
f"({ma_info.ma_crit_dmg:d})</highlight>\n\n"
blob += f"Damage: <highlight>{ma_info.ma_min_dmg:.2f} - {ma_info.ma_max_dmg:.2f} " \
f"({ma_info.ma_crit_dmg:.2f})</highlight>\n\n"
blob += "<header2>Shade</header2>\n"
blob += f"Speed: <highlight>{ma_info.shade_speed:.2f} / {ma_info.shade_speed:.2f} secs</highlight>\n"
blob += f"Damage: <highlight>{ma_info.shade_min_dmg:d} - {ma_info.shade_max_dmg:d} " \
f"({ma_info.shade_crit_dmg:d})</highlight>\n\n"
blob += f"Damage: <highlight>{ma_info.shade_min_dmg:.2f} - {ma_info.shade_max_dmg:.2f} " \
f"({ma_info.shade_crit_dmg:.2f})</highlight>\n\n"
blob += "<header2>All other professions</header2>\n"
blob += f"Speed: <highlight>{ma_info.gen_speed:.2f} / {ma_info.gen_speed:.2f} secs</highlight>\n"
blob += f"Damage: <highlight>{ma_info.gen_min_dmg:d} - {ma_info.gen_max_dmg:d} " \
f"({ma_info.gen_crit_dmg:d})</highlight>\n\n"
blob += f"Damage: <highlight>{ma_info.gen_min_dmg:.2f} - {ma_info.gen_max_dmg:.2f} " \
f"({ma_info.gen_crit_dmg:.2f})</highlight>\n\n"
return ChatBlob("Martial Arts Results", blob)
@@ -108,12 +108,12 @@ class CharacterInfoController:
blob += self.alts_controller.format_alt_list(alts)
more_info = self.text.paginate_single(ChatBlob("More Info", blob))
msg = self.text.format_char_info(char_info, online_status, True) + " " + more_info
msg = ChatBlob("More Info", blob, self.text.format_char_info(char_info, online_status, True)+" ")
# msg = self.text.format_char_info(char_info, online_status, True) + " " + more_info
elif char.char_id:
blob = "<notice>Note: Could not retrieve detailed info for character.</notice>\n\n"
blob += f"Name: <highlight>{char.name}</highlight>\n"
blob += f"Character ID: <highlight>{char.char_id:d}</highlight>\n"
blob += f"Character ID: <highlight>{char.char_id}</highlight>\n"
if online_status is not None:
blob += f"Online status: {'<green>Online</green>' if online_status else '<red>Offline</red>'}\n"
blob += self.get_name_history(char.char_id)