main.py

from twisted.internet import reactor

from app.common.database.repositories import channels, wrapper
from app.common.cache import status, usercount

from app.server import TcpBanchoFactory, HttpBanchoFactory

from app.common.constants import ANCHOR_ASCII_ART
from app.common.logging import Console, File
from app.objects.channel import Channel
from app.objects.player import Player
from app.tasks import (
    activities,
    events,
    pings
)

import logging
import config
import signal
import app
import os

logging.basicConfig(
    handlers=[Console, File],
    level=(
        logging.DEBUG
        if config.DEBUG
        else logging.INFO
    )
)

def setup():
    app.session.logger.info(f'{ANCHOR_ASCII_ART}')
    app.session.logger.info(f'Running anchor-{config.VERSION}')
    os.makedirs(config.DATA_PATH, exist_ok=True)

    app.session.logger.info('Loading channels...')

    for channel in channels.fetch_all():
        app.session.logger.info(f'  - {channel.name}')
        app.session.channels.append(
            Channel(
                channel.name,
                channel.topic,
                'BanchoBot',
                channel.read_permissions,
                channel.write_permissions,
                public=True
            )
        )

    app.session.logger.info('Loading bot...')

    app.session.players.add(
        bot_player := Player.bot_player()
    )
    app.session.bot_player = bot_player
    app.session.logger.info(f'  - {bot_player.name}')

    app.session.logger.info('Loading tasks...')
    app.session.tasks.submit(pings.ping_task)
    app.session.tasks.submit(events.event_listener)
    app.session.tasks.submit(activities.match_activity)

    # Reset usercount
    usercount.set(0)

    # Reset player statuses
    for key in status.get_all():
        player_id = key.split(':')[-1]
        status.delete(player_id)

def before_shutdown(*args):
    for player in app.session.players:
        # Enqueue server restart packet to all players
        # They should reconnect after 15 seconds
        player.enqueue_server_restart(15 * 1000)

    reactor.callLater(0.5, reactor.stop)

signal.signal(signal.SIGINT, before_shutdown)

def shutdown():
    # Reset usercount
    usercount.set(0)

    for player in app.session.players:
        status.delete(player.id)

    app.session.events.submit('shutdown')
    app.session.tasks.shutdown(cancel_futures=True, wait=False)

    def force_exit(*args):
        app.session.logger.warning("Force exiting...")
        os._exit(0)

    signal.signal(signal.SIGINT, force_exit)

def on_startup_fail(e: Exception):
    app.session.logger.fatal(f'Failed to start server: "{e}"')
    reactor.stop()

@wrapper.exception_wrapper(on_startup_fail)
def setup_servers():
    http_factory = HttpBanchoFactory()
    tcp_factory = TcpBanchoFactory()

    reactor.suggestThreadPoolSize(config.BANCHO_WORKERS)
    reactor.listenTCP(config.HTTP_PORT, http_factory)

    for port in config.TCP_PORTS:
        reactor.listenTCP(port, tcp_factory)

def main():
    reactor.addSystemEventTrigger('before', 'startup', setup)
    reactor.addSystemEventTrigger('before', 'startup', setup_servers)
    reactor.addSystemEventTrigger('after', 'shutdown', shutdown)
    reactor.run()

if __name__ == "__main__":
    main()