def __init__(self):
super().__init__()
self.client = None
self.log_rollup = {name: list() for name in config.get_server_names()}
start_date = datetime.datetime.now().replace(
minute=max(0,
datetime.datetime.now().minute - 3))
self.last_log_time = {
name: start_date
for name in config.get_server_names()
}
self.last_player_count_update = {
name: start_date
for name in config.get_server_names()
}
self.player_counts = {name: '' for name in config.get_server_names()}
async def start_service_managers():
await discord_service.get_service_manager().start()
await member_count.get_service_manager().start()
if config.get().cftools_application_id is not None:
await omega_service.get_service_manager().start()
for server_name in config.get_server_names():
asyncio.create_task(start_server(server_name))
async def flush_log(self):
await self.client.wait_until_ready()
for server_name in config.get_server_names():
if self.log_rollup[server_name] and datetime.timedelta(seconds=10) < \
datetime.datetime.now() - self.last_log_time[server_name]:
for message in build_fields(server_name,
self.log_rollup[server_name]):
await self.send_rolled_log(server_name, message)
self.last_log_time[server_name] = datetime.datetime.now()
self.log_rollup[server_name] = list()
async def on_message(self, message: discord.Message):
if message.author == self.user:
return
for server_name in config.get_server_names():
if config.get_server(server_name).rcon_password is not None:
if message.channel.id == config.get_server(
server_name).chat_channel_id:
await arguments.process_chat(server_name, message)
elif message.channel.id == config.get_server(server_name).admin_channel_id \
and message.content.startswith('--'):
args = shlex.split(message.content, comments=True)
try:
parsed_args, remaining_args = arguments.message_parser.parse_known_args(
args)
except (ValueError, argparse.ArgumentError):
log.info(f'invalid command {message.content}')
return
if remaining_args == args:
log.info(f'invalid command {message.content}')
asyncio.create_task(
discord_service.get_service_manager().send_message(
discord_service.Response(
server_name,
f'invalid command\n`{message.content}`')))
else:
await arguments.process_message_args(
server_name, parsed_args, message)
if message.channel.id in config.get(
).user_channel_ids and message.content.startswith('--'):
args = shlex.split(message.content, comments=True)
try:
parsed_args, remaining_args = arguments.user_message_parser.parse_known_args(
args)
if remaining_args == args:
return
except (ValueError, argparse.ArgumentError):
log.info(f'invalid command {message.content}')
asyncio.create_task(
discord_service.get_service_manager().send_message(
discord_service.UserResponse(
message.channel.id,
f'invalid command\n`{message.content}`')))
return
await arguments.process_user_message_args(message.channel.id,
parsed_args)
for custom_command in config.get().custom_commands:
custom_command: message_builder.Response = custom_command
if custom_command.enabled:
if message.channel.id in custom_command.channels or len(
custom_command.channels) == 0:
if re.match(custom_command.command, message.content):
embed = custom_command.generate()
if len(embed) <= 6000:
await message.channel.send(embed=embed)
else:
await message.channel.send(
f'Message longer than 6000 character limit: {len(embed)}'
)