import hashlib
import lxml.html

from atserver import AternosServer
from atconnect import AternosConnect
from aterrors import AternosCredentialsError

class Client:

	def __init__(self, username, md5=None, password=None):

		if (password == None) and (md5 == None):
			raise AttributeError('Password was not specified')

		if (password != None):
			self.__init__(
				username,
				md5=hashlib.md5(password.encode('utf-8'))\
				.hexdigest().lower()
			)
			return

		self.atconn = AternosConnect()

		self.token = self.atconn.get_token()
		self.sec = self.atconn.generate_sec()

		self.credentials = {
			'user': username,
			'password': md5
		}

		loginreq = self.atconn.request_cloudflare(
			f'https://aternos.org/panel/ajax/account/login.php?' + \
			f'SEC={self.sec}&TOKEN={self.token}',
			self.atconn.REQPOST, data=self.credentials
		)

		if loginreq.cookies.get('ATERNOS_SESSION', None) == None:
			raise AternosCredentialsError(
				'Check your username and password'
			)

	def get_servers(self):

		serverspage = self.atconn.request_cloudflare(
			'https://aternos.org/servers/',
			self.atconn.REQGET
		)
		serverstree = lxml.html.fromstring(serverspage.content)
		serverslist = serverstree.xpath('//div[@class="servers"]/div')

		servers = []
		for server in serverslist:
			servid = server.xpath('./div[@class="server-body"]/@data-id')[0]
			servers.append(AternosServer(servid, self.atconn))

		return servers