@drift-labs/sdk 2.98.0-beta.11 → 2.98.0-beta.13

package/VERSION

@@ -1 +1 @@
-2.98.0-beta.11
+2.98.0-beta.13

package/lib/browser/driftClient.js

@@ -3880,6 +3880,9 @@ class DriftClient {
         const spotMarketAccount = this.getSpotMarketAccount(marketIndex);
         const isSolMarket = spotMarketAccount.mint.equals(spotMarkets_1.WRAPPED_SOL_MINT);
         const createWSOLTokenAccount = isSolMarket && collateralAccountPublicKey.equals(this.wallet.publicKey);
+        // create associated token account because it may not exist
+        const associatedTokenAccountPublicKey = (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarketAccount.mint, this.wallet.publicKey, true);
+        addIfStakeIxs.push(await (0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(this.wallet.publicKey, associatedTokenAccountPublicKey, this.wallet.publicKey, spotMarketAccount.mint));
         let tokenAccount;
         if (!(await this.checkIfAccountExists(this.getUserStatsAccountPublicKey()))) {
             addIfStakeIxs.push(await this.getInitializeUserStatsIx());

package/lib/browser/index.d.ts

@@ -106,6 +106,7 @@ export * from './dlob/DLOBApiClient';
 export * from './dlob/types';
 export * from './dlob/orderBookLevels';
 export * from './userMap/userMap';
+export * from './userMap/referrerMap';
 export * from './userMap/userStatsMap';
 export * from './userMap/userMapConfig';
 export * from './math/bankruptcy';

package/lib/browser/index.js

@@ -129,6 +129,7 @@ __exportStar(require("./dlob/DLOBApiClient"), exports);
 __exportStar(require("./dlob/types"), exports);
 __exportStar(require("./dlob/orderBookLevels"), exports);
 __exportStar(require("./userMap/userMap"), exports);
+__exportStar(require("./userMap/referrerMap"), exports);
 __exportStar(require("./userMap/userStatsMap"), exports);
 __exportStar(require("./userMap/userMapConfig"), exports);
 __exportStar(require("./math/bankruptcy"), exports);

package/lib/browser/memcmp.d.ts

@@ -5,3 +5,6 @@ export declare function getUserWithOrderFilter(): MemcmpFilter;
 export declare function getUserWithAuctionFilter(): MemcmpFilter;
 export declare function getUserThatHasBeenLP(): MemcmpFilter;
 export declare function getUserWithName(name: string): MemcmpFilter;
+export declare function getUserStatsFilter(): MemcmpFilter;
+export declare function getUserStatsIsReferredFilter(): MemcmpFilter;
+export declare function getUserStatsIsReferredOrReferrerFilter(): MemcmpFilter;

package/lib/browser/memcmp.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getUserWithName = exports.getUserThatHasBeenLP = exports.getUserWithAuctionFilter = exports.getUserWithOrderFilter = exports.getNonIdleUserFilter = exports.getUserFilter = void 0;
+exports.getUserStatsIsReferredOrReferrerFilter = exports.getUserStatsIsReferredFilter = exports.getUserStatsFilter = exports.getUserWithName = exports.getUserThatHasBeenLP = exports.getUserWithAuctionFilter = exports.getUserWithOrderFilter = exports.getNonIdleUserFilter = exports.getUserFilter = void 0;
 const bs58_1 = __importDefault(require("bs58"));
 const anchor_1 = require("@coral-xyz/anchor");
 const userName_1 = require("./userName");
@@ -61,3 +61,30 @@ function getUserWithName(name) {
     };
 }
 exports.getUserWithName = getUserWithName;
+function getUserStatsFilter() {
+    return {
+        memcmp: {
+            offset: 0,
+            bytes: bs58_1.default.encode(anchor_1.BorshAccountsCoder.accountDiscriminator('UserStats')),
+        },
+    };
+}
+exports.getUserStatsFilter = getUserStatsFilter;
+function getUserStatsIsReferredFilter() {
+    return {
+        memcmp: {
+            offset: 188,
+            bytes: bs58_1.default.encode(Buffer.from(Uint8Array.from([2]))),
+        },
+    };
+}
+exports.getUserStatsIsReferredFilter = getUserStatsIsReferredFilter;
+function getUserStatsIsReferredOrReferrerFilter() {
+    return {
+        memcmp: {
+            offset: 188,
+            bytes: bs58_1.default.encode(Buffer.from(Uint8Array.from([3]))),
+        },
+    };
+}
+exports.getUserStatsIsReferredOrReferrerFilter = getUserStatsIsReferredOrReferrerFilter;

package/lib/browser/userMap/referrerMap.d.ts

@@ -0,0 +1,45 @@
+import { MemcmpFilter } from '@solana/web3.js';
+import { BulkAccountLoader } from '../accounts/bulkAccountLoader';
+import { DriftClient } from '../driftClient';
+import { ReferrerInfo } from '../types';
+export declare class ReferrerMap {
+    /**
+     * map from authority pubkey to ReferrerInfo.
+     * - if a user has not been entered into the map, the value is undefined
+     * - if a user has no referrer, the value is null
+     * - if a user has a referrer, the value is a ReferrerInfo object
+     */
+    private referrerMap;
+    private driftClient;
+    private bulkAccountLoader;
+    private parallelSync;
+    private fetchPromise?;
+    private fetchPromiseResolver;
+    /**
+     * Creates a new UserStatsMap instance.
+     *
+     * @param {DriftClient} driftClient - The DriftClient instance.
+     * @param {BulkAccountLoader} [bulkAccountLoader] - If not provided, a new BulkAccountLoader with polling disabled will be created.
+     */
+    constructor(driftClient: DriftClient, bulkAccountLoader?: BulkAccountLoader, parallelSync?: boolean);
+    /**
+     * Subscribe to all UserStats accounts.
+     */
+    subscribe(): Promise<void>;
+    has(authorityPublicKey: string): boolean;
+    get(authorityPublicKey: string): ReferrerInfo | undefined | null;
+    addReferrerInfo(authority: string, referrerInfo?: ReferrerInfo | null): Promise<void>;
+    /**
+     * Enforce that a UserStats will exist for the given authorityPublicKey,
+     * reading one from the blockchain if necessary.
+     * @param authorityPublicKey
+     * @returns
+     */
+    mustGet(authorityPublicKey: string): Promise<ReferrerInfo | null | undefined>;
+    values(): IterableIterator<ReferrerInfo | null>;
+    size(): number;
+    sync(): Promise<void>;
+    syncAll(): Promise<void>;
+    syncReferrer(referrerFilter: MemcmpFilter): Promise<void>;
+    unsubscribe(): Promise<void>;
+}

package/lib/browser/userMap/referrerMap.js

@@ -0,0 +1,186 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ReferrerMap = void 0;
+const web3_js_1 = require("@solana/web3.js");
+const bulkAccountLoader_1 = require("../accounts/bulkAccountLoader");
+const userStats_1 = require("../userStats");
+const pda_1 = require("../addresses/pda");
+const memcmp_1 = require("../memcmp");
+const bytes_1 = require("@coral-xyz/anchor/dist/cjs/utils/bytes");
+const DEFAULT_PUBLIC_KEY = web3_js_1.PublicKey.default.toBase58();
+class ReferrerMap {
+    /**
+     * Creates a new UserStatsMap instance.
+     *
+     * @param {DriftClient} driftClient - The DriftClient instance.
+     * @param {BulkAccountLoader} [bulkAccountLoader] - If not provided, a new BulkAccountLoader with polling disabled will be created.
+     */
+    constructor(driftClient, bulkAccountLoader, parallelSync) {
+        /**
+         * map from authority pubkey to ReferrerInfo.
+         * - if a user has not been entered into the map, the value is undefined
+         * - if a user has no referrer, the value is null
+         * - if a user has a referrer, the value is a ReferrerInfo object
+         */
+        this.referrerMap = new Map();
+        this.driftClient = driftClient;
+        if (!bulkAccountLoader) {
+            bulkAccountLoader = new bulkAccountLoader_1.BulkAccountLoader(driftClient.connection, driftClient.opts.commitment, 0);
+        }
+        this.bulkAccountLoader = bulkAccountLoader;
+        this.parallelSync = parallelSync !== undefined ? parallelSync : true;
+    }
+    /**
+     * Subscribe to all UserStats accounts.
+     */
+    async subscribe() {
+        if (this.size() > 0) {
+            return;
+        }
+        await this.driftClient.subscribe();
+        await this.sync();
+    }
+    has(authorityPublicKey) {
+        return this.referrerMap.has(authorityPublicKey);
+    }
+    get(authorityPublicKey) {
+        return this.referrerMap.get(authorityPublicKey);
+    }
+    async addReferrerInfo(authority, referrerInfo) {
+        if (referrerInfo || referrerInfo === null) {
+            this.referrerMap.set(authority, referrerInfo);
+        }
+        else if (referrerInfo === undefined) {
+            const userStat = new userStats_1.UserStats({
+                driftClient: this.driftClient,
+                userStatsAccountPublicKey: (0, pda_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, new web3_js_1.PublicKey(authority)),
+                accountSubscription: {
+                    type: 'polling',
+                    accountLoader: this.bulkAccountLoader,
+                },
+            });
+            await userStat.fetchAccounts();
+            const newReferrerInfo = userStat.getReferrerInfo();
+            if (newReferrerInfo) {
+                this.referrerMap.set(authority.toString(), newReferrerInfo);
+            }
+            else {
+                this.referrerMap.set(authority.toString(), null);
+            }
+        }
+    }
+    /**
+     * Enforce that a UserStats will exist for the given authorityPublicKey,
+     * reading one from the blockchain if necessary.
+     * @param authorityPublicKey
+     * @returns
+     */
+    async mustGet(authorityPublicKey) {
+        if (!this.has(authorityPublicKey)) {
+            await this.addReferrerInfo(authorityPublicKey);
+        }
+        return this.get(authorityPublicKey);
+    }
+    values() {
+        return this.referrerMap.values();
+    }
+    size() {
+        return this.referrerMap.size;
+    }
+    async sync() {
+        if (this.fetchPromise) {
+            return this.fetchPromise;
+        }
+        this.fetchPromise = new Promise((resolver) => {
+            this.fetchPromiseResolver = resolver;
+        });
+        try {
+            if (this.parallelSync) {
+                await Promise.all([
+                    this.syncAll(),
+                    this.syncReferrer((0, memcmp_1.getUserStatsIsReferredFilter)()),
+                    this.syncReferrer((0, memcmp_1.getUserStatsIsReferredOrReferrerFilter)()),
+                ]);
+            }
+            else {
+                await this.syncAll();
+                await this.syncReferrer((0, memcmp_1.getUserStatsIsReferredFilter)());
+                await this.syncReferrer((0, memcmp_1.getUserStatsIsReferredOrReferrerFilter)());
+            }
+        }
+        catch (e) {
+            console.error('error in referrerMap.sync', e);
+        }
+        finally {
+            this.fetchPromiseResolver();
+            this.fetchPromise = undefined;
+        }
+    }
+    async syncAll() {
+        const rpcRequestArgs = [
+            this.driftClient.program.programId.toBase58(),
+            {
+                commitment: this.driftClient.opts.commitment,
+                filters: [(0, memcmp_1.getUserStatsFilter)()],
+                encoding: 'base64',
+                dataSlice: {
+                    offset: 0,
+                    length: 0,
+                },
+                withContext: true,
+            },
+        ];
+        const rpcJSONResponse = 
+        // @ts-ignore
+        await this.driftClient.connection._rpcRequest('getProgramAccounts', rpcRequestArgs);
+        const rpcResponseAndContext = rpcJSONResponse.result;
+        for (const account of rpcResponseAndContext.value) {
+            // only add if it isn't already in the map
+            // so that if syncReferrer already set it, we dont overwrite
+            if (!this.has(account.pubkey)) {
+                this.addReferrerInfo(account.pubkey, null);
+            }
+        }
+    }
+    async syncReferrer(referrerFilter) {
+        const rpcRequestArgs = [
+            this.driftClient.program.programId.toBase58(),
+            {
+                commitment: this.driftClient.opts.commitment,
+                filters: [(0, memcmp_1.getUserStatsFilter)(), referrerFilter],
+                encoding: 'base64',
+                dataSlice: {
+                    offset: 0,
+                    length: 72,
+                },
+                withContext: true,
+            },
+        ];
+        const rpcJSONResponse = 
+        // @ts-ignore
+        await this.driftClient.connection._rpcRequest('getProgramAccounts', rpcRequestArgs);
+        const rpcResponseAndContext = rpcJSONResponse.result;
+        const batchSize = 1000;
+        for (let i = 0; i < rpcResponseAndContext.value.length; i += batchSize) {
+            const batch = rpcResponseAndContext.value.slice(i, i + batchSize);
+            await Promise.all(batch.map(async (programAccount) => {
+                // @ts-ignore
+                const buffer = Buffer.from(programAccount.account.data[0], programAccount.account.data[1]);
+                const authority = bytes_1.bs58.encode(buffer.subarray(8, 40));
+                const referrer = bytes_1.bs58.encode(buffer.subarray(40, 72));
+                const referrerKey = new web3_js_1.PublicKey(referrer);
+                this.addReferrerInfo(authority, referrer === DEFAULT_PUBLIC_KEY
+                    ? null
+                    : {
+                        referrer: (0, pda_1.getUserAccountPublicKeySync)(this.driftClient.program.programId, referrerKey, 0),
+                        referrerStats: (0, pda_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, referrerKey),
+                    });
+            }));
+            await new Promise((resolve) => setTimeout(resolve, 0));
+        }
+    }
+    async unsubscribe() {
+        this.referrerMap.clear();
+    }
+}
+exports.ReferrerMap = ReferrerMap;

package/lib/node/driftClient.js

@@ -3880,6 +3880,9 @@ class DriftClient {
         const spotMarketAccount = this.getSpotMarketAccount(marketIndex);
         const isSolMarket = spotMarketAccount.mint.equals(spotMarkets_1.WRAPPED_SOL_MINT);
         const createWSOLTokenAccount = isSolMarket && collateralAccountPublicKey.equals(this.wallet.publicKey);
+        // create associated token account because it may not exist
+        const associatedTokenAccountPublicKey = (0, spl_token_1.getAssociatedTokenAddressSync)(spotMarketAccount.mint, this.wallet.publicKey, true);
+        addIfStakeIxs.push(await (0, spl_token_1.createAssociatedTokenAccountIdempotentInstruction)(this.wallet.publicKey, associatedTokenAccountPublicKey, this.wallet.publicKey, spotMarketAccount.mint));
         let tokenAccount;
         if (!(await this.checkIfAccountExists(this.getUserStatsAccountPublicKey()))) {
             addIfStakeIxs.push(await this.getInitializeUserStatsIx());

package/lib/node/index.d.ts

@@ -106,6 +106,7 @@ export * from './dlob/DLOBApiClient';
 export * from './dlob/types';
 export * from './dlob/orderBookLevels';
 export * from './userMap/userMap';
+export * from './userMap/referrerMap';
 export * from './userMap/userStatsMap';
 export * from './userMap/userMapConfig';
 export * from './math/bankruptcy';

package/lib/node/index.js

@@ -129,6 +129,7 @@ __exportStar(require("./dlob/DLOBApiClient"), exports);
 __exportStar(require("./dlob/types"), exports);
 __exportStar(require("./dlob/orderBookLevels"), exports);
 __exportStar(require("./userMap/userMap"), exports);
+__exportStar(require("./userMap/referrerMap"), exports);
 __exportStar(require("./userMap/userStatsMap"), exports);
 __exportStar(require("./userMap/userMapConfig"), exports);
 __exportStar(require("./math/bankruptcy"), exports);

package/lib/node/memcmp.d.ts

@@ -5,3 +5,6 @@ export declare function getUserWithOrderFilter(): MemcmpFilter;
 export declare function getUserWithAuctionFilter(): MemcmpFilter;
 export declare function getUserThatHasBeenLP(): MemcmpFilter;
 export declare function getUserWithName(name: string): MemcmpFilter;
+export declare function getUserStatsFilter(): MemcmpFilter;
+export declare function getUserStatsIsReferredFilter(): MemcmpFilter;
+export declare function getUserStatsIsReferredOrReferrerFilter(): MemcmpFilter;

package/lib/node/memcmp.js

@@ -3,7 +3,7 @@ var __importDefault = (this && this.__importDefault) || function (mod) {
     return (mod && mod.__esModule) ? mod : { "default": mod };
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getUserWithName = exports.getUserThatHasBeenLP = exports.getUserWithAuctionFilter = exports.getUserWithOrderFilter = exports.getNonIdleUserFilter = exports.getUserFilter = void 0;
+exports.getUserStatsIsReferredOrReferrerFilter = exports.getUserStatsIsReferredFilter = exports.getUserStatsFilter = exports.getUserWithName = exports.getUserThatHasBeenLP = exports.getUserWithAuctionFilter = exports.getUserWithOrderFilter = exports.getNonIdleUserFilter = exports.getUserFilter = void 0;
 const bs58_1 = __importDefault(require("bs58"));
 const anchor_1 = require("@coral-xyz/anchor");
 const userName_1 = require("./userName");
@@ -61,3 +61,30 @@ function getUserWithName(name) {
     };
 }
 exports.getUserWithName = getUserWithName;
+function getUserStatsFilter() {
+    return {
+        memcmp: {
+            offset: 0,
+            bytes: bs58_1.default.encode(anchor_1.BorshAccountsCoder.accountDiscriminator('UserStats')),
+        },
+    };
+}
+exports.getUserStatsFilter = getUserStatsFilter;
+function getUserStatsIsReferredFilter() {
+    return {
+        memcmp: {
+            offset: 188,
+            bytes: bs58_1.default.encode(Buffer.from(Uint8Array.from([2]))),
+        },
+    };
+}
+exports.getUserStatsIsReferredFilter = getUserStatsIsReferredFilter;
+function getUserStatsIsReferredOrReferrerFilter() {
+    return {
+        memcmp: {
+            offset: 188,
+            bytes: bs58_1.default.encode(Buffer.from(Uint8Array.from([3]))),
+        },
+    };
+}
+exports.getUserStatsIsReferredOrReferrerFilter = getUserStatsIsReferredOrReferrerFilter;

package/lib/node/userMap/referrerMap.d.ts

@@ -0,0 +1,45 @@
+import { MemcmpFilter } from '@solana/web3.js';
+import { BulkAccountLoader } from '../accounts/bulkAccountLoader';
+import { DriftClient } from '../driftClient';
+import { ReferrerInfo } from '../types';
+export declare class ReferrerMap {
+    /**
+     * map from authority pubkey to ReferrerInfo.
+     * - if a user has not been entered into the map, the value is undefined
+     * - if a user has no referrer, the value is null
+     * - if a user has a referrer, the value is a ReferrerInfo object
+     */
+    private referrerMap;
+    private driftClient;
+    private bulkAccountLoader;
+    private parallelSync;
+    private fetchPromise?;
+    private fetchPromiseResolver;
+    /**
+     * Creates a new UserStatsMap instance.
+     *
+     * @param {DriftClient} driftClient - The DriftClient instance.
+     * @param {BulkAccountLoader} [bulkAccountLoader] - If not provided, a new BulkAccountLoader with polling disabled will be created.
+     */
+    constructor(driftClient: DriftClient, bulkAccountLoader?: BulkAccountLoader, parallelSync?: boolean);
+    /**
+     * Subscribe to all UserStats accounts.
+     */
+    subscribe(): Promise<void>;
+    has(authorityPublicKey: string): boolean;
+    get(authorityPublicKey: string): ReferrerInfo | undefined | null;
+    addReferrerInfo(authority: string, referrerInfo?: ReferrerInfo | null): Promise<void>;
+    /**
+     * Enforce that a UserStats will exist for the given authorityPublicKey,
+     * reading one from the blockchain if necessary.
+     * @param authorityPublicKey
+     * @returns
+     */
+    mustGet(authorityPublicKey: string): Promise<ReferrerInfo | null | undefined>;
+    values(): IterableIterator<ReferrerInfo | null>;
+    size(): number;
+    sync(): Promise<void>;
+    syncAll(): Promise<void>;
+    syncReferrer(referrerFilter: MemcmpFilter): Promise<void>;
+    unsubscribe(): Promise<void>;
+}

package/lib/node/userMap/referrerMap.js

@@ -0,0 +1,186 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ReferrerMap = void 0;
+const web3_js_1 = require("@solana/web3.js");
+const bulkAccountLoader_1 = require("../accounts/bulkAccountLoader");
+const userStats_1 = require("../userStats");
+const pda_1 = require("../addresses/pda");
+const memcmp_1 = require("../memcmp");
+const bytes_1 = require("@coral-xyz/anchor/dist/cjs/utils/bytes");
+const DEFAULT_PUBLIC_KEY = web3_js_1.PublicKey.default.toBase58();
+class ReferrerMap {
+    /**
+     * Creates a new UserStatsMap instance.
+     *
+     * @param {DriftClient} driftClient - The DriftClient instance.
+     * @param {BulkAccountLoader} [bulkAccountLoader] - If not provided, a new BulkAccountLoader with polling disabled will be created.
+     */
+    constructor(driftClient, bulkAccountLoader, parallelSync) {
+        /**
+         * map from authority pubkey to ReferrerInfo.
+         * - if a user has not been entered into the map, the value is undefined
+         * - if a user has no referrer, the value is null
+         * - if a user has a referrer, the value is a ReferrerInfo object
+         */
+        this.referrerMap = new Map();
+        this.driftClient = driftClient;
+        if (!bulkAccountLoader) {
+            bulkAccountLoader = new bulkAccountLoader_1.BulkAccountLoader(driftClient.connection, driftClient.opts.commitment, 0);
+        }
+        this.bulkAccountLoader = bulkAccountLoader;
+        this.parallelSync = parallelSync !== undefined ? parallelSync : true;
+    }
+    /**
+     * Subscribe to all UserStats accounts.
+     */
+    async subscribe() {
+        if (this.size() > 0) {
+            return;
+        }
+        await this.driftClient.subscribe();
+        await this.sync();
+    }
+    has(authorityPublicKey) {
+        return this.referrerMap.has(authorityPublicKey);
+    }
+    get(authorityPublicKey) {
+        return this.referrerMap.get(authorityPublicKey);
+    }
+    async addReferrerInfo(authority, referrerInfo) {
+        if (referrerInfo || referrerInfo === null) {
+            this.referrerMap.set(authority, referrerInfo);
+        }
+        else if (referrerInfo === undefined) {
+            const userStat = new userStats_1.UserStats({
+                driftClient: this.driftClient,
+                userStatsAccountPublicKey: (0, pda_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, new web3_js_1.PublicKey(authority)),
+                accountSubscription: {
+                    type: 'polling',
+                    accountLoader: this.bulkAccountLoader,
+                },
+            });
+            await userStat.fetchAccounts();
+            const newReferrerInfo = userStat.getReferrerInfo();
+            if (newReferrerInfo) {
+                this.referrerMap.set(authority.toString(), newReferrerInfo);
+            }
+            else {
+                this.referrerMap.set(authority.toString(), null);
+            }
+        }
+    }
+    /**
+     * Enforce that a UserStats will exist for the given authorityPublicKey,
+     * reading one from the blockchain if necessary.
+     * @param authorityPublicKey
+     * @returns
+     */
+    async mustGet(authorityPublicKey) {
+        if (!this.has(authorityPublicKey)) {
+            await this.addReferrerInfo(authorityPublicKey);
+        }
+        return this.get(authorityPublicKey);
+    }
+    values() {
+        return this.referrerMap.values();
+    }
+    size() {
+        return this.referrerMap.size;
+    }
+    async sync() {
+        if (this.fetchPromise) {
+            return this.fetchPromise;
+        }
+        this.fetchPromise = new Promise((resolver) => {
+            this.fetchPromiseResolver = resolver;
+        });
+        try {
+            if (this.parallelSync) {
+                await Promise.all([
+                    this.syncAll(),
+                    this.syncReferrer((0, memcmp_1.getUserStatsIsReferredFilter)()),
+                    this.syncReferrer((0, memcmp_1.getUserStatsIsReferredOrReferrerFilter)()),
+                ]);
+            }
+            else {
+                await this.syncAll();
+                await this.syncReferrer((0, memcmp_1.getUserStatsIsReferredFilter)());
+                await this.syncReferrer((0, memcmp_1.getUserStatsIsReferredOrReferrerFilter)());
+            }
+        }
+        catch (e) {
+            console.error('error in referrerMap.sync', e);
+        }
+        finally {
+            this.fetchPromiseResolver();
+            this.fetchPromise = undefined;
+        }
+    }
+    async syncAll() {
+        const rpcRequestArgs = [
+            this.driftClient.program.programId.toBase58(),
+            {
+                commitment: this.driftClient.opts.commitment,
+                filters: [(0, memcmp_1.getUserStatsFilter)()],
+                encoding: 'base64',
+                dataSlice: {
+                    offset: 0,
+                    length: 0,
+                },
+                withContext: true,
+            },
+        ];
+        const rpcJSONResponse = 
+        // @ts-ignore
+        await this.driftClient.connection._rpcRequest('getProgramAccounts', rpcRequestArgs);
+        const rpcResponseAndContext = rpcJSONResponse.result;
+        for (const account of rpcResponseAndContext.value) {
+            // only add if it isn't already in the map
+            // so that if syncReferrer already set it, we dont overwrite
+            if (!this.has(account.pubkey)) {
+                this.addReferrerInfo(account.pubkey, null);
+            }
+        }
+    }
+    async syncReferrer(referrerFilter) {
+        const rpcRequestArgs = [
+            this.driftClient.program.programId.toBase58(),
+            {
+                commitment: this.driftClient.opts.commitment,
+                filters: [(0, memcmp_1.getUserStatsFilter)(), referrerFilter],
+                encoding: 'base64',
+                dataSlice: {
+                    offset: 0,
+                    length: 72,
+                },
+                withContext: true,
+            },
+        ];
+        const rpcJSONResponse = 
+        // @ts-ignore
+        await this.driftClient.connection._rpcRequest('getProgramAccounts', rpcRequestArgs);
+        const rpcResponseAndContext = rpcJSONResponse.result;
+        const batchSize = 1000;
+        for (let i = 0; i < rpcResponseAndContext.value.length; i += batchSize) {
+            const batch = rpcResponseAndContext.value.slice(i, i + batchSize);
+            await Promise.all(batch.map(async (programAccount) => {
+                // @ts-ignore
+                const buffer = Buffer.from(programAccount.account.data[0], programAccount.account.data[1]);
+                const authority = bytes_1.bs58.encode(buffer.subarray(8, 40));
+                const referrer = bytes_1.bs58.encode(buffer.subarray(40, 72));
+                const referrerKey = new web3_js_1.PublicKey(referrer);
+                this.addReferrerInfo(authority, referrer === DEFAULT_PUBLIC_KEY
+                    ? null
+                    : {
+                        referrer: (0, pda_1.getUserAccountPublicKeySync)(this.driftClient.program.programId, referrerKey, 0),
+                        referrerStats: (0, pda_1.getUserStatsAccountPublicKey)(this.driftClient.program.programId, referrerKey),
+                    });
+            }));
+            await new Promise((resolve) => setTimeout(resolve, 0));
+        }
+    }
+    async unsubscribe() {
+        this.referrerMap.clear();
+    }
+}
+exports.ReferrerMap = ReferrerMap;

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.98.0-beta.11",
+	"version": "2.98.0-beta.13",
 	"main": "lib/node/index.js",
 	"types": "lib/node/index.d.ts",
 	"browser": "./lib/browser/index.js",

package/src/driftClient.ts

@@ -11,11 +11,13 @@ import bs58 from 'bs58';
 import {
 	ASSOCIATED_TOKEN_PROGRAM_ID,
 	createAssociatedTokenAccountInstruction,
+	createAssociatedTokenAccountIdempotentInstruction,
 	createCloseAccountInstruction,
 	createInitializeAccountInstruction,
 	getAssociatedTokenAddress,
 	TOKEN_2022_PROGRAM_ID,
 	TOKEN_PROGRAM_ID,
+	getAssociatedTokenAddressSync,
 } from '@solana/spl-token';
 import {
 	DriftClientMetricsEvents,
@@ -7407,6 +7409,22 @@ export class DriftClient {
 		const createWSOLTokenAccount =
 			isSolMarket && collateralAccountPublicKey.equals(this.wallet.publicKey);
 
+		// create associated token account because it may not exist
+		const associatedTokenAccountPublicKey = getAssociatedTokenAddressSync(
+			spotMarketAccount.mint,
+			this.wallet.publicKey,
+			true
+		);
+
+		addIfStakeIxs.push(
+			await createAssociatedTokenAccountIdempotentInstruction(
+				this.wallet.publicKey,
+				associatedTokenAccountPublicKey,
+				this.wallet.publicKey,
+				spotMarketAccount.mint
+			)
+		);
+
 		let tokenAccount;
 
 		if (

package/src/index.ts

@@ -107,6 +107,7 @@ export * from './dlob/DLOBApiClient';
 export * from './dlob/types';
 export * from './dlob/orderBookLevels';
 export * from './userMap/userMap';
+export * from './userMap/referrerMap';
 export * from './userMap/userStatsMap';
 export * from './userMap/userMapConfig';
 export * from './math/bankruptcy';

package/src/memcmp.ts

@@ -56,3 +56,30 @@ export function getUserWithName(name: string): MemcmpFilter {
 		},
 	};
 }
+
+export function getUserStatsFilter(): MemcmpFilter {
+	return {
+		memcmp: {
+			offset: 0,
+			bytes: bs58.encode(BorshAccountsCoder.accountDiscriminator('UserStats')),
+		},
+	};
+}
+
+export function getUserStatsIsReferredFilter(): MemcmpFilter {
+	return {
+		memcmp: {
+			offset: 188,
+			bytes: bs58.encode(Buffer.from(Uint8Array.from([2]))),
+		},
+	};
+}
+
+export function getUserStatsIsReferredOrReferrerFilter(): MemcmpFilter {
+	return {
+		memcmp: {
+			offset: 188,
+			bytes: bs58.encode(Buffer.from(Uint8Array.from([3]))),
+		},
+	};
+}

package/src/userMap/referrerMap.ts

@@ -0,0 +1,273 @@
+import {
+	MemcmpFilter,
+	PublicKey,
+	RpcResponseAndContext,
+} from '@solana/web3.js';
+import { BulkAccountLoader } from '../accounts/bulkAccountLoader';
+import { DriftClient } from '../driftClient';
+import { ReferrerInfo } from '../types';
+import { UserStats } from '../userStats';
+import {
+	getUserAccountPublicKeySync,
+	getUserStatsAccountPublicKey,
+} from '../addresses/pda';
+import {
+	getUserStatsFilter,
+	getUserStatsIsReferredFilter,
+	getUserStatsIsReferredOrReferrerFilter,
+} from '../memcmp';
+import { bs58 } from '@coral-xyz/anchor/dist/cjs/utils/bytes';
+
+const DEFAULT_PUBLIC_KEY = PublicKey.default.toBase58();
+
+export class ReferrerMap {
+	/**
+	 * map from authority pubkey to ReferrerInfo.
+	 * - if a user has not been entered into the map, the value is undefined
+	 * - if a user has no referrer, the value is null
+	 * - if a user has a referrer, the value is a ReferrerInfo object
+	 */
+	private referrerMap = new Map<string, ReferrerInfo | null>();
+	private driftClient: DriftClient;
+	private bulkAccountLoader: BulkAccountLoader;
+	private parallelSync: boolean;
+
+	private fetchPromise?: Promise<void>;
+	private fetchPromiseResolver: () => void;
+
+	/**
+	 * Creates a new UserStatsMap instance.
+	 *
+	 * @param {DriftClient} driftClient - The DriftClient instance.
+	 * @param {BulkAccountLoader} [bulkAccountLoader] - If not provided, a new BulkAccountLoader with polling disabled will be created.
+	 */
+	constructor(
+		driftClient: DriftClient,
+		bulkAccountLoader?: BulkAccountLoader,
+		parallelSync?: boolean
+	) {
+		this.driftClient = driftClient;
+		if (!bulkAccountLoader) {
+			bulkAccountLoader = new BulkAccountLoader(
+				driftClient.connection,
+				driftClient.opts.commitment,
+				0
+			);
+		}
+		this.bulkAccountLoader = bulkAccountLoader;
+		this.parallelSync = parallelSync !== undefined ? parallelSync : true;
+	}
+
+	/**
+	 * Subscribe to all UserStats accounts.
+	 */
+	public async subscribe() {
+		if (this.size() > 0) {
+			return;
+		}
+
+		await this.driftClient.subscribe();
+		await this.sync();
+	}
+
+	public has(authorityPublicKey: string): boolean {
+		return this.referrerMap.has(authorityPublicKey);
+	}
+
+	public get(authorityPublicKey: string): ReferrerInfo | undefined | null {
+		return this.referrerMap.get(authorityPublicKey);
+	}
+
+	public async addReferrerInfo(
+		authority: string,
+		referrerInfo?: ReferrerInfo | null
+	) {
+		if (referrerInfo || referrerInfo === null) {
+			this.referrerMap.set(authority, referrerInfo);
+		} else if (referrerInfo === undefined) {
+			const userStat = new UserStats({
+				driftClient: this.driftClient,
+				userStatsAccountPublicKey: getUserStatsAccountPublicKey(
+					this.driftClient.program.programId,
+					new PublicKey(authority)
+				),
+				accountSubscription: {
+					type: 'polling',
+					accountLoader: this.bulkAccountLoader,
+				},
+			});
+			await userStat.fetchAccounts();
+
+			const newReferrerInfo = userStat.getReferrerInfo();
+
+			if (newReferrerInfo) {
+				this.referrerMap.set(authority.toString(), newReferrerInfo);
+			} else {
+				this.referrerMap.set(authority.toString(), null);
+			}
+		}
+	}
+
+	/**
+	 * Enforce that a UserStats will exist for the given authorityPublicKey,
+	 * reading one from the blockchain if necessary.
+	 * @param authorityPublicKey
+	 * @returns
+	 */
+	public async mustGet(
+		authorityPublicKey: string
+	): Promise<ReferrerInfo | null | undefined> {
+		if (!this.has(authorityPublicKey)) {
+			await this.addReferrerInfo(authorityPublicKey);
+		}
+		return this.get(authorityPublicKey);
+	}
+
+	public values(): IterableIterator<ReferrerInfo | null> {
+		return this.referrerMap.values();
+	}
+
+	public size(): number {
+		return this.referrerMap.size;
+	}
+
+	public async sync(): Promise<void> {
+		if (this.fetchPromise) {
+			return this.fetchPromise;
+		}
+
+		this.fetchPromise = new Promise((resolver) => {
+			this.fetchPromiseResolver = resolver;
+		});
+
+		try {
+			if (this.parallelSync) {
+				await Promise.all([
+					this.syncAll(),
+					this.syncReferrer(getUserStatsIsReferredFilter()),
+					this.syncReferrer(getUserStatsIsReferredOrReferrerFilter()),
+				]);
+			} else {
+				await this.syncAll();
+				await this.syncReferrer(getUserStatsIsReferredFilter());
+				await this.syncReferrer(getUserStatsIsReferredOrReferrerFilter());
+			}
+		} catch (e) {
+			console.error('error in referrerMap.sync', e);
+		} finally {
+			this.fetchPromiseResolver();
+			this.fetchPromise = undefined;
+		}
+	}
+
+	public async syncAll(): Promise<void> {
+		const rpcRequestArgs = [
+			this.driftClient.program.programId.toBase58(),
+			{
+				commitment: this.driftClient.opts.commitment,
+				filters: [getUserStatsFilter()],
+				encoding: 'base64',
+				dataSlice: {
+					offset: 0,
+					length: 0,
+				},
+				withContext: true,
+			},
+		];
+
+		const rpcJSONResponse: any =
+			// @ts-ignore
+			await this.driftClient.connection._rpcRequest(
+				'getProgramAccounts',
+				rpcRequestArgs
+			);
+
+		const rpcResponseAndContext: RpcResponseAndContext<
+			Array<{
+				pubkey: string;
+				account: {
+					data: [string, string];
+				};
+			}>
+		> = rpcJSONResponse.result;
+
+		for (const account of rpcResponseAndContext.value) {
+			// only add if it isn't already in the map
+			// so that if syncReferrer already set it, we dont overwrite
+			if (!this.has(account.pubkey)) {
+				this.addReferrerInfo(account.pubkey, null);
+			}
+		}
+	}
+
+	async syncReferrer(referrerFilter: MemcmpFilter): Promise<void> {
+		const rpcRequestArgs = [
+			this.driftClient.program.programId.toBase58(),
+			{
+				commitment: this.driftClient.opts.commitment,
+				filters: [getUserStatsFilter(), referrerFilter],
+				encoding: 'base64',
+				dataSlice: {
+					offset: 0,
+					length: 72,
+				},
+				withContext: true,
+			},
+		];
+
+		const rpcJSONResponse: any =
+			// @ts-ignore
+			await this.driftClient.connection._rpcRequest(
+				'getProgramAccounts',
+				rpcRequestArgs
+			);
+
+		const rpcResponseAndContext: RpcResponseAndContext<
+			Array<{
+				pubkey: string;
+				account: {
+					data: [string, string];
+				};
+			}>
+		> = rpcJSONResponse.result;
+
+		const batchSize = 1000;
+		for (let i = 0; i < rpcResponseAndContext.value.length; i += batchSize) {
+			const batch = rpcResponseAndContext.value.slice(i, i + batchSize);
+			await Promise.all(
+				batch.map(async (programAccount) => {
+					// @ts-ignore
+					const buffer = Buffer.from(
+						programAccount.account.data[0],
+						programAccount.account.data[1]
+					);
+					const authority = bs58.encode(buffer.subarray(8, 40));
+					const referrer = bs58.encode(buffer.subarray(40, 72));
+
+					const referrerKey = new PublicKey(referrer);
+					this.addReferrerInfo(
+						authority,
+						referrer === DEFAULT_PUBLIC_KEY
+							? null
+							: {
+									referrer: getUserAccountPublicKeySync(
+										this.driftClient.program.programId,
+										referrerKey,
+										0
+									),
+									referrerStats: getUserStatsAccountPublicKey(
+										this.driftClient.program.programId,
+										referrerKey
+									),
+							  }
+					);
+				})
+			);
+			await new Promise((resolve) => setTimeout(resolve, 0));
+		}
+	}
+
+	public async unsubscribe() {
+		this.referrerMap.clear();
+	}
+}