From f4d8f8652f4bd2c16487b3766e4ffecb2fb79cc5 Mon Sep 17 00:00:00 2001
From: Sosthene <contact@3ni.tech>
Date: Wed, 11 Jun 2025 21:43:50 +0200
Subject: [PATCH] Profile and Folder event handler return object

---
 src/router.ts | 27 +++++++++++++++++++--------
 1 file changed, 19 insertions(+), 8 deletions(-)

diff --git a/src/router.ts b/src/router.ts
index 8fc6e3c..9b7af30 100755
--- a/src/router.ts
+++ b/src/router.ts
@@ -247,19 +247,23 @@ export async function registerAllListeners() {
       const { profileData, accessToken, refreshToken } = event.data;
 
       // Validate the session token
-      if (!accessToken || !tokenService.validateToken(accessToken, event.origin)) {
+      if (!accessToken || await !tokenService.validateToken(accessToken, event.origin)) {
         throw new Error('Invalid or expired session token');
       }
 
       // Create profile
-      await services.createAndSendProfileTx(profileData); 
+      const { processId, process } = await services.createAndSendProfileTx(profileData); 
+
+      const res = {
+        processId,
+        process,
+        profileData,
+      };
       
       window.parent.postMessage(
         { 
           type: MessageType.PROFILE_CREATED,
-          profileData,
-          accessToken,
-          refreshToken
+          profileCreated: res,
         },
         event.origin
       );
@@ -331,6 +335,7 @@ export async function registerAllListeners() {
     if (event.data.type !== MessageType.CREATE_FOLDER) {
       return;
     }
+
     const tokenService = await TokenService.getInstance();
 
     if (!services.isPaired()) {
@@ -342,16 +347,22 @@ export async function registerAllListeners() {
     try {
       const { folderData, token } = event.data;
 
-      if (!token || !tokenService.validateToken(token, event.origin)) {
+      if (!token || await !tokenService.validateToken(token, event.origin)) {
         throw new Error('Invalid or expired session token');
       }
 
-      await services.createAndSendFolderTx(folderData);
+      const { processId, process } = await services.createAndSendFolderTx(folderData);
+      
+      const res = {
+        processId,
+        process,
+        folderData,
+      };
       
       window.parent.postMessage(
         { 
           type: MessageType.FOLDER_CREATED,
-          token
+          folderCreated: res,
         },
         event.origin
       );