Withdraw to Privy Wallet and Check Balance

import { PrivyClient } from '@privy-io/node';
import {
  createSolanaRpc,
  address,
  pipe,
  createNoopSigner,
  createTransactionMessage,
  setTransactionMessageFeePayerSigner,
  setTransactionMessageLifetimeUsingBlockhash,
  appendTransactionMessageInstructions,
  compileTransaction,
  getBase64EncodedWireTransaction,
} from '@solana/kit';
import { KaminoVault } from '@kamino-finance/klend-sdk';
import { Decimal } from 'decimal.js';

const PRIVY_APP_ID = 'your-privy-app-id';
const PRIVY_APP_SECRET = 'your-privy-app-secret';
const RPC_ENDPOINT = 'https://api.mainnet-beta.solana.com';
const VAULT_ADDRESS = 'HDsayqAsDWy3QvANGqh2yNraqcD8Fnjgh73Mhb3WRS5E'; // Steakhouse USDC vault
const AUTH_KEY_PRIVATE = 'your-auth-key-private';
const WALLET_ID = 'your-wallet-id';

const privy = new PrivyClient({
  appId: PRIVY_APP_ID,
  appSecret: PRIVY_APP_SECRET,
});

const rpc = createSolanaRpc(RPC_ENDPOINT);

const wallet = await privy.wallets().get(WALLET_ID);
const walletAddress = wallet.address;

const vault = new KaminoVault(rpc, address(VAULT_ADDRESS));
await vault.getState();

const userShares = await vault.getUserShares(address(walletAddress));
console.log(
  `Vault shares: ${userShares.unstakedShares?.toString() || '0'} unstaked, ${
    userShares.stakedShares?.toString() || '0'
  } staked`
);

const totalShares = parseFloat(userShares.totalShares?.toString() || '0');

if (totalShares === 0) throw new Error('No shares to withdraw');

const noopSigner = createNoopSigner(address(walletAddress));
const withdrawAmount = new Decimal(totalShares);

const bundle = await vault.withdrawIxs(noopSigner, withdrawAmount);
const withdrawInstructions = [...(bundle.withdrawIxs || [])];

if (!withdrawInstructions.length) throw new Error('No withdrawal instructions returned');

const { value: withdrawBlockhash } = await rpc.getLatestBlockhash({ commitment: 'finalized' }).send();
const withdrawTxMessage = pipe(
  createTransactionMessage({ version: 0 }),
  (tx) => setTransactionMessageFeePayerSigner(noopSigner, tx),
  (tx) => setTransactionMessageLifetimeUsingBlockhash(withdrawBlockhash, tx),
  (tx) => appendTransactionMessageInstructions(withdrawInstructions, tx)
);

const compiledWithdrawTx = compileTransaction(withdrawTxMessage);
const serializedWithdrawTx = getBase64EncodedWireTransaction(compiledWithdrawTx);

const withdrawSignResponse = await privy
  .wallets()
  .solana()
  .signTransaction(WALLET_ID, {
    transaction: serializedWithdrawTx,
    authorization_context: { authorization_private_keys: [AUTH_KEY_PRIVATE] },
  });

const withdrawSig = await rpc
  .sendTransaction(withdrawSignResponse.signed_transaction as any, {
    encoding: 'base64',
    skipPreflight: true,
  })
  .send();

console.log('Withdrawal successful! Signature:', withdrawSig);

const balance = await privy.wallets().balance.get(WALLET_ID, {
  asset: 'usdc',
  chain: 'solana',
});
const usdcBalance = balance.balances[0];
console.log(`USDC Balance: ${usdcBalance.display_values.usdc}`);