@drift-labs/sdk 0.2.0-master.21 → 0.2.0-master.24

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (88) hide show
  1. package/lib/accounts/types.d.ts +1 -0
  2. package/lib/addresses/pda.d.ts +3 -0
  3. package/lib/addresses/pda.js +23 -1
  4. package/lib/admin.d.ts +3 -0
  5. package/lib/admin.js +31 -0
  6. package/lib/clearingHouse.d.ts +9 -2
  7. package/lib/clearingHouse.js +132 -4
  8. package/lib/clearingHouseUser.js +5 -5
  9. package/lib/config.js +1 -1
  10. package/lib/constants/banks.d.ts +2 -0
  11. package/lib/constants/banks.js +9 -0
  12. package/lib/constants/numericConstants.d.ts +2 -0
  13. package/lib/constants/numericConstants.js +4 -1
  14. package/lib/events/sort.js +7 -10
  15. package/lib/events/types.d.ts +2 -1
  16. package/lib/events/types.js +1 -0
  17. package/lib/idl/clearing_house.json +1008 -147
  18. package/lib/index.d.ts +1 -0
  19. package/lib/index.js +1 -0
  20. package/lib/math/amm.js +2 -2
  21. package/lib/math/insurance.d.ts +4 -0
  22. package/lib/math/insurance.js +27 -0
  23. package/lib/math/margin.d.ts +1 -1
  24. package/lib/math/margin.js +5 -7
  25. package/lib/math/position.js +1 -1
  26. package/lib/types.d.ts +56 -20
  27. package/lib/types.js +0 -3
  28. package/package.json +1 -1
  29. package/src/addresses/marketAddresses.js +26 -0
  30. package/src/addresses/pda.ts +47 -0
  31. package/src/admin.ts +65 -0
  32. package/src/assert/assert.js +9 -0
  33. package/src/clearingHouse.ts +217 -9
  34. package/src/clearingHouseUser.ts +10 -6
  35. package/src/config.ts +1 -1
  36. package/src/constants/banks.js +42 -0
  37. package/src/constants/banks.ts +16 -0
  38. package/src/constants/markets.js +42 -0
  39. package/src/constants/numericConstants.js +41 -0
  40. package/src/constants/numericConstants.ts +4 -0
  41. package/src/events/eventList.js +77 -0
  42. package/src/events/sort.ts +10 -14
  43. package/src/events/txEventCache.js +71 -0
  44. package/src/events/types.ts +3 -0
  45. package/src/examples/makeTradeExample.js +80 -0
  46. package/src/factory/bigNum.js +390 -0
  47. package/src/factory/oracleClient.js +20 -0
  48. package/src/idl/clearing_house.json +1008 -147
  49. package/src/index.ts +1 -0
  50. package/src/math/amm.ts +2 -2
  51. package/src/math/auction.js +42 -0
  52. package/src/math/conversion.js +11 -0
  53. package/src/math/funding.js +248 -0
  54. package/src/math/insurance.ts +35 -0
  55. package/src/math/margin.ts +3 -10
  56. package/src/math/oracles.js +26 -0
  57. package/src/math/position.ts +2 -2
  58. package/src/math/repeg.js +128 -0
  59. package/src/math/state.js +15 -0
  60. package/src/math/trade.js +253 -0
  61. package/src/math/utils.js +26 -0
  62. package/src/math/utils.js.map +1 -0
  63. package/src/oracles/oracleClientCache.js +19 -0
  64. package/src/oracles/pythClient.js +46 -0
  65. package/src/oracles/quoteAssetOracleClient.js +32 -0
  66. package/src/oracles/switchboardClient.js +69 -0
  67. package/src/oracles/types.js +2 -0
  68. package/src/orderParams.js +20 -0
  69. package/src/slot/SlotSubscriber.js +39 -0
  70. package/src/token/index.js +38 -0
  71. package/src/tokenFaucet.js +189 -0
  72. package/src/tx/types.js +2 -0
  73. package/src/tx/utils.js +17 -0
  74. package/src/types.ts +64 -20
  75. package/src/userName.js +20 -0
  76. package/src/util/computeUnits.js +27 -0
  77. package/src/util/getTokenAddress.js +9 -0
  78. package/src/util/promiseTimeout.js +14 -0
  79. package/src/util/tps.js +27 -0
  80. package/src/wallet.js +35 -0
  81. package/my-script/.env +0 -7
  82. package/my-script/getUserStats.ts +0 -124
  83. package/my-script/multiConnections.ts +0 -137
  84. package/my-script/test-regex.ts +0 -11
  85. package/my-script/utils.ts +0 -52
  86. package/my-script/ww18NdhuLSQPCrHSx7V68eZJpe2y311heWeXJfSmP3Q.json +0 -1
  87. package/my-script/ww2z7N9TG1PLLUQGQF2VKzCFaPtQ5FBhRfeEAuy6c5C.json +0 -1
  88. package/my-script/ww3StJtTubhwssqAhvSSAc5ifCgKjzmF8hz7Gt2DmSa.json +0 -1
