new issues and requests commands

pydvpl_bot.py

@@ -1,5 +1,6 @@
 import os
 import time
+import json 
 import psutil
 import discord
 from dotenv import load_dotenv
@@ -30,6 +31,17 @@ UPDATE_INTERVAL = 60
 
 activity_start_time = 0
 
+# Define folder paths
+FEEDBACK_FOLDER = "feedback"
+
+ISSUES_FOLDER = os.path.join(FEEDBACK_FOLDER, "issues")
+
+REQUESTS_FOLDER = os.path.join(FEEDBACK_FOLDER, "requests")
+
+COMPRESS_FOLDER= os.path.join("received_to_compress")
+
+DECOMPRESS_FOLDER= os.path.join("received_to_decompress")
+
 
 # DEFINES ------------------------------------------------------ END
 
@@ -59,10 +71,53 @@ class PyDVPLHelpCommand(DefaultHelpCommand):
 
 # Ensure the existence of folders
 def ensure_folders():
-    if not os.path.exists("received_to_compress"):
-        os.makedirs("received_to_compress")
-    if not os.path.exists("received_to_decompress"):
-        os.makedirs("received_to_decompress")
+    for folder in [FEEDBACK_FOLDER, ISSUES_FOLDER, REQUESTS_FOLDER, COMPRESS_FOLDER, DECOMPRESS_FOLDER]:
+        if not os.path.exists(folder):
+            os.makedirs(folder)
+
+
+# FUNCTION ------------------------------------------------------ SPLIT
+
+
+async def list_feedback(ctx, folder_path, title):
+    files = os.listdir(folder_path)
+    feedback_embed = discord.Embed(title=title, color=discord.Color.blurple())
+    for file in files:
+        file_path = os.path.join(folder_path, file)
+        with open(file_path, "r") as f:
+            feedback_data = json.load(f)
+            for feedback in feedback_data:
+                author_name = feedback['author_name']
+                author_id = feedback['author_id']
+                timestamp = feedback['timestamp']
+                message = feedback['message']
+                feedback_embed.add_field(
+                    name=f"Author: {author_name} (ID: {author_id})",
+                    value=f"Timestamp: {timestamp}\nMessage: {message}",
+                    inline=False
+                )
+    if feedback_embed.fields:
+        await ctx.send(embed=feedback_embed)
+    else:
+        await ctx.send("No feedback found.")
+    await ctx.message.delete()
+
+
+# FUNCTION ------------------------------------------------------ SPLIT
+
+
+async def clean_directory(ctx, directory):
+    try:
+        file_count = 0
+        for filename in os.listdir(directory):
+            file_path = os.path.join(directory, filename)
+            if os.path.isfile(file_path):
+                os.remove(file_path)
+                file_count += 1
+        await ctx.send(f"All files cleaned successfully in {directory}. Total files cleaned: {file_count}.")
+    except Exception as e:
+        await ctx.send(f"An error occurred while cleaning files: {e}")
+    await ctx.message.delete()
 
 
 # HELPER ------------------------------------------------------ END
@@ -230,7 +285,7 @@ async def compress(ctx):
         return
 
     for attachment in attachments:
-        file_path = os.path.join("received_to_compress", attachment.filename)
+        file_path = os.path.join(COMPRESS_FOLDER, attachment.filename)
 
         # Check if the file is already in .dvpl format, if so, skip it
         if file_path.endswith(".dvpl"):
@@ -265,7 +320,7 @@ async def decompress(ctx):
         return
 
     for attachment in attachments:
-        file_path = os.path.join("received_to_decompress", attachment.filename)
+        file_path = os.path.join(DECOMPRESS_FOLDER, attachment.filename)
 
         # Check if the file is not in .dvpl format, if so, skip it
         if not file_path.endswith(".dvpl"):
@@ -324,46 +379,63 @@ async def invite(ctx):
     await ctx.message.delete()
 
 
