From 9b5dfeed82263075474010be5294163cd0afa754 Mon Sep 17 00:00:00 2001
From: turnoffthiscomputer <colin.remi07@gmail.com>
Date: Tue, 3 Sep 2024 15:44:50 +0200
Subject: [PATCH] fix wss proving error handling

---
 sdk/src/QRcode/OpenPassportQRcode.tsx | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/sdk/src/QRcode/OpenPassportQRcode.tsx b/sdk/src/QRcode/OpenPassportQRcode.tsx
index f8d482c89..aca547c30 100644
--- a/sdk/src/QRcode/OpenPassportQRcode.tsx
+++ b/sdk/src/QRcode/OpenPassportQRcode.tsx
@@ -108,6 +108,10 @@ const OpenPassportQRcode: React.FC<OpenPassportQRcodeProps> = ({
         case 'proof_generated':
           setProofStep(ProofSteps.PROOF_GENERATED);
           break;
+        case 'proof_generation_failed':
+          setSessionId(crypto.randomUUID());
+          setProofStep(ProofSteps.WAITING_FOR_MOBILE);
+          break;
       }
 
       if (data.proof) {
@@ -147,10 +151,6 @@ const OpenPassportQRcode: React.FC<OpenPassportQRcodeProps> = ({
       setProofStep(ProofSteps.WAITING_FOR_MOBILE);
     });
     newSocket.on('mobile_status', handleMobileStatus);
-    newSocket.on('proof_generation_failed', (data) => {
-      setProofStep(ProofSteps.WAITING_FOR_MOBILE);
-    });
-
     return () => {
       newSocket.disconnect();
     };