From 9e327f6c088169a05f71661a5409dc9bef34f923 Mon Sep 17 00:00:00 2001
From: AnisHADJARAB <anis.hadjarab@atos.net>
Date: Fri, 25 Oct 2024 14:49:55 +0000
Subject: [PATCH] put faucet message right after socket connection

---
 src/router.ts     | 6 ------
 src/websockets.ts | 6 ++++++
 2 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/src/router.ts b/src/router.ts
index 60a0893..0eae8d4 100644
--- a/src/router.ts
+++ b/src/router.ts
@@ -55,12 +55,6 @@ async function init(): Promise<boolean> {
             await services.restoreProcesses();
             await services.restoreMessages();
 
-            const amount = await services.getAmount();
-
-            if (amount === 0n) {
-                const faucetMsg = await services.createFaucetMessage();
-                await services.sendFaucetMessage(faucetMsg);
-            }
             if (services.isPaired()) { 
               isPaired = true;
               console.log("🚀 ~ setTimeout ~ isPaired:", isPaired)
diff --git a/src/websockets.ts b/src/websockets.ts
index 83236ed..57a0a09 100644
--- a/src/websockets.ts
+++ b/src/websockets.ts
@@ -13,6 +13,12 @@ class WebSocketClient {
             this.ws.onopen = async (event) => {
                 console.log('WebSocket connection established');
 
+                const amount = await services.getAmount();
+
+                if (amount === 0n) {
+                    const faucetMsg = await services.createFaucetMessage();
+                    await services.sendFaucetMessage(faucetMsg);
+                }
                 while (this.messageQueue.length > 0) {
                     const message = this.messageQueue.shift();
                     if (message) {