mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
797 lines
34 KiB
Python
797 lines
34 KiB
Python
import json
|
|
import logging
|
|
import tornado.web
|
|
import tornado.escape
|
|
import bleach
|
|
import time
|
|
import datetime
|
|
import os
|
|
|
|
from tornado import iostream
|
|
|
|
from app.classes.shared.console import console
|
|
from app.classes.shared.models import Users, installer
|
|
from app.classes.web.base_handler import BaseHandler
|
|
from app.classes.shared.models import db_helper, Servers
|
|
from app.classes.shared.helpers import helper
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class PanelHandler(BaseHandler):
|
|
|
|
@tornado.web.authenticated
|
|
def get(self, page):
|
|
error = bleach.clean(self.get_argument('error', "WTF Error!"))
|
|
|
|
template = "panel/denied.html"
|
|
|
|
now = time.time()
|
|
formatted_time = str(datetime.datetime.fromtimestamp(now).strftime('%Y-%m-%d %H:%M:%S'))
|
|
|
|
exec_user_data = json.loads(self.get_secure_cookie("user_data"))
|
|
exec_user_id = exec_user_data['user_id']
|
|
exec_user = db_helper.get_user(exec_user_id)
|
|
|
|
exec_user_role = set()
|
|
if exec_user['superuser'] == 1:
|
|
defined_servers = self.controller.list_defined_servers()
|
|
exec_user_role.add("Super User")
|
|
else:
|
|
logger.debug(exec_user['roles'])
|
|
for r in exec_user['roles']:
|
|
role = db_helper.get_role(r)
|
|
exec_user_role.add(role['role_name'])
|
|
defined_servers = db_helper.get_all_authorized_servers(exec_user_id)
|
|
|
|
page_data = {
|
|
# todo: make this actually pull and compare version data
|
|
'update_available': False,
|
|
'version_data': helper.get_version_string(),
|
|
'user_data': exec_user_data,
|
|
'user_role' : exec_user_role,
|
|
'server_stats': {
|
|
'total': len(defined_servers),
|
|
'running': len(self.controller.list_running_servers()),
|
|
'stopped': (len(self.controller.list_defined_servers()) - len(self.controller.list_running_servers()))
|
|
},
|
|
'menu_servers': defined_servers,
|
|
'hosts_data': db_helper.get_latest_hosts_stats(),
|
|
'show_contribute': helper.get_setting("show_contribute_link", True),
|
|
'error': error,
|
|
'time': formatted_time
|
|
}
|
|
|
|
# if no servers defined, let's go to the build server area
|
|
if page_data['server_stats']['total'] == 0 and page != "error" and page != "credits" and page != "contribute":
|
|
self.set_status(301)
|
|
self.redirect("/server/step1")
|
|
return
|
|
|
|
if page == 'unauthorized':
|
|
template = "panel/denied.html"
|
|
|
|
elif page == "error":
|
|
template = "public/error.html"
|
|
|
|
elif page == 'credits':
|
|
with open(helper.credits_cache) as republic_credits_will_do:
|
|
credits = json.load(republic_credits_will_do)
|
|
page_data["patreons"] = credits["patreons"]
|
|
page_data["staff"] = credits["staff"]
|
|
page_data["translations"] = credits["translations"]
|
|
template = "panel/credits.html"
|
|
|
|
elif page == 'contribute':
|
|
template = "panel/contribute.html"
|
|
|
|
elif page == "remove_server":
|
|
server_id = self.get_argument('id', None)
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
server_data = self.controller.get_server_data(server_id)
|
|
server_name = server_data['server_name']
|
|
|
|
db_helper.add_to_audit_log(exec_user_data['user_id'],
|
|
"Deleted server {} named {}".format(server_id, server_name),
|
|
server_id,
|
|
self.get_remote_ip())
|
|
|
|
self.controller.remove_server(server_id)
|
|
self.redirect("/panel/dashboard")
|
|
return
|
|
|
|
elif page == 'dashboard':
|
|
if exec_user['superuser'] == 1:
|
|
page_data['servers'] = db_helper.get_all_servers_stats()
|
|
else:
|
|
user_auth = db_helper.get_authorized_servers_stats(exec_user_id)
|
|
logger.debug("ASFR: {}".format(user_auth))
|
|
page_data['servers'] = user_auth
|
|
|
|
for s in page_data['servers']:
|
|
try:
|
|
data = json.loads(s['int_ping_results'])
|
|
s['int_ping_results'] = data
|
|
except:
|
|
pass
|
|
|
|
template = "panel/dashboard.html"
|
|
|
|
elif page == 'server_detail':
|
|
server_id = self.get_argument('id', None)
|
|
subpage = bleach.clean(self.get_argument('subpage', ""))
|
|
|
|
if server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
else:
|
|
# does this server id exist?
|
|
if not db_helper.server_id_exists(server_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
if exec_user['superuser'] != 1:
|
|
if not db_helper.server_id_authorized(server_id, exec_user_id):
|
|
if not db_helper.server_id_authorized_from_roles(int(server_id), exec_user_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return False
|
|
|
|
valid_subpages = ['term', 'logs', 'backup', 'config', 'files', 'admin_controls']
|
|
|
|
if subpage not in valid_subpages:
|
|
logger.debug('not a valid subpage')
|
|
subpage = 'term'
|
|
logger.debug('Subpage: "{}"'.format(subpage))
|
|
|
|
server = self.controller.get_server_obj(server_id)
|
|
# server_data isn't needed since the server_stats also pulls server data
|
|
page_data['server_data'] = db_helper.get_server_data_by_id(server_id)
|
|
page_data['server_stats'] = db_helper.get_server_stats_by_id(server_id)
|
|
page_data['get_players'] = lambda: self.controller.stats.get_server_players(server_id)
|
|
if subpage == "backup":
|
|
page_data['backup_config'] = db_helper.get_backup_config(server_id)
|
|
page_data['backup_list'] = server.list_backups()
|
|
|
|
def get_banned_players_html():
|
|
banned_players = helper.get_banned_players(server_id, db_helper)
|
|
if banned_players is None:
|
|
return """
|
|
<li class="playerItem banned">
|
|
<h3>Error while reading banned-players.json</h3>
|
|
</li>
|
|
"""
|
|
html = ""
|
|
for player in banned_players:
|
|
html += """
|
|
<li class="playerItem banned">
|
|
<h3>{}</h3>
|
|
<span>Banned by {} for reason: {}</span>
|
|
<button onclick="send_command_to_server('pardon {}')" type="button" class="btn btn-danger">Unban</button>
|
|
</li>
|
|
""".format(player['name'], player['source'], player['reason'], player['name'])
|
|
|
|
return html
|
|
if subpage == "admin_controls":
|
|
page_data['banned_players'] = get_banned_players_html()
|
|
|
|
# template = "panel/server_details.html"
|
|
template = "panel/server_{subpage}.html".format(subpage=subpage)
|
|
|
|
elif page == 'download_backup':
|
|
server_id = self.get_argument('id', None)
|
|
file = self.get_argument('file', "")
|
|
|
|
if server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
else:
|
|
# does this server id exist?
|
|
if not db_helper.server_id_exists(server_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
if exec_user['superuser'] != 1:
|
|
#if not db_helper.server_id_authorized(server_id, exec_user_id):
|
|
if not db_helper.server_id_authorized_from_roles(int(server_id), exec_user_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return False
|
|
|
|
server_info = db_helper.get_server_data_by_id(server_id)
|
|
backup_file = os.path.abspath(os.path.join(server_info["backup_path"], file))
|
|
if not helper.in_path(server_info["backup_path"], backup_file) \
|
|
or not os.path.isfile(backup_file):
|
|
self.redirect("/panel/error?error=Invalid path detected")
|
|
return
|
|
|
|
self.set_header('Content-Type', 'application/octet-stream')
|
|
self.set_header('Content-Disposition', 'attachment; filename=' + file)
|
|
chunk_size = 1024 * 1024 * 4 # 4 MiB
|
|
|
|
with open(backup_file, 'rb') as f:
|
|
while True:
|
|
chunk = f.read(chunk_size)
|
|
if not chunk:
|
|
break
|
|
try:
|
|
self.write(chunk) # write the chunk to response
|
|
self.flush() # send the chunk to client
|
|
except iostream.StreamClosedError:
|
|
# this means the client has closed the connection
|
|
# so break the loop
|
|
break
|
|
finally:
|
|
# deleting the chunk is very important because
|
|
# if many clients are downloading files at the
|
|
# same time, the chunks in memory will keep
|
|
# increasing and will eat up the RAM
|
|
del chunk
|
|
self.redirect("/panel/server_detail?id={}&subpage=backup".format(server_id))
|
|
|
|
elif page == 'backup_now':
|
|
server_id = self.get_argument('id', None)
|
|
|
|
if server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
else:
|
|
# does this server id exist?
|
|
if not db_helper.server_id_exists(server_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
if exec_user['superuser'] != 1:
|
|
#if not db_helper.server_id_authorized(server_id, exec_user_id):
|
|
if not db_helper.server_id_authorized_from_roles(int(server_id), exec_user_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return False
|
|
|
|
server = self.controller.get_server_obj(server_id).backup_server()
|
|
self.redirect("/panel/server_detail?id={}&subpage=backup".format(server_id))
|
|
|
|
elif page == 'panel_config':
|
|
auth_servers = {}
|
|
auth_role_servers = {}
|
|
users_list = []
|
|
role_users = {}
|
|
roles = db_helper.get_all_roles()
|
|
role_servers = []
|
|
user_roles = {}
|
|
for user in db_helper.get_all_users():
|
|
user_roles_list = db_helper.get_user_roles_names(user.user_id)
|
|
user_servers = db_helper.get_all_authorized_servers(user.user_id)
|
|
servers = []
|
|
for server in user_servers:
|
|
servers.append(server['server_name'])
|
|
new_item = {user.user_id: servers}
|
|
auth_servers.update(new_item)
|
|
data = {user.user_id: user_roles_list}
|
|
user_roles.update(data)
|
|
for role in roles:
|
|
role_servers = []
|
|
role = db_helper.get_role(role.role_id)
|
|
for serv_id in role['servers']:
|
|
role_servers.append(db_helper.get_server_data_by_id(serv_id)['server_name'])
|
|
data = {role['role_id']: role_servers}
|
|
auth_role_servers.update(data)
|
|
|
|
|
|
page_data['auth-servers'] = auth_servers
|
|
page_data['role-servers'] = auth_role_servers
|
|
page_data['user-roles'] = user_roles
|
|
|
|
if exec_user['superuser'] == 1:
|
|
page_data['users'] = db_helper.get_all_users()
|
|
page_data['roles'] = db_helper.get_all_roles()
|
|
else:
|
|
page_data['users'] = db_helper.user_query(exec_user['user_id'])
|
|
page_data['roles'] = db_helper.user_role_query(exec_user['user_id'])
|
|
|
|
for user in page_data['users']:
|
|
if user.user_id != exec_user['user_id']:
|
|
user.api_token = "********"
|
|
template = "panel/panel_config.html"
|
|
|
|
elif page == "add_user":
|
|
page_data['new_user'] = True
|
|
page_data['user'] = {}
|
|
page_data['user']['username'] = ""
|
|
page_data['user']['user_id'] = -1
|
|
page_data['user']['enabled'] = True
|
|
page_data['user']['superuser'] = False
|
|
page_data['user']['api_token'] = "N/A"
|
|
page_data['user']['created'] = "N/A"
|
|
page_data['user']['last_login'] = "N/A"
|
|
page_data['user']['last_ip'] = "N/A"
|
|
page_data['user']['last_update'] = "N/A"
|
|
page_data['user']['roles'] = set()
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
|
|
page_data['roles_all'] = db_helper.get_all_roles()
|
|
page_data['servers'] = []
|
|
page_data['servers_all'] = self.controller.list_defined_servers()
|
|
page_data['role-servers'] = []
|
|
template = "panel/panel_edit_user.html"
|
|
|
|
elif page == "edit_user":
|
|
user_id = self.get_argument('id', None)
|
|
user_servers = db_helper.get_authorized_servers(user_id)
|
|
role_servers = db_helper.get_authorized_servers_from_roles(user_id)
|
|
page_role_servers = []
|
|
servers = set()
|
|
for server in user_servers:
|
|
flag = False
|
|
for rserver in role_servers:
|
|
if rserver['server_id'] == server['server_id']:
|
|
flag = True
|
|
if not flag:
|
|
servers.add(server['server_id'])
|
|
for server in role_servers:
|
|
page_role_servers.append(server['server_id'])
|
|
page_data['new_user'] = False
|
|
page_data['user'] = db_helper.get_user(user_id)
|
|
page_data['servers'] = servers
|
|
page_data['role-servers'] = page_role_servers
|
|
page_data['roles_all'] = db_helper.get_all_roles()
|
|
page_data['servers_all'] = self.controller.list_defined_servers()
|
|
|
|
if user_id is None:
|
|
self.redirect("/panel/error?error=Invalid User ID")
|
|
return
|
|
elif not exec_user['superuser']:
|
|
page_data['servers'] = []
|
|
page_data['role-servers'] = []
|
|
page_data['roles_all'] = []
|
|
page_data['servers_all'] = []
|
|
|
|
if exec_user['user_id'] != page_data['user']['user_id']:
|
|
page_data['user']['api_token'] = "********"
|
|
template = "panel/panel_edit_user.html"
|
|
|
|
elif page == "remove_user":
|
|
user_id = bleach.clean(self.get_argument('id', None))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif user_id is None:
|
|
self.redirect("/panel/error?error=Invalid User ID")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
target_user = db_helper.get_user(user_id)
|
|
if not target_user:
|
|
self.redirect("/panel/error?error=Invalid User ID")
|
|
return
|
|
elif target_user['superuser']:
|
|
self.redirect("/panel/error?error=Cannot remove a superuser")
|
|
return
|
|
|
|
db_helper.remove_user(user_id)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Removed user {} (UID:{})".format(target_user['username'], user_id),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
elif page == "add_role":
|
|
user_roles = {}
|
|
for user in db_helper.get_all_users():
|
|
user_roles_list = db_helper.get_user_roles_names(user.user_id)
|
|
user_servers = db_helper.get_all_authorized_servers(user.user_id)
|
|
data = {user.user_id: user_roles_list}
|
|
user_roles.update(data)
|
|
page_data['new_role'] = True
|
|
page_data['role'] = {}
|
|
page_data['role']['role_name'] = ""
|
|
page_data['role']['role_id'] = -1
|
|
page_data['role']['created'] = "N/A"
|
|
page_data['role']['last_update'] = "N/A"
|
|
page_data['role']['servers'] = set()
|
|
page_data['user-roles'] = user_roles
|
|
page_data['users'] = db_helper.get_all_users()
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
|
|
page_data['servers_all'] = self.controller.list_defined_servers()
|
|
template = "panel/panel_edit_role.html"
|
|
|
|
elif page == "edit_role":
|
|
auth_servers = {}
|
|
|
|
user_roles = {}
|
|
for user in db_helper.get_all_users():
|
|
user_roles_list = db_helper.get_user_roles_names(user.user_id)
|
|
user_servers = db_helper.get_all_authorized_servers(user.user_id)
|
|
data = {user.user_id: user_roles_list}
|
|
user_roles.update(data)
|
|
page_data['new_role'] = False
|
|
role_id = self.get_argument('id', None)
|
|
page_data['role'] = db_helper.get_role(role_id)
|
|
page_data['servers_all'] = self.controller.list_defined_servers()
|
|
page_data['user-roles'] = user_roles
|
|
page_data['users'] = db_helper.get_all_users()
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif role_id is None:
|
|
self.redirect("/panel/error?error=Invalid Role ID")
|
|
return
|
|
|
|
template = "panel/panel_edit_role.html"
|
|
|
|
elif page == "remove_role":
|
|
role_id = bleach.clean(self.get_argument('id', None))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif role_id is None:
|
|
self.redirect("/panel/error?error=Invalid Role ID")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
target_role = db_helper.get_role(role_id)
|
|
if not target_role:
|
|
self.redirect("/panel/error?error=Invalid Role ID")
|
|
return
|
|
|
|
db_helper.remove_role(role_id)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Removed role {} (RID:{})".format(target_role['role_name'], role_id),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
elif page == "activity_logs":
|
|
page_data['audit_logs'] = db_helper.get_actity_log()
|
|
|
|
template = "panel/activity_logs.html"
|
|
|
|
self.render(
|
|
template,
|
|
data=page_data,
|
|
time=time,
|
|
utc_offset=(time.timezone * -1 / 60 / 60),
|
|
translate=self.translator.translate,
|
|
)
|
|
|
|
@tornado.web.authenticated
|
|
def post(self, page):
|
|
exec_user_data = json.loads(self.get_secure_cookie("user_data"))
|
|
exec_user_id = exec_user_data['user_id']
|
|
exec_user = db_helper.get_user(exec_user_id)
|
|
|
|
exec_user_role = set()
|
|
if exec_user['superuser'] == 1:
|
|
defined_servers = self.controller.list_defined_servers()
|
|
exec_user_role.add("Super User")
|
|
else:
|
|
defined_servers = self.controller.list_authorized_servers(exec_user_id)
|
|
for r in exec_user['roles']:
|
|
role = db_helper.get_role(r)
|
|
exec_user_role.add(role['role_name'])
|
|
|
|
if page == 'server_detail':
|
|
server_id = self.get_argument('id', None)
|
|
server_name = self.get_argument('server_name', None)
|
|
server_path = self.get_argument('server_path', None)
|
|
log_path = self.get_argument('log_path', None)
|
|
executable = self.get_argument('executable', None)
|
|
execution_command = self.get_argument('execution_command', None)
|
|
stop_command = self.get_argument('stop_command', None)
|
|
auto_start_delay = self.get_argument('auto_start_delay', '10')
|
|
server_ip = self.get_argument('server_ip', None)
|
|
server_port = self.get_argument('server_port', None)
|
|
executable_update_url = self.get_argument('executable_update_url', None)
|
|
auto_start = int(float(self.get_argument('auto_start', '0')))
|
|
crash_detection = int(float(self.get_argument('crash_detection', '0')))
|
|
logs_delete_after = int(float(self.get_argument('logs_delete_after', '0')))
|
|
subpage = self.get_argument('subpage', None)
|
|
|
|
if not exec_user['superuser']:
|
|
if not db_helper.server_id_authorized(server_id, exec_user_id):
|
|
if not db_helper.server_id_authorized_from_roles(server_id, exec_user_id):
|
|
self.redirect("/panel/error?error=Unauthorized access: invalid server id")
|
|
return
|
|
elif server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
else:
|
|
# does this server id exist?
|
|
if not db_helper.server_id_exists(server_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
Servers.update({
|
|
Servers.server_name: server_name,
|
|
Servers.path: server_path,
|
|
Servers.log_path: log_path,
|
|
Servers.executable: executable,
|
|
Servers.execution_command: execution_command,
|
|
Servers.stop_command: stop_command,
|
|
Servers.auto_start_delay: auto_start_delay,
|
|
Servers.server_ip: server_ip,
|
|
Servers.server_port: server_port,
|
|
Servers.auto_start: auto_start,
|
|
Servers.executable_update_url: executable_update_url,
|
|
Servers.crash_detection: crash_detection,
|
|
Servers.logs_delete_after: logs_delete_after,
|
|
}).where(Servers.server_id == server_id).execute()
|
|
|
|
self.controller.refresh_server_settings(server_id)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited server {} named {}".format(server_id, server_name),
|
|
server_id,
|
|
self.get_remote_ip())
|
|
|
|
self.redirect("/panel/server_detail?id={}&subpage=config".format(server_id))
|
|
|
|
if page == "server_backup":
|
|
logger.debug(self.request.arguments)
|
|
server_id = self.get_argument('id', None)
|
|
backup_path = bleach.clean(self.get_argument('backup_path', None))
|
|
max_backups = bleach.clean(self.get_argument('max_backups', None))
|
|
enabled = int(float(bleach.clean(self.get_argument('auto_enabled'), '0')))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif server_id is None:
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
else:
|
|
# does this server id exist?
|
|
if not db_helper.server_id_exists(server_id):
|
|
self.redirect("/panel/error?error=Invalid Server ID")
|
|
return
|
|
|
|
if backup_path is not None:
|
|
Servers.update({
|
|
Servers.backup_path: backup_path
|
|
}).where(Servers.server_id == server_id).execute()
|
|
db_helper.set_backup_config(server_id, max_backups=max_backups)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited server {}: updated backups".format(server_id),
|
|
server_id,
|
|
self.get_remote_ip())
|
|
self.tasks_manager.reload_schedule_from_db()
|
|
self.redirect("/panel/server_detail?id={}&subpage=backup".format(server_id))
|
|
|
|
elif page == "edit_user":
|
|
user_id = bleach.clean(self.get_argument('id', None))
|
|
username = bleach.clean(self.get_argument('username', None))
|
|
password0 = bleach.clean(self.get_argument('password0', None))
|
|
password1 = bleach.clean(self.get_argument('password1', None))
|
|
enabled = int(float(self.get_argument('enabled', '0')))
|
|
regen_api = int(float(self.get_argument('regen_api', '0')))
|
|
|
|
if not exec_user['superuser']:
|
|
user_data = {
|
|
"username": username,
|
|
"password": password0,
|
|
}
|
|
db_helper.update_user(user_id, user_data=user_data)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited user {} (UID:{}) password".format(username,
|
|
user_id),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
return
|
|
elif username is None or username == "":
|
|
self.redirect("/panel/error?error=Invalid username")
|
|
return
|
|
elif user_id is None:
|
|
self.redirect("/panel/error?error=Invalid User ID")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
if not db_helper.user_id_exists(user_id):
|
|
self.redirect("/panel/error?error=Invalid User ID")
|
|
return
|
|
|
|
if password0 != password1:
|
|
self.redirect("/panel/error?error=Passwords must match")
|
|
return
|
|
|
|
roles = set()
|
|
for role in db_helper.get_all_roles():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('role_{}_membership'.format(role.role_id), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
roles.add(role.role_id)
|
|
|
|
servers = set()
|
|
for server in self.controller.list_defined_servers():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('server_{}_access'.format(server['server_id']), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
servers.add(server['server_id'])
|
|
|
|
user_data = {
|
|
"username": username,
|
|
"password": password0,
|
|
"enabled": enabled,
|
|
"regen_api": regen_api,
|
|
"roles": roles,
|
|
"servers": servers,
|
|
}
|
|
db_helper.update_user(user_id, user_data=user_data)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited user {} (UID:{}) with roles {} and servers {}".format(username, user_id, roles, servers),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
|
|
elif page == "add_user":
|
|
username = bleach.clean(self.get_argument('username', None))
|
|
password0 = bleach.clean(self.get_argument('password0', None))
|
|
password1 = bleach.clean(self.get_argument('password1', None))
|
|
enabled = int(float(self.get_argument('enabled', '0')))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif username is None or username == "":
|
|
self.redirect("/panel/error?error=Invalid username")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
if db_helper.get_user_id_by_name(username) is not None:
|
|
self.redirect("/panel/error?error=User exists")
|
|
return
|
|
|
|
if password0 != password1:
|
|
self.redirect("/panel/error?error=Passwords must match")
|
|
return
|
|
|
|
roles = set()
|
|
for role in db_helper.get_all_roles():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('role_{}_membership'.format(role.role_id), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
roles.add(role.role_id)
|
|
|
|
servers = set()
|
|
for server in self.controller.list_defined_servers():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('server_{}_access'.format(server['server_id']), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
servers.add(server['server_id'])
|
|
|
|
user_id = db_helper.add_user(username, password=password0, enabled=enabled)
|
|
user_data = {
|
|
"roles": roles,
|
|
"servers": servers,
|
|
}
|
|
db_helper.update_user(user_id, user_data)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Added user {} (UID:{})".format(username, user_id),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited user {} (UID:{}) with roles {}".format(username, user_id, roles),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
elif page == "edit_role":
|
|
role_id = bleach.clean(self.get_argument('id', None))
|
|
role_name = bleach.clean(self.get_argument('role_name', None))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif role_name is None or role_name == "":
|
|
self.redirect("/panel/error?error=Invalid username")
|
|
return
|
|
elif role_id is None:
|
|
self.redirect("/panel/error?error=Invalid Role ID")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
if not db_helper.role_id_exists(role_id):
|
|
self.redirect("/panel/error?error=Invalid Role ID")
|
|
return
|
|
|
|
servers = set()
|
|
for server in self.controller.list_defined_servers():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('server_{}_access'.format(server['server_id']), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
servers.add(server['server_id'])
|
|
|
|
role_data = {
|
|
"role_name": role_name,
|
|
"servers": servers
|
|
}
|
|
db_helper.update_role(role_id, role_data=role_data)
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited role {} (RID:{}) with servers {}".format(role_name, role_id, servers),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
|
|
elif page == "add_role":
|
|
role_name = bleach.clean(self.get_argument('role_name', None))
|
|
|
|
if not exec_user['superuser']:
|
|
self.redirect("/panel/error?error=Unauthorized access: not superuser")
|
|
return
|
|
elif role_name is None or role_name == "":
|
|
self.redirect("/panel/error?error=Invalid role name")
|
|
return
|
|
else:
|
|
# does this user id exist?
|
|
if db_helper.get_roleid_by_name(role_name) is not None:
|
|
self.redirect("/panel/error?error=Role exists")
|
|
return
|
|
|
|
servers = set()
|
|
for server in self.controller.list_defined_servers():
|
|
argument = int(float(
|
|
bleach.clean(
|
|
self.get_argument('server_{}_access'.format(server['server_id']), '0')
|
|
)
|
|
))
|
|
if argument:
|
|
servers.add(server['server_id'])
|
|
|
|
role_id = db_helper.add_role(role_name)
|
|
db_helper.update_role(role_id, {"servers": servers})
|
|
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Added role {} (RID:{})".format(role_name, role_id),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
db_helper.add_to_audit_log(exec_user['user_id'],
|
|
"Edited role {} (RID:{}) with servers {}".format(role_name, role_id, servers),
|
|
server_id=0,
|
|
source_ip=self.get_remote_ip())
|
|
self.redirect("/panel/panel_config")
|
|
|
|
else:
|
|
self.set_status(404)
|
|
self.render(
|
|
"public/404.html",
|
|
translate=self.translator.translate,
|
|
)
|