@drift-labs/sdk 2.20.0-beta.1 → 2.20.0-beta.3

package/lib/addresses/pda.d.ts

@@ -16,3 +16,4 @@ export declare function getDriftSignerPublicKey(programId: PublicKey): PublicKey
 export declare function getSerumOpenOrdersPublicKey(programId: PublicKey, market: PublicKey): PublicKey;
 export declare function getSerumSignerPublicKey(programId: PublicKey, market: PublicKey, nonce: BN): PublicKey;
 export declare function getSerumFulfillmentConfigPublicKey(programId: PublicKey, market: PublicKey): PublicKey;
+export declare function getReferrerNamePublicKeySync(programId: PublicKey, nameBuffer: number[]): PublicKey;

package/lib/addresses/pda.js

@@ -23,7 +23,7 @@ var __importStar = (this && this.__importStar) || function (mod) {
     return result;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.getSerumFulfillmentConfigPublicKey = exports.getSerumSignerPublicKey = exports.getSerumOpenOrdersPublicKey = exports.getDriftSignerPublicKey = exports.getInsuranceFundStakeAccountPublicKey = exports.getInsuranceFundVaultPublicKey = exports.getSpotMarketVaultPublicKey = exports.getSpotMarketPublicKey = exports.getPerpMarketPublicKey = exports.getUserStatsAccountPublicKey = exports.getUserAccountPublicKeySync = exports.getUserAccountPublicKey = exports.getUserAccountPublicKeyAndNonce = exports.getDriftStateAccountPublicKey = exports.getDriftStateAccountPublicKeyAndNonce = void 0;
+exports.getReferrerNamePublicKeySync = exports.getSerumFulfillmentConfigPublicKey = exports.getSerumSignerPublicKey = exports.getSerumOpenOrdersPublicKey = exports.getDriftSignerPublicKey = exports.getInsuranceFundStakeAccountPublicKey = exports.getInsuranceFundVaultPublicKey = exports.getSpotMarketVaultPublicKey = exports.getSpotMarketPublicKey = exports.getPerpMarketPublicKey = exports.getUserStatsAccountPublicKey = exports.getUserAccountPublicKeySync = exports.getUserAccountPublicKey = exports.getUserAccountPublicKeyAndNonce = exports.getDriftStateAccountPublicKey = exports.getDriftStateAccountPublicKeyAndNonce = void 0;
 const web3_js_1 = require("@solana/web3.js");
 const anchor = __importStar(require("@project-serum/anchor"));
 async function getDriftStateAccountPublicKeyAndNonce(programId) {
@@ -119,3 +119,10 @@ function getSerumFulfillmentConfigPublicKey(programId, market) {
     ], programId)[0];
 }
 exports.getSerumFulfillmentConfigPublicKey = getSerumFulfillmentConfigPublicKey;
+function getReferrerNamePublicKeySync(programId, nameBuffer) {
+    return web3_js_1.PublicKey.findProgramAddressSync([
+        Buffer.from(anchor.utils.bytes.utf8.encode('referrer_name')),
+        Buffer.from(nameBuffer),
+    ], programId)[0];
+}
+exports.getReferrerNamePublicKeySync = getReferrerNamePublicKeySync;

package/lib/constants/perpMarkets.js

@@ -54,6 +54,16 @@ exports.DevnetPerpMarkets = [
         launchTs: 1677068931000,
         oracleSource: __1.OracleSource.PYTH_1M,
     },
