feat: added stop command

pterodactyl/ptero.py

@@ -20,6 +20,17 @@ class Pterodactyl(commands.Cog):
             startup_arguments=None
         )
 
+    async def get_headers(self, guild: discord.Guild):
+        """Returns the headers used to access the Pterodactyl API."""
+        if await self.config.guild(guild).api_key() is None:
+            raise LookupError("API Key not set.")
+        headers = {
+            "Authorization": f"Bearer {await self.config.guild(guild).api_key()}",
+            "Content-Type": "application/json",
+            "Accept": "application/json"
+        }
+        return headers
+
     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:
@@ -39,7 +50,89 @@ class Pterodactyl(commands.Cog):
             async with session.put(url, headers=headers, json=data) as response:
                 return response
 
+    power = app_commands.Group(name='power', description="Controls the server's power state.")
+
+    @power.command(name='start', description="Starts the server.")
+    @app_commands.guild_only()
+    async def start(self, interaction: discord.Interaction):
+        """Starts the server."""
+        await interaction.defer(ephemeral=True, thinking=True)
+        interaction_message = await interaction.original_response()
+        headers = await self.get_headers(interaction.guild)
+        response = requests.get(await self.get_url(interaction.guild, "resources"), headers=headers)
+        requests_json = response.json()
+        current_status = requests_json['attributes']['current_state']
+        if current_status == "offline":
+            passed_info = {
+                "headers": headers,
+                "interaction": interaction,
+                "guild": interaction.guild,
+                "signal": "start",
+                "target_signal": "running",
+                "message": "Server starting...",
+                "completed_message": "Server started!"
+            }
+            await interaction_message.edit(content="Are you sure you'd like to start the server?", view=self.PowerButtons(timeout=180, passed_info=passed_info))
+        else:
+            message = await interaction_message.edit(content="The server is already running!")
+            await message.delete(delay=3)
+
+    @power.command(name='restart', description="Restarts the server.")
+    @app_commands.guild_only()
+    async def restart(self, interaction: discord.Interaction):
+        """Restarts the server."""
+        await interaction.defer(ephemeral=True, thinking=True)
+        interaction_message = await interaction.original_response()
+        headers = await self.get_headers(interaction.guild)
+        response = requests.get(await self.get_url(interaction.guild, "resources"), headers=headers)
+        requests_json = response.json()
+        current_status = requests_json['attributes']['current_state']
+        if current_status == "running":
+            passed_info = {
+                "headers": headers,
+                "interaction": interaction,
+                "guild": interaction.guild,
+                "signal": "restart",
+                "target_signal": "running",
+                "message": "Server restarting...",
+                "completed_message": "Server restarted!"
+            }
+            await interaction_message.edit(content="Are you sure you'd like to restart the server?", view=self.PowerButtons(timeout=180, passed_info=passed_info))
+        elif current_status == "offline":
+            message = await interaction_message.edit(content="The server is offline!")
+            await message.delete(delay=3)
+        elif current_status == "starting":
+            message = await interaction_message.edit(content="The server is already starting!")
+            await message.delete(delay=3)
+
+    @power.command(name='stop', description="Stops the server.")
+    @app_commands.guild_only()
+    async def stop(self, interaction: discord.Interaction):
+        """Stops the server."""
+        await interaction.defer(ephemeral=True, thinking=True)
+        interaction_message = await interaction.original_response()
+        headers = await self.get_headers(interaction.guild)
+        response = requests.get(await self.get_url(interaction.guild, "resources"), headers=headers)
+        requests_json = response.json()
+        current_status = requests_json['attributes']['current_state']
+        if current_status == "running" or current_status == "starting":
+            passed_info = {
+                "headers": headers,
+                "interaction": interaction,
+                "guild": interaction.guild,
+                "signal": "stop",
+                "target_signal": "offline",
+                "message": "Server stopping...",
+                "completed_message": "Server stopped!"
+            }
+            await interaction_message.edit(content="Are you sure you'd like to stop the server?", view=self.PowerButtons(timeout=180, passed_info=passed_info))
+        elif current_status == "offline":
+            message = await interaction_message.edit(content="The server is already offline!")
+            await message.delete(delay=3)
+
+
     @app_commands.command(description="Updates the server.")
+    @app_commands.guild_only()
     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)
@@ -51,17 +144,13 @@ class Pterodactyl(commands.Cog):
             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)
+            await interaction_message.edit(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 = {
+            headers = await self.get_headers(interaction.guild)
-            "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)
@@ -123,11 +212,12 @@ class Pterodactyl(commands.Cog):
 
         @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, self.passed_info['guild'], "power"), headers=self.passed_info['headers'], json={"signal": "stop"})
+            headers = self.passed_info['headers']
+            requests.post(await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=headers, json={"signal": "stop"})
             await self.passed_info['interaction'].edit_original_response(content="Server stopping...", view=None)
             while True:
                 async with aiohttp.ClientSession() as session:
