python-aternos/python_aternos/atconf.py

import enum
import re
import lxml.html
from typing import Any, Dict, List, Optional
from typing import TYPE_CHECKING

from . import atconnect

if TYPE_CHECKING:
	from atserver import AternosServer

class ServerOpts(enum.Enum):
	players = 'max-players'
	gm = 'gamemode'
	difficulty = 'difficulty'
	whl = 'white-list'
	online = 'online-mode'
	pvp = 'pvp'
	cmdblock = 'enable-command-block'
	flight = 'allow-flight'
	animals = 'spawn-animals'
	monsters = 'spawn-monsters'
	villagers = 'spawn-npcs'
	nether = 'allow-nether'
	forcegm = 'force-gamemode'
	spawnlock = 'spawn-protection'
	cmds = 'allow-cheats'
	pack = 'resource-pack'

DAT_PREFIX = 'Data:'

class WorldOpts(enum.Enum):
	seed = 'randomseed'
	hardcore = 'hardcore'
	difficulty = 'difficulty'

DAT_GR_PREFIX = 'Data:GameRules:'

class WorldRules(enum.Enum):
	advs = 'announceadvancements'
	cmdout = 'commandblockoutput'
	elytra = 'disableelytramovementcheck'
	daynight = 'dodaylightcycle'
	entdrop = 'doentitydrops'
	fire = 'dofiretick'
	limitcraft = 'dolimitedcrafting'
	mobloot = 'domobloot'
	mobs = 'domobspawning'
	blockdrop = 'dotiledrops'
	weather = 'doweathercycle'
	keepinv = 'keepinventory'
	deathmsg = 'showdeathmessages'
	admincmdlog = 'logadmincommands'
	cmdlen = 'maxcommandchainlength'
	entcram = 'maxentitycramming'
	mobgrief = 'mobgriefing'
	regen = 'naturalregeneration'
	rndtick = 'randomtickspeed'
	spawnradius = 'spawnradius'
	reducedf3 = 'reduceddebuginfo'
	spectchunkgen = 'spectatorsgeneratechunks'
	cmdfb = 'sendcommandfeedback'

DAT_TYPE_WORLD = 0
DAT_TYPE_GR = 1

class Gamemode(enum.IntEnum):
	survival = 0
	creative = 1
	adventure = 2
	spectator = 3

class Difficulty(enum.IntEnum):
	peaceful = 0
	easy = 1
	normal = 2
	hard = 3

JDK = 'openjdk:{}'
OJ9 = 'adoptopenjdk:{}-jre-openj9-bionic'

FLAG_PROP_TYPE = 1

class AternosConfig:

	def __init__(self, atserv:'AternosServer') -> None:

		self.atserv = atserv

	@property
	def timezone(self) -> str:

		optreq = self.atserv.atserver_request(
			'https://aternos.org/options',
			atconnect.REQGET
		)
		opttree = lxml.html.fromstring(optreq)

		tzopt = opttree.xpath('//div[@class="options-other-input timezone-switch"]')[0]
		tztext = tzopt.xpath('.//div[@class="option current"]')[0].text
		return tztext.strip()

	@timezone.setter
	def timezone(self, value:str) -> None:

		matches_tz = re.search(r'(?:^[A-Z]\w+\/[A-Z]\w+$)|^UTC$', value)
		if matches_tz == None:
			raise AttributeError('Timezone must match zoneinfo format: Area/Location')

		self.atserv.atserver_request(
			'https://aternos.org/panel/ajax/timezone.php',
			atconnect.REQPOST, data={'timezone': value},
			sendtoken=True
		)

	@property
	def java_version(self) -> str:

		optreq = self.atserv.atserver_request(
			'https://aternos.org/options',
			atconnect.REQGET
		)
		opttree = lxml.html.fromstring(optreq)

		imgopt = opttree.xpath('//div[@class="options-other-input image-switch"]')[0]
		imgver = imgopt.xpath('.//div[@class="option current"]/@data-value')[0]
		return imgver

	@java_version.setter
	def java_version(self, value:str) -> None:
		matches_jdkver = re.search(r'^(?:adopt)?openjdk:(\d+)(?:-jre-openj9-bionic)?$', value)
		if matches_jdkver == None:
			raise AttributeError('Incorrect Java image version format!')

		self.atserv.atserver_request(
			'https://aternos.org/panel/ajax/image.php',
			atconnect.REQPOST, data={'image': value},
			sendtoken=True
		)

	#
	# server.properties
	#
	def set_server_prop(self, option:str, value:Any) -> None:
		self.__set_prop(
			'/server.properties',
			option, value
		)

	def get_server_props(self, flags:int=FLAG_PROP_TYPE) -> Dict[str,Any]:
		return self.__get_all_props('https://aternos.org/options', flags)

	def set_server_props(self, props:Dict[str,Any]) -> None:
		for key in props:
			set_server_prop(key, props[key])

	#
	# level.dat
	#
	def set_world_prop(
		self, option:str, value:Any,
		proptype:int, world:str='world') -> None:

		prefix = DAT_PREFIX
		if proptype == DAT_TYPE_GR:
			prefix = DAT_GR_PREFIX

		self.__set_prop(
			f'/{world}/level.dat',
			f'{prefix}{option}',
			value
		)

	def get_world_props(
		self, world:str='world',
		flags:int=FLAG_PROP_TYPE) -> Dict[str,Any]:

		self.__get_all_props(
			f'https://aternos.org/files/{world}/level.dat',
			flags, [DAT_PREFIX, DAT_GR_PREFIX]
		)

	def set_world_props(self, props:Dict[str,Any]) -> None:
		for key in props:
			set_world_prop(key, prop[key])

	#
	# helpers
	#
	def __set_prop(self, file:str, option:str, value:Any) -> None:

		self.atserv.atserver_request(
			'https://aternos.org/panel/ajax/config.php',
			atconnect.REQPOST, data={
				'file': file,
				'option': option,
				'value': value
			}, sendtoken=True
		)

	def __get_all_props(
		self, url:str, flags:int=FLAG_PROP_TYPE,
		prefixes:Optional[List[str]]=None) -> Dict[str,Any]:

		optreq = self.atserv.atserver_request(
			'https://aternos.org/options',
			atconnect.REQGET
		)
		opttree = lxml.html.fromstring(optreq.content)

		configs = opttree.xpath('//div[@class="config-options"]')
		for i in range(len(configs)):

			conf = configs[i]
			opts = conf.xpath('/div[contains(@class,"config-option ")]')
			result = {}

			for opt in opts:

				key = opt.xpath('.//span[@class="config-option-output-key"]')[0].text
				value = opt.xpath('.//span[@class="config-option-output-value"]')[0].text
				if prefixes != None:
					key = f'{prefixes[i]}{key}'

				opttype = opt.xpath('/@class').split(' ')[1]
				if flags == FLAG_PROP_TYPE:

					if opttype == 'config-option-number'\
					or opttype == 'config-option-select':
						value = int(value)

					elif opttype == 'config-option-toggle':
						value = bool(value)
				result[key] = value
		return result