use remember_web cookie instead of pterodactyl_session

This commit is contained in:
Joseph Montanaro
2020-10-29 23:10:58 -07:00
parent 38863a92cc
commit c895dc7627
2 changed files with 8 additions and 5 deletions

View File

@ -78,9 +78,10 @@ class Socket:
class Client:
def __init__(self, token, session_cookie):
def __init__(self, token, cookie_name, cookie_value):
self.token = token
self.session_cookie = session_cookie
self.cookie_name = cookie_name
self.cookie_value = cookie_value
self.baseurl = 'https://control.heavynode.com/api'
self.stats = []
# global state is icky, but it sure is convenient
@ -148,7 +149,7 @@ class Client:
await self.socket.connect_socket()
async def fetch_daemon_secret(self):
cookie = {'pterodactyl_session': self.session_cookie}
cookie = {self.cookie_name: self.cookie_value}
async with aiohttp.ClientSession(cookies=cookie) as session:
r = await session.get('https://control.heavynode.com/server/' + self.server['identifier'])
m = re.search('"daemonSecret"\s?:\s?"([^"]*)"', await r.text())