From 44462f68872fdc459838fe6327f01bb6b4e257eb Mon Sep 17 00:00:00 2001
From: AricRedemption <aricredemption@gmail.com>
Date: Fri, 19 Apr 2024 10:24:53 +0800
Subject: [PATCH] chore: change opening method for welcome page

---
 src/lib/utils.ts           |  8 +++++++
 src/lib/wallet.ts          | 44 +++++++++++++++++++++++++++++++++++---
 src/pages/manage/Index.vue |  4 ++--
 src/pages/wallet/Asset.vue |  6 +++---
 src/router.ts              |  8 +++----
 src/stores/WalletStore.ts  | 44 ++++++++++++++++----------------------
 6 files changed, 77 insertions(+), 37 deletions(-)

diff --git a/src/lib/utils.ts b/src/lib/utils.ts
index cd73d64f..b676eadd 100644
--- a/src/lib/utils.ts
+++ b/src/lib/utils.ts
@@ -21,6 +21,14 @@ export const goToPage = (path: string, created = false) => {
   }
 }
 
+export const goToTab = (path: string, created = false) => {
+  if (IS_DEV || !created) {
+    router.push(path)
+  } else {
+    window.open(`${window.location.href.split('#')[0]}#${path}`, '_blank')
+  }
+}
+
 export function cn(...inputs: ClassValue[]) {
   return twMerge(clsx(inputs))
 }
diff --git a/src/lib/wallet.ts b/src/lib/wallet.ts
index 63d9bc8a..20b8b3a0 100644
--- a/src/lib/wallet.ts
+++ b/src/lib/wallet.ts
@@ -1,8 +1,8 @@
 import { goToPage } from './utils'
-import { V3Wallet } from './types'
 import useStorage from '@/lib/storage'
-import { getCurrentAccountId } from './account'
+import { type V3Wallet } from './types'
 import { toast } from '@/components/ui/toast'
+import { getCurrentAccountId } from './account'
 
 const CURRENT_WALLET_ID = 'currentWalletId'
 const V3_WALLETS_STORAGE_KEY = 'wallets_v3'
@@ -61,7 +61,7 @@ export async function getV3CurrentWallet() {
   return wallet
 }
 
