@drift-labs/sdk 0.2.0-master.31 → 0.2.0-master.32

package/lib/dlob/DLOB.d.ts

@@ -43,6 +43,7 @@ export declare class DLOB {
      * @param silent set to true to prevent logging on inserts and removals
      */
     constructor(perpMarkets: PerpMarketAccount[], spotMarkets: SpotMarketAccount[], silent?: boolean);
+    clear(): void;
     /**
      * initializes a new DLOB instance
      *

package/lib/dlob/DLOB.js

@@ -68,6 +68,29 @@ class DLOB {
             });
         }
     }
+    clear() {
+        for (const marketType of this.openOrders.keys()) {
+            this.openOrders.get(marketType).clear();
+        }
+        this.openOrders.clear();
+        for (const marketType of this.orderLists.keys()) {
+            for (const marketIndex of this.orderLists.get(marketType).keys()) {
+                const marketNodeLists = this.orderLists
+                    .get(marketType)
+                    .get(marketIndex);
+                for (const side of Object.keys(marketNodeLists)) {
+                    for (const orderType of Object.keys(marketNodeLists[side])) {
+                        marketNodeLists[side][orderType].clear();
+                    }
+                }
+            }
+        }
+        this.orderLists.clear();
+        for (const marketType of this.marketIndexToAccount.keys()) {
+            this.marketIndexToAccount.get(marketType).clear();
+        }
+        this.marketIndexToAccount.clear();
+    }
     /**
      * initializes a new DLOB instance
      *
@@ -309,6 +332,9 @@ class DLOB {
         if (marketTypeStr === 'perp' && vAsk) {
             generatorList.push((0, NodeList_1.getVammNodeGenerator)(vAsk));
         }
+        if (generatorList.length === 0) {
+            throw new Error('No ask generators found');
+        }
         const askGenerators = generatorList.map((generator) => {
             return {
                 next: generator.next(),
@@ -358,6 +384,9 @@ class DLOB {
         if (marketTypeStr === 'perp' && vBid) {
             generatorList.push((0, NodeList_1.getVammNodeGenerator)(vBid));
         }
+        if (generatorList.length === 0) {
+            throw new Error('No bid generators found');
+        }
         const bidGenerators = generatorList.map((generator) => {
             return {
                 next: generator.next(),

package/lib/dlob/NodeList.d.ts

@@ -14,6 +14,7 @@ export declare class NodeList<NodeType extends keyof DLOBNodeMap> implements DLO
     length: number;
     nodeMap: Map<string, DLOBNodeMap[NodeType]>;
     constructor(nodeType: NodeType, sortDirection: SortDirection);
+    clear(): void;
     insert(order: Order, marketType: MarketTypeStr, market: PerpMarketAccount | SpotMarketAccount, userAccount: PublicKey): void;
     prependNode(currentNode: DLOBNodeMap[NodeType], newNode: DLOBNodeMap[NodeType]): boolean;
     update(order: Order, userAccount: PublicKey): void;

package/lib/dlob/NodeList.js

@@ -14,6 +14,11 @@ class NodeList {
         this.length = 0;
         this.nodeMap = new Map();
     }
+    clear() {
+        this.head = undefined;
+        this.length = 0;
+        this.nodeMap.clear();
+    }
     insert(order, marketType, market, userAccount) {
         if ((0, __1.isVariant)(order.status, 'init')) {
             return;

package/lib/types.d.ts

@@ -6,19 +6,19 @@ export declare class ExchangeStatus {
         active: {};
     };
     static readonly FUNDINGPAUSED: {
-        fundingpaused: {};
+        fundingPaused: {};
     };
     static readonly AMMPAUSED: {
-        ammpaused: {};
+        ammPaused: {};
     };
     static readonly FILLPAUSED: {
-        fillpaused: {};
+        fillPaused: {};
     };
     static readonly LIQPAUSED: {
-        liqpaused: {};
+        liqPaused: {};
     };
     static readonly WITHDRAWPAUSED: {
-        withdrawpaused: {};
+        withdrawPaused: {};
     };
     static readonly PAUSED: {
         paused: {};
@@ -32,19 +32,19 @@ export declare class MarketStatus {
         active: {};
     };
     static readonly FUNDINGPAUSED: {
-        fundingpaused: {};
+        fundingPaused: {};
     };
     static readonly AMMPAUSED: {
-        ammpaused: {};
+        ammPaused: {};
     };
     static readonly FILLPAUSED: {
-        fillpaused: {};
+        fillPaused: {};
     };
     static readonly WITHDRAWPAUSED: {
-        withdrawpaused: {};
+        withdrawPaused: {};
     };
     static readonly REDUCEONLY: {
-        reduceonly: {};
+        reduceOnly: {};
     };
     static readonly SETTLEMENT: {
         settlement: {};

package/lib/types.js

@@ -7,22 +7,22 @@ class ExchangeStatus {
 }
 exports.ExchangeStatus = ExchangeStatus;
 ExchangeStatus.ACTIVE = { active: {} };
-ExchangeStatus.FUNDINGPAUSED = { fundingpaused: {} };
-ExchangeStatus.AMMPAUSED = { ammpaused: {} };
-ExchangeStatus.FILLPAUSED = { fillpaused: {} };
-ExchangeStatus.LIQPAUSED = { liqpaused: {} };
-ExchangeStatus.WITHDRAWPAUSED = { withdrawpaused: {} };
+ExchangeStatus.FUNDINGPAUSED = { fundingPaused: {} };
+ExchangeStatus.AMMPAUSED = { ammPaused: {} };
+ExchangeStatus.FILLPAUSED = { fillPaused: {} };
+ExchangeStatus.LIQPAUSED = { liqPaused: {} };
+ExchangeStatus.WITHDRAWPAUSED = { withdrawPaused: {} };
 ExchangeStatus.PAUSED = { paused: {} };
 class MarketStatus {
 }
 exports.MarketStatus = MarketStatus;
 MarketStatus.INITIALIZED = { initialized: {} };
 MarketStatus.ACTIVE = { active: {} };
-MarketStatus.FUNDINGPAUSED = { fundingpaused: {} };
-MarketStatus.AMMPAUSED = { ammpaused: {} };
-MarketStatus.FILLPAUSED = { fillpaused: {} };
-MarketStatus.WITHDRAWPAUSED = { withdrawpaused: {} };
-MarketStatus.REDUCEONLY = { reduceonly: {} };
+MarketStatus.FUNDINGPAUSED = { fundingPaused: {} };
+MarketStatus.AMMPAUSED = { ammPaused: {} };
+MarketStatus.FILLPAUSED = { fillPaused: {} };
+MarketStatus.WITHDRAWPAUSED = { withdrawPaused: {} };
+MarketStatus.REDUCEONLY = { reduceOnly: {} };
 MarketStatus.SETTLEMENT = { settlement: {} };
 MarketStatus.DELISTED = { delisted: {} };
 class ContractType {

package/package.json

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

package/src/dlob/DLOB.ts

@@ -135,6 +135,32 @@ export class DLOB {
 		}
 	}
 
+	public clear() {
+		for (const marketType of this.openOrders.keys()) {
+			this.openOrders.get(marketType).clear();
+		}
+		this.openOrders.clear();
+
+		for (const marketType of this.orderLists.keys()) {
+			for (const marketIndex of this.orderLists.get(marketType).keys()) {
+				const marketNodeLists = this.orderLists
+					.get(marketType)
+					.get(marketIndex);
+				for (const side of Object.keys(marketNodeLists)) {
+					for (const orderType of Object.keys(marketNodeLists[side])) {
+						marketNodeLists[side][orderType].clear();
+					}
+				}
+			}
+		}
+		this.orderLists.clear();
+
+		for (const marketType of this.marketIndexToAccount.keys()) {
+			this.marketIndexToAccount.get(marketType).clear();
+		}
+		this.marketIndexToAccount.clear();
+	}
+
 	/**
 	 * initializes a new DLOB instance
 	 *
@@ -516,6 +542,9 @@ export class DLOB {
 		if (marketTypeStr === 'perp' && vAsk) {
 			generatorList.push(getVammNodeGenerator(vAsk));
 		}
+		if (generatorList.length === 0) {
+			throw new Error('No ask generators found');
+		}
 
 		const askGenerators = generatorList.map((generator) => {
 			return {
@@ -588,6 +617,10 @@ export class DLOB {
 		if (marketTypeStr === 'perp' && vBid) {
 			generatorList.push(getVammNodeGenerator(vBid));
 		}
+		if (generatorList.length === 0) {
+			throw new Error('No bid generators found');
+		}
+
 		const bidGenerators = generatorList.map((generator) => {
 			return {
 				next: generator.next(),

package/src/dlob/NodeList.ts

@@ -34,6 +34,12 @@ export class NodeList<NodeType extends keyof DLOBNodeMap>
 		private sortDirection: SortDirection
 	) {}
 
+	public clear() {
+		this.head = undefined;
+		this.length = 0;
+		this.nodeMap.clear();
+	}
+
 	public insert(
 		order: Order,
 		marketType: MarketTypeStr,

package/src/types.ts

@@ -5,22 +5,22 @@ import { BN, ZERO } from '.';
 
 export class ExchangeStatus {
 	static readonly ACTIVE = { active: {} };
-	static readonly FUNDINGPAUSED = { fundingpaused: {} };
-	static readonly AMMPAUSED = { ammpaused: {} };
-	static readonly FILLPAUSED = { fillpaused: {} };
-	static readonly LIQPAUSED = { liqpaused: {} };
-	static readonly WITHDRAWPAUSED = { withdrawpaused: {} };
+	static readonly FUNDINGPAUSED = { fundingPaused: {} };
+	static readonly AMMPAUSED = { ammPaused: {} };
+	static readonly FILLPAUSED = { fillPaused: {} };
+	static readonly LIQPAUSED = { liqPaused: {} };
+	static readonly WITHDRAWPAUSED = { withdrawPaused: {} };
 	static readonly PAUSED = { paused: {} };
 }
 
 export class MarketStatus {
 	static readonly INITIALIZED = { initialized: {} };
 	static readonly ACTIVE = { active: {} };
-	static readonly FUNDINGPAUSED = { fundingpaused: {} };
-	static readonly AMMPAUSED = { ammpaused: {} };
-	static readonly FILLPAUSED = { fillpaused: {} };
-	static readonly WITHDRAWPAUSED = { withdrawpaused: {} };
-	static readonly REDUCEONLY = { reduceonly: {} };
+	static readonly FUNDINGPAUSED = { fundingPaused: {} };
+	static readonly AMMPAUSED = { ammPaused: {} };
+	static readonly FILLPAUSED = { fillPaused: {} };
+	static readonly WITHDRAWPAUSED = { withdrawPaused: {} };
+	static readonly REDUCEONLY = { reduceOnly: {} };
 	static readonly SETTLEMENT = { settlement: {} };
 	static readonly DELISTED = { delisted: {} };
 }

package/tests/bn/test.ts

@@ -1,4 +1,4 @@
-import { BN } from '@project-serum/anchor';
+import { BN } from '../../src';
 import { expect } from 'chai';
 import { BigNum } from '../../src/factory/bigNum';
 import {

package/tests/dlob/test.ts

@@ -284,6 +284,85 @@ describe('DLOB Tests', () => {
 			expect(foundBids).to.equal(0);
 		}
 	});
+
+	it('Can clear DLOB', () => {
+		const vAsk = new BN(15);
+		const vBid = new BN(10);
+		const dlob = new DLOB(mockPerpMarkets, mockSpotMarkets, false);
+		const marketIndex = 0;
+
+		const slot = 12;
+		const oracle = {
+			price: vBid.add(vAsk).div(new BN(2)),
+			slot: new BN(slot),
+			confidence: new BN(1),
+			hasSufficientNumberOfDataPoints: true,
+		};
+
+		insertOrderToDLOB(
+			dlob,
+			Keypair.generate().publicKey,
+			OrderType.LIMIT,
+			MarketType.PERP,
+			0, // orderId
+			marketIndex,
+			new BN(9), // price
+			BASE_PRECISION, // quantity
+			PositionDirection.LONG,
+			vBid,
+			vAsk
+		);
+		insertOrderToDLOB(
+			dlob,
+			Keypair.generate().publicKey,
+			OrderType.LIMIT,
+			MarketType.PERP,
+			1, // orderId
+			marketIndex,
+			new BN(8), // price
+			BASE_PRECISION, // quantity
+			PositionDirection.LONG,
+			vBid,
+			vAsk
+		);
+		insertOrderToDLOB(
+			dlob,
+			Keypair.generate().publicKey,
+			OrderType.LIMIT,
+			MarketType.PERP,
+			2, // orderId
+			marketIndex,
+			new BN(8), // price
+			BASE_PRECISION, // quantity
+			PositionDirection.LONG,
+			vBid,
+			vAsk
+		);
+
+		const bids = dlob.getBids(
+			marketIndex,
+			undefined,
+			0,
+			MarketType.PERP,
+			oracle
+		);
+		let b = 0;
+		for (const _bid of bids) {
+			b++;
+		}
+		expect(b).to.equal(3);
+
+		dlob.clear();
+		let thrown = false;
+		try {
+			const bids1 = dlob.getBids(marketIndex, vBid, 0, MarketType.PERP, oracle);
+			bids1.next();
+		} catch (e) {
+			console.error(e);
+			thrown = true;
+		}
+		expect(thrown, 'should throw after clearing').to.equal(true);
+	});
 });
 
 describe('DLOB Perp Tests', () => {
@@ -393,7 +472,7 @@ describe('DLOB Perp Tests', () => {
 		const bids = dlob.getBids(marketIndex, vBid, slot, MarketType.PERP, oracle);
 
 		console.log('The Book Bids:');
-		const gotBids = [];
+		const gotBids: Array<DLOBNode> = [];
 		let countBids = 0;
 		for (const bid of bids) {
 			gotBids.push(bid);

package/src/assert/assert.js

@@ -1,9 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.assert = void 0;
-function assert(condition, error) {
-    if (!condition) {
-        throw new Error(error || 'Unspecified AssertionError');
-    }
-}
-exports.assert = assert;

package/src/events/eventList.js

@@ -1,77 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.EventList = void 0;
-class Node {
-    constructor(event, next, prev) {
-        this.event = event;
-        this.next = next;
-        this.prev = prev;
-    }
-}
-class EventList {
-    constructor(eventType, maxSize, sortFn, orderDirection) {
-        this.eventType = eventType;
-        this.maxSize = maxSize;
-        this.sortFn = sortFn;
-        this.orderDirection = orderDirection;
-        this.size = 0;
-    }
-    insert(event) {
-        this.size++;
-        const newNode = new Node(event);
-        if (this.head === undefined) {
-            this.head = this.tail = newNode;
-            return;
-        }
-        if (this.sortFn(this.head.event, newNode.event) ===
-            (this.orderDirection === 'asc' ? 'less than' : 'greater than')) {
-            this.head.prev = newNode;
-            newNode.next = this.head;
-            this.head = newNode;
-        }
-        else {
-            let currentNode = this.head;
-            while (currentNode.next !== undefined &&
-                this.sortFn(currentNode.next.event, newNode.event) !==
-                    (this.orderDirection === 'asc' ? 'less than' : 'greater than')) {
-                currentNode = currentNode.next;
-            }
-            newNode.next = currentNode.next;
-            if (currentNode.next !== undefined) {
-                newNode.next.prev = newNode;
-            }
-            currentNode.next = newNode;
-            newNode.prev = currentNode;
-        }
-        if (this.size > this.maxSize) {
-            this.detach();
-        }
-    }
-    detach() {
-        const node = this.tail;
-        if (node.prev !== undefined) {
-            node.prev.next = node.next;
-        }
-        else {
-            this.head = node.next;
-        }
-        if (node.next !== undefined) {
-            node.next.prev = node.prev;
-        }
-        else {
-            this.tail = node.prev;
-        }
-        this.size--;
-    }
-    toArray() {
-        return Array.from(this);
-    }
-    *[Symbol.iterator]() {
-        let node = this.head;
-        while (node) {
-            yield node.event;
-            node = node.next;
-        }
-    }
-}
-exports.EventList = EventList;

package/src/examples/makeTradeExample.js

@@ -1,157 +0,0 @@
-'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.getTokenAddress = void 0;
-const anchor_1 = require('@project-serum/anchor');
-const __1 = require('..');
-const spl_token_1 = require('@solana/spl-token');
-const web3_js_1 = require('@solana/web3.js');
-const __2 = require('..');
-const banks_1 = require('../constants/spotMarkets');
-const getTokenAddress = (mintAddress, userPubKey) => {
-	return spl_token_1.Token.getAssociatedTokenAddress(
-		new web3_js_1.PublicKey(`ATokenGPvbdGVxr1b2hvZbsiqW5xWH25efTNsLJA8knL`),
-		spl_token_1.TOKEN_PROGRAM_ID,
-		new web3_js_1.PublicKey(mintAddress),
-		new web3_js_1.PublicKey(userPubKey)
-	);
-};
-exports.getTokenAddress = getTokenAddress;
-const main = () =>
-	__awaiter(void 0, void 0, void 0, function* () {
-		// Initialize Drift SDK
-		const sdkConfig = __2.initialize({ env: 'devnet' });
-		// Set up the Wallet and Provider
-		const privateKey = process.env.BOT_PRIVATE_KEY; // stored as an array string
-		const keypair = web3_js_1.Keypair.fromSecretKey(
-			Uint8Array.from(JSON.parse(privateKey))
-		);
-		const wallet = new __1.Wallet(keypair);
-		// Set up the Connection
-		const rpcAddress = process.env.RPC_ADDRESS; // can use: https://api.devnet.solana.com for devnet; https://api.mainnet-beta.solana.com for mainnet;
-		const connection = new web3_js_1.Connection(rpcAddress);
-		// Set up the Provider
-		const provider = new anchor_1.AnchorProvider(
-			connection,
-			wallet,
-			anchor_1.AnchorProvider.defaultOptions()
-		);
-		// Check SOL Balance
-		const lamportsBalance = yield connection.getBalance(wallet.publicKey);
-		console.log('SOL balance:', lamportsBalance / Math.pow(10, 9));
-		// Misc. other things to set up
-		const usdcTokenAddress = yield exports.getTokenAddress(
-			sdkConfig.USDC_MINT_ADDRESS,
-			wallet.publicKey.toString()
-		);
-		// Set up the Drift Clearing House
-		const clearingHousePublicKey = new web3_js_1.PublicKey(
-			sdkConfig.CLEARING_HOUSE_PROGRAM_ID
-		);
-		const clearingHouse = new __2.ClearingHouse({
-			connection,
-			wallet: provider.wallet,
-			programID: clearingHousePublicKey,
-		});
-		yield clearingHouse.subscribe();
-		// Set up Clearing House user client
-		const user = new __2.ClearingHouseUser({
-			clearingHouse,
-			userAccountPublicKey: yield clearingHouse.getUserAccountPublicKey(),
-		});
-		//// Check if clearing house account exists for the current wallet
-		const userAccountExists = yield user.exists();
-		if (!userAccountExists) {
-			//// Create a Clearing House account by Depositing some USDC ($10,000 in this case)
-			const depositAmount = new anchor_1.BN(10000).mul(__2.QUOTE_PRECISION);
-			yield clearingHouse.initializeUserAccountAndDepositCollateral(
-				depositAmount,
-				yield exports.getTokenAddress(
-					usdcTokenAddress.toString(),
-					wallet.publicKey.toString()
-				),
-				banks_1.SpotMarkets['devnet'][0].marketIndex
-			);
-		}
-		yield user.subscribe();
-		// Get current price
-		const solMarketInfo = sdkConfig.PERP_MARKETS.find(
-			(market) => market.baseAssetSymbol === 'SOL'
-		);
-		const currentMarketPrice = __2.calculateMarkPrice(
-			clearingHouse.getMarketAccount(solMarketInfo.marketIndex),
-			undefined
-		);
-		const formattedPrice = __2.convertToNumber(
-			currentMarketPrice,
-			__2.PRICE_PRECISION
-		);
-		console.log(`Current Market Price is $${formattedPrice}`);
-		// Estimate the slippage for a $5000 LONG trade
-		const solMarketAccount = clearingHouse.getMarketAccount(
-			solMarketInfo.marketIndex
-		);
-		const longAmount = new anchor_1.BN(5000).mul(__2.QUOTE_PRECISION);
-		const slippage = __2.convertToNumber(
-			__2.calculateTradeSlippage(
-				__2.PositionDirection.LONG,
-				longAmount,
-				solMarketAccount,
-				'quote',
-				undefined
-			)[0],
-			__2.PRICE_PRECISION
-		);
-		console.log(
-			`Slippage for a $5000 LONG on the SOL market would be $${slippage}`
-		);
-		// Make a $5000 LONG trade
-		yield clearingHouse.openPosition(
-			__2.PositionDirection.LONG,
-			longAmount,
-			solMarketInfo.marketIndex
-		);
-		console.log(`LONGED $5000 SOL`);
-		// Reduce the position by $2000
-		const reduceAmount = new anchor_1.BN(2000).mul(__2.QUOTE_PRECISION);
-		yield clearingHouse.openPosition(
-			__2.PositionDirection.SHORT,
-			reduceAmount,
-			solMarketInfo.marketIndex
-		);
-		// Close the rest of the position
-		yield clearingHouse.closePosition(solMarketInfo.marketIndex);
-	});
-main();

package/src/token/index.js

@@ -1,38 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.parseTokenAccount = void 0;
-const spl_token_1 = require("@solana/spl-token");
-const web3_js_1 = require("@solana/web3.js");
-function parseTokenAccount(data) {
-    const accountInfo = spl_token_1.AccountLayout.decode(data);
-    accountInfo.mint = new web3_js_1.PublicKey(accountInfo.mint);
-    accountInfo.owner = new web3_js_1.PublicKey(accountInfo.owner);
-    accountInfo.amount = spl_token_1.u64.fromBuffer(accountInfo.amount);
-    if (accountInfo.delegateOption === 0) {
-        accountInfo.delegate = null;
-        // eslint-disable-next-line new-cap
-        accountInfo.delegatedAmount = new spl_token_1.u64(0);
-    }
-    else {
-        accountInfo.delegate = new web3_js_1.PublicKey(accountInfo.delegate);
-        accountInfo.delegatedAmount = spl_token_1.u64.fromBuffer(accountInfo.delegatedAmount);
-    }
-    accountInfo.isInitialized = accountInfo.state !== 0;
-    accountInfo.isFrozen = accountInfo.state === 2;
-    if (accountInfo.isNativeOption === 1) {
-        accountInfo.rentExemptReserve = spl_token_1.u64.fromBuffer(accountInfo.isNative);
-        accountInfo.isNative = true;
-    }
-    else {
-        accountInfo.rentExemptReserve = null;
-        accountInfo.isNative = false;
-    }
-    if (accountInfo.closeAuthorityOption === 0) {
-        accountInfo.closeAuthority = null;
-    }
-    else {
-        accountInfo.closeAuthority = new web3_js_1.PublicKey(accountInfo.closeAuthority);
-    }
-    return accountInfo;
-}
-exports.parseTokenAccount = parseTokenAccount;

package/src/tx/types.js

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/src/tx/utils.js

@@ -1,17 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.wrapInTx = void 0;
-const web3_js_1 = require("@solana/web3.js");
-const COMPUTE_UNITS_DEFAULT = 200000;
-function wrapInTx(instruction, computeUnits = 600000 // TODO, requires less code change
-) {
-    const tx = new web3_js_1.Transaction();
-    if (computeUnits != COMPUTE_UNITS_DEFAULT) {
-        tx.add(web3_js_1.ComputeBudgetProgram.requestUnits({
-            units: computeUnits,
-            additionalFee: 0,
-        }));
-    }
-    return tx.add(instruction);
-}
-exports.wrapInTx = wrapInTx;

package/src/util/computeUnits.js

@@ -1,27 +0,0 @@
-"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

@@ -1,9 +0,0 @@
-"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

@@ -1,14 +0,0 @@
-"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

@@ -1,27 +0,0 @@
-"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;