cloud_functions: convert env var NETWORK to sdk Network

Signed-off-by: bingyuyap <[email protected]>
wormhole-foundation · Jun 6, 2024 · c1499ae · c1499ae
1 parent fd37c95
commit c1499ae
Show file tree

Hide file tree

Showing 4 changed files with 13 additions and 18 deletions.
diff --git a/cloud_functions/src/computeNTTRateLimits.ts b/cloud_functions/src/computeNTTRateLimits.ts
@@ -1,12 +1,12 @@
 import {
-  assertEnvironmentVariable,
   NTT_MANAGER_CONTRACT,
   NTT_TOKENS,
   NTT_TRANSCEIVER_CONTRACT,
   NTT_SUPPORTED_CHAINS,
   getEvmTokenDecimals,
   getSolanaTokenDecimals,
   NTTRateLimit,
+  getNetwork,
 } from '@wormhole-foundation/wormhole-monitor-common';
 import { EvmPlatform, EvmChains } from '@wormhole-foundation/sdk-evm';
 import { SolanaPlatform } from '@wormhole-foundation/sdk-solana';
@@ -17,7 +17,8 @@ import { Storage } from '@google-cloud/storage';
 
 const storage = new Storage();
 let bucketName: string = 'wormhole-ntt-cache';
-if (assertEnvironmentVariable('NETWORK') === 'Testnet') {
+const network = getNetwork();
+if (network === 'Testnet') {
   bucketName = 'wormhole-ntt-cache-testnet';
 }
 
@@ -115,7 +116,6 @@ export async function computeNTTRateLimits(req: any, res: any) {
   }
 
   try {
-    const network = assertEnvironmentVariable('NETWORK') as Network;
     const managerContracts = NTT_MANAGER_CONTRACT[network];
 
     const rateLimits = await Promise.all(

diff --git a/cloud_functions/src/computeTotalSupplyAndLocked.ts b/cloud_functions/src/computeTotalSupplyAndLocked.ts
@@ -1,24 +1,23 @@
 import {
   NTT_MANAGER_CONTRACT,
   NTT_TOKENS,
-  assertEnvironmentVariable,
+  NTTTotalSupplyAndLockedData,
   derivePda,
+  getCustody,
+  getCustodyAmount,
   getEvmTokenDecimals,
   getEvmTotalSupply,
+  getNetwork,
   normalizeToDecimals,
 } from '@wormhole-foundation/wormhole-monitor-common';
 import { PublicKey } from '@solana/web3.js';
-import {
-  getCustody,
-  getCustodyAmount,
-  NTTTotalSupplyAndLockedData,
-} from '@wormhole-foundation/wormhole-monitor-common';
 import { Network, rpc, Chain, chainToChainId } from '@wormhole-foundation/sdk-base';
 import { Storage } from '@google-cloud/storage';
 
 const storage = new Storage();
 let bucketName: string = 'wormhole-ntt-cache';
-if (assertEnvironmentVariable('NETWORK') === 'Testnet') {
+const network = getNetwork();
+if (network === 'Testnet') {
   bucketName = 'wormhole-ntt-cache-testnet';
 }
 
@@ -96,7 +95,6 @@ export async function computeTotalSupplyAndLocked(req: any, res: any) {
   }
 
   try {
-    const network = assertEnvironmentVariable('NETWORK') as Network;
     const totalSupplyAndLocked = await fetchTotalSupplyAndLocked(network);
     await cloudStorageCache.save(JSON.stringify(totalSupplyAndLocked));
     res.status(200).send('Total supply and locked saved');

diff --git a/cloud_functions/src/getNTTRateLimits.ts b/cloud_functions/src/getNTTRateLimits.ts
@@ -2,7 +2,7 @@ import { Storage } from '@google-cloud/storage';
 import { chainIdToChain } from '@wormhole-foundation/sdk-base';
 import {
   NTTRateLimit,
-  assertEnvironmentVariable,
+  getNetwork,
   normalizeToDecimals,
 } from '@wormhole-foundation/wormhole-monitor-common';
 import { Gauge, register } from 'prom-client';
@@ -33,7 +33,7 @@ async function setCapacityGauge(
 const storage = new Storage();
 
 let bucketName: string = 'wormhole-ntt-cache';
-const network = assertEnvironmentVariable('NETWORK');
+const network = getNetwork();
 if (network === 'Testnet') {
   bucketName = 'wormhole-ntt-cache-testnet';
 }

diff --git a/cloud_functions/src/getTotalSupplyAndLocked.ts b/cloud_functions/src/getTotalSupplyAndLocked.ts
@@ -1,11 +1,8 @@
 import { Storage } from '@google-cloud/storage';
-import {
-  NTTRateLimit,
-  assertEnvironmentVariable,
-} from '@wormhole-foundation/wormhole-monitor-common';
+import { NTTRateLimit, getNetwork } from '@wormhole-foundation/wormhole-monitor-common';
 
 const storage = new Storage();
-const network = assertEnvironmentVariable('NETWORK');
+const network = getNetwork();
 let bucketName: string = 'wormhole-ntt-cache';
 if (network === 'Testnet') {
   bucketName = 'wormhole-ntt-cache-testnet';