Drift Client Architecture

import { AuthorityDrift } from '@drift-labs/common';
import { Connection } from '@solana/web3.js';

// Initialize the client
const authorityDrift = new AuthorityDrift({
  solanaRpcEndpoint: 'https://api.mainnet-beta.solana.com',
  driftEnv: 'mainnet-beta',
  wallet: myWallet, // IWalletV2 instance
  tradableMarkets: [], // Optional: filter markets to subscribe to
  selectedTradeMarket: marketId, // Optional: primary trading market
});

// Subscribe to all data sources
await authorityDrift.subscribe();

// Access real-time price data
const markPrices = authorityDrift.markPriceCache;
const oraclePrices = authorityDrift.oraclePriceCache;
const userAccounts = authorityDrift.userAccountCache;

// Listen for updates
authorityDrift.onMarkPricesUpdate((prices) => {
  console.log('Mark prices updated:', prices);
});

authorityDrift.onUserAccountUpdate((account) => {
  console.log('User account updated:', account);
});

// Execute trades
const txSig = await authorityDrift.openPerpOrder({
  marketIndex: 0, // SOL-PERP
  amount: new BN(1_000_000_000), // 1 SOL
  direction: PositionDirection.LONG,
  orderType: OrderType.MARKET,
});

// Clean up when done
await authorityDrift.unsubscribe();

// Update the active trading market for optimized polling
authorityDrift.updateSelectedTradeMarket(newMarketId);

// Update wallet authority and resubscribe
await authorityDrift.updateAuthority(newWallet, subAccountId);

import { CentralServerDrift } from '@drift-labs/common';
import { PublicKey, BN } from '@drift-labs/sdk';

// Initialize the server client
const serverDrift = new CentralServerDrift({
  solanaRpcEndpoint: 'https://api.mainnet-beta.solana.com',
  driftEnv: 'mainnet-beta',
  supportedPerpMarkets: [0, 1, 2], // SOL, BTC, ETH
  supportedSpotMarkets: [0, 1], // USDC, SOL
  additionalDriftClientConfig: {
    txVersion: 0,
    txParams: {
      computeUnits: 200000,
      computeUnitsPrice: 1000,
    },
  },
});

// Subscribe to market data
await serverDrift.subscribe();

// Create a deposit transaction for a user
const userAccountPubkey = new PublicKey('...');
const depositTxn = await serverDrift.getDepositTxn(
  userAccountPubkey,
  new BN(1_000_000), // 1 USDC
  0 // USDC market index
);

// Return transaction to client for signing
return depositTxn;

// Create user account and deposit
const { transaction, userAccountPublicKey, subAccountId } = 
  await serverDrift.getCreateAndDepositTxn(
    authority,
    amount,
    spotMarketIndex,
    {
      referrerName: 'my-referrer',
      accountName: 'Main Account',
    }
  );

// Delete user account
const deleteTxn = await serverDrift.getDeleteUserTxn(userAccountPubkey);

// Deposit collateral
const depositTxn = await serverDrift.getDepositTxn(
  userAccountPubkey,
  amount,
  spotMarketIndex
);

// Withdraw collateral
const withdrawTxn = await serverDrift.getWithdrawTxn(
  userAccountPubkey,
  amount,
  spotMarketIndex,
  { isBorrow: false, isMax: false }
);

// Open perpetual market order
await serverDrift.getOpenPerpMarketOrderTxn({
  userAccountPublicKey,
  marketIndex: 0,
  amount: new BN(1_000_000_000),
  direction: PositionDirection.LONG,
  useSwift: false, // Set to true for Swift (gasless) orders
});

// Open limit order
await serverDrift.getOpenPerpNonMarketOrderTxn({
  userAccountPublicKey,
  marketIndex: 0,
  amount: new BN(1_000_000_000),
  direction: PositionDirection.LONG,
  orderType: OrderType.LIMIT,
  limitPrice: new BN(50_000_000), // $50
});

// Edit existing order
const editTxn = await serverDrift.getEditOrderTxn(
  userAccountPubkey,
  orderId,
  {
    newLimitPrice: new BN(51_000_000),
    newBaseAmount: new BN(2_000_000_000),
  }
);

// Cancel orders
const cancelTxn = await serverDrift.getCancelOrdersTxn(
  userAccountPubkey,
  [orderId1, orderId2]
);

// Cancel all orders
const cancelAllTxn = await serverDrift.getCancelAllOrdersTxn(
  userAccountPubkey,
  MarketType.PERP, // Optional: filter by market type
  marketIndex, // Optional: filter by market
  direction // Optional: filter by direction
);

// Settle funding payments
const settleFundingTxn = await serverDrift.getSettleFundingTxn(
  userAccountPubkey
);

// Settle PnL for specific markets
const settlePnlTxn = await serverDrift.getSettlePnlTxn(
  userAccountPubkey,
  [0, 1] // Market indexes
);

// Swap between spot markets using Jupiter
const swapTxn = await serverDrift.getSwapTxn(
  userAccountPubkey,
  0, // From USDC
  1, // To SOL
  new BN(100_000_000), // 100 USDC
  {
    slippageBps: 50, // 0.5% slippage
    swapMode: 'ExactIn',
  }
);

private async driftClientContextWrapper<T>(
  userAccountPublicKey: PublicKey,
  operation: (user: User) => Promise<T>,
  externalWallet?: PublicKey
): Promise<T>

// Get a User object for reading account data
const user = await serverDrift.getUser(userAccountPubkey);

// Access user account data
const userAccount = user.getUserAccount();
const positions = user.getPerpPositions();
const spotPositions = user.getSpotPositions();
const orders = user.getOpenOrders();

// Don't forget to unsubscribe when done
await user.unsubscribe();

// Access the underlying DriftClient
const driftClient = authorityDrift.driftClient;

// Get market accounts
const perpMarket = driftClient.getPerpMarketAccount(0);
const spotMarket = driftClient.getSpotMarketAccount(0);

// Access oracle data
const oracleData = driftClient.getOracleDataForPerpMarket(0);

// Access market information
const markets = serverDrift.markets;

// Get available markets
const perpMarkets = markets.getPerpMarkets();
const spotMarkets = markets.getSpotMarkets();

// Get transaction parameters with dynamic priority fees
const txParams = authorityDrift.getTxParams({
  computeUnits: 300000, // Override compute units
});

// Use in transactions
const depositTxn = await serverDrift.getDepositTxn(
  userAccountPubkey,
  amount,
  spotMarketIndex,
  { txParams }
);