@drift-labs/sdk 2.0.11 → 2.0.12

package/src/userMap/userStatsMap.ts

@@ -6,15 +6,6 @@ import {
 	UserStats,
 	UserStatsSubscriptionConfig,
 	bulkPollingUserStatsSubscribe,
-	WrappedEvent,
-	DepositRecord,
-	FundingPaymentRecord,
-	LiquidationRecord,
-	OrderActionRecord,
-	SettlePnlRecord,
-	NewUserRecord,
-	LPRecord,
-	InsuranceFundStakeRecord,
 } from '..';
 import { ProgramAccount } from '@project-serum/anchor';
 import { PublicKey } from '@solana/web3.js';
@@ -96,73 +87,6 @@ export class UserStatsMap {
 		}
 	}
 
-	public async updateWithEventRecord(
-		record: WrappedEvent<any>,
-		userMap?: UserMap
-	) {
-		if (record.eventType === 'DepositRecord') {
-			const depositRecord = record as DepositRecord;
-			await this.mustGet(depositRecord.userAuthority.toString());
-		} else if (record.eventType === 'FundingPaymentRecord') {
-			const fundingPaymentRecord = record as FundingPaymentRecord;
-			await this.mustGet(fundingPaymentRecord.userAuthority.toString());
-		} else if (record.eventType === 'LiquidationRecord') {
-			if (!userMap) {
-				return;
-			}
-
-			const liqRecord = record as LiquidationRecord;
-
-			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 as OrderRecord;
-			await userMap.updateWithOrderRecord(orderRecord);
-		} else if (record.eventType === 'OrderActionRecord') {
-			if (!userMap) {
-				return;
-			}
-			const actionRecord = record as OrderActionRecord;
-
-			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 as SettlePnlRecord;
-			const user = await userMap.mustGet(settlePnlRecord.user.toString());
-			await this.mustGet(user.getUserAccount().authority.toString());
-		} else if (record.eventType === 'NewUserRecord') {
-			const newUserRecord = record as NewUserRecord;
-			await this.mustGet(newUserRecord.userAuthority.toString());
-		} else if (record.eventType === 'LPRecord') {
-			if (!userMap) {
-				return;
-			}
-			const lpRecord = record as LPRecord;
-			const user = await userMap.mustGet(lpRecord.user.toString());
-			await this.mustGet(user.getUserAccount().authority.toString());
-		} else if (record.eventType === 'InsuranceFundStakeRecord') {
-			const ifStakeRecord = record as InsuranceFundStakeRecord;
-			await this.mustGet(ifStakeRecord.userAuthority.toString());
-		}
-	}
-
 	public has(authorityPublicKey: string): boolean {
 		return this.userStatsMap.has(authorityPublicKey);
 	}

package/src/util/computeUnits.js

@@ -0,0 +1,27 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.findComputeUnitConsumption = void 0;
+function findComputeUnitConsumption(programId, connection, txSignature, commitment = 'confirmed') {
+    return __awaiter(this, void 0, void 0, function* () {
+        const tx = yield connection.getTransaction(txSignature, { commitment });
+        const computeUnits = [];
+        const regex = new RegExp(`Program ${programId.toString()} consumed ([0-9]{0,6}) of ([0-9]{0,7}) compute units`);
+        tx.meta.logMessages.forEach((logMessage) => {
+            const match = logMessage.match(regex);
+            if (match && match[1]) {
+                computeUnits.push(match[1]);
+            }
+        });
+        return computeUnits;
+    });
+}
+exports.findComputeUnitConsumption = findComputeUnitConsumption;

package/src/util/getTokenAddress.js

@@ -0,0 +1,9 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getTokenAddress = void 0;
+const spl_token_1 = require("@solana/spl-token");
+const web3_js_1 = require("@solana/web3.js");
+const getTokenAddress = (mintAddress, userPubKey) => {
+    return spl_token_1.Token.getAssociatedTokenAddress(spl_token_1.ASSOCIATED_TOKEN_PROGRAM_ID, spl_token_1.TOKEN_PROGRAM_ID, new web3_js_1.PublicKey(mintAddress), new web3_js_1.PublicKey(userPubKey));
+};
+exports.getTokenAddress = getTokenAddress;

package/src/util/promiseTimeout.js

@@ -0,0 +1,14 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.promiseTimeout = void 0;
+function promiseTimeout(promise, timeoutMs) {
+    let timeoutId;
+    const timeoutPromise = new Promise((resolve) => {
+        timeoutId = setTimeout(() => resolve(null), timeoutMs);
+    });
+    return Promise.race([promise, timeoutPromise]).then((result) => {
+        clearTimeout(timeoutId);
+        return result;
+    });
+}
+exports.promiseTimeout = promiseTimeout;

package/src/util/tps.js

@@ -0,0 +1,27 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.estimateTps = void 0;
+function estimateTps(programId, connection, failed) {
+    return __awaiter(this, void 0, void 0, function* () {
+        let signatures = yield connection.getSignaturesForAddress(programId, undefined, 'finalized');
+        if (failed) {
+            signatures = signatures.filter((signature) => signature.err);
+        }
+        const numberOfSignatures = signatures.length;
+        if (numberOfSignatures === 0) {
+            return 0;
+        }
+        return (numberOfSignatures /
+            (signatures[0].blockTime - signatures[numberOfSignatures - 1].blockTime));
+    });
+}
+exports.estimateTps = estimateTps;