From 89315cc0ed23fe5a0bce3bff9f241ed322c56202 Mon Sep 17 00:00:00 2001
From: Jennie Alles <jennieramida@gmail.com>
Date: Wed, 8 Nov 2023 19:06:24 +0700
Subject: [PATCH] fix(components): add remove saved account function

---
 src/lib/components/modal/account/RemoveSavedAccount.tsx | 6 ++++--
 src/lib/stores/account.ts                               | 6 ++++++
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/src/lib/components/modal/account/RemoveSavedAccount.tsx b/src/lib/components/modal/account/RemoveSavedAccount.tsx
index 461285d3f..1e16443c6 100644
--- a/src/lib/components/modal/account/RemoveSavedAccount.tsx
+++ b/src/lib/components/modal/account/RemoveSavedAccount.tsx
@@ -9,6 +9,7 @@ import { useCallback } from "react";
 
 import { ActionModal } from "../ActionModal";
 import { CustomIcon } from "lib/components/icon";
+import { useAccountStore } from "lib/providers/store";
 import type { AccountLocalInfo } from "lib/stores/account";
 import { truncate } from "lib/utils";
 
@@ -35,8 +36,9 @@ export function RemoveSavedAccountModal({
   ),
 }: RemoveSavedAccountModalProps) {
   const toast = useToast();
+  const { removeSavedAccount } = useAccountStore();
   const handleRemove = useCallback(() => {
-    // removeSavedCode(codeId);
+    removeSavedAccount(account.address);
 
     toast({
       title: `Removed \u2018${account.name}\u2019 from Saved Codes`,
@@ -46,7 +48,7 @@ export function RemoveSavedAccountModal({
       position: "bottom-right",
       icon: <CustomIcon name="check-circle-solid" color="success.main" />,
     });
-  }, [account.name, toast]);
+  }, [removeSavedAccount, account.address, account.name, toast]);
   return (
     <ActionModal
       title={
diff --git a/src/lib/stores/account.ts b/src/lib/stores/account.ts
index a86960f86..6e2415fa4 100644
--- a/src/lib/stores/account.ts
+++ b/src/lib/stores/account.ts
@@ -79,6 +79,12 @@ export class AccountStore {
     return this.savedAccounts[this.userKey]?.includes(address) ?? false;
   }
 
+  removeSavedAccount(address: Addr): void {
+    this.savedAccounts[this.userKey] = this.savedAccounts[this.userKey]?.filter(
+      (each) => each !== address
+    );
+  }
+
   getSavedAccounts(): AccountLocalInfo[] {
     const savedAccountsByUserKey = this.savedAccounts[this.userKey] ?? [];
     return savedAccountsByUserKey