@scallop-io/sui-scallop-sdk 0.45.0 → 0.46.1

package/dist/models/scallopCache.d.ts

@@ -1,7 +1,7 @@
 import { QueryClient, QueryClientConfig } from '@tanstack/query-core';
 import { SuiTxArg } from '@scallop-io/sui-kit';
 import { SuiKit } from '@scallop-io/sui-kit';
-import type { SuiObjectResponse, SuiObjectDataOptions, SuiObjectData, PaginatedObjectsResponse, GetOwnedObjectsParams, DevInspectResults, GetDynamicFieldsParams, DynamicFieldPage, GetDynamicFieldObjectParams } from '@mysten/sui.js/client';
+import type { SuiObjectResponse, SuiObjectDataOptions, SuiObjectData, PaginatedObjectsResponse, GetOwnedObjectsParams, DevInspectResults, GetDynamicFieldsParams, DynamicFieldPage, GetDynamicFieldObjectParams, GetBalanceParams } from '@mysten/sui.js/client';
 type QueryInspectTxnParams = {
     queryTarget: string;
     args: SuiTxArg[];
@@ -38,7 +38,7 @@ export declare class ScallopCache {
      * @description Cache protocol config call for 60 seconds.
      * @returns Promise<ProtocolConfig>
      */
-    private getProtocolConfig;
+    getProtocolConfig(): Promise<import("@mysten/sui.js/client").ProtocolConfig>;
     /**
      * @description Provides cache for inspectTxn of the SuiKit.
      * @param QueryInspectTxnParams
@@ -67,5 +67,9 @@ export declare class ScallopCache {
     queryGetOwnedObjects(input: GetOwnedObjectsParams): Promise<PaginatedObjectsResponse>;
     queryGetDynamicFields(input: GetDynamicFieldsParams): Promise<DynamicFieldPage>;
     queryGetDynamicFieldObject(input: GetDynamicFieldObjectParams): Promise<SuiObjectResponse>;
+    queryGetAllCoinBalances(owner: string): Promise<{
+        [k: string]: string;
+    }>;
+    queryGetCoinBalance(input: GetBalanceParams): Promise<string>;
 }
 export {};

package/dist/models/scallopQuery.d.ts

@@ -124,20 +124,7 @@ export declare class ScallopQuery {
      * @param ownerAddress - The owner address.
      * @return All coin amounts.
      */
-    getCoinAmounts(assetCoinNames?: SupportAssetCoins[], ownerAddress?: string): Promise<{
-        eth?: number | undefined;
-        btc?: number | undefined;
-        usdc?: number | undefined;
-        usdt?: number | undefined;
-        sui?: number | undefined;
-        apt?: number | undefined;
-        sol?: number | undefined;
-        cetus?: number | undefined;
-        afsui?: number | undefined;
-        hasui?: number | undefined;
-        vsui?: number | undefined;
-        sca?: number | undefined;
-    }>;
+    getCoinAmounts(assetCoinNames?: SupportAssetCoins[], ownerAddress?: string): Promise<import("../types").OptionalKeys<Record<SupportAssetCoins, number>>>;
     /**
      * Get asset coin amount.
      *
@@ -153,20 +140,7 @@ export declare class ScallopQuery {
      * @param ownerAddress - The owner address.
      * @return All market market coin amounts.
      */
-    getMarketCoinAmounts(marketCoinNames?: SupportMarketCoins[], ownerAddress?: string): Promise<{
-        seth?: number | undefined;
-        ssui?: number | undefined;
-        susdc?: number | undefined;
-        susdt?: number | undefined;
-        scetus?: number | undefined;
-        safsui?: number | undefined;
-        shasui?: number | undefined;
-        svsui?: number | undefined;
-        sbtc?: number | undefined;
-        sapt?: number | undefined;
-        ssol?: number | undefined;
-        ssca?: number | undefined;
-    }>;
+    getMarketCoinAmounts(marketCoinNames?: SupportMarketCoins[], ownerAddress?: string): Promise<import("../types").OptionalKeys<Record<SupportMarketCoins, number>>>;
     /**
      * Get market coin amount.
      *
@@ -182,6 +156,13 @@ export declare class ScallopQuery {
      * @return Asset coin price.
      */
     getPriceFromPyth(assetCoinName: SupportAssetCoins): Promise<number>;
+    /**
+     * Get prices from pyth fee object.
+     *
+     * @param assetCoinNames - Array of supported asset coin names.
+     * @return Array of asset coin prices.
+     */
+    getPricesFromPyth(assetCoinNames: SupportAssetCoins[]): Promise<Record<SupportAssetCoins, number>>;
     /**
      * Get spools data.
      *
@@ -384,6 +365,23 @@ export declare class ScallopQuery {
      * @return Total value locked.
      */
     getTvl(indexer?: boolean): Promise<import("../types").TotalValueLocked>;
+    /**
+     * Get all veSca from walletAdddress
+     * @param walletAddress
+     * @returns array of veSca
+     */
+    getVeScas(walletAddress: string): Promise<(import("../types").Vesca | undefined)[]>;
+    /**
+     * Get total vesca treasury with movecall
+     * @returns Promise<string | undefined>
+     */
+    getTotalVeScaTreasuryAmount(): Promise<string>;
+    /**
+     * Return binded veScaKeyId of walletAddress if exist
+     * @param walletAddress
+     * @returns veScaKeyId
+     */
+    getVeScaKeyIdFromReferralBindings(walletAddress: string): Promise<string | null>;
     /**
      * Get binded obligationId from a veScaKey if it exists.
      * @param veScaKey

package/dist/queries/coreQuery.d.ts

@@ -1,7 +1,7 @@
 import type { SuiObjectData } from '@mysten/sui.js/client';
 import type { SuiAddressArg } from '@scallop-io/sui-kit';
 import type { ScallopQuery } from '../models';
-import { Market, MarketPool, MarketCollateral, SupportAssetCoins, SupportPoolCoins, SupportCollateralCoins, ObligationQueryInterface, Obligation, SupportMarketCoins } from '../types';
+import { Market, MarketPool, MarketCollateral, SupportAssetCoins, SupportPoolCoins, SupportCollateralCoins, ObligationQueryInterface, Obligation, SupportMarketCoins, OptionalKeys } from '../types';
 /**
  * Query market data.
  *
@@ -122,20 +122,7 @@ export declare const queryObligation: (query: ScallopQuery, obligationId: SuiAdd
  * @param ownerAddress - The owner address.
  * @return All owned coin amounts.
  */
-export declare const getCoinAmounts: (query: ScallopQuery, assetCoinNames?: SupportAssetCoins[], ownerAddress?: string) => Promise<{
-    eth?: number | undefined;
-    btc?: number | undefined;
-    usdc?: number | undefined;
-    usdt?: number | undefined;
-    sui?: number | undefined;
-    apt?: number | undefined;
-    sol?: number | undefined;
-    cetus?: number | undefined;
-    afsui?: number | undefined;
-    hasui?: number | undefined;
-    vsui?: number | undefined;
-    sca?: number | undefined;
-}>;
+export declare const getCoinAmounts: (query: ScallopQuery, assetCoinNames?: SupportAssetCoins[], ownerAddress?: string) => Promise<OptionalKeys<Record<SupportAssetCoins, number>>>;
 /**
  * Query owned coin amount.
  *
@@ -153,20 +140,7 @@ export declare const getCoinAmount: (query: ScallopQuery, assetCoinName: Support
  * @param ownerAddress - The owner address.
  * @return All owned market coins amount.
  */
-export declare const getMarketCoinAmounts: (query: ScallopQuery, marketCoinNames?: SupportMarketCoins[], ownerAddress?: string) => Promise<{
-    seth?: number | undefined;
-    ssui?: number | undefined;
-    susdc?: number | undefined;
-    susdt?: number | undefined;
-    scetus?: number | undefined;
-    safsui?: number | undefined;
-    shasui?: number | undefined;
-    svsui?: number | undefined;
-    sbtc?: number | undefined;
-    sapt?: number | undefined;
-    ssol?: number | undefined;
-    ssca?: number | undefined;
-}>;
+export declare const getMarketCoinAmounts: (query: ScallopQuery, marketCoinNames?: SupportMarketCoins[], ownerAddress?: string) => Promise<OptionalKeys<Record<SupportMarketCoins, number>>>;
 /**
  * Query owned market coin (sCoin) amount.
  *

package/dist/queries/index.d.ts

@@ -4,3 +4,4 @@ export * from './borrowIncentiveQuery';
 export * from './priceQuery';
 export * from './portfolioQuery';
 export * from './vescaQuery';
+export * from './referralQuery';

package/dist/queries/priceQuery.d.ts

@@ -1,3 +1,4 @@
+import { SuiObjectData } from '@mysten/sui.js/src/client';
 import type { ScallopQuery } from '../models';
 import type { SupportAssetCoins } from '../types';
 /**
@@ -7,4 +8,5 @@ import type { SupportAssetCoins } from '../types';
  * @param assetCoinName - Specific support asset coin name.
  * @return Asset coin price.
  */
-export declare const getPythPrice: (query: ScallopQuery, assetCoinName: SupportAssetCoins) => Promise<number>;
+export declare const getPythPrice: (query: ScallopQuery, assetCoinName: SupportAssetCoins, priceFeedObject?: SuiObjectData | null) => Promise<number>;
+export declare const getPythPrices: (query: ScallopQuery, assetCoinNames: SupportAssetCoins[]) => Promise<Record<SupportAssetCoins, number>>;

package/dist/queries/referralQuery.d.ts

@@ -0,0 +1,7 @@
+import { ScallopQuery } from 'src/models';
+/**
+ * Query the veScaKeyId from the referral bindings table using the borrower address
+ * @param query
+ * @returns
+ */
+export declare const queryVeScaKeyIdFromReferralBindings: (query: ScallopQuery, refereeAddress: string) => Promise<string | null>;

package/dist/queries/vescaQuery.d.ts

@@ -1,5 +1,5 @@
 import { Vesca } from '../types';
-import type { SuiObjectData } from '@mysten/sui.js/client';
+import { type SuiObjectData } from '@mysten/sui.js/client';
 import type { ScallopQuery } from '../models';
 /**
  * Query all owned veSca key.
@@ -16,7 +16,7 @@ export declare const getVescaKeys: (query: ScallopQuery, ownerAddress?: string)
  * @param ownerAddress - The owner address.
  * @return Owned veScas.
  */
-export declare const getVeScas: (query: ScallopQuery, ownerAddress?: string) => Promise<Vesca[]>;
+export declare const getVeScas: (query: ScallopQuery, ownerAddress?: string) => Promise<(Vesca | undefined)[]>;
 /**
  * Get veSca data.
  *
@@ -26,3 +26,7 @@ export declare const getVeScas: (query: ScallopQuery, ownerAddress?: string) =>
  * @returns Vesca data.
  */
 export declare const getVeSca: (query: ScallopQuery, veScaKeyId?: string, ownerAddress?: string) => Promise<Vesca | undefined>;
+/**
+ * Get current total veSca treasury amount.
+ */
+export declare const getTotalVeScaTreasuryAmount: (query: ScallopQuery) => Promise<string>;

package/dist/types/address.d.ts

@@ -48,6 +48,7 @@ export interface AddressesInterface {
         id: string;
         adminCap: string;
         object: string;
+        config: string;
         pools: Partial<Record<SupportStakeMarketCoins, {
             id: string;
             rewardPoolId: string;
@@ -64,12 +65,26 @@ export interface AddressesInterface {
     };
     vesca: {
         id: string;
+        object: string;
         adminCap: string;
         tableId: string;
         table: string;
         treasury: string;
         config: string;
     };
+    referral: {
+        id: string;
+        version: string;
+        object: string;
+        adminCap: string;
+        referralBindings: string;
+        bindingTableId: string;
+        referralRevenuePool: string;
+        revenueTableId: string;
+        referralTiers: string;
+        tiersTableId: string;
+        authorizedWitnessList: string;
+    };
 }
 type AddressPathsProps<T> = T extends string ? [] : {
     [K in Extract<keyof T, string>]: [K, ...AddressPathsProps<T[K]>];

package/dist/types/builder/core.d.ts

@@ -20,6 +20,7 @@ export type CoreNormalMethods = {
     withdraw: (marketCoin: SuiObjectArg, poolCoinName: SupportPoolCoins) => TransactionResult;
     withdrawEntry: (marketCoin: SuiObjectArg, poolCoinName: SupportPoolCoins) => void;
     borrow: (obligation: SuiAddressArg, obligationKey: SuiAddressArg, amount: SuiTxArg, poolCoinName: SupportPoolCoins) => TransactionResult;
+    borrowWithReferral: (obligation: SuiAddressArg, obligationKey: SuiAddressArg, borrowReferral: SuiObjectArg, amount: SuiTxArg, poolCoinName: SupportPoolCoins) => TransactionResult;
     borrowEntry: (obligation: SuiAddressArg, obligationKey: SuiAddressArg, amount: SuiTxArg, poolCoinName: SupportPoolCoins) => void;
     repay: (obligation: SuiAddressArg, coin: SuiObjectArg, poolCoinName: SupportPoolCoins) => void;
     borrowFlashLoan: (amount: SuiTxArg, poolCoinName: SupportPoolCoins) => TransactionResult;
@@ -29,6 +30,7 @@ export type CoreQuickMethods = {
     addCollateralQuick: (amount: number, collateralCoinName: SupportCollateralCoins, obligationId?: SuiAddressArg) => Promise<void>;
     takeCollateralQuick: (amount: number, collateralCoinName: SupportCollateralCoins, obligationId?: SuiAddressArg, obligationKey?: SuiAddressArg) => Promise<TransactionResult>;
     borrowQuick: (amount: number, poolCoinName: SupportPoolCoins, obligationId?: SuiAddressArg, obligationKey?: SuiAddressArg) => Promise<TransactionResult>;
+    borrowWithReferralQuick: (amount: number, poolCoinName: SupportPoolCoins, borrowReferral: SuiObjectArg, obligationId?: SuiAddressArg, obligationKey?: SuiAddressArg) => Promise<TransactionResult>;
     depositQuick: (amount: number, poolCoinName: SupportPoolCoins) => Promise<TransactionResult>;
     withdrawQuick: (amount: number, poolCoinName: SupportPoolCoins) => Promise<TransactionResult>;
     repayQuick: (amount: number, poolCoinName: SupportPoolCoins, obligationId?: SuiAddressArg) => Promise<void>;

package/dist/types/builder/index.d.ts

@@ -2,8 +2,9 @@ import type { CoreTxBlock } from './core';
 import type { SpoolTxBlock } from './spool';
 import type { BorrowIncentiveTxBlock } from './borrowIncentive';
 import type { VeScaTxBlock } from './vesca';
+import type { ReferralTxBlock } from './referral';
 export type * from './core';
 export type * from './spool';
 export type * from './borrowIncentive';
 export type * from './vesca';
-export type ScallopTxBlock = CoreTxBlock & SpoolTxBlock & BorrowIncentiveTxBlock & VeScaTxBlock;
+export type ScallopTxBlock = CoreTxBlock & SpoolTxBlock & ReferralTxBlock & BorrowIncentiveTxBlock & VeScaTxBlock;

package/dist/types/builder/referral.d.ts

@@ -0,0 +1,30 @@
+import { SuiObjectArg, TransactionResult, SuiTxBlock as SuiKitTxBlock } from '@scallop-io/sui-kit';
+import { ScallopBuilder } from 'src/models';
+import { SupportPoolCoins } from '../constant';
+export type ReferralIds = {
+    referralPgkId: string;
+    referralBindings: string;
+    referralRevenuePool: string;
+    referralTiers: string;
+    authorizedWitnessList: string;
+    version: string;
+};
+export type ReferralNormalMethods = {
+    bindToReferral: (veScaKeyId: string) => void;
+    claimReferralTicket: (poolCoinName: SupportPoolCoins) => TransactionResult;
+    burnReferralTicket: (ticket: SuiObjectArg, poolCoinName: SupportPoolCoins) => void;
+    claimReferralRevenue: (veScaKey: SuiObjectArg, poolCoinName: SupportPoolCoins) => TransactionResult;
+};
+export type ReferralQuickMethods = {
+    claimReferralRevenueQuick: (veScaKey: SuiObjectArg, coinNames: SupportPoolCoins[]) => Promise<void>;
+};
+export type SuiTxBlockWithReferralNormalMethods = SuiKitTxBlock & ReferralNormalMethods;
+export type ReferralTxBlock = SuiTxBlockWithReferralNormalMethods & ReferralQuickMethods;
+export type GenerateReferralNormalMethod = (params: {
+    builder: ScallopBuilder;
+    txBlock: SuiKitTxBlock;
+}) => ReferralNormalMethods;
+export type GenerateReferralQuickMethod = (params: {
+    builder: ScallopBuilder;
+    txBlock: SuiTxBlockWithReferralNormalMethods;
+}) => ReferralQuickMethods;

package/dist/types/builder/vesca.d.ts

@@ -13,6 +13,7 @@ export type VeScaNormalMethods = {
     extendLockAmount: (veScaKey: SuiObjectArg, scaCoin: SuiObjectArg) => void;
     renewExpiredVeSca: (veScaKey: SuiObjectArg, scaCoin: SuiObjectArg, newUnlockAtInSecondTimestamp: SuiTxArg) => void;
     redeemSca: (veScaKey: SuiObjectArg) => TransactionResult;
+    mintEmptyVeSca: () => TransactionResult;
 };
 export type VeScaQuickMethods = {
     lockScaQuick(amountOrCoin?: SuiObjectArg | number, lockPeriodInDays?: number, autoCheck?: boolean): Promise<void>;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@scallop-io/sui-scallop-sdk",
-  "version": "0.45.0",
+  "version": "0.46.1",
   "description": "Typescript sdk for interacting with Scallop contract on SUI",
   "keywords": [
     "sui",
@@ -39,15 +39,16 @@
   ],
   "dependencies": {
     "@mysten/bcs": "^0.8.1",
-    "@mysten/sui.js": "^0.44.0",
+    "@mysten/sui.js": "^0.49.1",
     "@noble/curves": "^1.2.0",
     "@noble/hashes": "^1.3.2",
     "@pythnetwork/price-service-client": "^1.8.2",
-    "@pythnetwork/pyth-sui-js": "^1.2.4",
-    "@scallop-io/sui-kit": "^0.44.2",
+    "@pythnetwork/pyth-sui-js": "2.0.0",
+    "@scallop-io/sui-kit": "^0.45.0",
     "@scure/bip39": "^1.2.1",
     "@tanstack/query-core": "^5.28.0",
     "axios": "^1.6.0",
+    "bech32": "^2.0.0",
     "bignumber.js": "^9.1.2",
     "superstruct": "^1.0.3",
     "tweetnacl": "^1.0.3"
@@ -74,8 +75,8 @@
     "vitest": "^0.34.6"
   },
   "peerDependencies": {
-    "@mysten/sui.js": "^0.44.0",
-    "@scallop-io/sui-kit": "^0.44.0"
+    "@mysten/sui.js": "0.49.1",
+    "@scallop-io/sui-kit": "^0.44.2"
   },
   "lint-staged": {
     "**/*.ts": [

package/src/builders/borrowIncentiveBuilder.ts

@@ -8,7 +8,7 @@ import {
 } from '../queries';
 import { requireSender } from '../utils';
 import type { SuiObjectArg } from '@scallop-io/sui-kit';
-import type { ScallopBuilder } from '../models';
+import type { ScallopBuilder } from 'src/models';
 import type {
   BorrowIncentiveIds,
   GenerateBorrowIncentiveNormalMethod,
@@ -19,10 +19,7 @@ import type {
   VescaIds,
 } from '../types';
 import { requireVeSca } from './vescaBuilder';
-import {
-  IS_VE_SCA_TEST,
-  OLD_BORROW_INCENTIVE_PROTOCOL_ID,
-} from 'src/constants';
+import { OLD_BORROW_INCENTIVE_PROTOCOL_ID } from 'src/constants';
 
 /**
  * Check and get Obligation information from transaction block.
@@ -88,9 +85,7 @@ const requireObligationInfo = async (
 const generateBorrowIncentiveNormalMethod: GenerateBorrowIncentiveNormalMethod =
   ({ builder, txBlock }) => {
     const borrowIncentiveIds: BorrowIncentiveIds = {
-      borrowIncentivePkg: IS_VE_SCA_TEST
-        ? '0x4d5a7cefa4147b4ace0ca845b20437d6ac0d32e5f2f855171f745472c2576246'
-        : builder.address.get('borrowIncentive.id'),
+      borrowIncentivePkg: builder.address.get('borrowIncentive.id'),
       query: builder.address.get('borrowIncentive.query'),
       config: builder.address.get('borrowIncentive.config'),
       incentivePools: builder.address.get('borrowIncentive.incentivePools'),
@@ -228,11 +223,7 @@ const generateBorrowIncentiveQuickMethod: GenerateBorrowIncentiveQuickMethod =
               (txn.target ===
                 `${OLD_BORROW_INCENTIVE_PROTOCOL_ID}::user::unstake` ||
                 txn.target ===
-                  (IS_VE_SCA_TEST
-                    ? `${'0x4d5a7cefa4147b4ace0ca845b20437d6ac0d32e5f2f855171f745472c2576246'}::user::unstake`
-                    : `${builder.address.get(
-                        'borrowIncentive.id'
-                      )}::user::unstake`))
+                  `${builder.address.get('borrowIncentive.id')}::user::unstake`)
           );
 
         if (!obligationLocked || unstakeObligationBeforeStake) {
@@ -262,11 +253,7 @@ const generateBorrowIncentiveQuickMethod: GenerateBorrowIncentiveQuickMethod =
               (txn.target ===
                 `${OLD_BORROW_INCENTIVE_PROTOCOL_ID}::user::unstake` ||
                 txn.target ===
-                  (IS_VE_SCA_TEST
-                    ? `${'0x4d5a7cefa4147b4ace0ca845b20437d6ac0d32e5f2f855171f745472c2576246'}::user::unstake`
-                    : `${builder.address.get(
-                        'borrowIncentive.id'
-                      )}::user::unstake`))
+                  `${builder.address.get('borrowIncentive.id')}::user::unstake`)
           );
 
         if (!obligationLocked || unstakeObligationBeforeStake) {
@@ -276,8 +263,12 @@ const generateBorrowIncentiveQuickMethod: GenerateBorrowIncentiveQuickMethod =
               builder.query,
               veSca.keyId
             );
+
             // if bindedObligationId is equal to obligationId, then use it again
-            if (!bindedObligationId || bindedObligationId === obligationArg) {
+            if (
+              (!bindedObligationId || bindedObligationId === obligationArg) &&
+              veSca.currentVeScaBalance > 0
+            ) {
               txBlock.stakeObligationWithVesca(
                 obligationArg,
                 obligationtKeyArg,

package/src/builders/coreBuilder.ts

@@ -70,6 +70,10 @@ const generateCoreNormalMethod: GenerateCoreNormalMethod = ({
     coinDecimalsRegistry: builder.address.get('core.coinDecimalsRegistry'),
     xOracle: builder.address.get('core.oracles.xOracle'),
   };
+
+  const referralPkgId = builder.address.get('referral.id');
+  const referralWitnessType = `${referralPkgId}::scallop_referral_program::REFERRAL_WITNESS`;
+
   return {
     openObligation: () =>
       txBlock.moveCall(
@@ -160,6 +164,30 @@ const generateCoreNormalMethod: GenerateCoreNormalMethod = ({
         [coinType]
       );
     },
+    borrowWithReferral: (
+      obligation,
+      obligationKey,
+      borrowReferral,
+      amount,
+      poolCoinName
+    ) => {
+      const coinType = builder.utils.parseCoinType(poolCoinName);
+      return txBlock.moveCall(
+        `${coreIds.protocolPkg}::borrow::borrow_with_referral`,
+        [
+          coreIds.version,
+          obligation,
+          obligationKey,
+          coreIds.market,
+          coreIds.coinDecimalsRegistry,
+          borrowReferral,
+          amount,
+          coreIds.xOracle,
+          SUI_CLOCK_OBJECT_ID,
+        ],
+        [coinType, referralWitnessType]
+      );
+    },
     borrowEntry: (obligation, obligationKey, amount, poolCoinName) => {
       const coinType = builder.utils.parseCoinType(poolCoinName);
       return txBlock.moveCall(
@@ -319,6 +347,32 @@ const generateCoreQuickMethod: GenerateCoreQuickMethod = ({
         poolCoinName
       );
     },
+    borrowWithReferralQuick: async (
+      amount,
+      poolCoinName,
+      borrowReferral,
+      obligationId,
+      obligationKey
+    ) => {
+      const obligationInfo = await requireObligationInfo(
+        builder,
+        txBlock,
+        obligationId,
+        obligationKey
+      );
+      const obligationCoinNames = await builder.utils.getObligationCoinNames(
+        obligationInfo.obligationId
+      );
+      const updateCoinNames = [...obligationCoinNames, poolCoinName];
+      await updateOracles(builder, txBlock, updateCoinNames);
+      return txBlock.borrowWithReferral(
+        obligationInfo.obligationId,
+        obligationInfo.obligationKey as SuiAddressArg,
+        borrowReferral,
+        amount,
+        poolCoinName
+      );
+    },
     repayQuick: async (amount, poolCoinName, obligationId) => {
       const sender = requireSender(txBlock);
       const obligationInfo = await requireObligationInfo(

package/src/builders/index.ts

@@ -6,6 +6,7 @@ import { newBorrowIncentiveTxBlock } from './borrowIncentiveBuilder';
 import { newVeScaTxBlock } from './vescaBuilder';
 import type { ScallopBuilder } from '../models';
 import type { ScallopTxBlock } from '../types';
+import { newReferralTxBlock } from './referralBuilder';
 
 /**
  * Create a new ScallopTxBlock instance.
@@ -23,15 +24,17 @@ export const newScallopTxBlock = (
     builder,
     vescaTxBlock
   );
-  const spoolTxBlock = newSpoolTxBlock(builder, borrowIncentiveTxBlock);
+  const referralTxBlock = newReferralTxBlock(builder, borrowIncentiveTxBlock);
+  const spoolTxBlock = newSpoolTxBlock(builder, referralTxBlock);
   const coreTxBlock = newCoreTxBlock(builder, spoolTxBlock);
-
   return new Proxy(coreTxBlock, {
     get: (target, prop) => {
       if (prop in vescaTxBlock) {
         return Reflect.get(vescaTxBlock, prop);
       } else if (prop in borrowIncentiveTxBlock) {
         return Reflect.get(borrowIncentiveTxBlock, prop);
+      } else if (prop in referralTxBlock) {
+        return Reflect.get(referralTxBlock, prop);
       } else if (prop in spoolTxBlock) {
         return Reflect.get(spoolTxBlock, prop);
       }