fix: preserve imported token migrations

Imported token migrations were losing their parent wallet context
between wallet creation and the migration flow, which made token
migrations disappear or crash before users could review them.

Track the created wallet ids during import completion and carry the
parent wallet metadata into the migration payload so imported tokens
reach the migration scenes as first-class assets.

Author: samholmes

src/components/scenes/CreateWalletCompletionScene.tsx

@@ -1,4 +1,4 @@
-import type { EdgeCreateCurrencyWallet, EdgeCurrencyWallet } from 'edge-core-js'
+import type { EdgeCreateCurrencyWallet } from 'edge-core-js'
 import * as React from 'react'
 import { ActivityIndicator, View } from 'react-native'
 import { FlatList } from 'react-native-gesture-handler'
@@ -56,7 +56,8 @@ const CreateWalletCompletionComponent: React.FC<Props> = props => {
   const defaultIsoFiat = useSelector(state => state.ui.settings.defaultIsoFiat)
 
   const [done, setDone] = React.useState(false)
-  const [wallets, setWallets] = React.useState<EdgeCurrencyWallet[]>([])
+  const [createdWalletIdsByPluginId, setCreatedWalletIdsByPluginId] =
+    React.useState<Record<string, string>>({})
 
   const { newWalletItems, newTokenItems } = React.useMemo(
     () => splitCreateWalletItems(createWalletList),
@@ -156,14 +157,13 @@ const CreateWalletCompletionComponent: React.FC<Props> = props => {
       }
 
       // Save the created wallets
-      setWallets(
-        walletResults
-          .filter(
-            (result): result is { ok: true; result: EdgeCurrencyWallet } =>
-              result.ok
-          )
-          .map(result => result.result)
-      )
+      const newCreatedWalletIdsByPluginId: Record<string, string> = {}
+      walletResults.forEach((result, index) => {
+        if (!result.ok) return
+        newCreatedWalletIdsByPluginId[newWalletItems[index].pluginId] =
+          result.result.id
+      })
+      setCreatedWalletIdsByPluginId(newCreatedWalletIdsByPluginId)
 
       setDone(true)
       return () => {}
@@ -252,23 +252,28 @@ const CreateWalletCompletionComponent: React.FC<Props> = props => {
   })
 
   const handleMigrate = useHandler(() => {
-    // Transform filtered items into the structure expected by the migration component
-    const migrateWalletList: MigrateWalletItem[] = newWalletItems.map(
-      createWallet => {
-        const { key, pluginId } = createWallet
-        const wallet = wallets.find(
-          wallet => wallet.currencyInfo.pluginId === pluginId
-        )
+    const migrateWalletList: MigrateWalletItem[] = [
+      ...newWalletItems,
+      ...newTokenItems
+    ].map(createWallet => {
+      const { key, pluginId } = createWallet
+      const parentWalletItem =
+        createWallet.walletType == null
+          ? newWalletItems.find(item => item.pluginId === pluginId)
+          : createWallet
 
-        return {
-          ...createWallet,
-          createWalletId: wallet == null ? '' : wallet.id,
-          displayName: walletNames[key],
-          key,
-          type: 'create'
-        }
+      return {
+        ...createWallet,
+        createWalletId: createdWalletIdsByPluginId[pluginId] ?? '',
+        walletType: parentWalletItem?.walletType ?? createWallet.walletType,
+        displayName:
+          parentWalletItem == null
+            ? createWallet.displayName
+            : walletNames[parentWalletItem.key] ?? parentWalletItem.displayName,
+        key,
+        type: 'create' as const
       }
-    )
+    })
 
     // Navigate to the migration screen with the prepared list
     if (migrateWalletList.length > 0) {