diff --git a/webapp/titanembeds/blueprints/gateway/gateway.py b/webapp/titanembeds/blueprints/gateway/gateway.py index e8f9e26..73826ae 100644 --- a/webapp/titanembeds/blueprints/gateway/gateway.py +++ b/webapp/titanembeds/blueprints/gateway/gateway.py @@ -5,6 +5,7 @@ import functools from flask import request, session import time import json +import os class Gateway(Namespace): def teardown_db_session(self): @@ -12,7 +13,8 @@ class Gateway(Namespace): db.session.remove() def on_connect(self): - emit('hello') + gateway_identifier = os.environ.get("TITAN_GATEWAY_ID", None) + emit('hello', {"gateway_identifier": gateway_identifier}) def on_identify(self, data): guild_id = data["guild_id"] diff --git a/webapp/titanembeds/static/js/embed.js b/webapp/titanembeds/static/js/embed.js index 9d804d5..e17093e 100644 --- a/webapp/titanembeds/static/js/embed.js +++ b/webapp/titanembeds/static/js/embed.js @@ -2122,6 +2122,14 @@ socket.emit('identify', {"guild_id": guild_id, "visitor_mode": visitor_mode}); process_message_users_cache(); }); + + socket.on('hello', function (msg) { + var gateway_identifier = msg.gateway_identifier; + if (!gateway_identifier) { + gateway_identifier = "null"; + } + console.log("%c[TitanEmbeds]%cConnected to gateway via%c" + gateway_identifier, 'color:aqua;background-color:black;border:1px solid black;padding: 3px;', 'color:white;background-color:black;border:1px solid black;padding: 3px;', 'color:white;background-color:black;border:1px solid black;font-family: Courier;padding: 3px;'); + }); socket.on("disconnect", function () {