+# FUNCTION ------------------------------------------------------ SPLIT
+
+
+# Command: Feedback
+@bot.command(help="Send feedback or report issues", usage="<requests/issues> <message>")
+async def feedback(ctx, category: str, *, message: str):
+    if category.lower() not in ["requests", "issues"]:
+        await ctx.send("Invalid category. Please choose 'requests' or 'issues'.")
+        return
+
+    feedback_data = {
+        "author_id": ctx.author.id,
+        "author_name": ctx.author.name,
+        "timestamp": str(ctx.message.created_at),
+        "message": message
+    }
+
+    folder_path = REQUESTS_FOLDER if category.lower() == "requests" else ISSUES_FOLDER
+    file_path = os.path.join(folder_path, f"{category.lower()}.json")
+
+    try:
+        with open(file_path, "r") as f:
+            data = json.load(f)
+    except FileNotFoundError:
+        data = []
+
+    data.append(feedback_data)
+
+    with open(file_path, "w") as f:
+        json.dump(data, f, indent=4)
+
+    await ctx.send(f"Thank you for your {category.lower()}. It has been recorded.")
+
+
 # UTILITY USER_CMD ------------------------------------------------------ END
 
 
 # UTILITY ADMIN_CMD ------------------------------------------------------ START
 
 
-@bot.command(help="Cleans all files in received_to_compress and received_to_decompress directories.")
-async def clean(ctx):
-    # Check if the user invoking the command is authorized
+@bot.command(help="Cleans files in specific directories")
+async def clean(ctx, clean_mode):
     if str(ctx.author.id) != AUTHORIZED_USER_ID:
         await ctx.send("You are not authorized to use this command.")
         await ctx.message.delete()
         return
 
-    try:
-        # Initialize counts for files cleaned
-        compress_count = 0
-        decompress_count = 0
-
-        # Clean files in received_to_compress directory
-        for filename in os.listdir("received_to_compress"):
-            file_path = os.path.join("received_to_compress", filename)
-            if os.path.isfile(file_path):
-                os.remove(file_path)
-                compress_count += 1
-
-        # Clean files in received_to_decompress directory
-        for filename in os.listdir("received_to_decompress"):
-            file_path = os.path.join("received_to_decompress", filename)
-            if os.path.isfile(file_path):
-                os.remove(file_path)
-                decompress_count += 1
-
-        # Send message with the counts
-        message = f"All files cleaned successfully. Compressed files cleaned: {compress_count}, Decompressed files cleaned: {decompress_count}."
-        await ctx.send(message)
-        await ctx.message.delete()
-    except Exception as e:
-        await ctx.send(f"An error occurred while cleaning files: {e}")
-        await ctx.message.delete()
+    if clean_mode.lower() == 'compress':
+        await clean_directory(ctx, COMPRESS_FOLDER)
+    elif clean_mode.lower() == 'decompress':
+        await clean_directory(ctx, DECOMPRESS_FOLDER)
+    elif clean_mode.lower() == 'issues':
+        await clean_directory(ctx, ISSUES_FOLDER)
+    elif clean_mode.lower() == 'requests':
+        await clean_directory(ctx, REQUESTS_FOLDER)
+    else:
+        await ctx.send("Invalid clean mode. Please choose 'received_to_compress', 'received_to_decompress', 'feedback_issues', or 'feedback_requests'.")
 
 
 # FUNCTION ------------------------------------------------------ SPLIT
@@ -626,7 +698,7 @@ async def stats(ctx):
     await ctx.message.delete()
 
 
-@bot.command(help="List all servers the bot is on with their IDs")
+@bot.command(help="List issues, requests, or servers")
 async def list(ctx, list_type, *args):
     if str(ctx.author.id) != AUTHORIZED_USER_ID:
         await ctx.send("You are not authorized to use this command.")
@@ -634,12 +706,16 @@ async def list(ctx, list_type, *args):
         return
     
     if list_type.lower() == 'servers':
-        if len(args) < 1:
-            servers_info = "\n".join([f"{server.name} - ID: {server.id}" for server in bot.guilds])
-            await ctx.send(f"List of servers the bot is on:\n```{servers_info}```")
-            await ctx.message.delete()
+        servers_info = "\n".join([f"{server.name} - ID: {server.id}" for server in bot.guilds])
+        await ctx.send(f"List of servers the bot is on:\n```{servers_info}```")
+        await ctx.message.delete()
+    elif list_type.lower() == 'issues':
+        await list_feedback(ctx, ISSUES_FOLDER, "List of issues:")
+    elif list_type.lower() == 'requests':
+        await list_feedback(ctx, REQUESTS_FOLDER, "List of requests:")
+    else:
+        await ctx.send("Invalid list type. Please choose 'servers', 'issues', or 'requests'.")
 
-    
 
 # UTILITY ADMIN_CMD ------------------------------------------------------ END