1 files changed, 88 insertions, 0 deletions

diff --git a/cogs/compression.py b/cogs/compression.py
new file mode 100644
index 0000000..dfd96ae
--- /dev/null
+++ b/cogs/compression.py
@@ -0,0 +1,88 @@
+import discord
+from discord.ext import commands
+import asyncio
+import mimetypes
+import os 
+import subprocess
+import tempfile
+
+def filetype(filename):
+    mime_type, _ = mimetypes.guess_type(filename)
+    if mime_type:
+        if mime_type.startswith('video/'):
+            return "Video"
+        elif mime_type.startswith('audio/'):
+            return "Audio"
+        elif mime_type.startswith('image/'):
+            return "Image"
+    return "Unknown"
+
+class Compression(commands.Cog):
+    def __init__(self, bot):
+        self.bot = bot
+    
+    @commands.slash_command(
+        name="kompresja",
+        description="skompresuj wideło albo dźwienk :)",
+    )
+    @discord.option(
+        "plik",
+        description="plik do skompresowania",
+        required=True,
+        type=discord.Attachment
+    )
+    async def compression(self, ctx, plik):
+        await ctx.defer()
+        # Save files to temp directory inside the bot's root directory
+        bot_dir = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
+        temp_dir = os.path.join(bot_dir, "temp")
+        if not os.path.exists(temp_dir):
+            os.makedirs(temp_dir)
+        file_path = os.path.join(temp_dir, plik.filename)
+        await plik.save(file_path)
+            
+        input_path = file_path
+        output_path = os.path.join(temp_dir, f"better_{plik.filename}")
+        file_type = filetype(file_path)
+        if file_type == "Video":
+            # Execute ffmpeg command to compress video
+            ffmpeg_command = f"ffmpeg -i {input_path} -c:v libx264 -crf 51 -b:v 100 -vf 'fps=10' -preset ultrafast -c:a libopus -b:a 500 {output_path}"
+        elif file_type == "Audio":
+            ffmpeg_command = f"ffmpeg -i {input_path} -ar 8000 -ac 1 -b:a 16k -acodec libmp3lame -q:a 9 {output_path}"
+        elif file_type == "Image":
+            # Use ImageMagick's magick to heavily compress image (JPEG/PNG/GIF)
+            # For JPEG: quality 5, strip metadata, resize to 30% of original size
+            # For PNG: magick to JPEG, then compress
+            # For GIF: optimize and reduce colors
+            ext = os.path.splitext(input_path)[1].lower()
+            if ext in [".jpg", ".jpeg"]:
+                img_command = f"magick {input_path} -strip -resize 30% -quality 5 {output_path}"
+            elif ext == ".gif":
+                # Optimize GIF: reduce colors, optimize layers, and resize
+                img_command = f"magick {input_path} -coalesce -resize 30% -colors 32 -strip -layers Optimize {output_path}"
+            else:
+                # magick non-JPEG images to JPEG for maximum compression
+                output_path = os.path.splitext(output_path)[0] + ".jpg"
+                img_command = f"magick {input_path} -strip -resize 30% -quality 5 {output_path}"
+            ffmpeg_command = img_command
+        else:
+            await ctx.respond("Plik jest nieobsługiwany", ephemeral=True)
+            return
+        
+        try:
+            subprocess.run(ffmpeg_command, check=True, shell=True)
+            # Upload compressed file
+            await ctx.respond(file=discord.File(output_path))
+            # Delete compressed file
+            os.remove(input_path)
+            os.remove(output_path)
+        except subprocess.CalledProcessError as e:
+            await ctx.respond(f"Compression failed: {str(e)}", ephemeral=True)
+            if os.path.exists(input_path):
+                os.remove(input_path)
+            if os.path.exists(output_path):
+                os.remove(output_path)
+
+
+def setup(bot):
+    bot.add_cog(Compression(bot))
\ No newline at end of file