-                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=self.passed_info['headers']) as response:
+                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=headers) as response:
                         response_dict = await response.json()
                         if response_dict['attributes']['current_state'] == "offline":
                             await self.passed_info['interaction'].edit_original_response(content="\nServer stopped!")
@@ -136,13 +226,13 @@ class Pterodactyl(commands.Cog):
                             await asyncio.sleep(2)
                             continue
             for data in self.passed_info['updater_startup_vars']:
-                await Pterodactyl.put(self, url=await Pterodactyl.get_url(self, self.passed_info['guild'], "startup/variable"), headers=self.passed_info['headers'], data=data)
+                await Pterodactyl.put(self, url=await Pterodactyl.get_url(self, self.passed_info['guild'], "startup/variable"), headers=headers, data=data)
-            requests.post(url=await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=self.passed_info['headers'], json={"signal": "start"})
+            requests.post(url=await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=headers, json={"signal": "start"})
             await self.passed_info['interaction'].edit_original_response(content="Updater started...")
             await asyncio.sleep(2)
             while True:
                 async with aiohttp.ClientSession() as session:
-                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=self.passed_info['headers']) as response:
+                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=headers) as response:
                         response_dict = await response.json()
                         if response_dict['attributes']['current_state'] == "offline":
                             await self.passed_info['interaction'].edit_original_response(content="Updater finished!")
@@ -151,13 +241,13 @@ class Pterodactyl(commands.Cog):
                             await asyncio.sleep(1)
                             continue
             for data in self.passed_info['old_startup_vars']:
-                await Pterodactyl.put(self, await Pterodactyl.get_url(self, self.passed_info['guild'], "startup/variable"), self.passed_info['headers'], data)
+                await Pterodactyl.put(self, await Pterodactyl.get_url(self, self.passed_info['guild'], "startup/variable"), headers, data)
             await asyncio.sleep(2)
-            requests.post(url=await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=self.passed_info['headers'], json={"signal": "start"})
+            requests.post(url=await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=headers, json={"signal": "start"})
             await self.passed_info['interaction'].edit_original_response(content="Server starting...")
             while True:
                 async with aiohttp.ClientSession() as session:
-                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=self.passed_info['headers']) as response:
+                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=headers) as response:
                         response_dict = await response.json()
                         if response_dict['attributes']['current_state'] == "running":
                             await self.passed_info['interaction'].edit_original_response(content="Server started!\nUpdate process completed!")
@@ -171,9 +261,40 @@ class Pterodactyl(commands.Cog):
             message = await self.passed_info['interaction'].edit_original_response(content=f"Command cancelled.", view=None)
             await message.delete(delay=3)
 
-    @app_commands.command()
+    class PowerButtons(ui.View):
-    async def test(self, interaction: discord.Interaction, endpoint: str = None):
+        def __init__(self, timeout, passed_info):
-        """This does stuff!"""
+            super().__init__()
+            self.passed_info = passed_info
+            self.config = Config.get_conf(None, cog_name='Pterodactyl', identifier=457581387213637448123567)
+
+        @ui.button(label="Yes", style=discord.ButtonStyle.success)
+        async def yes_button(self, button:ui.Button, interaction:discord.Interaction):
+            headers = self.passed_info['headers']
+            requests.post(await Pterodactyl.get_url(self, self.passed_info['guild'], "power"), headers=headers, json={"signal": self.passed_info['signal']})
+            message = await self.passed_info['interaction'].edit_original_response(content=self.passed_info['message'], view=None)
+            while True:
+                async with aiohttp.ClientSession() as session:
+                    async with session.get(await Pterodactyl.get_url(self, self.passed_info['guild'], "resources"), headers=headers) as response:
+                        response_dict = await response.json()
+                        if response_dict['attributes']['current_state'] == self.passed_info['target_signal']:
+                            await message.edit(content=self.passed_info['completed_message'])
+                            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)
+
+
+    get_group = app_commands.Group(name='get', description="Retrieves information from the Pterodactyl API.")
+
+    @get_group.command(name='url', description="Retrieves the URL for the specified endpoint.")
+    @app_commands.guild_only()
+    async def retrieve_url(self, interaction: discord.Interaction, endpoint: str = None):
+        """Retrieves the URL for the specified endpoint."""
         try:
             if endpoint:
                 url = await self.get_url(interaction.guild, endpoint)
@@ -187,12 +308,14 @@ class Pterodactyl(commands.Cog):
     configure = app_commands.Group(name="config", description="Configures the Pterodactyl cog.")
 
     @configure.command()
+    @app_commands.guild_only()
     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):
+    @app_commands.guild_only()
+    async def update(self, interaction: discord.Interaction):
         """Sets the startup arguments for the update command."""
         await interaction.response.send_modal(self.StartupConfigModal(self.config))