From b2579230a999b76982945a814ff80491608e70ef Mon Sep 17 00:00:00 2001
From: Anthony Laibe <anthony@laibe.cc>
Date: Tue, 8 Feb 2022 14:32:21 +0100
Subject: [PATCH] feat: prepare token service for multichain

---
 src/app_service/service/network/service.nim       | 5 +++++
 src/app_service/service/settings/dto/settings.nim | 2 +-
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/src/app_service/service/network/service.nim b/src/app_service/service/network/service.nim
index 216c2d5b48..59ab1a40d7 100644
--- a/src/app_service/service/network/service.nim
+++ b/src/app_service/service/network/service.nim
@@ -71,6 +71,11 @@ method getNetwork*(self: Service, chainId: int): NetworkDto =
     if chainId == network.chainId:
       return network
 
+method getNetwork*(self: Service, chainId: int): NetworkDto =
+  for network in self.getNetworks():
+    if chainId == network.chainId:
+      return network
+
 method getNetwork*(self: Service, networkType: NetworkType): NetworkDto =
   for network in self.getNetworks():
     if networkType.toChainId() == network.chainId:
diff --git a/src/app_service/service/settings/dto/settings.nim b/src/app_service/service/settings/dto/settings.nim
index 469c880a88..477bacf0ea 100644
--- a/src/app_service/service/settings/dto/settings.nim
+++ b/src/app_service/service/settings/dto/settings.nim
@@ -106,7 +106,7 @@ type
     telemetryServerUrl*: string
     fleet*: string
     currentUserStatus*: CurrentUserStatus
-    walletVisibleTokens*: Table[int, seq[string]] # [chainId, seq[tokenIds]]
+    walletVisibleTokens*: Table[int, seq[string]]
     nodeConfig*: JsonNode
     wakuBloomFilterMode*: bool
     recentStickerHashes*: seq[string]