Rename bs to bombsquad_server

bombsquad_server

@@ -0,0 +1,912 @@
+#!/usr/bin/env -S python3.10 -O
+
+# Released under the MIT License. See LICENSE for details.
+
+from __future__ import annotations
+import os,signal,subprocess, sys, time, _thread, logging, platform, shutil, json
+from pathlib import Path
+from threading import Lock, Thread, current_thread, Timer
+from typing import TYPE_CHECKING
+from nbstreamreader import NonBlockingStreamReader as NBSR
+from datetime import datetime
+#import requests
+
+ERROR_LOGGING=False
+
+def migrate_to_aarch():
+    maps = ["BridgitMash.so","FloatingIsland.so","InTheAir.so"]
+    games = ["CanonFight.so","DuelElimination.so","FlappyBird.so","LaserTracer.so","MonkeyClimb.so","OneNightNoStand.so","RealSoccer.so",
+    "SquidRace.so","StumbleRace.so","SubwayRun.so","UFOAttackGame.so"]
+    features = ["StumbledScoreScreen.so"]
+    tools = ["corelib.so"]
+    root = os.path.realpath(".")+"/dist/ba_root"
+    for map in maps:
+        shutil.copy(root+"/mods/aarch64/"+map,root+'/mods/maps/'+map)
+    for game in games:
+        shutil.copyfile(root+"/mods/aarch64/"+game,root+'/mods/games/'+game)
+    for f in features:
+        shutil.copyfile(root+"/mods/aarch64/"+f,root+'/mods/features/'+f)
+    for t in tools:
+        shutil.copyfile(root+"/mods/aarch64/"+t,root+'/mods/tools/'+t)
+    with open(".we_are_good","w") as f:
+        pass
+
+# by default we have x86_64 setup
+# if we found aarch64 system copy required files
+if platform.processor() == 'aarch64':
+    print("We are on aarch64 system")
+    if os.path.exists(".we_are_good"):
+        pass
+    else:
+        migrate_to_aarch()
+
+# We make use of the bacommon and efro packages as well as site-packages
+# included with our bundled Ballistica dist, so we need to add those paths
+# before we import them.
+sys.path += [
+    str(Path(Path(__file__).parent, 'dist', 'ba_data', 'python')),
+    str(Path(Path(__file__).parent, 'dist', 'ba_data', 'python-site-packages'))
+]
+
+from bacommon.servermanager import ServerConfig, StartServerModeCommand
+from efro.dataclassio import dataclass_from_dict, dataclass_validate
+from efro.error import CleanError
+from efro.terminal import Clr
+
+if TYPE_CHECKING:
+    from typing import Optional, Union
+    from types import FrameType
+    from bacommon.servermanager import ServerCommand
+
+VERSION_STR = '1.3'
+print("----------------------------------------------------------------------------------------------------------------------------------------")
+print()
+
+
+class ServerManagerApp:
+    """ An app which manages BallisticaCore server execution.
+
+    Handles configuring, launching, re-launching, and otherwise
+    managing BallisticaCore operating in server mode.
+    """
+
+    # How many seconds we wait after asking our subprocess to do an immediate
+    # shutdown before bringing down the hammer.
+    IMMEDIATE_SHUTDOWN_TIME_LIMIT = 5.0
+
+    def __init__(self) -> None:
+        self._config_path = 'config.yaml'
+        self._user_provided_config_path = False
+        self._config = ServerConfig()
+        self._ba_root_path = os.path.abspath('dist/ba_root')
+        self._interactive = sys.stdin.isatty()
+        self._wrapper_shutdown_desired = False
+        self._done = False
+        self._subprocess_commands: list[Union[str, ServerCommand]] = []
+        self._subprocess_commands_lock = Lock()
+        self._subprocess_force_kill_time: Optional[float] = None
+        self._auto_restart = True
+        self._config_auto_restart = True
+        self._config_mtime: Optional[float] = None
+        self._last_config_mtime_check_time: Optional[float] = None
+        self._should_report_subprocess_error = False
+        self._running = False
+        self._interpreter_start_time: Optional[float] = None
+        self._subprocess: Optional[subprocess.Popen[bytes]] = None
+        self._subprocess_launch_time: Optional[float] = None
+        self._subprocess_sent_config_auto_restart = False
+        self._subprocess_sent_clean_exit = False
+        self._subprocess_sent_unclean_exit = False
+        self._subprocess_thread: Optional[Thread] = None
+        self._subprocess_exited_cleanly: Optional[bool] = None
+        self.nbsr = None
+        # This may override the above defaults.
+        self._parse_command_line_args()
+
+        # Do an initial config-load. If the config is invalid at this point
+        # we can cleanly die (we're more lenient later on reloads).
+        self.load_config(strict=True, print_confirmation=False)
+
+    @property
+    def config(self) -> ServerConfig:
+        """The current config for the app."""
+        return self._config
+
+    @config.setter
+    def config(self, value: ServerConfig) -> None:
+        dataclass_validate(value)
+        self._config = value
+
+    def _prerun(self) -> None:
+        """Common code at the start of any run."""
+
+        # Make sure we don't call run multiple times.
+        if self._running:
+            raise RuntimeError('Already running.')
+        self._running = True
+
+        dbgstr = 'debug' if __debug__ else 'opt'
+        logging.info(f'{Clr.CYN}{Clr.BLD}ROCKY AND VORTEX OFFICIAL is {VERSION_STR}'
+            f' starting up in ({dbgstr} mode).{Clr.RST}')
+
+        # Python will handle SIGINT for us (as KeyboardInterrupt) but we
+        # need to register a SIGTERM handler so we have a chance to clean
+        # up our subprocess when someone tells us to die. (and avoid
+        # zombie processes)
+        signal.signal(signal.SIGTERM, self._handle_term_signal)
+
+        # During a run, we make the assumption that cwd is the dir
+        # containing this script, so make that so. Up until now that may
+        # not be the case (we support being called from any location).
+        os.chdir(os.path.abspath(os.path.dirname(__file__)))
+
+        # Fire off a background thread to wrangle our server binaries.
+        self._subprocess_thread = Thread(target=self._bg_thread_main)
+        self._subprocess_thread.start()
+
+    def _postrun(self) -> None:
+        """Common code at the end of any run."""
+        logging.info(f"{Clr.RED}{Clr.BLD}Server is shutting down.. Please wait!{Clr.RST}")
+
+        assert self._subprocess_thread is not None
+        if self._subprocess_thread.is_alive():
+            logging.info(f"{Clr.RED}{Clr.BLD}Killing all subprocess safely..{Clr.RST}")
+
+        # Mark ourselves as shutting down and wait for the process to wrap up.
+        self._done = True
+        self._subprocess_thread.join()
+
+        # If there's a server error we should care about, exit the
+        # entire wrapper uncleanly.
+        if self._should_report_subprocess_error:
+            raise CleanError('Server subprocess exited uncleanly.')
+
+    def run(self) -> None:
+        """Do the thing."""
+        if self._interactive:
+            self._run_interactive()
+        else:
+            self._run_noninteractive()
+
+    def _run_noninteractive(self) -> None:
+        """Run the app loop to completion noninteractively."""
+        self._prerun()
+        try:
+            while True:
+                time.sleep(1.234)
+        except KeyboardInterrupt:
+            # Gracefully bow out if we kill ourself via keyboard.
+            pass
+        except SystemExit:
+            # We get this from the builtin quit(), our signal handler, etc.
+            # Need to catch this so we can clean up, otherwise we'll be
+            # left in limbo with our process thread still running.
+            pass
+        self._postrun()
+
+    def _run_interactive(self) -> None:
+        """Run the app loop to completion interactively."""
+        import code
+        self._prerun()
+
+        # Print basic usage info for interactive mode.
+        logging.info(f"{Clr.CYN}{Clr.BLD}Interactive mode enabled use the 'mgr' object to interact with the server. Type 'help(mgr)' for more information.{Clr.RST}")
+        context = {'__name__': '__console__', '__doc__': None, 'mgr': self}
+
+        # Enable tab-completion if possible.
+        self._enable_tab_completion(context)
+
+        # Now just sit in an interpreter.
+        # TODO: make it possible to use IPython if the user has it available.
+        try:
+            self._interpreter_start_time = time.time()
+            code.interact(local=context, banner='', exitmsg='')
+        except SystemExit:
+            # We get this from the builtin quit(), our signal handler, etc.
+            # Need to catch this so we can clean up, otherwise we'll be
+            # left in limbo with our process thread still running.
+            pass
+        except BaseException as exc:
+            print(
+                f'Unexpected interpreter exception:'
+                f' {exc} ({type(exc)})',
+                flush=True)
+
+        self._postrun()
+
+    def cmd(self, statement: str) -> None:
+        """Exec a Python command on the current running server subprocess.
+
+        Note that commands are executed asynchronously and no status or
+        return value is accessible from this manager app.
+        """
+        if not isinstance(statement, str):
+            raise TypeError(f'Expected a string arg; got {type(statement)}')
+        with self._subprocess_commands_lock:
+            self._subprocess_commands.append(statement)
+        self._block_for_command_completion()
+
+    def _block_for_command_completion(self) -> None:
+        # Ideally we'd block here until the command was run so our prompt would
+        # print after it's results. We currently don't get any response from
+        # the app so the best we can do is block until our bg thread has sent
+        # it. In the future we can perhaps add a proper 'command port'
+        # interface for proper blocking two way communication.
+        while True:
+            with self._subprocess_commands_lock:
+                if not self._subprocess_commands:
+                    break
+            time.sleep(0.1)
+
+        # One last short delay so if we come out *just* as the command is sent
+        # we'll hopefully still give it enough time to process/print.
+        time.sleep(0.1)
+
+    def screenmessage(self,
+                      message: str,
+                      color: Optional[tuple[float, float, float]] = None,
+                      clients: Optional[list[int]] = None) -> None:
+        """Display a screen-message.
+
+        This will have no name attached and not show up in chat history.
+        They will show up in replays, however (unless clients is passed).
+        """
+        from bacommon.servermanager import ScreenMessageCommand
+        self._enqueue_server_command(
+            ScreenMessageCommand(message=message, color=color,
+                                 clients=clients))
+
+    def chatmessage(self,
+                    message: str,
+                    clients: Optional[list[int]] = None) -> None:
+        """Send a chat message from the server.
+
+        This will have the server's name attached and will be logged
+        in client chat windows, just like other chat messages.
+        """
+        from bacommon.servermanager import ChatMessageCommand
+        self._enqueue_server_command(
+            ChatMessageCommand(message=message, clients=clients))
+
+    def clientlist(self) -> None:
+        """Print a list of connected clients."""
+        from bacommon.servermanager import ClientListCommand
+        self._enqueue_server_command(ClientListCommand())
+        self._block_for_command_completion()
+
+    def kick(self, client_id: int, ban_time: Optional[int] = None) -> None:
+        """Kick the client with the provided id.
+
+        If ban_time is provided, the client will be banned for that
+        length of time in seconds. If it is None, ban duration will
+        be determined automatically. Pass 0 or a negative number for no
+        ban time.
+        """
+        from bacommon.servermanager import KickCommand
+        self._enqueue_server_command(
+            KickCommand(client_id=client_id, ban_time=ban_time))
+
+    def restart(self, immediate: bool = True) -> None:
+        """Restart the server subprocess.
+
+        By default, the current server process will exit immediately.
+        If 'immediate' is passed as False, however, it will instead exit at
+        the next clean transition point (the end of a series, etc).
+        """
+        from bacommon.servermanager import ShutdownCommand, ShutdownReason
+        self._enqueue_server_command(
+            ShutdownCommand(reason=ShutdownReason.RESTARTING,
+                            immediate=immediate))
+
+        # If we're asking for an immediate restart but don't get one within
+        # the grace period, bring down the hammer.
+        if immediate:
+            self._subprocess_force_kill_time = (
+                    time.time() + self.IMMEDIATE_SHUTDOWN_TIME_LIMIT)
+
+    def shutdown(self, immediate: bool = True) -> None:
+        """Shut down the server subprocess and exit the wrapper.
+
+        By default, the current server process will exit immediately.
+        If 'immediate' is passed as False, however, it will instead exit at
+        the next clean transition point (the end of a series, etc).
+        """
+        from bacommon.servermanager import ShutdownCommand, ShutdownReason
+        self._enqueue_server_command(
+            ShutdownCommand(reason=ShutdownReason.NONE, immediate=immediate))
+
+        # An explicit shutdown means we know to bail completely once this
+        # subprocess completes.
+        self._wrapper_shutdown_desired = True
+
+        # If we're asking for an immediate shutdown but don't get one within
+        # the grace period, bring down the hammer.
+        if immediate:
+            self._subprocess_force_kill_time = (
+                    time.time() + self.IMMEDIATE_SHUTDOWN_TIME_LIMIT)
+
+    def _parse_command_line_args(self) -> None:
+        """Parse command line args."""
+        # pylint: disable=too-many-branches
+
+        i = 1
+        argc = len(sys.argv)
+        did_set_interactive = False
+        while i < argc:
+            arg = sys.argv[i]
+            if arg == '--help':
+                self.print_help()
+                sys.exit(0)
+            elif arg == '--config':
+                if i + 1 >= argc:
+                    raise CleanError('Expected a config path as next arg.')
+                path = sys.argv[i + 1]
+                if not os.path.exists(path):
+                    raise CleanError(
+                        f"Supplied path does not exist: '{path}'.")
+                # We need an abs path because we may be in a different
+                # cwd currently than we will be during the run.
+                self._config_path = os.path.abspath(path)
+                self._user_provided_config_path = True
+                i += 2
+            elif arg == '--root':
+                if i + 1 >= argc:
+                    raise CleanError('Expected a path as next arg.')
+                path = sys.argv[i + 1]
+                # Unlike config_path, this one doesn't have to exist now.
+                # We do however need an abs path because we may be in a
+                # different cwd currently than we will be during the run.
+                self._ba_root_path = os.path.abspath(path)
+                i += 2
+            elif arg == '--interactive':
+                if did_set_interactive:
+                    raise CleanError('interactive/noninteractive can only'
+                                     ' be specified once.')
+                self._interactive = True
+                did_set_interactive = True
+                i += 1
+            elif arg == '--noninteractive':
+                if did_set_interactive:
+                    raise CleanError('interactive/noninteractive can only'
+                                     ' be specified once.')
+                self._interactive = False
+                did_set_interactive = True
+                i += 1
+            elif arg == '--no-auto-restart':
+                self._auto_restart = False
+                i += 1
+            elif arg == '--no-config-auto-restart':
+                self._config_auto_restart = False
+                i += 1
+            else:
+                raise CleanError(f"Invalid arg: '{arg}'.")
+
+    @classmethod
+    def _par(cls, txt: str) -> str:
+        """Spit out a pretty paragraph for our help text."""
+        import textwrap
+        ind = ' ' * 2
+        out = textwrap.fill(txt, 80, initial_indent=ind, subsequent_indent=ind)
+        return f'{out}\n'
+
+    @classmethod
+    def print_help(cls) -> None:
+        """Print app help."""
+        filename = os.path.basename(__file__)
+        out = (
+                f'{Clr.BLD}{filename} usage:{Clr.RST}\n' + cls._par(
+            'This script handles configuring, launching, re-launching,'
+            ' and otherwise managing BallisticaCore operating'
+            ' in server mode. It can be run with no arguments, but'
+            ' accepts the following optional ones:') + f'\n'
+                                                       f'{Clr.BLD}--help:{Clr.RST}\n'
+                                                       f'  Show this help.\n'
+                                                       f'\n'
+                                                       f'{Clr.BLD}--config [path]{Clr.RST}\n' + cls._par(
+            'Set the config file read by the server script. The config'
+            ' file contains most options for what kind of game to host.'
+            ' It should be in yaml format. Note that yaml is backwards'
+            ' compatible with json so you can just write json if you'
+            ' want to. If not specified, the script will look for a'
+            ' file named \'config.yaml\' in the same directory as the'
+            ' script.') + '\n'
+                          f'{Clr.BLD}--root [path]{Clr.RST}\n' + cls._par(
+            'Set the ballistica root directory. This is where the server'
+            ' binary will read and write its caches, state files,'
+            ' downloaded assets to, etc. It needs to be a writable'
+            ' directory. If not specified, the script will use the'
+            ' \'dist/ba_root\' directory relative to itself.') + '\n'
+                                                                 f'{Clr.BLD}--interactive{Clr.RST}\n'
+                                                                 f'{Clr.BLD}--noninteractive{Clr.RST}\n' + cls._par(
+            'Specify whether the script should run interactively.'
+            ' In interactive mode, the script creates a Python interpreter'
+            ' and reads commands from stdin, allowing for live interaction'
+            ' with the server. The server script will then exit when '
+            'end-of-file is reached in stdin. Noninteractive mode creates'
+            ' no interpreter and is more suited to being run in automated'
+            ' scenarios. By default, interactive mode will be used if'
+            ' a terminal is detected and noninteractive mode otherwise.') +
+                '\n'
+                f'{Clr.BLD}--no-auto-restart{Clr.RST}\n' +
+                cls._par('Auto-restart is enabled by default, which means the'
+                         ' server manager will restart the server binary whenever'
+                         ' it exits (even when uncleanly). Disabling auto-restart'
+                         ' will cause the server manager to instead exit after a'
+                         ' single run and also to return error codes if the'
+                         ' server binary did so.') + '\n'
+                                                     f'{Clr.BLD}--no-config-auto-restart{Clr.RST}\n' + cls._par(
+            'By default, when auto-restart is enabled, the server binary'
+            ' will be automatically restarted if changes to the server'
+            ' config file are detected. This disables that behavior.'))
+        print(out)
+
+    def load_config(self, strict: bool, print_confirmation: bool) -> None:
+        """Load the config.
+
+        If strict is True, errors will propagate upward.
+        Otherwise, warnings will be printed and repeated attempts will be
+        made to load the config. Eventually the function will give up
+        and leave the existing config as-is.
+        """
+        retry_seconds = 3
+        maxtries = 11
+        for trynum in range(maxtries):
+            try:
+                self._config = self._load_config_from_file(
+                    print_confirmation=print_confirmation)
+                return
+            except Exception as exc:
+                if strict:
+                    raise CleanError(
+                        f'Error loading config file:\n{exc}') from exc
+                print(f'{Clr.RED}Error loading config file:\n{exc}.{Clr.RST}',
+                      flush=True)
+                if trynum == maxtries - 1:
+                    print(
+                        f'{Clr.RED}Max-tries reached; giving up.'
+                        f' Existing config values will be used.{Clr.RST}',
+                        flush=True)
+                    break
+                print(
+                    f'{Clr.CYN}Please correct the error.'
+                    f' Will re-attempt load in {retry_seconds}'
+                    f' seconds. (attempt {trynum + 1} of'
+                    f' {maxtries - 1}).{Clr.RST}',
+                    flush=True)
+
+                for _j in range(retry_seconds):
+                    # If the app is trying to die, drop what we're doing.
+                    if self._done:
+                        return
+                    time.sleep(1)
+
+    def _load_config_from_file(self, print_confirmation: bool) -> ServerConfig:
+
+        out: Optional[ServerConfig] = None
+
+        if not os.path.exists(self._config_path):
+
+            # Special case:
+            # If the user didn't specify a particular config file, allow
+            # gracefully falling back to defaults if the default one is
+            # missing.
+            if not self._user_provided_config_path:
+                if print_confirmation:
+                    print(
+                        f'{Clr.YLW}Default config file not found'
+                        f' (\'{self._config_path}\'); using default'
+                        f' settings.{Clr.RST}',
+                        flush=True)
+                self._config_mtime = None
+                self._last_config_mtime_check_time = time.time()
+                return ServerConfig()
+
+            # Don't be so lenient if the user pointed us at one though.
+            raise RuntimeError(
+                f"Config file not found: '{self._config_path}'.")
+
+        import yaml
+        with open(self._config_path, encoding='utf-8') as infile:
+            user_config_raw = yaml.safe_load(infile.read())
+
+        # An empty config file will yield None, and that's ok.
+        if user_config_raw is not None:
+            out = dataclass_from_dict(ServerConfig, user_config_raw)
+
+        # Update our known mod-time since we know it exists.
+        self._config_mtime = Path(self._config_path).stat().st_mtime
+        self._last_config_mtime_check_time = time.time()
+
+        # Go with defaults if we weren't able to load anything.
+        if out is None:
+            out = ServerConfig()
+
+        if print_confirmation:
+            logging.info(f"{Clr.CYN}{Clr.BLD}Loaded all server configuration files successfully.{Clr.RST}")
+        return out
+
+    def _enable_tab_completion(self, locs: dict) -> None:
+        """Enable tab-completion on platforms where available (linux/mac)."""
+        try:
+            import readline
+            import rlcompleter
+            readline.set_completer(rlcompleter.Completer(locs).complete)
+            readline.parse_and_bind('tab:complete')
+        except ImportError:
+            # This is expected (readline doesn't exist under windows).
+            pass
+
+    def _bg_thread_main(self) -> None:
+        """Top level method run by our bg thread."""
+        while not self._done:
+            self._run_server_cycle()
+
+    def _handle_term_signal(self, sig: int, frame: FrameType) -> None:
+        """Handle signals (will always run in the main thread)."""
+        del sig, frame  # Unused.
+        sys.exit(1 if self._should_report_subprocess_error else 0)
+
+    def _run_server_cycle(self) -> None:
+        """Spin up the server subprocess and run it until exit."""
+        # pylint: disable=consider-using-with
+
+        # Reload our config, and update our overall behavior based on it.
+        # We do non-strict this time to give the user repeated attempts if
+        # if they mess up while modifying the config on the fly.
+        self.load_config(strict=False, print_confirmation=True)
+
+        self._prep_subprocess_environment()
+
+        # Launch the binary and grab its stdin;
+        # we'll use this to feed it commands.
+        self._subprocess_launch_time = time.time()
+
+        # Set an environment var so the server process knows its being
+        # run under us. This causes it to ignore ctrl-c presses and other
+        # slight behavior tweaks. Hmm; should this be an argument instead?
+        os.environ['BA_SERVER_WRAPPER_MANAGED'] = '1'
+        os.environ['BA_DEVICE_NAME'] = self._config.party_name
+        logging.info(f"{Clr.CYN}{Clr.BLD}Launching server sub-process. Hang tight!{Clr.RST}")
+        binary_name = ('BallisticaCoreHeadless.exe'
+                       if os.name == 'nt' else './bombsquad_headless')
+        if platform.processor() == 'aarch64':
+            binary_name = './bombsquad_headless_aarch64'
+        assert self._ba_root_path is not None
+        self._subprocess = None
+
+        # Launch!
+        try:
+            if ERROR_LOGGING:
+                self._subprocess = subprocess.Popen(
+                    [binary_name, '-cfgdir', self._ba_root_path],
+                    stdin=subprocess.PIPE,
+                    stdout=subprocess.PIPE,
+                    stderr=subprocess.PIPE,
+                    cwd='dist')
+
+                self.nbsr = NBSR(self._subprocess.stdout)
+                self.nbsrerr = NBSR(self._subprocess.stderr)
+            else:
+                self._subprocess = subprocess.Popen(
+                    [binary_name, '-cfgdir', self._ba_root_path],
+                    stdin=subprocess.PIPE,
+                    cwd='dist')
+                
+        except Exception as exc:
+            self._subprocess_exited_cleanly = False
+            print(
+                f'{Clr.RED}Error launching server subprocess: {exc}{Clr.RST}',
+                flush=True)
+
+        # Do the thing.
+        try:
+            self._run_subprocess_until_exit()
+
+        except Exception as exc:
+            print(f'{Clr.RED}Error running server subprocess: {exc}{Clr.RST}',
+                  flush=True)
+
+        self._kill_subprocess()
+
+        assert self._subprocess_exited_cleanly is not None
+
+        # EW: it seems that if we die before the main thread has fully started
+        # up the interpreter, its possible that it will not break out of its
+        # loop via the usual SystemExit that gets sent when we die.
+        if self._interactive:
+            while (self._interpreter_start_time is None
+                   or time.time() - self._interpreter_start_time < 0.5):
+                time.sleep(0.1)
+
+        # Avoid super fast death loops.
+        if (not self._subprocess_exited_cleanly and self._auto_restart
+                and not self._done):
+            time.sleep(5.0)
+
+        # If they don't want auto-restart, we'll exit the whole wrapper.
+        # (and with an error code if things ended badly).
+        if not self._auto_restart:
+            self._wrapper_shutdown_desired = True
+            if not self._subprocess_exited_cleanly:
+                self._should_report_subprocess_error = True
+
+        self._reset_subprocess_vars()
+
+        # If we want to die completely after this subprocess has ended,
+        # tell the main thread to die.
+        if self._wrapper_shutdown_desired:
+
+            # Only do this if the main thread is not already waiting for
+            # us to die; otherwise it can lead to deadlock.
+            # (we hang in os.kill while main thread is blocked in Thread.join)
+            if not self._done:
+                self._done = True
+
+                # This should break the main thread out of its blocking
+                # interpreter call.
+                os.kill(os.getpid(), signal.SIGTERM)
+
+    def _prep_subprocess_environment(self) -> None:
+        """Write files that must exist at process launch."""
+
+        assert self._ba_root_path is not None
+        os.makedirs(self._ba_root_path, exist_ok=True)
+        cfgpath = os.path.join(self._ba_root_path, 'config.json')
+        if os.path.exists(cfgpath):
+            with open(cfgpath, encoding='utf-8') as infile:
+                bincfg = json.loads(infile.read())
+        else:
+            bincfg = {}
+
+        # Some of our config values translate directly into the
+        # ballisticacore config file; the rest we pass at runtime.
+        bincfg['Port'] = self._config.port
+        bincfg['Auto Balance Teams'] = self._config.auto_balance_teams
+        bincfg['Show Tutorial'] = self._config.show_tutorial
+
+        if self._config.team_names is not None:
+            bincfg['Custom Team Names'] = self._config.team_names
+        elif 'Custom Team Names' in bincfg:
+            del bincfg['Custom Team Names']
+
+        if self._config.team_colors is not None:
+            bincfg['Custom Team Colors'] = self._config.team_colors
+        elif 'Custom Team Colors' in bincfg:
+            del bincfg['Custom Team Colors']
+
+        bincfg['Idle Exit Minutes'] = self._config.idle_exit_minutes
+        with open(cfgpath, 'w', encoding='utf-8') as outfile:
+            outfile.write(json.dumps(bincfg))
+
+    def _enqueue_server_command(self, command: ServerCommand) -> None:
+        """Enqueue a command to be sent to the server.
+
+        Can be called from any thread.
+        """
+        with self._subprocess_commands_lock:
+            self._subprocess_commands.append(command)
+
+    def _send_server_command(self, command: ServerCommand) -> None:
+        """Send a command to the server.
+
+        Must be called from the server process thread.
+        """
+        import pickle
+        assert current_thread() is self._subprocess_thread
+        assert self._subprocess is not None
+        assert self._subprocess.stdin is not None
+        val = repr(pickle.dumps(command))
+        assert '\n' not in val
+        execcode = (f'import ba._servermode;'
+                    f' ba._servermode._cmd({val})\n').encode()
+        self._subprocess.stdin.write(execcode)
+        self._subprocess.stdin.flush()
+
+    def _run_subprocess_until_exit(self) -> None:
+        if self._subprocess is None:
+            return
+
+        assert current_thread() is self._subprocess_thread
+        assert self._subprocess.stdin is not None
+
+        # Send the initial server config which should kick things off.
+        # (but make sure its values are still valid first)
+        dataclass_validate(self._config)
+        self._send_server_command(StartServerModeCommand(self._config))
+
+        while True:
+
+            # If the app is trying to shut down, nope out immediately.
+            if self._done:
+                break
+            # output=self._subprocess.stdout.readline()
+            # print(output)
+            if ERROR_LOGGING:
+                out = self.nbsr.readline(0.1)
+                out2 = self.nbsrerr.readline(0.1)
+                if out:
+                    sys.stdout.write(out.decode("utf-8"))
+                    _thread.start_new_thread(dump_logs, (out.decode("utf-8"),))
+                if out2:
+                    sys.stdout.write(out2.decode("utf-8"))
+                    _thread.start_new_thread(dump_logs, (out2.decode("utf-8"),))
+            # Pass along any commands to our process.
+            with self._subprocess_commands_lock:
+                for incmd in self._subprocess_commands:
+                    # If we're passing a raw string to exec, no need to wrap it
+                    # in any proper structure.
+                    if isinstance(incmd, str):
+                        self._subprocess.stdin.write((incmd + '\n').encode())
+                        self._subprocess.stdin.flush()
+                    else:
+                        self._send_server_command(incmd)
+                self._subprocess_commands = []
+
+            # Request restarts/shut-downs for various reasons.
+            self._request_shutdowns_or_restarts()
+
+            # If they want to force-kill our subprocess, simply exit this
+            # loop; the cleanup code will kill the process if its still
+            # alive.
+
+            if (self._subprocess_force_kill_time is not None
+                    and time.time() > self._subprocess_force_kill_time):
+                print(
+                    f'{Clr.CYN}Immediate shutdown time limit'
+                    f' ({self.IMMEDIATE_SHUTDOWN_TIME_LIMIT:.1f} seconds)'
+                    f' expired; force-killing subprocess...{Clr.RST}',
+                    flush=True)
+                break
+
+            # Watch for the server process exiting..
+            code: Optional[int] = self._subprocess.poll()
+            if code is not None:
+                clr = Clr.CYN if code == 0 else Clr.RED
+                print(
+                    f'{clr}Server subprocess exited'
+                    f' with code {code}.{Clr.RST}',
+                    flush=True)
+                self._subprocess_exited_cleanly = (code == 0)
+                break
+
+            time.sleep(0.25)
+
+    def _request_shutdowns_or_restarts(self) -> None:
+        # pylint: disable=too-many-branches
+        assert current_thread() is self._subprocess_thread
+        assert self._subprocess_launch_time is not None
+        now = time.time()
+        minutes_since_launch = (now - self._subprocess_launch_time) / 60.0
+
+        # If we're doing auto-restart with config changes, handle that.
+        if (self._auto_restart and self._config_auto_restart
+                and not self._subprocess_sent_config_auto_restart):
+            if (self._last_config_mtime_check_time is None
+                    or (now - self._last_config_mtime_check_time) > 3.123):
+                self._last_config_mtime_check_time = now
+                mtime: Optional[float]
+                if os.path.isfile(self._config_path):
+                    mtime = Path(self._config_path).stat().st_mtime
+                else:
+                    mtime = None
+                if mtime != self._config_mtime:
+                    print(
+                        f'{Clr.CYN}Config-file change detected;'
+                        f' requesting immediate restart.{Clr.RST}',
+                        flush=True)
+                    self.restart(immediate=True)
+                    self._subprocess_sent_config_auto_restart = True
+
+        # Attempt clean exit if our clean-exit-time passes.
+        # (and enforce a 6 hour max if not provided)
+        clean_exit_minutes = 360.0
+        if self._config.clean_exit_minutes is not None:
+            clean_exit_minutes = min(clean_exit_minutes,
+                                     self._config.clean_exit_minutes)
+        if clean_exit_minutes is not None:
+            if (minutes_since_launch > clean_exit_minutes
+                    and not self._subprocess_sent_clean_exit):
+                opname = 'restart' if self._auto_restart else 'shutdown'
+                print(
+                    f'{Clr.CYN}clean_exit_minutes'
+                    f' ({clean_exit_minutes})'
+                    f' elapsed; requesting soft'
+                    f' {opname}.{Clr.RST}',
+                    flush=True)
+                if self._auto_restart:
+                    self.restart(immediate=False)
+                else:
+                    self.shutdown(immediate=False)
+                self._subprocess_sent_clean_exit = True
+
+        # Attempt unclean exit if our unclean-exit-time passes.
+        # (and enforce a 7 hour max if not provided)
+        unclean_exit_minutes = 420.0
+        if self._config.unclean_exit_minutes is not None:
+            unclean_exit_minutes = min(unclean_exit_minutes,
+                                       self._config.unclean_exit_minutes)
+        if unclean_exit_minutes is not None:
+            if (minutes_since_launch > unclean_exit_minutes
+                    and not self._subprocess_sent_unclean_exit):
+                opname = 'restart' if self._auto_restart else 'shutdown'
+                print(
+                    f'{Clr.CYN}unclean_exit_minutes'
+                    f' ({unclean_exit_minutes})'
+                    f' elapsed; requesting immediate'
+                    f' {opname}.{Clr.RST}',
+                    flush=True)
+                if self._auto_restart:
+                    self.restart(immediate=True)
+                else:
+                    self.shutdown(immediate=True)
+                self._subprocess_sent_unclean_exit = True
+
+    def _reset_subprocess_vars(self) -> None:
+        self._subprocess = None
+        self._subprocess_launch_time = None
+        self._subprocess_sent_config_auto_restart = False
+        self._subprocess_sent_clean_exit = False
+        self._subprocess_sent_unclean_exit = False
+        self._subprocess_force_kill_time = None
+        self._subprocess_exited_cleanly = None
+
+    def _kill_subprocess(self) -> None:
+        """End the server subprocess if it still exists."""
+        assert current_thread() is self._subprocess_thread
+        if self._subprocess is None:
+            return
+
+        # First, ask it nicely to die and give it a moment.
+        # If that doesn't work, bring down the hammer.
+        self._subprocess.terminate()
+        try:
+            self._subprocess.wait(timeout=10)
+            self._subprocess_exited_cleanly = (
+                    self._subprocess.returncode == 0)
+        except subprocess.TimeoutExpired:
+            self._subprocess_exited_cleanly = False
+            self._subprocess.kill()
+        logging.info(f"{Clr.RED}{Clr.BLD}Server shut down completed successfully!{Clr.RST}")
+
+
+#PING THE SERVER STATUS SERVER
+#WE NEED TO AGREE ON THE SERVER ID'S
+# import requests
+# from threading import Timer
+
+# def ping_online_server():
+#     try:
+#         r = requests.get(url="https://serverstatus.professorfish.repl.co/statuses")
+#         data = r.json()
+#     except Exception as e:
+#         print(e)
+#         print("FAILED TO PING THE SERVER STATUS SERVER")
+#     finally:
+#         Timer(10, ping_online_server).start()
+def main() -> None:
+    """Run the BallisticaCore server manager."""
+    try:
+        # ping_online_server()
+        ServerManagerApp().run()
+
+	
+    except CleanError as exc:
+        # For clean errors, do a simple print and fail; no tracebacks/etc.
+        # Any others will bubble up and give us the usual mess.
+        exc.pretty_print()
+        sys.exit(1)
+
+
+def dump_logs(msg):
+    if os.path.isfile('logs.log'):
+        size = os.path.getsize('logs.log')
+
+        if size > 2000000:
+            os.remove('logs.log')
+
+    with open("logs.log", "a") as f:
+        f.write(msg)
+
+
+if __name__ == '__main__':
+    main()