init commit

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.12-slim
+
+WORKDIR /code
+
+ENV PYTHONUNBUFFERED 1
+
+RUN pip install --no-cache-dir -U discord pydvpl
+
+COPY ./pydvpl_bot.py /code
+
+CMD ["python", "pydvpl_bot.py"]

README.md

@@ -1,8 +1,8 @@
 ---
 title: PyDVPL Discord Bot
-emoji: 🏃
-colorFrom: indigo
-colorTo: yellow
+emoji: 🌍
+colorFrom: pink
+colorTo: indigo
 sdk: docker
 pinned: false
 ---

pydvpl_bot.py

@@ -0,0 +1,91 @@
+import os
+import discord
+from discord.ext import commands
+from pydvpl.dvpl import compress_dvpl, decompress_dvpl
+
+# Define your Discord bot token here
+TOKEN = os.getenv('DISCORD_TOKEN')
+
+# Set up bot command prefix
+bot_intents = discord.Intents.all()
+
+# Initialize the bot with intents
+bot = commands.Bot(command_prefix=">>", intents=bot_intents)
+
+# 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")
+
+@bot.event
+async def on_ready():
+    print(f'{bot.user} has connected to Discord!')
+    ensure_folders()
+
+async def compress_file(file_path):
+    try:
+        with open(file_path, "rb") as f:
+            file_data = f.read()
+        compressed_data = compress_dvpl(file_data)
+        compressed_file_path = file_path + ".dvpl"  # Example: Add .dvpl extension
+        with open(compressed_file_path, "wb") as f:
+            f.write(compressed_data)
+        return compressed_file_path
+    except Exception as e:
+        print(f"Compression failed: {e}")
+        return None
+
+async def decompress_file(file_path):
+    try:
+        with open(file_path, "rb") as f:
+            file_data = f.read()
+        decompressed_data = decompress_dvpl(file_data)
+        decompressed_file_path = os.path.splitext(file_path)[0]  # Example: Remove .dvpl extension
+        with open(decompressed_file_path, "wb") as f:
+            f.write(decompressed_data)
+        return decompressed_file_path
+    except Exception as e:
+        print(f"Decompression failed: {e}")
+        return None
+
+@bot.command()
+async def compress(ctx):
+    attachments = ctx.message.attachments
+    
+    if not attachments:
+        await ctx.send("No file attached.")
+        return
+
+    for attachment in attachments:
+        file_path = os.path.join("received_to_compress", attachment.filename)
+        await attachment.save(file_path)
+        compressed_file_path = await compress_file(file_path)
+        if compressed_file_path:
+            await ctx.send(file=discord.File(compressed_file_path))
+            os.remove(file_path)  # Delete the original file
+            os.remove(compressed_file_path)  # Delete the compressed file
+        else:
+            await ctx.send("Failed to compress the file.")
+
+@bot.command()
+async def decompress(ctx):
+    attachments = ctx.message.attachments
+    
+    if not attachments:
+        await ctx.send("No file attached.")
+        return
+
+    for attachment in attachments:
+        file_path = os.path.join("received_to_decompress", attachment.filename)
+        await attachment.save(file_path)
+        decompressed_file_path = await decompress_file(file_path)
+        if decompressed_file_path:
+            await ctx.send(file=discord.File(decompressed_file_path))
+            os.remove(file_path)  # Delete the original file
+            os.remove(decompressed_file_path)  # Delete the decompressed file
+        else:
+            await ctx.send("Failed to decompress the file.")
+
+bot.run(TOKEN)

received_to_compress/example.txt

@@ -0,0 +1 @@
+example.txt

received_to_decompress/example.txt

@@ -0,0 +1 @@
+example.txt