@@ -0,0 +1,71 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.TxEventCache = void 0;
4
+ class Node {
5
+ constructor(key, value, next, prev) {
6
+ this.key = key;
7
+ this.value = value;
8
+ this.next = next;
9
+ this.prev = prev;
10
+ }
11
+ }
12
+ // lru cache
13
+ class TxEventCache {
14
+ constructor(maxTx = 1024) {
15
+ this.maxTx = maxTx;
16
+ this.size = 0;
17
+ this.cacheMap = {};
18
+ }
19
+ add(key, events) {
20
+ const existingNode = this.cacheMap[key];
21
+ if (existingNode) {
22
+ this.detach(existingNode);
23
+ this.size--;
24
+ }
25
+ else if (this.size === this.maxTx) {
26
+ delete this.cacheMap[this.tail.key];
27
+ this.detach(this.tail);
28
+ this.size--;
29
+ }
30
+ // Write to head of LinkedList
31
+ if (!this.head) {
32
+ this.head = this.tail = new Node(key, events);
33
+ }
34
+ else {
35
+ const node = new Node(key, events, this.head);
36
+ this.head.prev = node;
37
+ this.head = node;
38
+ }
39
+ // update cacheMap with LinkedList key and Node reference
40
+ this.cacheMap[key] = this.head;
41
+ this.size++;
42
+ }
43
+ has(key) {
44
+ return this.cacheMap.hasOwnProperty(key);
45
+ }
46
+ get(key) {
47
+ var _a;
48
+ return (_a = this.cacheMap[key]) === null || _a === void 0 ? void 0 : _a.value;
49
+ }
50
+ detach(node) {
51
+ if (node.prev !== undefined) {
52
+ node.prev.next = node.next;
53
+ }
54
+ else {
55
+ this.head = node.next;
56
+ }
57
+ if (node.next !== undefined) {
58
+ node.next.prev = node.prev;
59
+ }
60
+ else {
61
+ this.tail = node.prev;
62
+ }
63
+ }
64
+ clear() {
65
+ this.head = undefined;
66
+ this.tail = undefined;
67
+ this.size = 0;
68
+ this.cacheMap = {};
69
+ }
70
+ }
71
+ exports.TxEventCache = TxEventCache;
@@ -5,6 +5,7 @@ import {
5
5
  FundingRateRecord,
6
6
  LiquidationRecord,
7
7
  NewUserRecord,
8
+ OrderActionRecord,
8
9
  OrderRecord,
9
10
  SettlePnlRecord,
10
11
  } from '../index';
