import os import sys import json from threading import Thread import time import argparse import logging.config import signal import peewee from app.classes.models.users import HelperUsers from app.classes.shared.console import Console from app.classes.shared.helpers import Helpers console = Console() helper = Helpers() if helper.check_root(): Console.critical( "Root detected. Root/Admin access denied. " "Run Crafty again with non-elevated permissions." ) time.sleep(5) Console.critical("Crafty shutting down. Root/Admin access denied.") sys.exit(0) # pylint: disable=wrong-import-position try: from app.classes.models.base_model import database_proxy from app.classes.shared.main_models import DatabaseBuilder from app.classes.shared.tasks import TasksManager from app.classes.shared.main_controller import Controller from app.classes.shared.migration import MigrationManager from app.classes.shared.command import MainPrompt except ModuleNotFoundError as err: helper.auto_installer_fix(err) def do_intro(): logger.info("***** Crafty Controller Started *****") version = helper.get_version_string() intro = f""" {'/' * 75} #{("Welcome to Crafty Controller - v." + version).center(73, " ")}# {'/' * 75} #{"Server Manager / Web Portal for your Minecraft server".center(73, " ")}# #{"Homepage: www.craftycontrol.com".center(73, " ")}# {'/' * 75} """ Console.magenta(intro) def setup_logging(debug=True): logging_config_file = os.path.join(os.path.curdir, "app", "config", "logging.json") if os.path.exists(logging_config_file): # open our logging config file with open(logging_config_file, "rt", encoding="utf-8") as f: logging_config = json.load(f) if debug: logging_config["loggers"][""]["level"] = "DEBUG" logging.config.dictConfig(logging_config) else: logging.basicConfig(level=logging.DEBUG) logging.warning(f"Unable to read logging config from {logging_config_file}") Console.critical(f"Unable to read logging config from {logging_config_file}") # Our Main Starter if __name__ == "__main__": parser = argparse.ArgumentParser("Crafty Controller - A Server Management System") parser.add_argument( "-i", "--ignore", action="store_true", help="Ignore session.lock files" ) parser.add_argument( "-v", "--verbose", action="store_true", help="Sets logging level to debug." ) parser.add_argument( "-d", "--daemon", action="store_true", help="Runs Crafty in daemon mode (no prompt)", ) args = parser.parse_args() helper.ensure_logging_setup() setup_logging(debug=args.verbose) # setting up the logger object logger = logging.getLogger(__name__) Console.cyan(f"Logging set to: {logger.level}") peewee_logger = logging.getLogger("peewee") peewee_logger.setLevel(logging.INFO) # print our pretty start message do_intro() # our session file, helps prevent multiple controller agents on the same machine. helper.create_session_file(ignore=args.ignore) # start the database database = peewee.SqliteDatabase( helper.db_path, pragmas={"journal_mode": "wal", "cache_size": -1024 * 10} ) database_proxy.initialize(database) migration_manager = MigrationManager(database, helper) migration_manager.up() # Automatically runs migrations # do our installer stuff user_helper = HelperUsers(database, helper) installer = DatabaseBuilder(database, helper, user_helper) FRESH_INSTALL = installer.is_fresh_install() if FRESH_INSTALL: Console.debug("Fresh install detected") Console.warning( f"We have detected a fresh install. Please be sure to forward " f"Crafty's port, {helper.get_setting('https_port')}, " f"through your router/firewall if you would like to be able " f"to access Crafty remotely." ) installer.default_settings() else: Console.debug("Existing install detected") # now the tables are created, we can load the tasks_manger and server controller controller = Controller(database, helper) tasks_manager = TasksManager(helper, controller) tasks_manager.start_webserver() def sigterm_handler(*sig): print() # for newline logger.info( f"Recieved {signal.Signals(sig[0]).name} [{sig[0]}], stopping Crafty..." ) Console.info( f"Recieved {signal.Signals(sig[0]).name} [{sig[0]}], stopping Crafty..." ) tasks_manager._main_graceful_exit() Crafty.universal_exit() signal.signal(signal.SIGTERM, sigterm_handler) # slowing down reporting just for a 1/2 second so messages look cleaner time.sleep(0.5) # init servers logger.info("Initializing all servers defined") Console.info("Initializing all servers defined") controller.init_all_servers() servers = controller.list_defined_servers() def tasks_starter(): # start stats logging tasks_manager.start_stats_recording() # once the controller is up and stats are logging, we can kick off # the scheduler officially tasks_manager.start_scheduler() # refresh our cache and schedule for every 12 hoursour cache refresh # for serverjars.com tasks_manager.serverjar_cache_refresher() tasks_starter_thread = Thread(target=tasks_starter, name="tasks_starter") tasks_starter_thread.start() def internet_check(): logger.info("Checking Internet. This may take a minute.") Console.info("\nChecking Internet. This may take a minute.") if not helper.check_internet(): logger.warning( "We have detected the machine running Crafty has no " "connection to the internet. Client connections to " "the server may be limited." ) Console.warning( "We have detected the machine running Crafty has no " "connection to the internet. Client connections to " "the server may be limited." ) internet_check_thread = Thread(target=internet_check, name="internet_check") internet_check_thread.start() if not controller.check_system_user(): controller.add_system_user() Crafty = MainPrompt(helper, tasks_manager, migration_manager) project_root = os.path.dirname(__file__) controller.set_project_root(project_root) controller.clear_unexecuted_commands() controller.clear_support_status() if not args.daemon: try: Crafty.cmdloop() except KeyboardInterrupt: print() # for newline logger.info("Recieved SIGINT, stopping Crafty...") Console.info("Recieved SIGINT, stopping Crafty...") tasks_manager._main_graceful_exit() Crafty.universal_exit() else: print("Crafty started in daemon mode, no shell will be printed") while True: try: if tasks_manager.get_main_thread_run_status(): break time.sleep(1) except KeyboardInterrupt: logger.info("Recieved SIGINT, stopping Crafty...") Console.info("Recieved SIGINT, stopping Crafty...") break tasks_manager._main_graceful_exit() Crafty.universal_exit()