"""Manage server-side startup"""
import sys

# api must be loaded to register the api callbacks, even though we don't
# use it directly from this module.
from hil import api  # pylint: disable=unused-import

from hil import model, auth
from hil.class_resolver import build_class_map_for
from hil.network_allocator import get_network_allocator


def register_drivers():
    """Put all of the loaded drivers somewhere where the server can find them.

    This must be run *after* extensions have been loaded.
    """
    build_class_map_for(model.Switch)


def validate_state():
    """Do some sanity checking before kicking things off. In particular:

    * Make sure we have extensions loaded for:
      * a network allocator
      * an auth backend

    (More checks may be added in the future).

    If any of the checks fail, ``validate_state`` aborts the program.
    """
    if get_network_allocator() is None:
        sys.exit("ERROR: No network allocator registered; make sure your "
                 "hil.cfg loads an extension which provides the network "
                 "allocator.")
    if auth.get_auth_backend() is None:
        sys.exit("ERROR: No authentication/authorization backend registered; "
                 "make sure your hil.cfg loads an extension which provides "
                 "the auth backend.")


def stop_orphan_consoles():
    """Stop any orphaned console logging processes.

    These may exist if HIL was shut down uncleanly.
    """
    # Stop all orphan console logging processes on startup
    nodes = model.Node.query.all()
    for node in nodes:
        node.obm.stop_console()
        node.obm.delete_console()


def init():
    """Set up the api server's internal state.

    This is a convenience wrapper that calls the other setup routines in
    this module in the correct order, as well as ``model.init_db``
    """
    register_drivers()
    validate_state()
    model.init_db()

"""Manage server-side startup"""
import sys

# api must be loaded to register the api callbacks, even though we don't
# use it directly from this module.
from hil import api  # pylint: disable=unused-import

from hil import model, auth
from hil.class_resolver import build_class_map_for
from hil.network_allocator import get_network_allocator


def register_drivers():
    """Put all of the loaded drivers somewhere where the server can find them.

    This must be run *after* extensions have been loaded.
    """
    build_class_map_for(model.Switch)


def validate_state():
    """Do some sanity checking before kicking things off. In particular:

    * Make sure we have extensions loaded for:
      * a network allocator
      * an auth backend

    (More checks may be added in the future).

    If any of the checks fail, ``validate_state`` aborts the program.
    """
    if get_network_allocator() is None:
        sys.exit("ERROR: No network allocator registered; make sure your "
                 "hil.cfg loads an extension which provides the network "
                 "allocator.")
    if auth.get_auth_backend() is None:
        sys.exit("ERROR: No authentication/authorization backend registered; "
                 "make sure your hil.cfg loads an extension which provides "
                 "the auth backend.")


def stop_orphan_consoles():
    """Stop any orphaned console logging processes.

    These may exist if HIL was shut down uncleanly.
    """
    # Stop all orphan console logging processes on startup
    nodes = model.Node.query.all()
    for node in nodes:
        node.disable_obm()


def init():
    """Set up the api server's internal state.

    This is a convenience wrapper that calls the other setup routines in
    this module in the correct order, as well as ``model.init_db``
    """
    register_drivers()
    validate_state()
    model.init_db()

Change stop_orphan_console to disable the obms instead