@@ -28,6 +29,7 @@ export const DefaultEventSubscriptionOptions: EventSubscriptionOptions = {
28
29
  'FundingPaymentRecord',
29
30
  'LiquidationRecord',
30
31
  'OrderRecord',
32
+ 'OrderActionRecord',
31
33
  'FundingRateRecord',
32
34
  'NewUserRecord',
33
35
  'SettlePnlRecord',
@@ -63,6 +65,7 @@ export type EventMap = {
63
65
  LiquidationRecord: Event<LiquidationRecord>;
64
66
  FundingRateRecord: Event<FundingRateRecord>;
65
67
  OrderRecord: Event<OrderRecord>;
68
+ OrderActionRecord: Event<OrderActionRecord>;
66
69
  SettlePnlRecord: Event<SettlePnlRecord>;
67
70
  NewUserRecord: Event<NewUserRecord>;
68
71
  };
@@ -0,0 +1,80 @@
1
+ "use strict";
2
+ var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
3
+ function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
4
+ return new (P || (P = Promise))(function (resolve, reject) {
5
+ function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
6
+ function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
7
+ function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
8
+ step((generator = generator.apply(thisArg, _arguments || [])).next());
9
+ });
10
+ };
11
+ Object.defineProperty(exports, "__esModule", { value: true });
12
+ exports.getTokenAddress = void 0;
13
+ const anchor_1 = require("@project-serum/anchor");
14
+ const __1 = require("..");
15
+ const spl_token_1 = require("@solana/spl-token");
16
+ const web3_js_1 = require("@solana/web3.js");
17
+ const __2 = require("..");
18
+ const banks_1 = require("../constants/banks");
19
+ const getTokenAddress = (mintAddress, userPubKey) => {
20
+ 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));
21
+ };
22
+ exports.getTokenAddress = getTokenAddress;
23
+ const main = () => __awaiter(void 0, void 0, void 0, function* () {
24
+ // Initialize Drift SDK
25
+ const sdkConfig = __2.initialize({ env: 'devnet' });
26
+ // Set up the Wallet and Provider
27
+ const privateKey = process.env.BOT_PRIVATE_KEY; // stored as an array string
28
+ const keypair = web3_js_1.Keypair.fromSecretKey(Uint8Array.from(JSON.parse(privateKey)));
29
+ const wallet = new __1.Wallet(keypair);
30
+ // Set up the Connection
31
+ const rpcAddress = process.env.RPC_ADDRESS; // can use: https://api.devnet.solana.com for devnet; https://api.mainnet-beta.solana.com for mainnet;
32
+ const connection = new web3_js_1.Connection(rpcAddress);
33
+ // Set up the Provider
34
+ const provider = new anchor_1.AnchorProvider(connection, wallet, anchor_1.AnchorProvider.defaultOptions());
35
+ // Check SOL Balance
36
+ const lamportsBalance = yield connection.getBalance(wallet.publicKey);
37
+ console.log('SOL balance:', lamportsBalance / Math.pow(10, 9));
38
+ // Misc. other things to set up
39
+ const usdcTokenAddress = yield exports.getTokenAddress(sdkConfig.USDC_MINT_ADDRESS, wallet.publicKey.toString());
40
+ // Set up the Drift Clearing House
41
+ const clearingHousePublicKey = new web3_js_1.PublicKey(sdkConfig.CLEARING_HOUSE_PROGRAM_ID);
42
+ const clearingHouse = new __2.ClearingHouse({
43
+ connection,
44
+ wallet: provider.wallet,
45
+ programID: clearingHousePublicKey,
46
+ });
47
+ yield clearingHouse.subscribe();
48
+ // Set up Clearing House user client
49
+ const user = new __2.ClearingHouseUser({
50
+ clearingHouse,
51
+ userAccountPublicKey: yield clearingHouse.getUserAccountPublicKey(),
52
+ });
53
+ //// Check if clearing house account exists for the current wallet
54
+ const userAccountExists = yield user.exists();
55
+ if (!userAccountExists) {
56
+ //// Create a Clearing House account by Depositing some USDC ($10,000 in this case)
57
+ const depositAmount = new anchor_1.BN(10000).mul(__2.QUOTE_PRECISION);
58
+ yield clearingHouse.initializeUserAccountAndDepositCollateral(depositAmount, yield exports.getTokenAddress(usdcTokenAddress.toString(), wallet.publicKey.toString()), banks_1.Banks['devnet'][0].bankIndex);
59
+ }
60
+ yield user.subscribe();
61
+ // Get current price
62
+ const solMarketInfo = sdkConfig.MARKETS.find((market) => market.baseAssetSymbol === 'SOL');
63
+ const currentMarketPrice = __2.calculateMarkPrice(clearingHouse.getMarketAccount(solMarketInfo.marketIndex), undefined);
64
+ const formattedPrice = __2.convertToNumber(currentMarketPrice, __2.MARK_PRICE_PRECISION);
65
+ console.log(`Current Market Price is $${formattedPrice}`);
66
+ // Estimate the slippage for a $5000 LONG trade
67
+ const solMarketAccount = clearingHouse.getMarketAccount(solMarketInfo.marketIndex);
68
+ const longAmount = new anchor_1.BN(5000).mul(__2.QUOTE_PRECISION);
69
+ const slippage = __2.convertToNumber(__2.calculateTradeSlippage(__2.PositionDirection.LONG, longAmount, solMarketAccount, 'quote', undefined)[0], __2.MARK_PRICE_PRECISION);
70
+ console.log(`Slippage for a $5000 LONG on the SOL market would be $${slippage}`);
71
+ // Make a $5000 LONG trade
72
+ yield clearingHouse.openPosition(__2.PositionDirection.LONG, longAmount, solMarketInfo.marketIndex);
73
+ console.log(`LONGED $5000 SOL`);
74
+ // Reduce the position by $2000
75
+ const reduceAmount = new anchor_1.BN(2000).mul(__2.QUOTE_PRECISION);
76
+ yield clearingHouse.openPosition(__2.PositionDirection.SHORT, reduceAmount, solMarketInfo.marketIndex);
77
+ // Close the rest of the position
78
+ yield clearingHouse.closePosition(solMarketInfo.marketIndex);
79
+ });
80
+ main();
@@ -0,0 +1,390 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.BigNum = void 0;
4
+ const anchor_1 = require("@project-serum/anchor");
5
+ const assert_1 = require("../assert/assert");
6
+ const numericConstants_1 = require("./../constants/numericConstants");
7
+ class BigNum {
8
+ constructor(val, precisionVal = new anchor_1.BN(0)) {
9
+ this.toString = (base, length) => this.val.toString(base, length);
10
+ this.val = new anchor_1.BN(val);
11
+ this.precision = new anchor_1.BN(precisionVal);
12
+ }
13
+ bigNumFromParam(bn) {
14
+ return anchor_1.BN.isBN(bn) ? BigNum.from(bn) : bn;
15
+ }
16
+ add(bn) {
17
+ assert_1.assert(bn.precision.eq(this.precision), 'Adding unequal precisions');
18
+ return BigNum.from(this.val.add(bn.val), this.precision);
19
+ }
20
+ sub(bn) {
21
+ assert_1.assert(bn.precision.eq(this.precision), 'Subtracting unequal precisions');
22
+ return BigNum.from(this.val.sub(bn.val), this.precision);
23
+ }
24
+ mul(bn) {
25
+ const mulVal = this.bigNumFromParam(bn);
26
+ return BigNum.from(this.val.mul(mulVal.val), this.precision.add(mulVal.precision));
27
+ }
28
+ /**
29
+ * Multiplies by another big number then scales the result down by the big number's precision so that we're in the same precision space
30
+ * @param bn
31
+ * @returns
32
+ */
33
+ scalarMul(bn) {
34
+ if (anchor_1.BN.isBN(bn))
35
+ return BigNum.from(this.val.mul(bn), this.precision);
36
+ return BigNum.from(this.val.mul(bn.val), this.precision.add(bn.precision)).shift(bn.precision.neg());
37
+ }
38
+ div(bn) {
39
+ if (anchor_1.BN.isBN(bn))
40
+ return BigNum.from(this.val.div(bn), this.precision);
41
+ return BigNum.from(this.val.div(bn.val), this.precision.sub(bn.precision));
42
+ }
43
+ /**
44
+ * Shift precision up or down
45
+ * @param exponent
46
+ * @param skipAdjustingPrecision
47
+ * @returns
48
+ */
49
+ shift(exponent, skipAdjustingPrecision = false) {
50
+ const shiftVal = typeof exponent === 'number' ? new anchor_1.BN(exponent) : exponent;
51
+ return BigNum.from(shiftVal.isNeg()
52
+ ? this.val.div(new anchor_1.BN(10).pow(shiftVal))
53
+ : this.val.mul(new anchor_1.BN(10).pow(shiftVal)), skipAdjustingPrecision ? this.precision : this.precision.add(shiftVal));
54
+ }
55
+ /**
56
+ * Shift to a target precision
57
+ * @param targetPrecision
58
+ * @returns
59
+ */
60
+ shiftTo(targetPrecision) {
61
+ return this.shift(targetPrecision.sub(this.precision));
62
+ }
63
+ /**
64
+ * Scale the number by a fraction
65
+ * @param numerator
66
+ * @param denominator
67
+ * @returns
68
+ */
69
+ scale(numerator, denominator) {
70
+ return this.mul(BigNum.from(new anchor_1.BN(numerator))).div(new anchor_1.BN(denominator));
71
+ }
72
+ toPercentage(denominator, precision) {
73
+ return this.shift(precision)
74
+ .shift(2, true)
75
+ .div(denominator)
76
+ .toPrecision(precision);
77
+ }
78
+ gt(bn, ignorePrecision) {
79
+ const comparisonVal = this.bigNumFromParam(bn);
80
+ if (!ignorePrecision && !comparisonVal.eq(numericConstants_1.ZERO)) {
81
+ assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
82
+ }
83
+ return this.val.gt(comparisonVal.val);
84
+ }
85
+ lt(bn, ignorePrecision) {
86
+ const comparisonVal = this.bigNumFromParam(bn);
87
+ if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
88
+ assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
89
+ }
90
+ return this.val.lt(comparisonVal.val);
91
+ }
92
+ gte(bn, ignorePrecision) {
93
+ const comparisonVal = this.bigNumFromParam(bn);
94
+ if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
95
+ assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
96
+ }
97
+ return this.val.gte(comparisonVal.val);
98
+ }
99
+ lte(bn, ignorePrecision) {
100
+ const comparisonVal = this.bigNumFromParam(bn);
101
+ if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
102
+ assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
103
+ }
104
+ return this.val.lte(comparisonVal.val);
105
+ }
106
+ eq(bn, ignorePrecision) {
107
+ const comparisonVal = this.bigNumFromParam(bn);
108
+ if (!ignorePrecision && !comparisonVal.val.eq(numericConstants_1.ZERO)) {
109
+ assert_1.assert(comparisonVal.precision.eq(this.precision), 'Trying to compare numbers with different precision. Yo can opt to ignore precision using the ignorePrecision parameter');
110
+ }
111
+ return this.val.eq(comparisonVal.val);
112
+ }
113
+ eqZero() {
114
+ return this.val.eq(numericConstants_1.ZERO);
115
+ }
116
+ gtZero() {
117
+ return this.val.gt(numericConstants_1.ZERO);
118
+ }
119
+ ltZero() {
120
+ return this.val.lt(numericConstants_1.ZERO);
121
+ }
122
+ gteZero() {
123
+ return this.val.gte(numericConstants_1.ZERO);
124
+ }
125
+ lteZero() {
126
+ return this.val.lte(numericConstants_1.ZERO);
127
+ }
128
+ abs() {
129
+ return new BigNum(this.val.abs(), this.precision);
130
+ }
131
+ neg() {
132
+ return new BigNum(this.val.neg(), this.precision);
133
+ }
134
+ /**
135
+ * Pretty print the underlying value in human-readable form. Depends on precision being correct for the output string to be correct
136
+ * @returns
137
+ */
138
+ print() {
139
+ assert_1.assert(this.precision.gte(numericConstants_1.ZERO), 'Tried to print a BN with precision lower than zero');
140
+ const plainString = this.toString();
141
+ const precisionNum = this.precision.toNumber();
142
+ // make a string with at least the precisionNum number of zeroes
143
+ let printString = [
144
+ ...Array(this.precision.toNumber()).fill(0),
145
+ ...plainString.split(''),
146
+ ].join('');
147
+ // inject decimal
148
+ printString =
149
+ printString.substring(0, printString.length - precisionNum) +
150
+ BigNum.delim +
151
+ printString.substring(printString.length - precisionNum);
152
+ // remove leading zeroes
153
+ printString = printString.replace(/^0+/, '');
154
+ // add zero if leading delim
155
+ if (this.isNeg()) {
156
+ if (printString[1] === BigNum.delim)
157
+ printString = printString.replace('-.', '-0.');
158
+ }
159
+ else {
160
+ if (printString[0] === BigNum.delim)
161
+ printString = `0${printString}`;
162
+ }
163
+ // remove trailing delim
164
+ if (printString[printString.length - 1] === BigNum.delim)
165
+ printString = printString.slice(0, printString.length - 1);
166
+ return printString;
167
+ }
168
+ prettyPrint(useTradePrecision, precisionOverride) {
169
+ const [leftSide, rightSide] = this.printShort(useTradePrecision, precisionOverride).split(BigNum.delim);
170
+ let formattedLeftSide = leftSide;
171
+ const isNeg = formattedLeftSide.includes('-');
172
+ if (isNeg) {
173
+ formattedLeftSide = formattedLeftSide.replace('-', '');
174
+ }
175
+ let index = formattedLeftSide.length - 3;
176
+ while (index >= 1) {
177
+ const formattedLeftSideArray = formattedLeftSide.split('');
178
+ formattedLeftSideArray.splice(index, 0, BigNum.spacer);
179
+ formattedLeftSide = formattedLeftSideArray.join('');
180
+ index -= 3;
181
+ }
182
+ return `${isNeg ? '-' : ''}${formattedLeftSide}${rightSide ? `${BigNum.delim}${rightSide}` : ''}`;
183
+ }
184
+ /**
185
+ * Print and remove unnecessary trailing zeroes
186
+ * @returns
187
+ */
188
+ printShort(useTradePrecision, precisionOverride) {
189
+ const printVal = precisionOverride
190
+ ? this.toPrecision(precisionOverride)
191
+ : useTradePrecision
192
+ ? this.toTradePrecision()
193
+ : this.print();
194
+ return printVal.replace(/0+$/g, '').replace(/\.$/, '').replace(/,$/, '');
195
+ }
196
+ debug() {
197
+ console.log(`${this.toString()} | ${this.print()} | ${this.precision.toString()}`);
198
+ }
199
+ /**
200
+ * Pretty print with the specified number of decimal places
201
+ * @param fixedPrecision
202
+ * @returns
203
+ */
204
+ toFixed(fixedPrecision) {
205
+ const printString = this.print();
206
+ const [leftSide, rightSide] = printString.split(BigNum.delim);
207
+ const filledRightSide = [
208
+ ...(rightSide !== null && rightSide !== void 0 ? rightSide : '').slice(0, fixedPrecision),
209
+ ...Array(fixedPrecision).fill('0'),
210
+ ]
211
+ .slice(0, fixedPrecision)
212
+ .join('');
213
+ return `${leftSide}${BigNum.delim}${filledRightSide}`;
214
+ }
215
+ /**
216
+ * Pretty print to the specified number of significant figures
217
+ * @param fixedPrecision
218
+ * @returns
219
+ */
220
+ toPrecision(fixedPrecision, trailingZeroes = false) {
221
+ const printString = this.print();
222
+ let precisionPrintString = printString.slice(0, fixedPrecision + 1);
223
+ if (!precisionPrintString.includes(BigNum.delim) ||
224
+ precisionPrintString[precisionPrintString.length - 1] === BigNum.delim) {
225
+ precisionPrintString = printString.slice(0, fixedPrecision);
226
+ }
227
+ const pointsOfPrecision = precisionPrintString.replace(BigNum.delim, '').length;
228
+ if (pointsOfPrecision < fixedPrecision) {
229
+ precisionPrintString = [
230
+ ...precisionPrintString.split(''),
231
+ ...Array(fixedPrecision - pointsOfPrecision).fill('0'),
232
+ ].join('');
233
+ }
234
+ if (!precisionPrintString.includes(BigNum.delim)) {
235
+ const delimFullStringLocation = printString.indexOf(BigNum.delim);
236
+ let skipExponent = false;
237
+ if (delimFullStringLocation === -1) {
238
+ // no decimal, not missing any precision
239
+ skipExponent = true;
240
+ }
241
+ if (precisionPrintString[precisionPrintString.length - 1] === BigNum.delim) {
242
+ // decimal is at end of string, not missing any precision, do nothing
243
+ skipExponent = true;
244
+ }
245
+ if (printString.indexOf(BigNum.delim) === fixedPrecision) {
246
+ // decimal is at end of string, not missing any precision, do nothing
247
+ skipExponent = true;
248
+ }
249
+ if (!skipExponent) {
250
+ const exponent = delimFullStringLocation - fixedPrecision;
251
+ if (trailingZeroes) {
252
+ precisionPrintString = `${precisionPrintString}${Array(exponent)
253
+ .fill('0')
254
+ .join('')}`;
255
+ }
256
+ else {
257
+ precisionPrintString = `${precisionPrintString}e${exponent}`;
258
+ }
259
+ }
260
+ }
261
+ return precisionPrintString;
262
+ }
263
+ toTradePrecision() {
264
+ return this.toPrecision(6, true);
265
+ }
266
+ /**
267
+ * Print dollar formatted value. Defaults to fixed decimals two unless a given precision is given.
268
+ * @param useTradePrecision
269
+ * @param precisionOverride
270
+ * @returns
271
+ */
272
+ toNotional(useTradePrecision, precisionOverride) {
273
+ var _a;
274
+ const prefix = `${this.lt(BigNum.zero()) ? `-` : ``}$`;
275
+ const usingCustomPrecision = true && (useTradePrecision || precisionOverride);
276
+ let val = usingCustomPrecision
277
+ ? this.prettyPrint(useTradePrecision, precisionOverride)
278
+ : BigNum.fromPrint(this.toFixed(2), new anchor_1.BN(2)).prettyPrint();
279
+ // Append two trailing zeroes if not using custom precision
280
+ if (!usingCustomPrecision) {
281
+ const [_, rightSide] = val.split(BigNum.delim);
282
+ const trailingLength = (_a = rightSide === null || rightSide === void 0 ? void 0 : rightSide.length) !== null && _a !== void 0 ? _a : 0;
283
+ if (trailingLength < 2) {
284
+ const numHasDecimals = this.print().includes(BigNum.delim);
285
+ // Handle case where pretty print won't include the decimal point
286
+ if (trailingLength === 0 && numHasDecimals) {
287
+ val = `${val}.00`;
288
+ }
289
+ else {
290
+ val = `${val}${new Array(2 - trailingLength).fill('0').join('')}`;
291
+ }
292
+ }
293
+ }
294
+ return `${prefix}${val.replace('-', '')}`;
295
+ }
296
+ toMillified(precision = 3) {
297
+ const stringVal = this.print();
298
+ const [leftSide] = stringVal.split(BigNum.delim);
299
+ if (!leftSide) {
300
+ return this.shift(new anchor_1.BN(precision)).toPrecision(precision, true);
301
+ }
302
+ if (leftSide.length <= 3) {
303
+ return this.shift(new anchor_1.BN(precision)).toPrecision(precision, true);
304
+ }
305
+ const unitTicks = ['', 'K', 'M', 'G', 'T', 'P', 'E', 'Z', 'Y'];
306
+ const unitNumber = Math.floor((leftSide.length - 1) / 3);
307
+ const unit = unitTicks[unitNumber];
308
+ let leadDigits = leftSide.slice(0, precision);
309
+ if (leadDigits.length < precision) {
310
+ leadDigits = [
311
+ ...leadDigits.split(''),
312
+ ...Array(precision - leadDigits.length).fill('0'),
313
+ ].join('');
314
+ }
315
+ const decimalLocation = leftSide.length - 3 * unitNumber;
316
+ let leadString = '';
317
+ if (decimalLocation >= precision) {
318
+ leadString = `${leadDigits}`;
319
+ }
320
+ else {
321
+ leadString = `${leadDigits.slice(0, decimalLocation)}${BigNum.delim}${leadDigits.slice(decimalLocation)}`;
322
+ }
323
+ return `${leadString}${unit}`;
324
+ }
325
+ toJSON() {
326
+ return {
327
+ val: this.val.toString(),
328
+ precision: this.precision.toString(),
329
+ };
330
+ }
331
+ isNeg() {
332
+ return this.lt(numericConstants_1.ZERO, true);
333
+ }
334
+ isPos() {
335
+ return !this.isNeg();
336
+ }
337
+ /**
338
+ * Get the numerical value of the BigNum. This can break if the BigNum is too large.
339
+ * @returns
340
+ */
341
+ toNum() {
342
+ return parseFloat(this.print());
343
+ }
344
+ static fromJSON(json) {
345
+ return BigNum.from(new anchor_1.BN(json.val), new anchor_1.BN(json.precision));
346
+ }
347
+ /**
348
+ * Create a BigNum instance
349
+ * @param val
350
+ * @param precision
351
+ * @returns
352
+ */
353
+ static from(val = numericConstants_1.ZERO, precision) {
354
+ assert_1.assert(new anchor_1.BN(precision).lt(new anchor_1.BN(100)), 'Tried to create a bignum with precision higher than 10^100');
355
+ return new BigNum(val, precision);
356
+ }
357
+ /**
358
+ * Create a BigNum instance from a printed BigNum
359
+ * @param val
360
+ * @param precisionOverride
361
+ * @returns
362
+ */
363
+ static fromPrint(val, precisionShift) {
364
+ var _a;
365
+ // Handle empty number edge cases
366
+ if (!val)
367
+ return BigNum.from(numericConstants_1.ZERO, precisionShift);
368
+ if (!val.replace(BigNum.delim, ''))
369
+ return BigNum.from(numericConstants_1.ZERO, precisionShift);
370
+ const [leftSide, rightSide] = val.split(BigNum.delim);
371
+ const rawBn = new anchor_1.BN(`${leftSide !== null && leftSide !== void 0 ? leftSide : ''}${rightSide !== null && rightSide !== void 0 ? rightSide : ''}`);
372
+ const rightSideLength = (_a = rightSide === null || rightSide === void 0 ? void 0 : rightSide.length) !== null && _a !== void 0 ? _a : 0;
373
+ const totalShift = precisionShift
374
+ ? precisionShift.sub(new anchor_1.BN(rightSideLength))
375
+ : numericConstants_1.ZERO;
376
+ return BigNum.from(rawBn, precisionShift).shift(totalShift, true);
377
+ }
378
+ static max(a, b) {
379
+ return a.gt(b) ? a : b;
380
+ }
381
+ static min(a, b) {
382
+ return a.lt(b) ? a : b;
383
+ }
384
+ static zero(precision) {
385
+ return BigNum.from(0, precision);
386
+ }
387
+ }
388
+ exports.BigNum = BigNum;
389
+ BigNum.delim = '.';
390
+ BigNum.spacer = ',';
@@ -0,0 +1,20 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.getOracleClient = void 0;
4
+ const types_1 = require("../types");
5
+ const pythClient_1 = require("../oracles/pythClient");
6
+ const switchboardClient_1 = require("../oracles/switchboardClient");
7
+ const quoteAssetOracleClient_1 = require("../oracles/quoteAssetOracleClient");
8
+ function getOracleClient(oracleSource, connection) {
9
+ if (types_1.isVariant(oracleSource, 'pyth')) {
10
+ return new pythClient_1.PythClient(connection);
11
+ }
12
+ if (types_1.isVariant(oracleSource, 'switchboard')) {
13
+ return new switchboardClient_1.SwitchboardClient(connection);
14
+ }
15
+ if (types_1.isVariant(oracleSource, 'quoteAsset')) {
16
+ return new quoteAssetOracleClient_1.QuoteAssetOracleClient();
17
+ }
18
+ throw new Error(`Unknown oracle source ${oracleSource}`);
19
+ }
20
+ exports.getOracleClient = getOracleClient;