fix: type & exports

bgd-labs · Sep 12, 2023 · 096c587 · 096c587
1 parent 181d303
commit 096c587
Show file tree

Hide file tree

Showing 5 changed files with 7 additions and 7 deletions.
diff --git a/src/commands/fork.ts b/src/commands/fork.ts
@@ -2,7 +2,7 @@ import { Command } from '@commander-js/extra-typings';
 import { tenderly } from '../utils/tenderlyClient';
 import { getGovernance } from '../govv3/governance';
 import { GovernanceV3Ethereum } from '@bgd-labs/aave-address-book';
-import { RPC_MAP, mainnetClient } from '../utils/rpcClients';
+import { CHAIN_ID_CLIENT_MAP, mainnetClient } from '../utils/rpcClients';
 import { getPayloadsController } from '../govv3/payloadsController';
 import { Hex, PublicClient } from 'viem';
 
@@ -54,7 +54,7 @@ export function addCommand(program: Command) {
  if (!payloadsControllerAddress) throw new Error('you need to provide a payloadsController');
  const payloadsController = getPayloadsController(
  payloadsControllerAddress as Hex,
- RPC_MAP[forkConfig.chainId as keyof typeof RPC_MAP] as PublicClient
+ CHAIN_ID_CLIENT_MAP[forkConfig.chainId as keyof typeof CHAIN_ID_CLIENT_MAP] as PublicClient
  );
  const payload = await payloadsController.getSimulationPayloadForExecution(Number(payloadId));
  const fork = await tenderly.fork({

diff --git a/src/commands/governance.ts b/src/commands/governance.ts
@@ -2,7 +2,7 @@ import { Command } from '@commander-js/extra-typings';
 import { simulateProposal } from '../govv3/simulate';
 import { GovernanceV3Goerli, IVotingMachineWithProofs_ABI, IVotingPortal_ABI } from '@bgd-labs/aave-address-book';
 import { HUMAN_READABLE_STATE, ProposalState, getGovernance } from '../govv3/governance';
-import { RPC_MAP, goerliClient } from '../utils/rpcClients';
+import { CHAIN_ID_CLIENT_MAP, goerliClient } from '../utils/rpcClients';
 import { logError, logInfo, logSuccess } from '../utils/logger';
 import {
  Hex,
@@ -164,8 +164,7 @@ export function addCommand(program: Command) {
  const votingMachine = getContract({
  address: machine,
  abi: IVotingMachineWithProofs_ABI,
- publicClient: RPC_MAP[Number(chainId) as keyof typeof RPC_MAP] as PublicClient,
- // walletClient: createWalletClient({ account: '0x0', chain: { id: Number(chainId) } as any, transport: http() }),
+ publicClient: CHAIN_ID_CLIENT_MAP[Number(chainId) as keyof typeof CHAIN_ID_CLIENT_MAP] as PublicClient,
  });
  const proofs = await governance.getVotingProofs(proposalId, voter as Hex, chainId);
  const encodedData = encodeFunctionData({

diff --git a/src/govv3/governance.ts b/src/govv3/governance.ts
@@ -40,7 +40,7 @@ export enum ProposalState {
  Expired,
 }
 
-export interface Governance<T extends WalletClient | undefined> {
+export interface Governance<T extends WalletClient | undefined = undefined> {
  governanceContract: GetContractReturnType<typeof IGovernanceCore_ABI, PublicClient, WalletClient>;
  cacheLogs: () => Promise<{
  createdLogs: Array<CreatedLog>;

diff --git a/src/index.ts b/src/index.ts
@@ -5,3 +5,4 @@ export * from './govv3/payloadsController';
 export * from './govv3/generatePayloadReport';
 export * from './utils/tenderlyClient';
 export * from './utils/logger';
+export * from './utils/rpcClients';
diff --git a/src/utils/rpcClients.ts b/src/utils/rpcClients.ts
@@ -36,7 +36,7 @@ export const sepoliaClient = createPublicClient({ chain: sepolia, transport: htt
 
 export const goerliClient = createPublicClient({ chain: goerli, transport: http(process.env.RPC_GOERLI) });
 
-export const RPC_MAP = {
+export const CHAIN_ID_CLIENT_MAP = {
  [mainnet.id]: mainnetClient,
  [arbitrum.id]: arbitrumClient,
  [polygon.id]: polygonClient,