@drift-labs/sdk 0.1.19-master.2 → 0.1.20

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 (106) hide show
  1. package/lib/accounts/bulkAccountLoader.d.ts +32 -0
  2. package/lib/accounts/bulkAccountLoader.d.ts.map +1 -0
  3. package/lib/accounts/bulkAccountLoader.js +166 -0
  4. package/lib/accounts/bulkUserSubscription.d.ts +8 -0
  5. package/lib/accounts/bulkUserSubscription.d.ts.map +1 -0
  6. package/lib/accounts/bulkUserSubscription.js +28 -0
  7. package/lib/accounts/pollingClearingHouseAccountSubscriber.d.ts +46 -0
  8. package/lib/accounts/pollingClearingHouseAccountSubscriber.d.ts.map +1 -0
  9. package/lib/accounts/pollingClearingHouseAccountSubscriber.js +206 -0
  10. package/lib/accounts/pollingTokenAccountSubscriber.d.ts +26 -0
  11. package/lib/accounts/pollingTokenAccountSubscriber.d.ts.map +1 -0
  12. package/lib/accounts/pollingTokenAccountSubscriber.js +79 -0
  13. package/lib/accounts/pollingUserAccountSubscriber.d.ts +31 -0
  14. package/lib/accounts/pollingUserAccountSubscriber.d.ts.map +1 -0
  15. package/lib/accounts/pollingUserAccountSubscriber.js +123 -0
  16. package/lib/accounts/types.d.ts +30 -0
  17. package/lib/accounts/types.d.ts.map +1 -1
  18. package/lib/accounts/utils.d.ts +2 -0
  19. package/lib/accounts/utils.d.ts.map +1 -0
  20. package/lib/accounts/utils.js +7 -0
  21. package/lib/accounts/webSocketAccountSubscriber.d.ts +6 -2
  22. package/lib/accounts/webSocketAccountSubscriber.d.ts.map +1 -1
  23. package/lib/accounts/webSocketAccountSubscriber.js +43 -12
  24. package/lib/accounts/{defaultClearingHouseAccountSubscriber.d.ts → webSocketClearingHouseAccountSubscriber.d.ts} +4 -2
  25. package/lib/accounts/webSocketClearingHouseAccountSubscriber.d.ts.map +1 -0
  26. package/lib/accounts/{defaultClearingHouseAccountSubscriber.js → webSocketClearingHouseAccountSubscriber.js} +5 -4
  27. package/lib/accounts/{defaultUserAccountSubscriber.d.ts → webSocketUserAccountSubscriber.d.ts} +4 -2
  28. package/lib/accounts/webSocketUserAccountSubscriber.d.ts.map +1 -0
  29. package/lib/accounts/{defaultUserAccountSubscriber.js → webSocketUserAccountSubscriber.js} +5 -4
  30. package/lib/admin.d.ts.map +1 -1
  31. package/lib/admin.js +6 -14
  32. package/lib/clearingHouse.d.ts +12 -2
  33. package/lib/clearingHouse.d.ts.map +1 -1
  34. package/lib/clearingHouse.js +28 -17
  35. package/lib/clearingHouseUser.d.ts +11 -3
  36. package/lib/clearingHouseUser.d.ts.map +1 -1
  37. package/lib/clearingHouseUser.js +46 -36
  38. package/lib/constants/markets.d.ts +2 -1
  39. package/lib/constants/markets.d.ts.map +1 -1
  40. package/lib/constants/markets.js +20 -15
  41. package/lib/constants/numericConstants.d.ts +3 -1
  42. package/lib/constants/numericConstants.d.ts.map +1 -1
  43. package/lib/constants/numericConstants.js +15 -17
  44. package/lib/examples/makeTradeExample.js +6 -6
  45. package/lib/factory/clearingHouse.d.ts +26 -0
  46. package/lib/factory/clearingHouse.d.ts.map +1 -0
  47. package/lib/factory/clearingHouse.js +64 -0
  48. package/lib/factory/clearingHouseUser.d.ts +20 -0
  49. package/lib/factory/clearingHouseUser.d.ts.map +1 -0
  50. package/lib/factory/clearingHouseUser.js +34 -0
  51. package/lib/index.d.ts +8 -2
  52. package/lib/index.d.ts.map +1 -1
  53. package/lib/index.js +9 -2
  54. package/lib/math/amm.js +7 -7
  55. package/lib/math/conversion.d.ts +1 -1
  56. package/lib/math/conversion.d.ts.map +1 -1
  57. package/lib/math/conversion.js +1 -1
  58. package/lib/math/funding.js +1 -1
  59. package/lib/math/insuranceFund.d.ts +2 -1
  60. package/lib/math/insuranceFund.d.ts.map +1 -1
  61. package/lib/math/insuranceFund.js +3 -6
  62. package/lib/math/market.js +1 -1
  63. package/lib/math/position.d.ts +2 -1
  64. package/lib/math/position.d.ts.map +1 -1
  65. package/lib/math/position.js +3 -6
  66. package/lib/math/trade.js +16 -16
  67. package/lib/math/utils.d.ts +2 -1
  68. package/lib/math/utils.d.ts.map +1 -1
  69. package/lib/math/utils.js +3 -3
  70. package/lib/mockUSDCFaucet.d.ts +2 -1
  71. package/lib/mockUSDCFaucet.d.ts.map +1 -1
  72. package/lib/pythClient.js +1 -1
  73. package/lib/token/index.d.ts +4 -0
  74. package/lib/token/index.d.ts.map +1 -0
  75. package/lib/token/index.js +38 -0
  76. package/lib/types.d.ts +2 -1
  77. package/lib/types.d.ts.map +1 -1
  78. package/package.json +11 -3
  79. package/src/accounts/bulkAccountLoader.ts +207 -0
  80. package/src/accounts/bulkUserSubscription.ts +28 -0
  81. package/src/accounts/pollingClearingHouseAccountSubscriber.ts +293 -0
  82. package/src/accounts/pollingTokenAccountSubscriber.ts +99 -0
  83. package/src/accounts/pollingUserAccountSubscriber.ts +166 -0
  84. package/src/accounts/types.ts +38 -0
  85. package/src/accounts/utils.ts +3 -0
  86. package/src/accounts/webSocketAccountSubscriber.ts +67 -17
  87. package/src/accounts/{defaultClearingHouseAccountSubscriber.ts → webSocketClearingHouseAccountSubscriber.ts} +4 -1
  88. package/src/accounts/{defaultUserAccountSubscriber.ts → webSocketUserAccountSubscriber.ts} +4 -1
  89. package/src/admin.ts +8 -18
  90. package/src/clearingHouse.ts +25 -17
  91. package/src/clearingHouseUser.ts +26 -6
  92. package/src/constants/markets.ts +9 -1
  93. package/src/constants/numericConstants.ts +2 -1
  94. package/src/examples/makeTradeExample.ts +4 -1
  95. package/src/factory/clearingHouse.ts +125 -0
  96. package/src/factory/clearingHouseUser.ts +73 -0
  97. package/src/index.ts +8 -2
  98. package/src/math/conversion.ts +1 -1
  99. package/src/math/insuranceFund.ts +1 -1
  100. package/src/math/position.ts +1 -1
  101. package/src/math/utils.ts +1 -1
  102. package/src/mockUSDCFaucet.ts +1 -1
  103. package/src/token/index.ts +37 -0
  104. package/src/types.ts +1 -1
  105. package/lib/accounts/defaultClearingHouseAccountSubscriber.d.ts.map +0 -1
  106. package/lib/accounts/defaultUserAccountSubscriber.d.ts.map +0 -1
