mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
203 lines
9.4 KiB
Python
203 lines
9.4 KiB
Python
import sys
|
|
import json
|
|
import logging
|
|
import os
|
|
import shutil
|
|
|
|
from app.classes.shared.console import console
|
|
from app.classes.web.base_handler import BaseHandler
|
|
from app.classes.shared.models import db_helper
|
|
from app.classes.minecraft.serverjars import server_jar_obj
|
|
from app.classes.shared.helpers import helper
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
import tornado.web
|
|
import tornado.escape
|
|
import bleach
|
|
|
|
except ModuleNotFoundError as e:
|
|
logger.critical("Import Error: Unable to load {} module".format(e.name), exc_info=True)
|
|
console.critical("Import Error: Unable to load {} module".format(e.name))
|
|
sys.exit(1)
|
|
|
|
|
|
class ServerHandler(BaseHandler):
|
|
|
|
@tornado.web.authenticated
|
|
def get(self, page):
|
|
# name = tornado.escape.json_decode(self.current_user)
|
|
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'])
|
|
|
|
template = "public/404.html"
|
|
|
|
page_data = {
|
|
'version_data': helper.get_version_string(),
|
|
'user_data': exec_user_data,
|
|
'user_role' : exec_user_role,
|
|
'server_stats': {
|
|
'total': len(self.controller.list_defined_servers()),
|
|
'running': len(self.controller.list_running_servers()),
|
|
'stopped': (len(self.controller.list_defined_servers()) - len(self.controller.list_running_servers()))
|
|
},
|
|
'hosts_data': db_helper.get_latest_hosts_stats(),
|
|
'menu_servers': defined_servers,
|
|
'show_contribute': helper.get_setting("show_contribute_link", True)
|
|
}
|
|
|
|
if page == "step1":
|
|
|
|
page_data['server_types'] = server_jar_obj.get_serverjar_data_sorted()
|
|
template = "server/wizard.html"
|
|
|
|
self.render(
|
|
template,
|
|
data=page_data,
|
|
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)
|
|
|
|
template = "public/404.html"
|
|
page_data = {
|
|
'version_data': "version_data_here",
|
|
'user_data': exec_user_data,
|
|
'show_contribute': helper.get_setting("show_contribute_link", True)
|
|
}
|
|
|
|
if page == "command":
|
|
server_id = bleach.clean(self.get_argument("id", None))
|
|
command = bleach.clean(self.get_argument("command", None))
|
|
|
|
if server_id is not None:
|
|
if command == "clone_server":
|
|
def is_name_used(name):
|
|
for server in db_helper.get_all_defined_servers():
|
|
if server['server_name'] == name:
|
|
return True
|
|
return
|
|
|
|
server_data = db_helper.get_server_data_by_id(server_id)
|
|
server_uuid = server_data.get('server_uuid')
|
|
new_server_name = server_data.get('server_name') + " (Copy)"
|
|
|
|
name_counter = 1
|
|
while is_name_used(new_server_name):
|
|
name_counter += 1
|
|
new_server_name = server_data.get('server_name') + " (Copy {})".format(name_counter)
|
|
|
|
new_server_uuid = helper.create_uuid()
|
|
while os.path.exists(os.path.join(helper.servers_dir, new_server_uuid)):
|
|
new_server_uuid = helper.create_uuid()
|
|
new_server_path = os.path.join(helper.servers_dir, new_server_uuid)
|
|
|
|
# copy the old server
|
|
shutil.copytree(server_data.get('path'), new_server_path)
|
|
|
|
# TODO get old server DB data to individual variables
|
|
stop_command = server_data.get('stop_command')
|
|
new_server_command = str(server_data.get('execution_command')).replace(server_uuid, new_server_uuid)
|
|
new_executable = server_data.get('executable')
|
|
new_server_log_file = str(server_data.get('log_path')).replace(server_uuid, new_server_uuid)
|
|
auto_start = server_data.get('auto_start')
|
|
auto_start_delay = server_data.get('auto_start_delay')
|
|
crash_detection = server_data.get('crash_detection')
|
|
server_port = server_data.get('server_port')
|
|
|
|
db_helper.create_server(new_server_name, new_server_uuid, new_server_path, "", new_server_command, new_executable, new_server_log_file, stop_command, server_port)
|
|
|
|
self.controller.init_all_servers()
|
|
|
|
return
|
|
|
|
db_helper.send_command(exec_user_data['user_id'], server_id, self.get_remote_ip(), command)
|
|
|
|
if page == "step1":
|
|
|
|
server = bleach.clean(self.get_argument('server', ''))
|
|
server_name = bleach.clean(self.get_argument('server_name', ''))
|
|
min_mem = bleach.clean(self.get_argument('min_memory', ''))
|
|
max_mem = bleach.clean(self.get_argument('max_memory', ''))
|
|
port = bleach.clean(self.get_argument('port', ''))
|
|
import_type = bleach.clean(self.get_argument('create_type', ''))
|
|
import_server_path = bleach.clean(self.get_argument('server_path', ''))
|
|
import_server_jar = bleach.clean(self.get_argument('server_jar', ''))
|
|
server_parts = server.split("|")
|
|
|
|
if not server_name:
|
|
self.redirect("/panel/error?error=Server name cannot be empty!")
|
|
return
|
|
|
|
if import_type == 'import_jar':
|
|
good_path = self.controller.verify_jar_server(import_server_path, import_server_jar)
|
|
|
|
if not good_path:
|
|
self.redirect("/panel/error?error=Server path or Server Jar not found!")
|
|
return
|
|
|
|
new_server_id = self.controller.import_jar_server(server_name, import_server_path,import_server_jar, min_mem, max_mem, port)
|
|
db_helper.add_to_audit_log(exec_user_data['user_id'],
|
|
"imported a jar server named \"{}\"".format(server_name), # Example: Admin imported a server named "old creative"
|
|
new_server_id,
|
|
self.get_remote_ip())
|
|
elif import_type == 'import_zip':
|
|
# here import_server_path means the zip path
|
|
good_path = self.controller.verify_zip_server(import_server_path)
|
|
if not good_path:
|
|
self.redirect("/panel/error?error=Zip file not found!")
|
|
return
|
|
|
|
new_server_id = self.controller.import_zip_server(server_name, import_server_path,import_server_jar, min_mem, max_mem, port)
|
|
if new_server_id == "false":
|
|
self.redirect("/panel/error?error=Zip file not accessible! You can fix this permissions issue with sudo chown -R crafty:crafty {} And sudo chmod 2775 -R {}".format(import_server_path, import_server_path))
|
|
return
|
|
db_helper.add_to_audit_log(exec_user_data['user_id'],
|
|
"imported a zip server named \"{}\"".format(server_name), # Example: Admin imported a server named "old creative"
|
|
new_server_id,
|
|
self.get_remote_ip())
|
|
else:
|
|
if len(server_parts) != 2:
|
|
self.redirect("/panel/error?error=Invalid server data")
|
|
return
|
|
server_type, server_version = server_parts
|
|
# TODO: add server type check here and call the correct server add functions if not a jar
|
|
role_ids = db_helper.get_user_roles_id(exec_user_id)
|
|
new_server_id = self.controller.create_jar_server(server_type, server_version, server_name, min_mem, max_mem, port)
|
|
db_helper.add_to_audit_log(exec_user_data['user_id'],
|
|
"created a {} {} server named \"{}\"".format(server_version, str(server_type).capitalize(), server_name), # Example: Admin created a 1.16.5 Bukkit server named "survival"
|
|
new_server_id,
|
|
self.get_remote_ip())
|
|
|
|
#These lines create a new Role for the Server with full permissions and add the user to it
|
|
role_id = db_helper.add_role("Creator of Server with id={}".format(new_server_id))
|
|
db_helper.add_role_server(new_server_id, role_id, "11111111")
|
|
db_helper.add_role_to_user(exec_user_id, role_id)
|
|
|
|
self.controller.stats.record_stats()
|
|
self.redirect("/panel/dashboard")
|
|
|
|
self.render(
|
|
template,
|
|
data=page_data,
|
|
translate=self.translator.translate,
|
|
)
|