Update components/middleware.py

components/middleware.py

@@ -20,7 +20,7 @@ from jwt.exceptions import (
 
 import datetime
 import time
-
+import base64
 #from geopy.geocoders import Nominatim
 #from geopy.exc import GeocoderServiceError
 #geolocator = Nominatim(user_agent="velocity")
@@ -35,17 +35,24 @@ import time
 
 
 
-def SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange):
+def SignatureVerification(mpc,X_401_Addr,X_401_Nonce,X_401_Sign,challange):
 
     
             if X_401_Addr and X_401_Nonce and X_401_Sign:
 
                 try:
 
-                    signature_bytes = base58.b58decode(X_401_Sign)
+                    
+                    signature_bytes=None
+                    
+                    if mpc==true:
+                        signature_bytes = base64.b64decode(X_401_Sign)
+                    else:
+                        signature_bytes=base58.b58decode(X_401_Sign)
+                               
+                    
                     verify_key = VerifyKey(base58.b58decode(X_401_Addr))
                     payload_bytes = challange.encode("utf-8")
-    
                     verify_key.verify(payload_bytes, signature_bytes)
                     return True
                          
@@ -265,6 +272,7 @@ class x401Kit(BaseHTTPMiddleware):
             mint_amount=float(request.headers.get("mint_amount"))
             print(mint_amount)
             helius_api_key = request.headers.get("helius_api_key")
+            mpc=request.headers.get("mpc")
         
 
            
@@ -272,7 +280,7 @@ class x401Kit(BaseHTTPMiddleware):
 
             challange=f"CHALLENGE::{X_401_Nonce}::{request.url.path}::VELOCITY401"
 
-            signverify=SignatureVerification(X_401_Addr,X_401_Nonce,X_401_Sign,challange)
+            signverify=SignatureVerification(mpc,X_401_Addr,X_401_Nonce,X_401_Sign,challange)
             tokenverify=None
             if required_mint=="no":
                 tokenverify=True