mirror of
https://github.com/hubHarmony/servii-backend.git
synced 2024-11-18 05:50:31 +00:00
[+] /FetchServers now return the amount of connected players at this time
This commit is contained in:
commit
87f92934ed
@ -32,7 +32,7 @@ def set_subdomain(user: UserRecord, subdomain: str) -> tuple[HTTPStatus, Union[s
|
||||
try:
|
||||
record: SRVRecord = cloudflare_manager.add_cloudflare_dns(subdomain, port)
|
||||
record_id: str = record.id
|
||||
if record_id is None:
|
||||
if record_id == "":
|
||||
return HTTPStatus.BAD_REQUEST, "Failed to add record to cloudflare"
|
||||
firebase_manager.update_firestore(user_id, {'cloudflare_entry': str(record_id)})
|
||||
except Exception as e:
|
||||
@ -48,7 +48,16 @@ def fetch_servers(user: UserRecord) -> tuple[HTTPStatus, Union[str, list]]:
|
||||
server_ref = firebase_manager.firestore_database.collection('users').document(user_id).collection('servers')
|
||||
try:
|
||||
servers = list(server_ref.stream())
|
||||
servers_data = [doc.to_dict() for doc in servers]
|
||||
servers_data: list[dict[str, any]] = []
|
||||
for doc in servers:
|
||||
server_info = doc.to_dict()
|
||||
if server_info.get('running', False):
|
||||
try:
|
||||
connected_players = mc_manager.get_online_players(server_info.get('port', 25565))
|
||||
except Exception as e:
|
||||
connected_players = 0
|
||||
server_info['onlinePlayers'] = connected_players
|
||||
servers_data.append(server_info)
|
||||
if not servers_data:
|
||||
account_create(user)
|
||||
return HTTPStatus.OK, servers_data
|
||||
@ -99,7 +108,10 @@ def server_create(user: UserRecord, name: str, version: str, framework: str = "p
|
||||
file_manager.create_folder(server_path)
|
||||
file_manager.copy_folder_contents(server_template_path, server_path)
|
||||
file_manager.copy_folder_contents("servers/shared", server_path)
|
||||
file_manager.update_server_property(server_path + "/server.properties", "server-port", port)
|
||||
prop_path: str = server_path + "/server.properties"
|
||||
file_manager.update_server_property(prop_path, "server-port", port)
|
||||
file_manager.update_server_property(prop_path, "query.port", port)
|
||||
file_manager.update_server_property(prop_path, "enable-query", "true")
|
||||
return HTTPStatus.CREATED, f"Successfully created server '{name}'."
|
||||
except Exception as e:
|
||||
file_manager.log_error(type(e).__name__, str(e))
|
||||
|
@ -1,6 +1,8 @@
|
||||
import subprocess
|
||||
import shlex
|
||||
import time
|
||||
import mcipc.query
|
||||
import mcipc.query.client
|
||||
from typing import Union
|
||||
|
||||
|
||||
@ -63,6 +65,12 @@ class MinecraftServerManager:
|
||||
return server_id
|
||||
return None
|
||||
|
||||
def get_online_players(self, port)->int:
|
||||
with mcipc.query.Client('127.0.0.1', port) as client:
|
||||
stats: mcipc.query.proto.FullStats = client.stats(full=True)
|
||||
stats: int = stats.num_players
|
||||
return stats
|
||||
|
||||
def set_cooldown(self, user_id):
|
||||
expiry_timestamp = time.time() + 30
|
||||
self.cooldowns[user_id] = expiry_timestamp
|
||||
|
@ -1,4 +1,3 @@
|
||||
import file_manager
|
||||
import firebase_manager
|
||||
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user