mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
322 lines
12 KiB
Python
322 lines
12 KiB
Python
import os
|
|
import sys
|
|
import logging
|
|
import datetime
|
|
|
|
from app.classes.shared.helpers import helper
|
|
from app.classes.shared.console import console
|
|
from app.classes.shared.main_models import db_helper
|
|
from app.classes.models.users import Users, users_helper
|
|
from app.classes.models.servers import Servers, servers_helper
|
|
from app.classes.web.websocket_helper import websocket_helper
|
|
|
|
|
|
logger = logging.getLogger(__name__)
|
|
peewee_logger = logging.getLogger('peewee')
|
|
peewee_logger.setLevel(logging.INFO)
|
|
|
|
try:
|
|
from peewee import *
|
|
from playhouse.shortcuts import model_to_dict
|
|
from enum import Enum
|
|
import yaml
|
|
|
|
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)
|
|
|
|
database = SqliteDatabase(helper.db_path, pragmas={
|
|
'journal_mode': 'wal',
|
|
'cache_size': -1024 * 10})
|
|
|
|
#************************************************************************************************
|
|
# Audit_Log Class
|
|
#************************************************************************************************
|
|
class Audit_Log(Model):
|
|
audit_id = AutoField()
|
|
created = DateTimeField(default=datetime.datetime.now)
|
|
user_name = CharField(default="")
|
|
user_id = IntegerField(default=0, index=True)
|
|
source_ip = CharField(default='127.0.0.1')
|
|
server_id = IntegerField(default=None, index=True) # When auditing global events, use server ID 0
|
|
log_msg = TextField(default='')
|
|
|
|
class Meta:
|
|
database = database
|
|
|
|
|
|
#************************************************************************************************
|
|
# Host_Stats Class
|
|
#************************************************************************************************
|
|
class Host_Stats(Model):
|
|
time = DateTimeField(default=datetime.datetime.now, index=True)
|
|
boot_time = CharField(default="")
|
|
cpu_usage = FloatField(default=0)
|
|
cpu_cores = IntegerField(default=0)
|
|
cpu_cur_freq = FloatField(default=0)
|
|
cpu_max_freq = FloatField(default=0)
|
|
mem_percent = FloatField(default=0)
|
|
mem_usage = CharField(default="")
|
|
mem_total = CharField(default="")
|
|
disk_json = TextField(default="")
|
|
|
|
class Meta:
|
|
table_name = "host_stats"
|
|
database = database
|
|
|
|
|
|
#************************************************************************************************
|
|
# Commands Class
|
|
#************************************************************************************************
|
|
class Commands(Model):
|
|
command_id = AutoField()
|
|
created = DateTimeField(default=datetime.datetime.now)
|
|
server_id = ForeignKeyField(Servers, backref='server', index=True)
|
|
user = ForeignKeyField(Users, backref='user', index=True)
|
|
source_ip = CharField(default='127.0.0.1')
|
|
command = CharField(default='')
|
|
executed = BooleanField(default=False)
|
|
|
|
class Meta:
|
|
table_name = "commands"
|
|
database = database
|
|
|
|
|
|
#************************************************************************************************
|
|
# Webhooks Class
|
|
#************************************************************************************************
|
|
class Webhooks(Model):
|
|
id = AutoField()
|
|
name = CharField(max_length=64, unique=True, index=True)
|
|
method = CharField(default="POST")
|
|
url = CharField(unique=True)
|
|
event = CharField(default="")
|
|
send_data = BooleanField(default=True)
|
|
|
|
class Meta:
|
|
table_name = "webhooks"
|
|
database = database
|
|
|
|
|
|
#************************************************************************************************
|
|
# Schedules Class
|
|
#************************************************************************************************
|
|
class Schedules(Model):
|
|
schedule_id = IntegerField(unique=True, primary_key=True)
|
|
server_id = ForeignKeyField(Servers, backref='schedule_server')
|
|
enabled = BooleanField()
|
|
action = CharField()
|
|
interval = IntegerField()
|
|
interval_type = CharField()
|
|
start_time = CharField(null=True)
|
|
command = CharField(null=True)
|
|
comment = CharField()
|
|
|
|
class Meta:
|
|
table_name = 'schedules'
|
|
database = database
|
|
|
|
|
|
#************************************************************************************************
|
|
# Backups Class
|
|
#************************************************************************************************
|
|
class Backups(Model):
|
|
directories = CharField(null=True)
|
|
max_backups = IntegerField()
|
|
server_id = ForeignKeyField(Servers, backref='backups_server')
|
|
schedule_id = ForeignKeyField(Schedules, backref='backups_schedule')
|
|
|
|
class Meta:
|
|
table_name = 'backups'
|
|
database = database
|
|
|
|
class helpers_management:
|
|
|
|
#************************************************************************************************
|
|
# Host_Stats Methods
|
|
#************************************************************************************************
|
|
@staticmethod
|
|
def get_latest_hosts_stats():
|
|
query = Host_Stats.select().order_by(Host_Stats.id.desc()).get()
|
|
return model_to_dict(query)
|
|
|
|
#************************************************************************************************
|
|
# Commands Methods
|
|
#************************************************************************************************
|
|
@staticmethod
|
|
def add_command(server_id, user_id, remote_ip, command):
|
|
Commands.insert({
|
|
Commands.server_id: server_id,
|
|
Commands.user: user_id,
|
|
Commands.source_ip: remote_ip,
|
|
Commands.command: command
|
|
}).execute()
|
|
|
|
@staticmethod
|
|
def get_unactioned_commands():
|
|
query = Commands.select().where(Commands.executed == 0)
|
|
return db_helper.return_rows(query)
|
|
|
|
@staticmethod
|
|
def mark_command_complete(command_id=None):
|
|
if command_id is not None:
|
|
logger.debug("Marking Command {} completed".format(command_id))
|
|
Commands.update({
|
|
Commands.executed: True
|
|
}).where(Commands.command_id == command_id).execute()
|
|
|
|
#************************************************************************************************
|
|
# Audit_Log Methods
|
|
#************************************************************************************************
|
|
@staticmethod
|
|
def get_actity_log():
|
|
q = Audit_Log.select()
|
|
return db_helper.return_db_rows(q)
|
|
|
|
@staticmethod
|
|
def add_to_audit_log(user_id, log_msg, server_id=None, source_ip=None):
|
|
logger.debug("Adding to audit log User:{} - Message: {} ".format(user_id, log_msg))
|
|
user_data = users_helper.get_user(user_id)
|
|
|
|
audit_msg = "{} {}".format(str(user_data['username']).capitalize(), log_msg)
|
|
|
|
websocket_helper.broadcast('notification', audit_msg)
|
|
|
|
Audit_Log.insert({
|
|
Audit_Log.user_name: user_data['username'],
|
|
Audit_Log.user_id: user_id,
|
|
Audit_Log.server_id: server_id,
|
|
Audit_Log.log_msg: audit_msg,
|
|
Audit_Log.source_ip: source_ip
|
|
}).execute()
|
|
|
|
@staticmethod
|
|
def add_to_audit_log_raw(user_name, user_id, server_id, log_msg, source_ip):
|
|
Audit_Log.insert({
|
|
Audit_Log.user_name: user_name,
|
|
Audit_Log.user_id: user_id,
|
|
Audit_Log.server_id: server_id,
|
|
Audit_Log.log_msg: log_msg,
|
|
Audit_Log.source_ip: source_ip
|
|
}).execute()
|
|
|
|
#************************************************************************************************
|
|
# Schedules Methods
|
|
#************************************************************************************************
|
|
@staticmethod
|
|
def create_scheduled_task(server_id, action, interval, interval_type, start_time, command, comment=None, enabled=True):
|
|
sch_id = Schedules.insert({
|
|
Schedules.server_id: server_id,
|
|
Schedules.action: action,
|
|
Schedules.enabled: enabled,
|
|
Schedules.interval: interval,
|
|
Schedules.interval_type: interval_type,
|
|
Schedules.start_time: start_time,
|
|
Schedules.command: command,
|
|
Schedules.comment: comment
|
|
}).execute()
|
|
return sch_id
|
|
|
|
@staticmethod
|
|
def delete_scheduled_task(schedule_id):
|
|
sch = Schedules.get(Schedules.schedule_id == schedule_id)
|
|
return Schedules.delete_instance(sch)
|
|
|
|
@staticmethod
|
|
def update_scheduled_task(schedule_id, updates):
|
|
Schedules.update(updates).where(Schedules.schedule_id == schedule_id).execute()
|
|
|
|
@staticmethod
|
|
def delete_scheduled_task_by_server(server_id):
|
|
Schedules.delete().where(Schedules.server_id == int(server_id)).execute()
|
|
|
|
@staticmethod
|
|
def get_scheduled_task(schedule_id):
|
|
return model_to_dict(Schedules.get(Schedules.schedule_id == schedule_id)).execute()
|
|
|
|
@staticmethod
|
|
def get_schedules_by_server(server_id):
|
|
return Schedules.select().where(Schedules.server_id == server_id).execute()
|
|
|
|
@staticmethod
|
|
def get_schedules_all():
|
|
return Schedules.select().execute()
|
|
|
|
@staticmethod
|
|
def get_schedules_enabled():
|
|
return Schedules.select().where(Schedules.enabled == True).execute()
|
|
|
|
#************************************************************************************************
|
|
# Backups Methods
|
|
#************************************************************************************************
|
|
@staticmethod
|
|
def get_backup_config(server_id):
|
|
try:
|
|
row = Backups.select().where(Backups.server_id == server_id).join(Schedules).join(Servers)[0]
|
|
conf = {
|
|
"backup_path": row.server_id.backup_path,
|
|
"directories": row.directories,
|
|
"max_backups": row.max_backups,
|
|
"auto_enabled": row.schedule_id.enabled,
|
|
"server_id": row.server_id.server_id
|
|
}
|
|
except IndexError:
|
|
conf = {
|
|
"backup_path": None,
|
|
"directories": None,
|
|
"max_backups": 0,
|
|
"auto_enabled": True,
|
|
"server_id": server_id
|
|
}
|
|
return conf
|
|
|
|
@staticmethod
|
|
def set_backup_config(server_id: int, backup_path: str = None, max_backups: int = None, auto_enabled: bool = True):
|
|
logger.debug("Updating server {} backup config with {}".format(server_id, locals()))
|
|
try:
|
|
row = Backups.select().where(Backups.server_id == server_id).join(Schedules).join(Servers)[0]
|
|
new_row = False
|
|
conf = {}
|
|
schd = {}
|
|
except IndexError:
|
|
conf = {
|
|
"directories": None,
|
|
"max_backups": 0,
|
|
"server_id": server_id
|
|
}
|
|
schd = {
|
|
"enabled": True,
|
|
"action": "backup_server",
|
|
"interval_type": "days",
|
|
"interval": 1,
|
|
"start_time": "00:00",
|
|
"server_id": server_id,
|
|
"comment": "Default backup job"
|
|
}
|
|
new_row = True
|
|
if max_backups is not None:
|
|
conf['max_backups'] = max_backups
|
|
schd['enabled'] = bool(auto_enabled)
|
|
if not new_row:
|
|
with database.atomic():
|
|
if backup_path is not None:
|
|
u1 = Servers.update(backup_path=backup_path).where(Servers.server_id == server_id).execute()
|
|
else:
|
|
u1 = 0
|
|
u2 = Backups.update(conf).where(Backups.server_id == server_id).execute()
|
|
u3 = Schedules.update(schd).where(Schedules.schedule_id == row.schedule_id).execute()
|
|
logger.debug("Updating existing backup record. {}+{}+{} rows affected".format(u1, u2, u3))
|
|
else:
|
|
with database.atomic():
|
|
conf["server_id"] = server_id
|
|
if backup_path is not None:
|
|
u = Servers.update(backup_path=backup_path).where(Servers.server_id == server_id)
|
|
s = Schedules.create(**schd)
|
|
conf['schedule_id'] = s.schedule_id
|
|
b = Backups.create(**conf)
|
|
logger.debug("Creating new backup record.")
|
|
|
|
|
|
management_helper = helpers_management()
|