crafty-4/app/classes/controllers/servers_controller.py
computergeek125 42ca4bb973 Refactored the yes to not pass objects in import
Merge Conflicts ᴙ Us
2022-04-11 00:23:55 -05:00

261 lines
9.1 KiB
Python

import os
import logging
import json
from app.classes.controllers.roles_controller import Roles_Controller
from app.classes.models.servers import helper_servers
from app.classes.models.users import helper_users, ApiKeys
from app.classes.models.server_permissions import (
Permissions_Servers,
Enum_Permissions_Server,
)
from app.classes.shared.helpers import Helpers
from app.classes.shared.main_models import db_shortcuts
logger = logging.getLogger(__name__)
class Servers_Controller:
def __init__(self, servers_helper):
self.servers_helper = servers_helper
# **********************************************************************************
# Generic Servers Methods
# **********************************************************************************
def create_server(
self,
name: str,
server_uuid: str,
server_dir: str,
backup_path: str,
server_command: str,
server_file: str,
server_log_file: str,
server_stop: str,
server_type: str,
server_port=25565,
):
return self.servers_helper.create_server(
name,
server_uuid,
server_dir,
backup_path,
server_command,
server_file,
server_log_file,
server_stop,
server_type,
server_port,
)
@staticmethod
def get_server_obj(server_id):
return helper_servers.get_server_obj(server_id)
@staticmethod
def update_server(server_obj):
return helper_servers.update_server(server_obj)
@staticmethod
def set_download(server_id):
return helper_servers.set_download(server_id)
@staticmethod
def finish_download(server_id):
return helper_servers.finish_download(server_id)
@staticmethod
def get_download_status(server_id):
return helper_servers.get_download_status(server_id)
def remove_server(self, server_id):
roles_list = Permissions_Servers.get_roles_from_server(server_id)
for role in roles_list:
role_id = role.role_id
role_data = Roles_Controller.get_role_with_servers(role_id)
role_data["servers"] = {server_id}
Permissions_Servers.delete_roles_permissions(role_id, role_data["servers"])
Permissions_Servers.remove_roles_of_server(server_id)
self.servers_helper.remove_server(server_id)
@staticmethod
def get_server_data_by_id(server_id):
return helper_servers.get_server_data_by_id(server_id)
# **********************************************************************************
# Servers Methods
# **********************************************************************************
@staticmethod
def get_all_defined_servers():
return helper_servers.get_all_defined_servers()
@staticmethod
def get_authorized_servers(user_id):
server_data = []
user_roles = helper_users.user_role_query(user_id)
for us in user_roles:
role_servers = Permissions_Servers.get_role_servers_from_role_id(us.role_id)
for role in role_servers:
server_data.append(helper_servers.get_server_data_by_id(role.server_id))
return server_data
@staticmethod
def get_all_servers_stats():
return helper_servers.get_all_servers_stats()
@staticmethod
def get_authorized_servers_stats_api_key(api_key: ApiKeys):
server_data = []
authorized_servers = Servers_Controller.get_authorized_servers(
api_key.user.user_id
)
for s in authorized_servers:
latest = helper_servers.get_latest_server_stats(s.get("server_id"))
key_permissions = Permissions_Servers.get_api_key_permissions_list(
api_key, s.get("server_id")
)
if Enum_Permissions_Server.Commands in key_permissions:
user_command_permission = True
else:
user_command_permission = False
server_data.append(
{
"server_data": s,
"stats": db_shortcuts.return_rows(latest)[0],
"user_command_permission": user_command_permission,
}
)
return server_data
@staticmethod
def get_authorized_servers_stats(user_id):
server_data = []
authorized_servers = Servers_Controller.get_authorized_servers(user_id)
for s in authorized_servers:
latest = helper_servers.get_latest_server_stats(s.get("server_id"))
# TODO
user_permissions = Permissions_Servers.get_user_id_permissions_list(
user_id, s.get("server_id")
)
if Enum_Permissions_Server.Commands in user_permissions:
user_command_permission = True
else:
user_command_permission = False
server_data.append(
{
"server_data": s,
"stats": db_shortcuts.return_rows(latest)[0],
"user_command_permission": user_command_permission,
}
)
return server_data
@staticmethod
def get_server_friendly_name(server_id):
return helper_servers.get_server_friendly_name(server_id)
# **********************************************************************************
# Servers_Stats Methods
# **********************************************************************************
@staticmethod
def get_server_stats_by_id(server_id):
return helper_servers.get_server_stats_by_id(server_id)
@staticmethod
def server_id_exists(server_id):
return helper_servers.server_id_exists(server_id)
@staticmethod
def get_server_type_by_id(server_id):
return helper_servers.get_server_type_by_id(server_id)
@staticmethod
def server_id_authorized(server_id_a, user_id):
user_roles = helper_users.user_role_query(user_id)
for role in user_roles:
for server_id_b in Permissions_Servers.get_role_servers_from_role_id(
role.role_id
):
if str(server_id_a) == str(server_id_b.server_id):
return True
return False
@staticmethod
def is_crashed(server_id):
return helper_servers.is_crashed(server_id)
@staticmethod
def server_id_authorized_api_key(server_id: str, api_key: ApiKeys) -> bool:
# TODO
return Servers_Controller.server_id_authorized(server_id, api_key.user.user_id)
# There is no view server permission
# permission_helper.both_have_perm(api_key)
def set_update(self, server_id, value):
return self.servers_helper.set_update(server_id, value)
@staticmethod
def get_TTL_without_player(server_id):
return helper_servers.get_TTL_without_player(server_id)
@staticmethod
def can_stop_no_players(server_id, time_limit):
return helper_servers.can_stop_no_players(server_id, time_limit)
def set_waiting_start(self, server_id, value):
self.servers_helper.set_waiting_start(server_id, value)
@staticmethod
def get_waiting_start(server_id):
return helper_servers.get_waiting_start(server_id)
@staticmethod
def get_update_status(server_id):
return helper_servers.get_update_status(server_id)
# **********************************************************************************
# Servers Helpers Methods
# **********************************************************************************
@staticmethod
def get_banned_players(server_id):
stats = helper_servers.get_server_stats_by_id(server_id)
server_path = stats["server_id"]["path"]
path = os.path.join(server_path, "banned-players.json")
try:
with open(
Helpers.get_os_understandable_path(path), encoding="utf-8"
) as file:
content = file.read()
file.close()
except Exception as ex:
print(ex)
return None
return json.loads(content)
def check_for_old_logs(self):
servers = helper_servers.get_all_defined_servers()
for server in servers:
logs_path = os.path.split(server["log_path"])[0]
latest_log_file = os.path.split(server["log_path"])[1]
logs_delete_after = int(server["logs_delete_after"])
if logs_delete_after == 0:
continue
log_files = list(
filter(lambda val: val != latest_log_file, os.listdir(logs_path))
)
for log_file in log_files:
log_file_path = os.path.join(logs_path, log_file)
if Helpers.check_file_exists(
log_file_path
) and Helpers.is_file_older_than_x_days(
log_file_path, logs_delete_after
):
os.remove(log_file_path)