-export async function getActiveWalletAccount() {
+export async function getActiveWalletOnlyAccount() {
   const walletId = await getCurrentWalletId()
   if (!walletId) {
     throw new Error('current wallet id not found')
@@ -91,6 +91,44 @@ export async function getActiveWalletAccount() {
   return wallet
 }
 
+export async function getInactiveWallets() {
+  const currentWalletId = await getCurrentWalletId()
+  if (!currentWalletId) {
+    throw new Error('Current wallet id not found.')
+  }
+  const wallets = await getV3Wallets()
+  if (!wallets.length) {
+    throw new Error('No wallets found. Please create a wallet first.')
+  }
+  return wallets.filter((wallet) => wallet.id !== currentWalletId)
+}
+
+export async function getActiveWalletOtherAccounts() {
+  const walletId = await getCurrentWalletId()
+  if (!walletId) {
+    throw new Error('current wallet id not found')
+  }
+  const wallets = await getV3Wallets()
+  if (!wallets.length) {
+    throw new Error('wallets not found')
+  }
+  const wallet = wallets.find((wallet) => wallet.id === walletId)
+  if (!wallet) {
+    throw new Error('wallet not found')
+  }
+
+  const { accounts } = wallet
+  if (!accounts || !accounts.length) {
+    throw new Error('wallet does not have any accounts')
+  }
+  const currentAccountId = await getCurrentAccountId()
+  if (!currentAccountId) {
+    throw new Error('current account id not found')
+  }
+  wallet.accounts = accounts.filter((account) => account.id !== currentAccountId)
+  return wallet
+}
+
 export async function getV3CurrentAccount() {
   const walletId = await getCurrentWalletId()
   const wallets = await getV3Wallets()
diff --git a/src/pages/manage/Index.vue b/src/pages/manage/Index.vue
index 7724ad38..b801bd87 100644
--- a/src/pages/manage/Index.vue
+++ b/src/pages/manage/Index.vue
@@ -1,12 +1,12 @@
 <script setup lang="ts">
 import { ref } from 'vue'
-import { goToPage } from '@/lib/utils'
 import { EditName } from '@/components'
 import { type V3Wallet } from '@/lib/types'
 import Avatar from '@/components/Avatar.vue'
 import { totalBalance } from '@/lib/balance'
 import { toast } from '@/components/ui/toast'
 import AddIcon from '@/assets/icons-v3/add.svg'
+import { goToTab, goToPage } from '@/lib/utils'
 import { WalletsStore } from '@/stores/WalletStore'
 import PencilIcon from '@/assets/icons-v3/pencil.svg'
 import { FlexBox, Divider, Button } from '@/components'
@@ -145,7 +145,7 @@ const updataAccountName = (walletId: string, accountId: string, accountName: str
       </FlexBox>
       <Button
         type="primary"
-        @click="goToPage('/welcome', true)"
+        @click="goToTab('/welcome', true)"
         class="py-6 absolute bottom-6 w-61.5 left-1/2 -translate-x-1/2"
       >
         Add Wallet
diff --git a/src/pages/wallet/Asset.vue b/src/pages/wallet/Asset.vue
index f2d91a6f..8550af68 100644
--- a/src/pages/wallet/Asset.vue
+++ b/src/pages/wallet/Asset.vue
@@ -16,9 +16,12 @@ import FilterIcon from '@/assets/icons-v3/filter.svg'
 import ToggleIcon from '@/assets/icons-v3/toggle.svg'
 import ArrowUpIcon from '@/assets/icons-v3/arrow-up.svg'
 import SelectorIcon from '@/assets/icons-v3/selector.svg'
+import { setV3AddressTypeStorage } from '@/lib/addressType'
 import { getTags, BTCAsset, MVCAsset } from '@/data/assets'
 import ArrowDownIcon from '@/assets/icons-v3/arrow-down.svg'
 import ArrowLeftIcon from '@/assets/icons-v3/arrow-left.svg'
+import { AddressType, Chain } from '@metalet/utxo-wallet-service'
+import { useChainWalletsStore } from '@/stores/ChainWalletsStore'
 import SuccessCheckedIcon from '@/assets/icons-v3/success-checked.svg'
 import { useExchangeRatesQuery, CoinCategory } from '@/queries/exchange-rates'
 import { DropdownMenu, DropdownMenuContent, DropdownMenuItem, DropdownMenuTrigger } from '@/components/ui/dropdown-menu'
@@ -30,9 +33,6 @@ import {
   DrawerTitle,
   DrawerDescription,
 } from '@/components/ui/drawer'
-import { setV3AddressTypeStorage } from '@/lib/addressType'
-import { AddressType, Chain } from '@metalet/utxo-wallet-service'
-import { useChainWalletsStore } from '@/stores/ChainWalletsStore'
 
 const isOpen = ref(false)
 const route = useRoute()
diff --git a/src/router.ts b/src/router.ts
index 09627515..0e2816c4 100644
--- a/src/router.ts
+++ b/src/router.ts
@@ -1,12 +1,12 @@
 import useStorage from './lib/storage'
-import { goToPage } from '@/lib/utils'
+import { goToPage, goToTab } from '@/lib/utils'
 import { IS_DEV } from '@/data/config'
 import * as VueRouter from 'vue-router'
 import { assetList } from '@/lib/balance'
+import { needMigrate } from './lib/migrate'
 import Wallet from './pages/wallet/Index.vue'
 import { getCurrentAccountId } from './lib/account'
-import { getCurrentWalletId, getV3Wallets, hasWallets } from './lib/wallet'
-import { needMigrate } from './lib/migrate'
+import { getCurrentWalletId, hasWallets } from './lib/wallet'
 
 const storage = useStorage()
 
@@ -475,7 +475,7 @@ router.beforeEach(async (to, _, next) => {
     if (await hasWallets()) {
       next('/manage/wallets')
     } else {
-      goToPage('/welcome', true)
+      goToTab('/welcome', true)
       next('/welcome')
     }
   } else {
diff --git a/src/stores/WalletStore.ts b/src/stores/WalletStore.ts
index 2d512c4d..9a761635 100644
--- a/src/stores/WalletStore.ts
+++ b/src/stores/WalletStore.ts
@@ -1,29 +1,31 @@
 import { reactive } from 'vue'
 import { getNet } from '@/lib/network'
-import { goToPage } from '@/lib/utils'
 import { toast } from '@/components/ui/toast'
+import { goToPage, goToTab } from '@/lib/utils'
 import { getCurrentAccountId } from '@/lib/account'
 import { getV3AddressTypeStorage } from '@/lib/addressType'
 import { WalletManager } from '@metalet/utxo-wallet-service'
 import { Chain, type Net } from '@metalet/utxo-wallet-service'
-import { getV3Wallets, getActiveWalletAccount, getCurrentWalletId, hasV3Wallets } from '@/lib/wallet'
+import {
+  hasV3Wallets,
+  getCurrentWalletId,
+  getInactiveWallets,
+  getActiveWalletOnlyAccount,
+  getActiveWalletOtherAccounts,
+} from '@/lib/wallet'
 
 let walletManager: WalletManager | null = null
 
 const initWalletManager = async (): Promise<WalletManager> => {
   try {
-    const activeWallet = await getActiveWalletAccount()
+    const activeWallet = await getActiveWalletOnlyAccount()
     const walletsOptions = [
       {
         id: activeWallet.id,
         mnemonic: activeWallet.mnemonic,
         name: activeWallet.name,
         mvcTypes: activeWallet.mvcTypes,
-        accountsOptions: activeWallet.accounts.map(({ id, name, addressIndex }) => ({
-          id,
-          name,
-          addressIndex,
-        })),
+        accountsOptions: activeWallet.accounts.map(({ id, name, addressIndex }) => ({ id, name, addressIndex })),
       },
     ]
     const network = getNet() as Net
@@ -35,7 +37,7 @@ const initWalletManager = async (): Promise<WalletManager> => {
     if (await hasV3Wallets()) {
       goToPage('/manage/wallets')
     } else {
-      goToPage('/welcome', true)
+      goToTab('/welcome', true)
     }
     throw err
   }
@@ -54,38 +56,30 @@ const getWalletManager = async (): Promise<WalletManager> => {
 
 const loadOtherAccounts = async () => {
   const manager = await getWalletManager()
-  const wallets = await getV3Wallets()
-  const activeWallet = await getActiveWalletAccount()
-  const currentAccountId = await getCurrentAccountId()
-  if (!activeWallet || !currentAccountId) {
-    const wallets = await getV3Wallets()
-    if (wallets.length) {
+  const activeWallet = await getActiveWalletOtherAccounts()
+  if (!activeWallet) {
+    if (await hasV3Wallets()) {
       goToPage('/manage/wallets')
     } else {
-      goToPage('/welcome', true)
+      goToTab('/welcome', true)
     }
     return
   }
-  const accounts = activeWallet.accounts.filter((account) => account.id !== currentAccountId)
-  for (let account of accounts) {
+  for (let account of activeWallet.accounts) {
     manager.addAccount(activeWallet.id, {
       id: account.id,
       name: account.name,
       addressIndex: account.addressIndex,
     })
   }
-  const otherWallets = wallets.filter((wallet) => wallet.id !== activeWallet.id)
-  for (let wallet of otherWallets) {
+  const inactiveWallets = await getInactiveWallets()
+  for (let wallet of inactiveWallets) {
     manager.addWallet({
       id: wallet.id,
       name: wallet.name,
       mnemonic: wallet.mnemonic,
       mvcTypes: wallet.mvcTypes,
-      accountsOptions: wallet.accounts.map((account) => ({
-        id: account.id,
-        name: account.name,
-        addressIndex: account.addressIndex,
-      })),
+      accountsOptions: wallet.accounts.map(({ id, name, addressIndex }) => ({ id, name, addressIndex })),
     })
   }
 }