@@ -0,0 +1,38 @@
1
+ "use strict";
2
+ Object.defineProperty(exports, "__esModule", { value: true });
3
+ exports.parseTokenAccount = void 0;
4
+ const spl_token_1 = require("@solana/spl-token");
5
+ const web3_js_1 = require("@solana/web3.js");
6
+ function parseTokenAccount(data) {
7
+ const accountInfo = spl_token_1.AccountLayout.decode(data);
8
+ accountInfo.mint = new web3_js_1.PublicKey(accountInfo.mint);
9
+ accountInfo.owner = new web3_js_1.PublicKey(accountInfo.owner);
10
+ accountInfo.amount = spl_token_1.u64.fromBuffer(accountInfo.amount);
11
+ if (accountInfo.delegateOption === 0) {
12
+ accountInfo.delegate = null;
13
+ // eslint-disable-next-line new-cap
14
+ accountInfo.delegatedAmount = new spl_token_1.u64(0);
15
+ }
16
+ else {
17
+ accountInfo.delegate = new web3_js_1.PublicKey(accountInfo.delegate);
18
+ accountInfo.delegatedAmount = spl_token_1.u64.fromBuffer(accountInfo.delegatedAmount);
19
+ }
20
+ accountInfo.isInitialized = accountInfo.state !== 0;
21
+ accountInfo.isFrozen = accountInfo.state === 2;
22
+ if (accountInfo.isNativeOption === 1) {
23
+ accountInfo.rentExemptReserve = spl_token_1.u64.fromBuffer(accountInfo.isNative);
24
+ accountInfo.isNative = true;
25
+ }
26
+ else {
27
+ accountInfo.rentExemptReserve = null;
28
+ accountInfo.isNative = false;
29
+ }
30
+ if (accountInfo.closeAuthorityOption === 0) {
31
+ accountInfo.closeAuthority = null;
32
+ }
33
+ else {
34
+ accountInfo.closeAuthority = new web3_js_1.PublicKey(accountInfo.closeAuthority);
35
+ }
36
+ return accountInfo;
37
+ }
38
+ exports.parseTokenAccount = parseTokenAccount;
package/lib/types.d.ts CHANGED
@@ -1,5 +1,6 @@
1
+ /// <reference types="bn.js" />
1
2
  import { PublicKey, Transaction } from '@solana/web3.js';
