monolith/threshold

#!/usr/bin/env python
from twisted.internet import reactor
from twisted.internet.ssl import DefaultOpenSSLContextFactory
import sys
from signal import signal, SIGINT

# from twisted.python import log
# from sys import stdout
# log.startLogging(stdout)
from sys import stdout, stderr  # Import again because we want to override
from codecs import getwriter  # fix printing odd shit to the terminal


import main

from utils.cleanup import handler


from utils.logging.log import log
from utils.loaders.command_loader import loadCommands
from core.server import ServerFactory
from core.relay import RelayFactory
import modules.counters
import core.logstash

main.initMain()

if "--debug" in sys.argv:  # yes really
    main.config["Debug"] = True
if "--trace" in sys.argv:
    main.config["Trace"] = True

loadCommands()


core.logstash.init_logstash()
signal(SIGINT, handler)  # Handle Ctrl-C and run the cleanup routine
stdout = getwriter("utf8")(stdout)  # this is a generic fix but we all know
stderr = getwriter("utf8")(stderr)  # it's just for the retards on Rizon using
#  unicode quit messages for no reason

if __name__ == "__main__":
    listener = ServerFactory()
    if main.config["Listener"]["UseSSL"] is True:
        reactor.listenSSL(
            main.config["Listener"]["Port"],
            listener,
            DefaultOpenSSLContextFactory(
                main.certPath + main.config["Key"],
                main.certPath + main.config["Certificate"],
            ),
            interface=main.config["Listener"]["Address"],
        )
        log(
            "Threshold running with SSL on %s:%s"
            % (main.config["Listener"]["Address"], main.config["Listener"]["Port"])
        )
    else:
        reactor.listenTCP(
            main.config["Listener"]["Port"],
            listener,
            interface=main.config["Listener"]["Address"],
        )
        log("Threshold running on %s:%s" % (main.config["Listener"]["Address"], main.config["Listener"]["Port"]))
    if main.config["RelayAPI"]["Enabled"]:
        relay = RelayFactory()
        if main.config["RelayAPI"]["UseSSL"] is True:
            reactor.listenSSL(
                main.config["RelayAPI"]["Port"],
                relay,
                DefaultOpenSSLContextFactory(
                    main.certPath + main.config["Key"],
                    main.certPath + main.config["Certificate"],
                ),
                interface=main.config["RelayAPI"]["Address"],
            )
            log(
                "Threshold relay running with SSL on %s:%s"
                % (main.config["RelayAPI"]["Address"], main.config["RelayAPI"]["Port"])
            )
        else:
            reactor.listenTCP(
                main.config["RelayAPI"]["Port"],
                relay,
                interface=main.config["RelayAPI"]["Address"],
            )
            log(
                "Threshold relay running on %s:%s"
                % (main.config["RelayAPI"]["Address"], main.config["RelayAPI"]["Port"])
            )
    for net in main.network.keys():
        main.network[net].start_bots()
    modules.counters.setupCounterLoop()
    reactor.run()