mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
362 lines
13 KiB
Python
362 lines
13 KiB
Python
import os
|
|
import json
|
|
import threading
|
|
import time
|
|
import logging
|
|
from datetime import datetime
|
|
import requests
|
|
|
|
from app.classes.controllers.servers_controller import ServersController
|
|
from app.classes.models.server_permissions import PermissionsServers
|
|
from app.classes.shared.file_helpers import FileHelpers
|
|
from app.classes.shared.websocket_manager import WebSocketManager
|
|
|
|
logger = logging.getLogger(__name__)
|
|
PAPERJARS = ["paper", "folia"]
|
|
|
|
|
|
class ServerJars:
|
|
def __init__(self, helper):
|
|
self.helper = helper
|
|
self.base_url = "https://serverjars.com"
|
|
self.paper_base = "https://api.papermc.io"
|
|
|
|
@staticmethod
|
|
def get_paper_jars():
|
|
return PAPERJARS
|
|
|
|
def get_paper_versions(self, project):
|
|
"""
|
|
Retrieves a list of versions for a specified project from the PaperMC API.
|
|
|
|
Parameters:
|
|
project (str): The project name to query for available versions.
|
|
|
|
Returns:
|
|
list: A list of version strings available for the project. Returns an empty
|
|
list if the API call fails or if no versions are found.
|
|
|
|
This function makes a GET request to the PaperMC API to fetch available project
|
|
versions, The versions are returned in reverse order, with the most recent
|
|
version first.
|
|
"""
|
|
try:
|
|
response = requests.get(
|
|
f"{self.paper_base}/v2/projects/{project}/", timeout=2
|
|
)
|
|
response.raise_for_status()
|
|
api_data = response.json()
|
|
except Exception as e:
|
|
logger.error(f"Error loading project versions for {project}: {e}")
|
|
return []
|
|
|
|
versions = api_data.get("versions", [])
|
|
versions.reverse() # Ensure the most recent version comes first
|
|
return versions
|
|
|
|
def get_paper_build(self, project, version):
|
|
"""
|
|
Fetches the latest build for a specified project and version from PaperMC API.
|
|
|
|
Parameters:
|
|
project (str): Project name, typically a server software like 'paper'.
|
|
version (str): Project version to fetch the build number for.
|
|
|
|
Returns:
|
|
int or None: Latest build number if successful, None if not or on error.
|
|
|
|
This method attempts to query the PaperMC API for the latest build and
|
|
handles exceptions by logging errors and returning None.
|
|
"""
|
|
try:
|
|
response = requests.get(
|
|
f"{self.paper_base}/v2/projects/{project}/versions/{version}/builds/",
|
|
timeout=2,
|
|
)
|
|
response.raise_for_status()
|
|
api_data = response.json()
|
|
except Exception as e:
|
|
logger.error(f"Error fetching build for {project} {version}: {e}")
|
|
return None
|
|
|
|
builds = api_data.get("builds", [])
|
|
return builds[-1] if builds else None
|
|
|
|
def get_fetch_url(self, jar, server, version):
|
|
"""
|
|
Constructs the URL for downloading a server JAR file based on the server type.
|
|
|
|
Supports two main types of server JAR sources:
|
|
- ServerJars API for servers not in PAPERJARS.
|
|
- Paper API for servers available through the Paper project.
|
|
|
|
Parameters:
|
|
jar (str): Name of the JAR file.
|
|
server (str): Server software name (e.g., "paper").
|
|
version (str): Server version.
|
|
|
|
Returns:
|
|
str or None: URL for downloading the JAR file, or None if URL cannot be
|
|
constructed or an error occurs.
|
|
"""
|
|
try:
|
|
# Check if the server type is not specifically handled by Paper.
|
|
if server not in PAPERJARS:
|
|
return f"{self.base_url}/api/fetchJar/{jar}/{server}/{version}"
|
|
|
|
# For Paper servers, attempt to get the build for the specified version.
|
|
paper_build_info = self.get_paper_build(server, version)
|
|
if paper_build_info is None:
|
|
# Log an error or handle the case where paper_build_info is None
|
|
logger.error(
|
|
"Error: Unable to get build information for server:"
|
|
f" {server}, version: {version}"
|
|
)
|
|
return None
|
|
|
|
build = paper_build_info.get("build")
|
|
if not build:
|
|
# Log an error or handle the case where build is None or not found
|
|
logger.error(
|
|
f"Error: Build number not found for server:"
|
|
f" {server}, version: {version}"
|
|
)
|
|
return None
|
|
|
|
# Construct and return the URL for downloading the Paper server JAR.
|
|
return (
|
|
f"{self.paper_base}/v2/projects/{server}/versions/{version}/"
|
|
f"builds/{build}/downloads/{server}-{version}-{build}.jar"
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"An error occurred while constructing fetch URL: {e}")
|
|
return None
|
|
|
|
def _get_api_result(self, call_url: str):
|
|
full_url = f"{self.base_url}{call_url}"
|
|
|
|
try:
|
|
response = requests.get(full_url, timeout=2)
|
|
response.raise_for_status()
|
|
api_data = json.loads(response.content)
|
|
except Exception as e:
|
|
logger.error(f"Unable to load {full_url} api due to error: {e}")
|
|
return {}
|
|
|
|
api_result = api_data.get("status")
|
|
api_response = api_data.get("response", {})
|
|
|
|
if api_result != "success":
|
|
logger.error(f"Api returned a failed status: {api_result}")
|
|
return {}
|
|
|
|
return api_response
|
|
|
|
def _read_cache(self):
|
|
cache_file = self.helper.serverjar_cache
|
|
cache = {}
|
|
try:
|
|
with open(cache_file, "r", encoding="utf-8") as f:
|
|
cache = json.load(f)
|
|
|
|
except Exception as e:
|
|
logger.error(f"Unable to read serverjars.com cache file: {e}")
|
|
|
|
return cache
|
|
|
|
def get_serverjar_data(self):
|
|
data = self._read_cache()
|
|
return data.get("types")
|
|
|
|
def _check_api_alive(self):
|
|
logger.info("Checking serverjars.com API status")
|
|
|
|
check_url = f"{self.base_url}/api/fetchTypes"
|
|
try:
|
|
response = requests.get(check_url, timeout=2)
|
|
|
|
if response.status_code in [200, 201]:
|
|
logger.info("Serverjars.com API is alive")
|
|
return True
|
|
except Exception as e:
|
|
logger.error(f"Unable to connect to serverjar.com api due to error: {e}")
|
|
return {}
|
|
|
|
logger.error("unable to contact serverjars.com api")
|
|
return False
|
|
|
|
def manual_refresh_cache(self):
|
|
cache_file = self.helper.serverjar_cache
|
|
|
|
# debug override
|
|
# cache_old = True
|
|
|
|
# if the API is down... we bomb out
|
|
if not self._check_api_alive():
|
|
return False
|
|
|
|
logger.info("Manual Refresh requested.")
|
|
now = datetime.now()
|
|
data = {
|
|
"last_refreshed": now.strftime("%m/%d/%Y, %H:%M:%S"),
|
|
"types": {},
|
|
}
|
|
|
|
jar_types = self._get_server_type_list()
|
|
data["types"].update(jar_types)
|
|
for s in data["types"]:
|
|
data["types"].update({s: dict.fromkeys(data["types"].get(s), {})})
|
|
for j in data["types"].get(s):
|
|
versions = self._get_jar_details(j, s)
|
|
data["types"][s].update({j: versions})
|
|
for item in PAPERJARS:
|
|
data["types"]["servers"][item] = self.get_paper_versions(item)
|
|
# save our cache
|
|
try:
|
|
with open(cache_file, "w", encoding="utf-8") as f:
|
|
f.write(json.dumps(data, indent=4))
|
|
logger.info("Cache file refreshed")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Unable to update serverjars.com cache file: {e}")
|
|
|
|
def refresh_cache(self):
|
|
cache_file = self.helper.serverjar_cache
|
|
cache_old = self.helper.is_file_older_than_x_days(cache_file)
|
|
|
|
# debug override
|
|
# cache_old = True
|
|
|
|
# if the API is down... we bomb out
|
|
if not self._check_api_alive():
|
|
return False
|
|
|
|
logger.info("Checking Cache file age")
|
|
# if file is older than 1 day
|
|
|
|
if cache_old:
|
|
logger.info("Cache file is over 1 day old, refreshing")
|
|
now = datetime.now()
|
|
data = {
|
|
"last_refreshed": now.strftime("%m/%d/%Y, %H:%M:%S"),
|
|
"types": {},
|
|
}
|
|
|
|
jar_types = self._get_server_type_list()
|
|
data["types"].update(jar_types)
|
|
for s in data["types"]:
|
|
data["types"].update({s: dict.fromkeys(data["types"].get(s), {})})
|
|
for j in data["types"].get(s):
|
|
versions = self._get_jar_details(j, s)
|
|
data["types"][s].update({j: versions})
|
|
for item in PAPERJARS:
|
|
data["types"]["servers"][item] = self.get_paper_versions(item)
|
|
# save our cache
|
|
try:
|
|
with open(cache_file, "w", encoding="utf-8") as f:
|
|
f.write(json.dumps(data, indent=4))
|
|
logger.info("Cache file refreshed")
|
|
|
|
except Exception as e:
|
|
logger.error(f"Unable to update serverjars.com cache file: {e}")
|
|
|
|
def _get_jar_details(self, server_type, jar_type="servers"):
|
|
url = f"/api/fetchAll/{jar_type}/{server_type}"
|
|
response = self._get_api_result(url)
|
|
temp = []
|
|
for v in response:
|
|
temp.append(v.get("version"))
|
|
time.sleep(0.5)
|
|
return temp
|
|
|
|
def _get_server_type_list(self):
|
|
url = "/api/fetchTypes/"
|
|
response = self._get_api_result(url)
|
|
if "bedrock" in response.keys():
|
|
# remove pocketmine from options
|
|
del response["bedrock"]
|
|
return response
|
|
|
|
def download_jar(self, jar, server, version, path, server_id):
|
|
update_thread = threading.Thread(
|
|
name=f"server_download-{server_id}-{server}-{version}",
|
|
target=self.a_download_jar,
|
|
daemon=True,
|
|
args=(jar, server, version, path, server_id),
|
|
)
|
|
update_thread.start()
|
|
|
|
def a_download_jar(self, jar, server, version, path, server_id):
|
|
"""
|
|
Downloads a server JAR file and performs post-download actions including
|
|
notifying users and setting import status.
|
|
|
|
This method waits for the server registration to complete, retrieves the
|
|
download URL for the specified server JAR file.
|
|
|
|
Upon successful download, it either runs the installer for
|
|
Forge servers or simply finishes the import process for other types. It
|
|
notifies server users about the completion of the download.
|
|
|
|
Parameters:
|
|
- jar (str): The name of the JAR file to download.
|
|
- server (str): The type of server software (e.g., 'forge', 'paper').
|
|
- version (str): The version of the server software.
|
|
- path (str): The local filesystem path where the JAR file will be saved.
|
|
- server_id (str): The unique identifier for the server being updated or
|
|
imported, used for notifying users and setting the import status.
|
|
|
|
Returns:
|
|
- bool: True if the JAR file was successfully downloaded and saved;
|
|
False otherwise.
|
|
|
|
The method ensures that the server is properly registered before proceeding
|
|
with the download and handles exceptions by logging errors and reverting
|
|
the import status if necessary.
|
|
"""
|
|
# delaying download for server register to finish
|
|
time.sleep(3)
|
|
|
|
fetch_url = self.get_fetch_url(jar, server, version)
|
|
if not fetch_url:
|
|
return False
|
|
|
|
server_users = PermissionsServers.get_server_user_list(server_id)
|
|
|
|
# Make sure the server is registered before updating its stats
|
|
while True:
|
|
try:
|
|
ServersController.set_import(server_id)
|
|
for user in server_users:
|
|
WebSocketManager().broadcast_user(user, "send_start_reload", {})
|
|
break
|
|
except Exception as ex:
|
|
logger.debug(f"Server not registered yet. Delaying download - {ex}")
|
|
|
|
# Initiate Download
|
|
jar_dir = os.path.dirname(path)
|
|
jar_name = os.path.basename(path)
|
|
logger.info(fetch_url)
|
|
success = FileHelpers.ssl_get_file(fetch_url, jar_dir, jar_name)
|
|
|
|
# Post-download actions
|
|
if success:
|
|
if server == "forge":
|
|
# If this is the newer Forge version, run the installer
|
|
ServersController.finish_import(server_id, True)
|
|
else:
|
|
ServersController.finish_import(server_id)
|
|
|
|
# Notify users
|
|
for user in server_users:
|
|
WebSocketManager().broadcast_user(
|
|
user, "notification", "Executable download finished"
|
|
)
|
|
time.sleep(3) # Delay for user notification
|
|
WebSocketManager().broadcast_user(user, "send_start_reload", {})
|
|
else:
|
|
logger.error(f"Unable to save jar to {path} due to download failure.")
|
|
ServersController.finish_import(server_id)
|
|
|
|
return success
|