272 lines
15 KiB
Python
272 lines
15 KiB
Python
import asyncio
|
|
import aiohttp
|
|
import discord
|
|
import requests
|
|
from discord import ui
|
|
from redbot.core import commands, app_commands, Config
|
|
|
|
class Pterodactyl(commands.Cog):
|
|
"""Pterodactyl allows you to manage your Pterodactyl Panel from Discord."""
|
|
|
|
def __init__(self, bot):
|
|
self.bot = bot
|
|
self.config = Config.get_conf(self, identifier=457581387213637448123567)
|
|
self.config.register_guild(
|
|
base_url=None,
|
|
api_key=None,
|
|
server_id=None,
|
|
startup_jar=None,
|
|
startup_arguments=None
|
|
)
|
|
|
|
async def get_url(self, guild, endpoint = None):
|
|
"""Returns the base url for the server's API, or the url for a specific API endpoint if one is provided."""
|
|
if await self.config.guild(guild).server_id() is None:
|
|
raise LookupError("Server ID not set.")
|
|
elif await self.config.guild(guild).base_url() is None:
|
|
raise LookupError("Base URL not set.")
|
|
base_url = await self.config.guild(guild).base_url()
|
|
server_id = await self.config.guild(guild).server_id()
|
|
url = f"https://{base_url}/api/client/servers/{server_id}"
|
|
if endpoint:
|
|
url += '/' + endpoint
|
|
return url
|
|
|
|
async def put(self, url: str, headers: dict, data: dict):
|
|
"""Sends an asyncio PUT request to the specified URL with the specified headers and data."""
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.put(url, headers=headers, json=data) as response:
|
|
return response
|
|
|
|
@app_commands.command(description="Updates the server.")
|
|
async def update(self, interaction: discord.Interaction):
|
|
"""Updates the server using the arguments provided in the server's configuration."""
|
|
await interaction.response.defer(ephemeral=True, thinking=True)
|
|
interaction_message = await interaction.original_response()
|
|
if await self.config.guild(interaction.guild).api_key() is None:
|
|
await interaction_message.edit(f"Something went wrong.\nError: `API Key not set.`", ephemeral=True)
|
|
raise LookupError("API Key not set.")
|
|
elif await self.config.guild(interaction.guild).startup_jar() is None:
|
|
await interaction_message.edit(f"Something went wrong.\nError: `Startup jar not set.`", ephemeral=True)
|
|
raise LookupError("Startup jar not set.")
|
|
elif await self.config.guild(interaction.guild).startup_arguments() is None:
|
|
await interaction.response.send_message(f"Something went wrong.\nError: `Startup arguments not set.`", ephemeral=True)
|
|
raise LookupError("Startup arguments not set.")
|
|
else:
|
|
api_key = await self.config.guild(interaction.guild).api_key()
|
|
startup_jar = await self.config.guild(interaction.guild).startup_jar()
|
|
startup_commands = await self.config.guild(interaction.guild).startup_arguments()
|
|
headers = {
|
|
"Authorization": f"Bearer {api_key}",
|
|
"Content-Type": "application/json",
|
|
"Accept": "application/json"
|
|
}
|
|
response = requests.get(await self.get_url(interaction.guild, "resources"), headers=headers)
|
|
response_dict = response.json()
|
|
list_var = requests.get(await self.get_url(interaction.guild, "startup"), headers=headers)
|
|
list_var_response_dict = list_var.json()
|
|
updater_startup_vars = [
|
|
{
|
|
"key": "FLAGS",
|
|
"value": startup_commands
|
|
},
|
|
{
|
|
"key": "SERVER_JARFILE",
|
|
"value": startup_jar
|
|
}
|
|
]
|
|
old_startup_vars = [
|
|
{
|
|
"key": "FLAGS",
|
|
"value": list_var_response_dict['data'][4]['attributes']['server_value']
|
|
},
|
|
{
|
|
"key": "SERVER_JARFILE",
|
|
"value": list_var_response_dict['data'][0]['attributes']['server_value']
|
|
}
|
|
]
|
|
if response_dict['attributes']['current_state'] == "offline":
|
|
for data in updater_startup_vars:
|
|
await self.put(await self.get_url(interaction.guild, "startup/variable"), headers, data)
|
|
requests.post(await self.get_url(interaction.guild, "power"), headers=headers, json={"signal": "start"})
|
|
await interaction_message.edit(content="Packwiz installer started...")
|
|
await asyncio.sleep(1)
|
|
while True:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(await self.get_url(interaction.guild, "resources"), headers=headers) as response:
|
|
response_dict = await response.json()
|
|
if response_dict['attributes']['current_state'] == "offline":
|
|
await interaction_message.edit(content="Packwiz installer finished.")
|
|
break
|
|
else:
|
|
await asyncio.sleep(1)
|
|
continue
|
|
for data in old_startup_vars:
|
|
await self.put(await self.get_url(interaction.guild, "startup/variable"), headers, data)
|
|
await interaction_message.edit(content="Packwiz installer finished.\nUpdate process completed!")
|
|
elif response_dict['attributes']['current_state'] == "running" or response_dict['attributes']['current_state'] == "starting":
|
|
passed_info = {
|
|
"headers": headers,
|
|
"updater_startup_vars": updater_startup_vars,
|
|
"old_startup_vars": old_startup_vars,
|
|
"interaction": interaction
|
|
}
|
|
await interaction_message.edit(content="The server is already running! Are you sure you'd like to stop the server for updates?", view=self.UpdateButtons(timeout=180, passed_info=passed_info))
|
|
|
|
class UpdateButtons(ui.View):
|
|
def __init__(self, timeout, passed_info):
|
|
super().__init__()
|
|
self.passed_info = passed_info
|
|
|
|
@ui.button(label="Yes", style=discord.ButtonStyle.success)
|
|
async def yes_button(self, button:ui.Button, interaction:discord.Interaction):
|
|
requests.post(await Pterodactyl.get_url(self, interaction.guild, "power"), headers=self.passed_info['headers'], json={"signal": "stop"})
|
|
await self.passed_info['interaction'].edit_original_response("Server stopping...", view=None)
|
|
while True:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(await Pterodactyl.get_url(self, interaction.guild, "resources"), headers=self.passed_info['headers']) as response:
|
|
response_dict = await response.json()
|
|
if response_dict['attributes']['current_state'] == "offline":
|
|
await self.passed_info['interaction'].edit_original_response("\nServer stopped!")
|
|
break
|
|
else:
|
|
await asyncio.sleep(2)
|
|
continue
|
|
for data in self.passed_info['updater_startup_vars']:
|
|
await Pterodactyl.put(await Pterodactyl.get_url(self, interaction.guild, "startup/variable"), self.passed_info['headers'], data)
|
|
requests.post(await Pterodactyl.get_url(self, interaction.guild, "power"), headers=self.passed_info['headers'], json={"signal": "start"})
|
|
await self.passed_info['interaction'].edit_original_response("Packwiz installer started...")
|
|
await asyncio.sleep(1)
|
|
while True:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(await Pterodactyl.get_url(self, interaction.guild, "resources"), headers=self.passed_info['headers']) as response:
|
|
response_dict = await response.json()
|
|
if response_dict['attributes']['current_state'] == "offline":
|
|
await self.passed_info['interaction'].edit_original_response("Packwiz installer finished!")
|
|
break
|
|
else:
|
|
await asyncio.sleep(1)
|
|
continue
|
|
for data in self.passed_info['old_startup_vars']:
|
|
await Pterodactyl.put(Pterodactyl.get_url(self, interaction.guild, "startup/variable"), self.passed_info['headers'], data)
|
|
asyncio.sleep(1)
|
|
requests.post(await Pterodactyl.get_url(self, interaction.guild, "power"), self.passed_info['headers'], json={"signal": "start"})
|
|
await self.passed_info['interaction'].edit_original_response("Server starting...")
|
|
while True:
|
|
async with aiohttp.ClientSession() as session:
|
|
async with session.get(await Pterodactyl.get_url(self, interaction.guild, "resources"), headers=self.passed_info['headers']) as response:
|
|
response_dict = await response.json()
|
|
if response_dict['attributes']['current_state'] == "running":
|
|
await self.passed_info['interaction'].edit_original_response("Server started!\nUpdate process completed!")
|
|
break
|
|
else:
|
|
await asyncio.sleep(1)
|
|
continue
|
|
|
|
@ui.button(label="No", style=discord.ButtonStyle.danger)
|
|
async def no_button(self, button:ui.Button, interaction:discord.Interaction):
|
|
message = await self.passed_info['interaction'].edit_original_response(content=f"Command cancelled.", view=None)
|
|
await message.delete(delay=3)
|
|
|
|
@app_commands.command()
|
|
async def test(self, interaction: discord.Interaction, endpoint: str = None):
|
|
"""This does stuff!"""
|
|
try:
|
|
if endpoint:
|
|
url = await self.get_url(interaction.guild, endpoint)
|
|
else:
|
|
url = await self.get_url(interaction.guild)
|
|
except LookupError as e:
|
|
await interaction.response.send_message(f"Something went wrong.\nError: `{e}`", ephemeral=True)
|
|
return
|
|
await interaction.response.send_message(url, ephemeral=True)
|
|
|
|
configure = app_commands.Group(name="config", description="Configures the Pterodactyl cog.")
|
|
|
|
@configure.command()
|
|
async def api(self, interaction: discord.Interaction):
|
|
"""Sets the information used to access the Pterdoactyl API."""
|
|
await interaction.response.send_modal(self.APIConfigModal(self.config))
|
|
|
|
@configure.command()
|
|
async def startup(self, interaction: discord.Interaction):
|
|
"""Sets the startup arguments for the update command."""
|
|
await interaction.response.send_modal(self.StartupConfigModal(self.config))
|
|
|
|
class APIConfigModal(discord.ui.Modal, title="Pterodactyl Manager Configuration"):
|
|
def __init__(self, config):
|
|
super().__init__()
|
|
self.config = config
|
|
base_url = discord.ui.TextInput(
|
|
label="Base URL",
|
|
placeholder="Input your Pterodactyl Panel's Base URL here, without HTTPS or HTTP.",
|
|
style=discord.TextStyle.paragraph,
|
|
required=False,
|
|
max_length=300
|
|
)
|
|
api_key = discord.ui.TextInput(
|
|
label="API Key",
|
|
placeholder="Input your Pterodactyl Client API Key here.",
|
|
style=discord.TextStyle.short,
|
|
required=False,
|
|
max_length=300
|
|
)
|
|
server_id = discord.ui.TextInput(
|
|
label="Server ID",
|
|
placeholder="Input your Pterodactyl server's Server ID here.",
|
|
style=discord.TextStyle.short,
|
|
required=False,
|
|
max_length=300
|
|
)
|
|
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
message = ""
|
|
if self.base_url.value != "":
|
|
await self.config.guild(interaction.guild).base_url.set(self.base_url.value)
|
|
message += f"- Base URL set to\n - `{self.base_url.value}`\n"
|
|
if self.api_key.value != "":
|
|
await self.config.guild(interaction.guild).api_key.set(self.api_key.value)
|
|
trimmed_api_key = self.api_key.value[:16]
|
|
message += f"- API Key set to\n - `{trimmed_api_key}` - Trimmed for security\n"
|
|
if self.server_id.value != "":
|
|
await self.config.guild(interaction.guild).server_id.set(self.server_id.value)
|
|
message += f"- Server ID set to\n - `{self.server_id.value}`\n"
|
|
if message == "":
|
|
trimmed_api_key = str(await self.config.guild(interaction.guild).api_key())[:16]
|
|
send = f"No changes were made.\nCurrent configuration:\n- Base URL:\n - `{await self.config.guild(interaction.guild).base_url()}`\n- API Key:\n - `{trimmed_api_key}` - Trimmed for security\n- Server ID:\n - `{await self.config.guild(interaction.guild).server_id()}`"
|
|
else:
|
|
send = f"Configuration changed:\n{message}"
|
|
await interaction.response.send_message(send, ephemeral=True)
|
|
|
|
class StartupConfigModal(discord.ui.Modal, title="Pterodactyl Manager Configuration"):
|
|
def __init__(self, config):
|
|
super().__init__()
|
|
self.config = config
|
|
startup_jar = discord.ui.TextInput(
|
|
label="Startup .jar",
|
|
placeholder="Input the name of your updater .jar here.",
|
|
style=discord.TextStyle.short,
|
|
required=False,
|
|
max_length=300
|
|
)
|
|
startup_arguments = discord.ui.TextInput(
|
|
label="Startup Arguments",
|
|
placeholder="Input your startup arguments here.\nExample:\n-g -s server https://repository.link/here",
|
|
style=discord.TextStyle.paragraph,
|
|
required=False,
|
|
max_length=1000
|
|
)
|
|
|
|
async def on_submit(self, interaction: discord.Interaction):
|
|
message = ""
|
|
if self.startup_jar.value != "":
|
|
await self.config.guild(interaction.guild).startup_jar.set(self.startup_jar.value)
|
|
message += f"- Startup jar set to\n - `{self.startup_jar.value}`\n"
|
|
if self.startup_arguments.value != "":
|
|
await self.config.guild(interaction.guild).startup_arguments.set(self.startup_arguments.value)
|
|
message += f"- Startup arguments set to:\n```{self.startup_arguments.value}```\n"
|
|
if message == "":
|
|
send = f"No changes were made.\nCurrent configuration:\n- Startup jar: `{await self.config.guild(interaction.guild).startup_jar()}`\n- Startup arguments:\n```{await self.config.guild(interaction.guild).startup_arguments()}```"
|
|
else:
|
|
send = f"Configuration changed:\n{message}"
|
|
await interaction.response.send_message(send, ephemeral=True)
|