#!/usr/bin/env python
from asyncio import gather, get_event_loop, run
from typing import TypedDict
from aiohttp import ClientSession, ClientWebSocketResponse

async def transmit(ws: ClientWebSocketResponse):
	loop = get_event_loop()
	while not ws.closed:
		await ws.send_json({
			'args': [
				await loop.run_in_executor(None, input),
			],
			'event': 'send command',
		})

class Response(TypedDict):
	args: list[str]
	event: str
	extra: str

async def receive(ws: ClientWebSocketResponse):
	async for response in ws:
		data: Response = response.json()
		if data['event'] == 'console output':
			print(' '.join(data['args']))

class Handshake(TypedDict):
	socket: str
	token: str

async def main():
	async with ClientSession(cookies = {
		'pterodactyl_session': '<按需替换>',
	}, headers = {
		'Origin': 'https://<按需替换>'
	}) as session:
		data: Handshake
		async with session.get('https://<按需替换>/api/client/servers/<按需替换>/websocket') as request:
			data = (await request.json())['data']
		async with session.ws_connect(data['socket']) as ws:
			await ws.send_json({
				'args': [
					data['token'],
				],
				'event': 'auth',
			})
			await gather(receive(ws), transmit(ws))

if __name__ == '__main__':
	run(main())