@drift-labs/sdk 2.0.12 → 2.0.14

package/lib/userMap/userMap.d.ts

@@ -1,4 +1,4 @@
-import { User, DriftClient, OrderRecord, UserSubscriptionConfig } from '..';
+import { User, DriftClient, OrderRecord, UserSubscriptionConfig, WrappedEvent } from '..';
 import { PublicKey } from '@solana/web3.js';
 export interface UserMapInterface {
     fetchAllUsers(): Promise<void>;
@@ -37,6 +37,7 @@ export declare class UserMap implements UserMapInterface {
      */
     getUserAuthority(key: string): PublicKey | undefined;
     updateWithOrderRecord(record: OrderRecord): Promise<void>;
+    updateWithEventRecord(record: WrappedEvent<any>): Promise<void>;
     values(): IterableIterator<User>;
     size(): number;
 }

package/lib/userMap/userMap.js

@@ -80,6 +80,46 @@ class UserMap {
             await this.addPubkey(record.user);
         }
     }
+    async updateWithEventRecord(record) {
+        if (record.eventType === 'DepositRecord') {
+            const depositRecord = record;
+            await this.mustGet(depositRecord.user.toString());
+        }
+        else if (record.eventType === 'FundingPaymentRecord') {
+            const fundingPaymentRecord = record;
+            await this.mustGet(fundingPaymentRecord.user.toString());
+        }
+        else if (record.eventType === 'LiquidationRecord') {
+            const liqRecord = record;
+            await this.mustGet(liqRecord.user.toString());
+            await this.mustGet(liqRecord.liquidator.toString());
+        }
+        else if (record.eventType === 'OrderRecord') {
+            const orderRecord = record;
+            await this.updateWithOrderRecord(orderRecord);
+        }
+        else if (record.eventType === 'OrderActionRecord') {
+            const actionRecord = record;
+            if (actionRecord.taker) {
+                await this.mustGet(actionRecord.taker.toString());
+            }
+            if (actionRecord.maker) {
+                await this.mustGet(actionRecord.maker.toString());
+            }
+        }
+        else if (record.eventType === 'SettlePnlRecord') {
+            const settlePnlRecord = record;
+            await this.mustGet(settlePnlRecord.user.toString());
+        }
+        else if (record.eventType === 'NewUserRecord') {
+            const newUserRecord = record;
+            await this.mustGet(newUserRecord.user.toString());
+        }
+        else if (record.eventType === 'LPRecord') {
+            const lpRecord = record;
+            await this.mustGet(lpRecord.user.toString());
+        }
+    }
     values() {
         return this.userMap.values();
     }

package/lib/userMap/userStatsMap.d.ts

@@ -1,4 +1,4 @@
-import { DriftClient, OrderRecord, UserStats, UserStatsSubscriptionConfig } from '..';
+import { DriftClient, OrderRecord, UserStats, UserStatsSubscriptionConfig, WrappedEvent } from '..';
 import { PublicKey } from '@solana/web3.js';
 import { UserMap } from './userMap';
 export declare class UserStatsMap {
@@ -12,6 +12,7 @@ export declare class UserStatsMap {
     fetchAllUserStats(): Promise<void>;
     addUserStat(authority: PublicKey): Promise<void>;
     updateWithOrderRecord(record: OrderRecord, userMap: UserMap): Promise<void>;
+    updateWithEventRecord(record: WrappedEvent<any>, userMap?: UserMap): Promise<void>;
     has(authorityPublicKey: string): boolean;
     get(authorityPublicKey: string): UserStats;
     mustGet(authorityPublicKey: string): Promise<UserStats>;

package/lib/userMap/userStatsMap.js

@@ -49,6 +49,71 @@ class UserStatsMap {
             this.addUserStat(user.getUserAccount().authority);
         }
     }
+    async updateWithEventRecord(record, userMap) {
+        if (record.eventType === 'DepositRecord') {
+            const depositRecord = record;
+            await this.mustGet(depositRecord.userAuthority.toString());
+        }
+        else if (record.eventType === 'FundingPaymentRecord') {
+            const fundingPaymentRecord = record;
+            await this.mustGet(fundingPaymentRecord.userAuthority.toString());
+        }
+        else if (record.eventType === 'LiquidationRecord') {
+            if (!userMap) {
+                return;
+            }
+            const liqRecord = record;
+            const user = await userMap.mustGet(liqRecord.user.toString());
+            await this.mustGet(user.getUserAccount().authority.toString());
+            const liquidatorUser = await userMap.mustGet(liqRecord.liquidator.toString());
+            await this.mustGet(liquidatorUser.getUserAccount().authority.toString());
+        }
+        else if (record.eventType === 'OrderRecord') {
+            if (!userMap) {
+                return;
+            }
+            const orderRecord = record;
+            await userMap.updateWithOrderRecord(orderRecord);
+        }
+        else if (record.eventType === 'OrderActionRecord') {
+            if (!userMap) {
+                return;
+            }
+            const actionRecord = record;
+            if (actionRecord.taker) {
+                const taker = await userMap.mustGet(actionRecord.taker.toString());
+                await this.mustGet(taker.getUserAccount().authority.toString());
+            }
+            if (actionRecord.maker) {
+                const maker = await userMap.mustGet(actionRecord.maker.toString());
+                await this.mustGet(maker.getUserAccount().authority.toString());
+            }
+        }
+        else if (record.eventType === 'SettlePnlRecord') {
+            if (!userMap) {
+                return;
+            }
+            const settlePnlRecord = record;
+            const user = await userMap.mustGet(settlePnlRecord.user.toString());
+            await this.mustGet(user.getUserAccount().authority.toString());
+        }
+        else if (record.eventType === 'NewUserRecord') {
+            const newUserRecord = record;
+            await this.mustGet(newUserRecord.userAuthority.toString());
+        }
+        else if (record.eventType === 'LPRecord') {
+            if (!userMap) {
+                return;
+            }
+            const lpRecord = record;
+            const user = await userMap.mustGet(lpRecord.user.toString());
+            await this.mustGet(user.getUserAccount().authority.toString());
+        }
+        else if (record.eventType === 'InsuranceFundStakeRecord') {
+            const ifStakeRecord = record;
+            await this.mustGet(ifStakeRecord.userAuthority.toString());
+        }
+    }
     has(authorityPublicKey) {
         return this.userStatsMap.has(authorityPublicKey);
     }

package/package.json

@@ -1,6 +1,6 @@
 {
 	"name": "@drift-labs/sdk",
-	"version": "2.0.12",
+	"version": "2.0.14",
 	"main": "lib/index.js",
 	"types": "lib/index.d.ts",
 	"author": "crispheaney",