🧹 Stargate state snapshot: Collection Token information [5/N] (#24)

Author: janjakubnanista

packages/stg-evm-v2/devtools/src/asset/sdk.ts

@@ -79,6 +79,11 @@ export class Asset extends Ownable implements IAsset {
         }
     }
 
+    @AsyncRetriable()
+    async getToken(): Promise<OmniAddress | undefined> {
+        return ignoreZero(await this.contract.contract.token())
+    }
+
     @AsyncRetriable()
     async getLPToken(): Promise<OmniAddress | undefined> {
         // We want to return undefined if there is no lpToken function defined on the asset as opposed to throwing

packages/stg-evm-v2/devtools/src/asset/types.ts

@@ -34,6 +34,7 @@ export interface IAsset extends IOmniSDK, IOwnable {
     isPaused(): Promise<boolean>
     setPaused(pause: boolean): Promise<OmniTransaction>
 
+    getToken(): Promise<OmniAddress | undefined>
     getLPToken(): Promise<OmniAddress | undefined>
 }
 

packages/stg-evm-v2/tasks/snapshot.ts

@@ -172,11 +172,12 @@ const createCollectAsset =
         const sdk = await createSdk(point)
 
         logger.verbose(`Collecting basic information`)
-        const [owner, paused, addressConfig, lpTokenAddress] = await Promise.all([
+        const [owner, paused, addressConfig, lpTokenAddress, tokenAddress] = await Promise.all([
             sdk.getOwner(),
             sdk.isPaused(),
             sdk.getAddressConfig(),
             sdk.getLPToken(),
+            sdk.getToken(),
         ])
 
         // Now we'll check the OFT paths
@@ -199,13 +200,17 @@ const createCollectAsset =
         const lpToken =
             lpTokenAddress == null ? undefined : await collectERC20({ address: lpTokenAddress, eid: point.eid })
 
+        // If the asset has an LP token attached, we'll collect its information
+        const token = tokenAddress == null ? undefined : await collectERC20({ address: tokenAddress, eid: point.eid })
+
         // For now we'll collect nothing at all
         const snapshot: AssetSnapshot = {
             address: point.address,
             addressConfig,
             owner,
             paused,
             lpToken,
+            token,
             oftPaths: oftPaths.flat(),
         }
 
@@ -313,6 +318,7 @@ interface AssetSnapshot {
     address: OmniAddress
     addressConfig: AddressConfig
     lpToken?: ERC20TokenSnapshot
+    token?: ERC20TokenSnapshot
     oftPaths: OFTPath[]
 }