Update components/middleware.py

components/middleware.py

@@ -25,6 +25,8 @@ import time
 #from geopy.exc import GeocoderServiceError
 #geolocator = Nominatim(user_agent="velocity")
 
+from mongodbconnection import provideClient
+
 from privy import PrivyAPI
 
 client = PrivyAPI(
@@ -259,12 +261,71 @@ class x401Kit(BaseHTTPMiddleware):
         
 
 
+
+            
             NONCE=SecretNonceGenerator()
+
+            aud=request.headers.get("origin")
+            MPC_=request.headers.get("mpc")
+            MPC_ID=request.headers.get("mpc_id")
+            MPC_STATUS=request.headers.get("mpc_status")
+            
+
+            if MPC=="true" and MPC_STATUS=="active":
+                
+                document=coll.find_one({"onwer":"system"})
+                for doc in  document:
+                    if doc["id"]==MPC_ID:
+                        decoded=verifyJWT(doc["x-jwt"],aud)
+                        return JSONResponse(
+                            content={"status":"active","token":decoded,"message":"identified already"},
+                            status_code=200,
+                            headers={
+    
+                                     "Access-Control-Allow-Origin": "*",
+                                    "Access-Control-Allow-Credentials": "true"                        
+                            
+                                })
+
+            elif MPC=="true" and MPC_STATUS=="create":
+                wallet = client.wallets.create( chain_type="solana")
+                payload={
+                             "id":wallet.id,
+                              "x-jwt":""
+                        }
+                        
+                coll.update_one(
+                            {"owner": "system"},
+                            {"$push": {"mpc": payload}}
+                        )
+
+                return JSONResponse(
+                        content={"status":"ready","id":wallet.id},
+                        status_code+200,
+                        headers={
+                            
+                            "Access-Control-Allow-Origin":"*",
+                             "Access-Control-Allow-Credentials": "true"
+                            
+                        }
+                )
+                
+
+            elif MPC=="true" and MPC_STATUS=="ready":
+
+
+                
+                        
+                       
+                        
+                        
+                
+                
             
             X_401_Nonce=request.headers.get("X-401-Nonce")
             X_401_Sign=request.headers.get("X-401-Signature")
             X_401_Addr=request.headers.get("X-401-Addr")
-            aud=request.headers.get("origin")
+            
             client_jwt=request.headers.get("x-jwt")
         
 
@@ -274,7 +335,7 @@ class x401Kit(BaseHTTPMiddleware):
             if client_jwt:
                         decoded=verifyJWT(client_jwt,aud)
                         return JSONResponse(
-                                content={"status":"identified","token":X_401_Addr,"message":"identified already"},
+                                content={"status":"identified","token":decoded,"message":"identified already"},
                                 status_code=200,
                                 headers={