def func_wrapper(parent_obj):
input_user = parent_obj.input_author
input_channel = parent_obj.input_channel
user_roles = parent_obj.authenticator.get_single_user_bot_roles(input_user.id)
# print("DECORATOR auth.cmd function. User_id %d Roles: %s " % (user_id, user_roles))
# User not present?
u_acl = parent_obj.authenticator.acl
user_check = u_acl.check_any(user_roles, "command", command)
# CHECK FOR CHANNEL PERMISSION
if not isinstance(input_channel, discord.channel.DMChannel):
if len(config.ALLOWED_CHANNELS):
if input_channel.id not in config.ALLOWED_CHANNELS and not input_channel.id == input_user:
return False
if len(config.DENY_CHANNELS):
if input_channel.id in config.DENY_CHANNELS and not input_channel.id == input_user:
return False
# CHECK FOR USER PERMISSION
if not user_check and config.AUTHENTICATION:
output_content = messagecomposer.prettify(errors.error_auth(command), "BLOCK")
user_permissions = u_acl.which_permissions_any(user_roles, "command")
output_content += messagecomposer.prettify("Commands you can use: %s" % user_permissions, "BLOCK")
return [{"destination": parent_obj.input_author,
"content": output_content,
'decoration': "BLOCK"}]
return func(parent_obj)
def cmd_users(self):
output_channel = self.input_author
output_broadcast = False
if "reload" in self.lp.key_words:
self.authenticator.reload_discord_users()
return {
"destination": output_channel,
"content": messagecomposer.prettify("Users Reloaded!\n",
"CSS"),
'broadcast': output_broadcast
}
# Find the Bot_Roles
b_role = False
bot_roles_list = self.authenticator.roles.get_bot_roles_list()
regex = ""
if self.lp.param:
for role in bot_roles_list:
regex += "%s|" % role
regex = r"\b(" + regex[:-1] + r")\b"
reg = re.search(regex, self.lp.param)
if reg:
b_role = reg.group(0)
if "all" not in self.lp.key_words and b_role not in bot_roles_list:
output_content = "Bot Role not found! Type !roles to list them or !users all"
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
output_content = "USERS\n=====\n"
get_key = operator.attrgetter("name")
for user in (sorted(self.authenticator.users.values(),
key=lambda mbr: get_key(mbr).lower())):
if "all" in self.lp.key_words or b_role in user.b_roles:
user_name = user.name
user_bot_roles = ""
for user_b_role in user.b_roles:
user_bot_roles += ".%s " % user_b_role
user_guilds = ""
for guild in user.guilds:
user_guilds += "{%s} " % guild
user_guilds = user_guilds[:-1]
output_content += "- [%s] %s - %s\n" % (user_name, user_guilds,
user_bot_roles)
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
def cmd_earthquake(self):
output_channel = self.input_channel
output_broadcast = False
if self.lp.my_date:
for merb in self.merbs.merbs:
merb.update_pop(self.lp.my_date, str(self.input_author))
self.merbs.save_timers()
output_date = timeh.change_tz(
timeh.naive_to_tz(self.lp.my_date, "UTC"), self.lp.timezone)
output_content = "Earthquake! All pop times updated [%s] %s, signed by %s" % \
(output_date.strftime(config.DATE_FORMAT_PRINT),
self.lp.timezone,
self.input_author
)
output_broadcast = self.get_broadcast_channels()
else:
output_content = errors.error_param(self.lp.cmd,
"Time Syntax Error. ")
output_channel = self.input_author
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
def cmd_target(self):
output_channel = self.input_channel
output_broadcast = False
if self.lp.merb_found:
if "off" in self.lp.key_words:
self.lp.merb_found.target = False
output_content = "Target OFF for [%s] " % self.lp.merb_found.name
else:
self.lp.merb_found.target = True
output_content = "Target ON for [%s]" % self.lp.merb_found.name
output_content += "- signed by %s" % self.input_author.name
self.merbs.save_targets()
output_broadcast = self.get_broadcast_channels()
# If there is a guessed merb
elif self.lp.merb_guessed:
output_content = "Merb not found. Did you mean %s?" % self.lp.merb_guessed.name
output_channel = self.input_author
else:
output_content = errors.error_param(self.lp.cmd,
"Missing Parameter. ")
output_channel = self.input_author
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
"broadcast": output_broadcast
}
def cmd_echo(self):
return {
"destination": self.input_channel,
"content": messagecomposer.prettify(self.lp.param, "CSS"),
'broadcast': self.get_broadcast_channels()
}
def leave_all_guilds(self):
return {
"destination": self.input_channel,
"content": messagecomposer.prettify("Farewell!", "CSS"),
'broadcast': self.get_broadcast_channels(),
'action': 'leave_all_guilds'
}
def cmd_merbs(self):
output_content = messagecomposer.output_list(self.merbs.get_all_meta())
return {
"destination": self.input_author,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': False
}
async def minute_digest():
tic = 60
while True:
await asyncio.sleep(tic)
now = timeh.now()
for merb in merbs.merbs:
# update merb eta
merb.eta = merb.get_new_eta()
minutes_diff = (merb.eta - now).total_seconds() // 60.0
# broadcast the alarm 30 minutes before a target spawns
if merb.target and minutes_diff == 30:
print_info = merb.print_short_info(v_trackers=True) + "\n"
message = config.TAG_ALERT + "\n" + messagecomposer.prettify(
print_info, "RED")
try:
await client.get_channel(config.BROADCAST_CHANNEL
).send(message)
logger_sirken.info(f"Broadcast alarm sent for {merb.name}")
except Exception as e:
logger_sirken.error(
f"Broadcast alarm not sent! Error: {e}")
# send a pm to watchers
for user in watch.users:
destination = discord.utils.get(client.get_all_members(),
id=user)
if watch.check(user, merb.name,
minutes_diff) and not merb.is_in_window():
try:
await destination.send(
messagecomposer.prettify(merb.print_short_info(),
"CSS"))
logging.info(
"PM Alarm sent to %s: %s | ETA: %s | DIFF MINUTES: %s"
% (user, merb.name, merb.eta, minutes_diff))
except Exception as e:
logger_sirken.error(f"PM Alarm not sent! Error: {e}")
# UPDATE EMBED TIMERS
try:
await embed_timers.update_message(client, merbs, trackers)
# logger_sirken.info("Embed timers updated.")
except Exception as e:
logger_sirken.error(f"Embed timers not updated! Error: {e}")
def leave_all_guilds(self):
return {
'destination': self.input_channel,
'content': messagecomposer.prettify("Farewell!", "CSS"),
'broadcast': self.broadcast_channels,
'action': 'leave_all_guilds'
}
def update_merb(self, mode="tod"):
output_channel = self.input_channel
output_broadcast = False
# Check for approx, exact for default
approx = 1
approx_output = ""
if "approx" in self.lp.key_words:
approx = 0
approx_output = "~"
# If there is a Merb
if self.lp.merb_found:
# Check if we have a date
if self.lp.my_date:
# UPDATE THE TOD
if mode == "tod":
self.lp.merb_found.update_tod(self.lp.my_date,
str(self.input_author),
approx)
if mode == "pop":
self.lp.merb_found.update_pop(self.lp.my_date,
str(self.input_author))
# save merbs
self.merbs.save_timers()
output_date = timeh.change_tz(
timeh.naive_to_tz(self.lp.my_date, "UTC"),
self.lp.timezone)
output_content = "[%s] updated! New %s: {%s %s} - %ssigned by %s" % \
(self.lp.merb_found.name,
mode,
output_date.strftime(config.DATE_FORMAT_PRINT),
self.lp.timezone,
approx_output,
self.input_author.name)
output_broadcast = self.get_broadcast_channels()
else:
output_content = errors.error_param(self.lp.cmd,
"Time Syntax Error. ")
output_channel = self.input_author
# If there is a guessed merb
elif self.lp.merb_guessed:
output_content = "Merb not found. Did you mean %s?" % self.lp.merb_guessed.name
output_channel = self.input_author
# If no merb
else:
output_channel = self.input_author
output_content = errors.error_merb_not_found()
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
def cmd_get(self):
output_channel = self.input_channel
output_broadcast = False
# print merbs in target
if "target" in self.lp.key_words:
output_content = "NEXT TARGETS\n"
output_content += "=" * (len(output_content) - 1) + "\n\n"
output_content += messagecomposer.output_list(
self.merbs.get_all_targets())
# print merbs by tag
elif self.lp.tag:
output_content = "#%s\n" % self.lp.tag.upper()
output_content += "=" * len(self.lp.tag) + "\n\n"
output_content += messagecomposer.output_list(
self.merbs.get_all_by_tag(self.lp.tag))
# print only merbs in windows
elif "window" in self.lp.key_words:
output_content = "MERBS IN WINDOW\n"
output_content += "=" * len(output_content) + "\n\n"
output_content += messagecomposer.output_list(
self.merbs.get_all_window())
# print a list of all merbs
elif "all" in self.lp.key_words:
output_content = messagecomposer.output_list(
self.merbs.get_all(self.lp.timezone, 'countdown'))
# print single merb
elif self.lp.merb_found:
if "info" in self.lp.key_words:
output_content = self.lp.merb_found.print_long_info(
self.lp.timezone)
else:
output_content = self.lp.merb_found.print_short_info()
# no parameter recognized but a guessed merb
elif self.lp.merb_guessed:
output_content = "Merb not found. Did you mean %s?" % self.lp.merb_guessed.name
output_channel = self.input_author
# no parameter recognized
else:
output_content = errors.error_param(self.lp.cmd,
"Missing Parameter. ")
output_channel = self.input_author
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
def cmd_set_role(self):
output_channel = self.input_author
output_broadcast = False
output_content = ""
b_role = d_role = False
if self.lp.param:
# Find the discord_role_id
reg = re.search(r"\b(\d+)\b", self.lp.param)
if reg:
d_role = self.authenticator.roles.check_discord_role(
reg.group(0))
# Find the Bot_Roles
bot_roles_list = self.authenticator.roles.get_bot_roles_list()
regex = ""
for role in bot_roles_list:
regex += "%s|" % role
regex = r"\b(" + regex[:-1] + r")\b"
reg = re.search(regex, self.lp.param)
if reg:
b_role = reg.group(0)
if not isinstance(d_role, auth.DiscordRole):
output_content = "Discord Role ID not found! Type !roles to list them"
elif not b_role:
output_content = "Bot Role not found! Type !roles to list them"
else:
self.authenticator.roles.assign_discord_role_to_bot_role(
str(d_role.id), b_role)
output_content += "Discord Role [%s] {%s} assigned to Bot Role [%s]\n\n" %\
(d_role.name, d_role.id, b_role)
output_content += "Commands for this new role:\n%s" %\
self.authenticator.acl.which_permissions_any([b_role], "command")
self.authenticator.reload_discord_roles()
self.authenticator.reload_discord_users()
else:
output_content = errors.error_param(self.lp.cmd,
"Missing Parameter. ")
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
'broadcast': output_broadcast
}
def cmd_watch(self):
output_channel = self.input_author
output_broadcast = False
if self.lp.merb_found:
# search for minutes param
minutes = 30
reg_min = re.search(r"\b(\d+)\b", self.lp.param)
if reg_min:
minutes = int(reg_min.group(0))
off = False
if "off" in self.lp.key_words:
off = True
output_content = "Track OFF for [%s]" % self.lp.merb_found.name
else:
output_content = "Track ON for [%s], I will alert you %d before ETA" % \
(self.lp.merb_found.name, minutes)
self.watch.switch(self.input_author.id, self.lp.merb_found.name,
minutes, off)
# If there is a guessed merb
elif self.lp.merb_guessed:
output_content = "Merb not found. Did you mean %s?" % self.lp.merb_guessed.name
output_channel = self.input_author
# if no merb is passed but OFF parameter is, toggle off all alarms
elif "off" in self.lp.key_words:
self.watch.all_off(self.input_author.id)
output_content = "All alarms are set to OFF"
# If not params are passed get the full list of tracked merbs
else:
tracked_merbs = self.watch.get_all(self.input_author.id)
if not tracked_merbs:
output_content = "No merbs tracked :("
else:
output_content = ""
for tmerb in tracked_merbs:
output_content += '[%s] will alert %d minutes before ETA\n' % (
tmerb, tracked_merbs[tmerb])
return {
"destination": output_channel,
"content": messagecomposer.prettify(output_content, "CSS"),
"broadcast": output_broadcast
}
def process(self, author, channel, line):
t_start = timer()
self.lp.process(line)
# continue only if there is a command
if not self.lp.cmd:
return False
self.input_author = author
self.input_channel = channel
cmd_list = {
"about": self.cmd_about, # About
"help": self.cmd_help, # Help
"hi": self.cmd_help,
"get": self.cmd_get, # Get a single Merb
"tod": self.cmd_tod, # Update a Merb Status
"pop": self.cmd_pop, # Set pop time to now
"watch": self.cmd_watch, # Watch a merb
"target": self.cmd_target,
"earthquake": self.cmd_earthquake, # Reset all pop times to now
"merbs": self.cmd_merbs, # Get Aliases
"roles": self.cmd_roles,
"setrole": self.cmd_set_role,
"users": self.cmd_users,
"echo": self.cmd_echo,
}
func = cmd_list.get(
self.lp.cmd, lambda: {
"destination": self.input_author,
"content": messagecomposer.prettify(errors.error_command()),
"broadcast": False
})
output = func()
t_end = timer()
processing_time = round(t_end - t_start, 5)
logger.info("%s - %s (%s)" % (messagecomposer.simple_username(
str(self.input_author)), line, processing_time))
# clearing the line
self.lp.clear()
return output
async def on_message(input_message):
# Skip self messages
if input_message.author == client.user:
return
t_input_process_start = timer()
response_messages = sirken_cmds.process(input_message)
t_input_process_end = timer()
# Do nothing if there are no responses
if not response_messages:
return
# calculate the processing time and print on log
processing_time = round(t_input_process_end - t_input_process_start, 5)
logger_io.info("INPUT: %s - %s (%s) " %
(input_message.author.name, input_message.content,
processing_time))
# Loop the output messages list and cut messages too long for discord
output_messages = list()
for raw_message in response_messages:
if len(raw_message['content']) > config.MAX_MESSAGE_LENGTH:
for trunk in messagecomposer.message_cut(
raw_message['content'], config.MAX_MESSAGE_LENGTH):
output_messages.append({
'destination':
raw_message['destination'],
'content':
trunk,
'decoration':
raw_message['decoration']
})
else:
# if length is under limit, just copy the message in the output list
output_messages.append(raw_message)
for output_m in output_messages:
# Send the decorated messages
await output_m['destination'].send(
messagecomposer.prettify(output_m['content'],
output_m['decoration']))
logger_io.debug("OUTPUT: %s - %s" %
(output_m['destination'], output_m['content']))
def cmd_roles(self):
output_channel = self.input_author
output_broadcast = False
output_content = ""
# Reload Roles
if "reload" in self.lp.key_words:
self.authenticator.reload_discord_roles()
self.authenticator.reload_discord_users()
output_content += "{Roles Reloaded}\n\n"
output_discord_roles_content = "DISCORD ROLES\n=============\n"
for d_role in self.authenticator.roles.discord_roles:
converted_role = self.authenticator.roles.convert_discord_role_into_bot_role(
str(d_role.id))
output_discord_roles_content += "- [%s server] %s (%d) -> %s\n" % \
(d_role.guild,
d_role.name,
d_role.id,
converted_role)
output_bot_roles_content = "\nBOT ROLES\n=========\n"
for b_role in self.authenticator.roles.bot_roles:
output_bot_roles_content += "- %s\n" % b_role
for resource in self.authenticator.roles.bot_roles[b_role]:
output_bot_roles_content += " - %s " % resource
for permission in self.authenticator.roles.bot_roles[b_role][
resource]:
output_bot_roles_content += "[%s] " % permission
output_bot_roles_content += "\n"
output_bot_roles_content += "\n"
return {
"destination":
output_channel,
"content":
messagecomposer.prettify(output_discord_roles_content +
output_bot_roles_content),
'broadcast':
output_broadcast
}
async def minute_digest():
tic = 60
while True:
await asyncio.sleep(tic)
now = timeh.now()
for merb in merbs.merbs:
# update merb eta
merb.eta = merb.get_eta()
minutes_diff = (merb.eta - now).total_seconds() // 60.0
for user in watch.users:
destination = discord.utils.get(client.get_all_members(),
id=user)
if watch.check(user, merb.name,
minutes_diff) and not merb.in_window():
await destination.send(
messagecomposer.prettify(merb.print_short_info(),
"CSS")[0])
logging.debug(
"ALARM TO %s: %s | ETA: %s | DIFF MINUTES: %s" %
(user, merb.name, merb.eta, minutes_diff))
