2460182f4f
XMPP is now async and works - Closes #27 Full async - Closes #22 Individual log files - Closes #21
103 lines
3.3 KiB
Python
103 lines
3.3 KiB
Python
#!/usr/bin/env python3
|
|
|
|
import logging
|
|
import bumper
|
|
import sys, socket
|
|
import time
|
|
import platform
|
|
import os
|
|
#os.environ['PYTHONASYNCIODEBUG'] = '1' # Uncomment to enable ASYNCIODEBUG
|
|
import asyncio
|
|
|
|
|
|
async def main():
|
|
try:
|
|
loop = asyncio.get_event_loop()
|
|
except:
|
|
loop = asyncio.new_event_loop()
|
|
|
|
args = sys.argv
|
|
listen_host = ""
|
|
|
|
if len(args) > 0:
|
|
if "--debug" in args:
|
|
logging.basicConfig(
|
|
level=logging.DEBUG,
|
|
format="[%(asctime)s] :: %(levelname)s :: %(name)s :: %(module)s :: %(funcName)s :: %(lineno)d :: %(message)s",
|
|
)
|
|
loop.set_debug(True) # Set asyncio loop to debug
|
|
#logging.getLogger("asyncio").setLevel(logging.DEBUG) # Show debug asyncio logs (disabled in init, uncomment for debugging asyncio)
|
|
else:
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="[%(asctime)s] :: %(levelname)s :: %(name)s :: %(message)s",
|
|
)
|
|
|
|
if "--listen" in args:
|
|
listen_host = args[args.index("--listen") + 1]
|
|
|
|
if listen_host == "":
|
|
if platform.system() == "Darwin": # If a Mac, use 0.0.0.0 for listening
|
|
listen_host = "0.0.0.0"
|
|
else:
|
|
listen_host = socket.gethostbyname(socket.gethostname())
|
|
|
|
conf_address_443 = (listen_host, 443)
|
|
conf_address_8007 = (listen_host, 8007)
|
|
xmpp_address = (listen_host, 5223)
|
|
mqtt_address = (listen_host, 8883)
|
|
|
|
xmpp_server = bumper.XMPPServer(
|
|
xmpp_address
|
|
)
|
|
mqtt_server = bumper.MQTTServer(mqtt_address)
|
|
mqtt_helperbot = bumper.MQTTHelperBot(mqtt_address)
|
|
conf_server = bumper.ConfServer(
|
|
conf_address_443, usessl=True, helperbot=mqtt_helperbot
|
|
)
|
|
conf_server_2 = bumper.ConfServer(
|
|
conf_address_8007, usessl=False, helperbot=mqtt_helperbot
|
|
)
|
|
|
|
# Start web servers
|
|
conf_server.confserver_app()
|
|
task_conf_server = asyncio.create_task(conf_server.start_server())
|
|
bumper.bumperlog.debug("task_conf_server added")
|
|
await task_conf_server
|
|
|
|
conf_server_2.confserver_app()
|
|
task_conf_server2 = asyncio.create_task(conf_server_2.start_server())
|
|
bumper.bumperlog.debug("task_conf_server2 added")
|
|
await task_conf_server2
|
|
|
|
# Start MQTT Server
|
|
task_mqtt_server = asyncio.create_task(mqtt_server.broker_coro())
|
|
bumper.bumperlog.debug("task_mqtt_server added")
|
|
await task_mqtt_server
|
|
|
|
# Start MQTT Helperbot
|
|
task_mqtt_helperbot = asyncio.create_task(mqtt_helperbot.start_helper_bot())
|
|
bumper.bumperlog.debug("task_mqtt_helperbot added")
|
|
await task_mqtt_helperbot
|
|
|
|
# Start XMPP Server
|
|
task_xmpp_server = asyncio.create_task(xmpp_server.async_server())
|
|
bumper.bumperlog.debug("task_xmpp_server added")
|
|
await task_xmpp_server
|
|
|
|
while True:
|
|
try:
|
|
await asyncio.sleep(30)
|
|
bumper.revoke_expired_tokens()
|
|
#disconnected_clients = bumper.get_disconnected_xmpp_clients()
|
|
#for client in disconnected_clients:
|
|
# xmpp_server.remove_client_byuid(client["userid"])
|
|
|
|
except KeyboardInterrupt:
|
|
bumper.bumperlog.info("Bumper Exiting - Keyboard Interrupt")
|
|
print("Bumper Exiting")
|
|
exit(0)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
asyncio.run(main())
|