python-aternos/python_aternos/atfm.py

import lxml.html
from typing import Union, Optional, List
from typing import TYPE_CHECKING

from .atfile import AternosFile, FileType
if TYPE_CHECKING:
	from .atserver import AternosServer

class FileManager:

	"""Aternos file manager class for viewing files structure
	
	:param atserv: :class:`python_aternos.atserver.AternosServer` instance
	:type atserv: python_aternos.atserver.AternosServer
	"""

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

		self.atserv = atserv

	def listdir(self, path:str='') -> List[AternosFile]:

		"""Requests a list of files
		in the specified directory

		:param path: Directory
		(an empty string means root), defaults to ''
		:type path: str, optional
		:return: List of :class:`python_aternos.atfile.AternosFile`
		:rtype: List[AternosFile]
		"""

		path = path.lstrip('/')
		filesreq = self.atserv.atserver_request(
			f'https://aternos.org/files/{path}', 'GET'
		)
		filestree = lxml.html.fromstring(filesreq.content)
		fileslist = filestree.xpath('//div[contains(concat(" ",normalize-space(@class)," ")," file ")]')

		files = []
		for f in fileslist:

			ftype_raw = f.xpath('@data-type')[0]
			ftype = FileType.file \
				if ftype_raw == 'file' \
				else FileType.directory

			fsize_raw = f.xpath('./div[@class="filesize"]')
			fsize = 0
			if len(fsize_raw) > 0:

				fsize_text = fsize_raw[0].text.strip()
				fsize_num = fsize_text[:fsize_text.rfind(' ')]
				fsize_msr = fsize_text[fsize_text.rfind(' ')+1:]

				try:
					fsize = self.convert_size(float(fsize_num), fsize_msr)
				except ValueError:
					fsize = -1

			fullpath = f.xpath('@data-path')[0]
			filepath = fullpath[:fullpath.rfind('/')]
			filename = fullpath[fullpath.rfind('/'):]
			files.append(
				AternosFile(
					self.atserv,
					filepath, filename,
					ftype, fsize
				)
			)

		return files

	def convert_size(self, num:Union[int,float], measure:str) -> float:

		"""Converts "human" file size to size in bytes

		:param num: Size
		:type num: Union[int,float]
		:param measure: Units (B, kB, MB, GB)
		:type measure: str
		:return: Size in bytes
		:rtype: float
		"""

		measure_match = {
			'B': 1,
			'kB': 1000,
			'MB': 1000000,
			'GB': 1000000000
		}
		try:
			return num * measure_match[measure]
		except KeyError:
			return -1

	def get_file(self, path:str) -> Optional[AternosFile]:

		"""Returns :class:`python_aternos.atfile.AternosFile`
		instance by its path

		:param path: Path to file including its filename
		:type path: str
		:return: _description_
		:rtype: Optional[AternosFile]
		"""

		filepath = path[:path.rfind('/')]
		filename = path[path.rfind('/'):]

		filedir = self.listdir(filepath)
		for file in filedir:
			if file.name == filename:
				return file

		return None

	def dl_file(self, path:str) -> bytes:

		"""Returns the file content in bytes (downloads it)

		:param path: Path to file including its filename
		:type path: str
		:return: File content
		:rtype: bytes
		"""

		file = self.atserv.atserver_request(
			f'https://aternos.org/panel/ajax/files/download.php?' + \
			f'file={path.replace("/","%2F")}',
			'GET'
		)

		return file.content

	def dl_world(self, world:str='world') -> bytes:

		"""Returns the world zip file content
		by its name (downloads it)

		:param world: Name of world, defaults to 'world'
		:type world: str, optional
		:return: Zip file content
		:rtype: bytes
		"""

		world = self.atserv.atserver_request(
			f'https://aternos.org/panel/ajax/worlds/download.php?' + \
			f'world={world.replace("/","%2F")}',
			'GET'
		)

		return world.content