Update app.py

app.py

@@ -71,31 +71,50 @@ class ZGMApiClient:
 # Khởi tạo client
 client = ZGMApiClient("https://gm.haivl.wtf", ZGM_API_TOKEN)
 
-# Hàm lấy key từ API trên VPS
-def get_key_from_api() -> tuple[Optional[str], str]:
+# Hàm kiểm tra key từ API trên VPS mà không xóa
+def validate_key_from_api(access_key: str) -> tuple[bool, str]:
     try:
-        response = requests.get(f"{KEY_API_URL}/api/keys/get", timeout=5)
-        print(f"API Response: {response.text}")  # Debug log
+        response = requests.get(f"{KEY_API_URL}/api/keys/validate", params={"key": access_key}, timeout=5)
+        print(f"Validate API Response: {response.text}")  # Debug log
         response.raise_for_status()
         data = response.json()
-        print(f"Extracted Key: {data['key']}")  # Debug log
-        return data["key"], f"Còn lại {data['remaining']} key."
+        if data["valid"]:
+            return True, f"Còn lại {data['remaining']} key."
+        else:
+            return False, "Key không hợp lệ! Vui lòng nhập key hợp lệ để tiếp tục."
+    except requests.exceptions.RequestException as e:
+        print(f"Validate API Error: {str(e)}")  # Debug log
+        return False, f"Lỗi khi kiểm tra key từ API: {str(e)}"
+
+# Hàm xóa key từ API trên VPS
+def remove_key_from_api(access_key: str) -> tuple[bool, str]:
+    try:
+        response = requests.delete(f"{KEY_API_URL}/api/keys/remove", params={"key": access_key}, timeout=5)
+        print(f"Remove API Response: {response.text}")  # Debug log
+        response.raise_for_status()
+        data = response.json()
+        if data["success"]:
+            return True, f"Còn lại {data['remaining']} key."
+        else:
+            return False, "Không thể xóa key."
     except requests.exceptions.RequestException as e:
-        print(f"API Error: {str(e)}")  # Debug log
-        return None, f"Lỗi khi lấy key từ API: {str(e)}"
+        print(f"Remove API Error: {str(e)}")  # Debug log
+        return False, f"Lỗi khi xóa key từ API: {str(e)}"
 
-# Hàm xác thực và xóa key (gọi API một lần)
+# Hàm xác thực và xóa key
 def validate_and_remove_key(access_key: str, result: Dict) -> tuple[bool, str]:
-    key_from_api, message = get_key_from_api()  # Gọi API một lần
-    print(f"Input Key: {access_key}, API Key: {key_from_api}")  # Debug log
-    if not key_from_api:
+    # Kiểm tra key có hợp lệ không
+    is_valid, message = validate_key_from_api(access_key)
+    print(f"Input Key: {access_key}, Valid: {is_valid}")  # Debug log
+    if not is_valid:
         return False, message
-    if access_key.strip() != key_from_api.strip():  # So sánh sau khi loại bỏ whitespace
-        return False, "Key không hợp lệ! Vui lòng nhập key hợp lệ để tiếp tục."
     
-    # Chỉ xóa key trên VPS khi thuê thành công (đã gọi API xóa trong get_key_from_api)
+    # Chỉ xóa key khi thuê OTP thành công
     if result.get("code") == 200 and result.get("data") and "orderId" in result["data"] and "email" in result["data"]:
-        remaining_keys = int(message.split("Còn lại ")[1].split(" key")[0])
+        success, remove_message = remove_key_from_api(access_key)
+        if not success:
+            return True, f"Thuê OTP thành công nhưng {remove_message}"
+        remaining_keys = int(remove_message.split("Còn lại ")[1].split(" key")[0])
         remaining_message = f"Còn lại {remaining_keys} key."
         if remaining_keys < 100:
             remaining_message += " Số lượng key còn ít! Vui lòng tạo thêm key mới."