Add vault native functions

v4-client-js/examples/native_examples.ts

@@ -15,6 +15,9 @@ import {
   withdraw,
   withdrawToIBC,
   wrappedError,
+  getMegavaultOwnerShares,
+  depositToMegavault,
+  withdrawFromMegavault,
 } from '../src/clients/native';
 import { DYDX_TEST_ADDRESS, DYDX_TEST_MNEMONIC } from './constants';
 
@@ -47,6 +50,19 @@ async function test(): Promise<void> {
     const userStats = await getUserStats(payload);
     console.log(userStats);
 
+    const balances1 = await getAccountBalances();
+    console.log(balances1);
+
+    const vaultOwnerShares = await getMegavaultOwnerShares(payload);
+    console.log(vaultOwnerShares);
+
+    // const depositResult = await depositToMegavault(0, 2);
+    // console.log(depositResult);
+
+    const withdrawResult = await withdrawFromMegavault(0, 1, 0);
+    console.log(withdrawResult);
+
+
     const sendTokenPayload = {
       subaccountNumber: 0,
       amount: '10', // Dydx Token

v4-client-js/src/clients/constants.ts

@@ -23,13 +23,15 @@ export const MAINNET_CHAIN_ID = 'dydx-mainnet-1';
 // ------------ API URLs ------------
 export enum IndexerApiHost {
   TESTNET = 'https://indexer.v4testnet.dydx.exchange/',
+  STAGING = 'https://indexer.v4staging.dydx.exchange/',
   LOCAL = 'http://localhost:3002',
   // For the deployment by DYDX token holders
   MAINNET = 'https://indexer.dydx.trade',
 }
 
 export enum IndexerWSHost {
   TESTNET = 'wss://dydx-testnet.imperator.co/v4/ws',
+  STAGING = 'wss://indexer.v4staging.dydx.exchange/v4/ws',
   LOCAL = 'ws://localhost:3003',
   // For the deployment by DYDX token holders
   MAINNET = 'wss://indexer.dydx.trade/v4/ws',
@@ -41,6 +43,7 @@ export enum FaucetApiHost {
 
 export enum ValidatorApiHost {
   TESTNET = 'https://test-dydx.kingnodes.com',
+  STAGING = 'https://validator.v4staging.dydx.exchange',
   LOCAL = 'http://localhost:26657',
   // For the deployment by DYDX token holders
   MAINNET = 'https://dydx-ops-rpc.kingnodes.com:443',
@@ -262,6 +265,24 @@ export class Network {
     return new Network('testnet', indexerConfig, validatorConfig);
   }
 
+  static staging(): Network {
+    const indexerConfig = new IndexerConfig(IndexerApiHost.STAGING, IndexerWSHost.STAGING);
+    const validatorConfig = new ValidatorConfig(
+      ValidatorApiHost.STAGING,
+      TESTNET_CHAIN_ID,
+      {
+        CHAINTOKEN_DENOM: 'adv4tnt',
+        USDC_DENOM: 'ibc/8E27BA2D5493AF5636760E354E46004562C46AB7EC0CC4C1CA14E9E20E2545B5',
+        USDC_GAS_DENOM: 'uusdc',
+        USDC_DECIMALS: 6,
+        CHAINTOKEN_DECIMALS: 18,
+      },
+      undefined,
+      'Client Example',
+    );
+    return new Network('staging', indexerConfig, validatorConfig);
+  }
+
   static local(): Network {
     const indexerConfig = new IndexerConfig(IndexerApiHost.LOCAL, IndexerWSHost.LOCAL);
     const validatorConfig = new ValidatorConfig(

v4-client-js/src/clients/native.ts

@@ -1372,21 +1372,72 @@ export async function getMegavaultOwnerShares(payload: string): Promise<string>
   }
 }
 
-export async function getMegavaultWithdrawalInfo(payload: string): Promise<string> {
+export async function getMegavaultWithdrawalInfo(
+  sharesToWithdraw: bigint
+): Promise<string> {
   try {
     const client = globalThis.client;
     if (client === undefined) {
       throw new UserError('client is not connected. Call connectClient() first');
     }
-    const json = JSON.parse(payload);
-    const address = json.address;
-    if (address === undefined) {
-      throw new UserError('address is not set');
-    }
     const response =
-      await globalThis.client?.validatorClient.get.getMegavaultWithdrawalInfo(address);
+      await globalThis.client?.validatorClient.get.getMegavaultWithdrawalInfo(sharesToWithdraw);
     return encodeJson(response);
   } catch (e) {
     return wrappedError(e);
   }
-}
+}
+
+export async function depositToMegavault(
+  subaccountNumber: number,
+  amountUsdc: number
+): Promise<string> {
+  try {
+    const client = globalThis.client;
+    if (client === undefined) {
+      throw new UserError('client is not connected. Call connectNetwork() first');
+    }
+    const wallet = globalThis.wallet;
+    if (wallet === undefined) {
+      throw new UserError('wallet is not set. Call connectWallet() first');
+    }
+    const subaccount = new SubaccountInfo(wallet, subaccountNumber);
+    const tx = await client.depositToMegavault(
+      subaccount,
+      amountUsdc,
+      Method.BroadcastTxCommit,
+    );
+    return encodeJson(tx);
+  } catch (error) {
+    return wrappedError(error);
+  }
+}
+
+export async function withdrawFromMegavault(
+  subaccountNumber: number,
+  shares: number,
+  minAmount: number,
+): Promise<string> {
+  try {
+    const client = globalThis.client;
+    if (client === undefined) {
+      throw new UserError('client is not connected. Call connectNetwork() first');
+    }
+    const wallet = globalThis.wallet;
+    if (wallet === undefined) {
+      throw new UserError('wallet is not set. Call connectWallet() first');
+    }
+    const subaccount = new SubaccountInfo(wallet, subaccountNumber);
+    const tx = await client.withdrawFromMegavault(
+      subaccount,
+      shares,
+      minAmount,
+      Method.BroadcastTxCommit,
+    );
+    return encodeJson(tx);
+  } catch (error) {
+    return wrappedError(error);
+  }
+}
+
+