FInal changes

Author: surajstaabi

README.md

@@ -1,6 +1,6 @@
 # Discord Windows Controller Bot
 
-The Discord Windows Controller Bot is a Python bot designed to control Windows operations through Discord commands. It can perform tasks like locking your workstation and monitoring mouse movement in a specified channel. Follow these steps to set up and run the bot.
+The Discord Windows Controller Bot is a Python bot designed to control Windows operations through Discord commands. It can perform tasks like locking your workstation and monitoring mouse movement in a specified channel as well as saving files you send to the local workstation. Follow these steps to set up and run the bot.
 
 ## Step 1: Clone the Repository
 
@@ -97,6 +97,7 @@ If you still face issues, try the following:
 1. In the project folder, create a text file named "key.txt."
 2. Paste the copied bot token into the "key.txt" file and save it.
 3. Get your Channel ID and your own ID from Discord, and paste them into the "config.json" file.
+4. Create a folder where you would want the files to be saved to locally, and copy the path to the `file_save_path` variable
 
 Example:
 
@@ -105,7 +106,8 @@ Example:
     "command_channel_id": 1137276176451079374,
     "author_id": 637917562920429309,
     "log_channel_id": 1137693326543122128,
-    "mouse_log_channel_id": 1137693326543123128
+    "mouse_log_channel_id": 1137693326543123128,
+    "file_save_path": "insert/file/path"
 }
 ```
 

bot.py

@@ -5,6 +5,7 @@
 import asyncio
 import pyautogui
 from commands.mouse_movement import monitor_mouse_movement
+from aiohttp.client_exceptions import ClientConnectorError
 
 KEY_PATH = "key.txt"
 CONFIG_PATH = "config.json"
@@ -35,7 +36,7 @@ async def network_monitor():
 async def main():
     if os.path.exists(KEY_PATH):
         with open(KEY_PATH, "r") as f:
-            TOKEN = f.read()
+            TOKEN = f.read().strip()
     else:
         TOKEN = ""
 
@@ -87,14 +88,27 @@ async def on_message(message):
             from commands.help_command import execute_help_command
             await execute_help_command(client, message, config)
 
+    async def start_bot():
+        while True:
+            try:
+                # Start the network monitoring task
+                network_task = asyncio.create_task(network_monitor())
+                
+                # Start the client
+                await client.start(TOKEN)
+            except ClientConnectorError:
+                print("Failed to connect to Discord. Retrying in 30 seconds...")
+                await asyncio.sleep(30)
+            except KeyboardInterrupt:
+                print("Interrupt received. Shutting down gracefully...")
+                break
+            except Exception as e:
+                print(f"An unexpected error occurred: {e}")
+                print("Restarting the bot in 30 seconds...")
+                await asyncio.sleep(30)
+
     try:
-        # Start the network monitoring task
-        network_task = asyncio.create_task(network_monitor())
-        
-        # Start the client
-        await client.start(TOKEN)
-    except KeyboardInterrupt:
-        print("Interrupt received. Shutting down gracefully...")
+        await start_bot()
     finally:
         # Cancel all running tasks
         tasks = [t for t in asyncio.all_tasks() if t is not asyncio.current_task()]