crafty-4/main.py

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_manager 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, controller)

    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()