mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
99 lines
3.4 KiB
Python
99 lines
3.4 KiB
Python
import os
|
|
import time
|
|
import logging
|
|
import sys
|
|
import yaml
|
|
import asyncio
|
|
import shutil
|
|
import tempfile
|
|
import zipfile
|
|
from distutils import dir_util
|
|
|
|
from app.classes.shared.helpers import helper
|
|
from app.classes.shared.console import console
|
|
|
|
from app.classes.models.roles import roles_helper
|
|
from app.classes.models.server_permissions import server_permissions
|
|
from app.classes.models.users import users_helper
|
|
|
|
from app.classes.shared.server import Server
|
|
from app.classes.minecraft.server_props import ServerProps
|
|
from app.classes.minecraft.serverjars import server_jar_obj
|
|
from app.classes.minecraft.stats import Stats
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
class Roles_Controller:
|
|
|
|
@staticmethod
|
|
def get_all_roles():
|
|
return roles_helper.get_all_roles()
|
|
|
|
@staticmethod
|
|
def get_roleid_by_name(role_name):
|
|
return roles_helper.get_roleid_by_name(role_name)
|
|
|
|
@staticmethod
|
|
def get_role(role_id):
|
|
return roles_helper.get_role(role_id)
|
|
|
|
|
|
@staticmethod
|
|
def update_role(role_id: str, role_data = None, permissions_mask: str = "00000000"):
|
|
if role_data is None:
|
|
role_data = {}
|
|
base_data = Roles_Controller.get_role_with_servers(role_id)
|
|
up_data = {}
|
|
added_servers = set()
|
|
edited_servers = set()
|
|
removed_servers = set()
|
|
for key in role_data:
|
|
if key == "role_id":
|
|
continue
|
|
elif key == "servers":
|
|
added_servers = role_data['servers'].difference(base_data['servers'])
|
|
removed_servers = base_data['servers'].difference(role_data['servers'])
|
|
elif base_data[key] != role_data[key]:
|
|
up_data[key] = role_data[key]
|
|
up_data['last_update'] = helper.get_time_as_string()
|
|
logger.debug("role: {} +server:{} -server{}".format(role_data, added_servers, removed_servers))
|
|
for server in added_servers:
|
|
server_permissions.get_or_create(role_id, server, permissions_mask)
|
|
for server in base_data['servers']:
|
|
server_permissions.update_role_permission(role_id, server, permissions_mask)
|
|
# TODO: This is horribly inefficient and we should be using bulk queries but im going for functionality at this point
|
|
server_permissions.delete_roles_permissions(role_id, removed_servers)
|
|
if up_data:
|
|
roles_helper.update_role(role_id, up_data)
|
|
|
|
@staticmethod
|
|
def add_role(role_name):
|
|
return roles_helper.add_role(role_name)
|
|
|
|
@staticmethod
|
|
def remove_role(role_id):
|
|
role_data = Roles_Controller.get_role_with_servers(role_id)
|
|
server_permissions.delete_roles_permissions(role_id, role_data['servers'])
|
|
users_helper.remove_roles_from_role_id(role_id)
|
|
return roles_helper.remove_role(role_id)
|
|
|
|
@staticmethod
|
|
def role_id_exists(role_id):
|
|
return roles_helper.role_id_exists(role_id)
|
|
|
|
@staticmethod
|
|
def get_role_with_servers(role_id):
|
|
role = roles_helper.get_role(role_id)
|
|
|
|
if role:
|
|
servers_query = server_permissions.get_servers_from_role(role_id)
|
|
# TODO: this query needs to be narrower
|
|
servers = set()
|
|
for s in servers_query:
|
|
servers.add(s.server_id.server_id)
|
|
role['servers'] = servers
|
|
#logger.debug("role: ({}) {}".format(role_id, role))
|
|
return role
|
|
else:
|
|
#logger.debug("role: ({}) {}".format(role_id, {}))
|
|
return {} |