Update components/middleware.py

components/middleware.py

@@ -24,7 +24,8 @@ import time
 from geopy.geocoders import Nominatim
 from geopy.exc import GeocoderServiceError
 from .mongodbconnection import provideClient
-
+from eth_account import Account
+from eth_account.messages import encode_defunct
 
 
 
@@ -106,6 +107,27 @@ def SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange):
 
 
 
+
+
+
+
+
+def verify_ethereum_signature(address, signature, original_payload):
+  
+    try:
+     
+        message = encode_defunct(text=original_payload)
+
+        recovered_address = Account.recover_message(message, signature=signature)
+
+        return recovered_address.lower() == address.lower()
+    except Exception as e:
+        print(f"Verification failed: {e}")
+        return False
+
+
+
+
 def SecretNonceGenerator():
         random_bytes = secrets.token_bytes(32)
         return hashlib.sha256(random_bytes).hexdigest()
@@ -333,13 +355,24 @@ class x401Kit(BaseHTTPMiddleware):
             helius_api_key = request.headers.get("helius_api_key")
             geo_code=request.headers.get("geo_code")
             geo_code_locs=[request.headers.get("geo_code_locs")]
+            eth=request.headers.get("eth")            
 
            
         
 
             challange=f"CHALLENGE::{X_401_Nonce}::{request.url.path}::VELOCITY401"
 
-            signverify=SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange)
+        
+            signverify=None
+            if eth==true:
+                
+                signverify=verify_ethereum_signature(X_401_Addr,X_401_Sign,challange)
+
+            else:
+                signverify=SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange)
+            
+            
+
         
             tokenverify=None
             if required_mint =="no":