fix tmp dir being deleted before sending file

1 files changed, 4 insertions, 1 deletions

diff --git a/cogs/jarvis.py b/cogs/jarvis.py
index 0f4606b..21f71f6 100644
--- a/cogs/jarvis.py
+++ b/cogs/jarvis.py
@@ -4,6 +4,7 @@ from discord.ext import commands
 import requests
 import json
 import logging
+import os
 
 from cogs.compression import Compression
 from cogs.downloader import Downloader
@@ -78,7 +79,7 @@ class Jarvis(commands.Cog):
 
         if action == "download":
             # Extract the URL from the message using regex
-            url_match = re.search(r'(https?://\S+)', message.content).group(0)
+            url_match = re.search(r'https://(?:[A-Za-z0-9-]+\.)*[A-Za-z0-9-]{1,63}\.[A-Za-z]{2,63}/\S*', message.content).group(0)
             if not url_match:
                 await channel.send("Nie znaleziono poprawnego linku w odpowiedzi.")
                 return
@@ -91,10 +92,12 @@ class Jarvis(commands.Cog):
                     elif response["type"] == "file":
                         file_data = response["content"]
                         await channel.send("wysyłon:", file=discord.File(file_data))
+                        os.remove(response["temp_dir"])
                 else:
                     logger.error(f"Error: {response['content']}")
                     if 'details' in response:
                         logger.error(f"Details: {response['details']}")
+                    os.remove(response["temp_dir"])
 
             elif action == "compress":
                 pass