+    {
+        fullName: 'Polygon',
+        category: ['L2', 'Infra'],
+        symbol: 'MATIC-PERP',
+        baseAssetSymbol: 'MATIC',
+        marketIndex: 5,
+        oracle: new web3_js_1.PublicKey('FBirwuDFuRAu4iSGc7RGxN5koHB7EJM1wbCmyPuQoGur'),
+        launchTs: 1677690149000,
+        oracleSource: __1.OracleSource.PYTH,
+    },
 ];
 exports.MainnetPerpMarkets = [
     {
@@ -106,6 +116,16 @@ exports.MainnetPerpMarkets = [
         launchTs: 1677690149000,
         oracleSource: __1.OracleSource.PYTH_1M,
     },
+    {
+        fullName: 'Polygon',
+        category: ['L2', 'Infra'],
+        symbol: 'MATIC-PERP',
+        baseAssetSymbol: 'MATIC',
+        marketIndex: 5,
+        oracle: new web3_js_1.PublicKey('7KVswB9vkCgeM3SHP7aGDijvdRAHK8P5wi9JXViCrtYh'),
+        launchTs: 1677690149000,
+        oracleSource: __1.OracleSource.PYTH,
+    },
 ];
 exports.PerpMarkets = {
     devnet: exports.DevnetPerpMarkets,

package/lib/driftClient.d.ts

@@ -1,7 +1,7 @@
 /// <reference types="node" />
 /// <reference types="bn.js" />
 import { AnchorProvider, BN, Program } from '@project-serum/anchor';
-import { StateAccount, IWallet, PositionDirection, UserAccount, PerpMarketAccount, OrderParams, Order, SpotMarketAccount, SpotPosition, MakerInfo, TakerInfo, OptionalOrderParams, OrderType, ReferrerInfo, MarketType, TxParams, SerumV3FulfillmentConfigAccount, OrderTriggerCondition, PerpMarketExtendedInfo } from './types';
+import { StateAccount, IWallet, PositionDirection, UserAccount, PerpMarketAccount, OrderParams, Order, SpotMarketAccount, SpotPosition, MakerInfo, TakerInfo, OptionalOrderParams, OrderType, ReferrerInfo, MarketType, TxParams, SerumV3FulfillmentConfigAccount, ReferrerNameAccount, OrderTriggerCondition, PerpMarketExtendedInfo } from './types';
 import * as anchor from '@project-serum/anchor';
 import { Connection, PublicKey, TransactionSignature, ConfirmOptions, Transaction, TransactionInstruction, AccountMeta, Signer, AddressLookupTableAccount } from '@solana/web3.js';
 import { TokenFaucet } from './tokenFaucet';
@@ -96,6 +96,7 @@ export declare class DriftClient {
     initializeUserAccount(subAccountId?: number, name?: string, referrerInfo?: ReferrerInfo): Promise<[TransactionSignature, PublicKey]>;
     getInitializeUserInstructions(subAccountId?: number, name?: string, referrerInfo?: ReferrerInfo): Promise<[PublicKey, TransactionInstruction]>;
     getInitializeUserStatsIx(): Promise<TransactionInstruction>;
+    initializeReferrerName(name: string): Promise<TransactionSignature>;
     updateUserName(name: string, subAccountId?: number): Promise<TransactionSignature>;
     updateUserCustomMarginRatio(marginRatio: number, subAccountId?: number): Promise<TransactionSignature>;
     updateUserMarginTradingEnabled(marginTradingEnabled: boolean, subAccountId?: number): Promise<TransactionSignature>;
@@ -106,6 +107,7 @@ export declare class DriftClient {
     getUser(subAccountId?: number): User;
     getUsers(): User[];
     getUserStats(): UserStats;
+    fetchReferrerNameAccount(name: string): Promise<ReferrerNameAccount | undefined>;
     userStatsAccountPublicKey: PublicKey;
     getUserStatsAccountPublicKey(): PublicKey;
     getUserAccountPublicKey(): Promise<PublicKey>;

package/lib/driftClient.js

@@ -358,6 +358,24 @@ class DriftClient {
             },
         });
     }
+    async initializeReferrerName(name) {
+        const userAccountPublicKey = (0, pda_1.getUserAccountPublicKeySync)(this.program.programId, this.wallet.publicKey, 0);
+        const nameBuffer = (0, userName_1.encodeName)(name);
+        const referrerNameAccountPublicKey = await (0, pda_1.getReferrerNamePublicKeySync)(this.program.programId, nameBuffer);
+        const tx = await this.program.transaction.initializeReferrerName(nameBuffer, {
+            accounts: {
+                referrerName: referrerNameAccountPublicKey,
+                user: userAccountPublicKey,
+                authority: this.wallet.publicKey,
+                userStats: this.getUserStatsAccountPublicKey(),
+                payer: this.wallet.publicKey,
+                rent: anchor.web3.SYSVAR_RENT_PUBKEY,
+                systemProgram: anchor.web3.SystemProgram.programId,
+            },
+        });
+        const { txSig } = await this.sendTransaction(tx, [], this.opts);
+        return txSig;
+    }
     async updateUserName(name, subAccountId = 0) {
         const userAccountPublicKey = (0, pda_1.getUserAccountPublicKeySync)(this.program.programId, this.wallet.publicKey, subAccountId);
         const nameBuffer = (0, userName_1.encodeName)(name);
@@ -459,6 +477,11 @@ class DriftClient {
     getUserStats() {
         return this.userStats;
     }
+    async fetchReferrerNameAccount(name) {
+        const nameBuffer = (0, userName_1.encodeName)(name);
+        const referrerNameAccountPublicKey = (0, pda_1.getReferrerNamePublicKeySync)(this.program.programId, nameBuffer);
+        return (await this.program.account.referrerName.fetch(referrerNameAccountPublicKey));
+    }
     getUserStatsAccountPublicKey() {
         if (this.userStatsAccountPublicKey) {
             return this.userStatsAccountPublicKey;

package/lib/idl/drift.json

@@ -1,5 +1,5 @@
 {
-    "version": "2.20.0-beta.1",
+    "version": "2.20.0-beta.3",
     "name": "drift",
     "instructions": [
         {
@@ -93,6 +93,57 @@
             ],
             "args": []
         },
+        {
+            "name": "initializeReferrerName",
+            "accounts": [
+                {
+                    "name": "referrerName",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "user",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "userStats",
+                    "isMut": true,
+                    "isSigner": false
+                },
+                {
+                    "name": "authority",
+                    "isMut": false,
+                    "isSigner": true
+                },
+                {
+                    "name": "payer",
+                    "isMut": true,
+                    "isSigner": true
+                },
+                {
+                    "name": "rent",
+                    "isMut": false,
+                    "isSigner": false
+                },
+                {
+                    "name": "systemProgram",
+                    "isMut": false,
+                    "isSigner": false
+                }
+            ],
+            "args": [
+                {
+                    "name": "name",
+                    "type": {
+                        "array": [
+                            "u8",
+                            32
+                        ]
+                    }
+                }
+            ]
+        },
         {
             "name": "deposit",
             "accounts": [
@@ -4607,6 +4658,35 @@
                     }
                 ]
             }
+        },
+        {
+            "name": "ReferrerName",
+            "type": {
+                "kind": "struct",
+                "fields": [
+                    {
+                        "name": "authority",
+                        "type": "publicKey"
+                    },
+                    {
+                        "name": "user",
+                        "type": "publicKey"
+                    },
+                    {
+                        "name": "userStats",
+                        "type": "publicKey"
+                    },
+                    {
+                        "name": "name",
+                        "type": {
+                            "array": [
+                                "u8",
+                                32
+                            ]
+                        }
+                    }
+                ]
+            }
         }
     ],
     "types": [
@@ -6208,9 +6288,7 @@
                         "name": "Match",
                         "fields": [
                             "publicKey",
-                            {
-                                "defined": "usize"
-                            }
+                            "u16"
                         ]
                     }
                 ]

package/lib/types.d.ts

@@ -1010,6 +1010,12 @@ export type SerumV3FulfillmentConfigAccount = {
     serumOpenOrders: PublicKey;
     serumSignerNonce: BN;
 };
+export type ReferrerNameAccount = {
+    name: number[];
+    user: PublicKey;
+    authority: PublicKey;
+    userStats: PublicKey;
+};
 export type PerpMarketExtendedInfo = {
     marketIndex: number;
     /**

package/package.json

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

package/src/addresses/pda.ts

@@ -188,3 +188,16 @@ export function getSerumFulfillmentConfigPublicKey(
 		programId
 	)[0];
 }
+
+export function getReferrerNamePublicKeySync(
+	programId: PublicKey,
+	nameBuffer: number[]
+): PublicKey {
+	return PublicKey.findProgramAddressSync(
+		[
+			Buffer.from(anchor.utils.bytes.utf8.encode('referrer_name')),
+			Buffer.from(nameBuffer),
+		],
+		programId
+	)[0];
+}

package/src/constants/perpMarkets.ts

@@ -64,6 +64,16 @@ export const DevnetPerpMarkets: PerpMarketConfig[] = [
 		launchTs: 1677068931000,
 		oracleSource: OracleSource.PYTH_1M,
 	},
+	{
+		fullName: 'Polygon',
+		category: ['L2', 'Infra'],
+		symbol: 'MATIC-PERP',
+		baseAssetSymbol: 'MATIC',
+		marketIndex: 5,
+		oracle: new PublicKey('FBirwuDFuRAu4iSGc7RGxN5koHB7EJM1wbCmyPuQoGur'),
+		launchTs: 1677690149000, //todo
+		oracleSource: OracleSource.PYTH,
+	},
 ];
 
 export const MainnetPerpMarkets: PerpMarketConfig[] = [
@@ -117,6 +127,16 @@ export const MainnetPerpMarkets: PerpMarketConfig[] = [
 		launchTs: 1677690149000,
 		oracleSource: OracleSource.PYTH_1M,
 	},
+	{
+		fullName: 'Polygon',
+		category: ['L2', 'Infra'],
+		symbol: 'MATIC-PERP',
+		baseAssetSymbol: 'MATIC',
+		marketIndex: 5,
+		oracle: new PublicKey('7KVswB9vkCgeM3SHP7aGDijvdRAHK8P5wi9JXViCrtYh'),
+		launchTs: 1677690149000, //todo
+		oracleSource: OracleSource.PYTH,
+	},
 ];
 
 export const PerpMarkets: { [key in DriftEnv]: PerpMarketConfig[] } = {

package/src/driftClient.ts

@@ -25,6 +25,7 @@ import {
 	TxParams,
 	SerumV3FulfillmentConfigAccount,
 	isVariant,
+	ReferrerNameAccount,
 	OrderTriggerCondition,
 	isOneOfVariant,
 	PostOnlyParams,
@@ -58,6 +59,7 @@ import {
 	getDriftStateAccountPublicKey,
 	getInsuranceFundStakeAccountPublicKey,
 	getPerpMarketPublicKey,
+	getReferrerNamePublicKeySync,
 	getSerumFulfillmentConfigPublicKey,
 	getSerumSignerPublicKey,
 	getSpotMarketPublicKey,
@@ -584,6 +586,40 @@ export class DriftClient {
 		});
 	}
 
+	public async initializeReferrerName(
+		name: string
+	): Promise<TransactionSignature> {
+		const userAccountPublicKey = getUserAccountPublicKeySync(
+			this.program.programId,
+			this.wallet.publicKey,
+			0
+		);
+
+		const nameBuffer = encodeName(name);
+
+		const referrerNameAccountPublicKey = await getReferrerNamePublicKeySync(
+			this.program.programId,
+			nameBuffer
+		);
+
+		const tx = await this.program.transaction.initializeReferrerName(
+			nameBuffer,
+			{
+				accounts: {
+					referrerName: referrerNameAccountPublicKey,
+					user: userAccountPublicKey,
+					authority: this.wallet.publicKey,
+					userStats: this.getUserStatsAccountPublicKey(),
+					payer: this.wallet.publicKey,
+					rent: anchor.web3.SYSVAR_RENT_PUBKEY,
+					systemProgram: anchor.web3.SystemProgram.programId,
+				},
+			}
+		);
+		const { txSig } = await this.sendTransaction(tx, [], this.opts);
+		return txSig;
+	}
+
 	public async updateUserName(
 		name: string,
 		subAccountId = 0
@@ -760,6 +796,19 @@ export class DriftClient {
 		return this.userStats;
 	}
 
+	public async fetchReferrerNameAccount(
+		name: string
+	): Promise<ReferrerNameAccount | undefined> {
+		const nameBuffer = encodeName(name);
+		const referrerNameAccountPublicKey = getReferrerNamePublicKeySync(
+			this.program.programId,
+			nameBuffer
+		);
+		return (await this.program.account.referrerName.fetch(
+			referrerNameAccountPublicKey
+		)) as ReferrerNameAccount;
+	}
+
 	userStatsAccountPublicKey: PublicKey;
 	public getUserStatsAccountPublicKey(): PublicKey {
 		if (this.userStatsAccountPublicKey) {

package/src/idl/drift.json

@@ -1,5 +1,5 @@
 {
-  "version": "2.20.0-beta.1",
+  "version": "2.20.0-beta.3",
   "name": "drift",
   "instructions": [
     {
@@ -93,6 +93,57 @@
       ],
       "args": []
     },
+    {
+      "name": "initializeReferrerName",
+      "accounts": [
+        {
+          "name": "referrerName",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "user",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "userStats",
+          "isMut": true,
+          "isSigner": false
+        },
+        {
+          "name": "authority",
+          "isMut": false,
+          "isSigner": true
+        },
+        {
+          "name": "payer",
+          "isMut": true,
+          "isSigner": true
+        },
+        {
+          "name": "rent",
+          "isMut": false,
+          "isSigner": false
+        },
+        {
+          "name": "systemProgram",
+          "isMut": false,
+          "isSigner": false
+        }
+      ],
+      "args": [
+        {
+          "name": "name",
+          "type": {
+            "array": [
+              "u8",
+              32
+            ]
+          }
+        }
+      ]
+    },
     {
       "name": "deposit",
       "accounts": [
@@ -4607,6 +4658,35 @@
           }
         ]
       }
+    },
+    {
+      "name": "ReferrerName",
+      "type": {
+        "kind": "struct",
+        "fields": [
+          {
+            "name": "authority",
+            "type": "publicKey"
+          },
+          {
+            "name": "user",
+            "type": "publicKey"
+          },
+          {
+            "name": "userStats",
+            "type": "publicKey"
+          },
+          {
+            "name": "name",
+            "type": {
+              "array": [
+                "u8",
+                32
+              ]
+            }
+          }
+        ]
+      }
     }
   ],
   "types": [
@@ -6208,9 +6288,7 @@
             "name": "Match",
             "fields": [
               "publicKey",
-              {
-                "defined": "usize"
-              }
+              "u16"
             ]
           }
         ]

package/src/types.ts

@@ -999,6 +999,13 @@ export type SerumV3FulfillmentConfigAccount = {
 	serumSignerNonce: BN;
 };
 
+export type ReferrerNameAccount = {
+	name: number[];
+	user: PublicKey;
+	authority: PublicKey;
+	userStats: PublicKey;
+};
+
 export type PerpMarketExtendedInfo = {
 	marketIndex: number;
 	/**