del auth on disconnect

crustio · Oct 25, 2024 · 2997ca8 · 2997ca8
1 parent dd81f2b
commit 2997ca8
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/src/commands-handlers.ts b/src/commands-handlers.ts
@@ -210,6 +210,7 @@ export async function handleDisconnectCommand(msg: TelegramBot.Message): Promise
     }
 
     await connector.disconnect();
+    await setAuth(chatId, null);
 
     await bot.sendMessage(chatId, 'Wallet has been disconnected');
 }

diff --git a/src/ton-connect/storage.ts b/src/ton-connect/storage.ts
@@ -46,8 +46,12 @@ export async function setMode(chatId: number, mode: MODE): Promise<void> {
     await client.set(`storage_mode_${chatId}`, mode);
 }
 
-export async function setAuth(chatId: number, auth: string) {
-    await client.set(`auth_${chatId}`, auth);
+export async function setAuth(chatId: number, auth?: string | null) {
+    if (!auth) {
+        await client.del(`auth_${chatId}`);
+    } else {
+        await client.set(`auth_${chatId}`, auth);
+    }
 }
 
 export async function getAuth(chatId: number) {
-Original file line number
+Diff line change
@@ Expand Up @@
         }
         await connector.disconnect();
+        await setAuth(chatId, null);
         await bot.sendMessage(chatId, 'Wallet has been disconnected');
     }
@@ Expand Down @@