2
- import BN from 'bn.js';
3
+ import { BN } from '.';
3
4
  export declare class SwapDirection {
4
5
  static readonly ADD: {
5
6
  add: {};
@@ -1 +1 @@
1
- {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACzD,OAAO,EAAE,MAAM,OAAO,CAAC;AAGvB,qBAAa,aAAa;IACzB,MAAM,CAAC,QAAQ,CAAC,GAAG;;MAAe;IAClC,MAAM,CAAC,QAAQ,CAAC,MAAM;;MAAkB;CACxC;AACD,qBAAa,iBAAiB;IAC7B,MAAM,CAAC,QAAQ,CAAC,IAAI;;MAAgB;IACpC,MAAM,CAAC,QAAQ,CAAC,KAAK;;MAAiB;CACtC;AAED,qBAAa,YAAY;IACxB,MAAM,CAAC,QAAQ,CAAC,IAAI;;MAAgB;IACpC,MAAM,CAAC,QAAQ,CAAC,WAAW;;MAAuB;CAClD;AAED,oBAAY,SAAS;IACpB,IAAI,IAAI;IACR,GAAG,IAAI;IACP,IAAI,IAAI;CACR;AAED,oBAAY,gBAAgB,GACzB,GAAG,GACH,GAAG,GACH,IAAI,GACJ,IAAI,GACJ,KAAK,GACL,GAAG,GACH,GAAG,GACH,GAAG,CAAC;AAGP,oBAAY,mBAAmB,GAAG;IACjC,IAAI,EAAE,EAAE,CAAC;IACT,YAAY,EAAE,WAAW,EAAE,CAAC;CAC5B,CAAC;AAEF,oBAAY,qBAAqB,GAAG;IACnC,IAAI,EAAE,EAAE,CAAC;IACT,cAAc,EAAE,aAAa,EAAE,CAAC;CAChC,CAAC;AAEF,oBAAY,2BAA2B,GAAG;IACzC,IAAI,EAAE,EAAE,CAAC;IACT,YAAY,EAAE,mBAAmB,EAAE,CAAC;CACpC,CAAC;AAEF,oBAAY,yBAAyB,GAAG;IACvC,IAAI,EAAE,EAAE,CAAC;IACT,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;CACxC,CAAC;AAEF,oBAAY,4BAA4B,GAAG;IAC1C,IAAI,EAAE,EAAE,CAAC;IACT,qBAAqB,EAAE,oBAAoB,EAAE,CAAC;CAC9C,CAAC;AAEF,oBAAY,yBAAyB,GAAG;IACvC,IAAI,EAAE,EAAE,CAAC;IACT,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;CACxC,CAAC;AAEF,oBAAY,aAAa,GAAG;IAC3B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE;QACV,OAAO,CAAC,EAAE,GAAG,CAAC;QACd,QAAQ,CAAC,EAAE,GAAG,CAAC;KACf,CAAC;IACF,gBAAgB,EAAE,EAAE,CAAC;IACrB,wBAAwB,EAAE,EAAE,CAAC;IAC7B,MAAM,EAAE,EAAE,CAAC;CACX,CAAC;AAEF,oBAAY,mBAAmB,GAAG;IACjC,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,WAAW,EAAE,EAAE,CAAC;IAChB,mBAAmB,EAAE,EAAE,CAAC;IACxB,sBAAsB,EAAE,EAAE,CAAC;IAC3B,uBAAuB,EAAE,EAAE,CAAC;IAC5B,WAAW,EAAE,EAAE,CAAC;IAChB,kBAAkB,EAAE,EAAE,CAAC;IACvB,qBAAqB,EAAE,EAAE,CAAC;IAC1B,sBAAsB,EAAE,EAAE,CAAC;IAC3B,UAAU,EAAE,EAAE,CAAC;IACf,mBAAmB,EAAE,EAAE,CAAC;IACxB,oBAAoB,EAAE,EAAE,CAAC;IACzB,eAAe,EAAE,EAAE,CAAC;IACpB,YAAY,EAAE,EAAE,CAAC;IACjB,WAAW,EAAE,EAAE,CAAC;IAChB,OAAO,EAAE,EAAE,CAAC;CACZ,CAAC;AAEF,oBAAY,WAAW,GAAG;IACzB,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE;QACV,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,KAAK,CAAC,EAAE,GAAG,CAAC;KACZ,CAAC;IACF,eAAe,EAAE,EAAE,CAAC;IACpB,gBAAgB,EAAE,EAAE,CAAC;IACrB,eAAe,EAAE,EAAE,CAAC;IACpB,cAAc,EAAE,EAAE,CAAC;IACnB,GAAG,EAAE,EAAE,CAAC;IACR,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,aAAa,EAAE,EAAE,CAAC;IAClB,WAAW,EAAE,EAAE,CAAC;IAChB,WAAW,EAAE,OAAO,CAAC;IACrB,WAAW,EAAE,EAAE,CAAC;CAChB,CAAC;AAEF,oBAAY,iBAAiB,GAAG;IAC/B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,WAAW,EAAE,EAAE,CAAC;IAChB,WAAW,EAAE,EAAE,CAAC;IAChB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,eAAe,EAAE,EAAE,CAAC;IACpB,aAAa,EAAE,EAAE,CAAC;CAClB,CAAC;AAEF,oBAAY,oBAAoB,GAAG;IAClC,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,WAAW,EAAE,EAAE,CAAC;IAChB,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,qBAAqB,EAAE,EAAE,CAAC;IAC1B,wBAAwB,EAAE,EAAE,CAAC;IAC7B,yBAAyB,EAAE,EAAE,CAAC;CAC9B,CAAC;AAEF,oBAAY,iBAAiB,GAAG;IAC/B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,cAAc,EAAE,EAAE,CAAC;IACnB,oBAAoB,EAAE,EAAE,CAAC;IACzB,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,kBAAkB,EAAE,EAAE,CAAC;IACvB,UAAU,EAAE,SAAS,CAAC;IACtB,eAAe,EAAE,EAAE,CAAC;IACpB,UAAU,EAAE,EAAE,CAAC;IACf,aAAa,EAAE,EAAE,CAAC;IAClB,WAAW,EAAE,EAAE,CAAC;CAChB,CAAC;AAEF,oBAAY,YAAY,GAAG;IAC1B,KAAK,EAAE,SAAS,CAAC;IACjB,aAAa,EAAE,OAAO,CAAC;IACvB,cAAc,EAAE,OAAO,CAAC;IACxB,mBAAmB,EAAE,OAAO,CAAC;IAC7B,cAAc,EAAE,SAAS,CAAC;IAC1B,eAAe,EAAE,SAAS,CAAC;IAC3B,wBAAwB,EAAE,SAAS,CAAC;IACpC,oBAAoB,EAAE,MAAM,CAAC;IAC7B,cAAc,EAAE,SAAS,CAAC;IAC1B,uBAAuB,EAAE,SAAS,CAAC;IACnC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,kBAAkB,EAAE,EAAE,CAAC;IACvB,sBAAsB,EAAE,EAAE,CAAC;IAC3B,kBAAkB,EAAE,EAAE,CAAC;IACvB,OAAO,EAAE,SAAS,CAAC;IACnB,YAAY,EAAE,SAAS,CAAC;IACxB,cAAc,EAAE,SAAS,CAAC;IAC1B,kBAAkB,EAAE,SAAS,CAAC;IAC9B,qBAAqB,EAAE,SAAS,CAAC;IACjC,YAAY,EAAE,SAAS,CAAC;IACxB,kBAAkB,EAAE,SAAS,CAAC;IAC9B,0CAA0C,EAAE,EAAE,CAAC;IAC/C,4CAA4C,EAAE,EAAE,CAAC;IACjD,4CAA4C,EAAE,EAAE,CAAC;IACjD,8CAA8C,EAAE,EAAE,CAAC;IACnD,yCAAyC,EAAE,EAAE,CAAC;IAC9C,2CAA2C,EAAE,EAAE,CAAC;IAChD,4CAA4C,EAAE,EAAE,CAAC;IACjD,yCAAyC,EAAE,EAAE,CAAC;IAC9C,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,EAAE,EAAE,CAAC;IACb,iBAAiB,EAAE,EAAE,CAAC;IACtB,aAAa,EAAE,SAAS,CAAC;IACzB,YAAY,EAAE,SAAS,CAAC;IACxB,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,EAAE,CAAC;IACf,oBAAoB,EAAE,SAAS,CAAC;CAChC,CAAC;AAEF,oBAAY,cAAc,GAAG;IAC5B,YAAY,EAAE,EAAE,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;CAClB,CAAC;AAEF,oBAAY,MAAM,GAAG;IACpB,GAAG,EAAE,GAAG,CAAC;IACT,eAAe,EAAE,EAAE,CAAC;IACpB,mBAAmB,EAAE,EAAE,CAAC;IACxB,oBAAoB,EAAE,EAAE,CAAC;IACzB,WAAW,EAAE,OAAO,CAAC;IACrB,YAAY,EAAE,EAAE,CAAC;CACjB,CAAC;AAEF,oBAAY,GAAG,GAAG;IACjB,gBAAgB,EAAE,EAAE,CAAC;IACrB,KAAK,EAAE,EAAE,CAAC;IACV,qBAAqB,EAAE,EAAE,CAAC;IAC1B,eAAe,EAAE,EAAE,CAAC;IACpB,iBAAiB,EAAE,EAAE,CAAC;IACtB,iBAAiB,EAAE,EAAE,CAAC;IACtB,mBAAmB,EAAE,EAAE,CAAC;IACxB,mBAAmB,EAAE,EAAE,CAAC;IACxB,qBAAqB,EAAE,EAAE,CAAC;IAC1B,MAAM,EAAE,SAAS,CAAC;IAClB,YAAY,EAAE,YAAY,CAAC;IAC3B,aAAa,EAAE,EAAE,CAAC;IAClB,iBAAiB,EAAE,EAAE,CAAC;IACtB,aAAa,EAAE,EAAE,CAAC;IAClB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,iBAAiB,EAAE,EAAE,CAAC;IACtB,QAAQ,EAAE,EAAE,CAAC;IACb,gBAAgB,EAAE,EAAE,CAAC;IACrB,eAAe,EAAE,EAAE,CAAC;CACpB,CAAC;AAGF,oBAAY,YAAY,GAAG;IAC1B,eAAe,EAAE,EAAE,CAAC;IACpB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,WAAW,EAAE,EAAE,CAAC;IAChB,gBAAgB,EAAE,EAAE,CAAC;CACrB,CAAC;AAEF,oBAAY,oBAAoB,GAAG;IAClC,SAAS,EAAE,YAAY,EAAE,CAAC;IAC1B,IAAI,EAAE,SAAS,CAAC;CAChB,CAAC;AAEF,oBAAY,WAAW,GAAG;IACzB,SAAS,EAAE,SAAS,CAAC;IACrB,UAAU,EAAE,EAAE,CAAC;IACf,kBAAkB,EAAE,EAAE,CAAC;IACvB,SAAS,EAAE,SAAS,CAAC;IACrB,YAAY,EAAE,EAAE,CAAC;CACjB,CAAC;AAGF,MAAM,WAAW,OAAO;IACvB,eAAe,CAAC,EAAE,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACvD,mBAAmB,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IAChE,SAAS,EAAE,SAAS,CAAC;CACrB;AAED,oBAAY,YAAY,GAAG;IAC1B,YAAY,EAAE,EAAE,CAAC;IACjB,cAAc,EAAE,EAAE,CAAC;IACnB,kBAAkB,EAAE;QACnB,SAAS,EAAE;YACV,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,UAAU,EAAE;YACX,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,SAAS,EAAE;YACV,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,UAAU,EAAE;YACX,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;KACF,CAAC;IACF,gBAAgB,EAAE;QACjB,uBAAuB,EAAE,EAAE,CAAC;QAC5B,yBAAyB,EAAE,EAAE,CAAC;QAC9B,wBAAwB,EAAE,EAAE,CAAC;QAC7B,0BAA0B,EAAE,EAAE,CAAC;KAC/B,CAAC;CACF,CAAC;AAEF,oBAAY,gBAAgB,GAAG;IAC9B,eAAe,EAAE;QAChB,6BAA6B,EAAE,EAAE,CAAC;QAClC,+BAA+B,EAAE,EAAE,CAAC;KACpC,CAAC;IACF,QAAQ,EAAE;QACT,gBAAgB,EAAE,EAAE,CAAC;QACrB,yBAAyB,EAAE,EAAE,CAAC;QAC9B,gBAAgB,EAAE,EAAE,CAAC;KACrB,CAAC;IACF,kBAAkB,EAAE,OAAO,CAAC;CAC5B,CAAC"}
1
+ {"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../src/types.ts"],"names":[],"mappings":";AAAA,OAAO,EAAE,SAAS,EAAE,WAAW,EAAE,MAAM,iBAAiB,CAAC;AACzD,OAAO,EAAE,EAAE,EAAE,MAAM,GAAG,CAAC;AAGvB,qBAAa,aAAa;IACzB,MAAM,CAAC,QAAQ,CAAC,GAAG;;MAAe;IAClC,MAAM,CAAC,QAAQ,CAAC,MAAM;;MAAkB;CACxC;AACD,qBAAa,iBAAiB;IAC7B,MAAM,CAAC,QAAQ,CAAC,IAAI;;MAAgB;IACpC,MAAM,CAAC,QAAQ,CAAC,KAAK;;MAAiB;CACtC;AAED,qBAAa,YAAY;IACxB,MAAM,CAAC,QAAQ,CAAC,IAAI;;MAAgB;IACpC,MAAM,CAAC,QAAQ,CAAC,WAAW;;MAAuB;CAClD;AAED,oBAAY,SAAS;IACpB,IAAI,IAAI;IACR,GAAG,IAAI;IACP,IAAI,IAAI;CACR;AAED,oBAAY,gBAAgB,GACzB,GAAG,GACH,GAAG,GACH,IAAI,GACJ,IAAI,GACJ,KAAK,GACL,GAAG,GACH,GAAG,GACH,GAAG,CAAC;AAGP,oBAAY,mBAAmB,GAAG;IACjC,IAAI,EAAE,EAAE,CAAC;IACT,YAAY,EAAE,WAAW,EAAE,CAAC;CAC5B,CAAC;AAEF,oBAAY,qBAAqB,GAAG;IACnC,IAAI,EAAE,EAAE,CAAC;IACT,cAAc,EAAE,aAAa,EAAE,CAAC;CAChC,CAAC;AAEF,oBAAY,2BAA2B,GAAG;IACzC,IAAI,EAAE,EAAE,CAAC;IACT,YAAY,EAAE,mBAAmB,EAAE,CAAC;CACpC,CAAC;AAEF,oBAAY,yBAAyB,GAAG;IACvC,IAAI,EAAE,EAAE,CAAC;IACT,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;CACxC,CAAC;AAEF,oBAAY,4BAA4B,GAAG;IAC1C,IAAI,EAAE,EAAE,CAAC;IACT,qBAAqB,EAAE,oBAAoB,EAAE,CAAC;CAC9C,CAAC;AAEF,oBAAY,yBAAyB,GAAG;IACvC,IAAI,EAAE,EAAE,CAAC;IACT,kBAAkB,EAAE,iBAAiB,EAAE,CAAC;CACxC,CAAC;AAEF,oBAAY,aAAa,GAAG;IAC3B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE;QACV,OAAO,CAAC,EAAE,GAAG,CAAC;QACd,QAAQ,CAAC,EAAE,GAAG,CAAC;KACf,CAAC;IACF,gBAAgB,EAAE,EAAE,CAAC;IACrB,wBAAwB,EAAE,EAAE,CAAC;IAC7B,MAAM,EAAE,EAAE,CAAC;CACX,CAAC;AAEF,oBAAY,mBAAmB,GAAG;IACjC,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,WAAW,EAAE,EAAE,CAAC;IAChB,mBAAmB,EAAE,EAAE,CAAC;IACxB,sBAAsB,EAAE,EAAE,CAAC;IAC3B,uBAAuB,EAAE,EAAE,CAAC;IAC5B,WAAW,EAAE,EAAE,CAAC;IAChB,kBAAkB,EAAE,EAAE,CAAC;IACvB,qBAAqB,EAAE,EAAE,CAAC;IAC1B,sBAAsB,EAAE,EAAE,CAAC;IAC3B,UAAU,EAAE,EAAE,CAAC;IACf,mBAAmB,EAAE,EAAE,CAAC;IACxB,oBAAoB,EAAE,EAAE,CAAC;IACzB,eAAe,EAAE,EAAE,CAAC;IACpB,YAAY,EAAE,EAAE,CAAC;IACjB,WAAW,EAAE,EAAE,CAAC;IAChB,OAAO,EAAE,EAAE,CAAC;CACZ,CAAC;AAEF,oBAAY,WAAW,GAAG;IACzB,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,SAAS,EAAE;QACV,IAAI,CAAC,EAAE,GAAG,CAAC;QACX,KAAK,CAAC,EAAE,GAAG,CAAC;KACZ,CAAC;IACF,eAAe,EAAE,EAAE,CAAC;IACpB,gBAAgB,EAAE,EAAE,CAAC;IACrB,eAAe,EAAE,EAAE,CAAC;IACpB,cAAc,EAAE,EAAE,CAAC;IACnB,GAAG,EAAE,EAAE,CAAC;IACR,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,aAAa,EAAE,EAAE,CAAC;IAClB,WAAW,EAAE,EAAE,CAAC;IAChB,WAAW,EAAE,OAAO,CAAC;IACrB,WAAW,EAAE,EAAE,CAAC;CAChB,CAAC;AAEF,oBAAY,iBAAiB,GAAG;IAC/B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,WAAW,EAAE,EAAE,CAAC;IAChB,WAAW,EAAE,EAAE,CAAC;IAChB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,eAAe,EAAE,EAAE,CAAC;IACpB,aAAa,EAAE,EAAE,CAAC;CAClB,CAAC;AAEF,oBAAY,oBAAoB,GAAG;IAClC,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,WAAW,EAAE,EAAE,CAAC;IAChB,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,qBAAqB,EAAE,EAAE,CAAC;IAC1B,wBAAwB,EAAE,EAAE,CAAC;IAC7B,yBAAyB,EAAE,EAAE,CAAC;CAC9B,CAAC;AAEF,oBAAY,iBAAiB,GAAG;IAC/B,EAAE,EAAE,EAAE,CAAC;IACP,QAAQ,EAAE,EAAE,CAAC;IACb,aAAa,EAAE,SAAS,CAAC;IACzB,IAAI,EAAE,SAAS,CAAC;IAChB,OAAO,EAAE,OAAO,CAAC;IACjB,cAAc,EAAE,EAAE,CAAC;IACnB,oBAAoB,EAAE,EAAE,CAAC;IACzB,cAAc,EAAE,EAAE,CAAC;IACnB,eAAe,EAAE,EAAE,CAAC;IACpB,kBAAkB,EAAE,EAAE,CAAC;IACvB,UAAU,EAAE,SAAS,CAAC;IACtB,eAAe,EAAE,EAAE,CAAC;IACpB,UAAU,EAAE,EAAE,CAAC;IACf,aAAa,EAAE,EAAE,CAAC;IAClB,WAAW,EAAE,EAAE,CAAC;CAChB,CAAC;AAEF,oBAAY,YAAY,GAAG;IAC1B,KAAK,EAAE,SAAS,CAAC;IACjB,aAAa,EAAE,OAAO,CAAC;IACvB,cAAc,EAAE,OAAO,CAAC;IACxB,mBAAmB,EAAE,OAAO,CAAC;IAC7B,cAAc,EAAE,SAAS,CAAC;IAC1B,eAAe,EAAE,SAAS,CAAC;IAC3B,wBAAwB,EAAE,SAAS,CAAC;IACpC,oBAAoB,EAAE,MAAM,CAAC;IAC7B,cAAc,EAAE,SAAS,CAAC;IAC1B,uBAAuB,EAAE,SAAS,CAAC;IACnC,mBAAmB,EAAE,MAAM,CAAC;IAC5B,kBAAkB,EAAE,EAAE,CAAC;IACvB,sBAAsB,EAAE,EAAE,CAAC;IAC3B,kBAAkB,EAAE,EAAE,CAAC;IACvB,OAAO,EAAE,SAAS,CAAC;IACnB,YAAY,EAAE,SAAS,CAAC;IACxB,cAAc,EAAE,SAAS,CAAC;IAC1B,kBAAkB,EAAE,SAAS,CAAC;IAC9B,qBAAqB,EAAE,SAAS,CAAC;IACjC,YAAY,EAAE,SAAS,CAAC;IACxB,kBAAkB,EAAE,SAAS,CAAC;IAC9B,0CAA0C,EAAE,EAAE,CAAC;IAC/C,4CAA4C,EAAE,EAAE,CAAC;IACjD,4CAA4C,EAAE,EAAE,CAAC;IACjD,8CAA8C,EAAE,EAAE,CAAC;IACnD,yCAAyC,EAAE,EAAE,CAAC;IAC9C,2CAA2C,EAAE,EAAE,CAAC;IAChD,4CAA4C,EAAE,EAAE,CAAC;IACjD,yCAAyC,EAAE,EAAE,CAAC;IAC9C,YAAY,EAAE,YAAY,CAAC;IAC3B,QAAQ,EAAE,EAAE,CAAC;IACb,iBAAiB,EAAE,EAAE,CAAC;IACtB,aAAa,EAAE,SAAS,CAAC;IACzB,YAAY,EAAE,SAAS,CAAC;IACxB,gBAAgB,EAAE,gBAAgB,CAAC;IACnC,UAAU,EAAE,EAAE,CAAC;IACf,oBAAoB,EAAE,SAAS,CAAC;CAChC,CAAC;AAEF,oBAAY,cAAc,GAAG;IAC5B,YAAY,EAAE,EAAE,CAAC;IACjB,OAAO,EAAE,MAAM,EAAE,CAAC;CAClB,CAAC;AAEF,oBAAY,MAAM,GAAG;IACpB,GAAG,EAAE,GAAG,CAAC;IACT,eAAe,EAAE,EAAE,CAAC;IACpB,mBAAmB,EAAE,EAAE,CAAC;IACxB,oBAAoB,EAAE,EAAE,CAAC;IACzB,WAAW,EAAE,OAAO,CAAC;IACrB,YAAY,EAAE,EAAE,CAAC;CACjB,CAAC;AAEF,oBAAY,GAAG,GAAG;IACjB,gBAAgB,EAAE,EAAE,CAAC;IACrB,KAAK,EAAE,EAAE,CAAC;IACV,qBAAqB,EAAE,EAAE,CAAC;IAC1B,eAAe,EAAE,EAAE,CAAC;IACpB,iBAAiB,EAAE,EAAE,CAAC;IACtB,iBAAiB,EAAE,EAAE,CAAC;IACtB,mBAAmB,EAAE,EAAE,CAAC;IACxB,mBAAmB,EAAE,EAAE,CAAC;IACxB,qBAAqB,EAAE,EAAE,CAAC;IAC1B,MAAM,EAAE,SAAS,CAAC;IAClB,YAAY,EAAE,YAAY,CAAC;IAC3B,aAAa,EAAE,EAAE,CAAC;IAClB,iBAAiB,EAAE,EAAE,CAAC;IACtB,aAAa,EAAE,EAAE,CAAC;IAClB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,yBAAyB,EAAE,EAAE,CAAC;IAC9B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,0BAA0B,EAAE,EAAE,CAAC;IAC/B,iBAAiB,EAAE,EAAE,CAAC;IACtB,QAAQ,EAAE,EAAE,CAAC;IACb,gBAAgB,EAAE,EAAE,CAAC;IACrB,eAAe,EAAE,EAAE,CAAC;CACpB,CAAC;AAGF,oBAAY,YAAY,GAAG;IAC1B,eAAe,EAAE,EAAE,CAAC;IACpB,yBAAyB,EAAE,EAAE,CAAC;IAC9B,WAAW,EAAE,EAAE,CAAC;IAChB,gBAAgB,EAAE,EAAE,CAAC;CACrB,CAAC;AAEF,oBAAY,oBAAoB,GAAG;IAClC,SAAS,EAAE,YAAY,EAAE,CAAC;IAC1B,IAAI,EAAE,SAAS,CAAC;CAChB,CAAC;AAEF,oBAAY,WAAW,GAAG;IACzB,SAAS,EAAE,SAAS,CAAC;IACrB,UAAU,EAAE,EAAE,CAAC;IACf,kBAAkB,EAAE,EAAE,CAAC;IACvB,SAAS,EAAE,SAAS,CAAC;IACrB,YAAY,EAAE,EAAE,CAAC;CACjB,CAAC;AAGF,MAAM,WAAW,OAAO;IACvB,eAAe,CAAC,EAAE,EAAE,WAAW,GAAG,OAAO,CAAC,WAAW,CAAC,CAAC;IACvD,mBAAmB,CAAC,GAAG,EAAE,WAAW,EAAE,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC,CAAC;IAChE,SAAS,EAAE,SAAS,CAAC;CACrB;AAED,oBAAY,YAAY,GAAG;IAC1B,YAAY,EAAE,EAAE,CAAC;IACjB,cAAc,EAAE,EAAE,CAAC;IACnB,kBAAkB,EAAE;QACnB,SAAS,EAAE;YACV,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,UAAU,EAAE;YACX,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,SAAS,EAAE;YACV,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;QACF,UAAU,EAAE;YACX,cAAc,EAAE,EAAE,CAAC;YACnB,iBAAiB,EAAE,EAAE,CAAC;YACtB,mBAAmB,EAAE,EAAE,CAAC;SACxB,CAAC;KACF,CAAC;IACF,gBAAgB,EAAE;QACjB,uBAAuB,EAAE,EAAE,CAAC;QAC5B,yBAAyB,EAAE,EAAE,CAAC;QAC9B,wBAAwB,EAAE,EAAE,CAAC;QAC7B,0BAA0B,EAAE,EAAE,CAAC;KAC/B,CAAC;CACF,CAAC;AAEF,oBAAY,gBAAgB,GAAG;IAC9B,eAAe,EAAE;QAChB,6BAA6B,EAAE,EAAE,CAAC;QAClC,+BAA+B,EAAE,EAAE,CAAC;KACpC,CAAC;IACF,QAAQ,EAAE;QACT,gBAAgB,EAAE,EAAE,CAAC;QACrB,yBAAyB,EAAE,EAAE,CAAC;QAC9B,gBAAgB,EAAE,EAAE,CAAC;KACrB,CAAC;IACF,kBAAkB,EAAE,OAAO,CAAC;CAC5B,CAAC"}
package/package.json CHANGED
@@ -1,6 +1,6 @@
1
1
  {
2
2
  "name": "@drift-labs/sdk",
3
- "version": "0.1.19-master.2",
3
+ "version": "0.1.20",
4
4
  "main": "lib/index.js",
5
5
  "types": "lib/index.d.ts",
6
6
  "author": "crispheaney",
@@ -10,6 +10,7 @@
10
10
  "url": "git@github.com:drift-labs/protocol-v1.git"
11
11
  },
12
12
  "scripts": {
13
+ "lint": "eslint './**/*.{ts,tsx}' --quiet",
13
14
  "build": "yarn clean && tsc",
14
15
  "clean": "rm -rf lib",
15
16
  "patch-and-pub": "npm version patch --force && npm publish"
@@ -31,8 +32,15 @@
31
32
  "@pythnetwork/client": "^2.5.1",
32
33
  "@solana/spl-token": "^0.1.6",
33
34
  "@solana/web3.js": "^1.22.0",
34
- "@types/bn.js": "^5.1.0",
35
- "strict-event-emitter-types": "^2.0.0"
35
+ "strict-event-emitter-types": "^2.0.0",
36
+ "uuid": "^8.3.2"
37
+ },
38
+ "devDependencies": {
39
+ "@typescript-eslint/eslint-plugin": "^4.28.0",
40
+ "@typescript-eslint/parser": "^4.28.0",
41
+ "eslint": "^7.29.0",
42
+ "eslint-config-prettier": "^8.3.0",
43
+ "eslint-plugin-prettier": "^3.4.0"
36
44
  },
37
45
  "description": "SDK for Drift Protocol v1",
38
46
  "engines": {
@@ -0,0 +1,207 @@
1
+ import { Commitment, Connection, PublicKey } from '@solana/web3.js';
2
+ import { v4 as uuidv4 } from 'uuid';
3
+ import { AccountData } from './types';
4
+
5
+ type AccountToLoad = {
6
+ publicKey: PublicKey;
7
+ callbacks: Map<string, (buffer: Buffer) => void>;
8
+ };
9
+
10
+ const GET_MULTIPLE_ACCOUNTS_CHUNK_SIZE = 99;
11
+
12
+ export class BulkAccountLoader {
13
+ connection: Connection;
14
+ commitment: Commitment;
15
+ pollingFrequency: number;
16
+ accountsToLoad = new Map<string, AccountToLoad>();
17
+ accountData = new Map<string, AccountData>();
18
+ errorCallbacks = new Map<string, (e) => void>();
19
+ intervalId?: NodeJS.Timer;
20
+ // to handle clients spamming load
21
+ loadPromise?: Promise<void>;
22
+ loadPromiseResolver: () => void;
23
+
24
+ public constructor(
25
+ connection: Connection,
26
+ commitment: Commitment,
27
+ pollingFrequency: number
28
+ ) {
29
+ this.connection = connection;
30
+ this.commitment = commitment;
31
+ this.pollingFrequency = pollingFrequency;
32
+ }
33
+
34
+ public addAccount(
35
+ publicKey: PublicKey,
36
+ callback: (buffer: Buffer) => void
37
+ ): string {
38
+ const existingSize = this.accountsToLoad.size;
39
+
40
+ const callbackId = uuidv4();
41
+ const existingAccountToLoad = this.accountsToLoad.get(publicKey.toString());
42
+ if (existingAccountToLoad) {
43
+ existingAccountToLoad.callbacks.set(callbackId, callback);
44
+ } else {
45
+ const callbacks = new Map<string, (buffer: Buffer) => void>();
46
+ callbacks.set(callbackId, callback);
47
+ const newAccountToLoad = {
48
+ publicKey,
49
+ callbacks,
50
+ };
51
+ this.accountsToLoad.set(publicKey.toString(), newAccountToLoad);
52
+ }
53
+
54
+ if (existingSize === 0) {
55
+ this.startPolling();
56
+ }
57
+
58
+ // if a new account needs to be polled, remove the cached loadPromise in case client calls load immediately after
59
+ this.loadPromise = undefined;
60
+
61
+ return callbackId;
62
+ }
63
+
64
+ public removeAccount(publicKey: PublicKey, callbackId: string): void {
65
+ const existingAccountToLoad = this.accountsToLoad.get(publicKey.toString());
66
+ if (existingAccountToLoad) {
67
+ existingAccountToLoad.callbacks.delete(callbackId);
68
+ if (existingAccountToLoad.callbacks.size === 0) {
69
+ this.accountsToLoad.delete(existingAccountToLoad.publicKey.toString());
70
+ }
71
+ }
72
+
73
+ if (this.accountsToLoad.size === 0) {
74
+ this.stopPolling();
75
+ }
76
+ }
77
+
78
+ public addErrorCallbacks(callback: (error: Error) => void): string {
79
+ const callbackId = uuidv4();
80
+ this.errorCallbacks.set(callbackId, callback);
81
+ return callbackId;
82
+ }
83
+
84
+ public removeErrorCallbacks(callbackId: string): void {
85
+ this.errorCallbacks.delete(callbackId);
86
+ }
87
+
88
+ chunks<T>(array: readonly T[], size: number): T[][] {
89
+ return new Array(Math.ceil(array.length / size))
90
+ .fill(null)
91
+ .map((_, index) => index * size)
92
+ .map((begin) => array.slice(begin, begin + size));
93
+ }
94
+
95
+ public async load(): Promise<void> {
96
+ if (this.loadPromise) {
97
+ return this.loadPromise;
98
+ }
99
+ this.loadPromise = new Promise((resolver) => {
100
+ this.loadPromiseResolver = resolver;
101
+ });
102
+
103
+ const chunks = this.chunks(
104
+ Array.from(this.accountsToLoad.values()),
105
+ GET_MULTIPLE_ACCOUNTS_CHUNK_SIZE
106
+ );
107
+
108
+ await Promise.all(
109
+ chunks.map((chunk) => {
110
+ return this.loadChunk(chunk);
111
+ })
112
+ );
113
+
114
+ this.loadPromiseResolver();
115
+ this.loadPromise = undefined;
116
+ }
117
+
118
+ async loadChunk(accountsToLoad: AccountToLoad[]): Promise<void> {
119
+ if (accountsToLoad.length === 0) {
120
+ return;
121
+ }
122
+
123
+ const args = [
124
+ accountsToLoad.map((accountToLoad) => {
125
+ return accountToLoad.publicKey.toBase58();
126
+ }),
127
+ { commitment: this.commitment },
128
+ ];
129
+
130
+ let rpcResponse;
131
+ try {
132
+ // @ts-ignore
133
+ rpcResponse = await this.connection._rpcRequest(
134
+ 'getMultipleAccounts',
135
+ args
136
+ );
137
+ } catch (e) {
138
+ for (const [_, callback] of this.errorCallbacks) {
139
+ callback(e);
140
+ }
141
+ return;
142
+ }
143
+
144
+ const newSlot = rpcResponse.result.context.slot;
145
+
146
+ for (const i in accountsToLoad) {
147
+ const accountToLoad = accountsToLoad[i];
148
+ const key = accountToLoad.publicKey.toString();
149
+ const oldRPCResponse = this.accountData.get(key);
150
+
151
+ let newBuffer: Buffer | undefined = undefined;
152
+ if (rpcResponse.result.value[i]) {
153
+ const raw: string = rpcResponse.result.value[i].data[0];
154
+ const dataType = rpcResponse.result.value[i].data[1];
155
+ newBuffer = Buffer.from(raw, dataType);
156
+ }
157
+
158
+ if (!oldRPCResponse) {
159
+ this.accountData.set(key, {
160
+ slot: newSlot,
161
+ buffer: newBuffer,
162
+ });
163
+ this.handleAccountCallbacks(accountToLoad, newBuffer);
164
+ continue;
165
+ }
166
+
167
+ if (newSlot <= oldRPCResponse.slot) {
168
+ continue;
169
+ }
170
+
171
+ const oldBuffer = oldRPCResponse.buffer;
172
+ if (newBuffer && (!oldBuffer || !newBuffer.equals(oldBuffer))) {
173
+ this.accountData.set(key, {
174
+ slot: newSlot,
175
+ buffer: newBuffer,
176
+ });
177
+ this.handleAccountCallbacks(accountToLoad, newBuffer);
178
+ }
179
+ }
180
+ }
181
+
182
+ handleAccountCallbacks(accountToLoad: AccountToLoad, buffer: Buffer): void {
183
+ for (const [_, callback] of accountToLoad.callbacks) {
184
+ callback(buffer);
185
+ }
186
+ }
187
+
188
+ public getAccountData(publicKey: PublicKey): Buffer | undefined {
189
+ const accountData = this.accountData.get(publicKey.toString());
190
+ return accountData?.buffer;
191
+ }
192
+
193
+ public startPolling(): void {
194
+ if (this.intervalId) {
195
+ return;
196
+ }
197
+
198
+ this.intervalId = setInterval(this.load.bind(this), this.pollingFrequency);
199
+ }
200
+
201
+ public stopPolling(): void {
202
+ if (this.intervalId) {
203
+ clearInterval(this.intervalId);
204
+ this.intervalId = undefined;
205
+ }
206
+ }
207
+ }
@@ -0,0 +1,28 @@
1
+ import { ClearingHouseUser } from '../clearingHouseUser';
2
+ import { BulkAccountLoader } from './bulkAccountLoader';
3
+ import { PollingUserAccountSubscriber } from './pollingUserAccountSubscriber';
4
+
5
+ /**
6
+ * @param users
7
+ * @param accountLoader
8
+ */
9
+ export async function bulkPollingUserSubscribe(
10
+ users: ClearingHouseUser[],
11
+ accountLoader: BulkAccountLoader
12
+ ): Promise<void> {
13
+ await Promise.all(
14
+ users.map((user) => {
15
+ return (
16
+ user.accountSubscriber as PollingUserAccountSubscriber
17
+ ).addToAccountLoader();
18
+ })
19
+ );
20
+
21
+ await accountLoader.load();
22
+
23
+ await Promise.all(
24
+ users.map(async (user) => {
25
+ return user.subscribe();
26
+ })
27
+ );
28
+ }
@@ -0,0 +1,293 @@
1
+ import {
2
+ AccountToPoll,
3
+ ClearingHouseAccountEvents,
4
+ ClearingHouseAccountSubscriber,
5
+ ClearingHouseAccountTypes,
6
+ NotSubscribedError,
7
+ } from './types';
8
+ import { Program } from '@project-serum/anchor';
9
+ import StrictEventEmitter from 'strict-event-emitter-types';
10
+ import { EventEmitter } from 'events';
11
+ import {
12
+ DepositHistoryAccount,
13
+ ExtendedCurveHistoryAccount,
14
+ FundingPaymentHistoryAccount,
15
+ FundingRateHistoryAccount,
16
+ LiquidationHistoryAccount,
17
+ MarketsAccount,
18
+ StateAccount,
19
+ TradeHistoryAccount,
20
+ } from '../types';
21
+ import { getClearingHouseStateAccountPublicKey } from '../addresses';
22
+ import { BulkAccountLoader } from './bulkAccountLoader';
23
+ import { capitalize } from './utils';
24
+ import { ClearingHouseConfigType } from '../factory/clearingHouse';
25
+
26
+ export class PollingClearingHouseAccountSubscriber
27
+ implements ClearingHouseAccountSubscriber
28
+ {
29
+ isSubscribed: boolean;
30
+ program: Program;
31
+ eventEmitter: StrictEventEmitter<EventEmitter, ClearingHouseAccountEvents>;
32
+
33
+ accountLoader: BulkAccountLoader;
34
+ accountsToPoll = new Map<string, AccountToPoll>();
35
+ errorCallbackId?: string;
36
+
37
+ state?: StateAccount;
38
+ markets?: MarketsAccount;
39
+ tradeHistory?: TradeHistoryAccount;
40
+ depositHistory?: DepositHistoryAccount;
41
+ fundingPaymentHistory?: FundingPaymentHistoryAccount;
42
+ fundingRateHistory?: FundingRateHistoryAccount;
43
+ liquidationHistory?: LiquidationHistoryAccount;
44
+ extendedCurveHistory: ExtendedCurveHistoryAccount;
45
+
46
+ optionalExtraSubscriptions: ClearingHouseAccountTypes[] = [];
47
+
48
+ type: ClearingHouseConfigType = 'polling';
49
+
50
+ private isSubscribing = false;
51
+ private subscriptionPromise: Promise<boolean>;
52
+ private subscriptionPromiseResolver: (val: boolean) => void;
53
+
54
+ public constructor(program: Program, accountLoader: BulkAccountLoader) {
55
+ this.isSubscribed = false;
56
+ this.program = program;
57
+ this.eventEmitter = new EventEmitter();
58
+ this.accountLoader = accountLoader;
59
+ }
60
+
61
+ public async subscribe(
62
+ optionalSubscriptions?: ClearingHouseAccountTypes[]
63
+ ): Promise<boolean> {
64
+ if (this.isSubscribed) {
65
+ return true;
66
+ }
67
+
68
+ if (this.isSubscribing) {
69
+ return await this.subscriptionPromise;
70
+ }
71
+
72
+ this.optionalExtraSubscriptions = optionalSubscriptions;
73
+
74
+ this.isSubscribing = true;
75
+
76
+ this.subscriptionPromise = new Promise((res) => {
77
+ this.subscriptionPromiseResolver = res;
78
+ });
79
+
80
+ await this.updateAccountsToPoll();
81
+ await this.addToAccountLoader();
82
+ await this.fetch();
83
+ this.eventEmitter.emit('update');
84
+
85
+ this.isSubscribing = false;
86
+ this.isSubscribed = true;
87
+ this.subscriptionPromiseResolver(true);
88
+
89
+ return true;
90
+ }
91
+
92
+ async updateAccountsToPoll(): Promise<void> {
93
+ if (this.accountsToPoll.size > 0) {
94
+ return;
95
+ }
96
+
97
+ const statePublicKey = await getClearingHouseStateAccountPublicKey(
98
+ this.program.programId
99
+ );
100
+
101
+ const state = (await this.program.account.state.fetch(
102
+ statePublicKey
103
+ )) as StateAccount;
104
+
105
+ this.accountsToPoll.set(statePublicKey.toString(), {
106
+ key: 'state',
107
+ publicKey: statePublicKey,
108
+ eventType: 'stateAccountUpdate',
109
+ });
110
+
111
+ this.accountsToPoll.set(state.markets.toString(), {
112
+ key: 'markets',
113
+ publicKey: state.markets,
114
+ eventType: 'marketsAccountUpdate',
115
+ });
116
+
117
+ if (this.optionalExtraSubscriptions?.includes('tradeHistoryAccount')) {
118
+ this.accountsToPoll.set(state.tradeHistory.toString(), {
119
+ key: 'tradeHistory',
120
+ publicKey: state.tradeHistory,
121
+ eventType: 'tradeHistoryAccountUpdate',
122
+ });
123
+ }
124
+
125
+ if (this.optionalExtraSubscriptions?.includes('depositHistoryAccount')) {
126
+ this.accountsToPoll.set(state.depositHistory.toString(), {
127
+ key: 'depositHistory',
128
+ publicKey: state.depositHistory,
129
+ eventType: 'depositHistoryAccountUpdate',
130
+ });
131
+ }
132
+
133
+ if (
134
+ this.optionalExtraSubscriptions?.includes('fundingPaymentHistoryAccount')
135
+ ) {
136
+ this.accountsToPoll.set(state.fundingPaymentHistory.toString(), {
137
+ key: 'fundingPaymentHistory',
138
+ publicKey: state.fundingPaymentHistory,
139
+ eventType: 'fundingPaymentHistoryAccountUpdate',
140
+ });
141
+ }
142
+
143
+ if (
144
+ this.optionalExtraSubscriptions?.includes('fundingRateHistoryAccount')
145
+ ) {
146
+ this.accountsToPoll.set(state.fundingRateHistory.toString(), {
147
+ key: 'fundingRateHistory',
148
+ publicKey: state.fundingRateHistory,
149
+ eventType: 'fundingRateHistoryAccountUpdate',
150
+ });
151
+ }
152
+
153
+ if (this.optionalExtraSubscriptions?.includes('curveHistoryAccount')) {
154
+ this.accountsToPoll.set(state.extendedCurveHistory.toString(), {
155
+ key: 'extendedCurveHistory',
156
+ publicKey: state.extendedCurveHistory,
157
+ eventType: 'curveHistoryAccountUpdate',
158
+ });
159
+ }
160
+
161
+ if (
162
+ this.optionalExtraSubscriptions?.includes('liquidationHistoryAccount')
163
+ ) {
164
+ this.accountsToPoll.set(state.liquidationHistory.toString(), {
165
+ key: 'liquidationHistory',
166
+ publicKey: state.liquidationHistory,
167
+ eventType: 'liquidationHistoryAccountUpdate',
168
+ });
169
+ }
170
+ }
171
+
172
+ async addToAccountLoader(): Promise<void> {
173
+ for (const [_, accountToPoll] of this.accountsToPoll) {
174
+ accountToPoll.callbackId = this.accountLoader.addAccount(
175
+ accountToPoll.publicKey,
176
+ (buffer) => {
177
+ const account = this.program.account[
178
+ accountToPoll.key
179
+ ].coder.accounts.decode(capitalize(accountToPoll.key), buffer);
180
+ this[accountToPoll.key] = account;
181
+ // @ts-ignore
182
+ this.eventEmitter.emit(accountToPoll.eventType, account);
183
+ this.eventEmitter.emit('update');
184
+ }
185
+ );
186
+ }
187
+
188
+ this.errorCallbackId = this.accountLoader.addErrorCallbacks((error) => {
189
+ this.eventEmitter.emit('error', error);
190
+ });
191
+ }
192
+
193
+ public async fetch(): Promise<void> {
194
+ await this.accountLoader.load();
195
+ for (const [_, accountToPoll] of this.accountsToPoll) {
196
+ const buffer = this.accountLoader.getAccountData(accountToPoll.publicKey);
197
+ if (buffer) {
198
+ this[accountToPoll.key] = this.program.account[
199
+ accountToPoll.key
200
+ ].coder.accounts.decode(capitalize(accountToPoll.key), buffer);
201
+ }
202
+ }
203
+ }
204
+
205
+ public async unsubscribe(): Promise<void> {
206
+ if (!this.isSubscribed) {
207
+ return;
208
+ }
209
+
210
+ for (const [_, accountToPoll] of this.accountsToPoll) {
211
+ this.accountLoader.removeAccount(
212
+ accountToPoll.publicKey,
213
+ accountToPoll.callbackId
214
+ );
215
+ }
216
+
217
+ this.accountLoader.removeErrorCallbacks(this.errorCallbackId);
218
+ this.errorCallbackId = undefined;
219
+
220
+ this.accountsToPoll.clear();
221
+ this.isSubscribed = false;
222
+ }
223
+
224
+ assertIsSubscribed(): void {
225
+ if (!this.isSubscribed) {
226
+ throw new NotSubscribedError(
227
+ 'You must call `subscribe` before using this function'
228
+ );
229
+ }
230
+ }
231
+
232
+ assertOptionalIsSubscribed(
233
+ optionalSubscription: ClearingHouseAccountTypes
234
+ ): void {
235
+ if (!this.isSubscribed) {
236
+ throw new NotSubscribedError(
237
+ 'You must call `subscribe` before using this function'
238
+ );
239
+ }
240
+
241
+ if (!this.optionalExtraSubscriptions.includes(optionalSubscription)) {
242
+ throw new NotSubscribedError(
243
+ `You need to subscribe to the optional Clearing House account "${optionalSubscription}" to use this method`
244
+ );
245
+ }
246
+ }
247
+
248
+ public getStateAccount(): StateAccount {
249
+ this.assertIsSubscribed();
250
+ return this.state;
251
+ }
252
+
253
+ public getMarketsAccount(): MarketsAccount {
254
+ this.assertIsSubscribed();
255
+ return this.markets;
256
+ }
257
+
258
+ public getTradeHistoryAccount(): TradeHistoryAccount {
259
+ this.assertIsSubscribed();
260
+ this.assertOptionalIsSubscribed('tradeHistoryAccount');
261
+ return this.tradeHistory;
262
+ }
263
+
264
+ public getDepositHistoryAccount(): DepositHistoryAccount {
265
+ this.assertIsSubscribed();
266
+ this.assertOptionalIsSubscribed('depositHistoryAccount');
267
+ return this.depositHistory;
268
+ }
269
+
270
+ public getFundingPaymentHistoryAccount(): FundingPaymentHistoryAccount {
271
+ this.assertIsSubscribed();
272
+ this.assertOptionalIsSubscribed('fundingPaymentHistoryAccount');
273
+ return this.fundingPaymentHistory;
274
+ }
275
+
276
+ public getFundingRateHistoryAccount(): FundingRateHistoryAccount {
277
+ this.assertIsSubscribed();
278
+ this.assertOptionalIsSubscribed('fundingRateHistoryAccount');
279
+ return this.fundingRateHistory;
280
+ }
281
+
282
+ public getCurveHistoryAccount(): ExtendedCurveHistoryAccount {
283
+ this.assertIsSubscribed();
284
+ this.assertOptionalIsSubscribed('curveHistoryAccount');
285
+ return this.extendedCurveHistory;
286
+ }
287
+
288
+ public getLiquidationHistoryAccount(): LiquidationHistoryAccount {
289
+ this.assertIsSubscribed();
290
+ this.assertOptionalIsSubscribed('liquidationHistoryAccount');
291
+ return this.liquidationHistory;
292
+ }
293
+ }