def level(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
msg_args = message.split(' ')
if len(msg_args) > 1:
username = msg_args[0].lower()
new_level = int(msg_args[1])
if new_level >= source.level:
bot.whisper(source.username, 'You cannot promote someone to the same or higher level as you ({0}).'.format(source.level))
return False
# We create the user if the user didn't already exist in the database.
with bot.users.get_user_context(username) as user:
old_level = user.level
user.level = new_level
log_msg = '{}\'s user level changed from {} to {}'.format(
user.username_raw,
old_level,
new_level)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Userlevel edited', source, log_msg)
return True
bot.whisper(source.username, 'Usage: !level USERNAME NEW_LEVEL')
return False
def post(self, row_id, **options):
args = self.post_parser.parse_args()
try:
new_state = int(args['new_state'])
except (ValueError, KeyError):
return {'error': 'Invalid `new_state` parameter.'}, 400
with DBManager.create_session_scope() as db_session:
row = db_session.query(Banphrase).filter_by(id=row_id).one_or_none()
if not row:
return {
'error': 'Banphrase with this ID not found'
}, 404
row.enabled = True if new_state == 1 else False
db_session.commit()
payload = {
'id': row.id,
'new_state': row.enabled,
}
AdminLogManager.post('Banphrase toggled',
options['user'],
'Enabled' if row.enabled else 'Disabled',
row.phrase)
SocketClientManager.send('banphrase.update', payload)
return {'success': 'successful toggle', 'new_state': new_state}
def modules_edit(module_id, **options):
module_manager = ModuleManager(None).load(do_reload=False)
current_module = find(lambda m: m.ID == module_id, module_manager.all_modules)
if current_module is None:
return render_template('admin/module_404.html'), 404
sub_modules = []
for module in module_manager.all_modules:
module.db_module = None
with DBManager.create_session_scope() as db_session:
for db_module in db_session.query(Module):
module = find(lambda m: m.ID == db_module.id, module_manager.all_modules)
if module:
module.db_module = db_module
if module.PARENT_MODULE == current_module.__class__:
sub_modules.append(module)
if current_module.db_module is None:
return render_template('admin/module_404.html'), 404
if request.method == 'POST':
form_values = {key: value for key, value in request.form.items()}
res = current_module.parse_settings(**form_values)
if res is False:
return render_template('admin/module_404.html'), 404
current_module.db_module.settings = json.dumps(res)
db_session.commit()
settings = None
try:
settings = json.loads(current_module.db_module.settings)
except (TypeError, ValueError):
pass
current_module.load(settings=settings)
payload = {
'id': current_module.db_module.id,
}
SocketClientManager.send('module.update', payload)
AdminLogManager.post('Module edited', options['user'], current_module.NAME)
return render_template('admin/configure_module.html',
module=current_module,
sub_modules=sub_modules)
else:
settings = None
try:
settings = json.loads(current_module.db_module.settings)
except (TypeError, ValueError):
pass
current_module.load(settings=settings)
return render_template('admin/configure_module.html',
module=current_module,
sub_modules=sub_modules)
def remove_link_whitelist(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
id = None
try:
id = int(message)
except ValueError:
pass
link = self.db_session.query(WhitelistedLink).filter_by(id=id).one_or_none()
if link:
self.whitelisted_links.remove(link)
self.db_session.delete(link)
self.db_session.commit()
else:
bot.whisper(source.username, 'No link with the given id found')
return False
AdminLogManager.post('Whitelist link removed', source, link.domain)
bot.whisper(source.username, 'Successfully removed whitelisted link with id {0}'.format(link.id))
else:
bot.whisper(source.username, 'Usage: !remove link whitelist ID')
return False
def command_remove(command_id, **options):
with DBManager.create_session_scope() as db_session:
command = db_session.query(Command).filter_by(id=command_id).one_or_none()
if command is None:
return make_response(jsonify({'error': 'Invalid command ID'}), 404)
if command.level > options['user'].level:
abort(403)
log_msg = 'The !{} command has been removed'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command removed',
options['user'],
log_msg)
db_session.delete(command.data)
db_session.delete(command)
payload = {
'event': 'command.remove',
'data': {
'command_id': command_id
}
}
payload_bytes = json.dumps(payload).encode('utf-8')
try:
with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as client:
client.connect(config['sock']['sock_file'])
client.sendall(payload_bytes)
return make_response(jsonify({'success': 'good job'}))
except:
log.exception('???')
return make_response(jsonify({'error': 'Could not push update'}))
def remove_link_whitelist(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
id = None
try:
id = int(message)
except ValueError:
pass
link = self.db_session.query(WhitelistedLink).filter_by(
id=id).one_or_none()
if link:
self.whitelisted_links.remove(link)
self.db_session.delete(link)
self.db_session.commit()
else:
bot.whisper(source.username, 'No link with the given id found')
return False
AdminLogManager.post('Whitelist link removed', source, link.domain)
bot.whisper(
source.username,
'Successfully removed whitelisted link with id {0}'.format(
link.id))
else:
bot.whisper(source.username, 'Usage: !remove link whitelist ID')
return False
def level(bot, source, message, event, args):
if message:
msg_args = message.split(' ')
if len(msg_args) > 1:
username = msg_args[0].lower()
new_level = int(msg_args[1])
if new_level >= source.level:
bot.whisper(source.username, 'You cannot promote someone to the same or higher level as you ({0}).'.format(source.level))
return False
# We create the user if the user didn't already exist in the database.
user = bot.users[username]
old_level = user.level
user.level = new_level
log_msg = '{}\'s user level changed from {} to {}'.format(
user.username_raw,
old_level,
new_level)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Userlevel edited', source, log_msg)
return True
bot.whisper(source.username, 'Usage: !level USERNAME NEW_LEVEL')
return False
def add_link_blacklist(self, **options):
bot = options['bot']
message = options['message']
source = options['source']
options, new_links = self.parse_link_blacklist_arguments(message)
if new_links:
parts = new_links.split(' ')
try:
for link in parts:
if len(link) > 1:
self.blacklist_url(link, **options)
AdminLogManager.post('Blacklist link added', source,
link)
bot.whisper(source.username, 'Successfully added your links')
return True
except:
log.exception('Unhandled exception in add_link_blacklist')
bot.whisper(source.username,
'Some error occurred while adding your links')
return False
else:
bot.whisper(source.username, 'Usage: !add link blacklist LINK')
return False
def level(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
msg_args = message.split(' ')
if len(msg_args) > 1:
username = msg_args[0].lower()
new_level = int(msg_args[1])
if new_level >= source.level:
bot.whisper(source.username, 'You cannot promote someone to the same or higher level as you ({0}).'.format(source.level))
return False
# We create the user if the user didn't already exist in the database.
user = bot.users[username]
old_level = user.level
user.level = new_level
log_msg = '{}\'s user level changed from {} to {}'.format(
user.username_raw,
old_level,
new_level)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Userlevel edited', source, log_msg)
return True
bot.whisper(source.username, 'Usage: !level USERNAME NEW_LEVEL')
return False
def generic_toggle(route_key, row_id, **options):
valid_routes = {
'timer': Timer,
'banphrase': Banphrase,
'module': Module,
}
route_name = {
'timer': lambda x: x.name,
'banphrase': lambda x: x.phrase,
'module': lambda x: x.id,
}
route_title = {
'timer': 'Timer',
'banphrase': 'Banphrase',
'module': 'Module',
}
route_validator = {
'module': lambda x: validate_module(x.id)
}
if route_key not in valid_routes:
return make_response(jsonify({'error': 'Invalid route.'}), 400)
if 'new_state' not in request.form:
return make_response(jsonify({'error': 'Missing `new_state` parameter.'}), 400)
try:
new_state = int(request.form['new_state'])
except (ValueError, KeyError):
return make_response(jsonify({'error': 'Invalid `new_state` parameter.'}), 400)
route_value = valid_routes[route_key]
with DBManager.create_session_scope() as db_session:
row = db_session.query(route_value).filter_by(id=row_id).one_or_none()
if row:
validator = route_validator.get(route_key, None)
if validator is not None:
res = validator(row)
if not res:
return make_response(jsonify({'error': 'cannot modify {}'.format(route_key)}), 400)
row.enabled = True if new_state == 1 else False
db_session.commit()
payload = {
'{}_id'.format(route_key): row.id, # remove this
'id': row.id,
'new_state': row.enabled,
}
AdminLogManager.post('{title} toggled'.format(title=route_title[route_key]),
options['user'],
'Enabled' if row.enabled else 'Disabled',
route_name[route_key](row))
SocketClientManager.send('{}.update'.format(route_key), payload)
return make_response(jsonify({'success': 'successful toggle', 'new_state': new_state}))
else:
return make_response(jsonify({'error': 'invalid {} id'.format(route_key)}))
def set_title(bot, source, message, event, args):
# XXX: This should be a module
if message:
bot.twitchapi.set_title(bot.streamer, message)
log_msg = '{0} updated the title to "{1}"'.format(source.username_raw, message)
bot.say(log_msg)
AdminLogManager.add_entry("Title set", source, log_msg)
def add_command(bot, source, message, event, args):
"""Dispatch method for creating commands.
Usage: !add command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username, "Usage: !add command ALIAS [options] RESPONSE")
return False
options, response = bot.commands.parse_command_arguments(message_parts[1:])
options["added_by"] = source.id
if options is False:
bot.whisper(source.username, "Invalid command")
return False
alias_str = message_parts[0].replace("!", "").lower()
type = "say"
if options["whisper"] is True:
type = "whisper"
elif options["reply"] is True:
type = "reply"
elif response.startswith("/me") or response.startswith(".me"):
type = "me"
response = " ".join(response.split(" ")[1:])
elif options["whisper"] is False or options["reply"] is False:
type = "say"
action = {"type": type, "message": response}
command, new_command, alias_matched = bot.commands.create_command(alias_str, action=action, **options)
if new_command is True:
bot.whisper(source.username, "Added your command (ID: {command.id})".format(command=command))
log_msg = "The !{} command has been created".format(command.command.split("|")[0])
AdminLogManager.add_entry("Command created", source, log_msg)
return True
# At least one alias is already in use, notify the user to use !edit command instead
bot.whisper(
source.username,
"The alias {} is already in use. To edit that command, use !edit command instead of !add command.".format(
alias_matched
),
)
return False
def add_command(bot, source, message, event, args):
"""Dispatch method for creating commands.
Usage: !add command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username, 'Usage: !add command ALIAS [options] RESPONSE')
return False
options, response = bot.commands.parse_command_arguments(message_parts[1:])
options['added_by'] = source.id
if options is False:
bot.whisper(source.username, 'Invalid command')
return False
alias_str = message_parts[0].replace('!', '').lower()
type = 'say'
if options['whisper'] is True:
type = 'whisper'
elif options['reply'] is True:
type = 'reply'
elif response.startswith('/me') or response.startswith('.me'):
type = 'me'
response = ' '.join(response.split(' ')[1:])
elif options['whisper'] is False or options['reply'] is False:
type = 'say'
action = {
'type': type,
'message': response,
}
command, new_command, alias_matched = bot.commands.create_command(alias_str, action=action, **options)
if new_command is True:
bot.whisper(source.username, 'Added your command (ID: {command.id})'.format(command=command))
log_msg = 'The !{} command has been created'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command created',
source,
log_msg)
return True
# At least one alias is already in use, notify the user to use !edit command instead
bot.whisper(source.username, 'The alias {} is already in use. To edit that command, use !edit command instead of !add command.'.format(alias_matched))
return False
def get(self, timer_id, **options):
with DBManager.create_session_scope() as db_session:
timer = db_session.query(Timer).filter_by(id=timer_id).one_or_none()
if timer is None:
return {'error': 'Invalid timer ID'}, 404
AdminLogManager.post('Timer removed', options['user'], timer.name)
db_session.delete(timer)
SocketClientManager.send('timer.remove', {'id': timer.id})
return {'success': 'good job'}
def set_title(bot, source, message, event, args):
# XXX: This should be a module
if message:
bot.twitchapi.set_title(bot.streamer, message)
log_msg = '{0} updated the title to "{1}"'.format(
source.username_raw, message)
bot.say(log_msg)
AdminLogManager.add_entry('Title set', source, log_msg)
def get(self, banphrase_id, **options):
with DBManager.create_session_scope() as db_session:
banphrase = db_session.query(Banphrase).filter_by(id=banphrase_id).one_or_none()
if banphrase is None:
return {'error': 'Invalid banphrase ID'}, 404
AdminLogManager.post('Banphrase removed', options['user'], banphrase.phrase)
db_session.delete(banphrase)
db_session.delete(banphrase.data)
SocketClientManager.send('banphrase.remove', {'id': banphrase.id})
return {'success': 'good job'}, 200
def add_alias(bot, source, message, event, args):
"""Dispatch method for adding aliases to already-existing commands.
Usage: !add alias EXISTING_ALIAS NEW_ALIAS_1 NEW_ALIAS_2 ...
"""
if message:
message = message.replace('!', '').lower()
# Make sure we got both an existing alias and at least one new alias
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username,
'Usage: !add alias existingalias newalias')
return False
existing_alias = message_parts[0]
new_aliases = re.split('\|| ', ' '.join(message_parts[1:]))
added_aliases = []
already_used_aliases = []
if existing_alias not in bot.commands:
bot.whisper(
source.username,
'No command called "{0}" found'.format(existing_alias))
return False
command = bot.commands[existing_alias]
for alias in set(new_aliases):
if alias in bot.commands:
already_used_aliases.append(alias)
else:
added_aliases.append(alias)
bot.commands[alias] = command
if len(added_aliases) > 0:
new_aliases = '{}|{}'.format(command.command,
'|'.join(added_aliases))
bot.commands.edit_command(command, command=new_aliases)
bot.whisper(
source.username,
'Successfully added the aliases {0} to {1}'.format(
', '.join(added_aliases), existing_alias))
log_msg = 'The aliases {0} has been added to {1}'.format(
', '.join(added_aliases), existing_alias)
AdminLogManager.add_entry('Alias added', source, log_msg)
if len(already_used_aliases) > 0:
bot.whisper(
source.username,
'The following aliases were already in use: {0}'.format(
', '.join(already_used_aliases)))
else:
bot.whisper(source.username,
'Usage: !add alias existingalias newalias')
def banphrase_remove(banphrase_id, **options):
with DBManager.create_session_scope() as db_session:
banphrase = db_session.query(Banphrase).filter_by(
id=banphrase_id).one_or_none()
if banphrase is None:
return make_response(jsonify({'error': 'Invalid banphrase ID'}))
AdminLogManager.post('Banphrase removed', options['user'],
banphrase.phrase)
db_session.delete(banphrase)
db_session.delete(banphrase.data)
SocketClientManager.send('banphrase.remove',
{'banphrase_id': banphrase.id})
return make_response(jsonify({'success': 'good job'}))
def remove_command(bot, source, message, event, args):
if message:
id = None
command = None
try:
id = int(message)
except Exception:
pass
if id is None:
potential_cmd = ''.join(
message.split(' ')[:1]).lower().replace('!', '')
if potential_cmd in bot.commands:
command = bot.commands[potential_cmd]
log.info('got command: {0}'.format(command))
else:
for key, check_command in bot.commands.items():
if check_command.id == id:
command = check_command
break
if command is None:
bot.whisper(source.username,
'No command with the given parameters found')
return False
if command.id == -1:
bot.whisper(
source.username,
'That command is an internal command, it cannot be removed.'
)
return False
if source.level < 2000:
if command.action is not None and not command.action.type == 'message':
bot.whisper(
source.username,
'That command is not a normal command, it cannot be removed by you.'
)
return False
bot.whisper(
source.username,
'Successfully removed command with id {0}'.format(command.id))
log_msg = 'The !{} command has been removed'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command removed', source, log_msg)
bot.commands.remove_command(command)
else:
bot.whisper(source.username,
'Usage: !remove command (COMMAND_ID|COMMAND_ALIAS)')
def remove_alias(bot, source, message, event, args):
"""Dispatch method for removing aliases from a command.
Usage: !remove alias EXISTING_ALIAS_1 EXISTING_ALIAS_2"""
if message:
aliases = re.split('\|| ', message.lower())
log.info(aliases)
if len(aliases) < 1:
bot.whisper(source.username,
'Usage: !remove alias EXISTINGALIAS')
return False
num_removed = 0
commands_not_found = []
for alias in aliases:
if alias not in bot.commands:
commands_not_found.append(alias)
continue
command = bot.commands[alias]
current_aliases = command.command.split('|')
current_aliases.remove(alias)
if len(current_aliases) == 0:
bot.whisper(
source.username,
"{0} is the only remaining alias for this command and can't be removed."
.format(alias))
continue
new_aliases = '|'.join(current_aliases)
bot.commands.edit_command(command, command=new_aliases)
num_removed += 1
del bot.commands[alias]
log_msg = 'The alias {0} has been removed from {1}'.format(
alias,
new_aliases.split('|')[0])
AdminLogManager.add_entry('Alias removed', source, log_msg)
whisper_str = ''
if num_removed > 0:
whisper_str = 'Successfully removed {0} aliases.'.format(
num_removed)
if len(commands_not_found) > 0:
whisper_str += ' Aliases {0} not found'.format(
', '.join(commands_not_found))
if len(whisper_str) > 0:
bot.whisper(source.username, whisper_str)
else:
bot.whisper(source.username, 'Usage: !remove alias EXISTINGALIAS')
def add_alias(bot, source, message, event, args):
"""Dispatch method for adding aliases to already-existing commands.
Usage: !add alias EXISTING_ALIAS NEW_ALIAS_1 NEW_ALIAS_2 ...
"""
if message:
message = message.replace("!", "").lower()
# Make sure we got both an existing alias and at least one new alias
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username, "Usage: !add alias existingalias newalias")
return False
existing_alias = message_parts[0]
new_aliases = re.split("\|| ", " ".join(message_parts[1:]))
added_aliases = []
already_used_aliases = []
if existing_alias not in bot.commands:
bot.whisper(source.username, 'No command called "{0}" found'.format(existing_alias))
return False
command = bot.commands[existing_alias]
for alias in set(new_aliases):
if alias in bot.commands:
already_used_aliases.append(alias)
else:
added_aliases.append(alias)
bot.commands[alias] = command
if len(added_aliases) > 0:
new_aliases = "{}|{}".format(command.command, "|".join(added_aliases))
bot.commands.edit_command(command, command=new_aliases)
bot.whisper(
source.username,
"Successfully added the aliases {0} to {1}".format(", ".join(added_aliases), existing_alias),
)
log_msg = "The aliases {0} has been added to {1}".format(", ".join(added_aliases), existing_alias)
AdminLogManager.add_entry("Alias added", source, log_msg)
if len(already_used_aliases) > 0:
bot.whisper(
source.username,
"The following aliases were already in use: {0}".format(", ".join(already_used_aliases)),
)
else:
bot.whisper(source.username, "Usage: !add alias existingalias newalias")
def add_link_whitelist(self, **options):
bot = options['bot']
message = options['message']
source = options['source']
parts = message.split(' ')
try:
for link in parts:
self.whitelist_url(link)
AdminLogManager.post('Whitelist link added', source, link)
except:
log.exception('Unhandled exception in add_link')
bot.whisper(source.username, 'Some error occurred white adding your links')
return False
bot.whisper(source.username, 'Successfully added your links')
def permaban(bot, source, message, event, args):
if message:
msg_args = message.split(' ')
username = msg_args[0].lower()
user = bot.users[username]
if user.banned:
bot.whisper(source.username, 'User is already permabanned.')
return False
user.banned = True
message = message.lower()
log_msg = '{} has been permabanned'.format(user.username_raw)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Permaban added', source, log_msg)
def add_link_whitelist(self, **options):
bot = options['bot']
message = options['message']
source = options['source']
parts = message.split(' ')
try:
for link in parts:
self.whitelist_url(link)
AdminLogManager.post('Whitelist link added', source, link)
except:
log.exception('Unhandled exception in add_link')
bot.whisper(source.username,
'Some error occurred white adding your links')
return False
bot.whisper(source.username, 'Successfully added your links')
def remove_alias(bot, source, message, event, args):
"""Dispatch method for removing aliases from a command.
Usage: !remove alias EXISTING_ALIAS_1 EXISTING_ALIAS_2"""
if message:
aliases = re.split("\|| ", message.lower())
log.info(aliases)
if len(aliases) < 1:
bot.whisper(source.username, "Usage: !remove alias EXISTINGALIAS")
return False
num_removed = 0
commands_not_found = []
for alias in aliases:
if alias not in bot.commands:
commands_not_found.append(alias)
continue
command = bot.commands[alias]
current_aliases = command.command.split("|")
current_aliases.remove(alias)
if len(current_aliases) == 0:
bot.whisper(
source.username,
"{0} is the only remaining alias for this command and can't be removed.".format(alias),
)
continue
new_aliases = "|".join(current_aliases)
bot.commands.edit_command(command, command=new_aliases)
num_removed += 1
del bot.commands[alias]
log_msg = "The alias {0} has been removed from {1}".format(alias, new_aliases.split("|")[0])
AdminLogManager.add_entry("Alias removed", source, log_msg)
whisper_str = ""
if num_removed > 0:
whisper_str = "Successfully removed {0} aliases.".format(num_removed)
if len(commands_not_found) > 0:
whisper_str += " Aliases {0} not found".format(", ".join(commands_not_found))
if len(whisper_str) > 0:
bot.whisper(source.username, whisper_str)
else:
bot.whisper(source.username, "Usage: !remove alias EXISTINGALIAS")
def get(self, command_id, **options):
with DBManager.create_session_scope() as db_session:
command = db_session.query(Command).filter_by(id=command_id).one_or_none()
if command is None:
return {'error': 'Invalid command ID'}, 404
if command.level > options['user'].level:
return {'error': 'Unauthorized'}, 403
log_msg = 'The !{} command has been removed'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command removed',
options['user'],
log_msg)
db_session.delete(command.data)
db_session.delete(command)
if SocketClientManager.send('command.remove', {'command_id': command_id}) is True:
return {'success': 'good job'}, 200
else:
return {'error': 'could not push update'}, 500
def unpermaban(bot, source, message, event, args):
if message:
tmp_username = message.split(' ')[0].strip().lower()
user = bot.users.find(tmp_username)
if not user:
bot.whisper(source.username, 'No user with that name found.')
return False
if user.banned is False:
bot.whisper(source.username, 'User is not permabanned.')
return False
user.banned = False
message = message.lower()
log_msg = '{} is no longer permabanned'.format(user.username_raw)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Permaban remove', source, log_msg)
def permaban_command(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
msg_args = message.split(' ')
username = msg_args[0].lower()
user = bot.users[username]
if user.banned:
bot.whisper(source.username, 'User is already permabanned.')
return False
user.banned = True
message = message.lower()
log_msg = '{} has been permabanned'.format(user.username_raw)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Permaban added', source, log_msg)
def home(**options):
latest_logs_raw = AdminLogManager.get_entries()
cached_users = {}
with DBManager.create_session_scope() as db_session:
latest_logs = []
for log in latest_logs_raw:
log['user'] = cached_users[log['user_id']] if log['user_id'] in cached_users else db_session.query(User).filter_by(id=log['user_id']).one_or_none()
latest_logs.append(log)
return render_template('admin/home.html',
latest_logs=latest_logs)
def remove_command(bot, source, message, event, args):
if message:
id = None
command = None
try:
id = int(message)
except Exception:
pass
if id is None:
potential_cmd = ''.join(message.split(' ')[:1]).lower().replace('!', '')
if potential_cmd in bot.commands:
command = bot.commands[potential_cmd]
log.info('got command: {0}'.format(command))
else:
for key, check_command in bot.commands.items():
if check_command.id == id:
command = check_command
break
if command is None:
bot.whisper(source.username, 'No command with the given parameters found')
return False
if command.id == -1:
bot.whisper(source.username, 'That command is an internal command, it cannot be removed.')
return False
if source.level < 2000:
if command.action is not None and not command.action.type == 'message':
bot.whisper(source.username, 'That command is not a normal command, it cannot be removed by you.')
return False
bot.whisper(source.username, 'Successfully removed command with id {0}'.format(command.id))
log_msg = 'The !{} command has been removed'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command removed',
source,
log_msg)
bot.commands.remove_command(command)
else:
bot.whisper(source.username, 'Usage: !remove command (COMMAND_ID|COMMAND_ALIAS)')
def unpermaban_command(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
tmp_username = message.split(' ')[0].strip().lower()
user = bot.users.find(tmp_username)
if not user:
bot.whisper(source.username, 'No user with that name found.')
return False
if user.banned is False:
bot.whisper(source.username, 'User is not permabanned.')
return False
user.banned = False
message = message.lower()
log_msg = '{} is no longer permabanned'.format(user.username_raw)
bot.whisper(source.username, log_msg)
AdminLogManager.add_entry('Permaban remove', source, log_msg)
def add_link_blacklist(self, **options):
bot = options['bot']
message = options['message']
source = options['source']
options, new_links = self.parse_link_blacklist_arguments(message)
if new_links:
parts = new_links.split(' ')
try:
for link in parts:
if len(link) > 1:
self.blacklist_url(link, **options)
AdminLogManager.post('Blacklist link added', source, link)
bot.whisper(source.username, 'Successfully added your links')
return True
except:
log.exception('Unhandled exception in add_link_blacklist')
bot.whisper(source.username, 'Some error occurred while adding your links')
return False
else:
bot.whisper(source.username, 'Usage: !add link blacklist LINK')
return False
def remove_banphrase(self, **options):
message = options['message']
bot = options['bot']
source = options['source']
if message:
id = None
try:
id = int(message)
except ValueError:
pass
banphrase = bot.banphrase_manager.find_match(message=message, id=id)
if banphrase is None:
bot.whisper(source.username, 'No banphrase with the given parameters found')
return False
AdminLogManager.post('Banphrase removed', source, banphrase.phrase)
bot.whisper(source.username, 'Successfully removed banphrase with id {0}'.format(banphrase.id))
bot.banphrase_manager.remove_banphrase(banphrase)
else:
bot.whisper(source.username, 'Usage: !remove banphrase (BANPHRASE_ID)')
return False
def command_remove(command_id, **options):
with DBManager.create_session_scope() as db_session:
command = db_session.query(Command).filter_by(
id=command_id).one_or_none()
if command is None:
return make_response(jsonify({'error': 'Invalid command ID'}), 404)
if command.level > options['user'].level:
abort(403)
log_msg = 'The !{} command has been removed'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command removed', options['user'], log_msg)
db_session.delete(command.data)
db_session.delete(command)
payload = {'event': 'command.remove', 'data': {'command_id': command_id}}
payload_bytes = json.dumps(payload).encode('utf-8')
try:
with socket.socket(socket.AF_UNIX, socket.SOCK_STREAM) as client:
client.connect(config['sock']['sock_file'])
client.sendall(payload_bytes)
return make_response(jsonify({'success': 'good job'}))
except:
log.exception('???')
return make_response(jsonify({'error': 'Could not push update'}))
def add_banphrase(self, **options):
"""Method for creating and editing banphrases.
Usage: !add banphrase BANPHRASE [options]
Multiple options available:
--length LENGTH
--perma/--no-perma
--notify/--no-notify
"""
message = options['message']
bot = options['bot']
source = options['source']
if message:
options, phrase = bot.banphrase_manager.parse_banphrase_arguments(message)
if options is False:
bot.whisper(source.username, 'Invalid banphrase')
return False
options['added_by'] = source.id
options['edited_by'] = source.id
banphrase, new_banphrase = bot.banphrase_manager.create_banphrase(phrase, **options)
if new_banphrase is True:
bot.whisper(source.username, 'Added your banphrase (ID: {banphrase.id})'.format(banphrase=banphrase))
AdminLogManager.post('Banphrase added', source, phrase)
return True
banphrase.set(**options)
banphrase.data.set(edited_by=options['edited_by'])
DBManager.session_add_expunge(banphrase)
bot.banphrase_manager.commit()
bot.whisper(source.username, 'Updated your banphrase (ID: {banphrase.id}) with ({what})'.format(banphrase=banphrase, what=', '.join([key for key in options if key != 'added_by'])))
AdminLogManager.post('Banphrase edited', source, phrase)
def commands_create(**options):
session.pop('command_created_id', None)
session.pop('command_edited_id', None)
if request.method == 'POST':
if 'aliases' not in request.form:
abort(403)
alias_str = request.form.get('aliases', '').replace('!',
'').lower()
delay_all = request.form.get('cd', Command.DEFAULT_CD_ALL)
delay_user = request.form.get('usercd', Command.DEFAULT_CD_USER)
level = request.form.get('level', Command.DEFAULT_LEVEL)
cost = request.form.get('cost', 0)
try:
delay_all = int(delay_all)
delay_user = int(delay_user)
level = int(level)
cost = int(cost)
except ValueError:
abort(403)
if len(alias_str) == 0:
abort(403)
if delay_all < 0 or delay_all > 9999:
abort(403)
if delay_user < 0 or delay_user > 9999:
abort(403)
if level < 0 or level > 2000:
abort(403)
if cost < 0 or cost > 9999999:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'delay_all': delay_all,
'delay_user': delay_user,
'level': level,
'cost': cost,
'added_by': user.id,
}
valid_action_types = ['say', 'me', 'whisper', 'reply']
action_type = request.form.get('reply', 'say').lower()
if action_type not in valid_action_types:
abort(403)
response = request.form.get('response', '')
if len(response) == 0:
abort(403)
action = {'type': action_type, 'message': response}
options['action'] = action
command_manager = (CommandManager(
socket_manager=None,
module_manager=ModuleManager(None).load(),
bot=None).load(enabled=None))
command_aliases = []
for alias, command in command_manager.items():
command_aliases.append(alias)
if command.command and len(command.command) > 0:
command_aliases.extend(command.command.split('|'))
command_aliases = set(command_aliases)
alias_str = alias_str.replace(' ', '').replace('!', '').lower()
alias_list = alias_str.split('|')
alias_list = [alias for alias in alias_list if len(alias) > 0]
if len(alias_list) == 0:
return render_template('admin/create_command_fail.html')
for alias in alias_list:
if alias in command_aliases:
return render_template('admin/create_command_fail.html')
alias_str = '|'.join(alias_list)
command = Command(command=alias_str, **options)
command.data = CommandData(command.id, **options)
log_msg = 'The !{} command has been created'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command created', user, log_msg)
with DBManager.create_session_scope(
expire_on_commit=False) as db_session:
db_session.add(command)
db_session.add(command.data)
db_session.commit()
db_session.expunge(command)
db_session.expunge(command.data)
SocketClientManager.send('command.update',
{'command_id': command.id})
session['command_created_id'] = command.id
return redirect('/admin/commands/', 303)
else:
return render_template('admin/create_command.html')
def command_update(command_id, **extra_args):
if not request.method == 'POST':
return make_response(
jsonify({'error': 'Invalid request method. (Expected POST)'}), 400)
if len(request.form) == 0:
return make_response(jsonify({'error': 'Missing parameter to edit.'}),
400)
valid_names = [
'enabled', 'level', 'delay_all', 'delay_user', 'cost',
'can_execute_with_whisper', 'sub_only'
]
valid_action_names = ['type', 'message']
with DBManager.create_session_scope() as db_session:
command = db_session.query(Command).options(
joinedload(Command.data).joinedload(
CommandData.user)).filter_by(id=command_id).one_or_none()
if command is None:
return make_response(jsonify({'error': 'Invalid command ID'}), 404)
if command.level > extra_args['user'].level:
abort(403)
parsed_action = json.loads(command.action_json)
options = {
'edited_by': extra_args['user'].id,
}
for key in request.form:
if key.startswith('data_'):
name = key[5:]
value = request.form[key]
if name.startswith('action_'):
name = name[7:]
if name in valid_action_names and name in parsed_action and command.action.type == 'message':
value_type = type(parsed_action[name])
if value_type is bool:
parsed_value = True if value == '1' else False
elif value_type is int:
try:
parsed_value = int(value)
except ValueError:
continue
else:
parsed_value = value
parsed_action[name] = parsed_value
command.action_json = json.dumps(parsed_action)
else:
if name in valid_names:
value_type = type(getattr(command, name))
if value_type is bool:
parsed_value = True if value == '1' else False
elif value_type is int:
try:
parsed_value = int(value)
except ValueError:
continue
else:
parsed_value = value
options[name] = parsed_value
aj = json.loads(command.action_json)
old_message = ''
new_message = ''
try:
old_message = command.action.response
new_message = aj['message']
except:
pass
command.set(**options)
command.data.set(**options)
if len(old_message) > 0 and old_message != new_message:
log_msg = 'The !{} command has been updated from "{}" to "{}"'.format(
command.command.split('|')[0], old_message, new_message)
else:
log_msg = 'The !{} command has been updated'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command edited',
extra_args['user'],
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
if SocketClientManager.send('command.update',
{'command_id': command_id}) is True:
return make_response(jsonify({'success': 'good job'}))
else:
return make_response(jsonify({'error': 'Could not push update'}))
def timers_create(**options):
session.pop('timer_created_id', None)
session.pop('timer_edited_id', None)
if request.method == 'POST':
id = None
try:
if 'id' in request.form:
id = int(request.form['id'])
name = request.form['name'].strip()
interval_online = int(request.form['interval_online'])
interval_offline = int(request.form['interval_offline'])
message_type = request.form['message_type']
message = request.form['message'].strip()
except (KeyError, ValueError):
abort(403)
if interval_online < 0 or interval_offline < 0:
abort(403)
if message_type not in ['say', 'me']:
abort(403)
if len(message) == 0:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'name': name,
'interval_online': interval_online,
'interval_offline': interval_offline,
}
action = {
'type': message_type,
'message': message
}
options['action'] = action
if id is None:
timer = Timer(**options)
with DBManager.create_session_scope(expire_on_commit=False) as db_session:
if id is not None:
timer = db_session.query(Timer).filter_by(id=id).one_or_none()
if timer is None:
return redirect('/admin/timers/', 303)
old_message = ''
new_message = ''
try:
old_message = timer.action.response
new_message = action['message']
except:
pass
timer.set(**options)
if len(old_message) > 0 and old_message != new_message:
log_msg = 'Timer "{0}" has been updated from "{1}" to "{2}"'.format(
timer.name,
old_message,
new_message)
else:
log_msg = 'Timer "{0}" has been updated'.format(timer.name)
AdminLogManager.add_entry('Timer edited',
user,
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
else:
db_session.add(timer)
AdminLogManager.post('Timer added', user, timer.name)
SocketClientManager.send('timer.update', {'timer_id': timer.id})
if id is None:
session['timer_created_id'] = timer.id
else:
session['timer_edited_id'] = timer.id
return redirect('/admin/timers/', 303)
else:
return render_template('admin/create_timer.html')
def banphrases_create(**options):
session.pop('banphrase_created_id', None)
session.pop('banphrase_edited_id', None)
if request.method == 'POST':
id = None
try:
if 'id' in request.form:
id = int(request.form['id'])
name = request.form['name'].strip()
permanent = request.form.get('permanent', 'off')
warning = request.form.get('warning', 'off')
notify = request.form.get('notify', 'off')
case_sensitive = request.form.get('case_sensitive', 'off')
sub_immunity = request.form.get('sub_immunity', 'off')
length = int(request.form['length'])
phrase = request.form['phrase'].strip()
operator = request.form['operator'].strip().lower()
except (KeyError, ValueError):
abort(403)
permanent = True if permanent == 'on' else False
warning = True if warning == 'on' else False
notify = True if notify == 'on' else False
case_sensitive = True if case_sensitive == 'on' else False
sub_immunity = True if sub_immunity == 'on' else False
if len(name) == 0:
abort(403)
if len(phrase) == 0:
abort(403)
if length < 0 or length > 1209600:
abort(403)
valid_operators = ['contains', 'startswith', 'endswith']
if operator not in valid_operators:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'name': name,
'phrase': phrase,
'permanent': permanent,
'warning': warning,
'notify': notify,
'case_sensitive': case_sensitive,
'sub_immunity': sub_immunity,
'length': length,
'added_by': user.id,
'edited_by': user.id,
'operator': operator,
}
if id is None:
banphrase = Banphrase(**options)
banphrase.data = BanphraseData(banphrase.id, added_by=options['added_by'])
with DBManager.create_session_scope(expire_on_commit=False) as db_session:
if id is not None:
banphrase = db_session.query(Banphrase).options(joinedload(Banphrase.data)).filter_by(id=id).one_or_none()
if banphrase is None:
return redirect('/admin/banphrases/', 303)
banphrase.set(**options)
banphrase.data.set(edited_by=options['edited_by'])
log.info('Updated banphrase ID {} by user ID {}'.format(banphrase.id, options['edited_by']))
AdminLogManager.post('Banphrase edited', user, banphrase.phrase)
else:
db_session.add(banphrase)
db_session.add(banphrase.data)
log.info('Added a new banphrase by user ID {}'.format(options['added_by']))
AdminLogManager.post('Banphrase added', user, banphrase.phrase)
SocketClientManager.send('banphrase.update', {'banphrase_id': banphrase.id})
if id is None:
session['banphrase_created_id'] = banphrase.id
else:
session['banphrase_edited_id'] = banphrase.id
return redirect('/admin/banphrases/', 303)
else:
return render_template('admin/create_banphrase.html')
def command_update(command_id, **extra_args):
if not request.method == 'POST':
return make_response(jsonify({'error': 'Invalid request method. (Expected POST)'}), 400)
if len(request.form) == 0:
return make_response(jsonify({'error': 'Missing parameter to edit.'}), 400)
valid_names = [
'enabled',
'level',
'delay_all',
'delay_user',
'cost',
'can_execute_with_whisper',
'sub_only'
]
valid_action_names = [
'type',
'message'
]
with DBManager.create_session_scope() as db_session:
command = db_session.query(Command).options(joinedload(Command.data).joinedload(CommandData.user)).filter_by(id=command_id).one_or_none()
if command is None:
return make_response(jsonify({'error': 'Invalid command ID'}), 404)
if command.level > extra_args['user'].level:
abort(403)
parsed_action = json.loads(command.action_json)
options = {
'edited_by': extra_args['user'].id,
}
for key in request.form:
if key.startswith('data_'):
name = key[5:]
value = request.form[key]
if name.startswith('action_'):
name = name[7:]
if name in valid_action_names and name in parsed_action and command.action.type == 'message':
value_type = type(parsed_action[name])
if value_type is bool:
parsed_value = True if value == '1' else False
elif value_type is int:
try:
parsed_value = int(value)
except ValueError:
continue
else:
parsed_value = value
parsed_action[name] = parsed_value
command.action_json = json.dumps(parsed_action)
else:
if name in valid_names:
value_type = type(getattr(command, name))
if value_type is bool:
parsed_value = True if value == '1' else False
elif value_type is int:
try:
parsed_value = int(value)
except ValueError:
continue
else:
parsed_value = value
options[name] = parsed_value
aj = json.loads(command.action_json)
old_message = ''
new_message = ''
try:
old_message = command.action.response
new_message = aj['message']
except:
pass
command.set(**options)
command.data.set(**options)
if len(old_message) > 0 and old_message != new_message:
log_msg = 'The !{} command has been updated from "{}" to "{}"'.format(
command.command.split('|')[0],
old_message,
new_message)
else:
log_msg = 'The !{} command has been updated'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command edited',
extra_args['user'],
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
if SocketClientManager.send('command.update', {'command_id': command_id}) is True:
return make_response(jsonify({'success': 'good job'}))
else:
return make_response(jsonify({'error': 'Could not push update'}))
def edit_command(bot, source, message, event, args):
"""Dispatch method for editing commands.
Usage: !edit command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username,
'Usage: !add command ALIAS [options] RESPONSE')
return False
options, response = bot.commands.parse_command_arguments(
message_parts[1:])
options['edited_by'] = source.id
if options is False:
bot.whisper(source.username, 'Invalid command')
return False
alias = message_parts[0].replace('!', '').lower()
type = 'say'
if options['whisper'] is True:
type = 'whisper'
elif options['reply'] is True:
type = 'reply'
elif response.startswith('/me') or response.startswith('.me'):
type = 'me'
response = ' '.join(response.split(' ')[1:])
elif options['whisper'] is False or options['reply'] is False:
type = 'say'
action = {
'type': type,
'message': response,
}
command = bot.commands.get(alias, None)
if command is None:
bot.whisper(
source.username,
'No command found with the alias {}. Did you mean to create the command? If so, use !add command instead.'
.format(alias))
return False
old_message = ''
new_message = ''
if len(action['message']) > 0:
options['action'] = action
old_message = command.action.response
new_message = action['message']
elif not type == command.action.subtype:
options['action'] = {
'type': type,
'message': command.action.response,
}
bot.commands.edit_command(command, **options)
bot.whisper(
source.username,
'Updated the command (ID: {command.id})'.format(
command=command))
if len(new_message) > 0:
log_msg = 'The !{} command has been updated from "{}" to "{}"'.format(
command.command.split('|')[0], old_message, new_message)
else:
log_msg = 'The !{} command has been updated'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command edited',
source,
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
def modules_edit(module_id, **options):
module_manager = ModuleManager(None).load(do_reload=False)
current_module = find(lambda m: m.ID == module_id,
module_manager.all_modules)
if current_module is None:
return render_template('admin/module_404.html'), 404
sub_modules = []
for module in module_manager.all_modules:
module.db_module = None
with DBManager.create_session_scope() as db_session:
for db_module in db_session.query(Module):
module = find(lambda m: m.ID == db_module.id,
module_manager.all_modules)
if module:
module.db_module = db_module
if module.PARENT_MODULE == current_module.__class__:
sub_modules.append(module)
if current_module.db_module is None:
return render_template('admin/module_404.html'), 404
if request.method == 'POST':
form_values = {
key: value
for key, value in request.form.items()
}
res = current_module.parse_settings(**form_values)
if res is False:
return render_template('admin/module_404.html'), 404
current_module.db_module.settings = json.dumps(res)
db_session.commit()
settings = None
try:
settings = json.loads(current_module.db_module.settings)
except (TypeError, ValueError):
pass
current_module.load(settings=settings)
payload = {
'id': current_module.db_module.id,
}
SocketClientManager.send('module.update', payload)
AdminLogManager.post('Module edited', options['user'],
current_module.NAME)
return render_template('admin/configure_module.html',
module=current_module,
sub_modules=sub_modules)
else:
settings = None
try:
settings = json.loads(current_module.db_module.settings)
except (TypeError, ValueError):
pass
current_module.load(settings=settings)
return render_template('admin/configure_module.html',
module=current_module,
sub_modules=sub_modules)
def generic_toggle(route_key, row_id, **options):
valid_routes = {
'timer': Timer,
'banphrase': Banphrase,
'module': Module,
}
route_name = {
'timer': lambda x: x.name,
'banphrase': lambda x: x.phrase,
'module': lambda x: x.id,
}
route_title = {
'timer': 'Timer',
'banphrase': 'Banphrase',
'module': 'Module',
}
route_validator = {'module': lambda x: validate_module(x.id)}
if route_key not in valid_routes:
return make_response(jsonify({'error': 'Invalid route.'}), 400)
if 'new_state' not in request.form:
return make_response(
jsonify({'error': 'Missing `new_state` parameter.'}), 400)
try:
new_state = int(request.form['new_state'])
except (ValueError, KeyError):
return make_response(
jsonify({'error': 'Invalid `new_state` parameter.'}), 400)
route_value = valid_routes[route_key]
with DBManager.create_session_scope() as db_session:
row = db_session.query(route_value).filter_by(id=row_id).one_or_none()
if row:
validator = route_validator.get(route_key, None)
if validator is not None:
res = validator(row)
if not res:
return make_response(
jsonify(
{'error': 'cannot modify {}'.format(route_key)}),
400)
row.enabled = True if new_state == 1 else False
db_session.commit()
payload = {
'{}_id'.format(route_key): row.id, # remove this
'id': row.id,
'new_state': row.enabled,
}
AdminLogManager.post(
'{title} toggled'.format(title=route_title[route_key]),
options['user'], 'Enabled' if row.enabled else 'Disabled',
route_name[route_key](row))
SocketClientManager.send('{}.update'.format(route_key), payload)
return make_response(
jsonify({
'success': 'successful toggle',
'new_state': new_state
}))
else:
return make_response(
jsonify({'error': 'invalid {} id'.format(route_key)}))
def timers_create(**options):
session.pop('timer_created_id', None)
session.pop('timer_edited_id', None)
if request.method == 'POST':
id = None
try:
if 'id' in request.form:
id = int(request.form['id'])
name = request.form['name'].strip()
interval_online = int(request.form['interval_online'])
interval_offline = int(request.form['interval_offline'])
message_type = request.form['message_type']
message = request.form['message'].strip()
except (KeyError, ValueError):
abort(403)
if interval_online < 0 or interval_offline < 0:
abort(403)
if message_type not in ['say', 'me']:
abort(403)
if len(message) == 0:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'name': name,
'interval_online': interval_online,
'interval_offline': interval_offline,
}
action = {'type': message_type, 'message': message}
options['action'] = action
if id is None:
timer = Timer(**options)
with DBManager.create_session_scope(
expire_on_commit=False) as db_session:
if id is not None:
timer = db_session.query(Timer).filter_by(
id=id).one_or_none()
if timer is None:
return redirect('/admin/timers/', 303)
old_message = ''
new_message = ''
try:
old_message = timer.action.response
new_message = action['message']
except:
pass
timer.set(**options)
if len(old_message) > 0 and old_message != new_message:
log_msg = 'Timer "{0}" has been updated from "{1}" to "{2}"'.format(
timer.name, old_message, new_message)
else:
log_msg = 'Timer "{0}" has been updated'.format(
timer.name)
AdminLogManager.add_entry('Timer edited',
user,
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
else:
db_session.add(timer)
AdminLogManager.post('Timer added', user, timer.name)
SocketClientManager.send('timer.update', {'timer_id': timer.id})
if id is None:
session['timer_created_id'] = timer.id
else:
session['timer_edited_id'] = timer.id
return redirect('/admin/timers/', 303)
else:
return render_template('admin/create_timer.html')
def edit_command(bot, source, message, event, args):
"""Dispatch method for editing commands.
Usage: !edit command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username, "Usage: !add command ALIAS [options] RESPONSE")
return False
options, response = bot.commands.parse_command_arguments(message_parts[1:])
options["edited_by"] = source.id
if options is False:
bot.whisper(source.username, "Invalid command")
return False
alias = message_parts[0].replace("!", "").lower()
type = "say"
if options["whisper"] is True:
type = "whisper"
elif options["reply"] is True:
type = "reply"
elif response.startswith("/me") or response.startswith(".me"):
type = "me"
response = " ".join(response.split(" ")[1:])
elif options["whisper"] is False or options["reply"] is False:
type = "say"
action = {"type": type, "message": response}
command = bot.commands.get(alias, None)
if command is None:
bot.whisper(
source.username,
"No command found with the alias {}. Did you mean to create the command? If so, use !add command instead.".format(
alias
),
)
return False
old_message = ""
new_message = ""
if len(action["message"]) > 0:
options["action"] = action
old_message = command.action.response
new_message = action["message"]
elif not type == command.action.subtype:
options["action"] = {"type": type, "message": command.action.response}
bot.commands.edit_command(command, **options)
bot.whisper(source.username, "Updated the command (ID: {command.id})".format(command=command))
if len(new_message) > 0:
log_msg = 'The !{} command has been updated from "{}" to "{}"'.format(
command.command.split("|")[0], old_message, new_message
)
else:
log_msg = "The !{} command has been updated".format(command.command.split("|")[0])
AdminLogManager.add_entry(
"Command edited", source, log_msg, data={"old_message": old_message, "new_message": new_message}
)
def commands_create(**options):
session.pop('command_created_id', None)
session.pop('command_edited_id', None)
if request.method == 'POST':
if 'aliases' not in request.form:
abort(403)
alias_str = request.form.get('aliases', '').replace('!', '').lower()
delay_all = request.form.get('cd', Command.DEFAULT_CD_ALL)
delay_user = request.form.get('usercd', Command.DEFAULT_CD_USER)
level = request.form.get('level', Command.DEFAULT_LEVEL)
cost = request.form.get('cost', 0)
try:
delay_all = int(delay_all)
delay_user = int(delay_user)
level = int(level)
cost = int(cost)
except ValueError:
abort(403)
if len(alias_str) == 0:
abort(403)
if delay_all < 0 or delay_all > 9999:
abort(403)
if delay_user < 0 or delay_user > 9999:
abort(403)
if level < 0 or level > 2000:
abort(403)
if cost < 0 or cost > 9999999:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'delay_all': delay_all,
'delay_user': delay_user,
'level': level,
'cost': cost,
'added_by': user.id,
}
valid_action_types = ['say', 'me', 'whisper', 'reply']
action_type = request.form.get('reply', 'say').lower()
if action_type not in valid_action_types:
abort(403)
response = request.form.get('response', '')
if len(response) == 0:
abort(403)
action = {
'type': action_type,
'message': response
}
options['action'] = action
command_manager = (
CommandManager(
socket_manager=None,
module_manager=ModuleManager(None).load(),
bot=None).load(enabled=None))
command_aliases = []
for alias, command in command_manager.items():
command_aliases.append(alias)
if command.command and len(command.command) > 0:
command_aliases.extend(command.command.split('|'))
command_aliases = set(command_aliases)
alias_str = alias_str.replace(' ', '').replace('!', '').lower()
alias_list = alias_str.split('|')
alias_list = [alias for alias in alias_list if len(alias) > 0]
if len(alias_list) == 0:
return render_template('admin/create_command_fail.html')
for alias in alias_list:
if alias in command_aliases:
return render_template('admin/create_command_fail.html')
alias_str = '|'.join(alias_list)
command = Command(command=alias_str, **options)
command.data = CommandData(command.id, **options)
log_msg = 'The !{} command has been created'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command created',
user,
log_msg)
with DBManager.create_session_scope(expire_on_commit=False) as db_session:
db_session.add(command)
db_session.add(command.data)
db_session.commit()
db_session.expunge(command)
db_session.expunge(command.data)
SocketClientManager.send('command.update', {'command_id': command.id})
session['command_created_id'] = command.id
return redirect('/admin/commands/', 303)
else:
return render_template('admin/create_command.html')
def add_command(bot, source, message, event, args):
"""Dispatch method for creating commands.
Usage: !add command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username,
'Usage: !add command ALIAS [options] RESPONSE')
return False
options, response = bot.commands.parse_command_arguments(
message_parts[1:])
options['added_by'] = source.id
if options is False:
bot.whisper(source.username, 'Invalid command')
return False
alias_str = message_parts[0].replace('!', '').lower()
type = 'say'
if options['whisper'] is True:
type = 'whisper'
elif options['reply'] is True:
type = 'reply'
elif response.startswith('/me') or response.startswith('.me'):
type = 'me'
response = ' '.join(response.split(' ')[1:])
elif options['whisper'] is False or options['reply'] is False:
type = 'say'
action = {
'type': type,
'message': response,
}
command, new_command, alias_matched = bot.commands.create_command(
alias_str, action=action, **options)
if new_command is True:
bot.whisper(
source.username,
'Added your command (ID: {command.id})'.format(
command=command))
log_msg = 'The !{} command has been created'.format(
command.command.split('|')[0])
AdminLogManager.add_entry('Command created', source, log_msg)
return True
# At least one alias is already in use, notify the user to use !edit command instead
bot.whisper(
source.username,
'The alias {} is already in use. To edit that command, use !edit command instead of !add command.'
.format(alias_matched))
return False
def banphrases_create(**options):
session.pop('banphrase_created_id', None)
session.pop('banphrase_edited_id', None)
if request.method == 'POST':
id = None
try:
if 'id' in request.form:
id = int(request.form['id'])
name = request.form['name'].strip()
permanent = request.form.get('permanent', 'off')
warning = request.form.get('warning', 'off')
notify = request.form.get('notify', 'off')
case_sensitive = request.form.get('case_sensitive', 'off')
sub_immunity = request.form.get('sub_immunity', 'off')
length = int(request.form['length'])
phrase = request.form['phrase'].strip()
operator = request.form['operator'].strip().lower()
except (KeyError, ValueError):
abort(403)
permanent = True if permanent == 'on' else False
warning = True if warning == 'on' else False
notify = True if notify == 'on' else False
case_sensitive = True if case_sensitive == 'on' else False
sub_immunity = True if sub_immunity == 'on' else False
if len(name) == 0:
abort(403)
if len(phrase) == 0:
abort(403)
if length < 0 or length > 1209600:
abort(403)
valid_operators = ['contains', 'startswith', 'endswith']
if operator not in valid_operators:
abort(403)
user = options.get('user', None)
if user is None:
abort(403)
options = {
'name': name,
'phrase': phrase,
'permanent': permanent,
'warning': warning,
'notify': notify,
'case_sensitive': case_sensitive,
'sub_immunity': sub_immunity,
'length': length,
'added_by': user.id,
'edited_by': user.id,
'operator': operator,
}
if id is None:
banphrase = Banphrase(**options)
banphrase.data = BanphraseData(banphrase.id,
added_by=options['added_by'])
with DBManager.create_session_scope(
expire_on_commit=False) as db_session:
if id is not None:
banphrase = db_session.query(Banphrase).options(
joinedload(
Banphrase.data)).filter_by(id=id).one_or_none()
if banphrase is None:
return redirect('/admin/banphrases/', 303)
banphrase.set(**options)
banphrase.data.set(edited_by=options['edited_by'])
log.info('Updated banphrase ID {} by user ID {}'.format(
banphrase.id, options['edited_by']))
AdminLogManager.post('Banphrase edited', user,
banphrase.phrase)
else:
db_session.add(banphrase)
db_session.add(banphrase.data)
log.info('Added a new banphrase by user ID {}'.format(
options['added_by']))
AdminLogManager.post('Banphrase added', user,
banphrase.phrase)
SocketClientManager.send('banphrase.update',
{'banphrase_id': banphrase.id})
if id is None:
session['banphrase_created_id'] = banphrase.id
else:
session['banphrase_edited_id'] = banphrase.id
return redirect('/admin/banphrases/', 303)
else:
return render_template('admin/create_banphrase.html')
def edit_command(bot, source, message, event, args):
"""Dispatch method for editing commands.
Usage: !edit command ALIAS [options] RESPONSE
Multiple options available:
--whisper/--no-whisper
--reply/--no-reply
--modonly/--no-modonly
--cd CD
--usercd USERCD
--level LEVEL
--cost COST
"""
if message:
# Make sure we got both an alias and a response
message_parts = message.split()
if len(message_parts) < 2:
bot.whisper(source.username, 'Usage: !add command ALIAS [options] RESPONSE')
return False
options, response = bot.commands.parse_command_arguments(message_parts[1:])
options['edited_by'] = source.id
if options is False:
bot.whisper(source.username, 'Invalid command')
return False
alias = message_parts[0].replace('!', '').lower()
type = 'say'
if options['whisper'] is True:
type = 'whisper'
elif options['reply'] is True:
type = 'reply'
elif response.startswith('/me') or response.startswith('.me'):
type = 'me'
response = ' '.join(response.split(' ')[1:])
elif options['whisper'] is False or options['reply'] is False:
type = 'say'
action = {
'type': type,
'message': response,
}
command = bot.commands.get(alias, None)
if command is None:
bot.whisper(source.username, 'No command found with the alias {}. Did you mean to create the command? If so, use !add command instead.'.format(alias))
return False
old_message = ''
new_message = ''
if len(action['message']) > 0:
options['action'] = action
old_message = command.action.response
new_message = action['message']
elif not type == command.action.subtype:
options['action'] = {
'type': type,
'message': command.action.response,
}
bot.commands.edit_command(command, **options)
bot.whisper(source.username, 'Updated the command (ID: {command.id})'.format(command=command))
if len(new_message) > 0:
log_msg = 'The !{} command has been updated from "{}" to "{}"'.format(
command.command.split('|')[0],
old_message,
new_message)
else:
log_msg = 'The !{} command has been updated'.format(command.command.split('|')[0])
AdminLogManager.add_entry('Command edited',
source,
log_msg,
data={
'old_message': old_message,
'new_message': new_message,
})
