@haneullabs/deepbook-v3 0.1.0 → 1.0.11

package/src/contracts/deepbook/account.ts

@@ -1,13 +1,13 @@
-// Copyright (c) Mysten Labs, Inc.
-// SPDX-License-Identifier: Apache-2.0
+/**************************************************************
+ * THIS FILE IS GENERATED AND SHOULD NOT BE MANUALLY MODIFIED *
+ **************************************************************/
 
 /** Account module manages the account data for each user. */
 
-import { MoveStruct, normalizeMoveArguments } from '../utils/index.js';
-import type { RawTransactionArgument } from '../utils/index.js';
+import { MoveStruct, normalizeMoveArguments, type RawTransactionArgument } from '../utils/index.js';
 import { bcs } from '@haneullabs/haneul/bcs';
-import type { Transaction } from '@haneullabs/haneul/transactions';
-import * as vec_set from './deps/sui/vec_set.js';
+import { type Transaction } from '@haneullabs/haneul/transactions';
+import * as vec_set from './deps/haneul/vec_set.js';
 import * as balances from './balances.js';
 const $moduleName = '@deepbook/core::account';
 export const Account = new MoveStruct({
@@ -35,7 +35,7 @@ export interface OpenOrdersOptions {
 }
 export function openOrders(options: OpenOrdersOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -54,7 +54,7 @@ export interface TakerVolumeOptions {
 }
 export function takerVolume(options: TakerVolumeOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -73,7 +73,7 @@ export interface MakerVolumeOptions {
 }
 export function makerVolume(options: MakerVolumeOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -92,7 +92,7 @@ export interface TotalVolumeOptions {
 }
 export function totalVolume(options: TotalVolumeOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -111,7 +111,7 @@ export interface ActiveStakeOptions {
 }
 export function activeStake(options: ActiveStakeOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -130,7 +130,7 @@ export interface InactiveStakeOptions {
 }
 export function inactiveStake(options: InactiveStakeOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -149,7 +149,7 @@ export interface CreatedProposalOptions {
 }
 export function createdProposal(options: CreatedProposalOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -168,7 +168,7 @@ export interface VotedProposalOptions {
 }
 export function votedProposal(options: VotedProposalOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -187,7 +187,7 @@ export interface SettledBalancesOptions {
 }
 export function settledBalances(options: SettledBalancesOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::account::Account`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['self'];
 	return (tx: Transaction) =>
 		tx.moveCall({

package/src/contracts/deepbook/balance_manager.ts

@@ -1,5 +1,6 @@
-// Copyright (c) Mysten Labs, Inc.
-// SPDX-License-Identifier: Apache-2.0
+/**************************************************************
+ * THIS FILE IS GENERATED AND SHOULD NOT BE MANUALLY MODIFIED *
+ **************************************************************/
 
 /**
  * The BalanceManager is a shared object that holds all of the balances for
@@ -11,19 +12,22 @@
  * Generally, a high frequency trading engine will trade as the default owner.
  */
 
-import { MoveStruct, normalizeMoveArguments } from '../utils/index.js';
-import type { RawTransactionArgument } from '../utils/index.js';
+import {
+	MoveStruct,
+	MoveTuple,
+	normalizeMoveArguments,
+	type RawTransactionArgument,
+} from '../utils/index.js';
 import { bcs } from '@haneullabs/haneul/bcs';
-import type { Transaction } from '@haneullabs/haneul/transactions';
-import * as object from './deps/sui/object.js';
-import * as bag from './deps/sui/bag.js';
-import * as vec_set from './deps/sui/vec_set.js';
+import { type Transaction } from '@haneullabs/haneul/transactions';
+import * as bag from './deps/haneul/bag.js';
+import * as vec_set from './deps/haneul/vec_set.js';
 import * as type_name from './deps/std/type_name.js';
 const $moduleName = '@deepbook/core::balance_manager';
 export const BalanceManager = new MoveStruct({
 	name: `${$moduleName}::BalanceManager`,
 	fields: {
-		id: object.UID,
+		id: bcs.Address,
 		owner: bcs.Address,
 		balances: bag.Bag,
 		allow_listed: vec_set.VecSet(bcs.Address),
@@ -46,29 +50,62 @@ export const BalanceEvent = new MoveStruct({
 	},
 });
 export const BalanceKey = new MoveStruct({
-	name: `${$moduleName}::BalanceKey`,
+	name: `${$moduleName}::BalanceKey<phantom T>`,
 	fields: {
 		dummy_field: bcs.bool(),
 	},
 });
+export const ReferralKey = new MoveTuple({
+	name: `${$moduleName}::ReferralKey`,
+	fields: [bcs.Address],
+});
 export const TradeCap = new MoveStruct({
 	name: `${$moduleName}::TradeCap`,
 	fields: {
-		id: object.UID,
+		id: bcs.Address,
 		balance_manager_id: bcs.Address,
 	},
 });
 export const DepositCap = new MoveStruct({
 	name: `${$moduleName}::DepositCap`,
 	fields: {
-		id: object.UID,
+		id: bcs.Address,
 		balance_manager_id: bcs.Address,
 	},
 });
 export const WithdrawCap = new MoveStruct({
 	name: `${$moduleName}::WithdrawCap`,
 	fields: {
-		id: object.UID,
+		id: bcs.Address,
+		balance_manager_id: bcs.Address,
+	},
+});
+export const DeepBookReferral = new MoveStruct({
+	name: `${$moduleName}::DeepBookReferral`,
+	fields: {
+		id: bcs.Address,
+		owner: bcs.Address,
+	},
+});
+export const DeepBookPoolReferral = new MoveStruct({
+	name: `${$moduleName}::DeepBookPoolReferral`,
+	fields: {
+		id: bcs.Address,
+		owner: bcs.Address,
+		pool_id: bcs.Address,
+	},
+});
+export const DeepBookReferralCreatedEvent = new MoveStruct({
+	name: `${$moduleName}::DeepBookReferralCreatedEvent`,
+	fields: {
+		referral_id: bcs.Address,
+		owner: bcs.Address,
+	},
+});
+export const DeepBookReferralSetEvent = new MoveStruct({
+	name: `${$moduleName}::DeepBookReferralSetEvent`,
+	fields: {
+		referral_id: bcs.Address,
 		balance_manager_id: bcs.Address,
 	},
 });
@@ -101,7 +138,7 @@ export interface NewWithOwnerOptions {
 }
 export function newWithOwner(options: NewWithOwnerOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = ['address'] satisfies string[];
+	const argumentsTypes = ['address'] satisfies (string | null)[];
 	const parameterNames = ['Owner'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -121,7 +158,7 @@ export interface NewWithCustomOwnerOptions {
 /** Create a new balance manager with an owner. */
 export function newWithCustomOwner(options: NewWithCustomOwnerOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = ['address'] satisfies string[];
+	const argumentsTypes = ['address'] satisfies (string | null)[];
 	const parameterNames = ['owner'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -132,16 +169,16 @@ export function newWithCustomOwner(options: NewWithCustomOwnerOptions) {
 		});
 }
 export interface NewWithCustomOwnerAndCapsArguments {
-	owner: RawTransactionArgument<string>;
+	Owner: RawTransactionArgument<string>;
 }
 export interface NewWithCustomOwnerAndCapsOptions {
 	package?: string;
-	arguments: NewWithCustomOwnerAndCapsArguments | [owner: RawTransactionArgument<string>];
+	arguments: NewWithCustomOwnerAndCapsArguments | [Owner: RawTransactionArgument<string>];
 }
 export function newWithCustomOwnerAndCaps(options: NewWithCustomOwnerAndCapsOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = ['address'] satisfies string[];
-	const parameterNames = ['owner'];
+	const argumentsTypes = ['address'] satisfies (string | null)[];
+	const parameterNames = ['Owner'];
 	return (tx: Transaction) =>
 		tx.moveCall({
 			package: packageAddress,
@@ -150,6 +187,135 @@ export function newWithCustomOwnerAndCaps(options: NewWithCustomOwnerAndCapsOpti
 			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
 		});
 }
+export interface NewWithCustomOwnerCapsArguments {
+	deepbookRegistry: RawTransactionArgument<string>;
+	owner: RawTransactionArgument<string>;
+}
+export interface NewWithCustomOwnerCapsOptions {
+	package?: string;
+	arguments:
+		| NewWithCustomOwnerCapsArguments
+		| [deepbookRegistry: RawTransactionArgument<string>, owner: RawTransactionArgument<string>];
+	typeArguments: [string];
+}
+export function newWithCustomOwnerCaps(options: NewWithCustomOwnerCapsOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, 'address'] satisfies (string | null)[];
+	const parameterNames = ['deepbookRegistry', 'owner'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'new_with_custom_owner_caps',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+			typeArguments: options.typeArguments,
+		});
+}
+export interface SetReferralArguments {
+	BalanceManager: RawTransactionArgument<string>;
+	Referral: RawTransactionArgument<string>;
+	TradeCap: RawTransactionArgument<string>;
+}
+export interface SetReferralOptions {
+	package?: string;
+	arguments:
+		| SetReferralArguments
+		| [
+				BalanceManager: RawTransactionArgument<string>,
+				Referral: RawTransactionArgument<string>,
+				TradeCap: RawTransactionArgument<string>,
+		  ];
+}
+export function setReferral(options: SetReferralOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, null, null] satisfies (string | null)[];
+	const parameterNames = ['BalanceManager', 'Referral', 'TradeCap'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'set_referral',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface SetBalanceManagerReferralArguments {
+	balanceManager: RawTransactionArgument<string>;
+	referral: RawTransactionArgument<string>;
+	tradeCap: RawTransactionArgument<string>;
+}
+export interface SetBalanceManagerReferralOptions {
+	package?: string;
+	arguments:
+		| SetBalanceManagerReferralArguments
+		| [
+				balanceManager: RawTransactionArgument<string>,
+				referral: RawTransactionArgument<string>,
+				tradeCap: RawTransactionArgument<string>,
+		  ];
+}
+/** Set the referral for the balance manager. */
+export function setBalanceManagerReferral(options: SetBalanceManagerReferralOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, null, null] satisfies (string | null)[];
+	const parameterNames = ['balanceManager', 'referral', 'tradeCap'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'set_balance_manager_referral',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface UnsetReferralArguments {
+	BalanceManager: RawTransactionArgument<string>;
+	TradeCap: RawTransactionArgument<string>;
+}
+export interface UnsetReferralOptions {
+	package?: string;
+	arguments:
+		| UnsetReferralArguments
+		| [BalanceManager: RawTransactionArgument<string>, TradeCap: RawTransactionArgument<string>];
+}
+export function unsetReferral(options: UnsetReferralOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, null] satisfies (string | null)[];
+	const parameterNames = ['BalanceManager', 'TradeCap'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'unset_referral',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface UnsetBalanceManagerReferralArguments {
+	balanceManager: RawTransactionArgument<string>;
+	poolId: RawTransactionArgument<string>;
+	tradeCap: RawTransactionArgument<string>;
+}
+export interface UnsetBalanceManagerReferralOptions {
+	package?: string;
+	arguments:
+		| UnsetBalanceManagerReferralArguments
+		| [
+				balanceManager: RawTransactionArgument<string>,
+				poolId: RawTransactionArgument<string>,
+				tradeCap: RawTransactionArgument<string>,
+		  ];
+}
+/** Unset the referral for the balance manager. */
+export function unsetBalanceManagerReferral(options: UnsetBalanceManagerReferralOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, '0x2::object::ID', null] satisfies (string | null)[];
+	const parameterNames = ['balanceManager', 'poolId', 'tradeCap'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'unset_balance_manager_referral',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
 export interface BalanceArguments {
 	balanceManager: RawTransactionArgument<string>;
 }
@@ -161,7 +327,7 @@ export interface BalanceOptions {
 /** Returns the balance of a Coin in a balance manager. */
 export function balance(options: BalanceOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -182,7 +348,7 @@ export interface MintTradeCapOptions {
 /** Mint a `TradeCap`, only owner can mint a `TradeCap`. */
 export function mintTradeCap(options: MintTradeCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -202,7 +368,7 @@ export interface MintDepositCapOptions {
 /** Mint a `DepositCap`, only owner can mint. */
 export function mintDepositCap(options: MintDepositCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -222,7 +388,7 @@ export interface MintWithdrawCapOptions {
 /** Mint a `WithdrawCap`, only owner can mint. */
 export function mintWithdrawCap(options: MintWithdrawCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -248,10 +414,7 @@ export interface RevokeTradeCapOptions {
  */
 export function revokeTradeCap(options: RevokeTradeCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		'0x0000000000000000000000000000000000000000000000000000000000000002::object::ID',
-	] satisfies string[];
+	const argumentsTypes = [null, '0x2::object::ID'] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'tradeCapId'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -274,7 +437,7 @@ export interface GenerateProofAsOwnerOptions {
  */
 export function generateProofAsOwner(options: GenerateProofAsOwnerOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -300,10 +463,7 @@ export interface GenerateProofAsTraderOptions {
  */
 export function generateProofAsTrader(options: GenerateProofAsTraderOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		`${packageAddress}::balance_manager::TradeCap`,
-	] satisfies string[];
+	const argumentsTypes = [null, null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'tradeCap'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -327,10 +487,7 @@ export interface DepositOptions {
 /** Deposit funds to a balance manager. Only owner can call this directly. */
 export function deposit(options: DepositOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		`0x0000000000000000000000000000000000000000000000000000000000000002::coin::Coin<${options.typeArguments[0]}>`,
-	] satisfies string[];
+	const argumentsTypes = [null, null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'coin'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -360,11 +517,7 @@ export interface DepositWithCapOptions {
 /** Deposit funds into a balance manager by a `DepositCap` owner. */
 export function depositWithCap(options: DepositWithCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		`${packageAddress}::balance_manager::DepositCap`,
-		`0x0000000000000000000000000000000000000000000000000000000000000002::coin::Coin<${options.typeArguments[0]}>`,
-	] satisfies string[];
+	const argumentsTypes = [null, null, null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'depositCap', 'coin'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -394,11 +547,7 @@ export interface WithdrawWithCapOptions {
 /** Withdraw funds from a balance manager by a `WithdrawCap` owner. */
 export function withdrawWithCap(options: WithdrawWithCapOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		`${packageAddress}::balance_manager::WithdrawCap`,
-		'u64',
-	] satisfies string[];
+	const argumentsTypes = [null, null, 'u64'] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'withdrawCap', 'withdrawAmount'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -430,10 +579,7 @@ export interface WithdrawOptions {
  */
 export function withdraw(options: WithdrawOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		'u64',
-	] satisfies string[];
+	const argumentsTypes = [null, 'u64'] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'withdrawAmount'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -454,7 +600,7 @@ export interface WithdrawAllOptions {
 }
 export function withdrawAll(options: WithdrawAllOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -465,6 +611,92 @@ export function withdrawAll(options: WithdrawAllOptions) {
 			typeArguments: options.typeArguments,
 		});
 }
+export interface RegisterManagerArguments {
+	BalanceManager: RawTransactionArgument<string>;
+	Registry: RawTransactionArgument<string>;
+}
+export interface RegisterManagerOptions {
+	package?: string;
+	arguments:
+		| RegisterManagerArguments
+		| [BalanceManager: RawTransactionArgument<string>, Registry: RawTransactionArgument<string>];
+}
+export function registerManager(options: RegisterManagerOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, null] satisfies (string | null)[];
+	const parameterNames = ['BalanceManager', 'Registry'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'register_manager',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface RegisterBalanceManagerArguments {
+	balanceManager: RawTransactionArgument<string>;
+	registry: RawTransactionArgument<string>;
+}
+export interface RegisterBalanceManagerOptions {
+	package?: string;
+	arguments:
+		| RegisterBalanceManagerArguments
+		| [balanceManager: RawTransactionArgument<string>, registry: RawTransactionArgument<string>];
+}
+export function registerBalanceManager(options: RegisterBalanceManagerOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, null] satisfies (string | null)[];
+	const parameterNames = ['balanceManager', 'registry'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'register_balance_manager',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface GetReferralIdArguments {
+	BalanceManager: RawTransactionArgument<string>;
+}
+export interface GetReferralIdOptions {
+	package?: string;
+	arguments: GetReferralIdArguments | [BalanceManager: RawTransactionArgument<string>];
+}
+export function getReferralId(options: GetReferralIdOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null] satisfies (string | null)[];
+	const parameterNames = ['BalanceManager'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'get_referral_id',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface GetBalanceManagerReferralIdArguments {
+	balanceManager: RawTransactionArgument<string>;
+	poolId: RawTransactionArgument<string>;
+}
+export interface GetBalanceManagerReferralIdOptions {
+	package?: string;
+	arguments:
+		| GetBalanceManagerReferralIdArguments
+		| [balanceManager: RawTransactionArgument<string>, poolId: RawTransactionArgument<string>];
+}
+/** Get the referral id from the balance manager. */
+export function getBalanceManagerReferralId(options: GetBalanceManagerReferralIdOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null, '0x2::object::ID'] satisfies (string | null)[];
+	const parameterNames = ['balanceManager', 'poolId'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'get_balance_manager_referral_id',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
 export interface ValidateProofArguments {
 	balanceManager: RawTransactionArgument<string>;
 	proof: RawTransactionArgument<string>;
@@ -477,10 +709,7 @@ export interface ValidateProofOptions {
 }
 export function validateProof(options: ValidateProofOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [
-		`${packageAddress}::balance_manager::BalanceManager`,
-		`${packageAddress}::balance_manager::TradeProof`,
-	] satisfies string[];
+	const argumentsTypes = [null, null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager', 'proof'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -500,7 +729,7 @@ export interface OwnerOptions {
 /** Returns the owner of the balance_manager. */
 export function owner(options: OwnerOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -520,7 +749,7 @@ export interface IdOptions {
 /** Returns the owner of the balance_manager. */
 export function id(options: IdOptions) {
 	const packageAddress = options.package ?? '@deepbook/core';
-	const argumentsTypes = [`${packageAddress}::balance_manager::BalanceManager`] satisfies string[];
+	const argumentsTypes = [null] satisfies (string | null)[];
 	const parameterNames = ['balanceManager'];
 	return (tx: Transaction) =>
 		tx.moveCall({
@@ -530,3 +759,60 @@ export function id(options: IdOptions) {
 			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
 		});
 }
+export interface ReferralOwnerArguments {
+	Referral: RawTransactionArgument<string>;
+}
+export interface ReferralOwnerOptions {
+	package?: string;
+	arguments: ReferralOwnerArguments | [Referral: RawTransactionArgument<string>];
+}
+export function referralOwner(options: ReferralOwnerOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null] satisfies (string | null)[];
+	const parameterNames = ['Referral'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'referral_owner',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface BalanceManagerReferralOwnerArguments {
+	referral: RawTransactionArgument<string>;
+}
+export interface BalanceManagerReferralOwnerOptions {
+	package?: string;
+	arguments: BalanceManagerReferralOwnerArguments | [referral: RawTransactionArgument<string>];
+}
+export function balanceManagerReferralOwner(options: BalanceManagerReferralOwnerOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null] satisfies (string | null)[];
+	const parameterNames = ['referral'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'balance_manager_referral_owner',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}
+export interface BalanceManagerReferralPoolIdArguments {
+	referral: RawTransactionArgument<string>;
+}
+export interface BalanceManagerReferralPoolIdOptions {
+	package?: string;
+	arguments: BalanceManagerReferralPoolIdArguments | [referral: RawTransactionArgument<string>];
+}
+export function balanceManagerReferralPoolId(options: BalanceManagerReferralPoolIdOptions) {
+	const packageAddress = options.package ?? '@deepbook/core';
+	const argumentsTypes = [null] satisfies (string | null)[];
+	const parameterNames = ['referral'];
+	return (tx: Transaction) =>
+		tx.moveCall({
+			package: packageAddress,
+			module: 'balance_manager',
+			function: 'balance_manager_referral_pool_id',
+			arguments: normalizeMoveArguments(options.arguments, argumentsTypes, parameterNames),
+		});
+}