cswimr/SeaCogs
cswimr/SeaCogs
1
1
Fork 1
Code Issues 11 Pull requests 4 Projects 1 Packages 1 Wiki Activity Actions

Add Pterodactyl cog #19

Merged
cswimr merged 139 commits from pterodactyl into main 2024-03-02 00:07:42 -05:00
Conversation 0 Commits 139 Files changed 22 +1 -2
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files
Showing only changes of commit 5e3ab08d6d - Show all commits

3
pterodactyl/pterodactyl.py
View file

@ -37,7 +37,7 @@ class Pterodactyl(commands.Cog):
try: try:
client = PterodactylClient(base_url, api_key).client client = PterodactylClient(base_url, api_key).client
websocket_credentials = client.servers.get_websocket(server_id) websocket_credentials = client.servers.get_websocket(server_id)
self.logger.debug(f"Websocket connection details retrieved:\nSocket: {websocket_credentials['data']['socket']}\nToken: {websocket_credentials['data']['token']}") self.logger.debug("Websocket connection details retrieved:\nSocket: %s\nToken: %s", websocket_credentials['data']['socket'], websocket_credentials['data']['token'])
except exceptions.ClientConfigError as e: except exceptions.ClientConfigError as e:
self.logger.error(f'Failed to initialize Pterodactyl client: {e}') self.logger.error(f'Failed to initialize Pterodactyl client: {e}')
return return
@ -48,7 +48,6 @@ class Pterodactyl(commands.Cog):
async with websockets.connect(websocket_credentials['data']['socket'], extra_headers=extra_headers) as websocket: async with websockets.connect(websocket_credentials['data']['socket'], extra_headers=extra_headers) as websocket:
self.logger.debug("WebSocket connection established") self.logger.debug("WebSocket connection established")
# Send authentication token
auth_message = json.dumps({"event": "auth", "args": [websocket_credentials['data']['token']]}) auth_message = json.dumps({"event": "auth", "args": [websocket_credentials['data']['token']]})
await websocket.send(auth_message) await websocket.send(auth_message)
self.logger.debug("Authentication message sent") self.logger.debug("Authentication message sent")