mirror of
https://gitlab.com/crafty-controller/crafty-4.git
synced 2024-08-30 18:23:09 +00:00
211 lines
6.7 KiB
Python
211 lines
6.7 KiB
Python
import os
|
|
import sys
|
|
import json
|
|
import time
|
|
import logging
|
|
from datetime import datetime
|
|
|
|
from app.classes.shared.helpers import helper
|
|
from app.classes.shared.console import console
|
|
from app.classes.shared.models import Servers
|
|
from app.classes.minecraft.server_props import ServerProps
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
try:
|
|
import requests
|
|
|
|
except ModuleNotFoundError as e:
|
|
logger.critical("Import Error: Unable to load {} module".format(e, e.name))
|
|
console.critical("Import Error: Unable to load {} module".format(e, e.name))
|
|
sys.exit(1)
|
|
|
|
|
|
class ServerJars:
|
|
|
|
def _get_api_result(self, call_url: str):
|
|
base_url = "https://serverjars.com"
|
|
full_url = "{base}{call_url}".format(base=base_url, call_url=call_url)
|
|
|
|
r = requests.get(full_url, timeout=2)
|
|
|
|
if r.status_code not in [200, 201]:
|
|
return {}
|
|
|
|
try:
|
|
api_data = json.loads(r.content)
|
|
except Exception as e:
|
|
logger.error("Unable to parse serverjar.com api result due to error: {}".format(e))
|
|
return {}
|
|
|
|
api_result = api_data.get('status')
|
|
api_response = api_data.get('response', {})
|
|
|
|
if api_result != "success":
|
|
logger.error("Api returned a failed status: {}".format(api_result))
|
|
return {}
|
|
|
|
return api_response
|
|
|
|
@staticmethod
|
|
def _read_cache():
|
|
cache_file = helper.serverjar_cache
|
|
cache = {}
|
|
try:
|
|
with open(cache_file, "r") as f:
|
|
cache = json.load(f)
|
|
|
|
except Exception as e:
|
|
logger.error("Unable to read serverjars.com cache file: {}".format(e))
|
|
|
|
return cache
|
|
|
|
def get_serverjar_data(self):
|
|
data = self._read_cache()
|
|
return data.get('servers')
|
|
|
|
@staticmethod
|
|
def _check_api_alive():
|
|
logger.info("Checking serverjars.com API status")
|
|
|
|
check_url = "https://serverjars.com/api/fetchTypes"
|
|
r = requests.get(check_url, timeout=2)
|
|
|
|
if r.status_code in [200, 201]:
|
|
logger.info("Serverjars.com API is alive")
|
|
return True
|
|
|
|
logger.error("unable to contact Serverjars.com api")
|
|
return False
|
|
|
|
def refresh_cache(self):
|
|
|
|
cache_file = helper.serverjar_cache
|
|
cache_old = 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"),
|
|
'servers': {}
|
|
}
|
|
|
|
jar_types = self._get_server_type_list()
|
|
|
|
# for each jar type
|
|
for j in jar_types:
|
|
|
|
# for each server
|
|
for s in jar_types.get(j):
|
|
# jar versions for this server
|
|
versions = self._get_jar_details(s)
|
|
|
|
# add these versions (a list) to the dict with a key of the server type
|
|
data['servers'].update({
|
|
s: versions
|
|
})
|
|
|
|
# save our cache
|
|
try:
|
|
with open(cache_file, "w") as f:
|
|
f.write(json.dumps(data, indent=4))
|
|
logger.info("Cache file refreshed")
|
|
|
|
except Exception as e:
|
|
logger.error("Unable to update serverjars.com cache file: {}".format(e))
|
|
|
|
def _get_jar_details(self, jar_type='servers'):
|
|
url = '/api/fetchAll/{type}'.format(type=jar_type)
|
|
response = self._get_api_result(url)
|
|
temp = []
|
|
for v in response:
|
|
temp.append(v.get('version'))
|
|
time.sleep(.5)
|
|
return temp
|
|
|
|
def _get_server_type_list(self):
|
|
url = '/api/fetchTypes/'
|
|
response = self._get_api_result(url)
|
|
return response
|
|
|
|
def download_jar(self, server, version, path):
|
|
base_url = "https://serverjars.com/api/fetchJar/{server}/{version}".format(server=server, version=version)
|
|
r = requests.get(base_url, timeout=2)
|
|
if r.status_code in [200, 201]:
|
|
|
|
try:
|
|
with open(path, 'bw') as output:
|
|
output.write(r.content)
|
|
return True
|
|
except Exception as e:
|
|
logger.error("Unable to save jar to {path} due to error:{error}".format(path=path, error=e))
|
|
pass
|
|
|
|
logger.error("Got {} code from download, escaping".format(r.status_code))
|
|
return False
|
|
|
|
# todo: build server
|
|
def build_server(self, server: str, version: str, name: str, min_mem: int, max_mem: int, port: int):
|
|
server_id = helper.create_uuid()
|
|
server_dir = os.path.join(helper.servers_dir, server_id)
|
|
jar_file = "{server}-{version}.jar".format(server=server, version=version)
|
|
full_jar_path = os.path.join(server_dir, jar_file)
|
|
|
|
# make the dir - perhaps a UUID?
|
|
helper.ensure_dir_exists(server_dir)
|
|
|
|
# download the jar
|
|
self.download_jar(server, version, full_jar_path)
|
|
|
|
# todo: verify the MD5
|
|
|
|
# put data in the db
|
|
Servers.insert({
|
|
Servers.server_name: name,
|
|
Servers.server_uuid: server_id,
|
|
Servers.path: server_dir,
|
|
Servers.executable: jar_file,
|
|
Servers.execution_command: 'java -Xms{}G -Xmx{}G -jar /var/opt/minecraft/server/paperclip.jar nogui'.format(min_mem, max_mem),
|
|
Servers.auto_start: False,
|
|
Servers.auto_start_delay: 10,
|
|
Servers.crash_detection: False,
|
|
Servers.log_path:"{}/logs/latest.log".format(server_dir),
|
|
Servers.stop_command:'stop'
|
|
}).execute()
|
|
|
|
|
|
try:
|
|
# place a file in the dir saying it's owned by crafty
|
|
with open(os.path.join(server_dir, "crafty_managed.txt"), 'w') as f:
|
|
f.write("The server in this directory is managed by Crafty Controller.\n Leave this file alone please")
|
|
f.close()
|
|
|
|
# do a eula.txt
|
|
with open(os.path.join(server_dir, "eula.txt"), 'w') as f:
|
|
f.write("eula=true")
|
|
f.close()
|
|
|
|
# setup server.properties with the port
|
|
with open(os.path.join(server_dir, "server.properties"), "w") as f:
|
|
f.write("server_port={}".format(port))
|
|
f.close()
|
|
|
|
except Exception as e:
|
|
logger.error("Unable to create required server files due to :{}".format(e))
|
|
return False
|
|
|
|
return True
|
|
|
|
server_jar_obj = ServerJars()
|