cloud_functions: convert env var NETWORK to sdk Network

Signed-off-by: bingyuyap <bingyu.yap.21@gmail.com>

Author: bingyuyap

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(

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');

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';
 }

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';