Update app/discord_logic.py

app/discord_logic.py

@@ -1,22 +1,31 @@
 import httpx
 import os
-import asyncio
+import dns.resolver
+import socket
 
 class DiscordApp:
     def __init__(self):
         self.token = os.getenv("DISCORD_TOKEN")
-        self.base_url = "https://discord.com/api/v10"
         self.client = None
 
+    def resolve_dns(self, hostname):
+        """Tự tìm IP của hostname nếu hệ thống bị lỗi DNS"""
+        try:
+            # Thử dùng DNS của Google để tìm IP
+            resolver = dns.resolver.Resolver()
+            resolver.nameservers = ['8.8.8.8', '1.1.1.1']
+            answers = resolver.resolve(hostname, 'A')
+            return str(answers[0])
+        except Exception:
+            return None
+
     async def start(self):
-        # Khởi tạo HTTPX Client với cấu hình bỏ qua lỗi DNS tạm thời
         if not self.client:
             self.client = httpx.AsyncClient(
-                timeout=15.0,
-                follow_redirects=True,
-                verify=False # Tạm thời bỏ qua xác thực SSL nếu mạng chập chờn
+                timeout=20.0,
+                verify=False # Bỏ qua SSL check để tránh lỗi mismatch khi gọi bằng IP
             )
-        print("✅ [Discord] HTTPX Client started.")
+        print("✅ [Discord] HTTPX Client started with DNS Fix.")
 
     async def stop(self):
         if self.client:
@@ -24,22 +33,29 @@ class DiscordApp:
 
     async def get_user_info(self, uid: str):
         if not self.token:
-            return {"status": "error", "message": "Missing DISCORD_TOKEN"}
+            return {"status": "error", "message": "Missing TOKEN"}
         
-        if not self.client:
-            await self.start()
+        # Thử lấy IP của discord.com
+        ip = self.resolve_dns("discord.com")
+        if ip:
+            url = f"https://{ip}/api/v10/users/{uid}"
+            headers = {
+                "Authorization": f"Bot {self.token}",
+                "Host": "discord.com" # Quan trọng: Phải giữ Host header là discord.com
+            }
+        else:
+            # Nếu không tìm được IP thì dùng URL mặc định
+            url = f"https://discord.com/api/v10/users/{uid}"
+            headers = {"Authorization": f"Bot {self.token}"}
 
-        headers = {"Authorization": f"Bot {self.token}"}
-        
         try:
-            # Gọi trực tiếp API
-            resp = await self.client.get(f"{self.base_url}/users/{uid}", headers=headers)
-            
+            resp = await self.client.get(url, headers=headers)
             if resp.status_code == 200:
                 data = resp.json()
                 user_id = data.get("id")
                 avatar_hash = data.get("avatar")
                 
+                # Link avatar dùng CDN (thường CDN ít bị lỗi DNS hơn)
                 if avatar_hash:
                     fmt = "gif" if avatar_hash.startswith("a_") else "png"
                     avatar_url = f"https://cdn.discordapp.com/avatars/{user_id}/{avatar_hash}.{fmt}?size=1024"
@@ -55,8 +71,6 @@ class DiscordApp:
                         "avatar": avatar_url,
                     }
                 }
-            return {"status": "error", "message": f"Discord status: {resp.status_code}"}
-
+            return {"status": "error", "message": f"Discord trả về lỗi: {resp.status_code}"}
         except Exception as e:
-            # Nếu vẫn lỗi DNS, thử dùng link dự phòng (Alternative API)
-            return {"status": "error", "message": f"Mạng lỗi: {str(e)}"}
+            return {"status": "error", "message": f"Lỗi kết nối cuối cùng: {str(e)}"}