1929 lines
53 KiB
TypeScript
1929 lines
53 KiB
TypeScript
import { AnchorProvider, BN, Program, Provider } from '@project-serum/anchor';
|
|
import { getFeeRates, getFeeTier } from '@project-serum/serum';
|
|
import { Order } from '@project-serum/serum/lib/market';
|
|
import {
|
|
closeAccount,
|
|
initializeAccount,
|
|
WRAPPED_SOL_MINT,
|
|
} from '@project-serum/serum/lib/token-instructions';
|
|
import {
|
|
ASSOCIATED_TOKEN_PROGRAM_ID,
|
|
Token,
|
|
TOKEN_PROGRAM_ID,
|
|
} from '@solana/spl-token';
|
|
import {
|
|
AccountMeta,
|
|
Cluster,
|
|
Keypair,
|
|
MemcmpFilter,
|
|
PublicKey,
|
|
Signer,
|
|
SystemProgram,
|
|
SYSVAR_INSTRUCTIONS_PUBKEY,
|
|
SYSVAR_RENT_PUBKEY,
|
|
Transaction,
|
|
TransactionInstruction,
|
|
TransactionSignature,
|
|
} from '@solana/web3.js';
|
|
import bs58 from 'bs58';
|
|
import { Bank, MintInfo } from './accounts/bank';
|
|
import { Group } from './accounts/group';
|
|
import { I80F48 } from './accounts/I80F48';
|
|
import {
|
|
MangoAccount,
|
|
MangoAccountData,
|
|
TokenPosition,
|
|
} from './accounts/mangoAccount';
|
|
import { StubOracle } from './accounts/oracle';
|
|
import { OrderType, PerpMarket, Side } from './accounts/perp';
|
|
import {
|
|
Serum3Market,
|
|
Serum3OrderType,
|
|
Serum3SelfTradeBehavior,
|
|
Serum3Side,
|
|
} from './accounts/serum3';
|
|
import { SERUM3_PROGRAM_ID } from './constants';
|
|
import { Id } from './ids';
|
|
import { IDL, MangoV4 } from './mango_v4';
|
|
import { FlashLoanType } from './types';
|
|
import {
|
|
createAssociatedTokenAccountIdempotentInstruction,
|
|
getAssociatedTokenAddress,
|
|
I64_MAX_BN,
|
|
toNativeDecimals,
|
|
} from './utils';
|
|
import { simulate } from './utils/anchor';
|
|
import { sendTransaction } from './utils/rpc';
|
|
|
|
enum AccountRetriever {
|
|
Scanning,
|
|
Fixed,
|
|
}
|
|
|
|
export type IdsSource = 'api' | 'static' | 'get-program-accounts';
|
|
|
|
// TODO: replace ui values with native as input wherever possible
|
|
// TODO: replace token/market names with token or market indices
|
|
export class MangoClient {
|
|
private postSendTxCallback?: ({ txid: string }) => void;
|
|
private prioritizationFee: number;
|
|
|
|
constructor(
|
|
public program: Program<MangoV4>,
|
|
public programId: PublicKey,
|
|
public cluster: Cluster,
|
|
public opts: {
|
|
postSendTxCallback?: ({ txid }: { txid: string }) => void;
|
|
prioritizationFee?: number;
|
|
} = {},
|
|
public idsSource: IdsSource = 'api',
|
|
) {
|
|
this.prioritizationFee = opts?.prioritizationFee || 0;
|
|
this.postSendTxCallback = opts?.postSendTxCallback;
|
|
// TODO: evil side effect, but limited backtraces are a nightmare
|
|
Error.stackTraceLimit = 1000;
|
|
}
|
|
|
|
/// public
|
|
|
|
// Group
|
|
|
|
public async groupCreate(
|
|
groupNum: number,
|
|
testing: boolean,
|
|
version: number,
|
|
insuranceMintPk: PublicKey,
|
|
): Promise<TransactionSignature> {
|
|
const adminPk = (this.program.provider as AnchorProvider).wallet.publicKey;
|
|
return await this.program.methods
|
|
.groupCreate(groupNum, testing ? 1 : 0, version)
|
|
.accounts({
|
|
creator: adminPk,
|
|
payer: adminPk,
|
|
insuranceMint: insuranceMintPk,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async groupEdit(
|
|
group: Group,
|
|
admin: PublicKey | undefined,
|
|
fastListingAdmin: PublicKey | undefined,
|
|
testing: number | undefined,
|
|
version: number | undefined,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.groupEdit(
|
|
admin ?? null,
|
|
fastListingAdmin ?? null,
|
|
testing ?? null,
|
|
version ?? null,
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async groupClose(group: Group): Promise<TransactionSignature> {
|
|
const adminPk = (this.program.provider as AnchorProvider).wallet.publicKey;
|
|
return await this.program.methods
|
|
.groupClose()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
insuranceVault: group.insuranceVault,
|
|
admin: adminPk,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async getGroup(groupPk: PublicKey): Promise<Group> {
|
|
const groupAccount = await this.program.account.group.fetch(groupPk);
|
|
const group = Group.from(groupPk, groupAccount);
|
|
await group.reloadAll(this);
|
|
return group;
|
|
}
|
|
|
|
public async getGroupsForCreator(creatorPk: PublicKey): Promise<Group[]> {
|
|
const filters: MemcmpFilter[] = [
|
|
{
|
|
memcmp: {
|
|
bytes: creatorPk.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
];
|
|
|
|
const groups = (await this.program.account.group.all(filters)).map(
|
|
(tuple) => Group.from(tuple.publicKey, tuple.account),
|
|
);
|
|
groups.forEach((group) => group.reloadAll(this));
|
|
return groups;
|
|
}
|
|
|
|
public async getGroupForCreator(
|
|
creatorPk: PublicKey,
|
|
groupNum: number,
|
|
): Promise<Group> {
|
|
const bbuf = Buffer.alloc(4);
|
|
bbuf.writeUInt32LE(groupNum);
|
|
const filters: MemcmpFilter[] = [
|
|
{
|
|
memcmp: {
|
|
bytes: creatorPk.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
{
|
|
memcmp: {
|
|
bytes: bs58.encode(bbuf),
|
|
offset: 40,
|
|
},
|
|
},
|
|
];
|
|
const groups = (await this.program.account.group.all(filters)).map(
|
|
(tuple) => Group.from(tuple.publicKey, tuple.account),
|
|
);
|
|
await groups[0].reloadAll(this);
|
|
return groups[0];
|
|
}
|
|
|
|
// Tokens/Banks
|
|
|
|
public async tokenRegister(
|
|
group: Group,
|
|
mintPk: PublicKey,
|
|
oraclePk: PublicKey,
|
|
oracleConfFilter: number,
|
|
tokenIndex: number,
|
|
name: string,
|
|
adjustmentFactor: number,
|
|
util0: number,
|
|
rate0: number,
|
|
util1: number,
|
|
rate1: number,
|
|
maxRate: number,
|
|
loanFeeRate: number,
|
|
loanOriginationFeeRate: number,
|
|
maintAssetWeight: number,
|
|
initAssetWeight: number,
|
|
maintLiabWeight: number,
|
|
initLiabWeight: number,
|
|
liquidationFee: number,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.tokenRegister(
|
|
tokenIndex,
|
|
name,
|
|
{
|
|
confFilter: {
|
|
val: I80F48.fromNumber(oracleConfFilter).getData(),
|
|
},
|
|
} as any, // future: nested custom types dont typecheck, fix if possible?
|
|
{ adjustmentFactor, util0, rate0, util1, rate1, maxRate },
|
|
loanFeeRate,
|
|
loanOriginationFeeRate,
|
|
maintAssetWeight,
|
|
initAssetWeight,
|
|
maintLiabWeight,
|
|
initLiabWeight,
|
|
liquidationFee,
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
mint: mintPk,
|
|
oracle: oraclePk,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
rent: SYSVAR_RENT_PUBKEY,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async tokenRegisterTrustless(
|
|
group: Group,
|
|
mintPk: PublicKey,
|
|
oraclePk: PublicKey,
|
|
tokenIndex: number,
|
|
name: string,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.tokenRegisterTrustless(tokenIndex, name)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
fastListingAdmin: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
mint: mintPk,
|
|
oracle: oraclePk,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
rent: SYSVAR_RENT_PUBKEY,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async tokenEdit(
|
|
group: Group,
|
|
mintPk: PublicKey,
|
|
oracle: PublicKey,
|
|
oracleConfFilter: number,
|
|
groupInsuranceFund: boolean | undefined,
|
|
adjustmentFactor: number,
|
|
util0: number,
|
|
rate0: number,
|
|
util1: number,
|
|
rate1: number,
|
|
maxRate: number,
|
|
loanFeeRate: number,
|
|
loanOriginationFeeRate: number,
|
|
maintAssetWeight: number,
|
|
initAssetWeight: number,
|
|
maintLiabWeight: number,
|
|
initLiabWeight: number,
|
|
liquidationFee: number,
|
|
): Promise<TransactionSignature> {
|
|
const bank = group.getFirstBankByMint(mintPk);
|
|
const mintInfo = group.mintInfosMapByTokenIndex.get(bank.tokenIndex)!;
|
|
|
|
return await this.program.methods
|
|
.tokenEdit(
|
|
new BN(0),
|
|
oracle,
|
|
{
|
|
confFilter: {
|
|
val: I80F48.fromNumber(oracleConfFilter).getData(),
|
|
},
|
|
} as any, // future: nested custom types dont typecheck, fix if possible?
|
|
groupInsuranceFund ?? null,
|
|
{ adjustmentFactor, util0, rate0, util1, rate1, maxRate },
|
|
loanFeeRate,
|
|
loanOriginationFeeRate,
|
|
maintAssetWeight,
|
|
initAssetWeight,
|
|
maintLiabWeight,
|
|
initLiabWeight,
|
|
liquidationFee,
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
mintInfo: mintInfo.publicKey,
|
|
})
|
|
.remainingAccounts([
|
|
{
|
|
pubkey: bank.publicKey,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
} as AccountMeta,
|
|
])
|
|
.rpc({ skipPreflight: true });
|
|
}
|
|
|
|
public async tokenDeregister(
|
|
group: Group,
|
|
mintPk: PublicKey,
|
|
): Promise<TransactionSignature> {
|
|
const bank = group.getFirstBankByMint(mintPk);
|
|
const adminPk = (this.program.provider as AnchorProvider).wallet.publicKey;
|
|
|
|
const dustVaultPk = await getAssociatedTokenAddress(bank.mint, adminPk);
|
|
const ai = await this.program.provider.connection.getAccountInfo(
|
|
dustVaultPk,
|
|
);
|
|
if (!ai) {
|
|
const tx = new Transaction();
|
|
tx.add(
|
|
Token.createAssociatedTokenAccountInstruction(
|
|
ASSOCIATED_TOKEN_PROGRAM_ID,
|
|
TOKEN_PROGRAM_ID,
|
|
bank.mint,
|
|
dustVaultPk,
|
|
adminPk,
|
|
adminPk,
|
|
),
|
|
);
|
|
await this.program.provider.sendAndConfirm(tx);
|
|
}
|
|
|
|
return await this.program.methods
|
|
.tokenDeregister(bank.tokenIndex)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: adminPk,
|
|
mintInfo: group.mintInfosMapByTokenIndex.get(bank.tokenIndex)
|
|
?.publicKey,
|
|
dustVault: dustVaultPk,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.remainingAccounts(
|
|
[bank.publicKey, bank.vault].map(
|
|
(pk) =>
|
|
({ pubkey: pk, isWritable: true, isSigner: false } as AccountMeta),
|
|
),
|
|
)
|
|
.rpc();
|
|
}
|
|
|
|
public async getBanksForGroup(group: Group): Promise<Bank[]> {
|
|
return (
|
|
await this.program.account.bank.all([
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
])
|
|
).map((tuple) => Bank.from(tuple.publicKey, tuple.account));
|
|
}
|
|
|
|
public async getMintInfosForGroup(group: Group): Promise<MintInfo[]> {
|
|
return (
|
|
await this.program.account.mintInfo.all([
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
])
|
|
).map((tuple) => {
|
|
return MintInfo.from(tuple.publicKey, tuple.account);
|
|
});
|
|
}
|
|
|
|
public async getMintInfoForTokenIndex(
|
|
group: Group,
|
|
tokenIndex: number,
|
|
): Promise<MintInfo[]> {
|
|
const tokenIndexBuf = Buffer.alloc(2);
|
|
tokenIndexBuf.writeUInt16LE(tokenIndex);
|
|
return (
|
|
await this.program.account.mintInfo.all([
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
{
|
|
memcmp: {
|
|
bytes: bs58.encode(tokenIndexBuf),
|
|
offset: 40,
|
|
},
|
|
},
|
|
])
|
|
).map((tuple) => {
|
|
return MintInfo.from(tuple.publicKey, tuple.account);
|
|
});
|
|
}
|
|
|
|
// Stub Oracle
|
|
|
|
public async stubOracleCreate(
|
|
group: Group,
|
|
mintPk: PublicKey,
|
|
price: number,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.stubOracleCreate({ val: I80F48.fromNumber(price).getData() })
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
mint: mintPk,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async stubOracleClose(
|
|
group: Group,
|
|
oracle: PublicKey,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.stubOracleClose()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
oracle: oracle,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async stubOracleSet(
|
|
group: Group,
|
|
oraclePk: PublicKey,
|
|
price: number,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.stubOracleSet({ val: I80F48.fromNumber(price).getData() })
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
oracle: oraclePk,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async getStubOracle(
|
|
group: Group,
|
|
mintPk?: PublicKey,
|
|
): Promise<StubOracle[]> {
|
|
const filters = [
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
];
|
|
|
|
if (mintPk) {
|
|
filters.push({
|
|
memcmp: {
|
|
bytes: mintPk.toBase58(),
|
|
offset: 40,
|
|
},
|
|
});
|
|
}
|
|
|
|
return (await this.program.account.stubOracle.all(filters)).map((pa) =>
|
|
StubOracle.from(pa.publicKey, pa.account),
|
|
);
|
|
}
|
|
|
|
// MangoAccount
|
|
|
|
public async getOrCreateMangoAccount(
|
|
group: Group,
|
|
ownerPk: PublicKey,
|
|
accountNumber?: number,
|
|
name?: string,
|
|
): Promise<MangoAccount> {
|
|
// TODO: this function discards accountSize and name when the account exists already!
|
|
// TODO: this function always creates accounts for this.program.owner, and not
|
|
// ownerPk! It needs to get passed a keypair, and we need to add
|
|
// createMangoAccountForOwner
|
|
if (accountNumber === undefined) {
|
|
// Get any MangoAccount
|
|
// TODO: should probably sort by accountNum for deterministic output!
|
|
let mangoAccounts = await this.getMangoAccountsForOwner(group, ownerPk);
|
|
if (mangoAccounts.length === 0) {
|
|
await this.createMangoAccount(group, accountNumber, name);
|
|
mangoAccounts = await this.getMangoAccountsForOwner(group, ownerPk);
|
|
}
|
|
return mangoAccounts[0];
|
|
} else {
|
|
let account = await this.getMangoAccountForOwner(
|
|
group,
|
|
ownerPk,
|
|
accountNumber,
|
|
);
|
|
if (account === undefined) {
|
|
await this.createMangoAccount(group, accountNumber, name);
|
|
account = await this.getMangoAccountForOwner(
|
|
group,
|
|
ownerPk,
|
|
accountNumber,
|
|
);
|
|
}
|
|
return account;
|
|
}
|
|
}
|
|
|
|
public async createMangoAccount(
|
|
group: Group,
|
|
accountNumber?: number,
|
|
name?: string,
|
|
): Promise<TransactionSignature> {
|
|
const transaction = await this.program.methods
|
|
.accountCreate(accountNumber ?? 0, 8, 0, 0, 0, name ?? '')
|
|
.accounts({
|
|
group: group.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.transaction();
|
|
|
|
return await sendTransaction(
|
|
this.program.provider as AnchorProvider,
|
|
transaction,
|
|
{
|
|
postSendTxCallback: this.postSendTxCallback,
|
|
},
|
|
);
|
|
}
|
|
|
|
public async expandMangoAccount(
|
|
group: Group,
|
|
account: MangoAccount,
|
|
tokenCount: number,
|
|
serum3Count: number,
|
|
perpCount: number,
|
|
perpOoCount: number,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.accountExpand(tokenCount, serum3Count, perpCount, perpOoCount)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: account.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async editMangoAccount(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
name?: string,
|
|
delegate?: PublicKey,
|
|
): Promise<TransactionSignature> {
|
|
const transaction = await this.program.methods
|
|
.accountEdit(name ?? null, delegate ?? null)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.transaction();
|
|
|
|
return await sendTransaction(
|
|
this.program.provider as AnchorProvider,
|
|
transaction,
|
|
{
|
|
postSendTxCallback: this.postSendTxCallback,
|
|
},
|
|
);
|
|
}
|
|
|
|
public async getMangoAccount(mangoAccount: MangoAccount) {
|
|
return MangoAccount.from(
|
|
mangoAccount.publicKey,
|
|
await this.program.account.mangoAccount.fetch(mangoAccount.publicKey),
|
|
);
|
|
}
|
|
|
|
public async getMangoAccountForOwner(
|
|
group: Group,
|
|
ownerPk: PublicKey,
|
|
accountNumber: number,
|
|
): Promise<MangoAccount> {
|
|
return (await this.getMangoAccountsForOwner(group, ownerPk)).find(
|
|
(a) => a.accountNum == accountNumber,
|
|
);
|
|
}
|
|
|
|
public async getMangoAccountsForOwner(
|
|
group: Group,
|
|
ownerPk: PublicKey,
|
|
): Promise<MangoAccount[]> {
|
|
return (
|
|
await this.program.account.mangoAccount.all([
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
{
|
|
memcmp: {
|
|
bytes: ownerPk.toBase58(),
|
|
offset: 40,
|
|
},
|
|
},
|
|
])
|
|
).map((pa) => {
|
|
return MangoAccount.from(pa.publicKey, pa.account);
|
|
});
|
|
}
|
|
|
|
public async getAllMangoAccounts(group: Group): Promise<MangoAccount[]> {
|
|
return (
|
|
await this.program.account.mangoAccount.all([
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
])
|
|
).map((pa) => {
|
|
return MangoAccount.from(pa.publicKey, pa.account);
|
|
});
|
|
}
|
|
|
|
public async closeMangoAccount(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.accountClose()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
solDestination: mangoAccount.owner,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async computeAccountData(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
): Promise<MangoAccountData> {
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(AccountRetriever.Fixed, group, [
|
|
mangoAccount,
|
|
]);
|
|
|
|
// Use our custom simulate fn in utils/anchor.ts so signing the tx is not required
|
|
this.program.provider.simulate = simulate;
|
|
|
|
let res;
|
|
try {
|
|
res = await this.program.methods
|
|
.computeAccountData()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
})
|
|
.remainingAccounts(
|
|
healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({
|
|
pubkey: pk,
|
|
isWritable: false,
|
|
isSigner: false,
|
|
} as AccountMeta),
|
|
),
|
|
)
|
|
.simulate();
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
|
|
return MangoAccountData.from(
|
|
res.events.find((event) => (event.name = 'MangoAccountData')).data as any,
|
|
);
|
|
}
|
|
|
|
public async tokenDeposit(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
mintPk: PublicKey,
|
|
amount: number,
|
|
): Promise<TransactionSignature> {
|
|
const nativeAmount = toNativeDecimals(
|
|
amount,
|
|
group.getMintDecimals(mintPk),
|
|
).toNumber();
|
|
return await this.tokenDepositNative(
|
|
group,
|
|
mangoAccount,
|
|
mintPk,
|
|
nativeAmount,
|
|
);
|
|
}
|
|
|
|
public async tokenDepositNative(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
mintPk: PublicKey,
|
|
nativeAmount: number,
|
|
) {
|
|
const bank = group.getFirstBankByMint(mintPk);
|
|
|
|
const tokenAccountPk = await getAssociatedTokenAddress(
|
|
mintPk,
|
|
mangoAccount.owner,
|
|
);
|
|
|
|
let wrappedSolAccount: Keypair | undefined;
|
|
let preInstructions: TransactionInstruction[] = [];
|
|
let postInstructions: TransactionInstruction[] = [];
|
|
const additionalSigners: Signer[] = [];
|
|
if (mintPk.equals(WRAPPED_SOL_MINT)) {
|
|
wrappedSolAccount = new Keypair();
|
|
const lamports = nativeAmount + 1e7;
|
|
|
|
preInstructions = [
|
|
SystemProgram.createAccount({
|
|
fromPubkey: mangoAccount.owner,
|
|
newAccountPubkey: wrappedSolAccount.publicKey,
|
|
lamports,
|
|
space: 165,
|
|
programId: TOKEN_PROGRAM_ID,
|
|
}),
|
|
initializeAccount({
|
|
account: wrappedSolAccount.publicKey,
|
|
mint: WRAPPED_SOL_MINT,
|
|
owner: mangoAccount.owner,
|
|
}),
|
|
];
|
|
postInstructions = [
|
|
closeAccount({
|
|
source: wrappedSolAccount.publicKey,
|
|
destination: mangoAccount.owner,
|
|
owner: mangoAccount.owner,
|
|
}),
|
|
];
|
|
additionalSigners.push(wrappedSolAccount);
|
|
}
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(
|
|
AccountRetriever.Fixed,
|
|
group,
|
|
[mangoAccount],
|
|
[bank],
|
|
);
|
|
|
|
const transaction = await this.program.methods
|
|
.tokenDeposit(new BN(nativeAmount))
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
bank: bank.publicKey,
|
|
vault: bank.vault,
|
|
tokenAccount: wrappedSolAccount?.publicKey ?? tokenAccountPk,
|
|
tokenAuthority: mangoAccount.owner,
|
|
})
|
|
.remainingAccounts(
|
|
healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({ pubkey: pk, isWritable: false, isSigner: false } as AccountMeta),
|
|
),
|
|
)
|
|
.preInstructions(preInstructions)
|
|
.postInstructions(postInstructions)
|
|
.signers(additionalSigners)
|
|
.transaction();
|
|
|
|
return await sendTransaction(
|
|
this.program.provider as AnchorProvider,
|
|
transaction,
|
|
{
|
|
additionalSigners,
|
|
postSendTxCallback: this.postSendTxCallback,
|
|
},
|
|
);
|
|
}
|
|
|
|
public async tokenWithdraw(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
mintPk: PublicKey,
|
|
amount: number,
|
|
allowBorrow: boolean,
|
|
): Promise<TransactionSignature> {
|
|
const nativeAmount = toNativeDecimals(
|
|
amount,
|
|
group.getMintDecimals(mintPk),
|
|
).toNumber();
|
|
return await this.tokenWithdrawNative(
|
|
group,
|
|
mangoAccount,
|
|
mintPk,
|
|
nativeAmount,
|
|
allowBorrow,
|
|
);
|
|
}
|
|
|
|
public async tokenWithdrawNative(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
mintPk: PublicKey,
|
|
nativeAmount: number,
|
|
allowBorrow: boolean,
|
|
): Promise<TransactionSignature> {
|
|
const bank = group.getFirstBankByMint(mintPk);
|
|
|
|
const tokenAccountPk = await getAssociatedTokenAddress(
|
|
bank.mint,
|
|
mangoAccount.owner,
|
|
);
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(
|
|
AccountRetriever.Fixed,
|
|
group,
|
|
[mangoAccount],
|
|
[bank],
|
|
);
|
|
|
|
const transaction = await this.program.methods
|
|
.tokenWithdraw(new BN(nativeAmount), allowBorrow)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
bank: bank.publicKey,
|
|
vault: bank.vault,
|
|
tokenAccount: tokenAccountPk,
|
|
owner: mangoAccount.owner,
|
|
})
|
|
.remainingAccounts(
|
|
healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({ pubkey: pk, isWritable: false, isSigner: false } as AccountMeta),
|
|
),
|
|
)
|
|
.preInstructions([
|
|
// ensure withdraws don't fail with missing ATAs
|
|
await createAssociatedTokenAccountIdempotentInstruction(
|
|
mangoAccount.owner,
|
|
mangoAccount.owner,
|
|
bank.mint,
|
|
),
|
|
])
|
|
.transaction();
|
|
// .rpc({ skipPreflight: true });
|
|
|
|
return await sendTransaction(
|
|
this.program.provider as AnchorProvider,
|
|
transaction,
|
|
{
|
|
postSendTxCallback: this.postSendTxCallback,
|
|
},
|
|
);
|
|
}
|
|
|
|
// Serum
|
|
|
|
public async serum3RegisterMarket(
|
|
group: Group,
|
|
serum3MarketExternalPk: PublicKey,
|
|
baseBank: Bank,
|
|
quoteBank: Bank,
|
|
marketIndex: number,
|
|
name: string,
|
|
): Promise<TransactionSignature> {
|
|
return await this.program.methods
|
|
.serum3RegisterMarket(marketIndex, name)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
serumProgram: SERUM3_PROGRAM_ID[this.cluster],
|
|
serumMarketExternal: serum3MarketExternalPk,
|
|
baseBank: baseBank.publicKey,
|
|
quoteBank: quoteBank.publicKey,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async serum3deregisterMarket(
|
|
group: Group,
|
|
serum3MarketName: string,
|
|
): Promise<TransactionSignature> {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
return await this.program.methods
|
|
.serum3DeregisterMarket()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
serumMarket: serum3Market.publicKey,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async serum3GetMarkets(
|
|
group: Group,
|
|
baseTokenIndex?: number,
|
|
quoteTokenIndex?: number,
|
|
): Promise<Serum3Market[]> {
|
|
const bumpfbuf = Buffer.alloc(1);
|
|
bumpfbuf.writeUInt8(255);
|
|
|
|
const filters: MemcmpFilter[] = [
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
];
|
|
|
|
if (baseTokenIndex) {
|
|
const bbuf = Buffer.alloc(2);
|
|
bbuf.writeUInt16LE(baseTokenIndex);
|
|
filters.push({
|
|
memcmp: {
|
|
bytes: bs58.encode(bbuf),
|
|
offset: 40,
|
|
},
|
|
});
|
|
}
|
|
|
|
if (quoteTokenIndex) {
|
|
const qbuf = Buffer.alloc(2);
|
|
qbuf.writeUInt16LE(quoteTokenIndex);
|
|
filters.push({
|
|
memcmp: {
|
|
bytes: bs58.encode(qbuf),
|
|
offset: 42,
|
|
},
|
|
});
|
|
}
|
|
|
|
return (await this.program.account.serum3Market.all(filters)).map((tuple) =>
|
|
Serum3Market.from(tuple.publicKey, tuple.account),
|
|
);
|
|
}
|
|
|
|
public async serum3CreateOpenOrders(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
marketName: string,
|
|
): Promise<TransactionSignature> {
|
|
const serum3Market: Serum3Market = group.serum3MarketsMap.get(marketName)!;
|
|
|
|
return await this.program.methods
|
|
.serum3CreateOpenOrders()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: serum3Market.serumProgram,
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async serum3CloseOpenOrders(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
serum3MarketName: string,
|
|
): Promise<TransactionSignature> {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
const openOrders = mangoAccount.serum3.find(
|
|
(account) => account.marketIndex === serum3Market.marketIndex,
|
|
)?.openOrders;
|
|
|
|
return await this.program.methods
|
|
.serum3CloseOpenOrders()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: serum3Market.serumProgram,
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
openOrders,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async serum3PlaceOrder(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
serum3MarketName: string,
|
|
side: Serum3Side,
|
|
price: number,
|
|
size: number,
|
|
selfTradeBehavior: Serum3SelfTradeBehavior,
|
|
orderType: Serum3OrderType,
|
|
clientOrderId: number,
|
|
limit: number,
|
|
) {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
if (!mangoAccount.findSerum3Account(serum3Market.marketIndex)) {
|
|
await this.serum3CreateOpenOrders(group, mangoAccount, 'BTC/USDC');
|
|
mangoAccount = await this.getMangoAccount(mangoAccount);
|
|
}
|
|
|
|
const serum3MarketExternal =
|
|
group.serum3MarketExternalsMap.get(serum3MarketName)!;
|
|
|
|
const serum3MarketExternalVaultSigner =
|
|
await PublicKey.createProgramAddress(
|
|
[
|
|
serum3Market.serumMarketExternal.toBuffer(),
|
|
serum3MarketExternal.decoded.vaultSignerNonce.toArrayLike(
|
|
Buffer,
|
|
'le',
|
|
8,
|
|
),
|
|
],
|
|
SERUM3_PROGRAM_ID[this.cluster],
|
|
);
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(AccountRetriever.Fixed, group, [
|
|
mangoAccount,
|
|
]);
|
|
|
|
const limitPrice = serum3MarketExternal.priceNumberToLots(price);
|
|
const maxBaseQuantity = serum3MarketExternal.baseSizeNumberToLots(size);
|
|
const feeTier = getFeeTier(0, 0 /** TODO: fix msrm/srm balance */);
|
|
const rates = getFeeRates(feeTier);
|
|
const maxQuoteQuantity = new BN(
|
|
serum3MarketExternal.decoded.quoteLotSize.toNumber() *
|
|
(1 + rates.taker) /** TODO: fix taker/maker */,
|
|
).mul(
|
|
serum3MarketExternal
|
|
.baseSizeNumberToLots(size)
|
|
.mul(serum3MarketExternal.priceNumberToLots(price)),
|
|
);
|
|
|
|
return await this.program.methods
|
|
.serum3PlaceOrder(
|
|
side,
|
|
limitPrice,
|
|
maxBaseQuantity,
|
|
maxQuoteQuantity,
|
|
selfTradeBehavior,
|
|
orderType,
|
|
new BN(clientOrderId),
|
|
limit,
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
openOrders: mangoAccount.findSerum3Account(serum3Market.marketIndex)
|
|
?.openOrders,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: SERUM3_PROGRAM_ID[this.cluster],
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
marketBids: serum3MarketExternal.bidsAddress,
|
|
marketAsks: serum3MarketExternal.asksAddress,
|
|
marketEventQueue: serum3MarketExternal.decoded.eventQueue,
|
|
marketRequestQueue: serum3MarketExternal.decoded.requestQueue,
|
|
marketBaseVault: serum3MarketExternal.decoded.baseVault,
|
|
marketQuoteVault: serum3MarketExternal.decoded.quoteVault,
|
|
marketVaultSigner: serum3MarketExternalVaultSigner,
|
|
quoteBank: group.getFirstBankByTokenIndex(serum3Market.quoteTokenIndex)
|
|
.publicKey,
|
|
quoteVault: group.getFirstBankByTokenIndex(serum3Market.quoteTokenIndex)
|
|
.vault,
|
|
baseBank: group.getFirstBankByTokenIndex(serum3Market.baseTokenIndex)
|
|
.publicKey,
|
|
baseVault: group.getFirstBankByTokenIndex(serum3Market.baseTokenIndex)
|
|
.vault,
|
|
})
|
|
.remainingAccounts(
|
|
healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({ pubkey: pk, isWritable: false, isSigner: false } as AccountMeta),
|
|
),
|
|
)
|
|
.rpc();
|
|
}
|
|
|
|
async serum3CancelAllorders(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
serum3MarketName: string,
|
|
limit: number,
|
|
) {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
const serum3MarketExternal =
|
|
group.serum3MarketExternalsMap.get(serum3MarketName)!;
|
|
|
|
return await this.program.methods
|
|
.serum3CancelAllOrders(limit)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
openOrders: mangoAccount.findSerum3Account(serum3Market.marketIndex)
|
|
?.openOrders,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: SERUM3_PROGRAM_ID[this.cluster],
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
marketBids: serum3MarketExternal.bidsAddress,
|
|
marketAsks: serum3MarketExternal.asksAddress,
|
|
marketEventQueue: serum3MarketExternal.decoded.eventQueue,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
async serum3SettleFunds(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
serum3MarketName: string,
|
|
): Promise<TransactionSignature> {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
const serum3MarketExternal =
|
|
group.serum3MarketExternalsMap.get(serum3MarketName)!;
|
|
|
|
const serum3MarketExternalVaultSigner =
|
|
// TODO: put into a helper method, and remove copy pasta
|
|
await PublicKey.createProgramAddress(
|
|
[
|
|
serum3Market.serumMarketExternal.toBuffer(),
|
|
serum3MarketExternal.decoded.vaultSignerNonce.toArrayLike(
|
|
Buffer,
|
|
'le',
|
|
8,
|
|
),
|
|
],
|
|
SERUM3_PROGRAM_ID[this.cluster],
|
|
);
|
|
|
|
return await this.program.methods
|
|
.serum3SettleFunds()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
openOrders: mangoAccount.findSerum3Account(serum3Market.marketIndex)
|
|
?.openOrders,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: SERUM3_PROGRAM_ID[this.cluster],
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
marketBaseVault: serum3MarketExternal.decoded.baseVault,
|
|
marketQuoteVault: serum3MarketExternal.decoded.quoteVault,
|
|
marketVaultSigner: serum3MarketExternalVaultSigner,
|
|
quoteBank: group.getFirstBankByTokenIndex(serum3Market.quoteTokenIndex)
|
|
.publicKey,
|
|
quoteVault: group.getFirstBankByTokenIndex(serum3Market.quoteTokenIndex)
|
|
.vault,
|
|
baseBank: group.getFirstBankByTokenIndex(serum3Market.baseTokenIndex)
|
|
.publicKey,
|
|
baseVault: group.getFirstBankByTokenIndex(serum3Market.baseTokenIndex)
|
|
.vault,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
async serum3CancelOrder(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
serum3MarketName: string,
|
|
side: Serum3Side,
|
|
orderId: BN,
|
|
): Promise<TransactionSignature> {
|
|
const serum3Market = group.serum3MarketsMap.get(serum3MarketName)!;
|
|
|
|
const serum3MarketExternal =
|
|
group.serum3MarketExternalsMap.get(serum3MarketName)!;
|
|
|
|
return await this.program.methods
|
|
.serum3CancelOrder(side, orderId)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
openOrders: mangoAccount.findSerum3Account(serum3Market.marketIndex)
|
|
?.openOrders,
|
|
serumMarket: serum3Market.publicKey,
|
|
serumProgram: SERUM3_PROGRAM_ID[this.cluster],
|
|
serumMarketExternal: serum3Market.serumMarketExternal,
|
|
marketBids: serum3MarketExternal.bidsAddress,
|
|
marketAsks: serum3MarketExternal.asksAddress,
|
|
marketEventQueue: serum3MarketExternal.decoded.eventQueue,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
async getSerum3Orders(
|
|
group: Group,
|
|
serum3MarketName: string,
|
|
): Promise<Order[]> {
|
|
const serum3MarketExternal =
|
|
group.serum3MarketExternalsMap.get(serum3MarketName)!;
|
|
|
|
// TODO: filter for mango account
|
|
return await serum3MarketExternal.loadOrdersForOwner(
|
|
this.program.provider.connection,
|
|
group.publicKey,
|
|
);
|
|
}
|
|
|
|
/// perps
|
|
|
|
async perpCreateMarket(
|
|
group: Group,
|
|
oraclePk: PublicKey,
|
|
perpMarketIndex: number,
|
|
name: string,
|
|
oracleConfFilter: number,
|
|
baseTokenIndex: number,
|
|
baseTokenDecimals: number,
|
|
quoteTokenIndex: number,
|
|
quoteLotSize: number,
|
|
baseLotSize: number,
|
|
maintAssetWeight: number,
|
|
initAssetWeight: number,
|
|
maintLiabWeight: number,
|
|
initLiabWeight: number,
|
|
liquidationFee: number,
|
|
makerFee: number,
|
|
takerFee: number,
|
|
minFunding: number,
|
|
maxFunding: number,
|
|
impactQuantity: number,
|
|
): Promise<TransactionSignature> {
|
|
const bids = new Keypair();
|
|
const asks = new Keypair();
|
|
const eventQueue = new Keypair();
|
|
|
|
return await this.program.methods
|
|
.perpCreateMarket(
|
|
perpMarketIndex,
|
|
name,
|
|
{
|
|
confFilter: {
|
|
val: I80F48.fromNumber(oracleConfFilter).getData(),
|
|
},
|
|
} as any, // future: nested custom types dont typecheck, fix if possible?
|
|
baseTokenIndex,
|
|
baseTokenDecimals,
|
|
new BN(quoteLotSize),
|
|
new BN(baseLotSize),
|
|
maintAssetWeight,
|
|
initAssetWeight,
|
|
maintLiabWeight,
|
|
initLiabWeight,
|
|
liquidationFee,
|
|
makerFee,
|
|
takerFee,
|
|
minFunding,
|
|
maxFunding,
|
|
new BN(impactQuantity),
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
oracle: oraclePk,
|
|
bids: bids.publicKey,
|
|
asks: asks.publicKey,
|
|
eventQueue: eventQueue.publicKey,
|
|
payer: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.preInstructions([
|
|
// TODO: try to pick up sizes of bookside and eventqueue from IDL, so we can stay in sync with program
|
|
|
|
// book sides
|
|
SystemProgram.createAccount({
|
|
programId: this.program.programId,
|
|
space: 8 + 98584,
|
|
lamports:
|
|
await this.program.provider.connection.getMinimumBalanceForRentExemption(
|
|
8 + 98584,
|
|
),
|
|
fromPubkey: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
newAccountPubkey: bids.publicKey,
|
|
}),
|
|
SystemProgram.createAccount({
|
|
programId: this.program.programId,
|
|
space: 8 + 98584,
|
|
lamports:
|
|
await this.program.provider.connection.getMinimumBalanceForRentExemption(
|
|
8 + 98584,
|
|
),
|
|
fromPubkey: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
newAccountPubkey: asks.publicKey,
|
|
}),
|
|
// event queue
|
|
SystemProgram.createAccount({
|
|
programId: this.program.programId,
|
|
space: 8 + 4 * 2 + 8 + 488 * 208,
|
|
lamports:
|
|
await this.program.provider.connection.getMinimumBalanceForRentExemption(
|
|
8 + 4 * 2 + 8 + 488 * 208,
|
|
),
|
|
fromPubkey: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
newAccountPubkey: eventQueue.publicKey,
|
|
}),
|
|
])
|
|
.signers([bids, asks, eventQueue])
|
|
.rpc();
|
|
}
|
|
|
|
async perpEditMarket(
|
|
group: Group,
|
|
perpMarketName: string,
|
|
oracle: PublicKey,
|
|
oracleConfFilter: number,
|
|
baseTokenIndex: number,
|
|
baseTokenDecimals: number,
|
|
maintAssetWeight: number,
|
|
initAssetWeight: number,
|
|
maintLiabWeight: number,
|
|
initLiabWeight: number,
|
|
liquidationFee: number,
|
|
makerFee: number,
|
|
takerFee: number,
|
|
minFunding: number,
|
|
maxFunding: number,
|
|
impactQuantity: number,
|
|
): Promise<TransactionSignature> {
|
|
const perpMarket = group.perpMarketsMap.get(perpMarketName)!;
|
|
|
|
return await this.program.methods
|
|
.perpEditMarket(
|
|
oracle,
|
|
{
|
|
confFilter: {
|
|
val: I80F48.fromNumber(oracleConfFilter).getData(),
|
|
},
|
|
} as any, // future: nested custom types dont typecheck, fix if possible?
|
|
baseTokenIndex,
|
|
baseTokenDecimals,
|
|
maintAssetWeight,
|
|
initAssetWeight,
|
|
maintLiabWeight,
|
|
initLiabWeight,
|
|
liquidationFee,
|
|
makerFee,
|
|
takerFee,
|
|
minFunding,
|
|
maxFunding,
|
|
new BN(impactQuantity),
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
perpMarket: perpMarket.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
async perpCloseMarket(
|
|
group: Group,
|
|
perpMarketName: string,
|
|
): Promise<TransactionSignature> {
|
|
const perpMarket = group.perpMarketsMap.get(perpMarketName)!;
|
|
|
|
return await this.program.methods
|
|
.perpCloseMarket()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
admin: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
perpMarket: perpMarket.publicKey,
|
|
asks: perpMarket.asks,
|
|
bids: perpMarket.bids,
|
|
eventQueue: perpMarket.eventQueue,
|
|
solDestination: (this.program.provider as AnchorProvider).wallet
|
|
.publicKey,
|
|
})
|
|
.rpc();
|
|
}
|
|
|
|
public async perpGetMarkets(
|
|
group: Group,
|
|
baseTokenIndex?: number,
|
|
): Promise<PerpMarket[]> {
|
|
const bumpfbuf = Buffer.alloc(1);
|
|
bumpfbuf.writeUInt8(255);
|
|
|
|
const filters: MemcmpFilter[] = [
|
|
{
|
|
memcmp: {
|
|
bytes: group.publicKey.toBase58(),
|
|
offset: 8,
|
|
},
|
|
},
|
|
];
|
|
|
|
if (baseTokenIndex) {
|
|
const bbuf = Buffer.alloc(2);
|
|
bbuf.writeUInt16LE(baseTokenIndex);
|
|
filters.push({
|
|
memcmp: {
|
|
bytes: bs58.encode(bbuf),
|
|
offset: 40,
|
|
},
|
|
});
|
|
}
|
|
|
|
return (await this.program.account.perpMarket.all(filters)).map((tuple) =>
|
|
PerpMarket.from(tuple.publicKey, tuple.account),
|
|
);
|
|
}
|
|
|
|
async perpPlaceOrder(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
perpMarketName: string,
|
|
side: Side,
|
|
price: number,
|
|
quantity: number,
|
|
maxQuoteQuantity: number,
|
|
clientOrderId: number,
|
|
orderType: OrderType,
|
|
expiryTimestamp: number,
|
|
limit: number,
|
|
) {
|
|
const perpMarket = group.perpMarketsMap.get(perpMarketName)!;
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(AccountRetriever.Fixed, group, [
|
|
mangoAccount,
|
|
]);
|
|
|
|
const [nativePrice, nativeQuantity] = perpMarket.uiToNativePriceQuantity(
|
|
price,
|
|
quantity,
|
|
);
|
|
|
|
const maxQuoteQuantityLots = maxQuoteQuantity
|
|
? perpMarket.uiQuoteToLots(maxQuoteQuantity)
|
|
: I64_MAX_BN;
|
|
|
|
await this.program.methods
|
|
.perpPlaceOrder(
|
|
side,
|
|
nativePrice,
|
|
nativeQuantity,
|
|
maxQuoteQuantityLots,
|
|
new BN(clientOrderId),
|
|
orderType,
|
|
new BN(expiryTimestamp),
|
|
limit,
|
|
)
|
|
.accounts({
|
|
group: group.publicKey,
|
|
account: mangoAccount.publicKey,
|
|
perpMarket: perpMarket.publicKey,
|
|
asks: perpMarket.asks,
|
|
bids: perpMarket.bids,
|
|
eventQueue: perpMarket.eventQueue,
|
|
oracle: perpMarket.oracle,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.remainingAccounts(
|
|
healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({ pubkey: pk, isWritable: false, isSigner: false } as AccountMeta),
|
|
),
|
|
)
|
|
.rpc();
|
|
}
|
|
|
|
public async marginTrade({
|
|
group,
|
|
mangoAccount,
|
|
inputMintPk,
|
|
amountIn,
|
|
outputMintPk,
|
|
userDefinedInstructions,
|
|
// margin trade is a general function
|
|
// set flash_loan_type to FlashLoanType.swap if you desire the transaction to be recorded as a swap
|
|
flashLoanType,
|
|
}: {
|
|
group: Group;
|
|
mangoAccount: MangoAccount;
|
|
inputMintPk: PublicKey;
|
|
amountIn: number;
|
|
outputMintPk: PublicKey;
|
|
userDefinedInstructions: TransactionInstruction[];
|
|
flashLoanType: FlashLoanType;
|
|
}): Promise<TransactionSignature> {
|
|
const inputBank: Bank = group.getFirstBankByMint(inputMintPk);
|
|
const outputBank: Bank = group.getFirstBankByMint(outputMintPk);
|
|
|
|
if (!inputBank || !outputBank) throw new Error('Invalid token');
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(
|
|
AccountRetriever.Fixed,
|
|
group,
|
|
[mangoAccount],
|
|
[inputBank, outputBank],
|
|
);
|
|
const parsedHealthAccounts = healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({
|
|
pubkey: pk,
|
|
isWritable: false,
|
|
isSigner: false,
|
|
} as AccountMeta),
|
|
);
|
|
|
|
/*
|
|
* Find or create associated token accounts
|
|
*/
|
|
const inputTokenAccountPk = await getAssociatedTokenAddress(
|
|
inputBank.mint,
|
|
mangoAccount.owner,
|
|
);
|
|
const inputTokenAccExists =
|
|
await this.program.provider.connection.getAccountInfo(
|
|
inputTokenAccountPk,
|
|
);
|
|
const preInstructions = [];
|
|
if (!inputTokenAccExists) {
|
|
preInstructions.push(
|
|
await createAssociatedTokenAccountIdempotentInstruction(
|
|
mangoAccount.owner,
|
|
mangoAccount.owner,
|
|
inputBank.mint,
|
|
),
|
|
);
|
|
}
|
|
|
|
const outputTokenAccountPk = await getAssociatedTokenAddress(
|
|
outputBank.mint,
|
|
mangoAccount.owner,
|
|
);
|
|
const outputTokenAccExists =
|
|
await this.program.provider.connection.getAccountInfo(
|
|
outputTokenAccountPk,
|
|
);
|
|
if (!outputTokenAccExists) {
|
|
preInstructions.push(
|
|
await createAssociatedTokenAccountIdempotentInstruction(
|
|
mangoAccount.owner,
|
|
mangoAccount.owner,
|
|
outputBank.mint,
|
|
),
|
|
);
|
|
}
|
|
|
|
const inputBankAccount = {
|
|
pubkey: inputBank.publicKey,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
};
|
|
const outputBankAccount = {
|
|
pubkey: outputBank.publicKey,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
};
|
|
const inputBankVault = {
|
|
pubkey: inputBank.vault,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
};
|
|
const outputBankVault = {
|
|
pubkey: outputBank.vault,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
};
|
|
const inputATA = {
|
|
pubkey: inputTokenAccountPk,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
};
|
|
const outputATA = {
|
|
pubkey: outputTokenAccountPk,
|
|
isWritable: false,
|
|
isSigner: false,
|
|
};
|
|
const groupAM = {
|
|
pubkey: group.publicKey,
|
|
isWritable: false,
|
|
isSigner: false,
|
|
};
|
|
|
|
const flashLoanEndIx = await this.program.methods
|
|
.flashLoanEnd(flashLoanType)
|
|
.accounts({
|
|
account: mangoAccount.publicKey,
|
|
owner: (this.program.provider as AnchorProvider).wallet.publicKey,
|
|
})
|
|
.remainingAccounts([
|
|
...parsedHealthAccounts,
|
|
inputBankVault,
|
|
outputBankVault,
|
|
inputATA,
|
|
{
|
|
isWritable: true,
|
|
pubkey: outputTokenAccountPk,
|
|
isSigner: false,
|
|
},
|
|
groupAM,
|
|
])
|
|
.instruction();
|
|
|
|
const flashLoanBeginIx = await this.program.methods
|
|
.flashLoanBegin([
|
|
toNativeDecimals(amountIn, inputBank.mintDecimals),
|
|
new BN(
|
|
0,
|
|
) /* we don't care about borrowing the target amount, this is just a dummy */,
|
|
])
|
|
.accounts({
|
|
instructions: SYSVAR_INSTRUCTIONS_PUBKEY,
|
|
})
|
|
.remainingAccounts([
|
|
inputBankAccount,
|
|
outputBankAccount,
|
|
inputBankVault,
|
|
outputBankVault,
|
|
inputATA,
|
|
outputATA,
|
|
groupAM,
|
|
])
|
|
.instruction();
|
|
|
|
const tx = new Transaction();
|
|
for (const i of preInstructions) {
|
|
tx.add(i);
|
|
}
|
|
tx.add(flashLoanBeginIx);
|
|
for (const i of userDefinedInstructions) {
|
|
tx.add(i);
|
|
}
|
|
tx.add(flashLoanEndIx);
|
|
|
|
return await sendTransaction(this.program.provider as AnchorProvider, tx, {
|
|
postSendTxCallback: this.postSendTxCallback,
|
|
});
|
|
}
|
|
|
|
async updateIndexAndRate(group: Group, mintPk: PublicKey) {
|
|
// TODO: handle updating multiple banks
|
|
const bank = group.getFirstBankByMint(mintPk);
|
|
const mintInfo = group.mintInfosMapByMint.get(mintPk.toString())!;
|
|
|
|
await this.program.methods
|
|
.tokenUpdateIndexAndRate()
|
|
.accounts({
|
|
group: group.publicKey,
|
|
mintInfo: mintInfo.publicKey,
|
|
oracle: mintInfo.oracle,
|
|
instructions: SYSVAR_INSTRUCTIONS_PUBKEY,
|
|
})
|
|
.remainingAccounts([
|
|
{
|
|
pubkey: bank.publicKey,
|
|
isWritable: true,
|
|
isSigner: false,
|
|
} as AccountMeta,
|
|
])
|
|
.rpc();
|
|
}
|
|
|
|
/// liquidations
|
|
|
|
async liqTokenWithToken(
|
|
group: Group,
|
|
liqor: MangoAccount,
|
|
liqee: MangoAccount,
|
|
assetMintPk: PublicKey,
|
|
liabMintPk: PublicKey,
|
|
maxLiabTransfer: number,
|
|
) {
|
|
const assetBank: Bank = group.getFirstBankByMint(assetMintPk);
|
|
const liabBank: Bank = group.getFirstBankByMint(liabMintPk);
|
|
|
|
const healthRemainingAccounts: PublicKey[] =
|
|
this.buildHealthRemainingAccounts(
|
|
AccountRetriever.Scanning,
|
|
group,
|
|
[liqor, liqee],
|
|
[assetBank, liabBank],
|
|
);
|
|
|
|
const parsedHealthAccounts = healthRemainingAccounts.map(
|
|
(pk) =>
|
|
({
|
|
pubkey: pk,
|
|
isWritable:
|
|
pk.equals(assetBank.publicKey) || pk.equals(liabBank.publicKey)
|
|
? true
|
|
: false,
|
|
isSigner: false,
|
|
} as AccountMeta),
|
|
);
|
|
|
|
await this.program.methods
|
|
.liqTokenWithToken(assetBank.tokenIndex, liabBank.tokenIndex, {
|
|
val: I80F48.fromNumber(maxLiabTransfer).getData(),
|
|
})
|
|
.accounts({
|
|
group: group.publicKey,
|
|
liqor: liqor.publicKey,
|
|
liqee: liqee.publicKey,
|
|
liqorOwner: liqor.owner,
|
|
})
|
|
.remainingAccounts(parsedHealthAccounts)
|
|
.rpc();
|
|
}
|
|
|
|
/// static
|
|
|
|
static connect(
|
|
provider: Provider,
|
|
cluster: Cluster,
|
|
programId: PublicKey,
|
|
opts: any = {},
|
|
getIdsFromApi: IdsSource = 'api',
|
|
): MangoClient {
|
|
// TODO: use IDL on chain or in repository? decide...
|
|
// Alternatively we could fetch IDL from chain.
|
|
// const idl = await Program.fetchIdl(MANGO_V4_ID, provider);
|
|
const idl = IDL;
|
|
|
|
return new MangoClient(
|
|
new Program<MangoV4>(idl as MangoV4, programId, provider),
|
|
programId,
|
|
cluster,
|
|
opts,
|
|
getIdsFromApi,
|
|
);
|
|
}
|
|
|
|
static connectForGroupName(
|
|
provider: Provider,
|
|
groupName: string,
|
|
): MangoClient {
|
|
// TODO: use IDL on chain or in repository? decide...
|
|
// Alternatively we could fetch IDL from chain.
|
|
// const idl = await Program.fetchIdl(MANGO_V4_ID, provider);
|
|
const idl = IDL;
|
|
|
|
const id = Id.fromIdsByName(groupName);
|
|
|
|
return new MangoClient(
|
|
new Program<MangoV4>(
|
|
idl as MangoV4,
|
|
new PublicKey(id.mangoProgramId),
|
|
provider,
|
|
),
|
|
new PublicKey(id.mangoProgramId),
|
|
id.cluster,
|
|
);
|
|
}
|
|
|
|
/// private
|
|
|
|
public buildHealthRemainingAccounts(
|
|
retriever: AccountRetriever,
|
|
group: Group,
|
|
mangoAccounts: MangoAccount[],
|
|
banks?: Bank[] /** TODO for serum3PlaceOrder we are just ingoring this atm */,
|
|
): PublicKey[] {
|
|
if (retriever === AccountRetriever.Fixed) {
|
|
return this.buildFixedAccountRetrieverHealthAccounts(
|
|
group,
|
|
mangoAccounts[0],
|
|
banks,
|
|
);
|
|
} else {
|
|
return this.buildScanningAccountRetrieverHealthAccounts(
|
|
group,
|
|
mangoAccounts,
|
|
banks,
|
|
);
|
|
}
|
|
}
|
|
|
|
public buildFixedAccountRetrieverHealthAccounts(
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
banks?: Bank[] /** TODO for serum3PlaceOrder we are just ingoring this atm */,
|
|
): PublicKey[] {
|
|
const healthRemainingAccounts: PublicKey[] = [];
|
|
|
|
const allTokenIndices = mangoAccount.tokens.map(
|
|
(token) => token.tokenIndex,
|
|
);
|
|
if (banks) {
|
|
for (const bank of banks) {
|
|
if (allTokenIndices.indexOf(bank.tokenIndex) < 0) {
|
|
allTokenIndices[
|
|
mangoAccount.tokens.findIndex(
|
|
(token, index) =>
|
|
!token.isActive() &&
|
|
allTokenIndices[index] == TokenPosition.TokenIndexUnset,
|
|
)
|
|
] = bank.tokenIndex;
|
|
}
|
|
}
|
|
}
|
|
const mintInfos = allTokenIndices
|
|
.filter((index) => index != TokenPosition.TokenIndexUnset)
|
|
.map((tokenIndex) => group.mintInfosMapByTokenIndex.get(tokenIndex)!);
|
|
|
|
healthRemainingAccounts.push(
|
|
...mintInfos.map((mintInfo) => mintInfo.firstBank()),
|
|
);
|
|
healthRemainingAccounts.push(
|
|
...mintInfos.map((mintInfo) => mintInfo.oracle),
|
|
);
|
|
healthRemainingAccounts.push(
|
|
...mangoAccount.serum3
|
|
.filter((serum3Account) => serum3Account.marketIndex !== 65535)
|
|
.map((serum3Account) => serum3Account.openOrders),
|
|
);
|
|
healthRemainingAccounts.push(
|
|
...mangoAccount.perps
|
|
.filter((perp) => perp.marketIndex !== 65535)
|
|
.map(
|
|
(perp) =>
|
|
Array.from(group.perpMarketsMap.values()).filter(
|
|
(perpMarket) => perpMarket.perpMarketIndex === perp.marketIndex,
|
|
)[0].publicKey,
|
|
),
|
|
);
|
|
|
|
return healthRemainingAccounts;
|
|
}
|
|
|
|
public buildScanningAccountRetrieverHealthAccounts(
|
|
group: Group,
|
|
mangoAccounts: MangoAccount[],
|
|
banks?: Bank[] /** TODO for serum3PlaceOrder we are just ingoring this atm */,
|
|
): PublicKey[] {
|
|
const healthRemainingAccounts: PublicKey[] = [];
|
|
|
|
let tokenIndices = [];
|
|
for (const mangoAccount of mangoAccounts) {
|
|
tokenIndices.push(
|
|
...mangoAccount.tokens
|
|
.filter((token) => token.tokenIndex !== 65535)
|
|
.map((token) => token.tokenIndex),
|
|
);
|
|
}
|
|
tokenIndices = [...new Set(tokenIndices)];
|
|
|
|
if (banks?.length) {
|
|
for (const bank of banks) {
|
|
tokenIndices.push(bank.tokenIndex);
|
|
}
|
|
}
|
|
const mintInfos = [...new Set(tokenIndices)].map(
|
|
(tokenIndex) => group.mintInfosMapByTokenIndex.get(tokenIndex)!,
|
|
);
|
|
healthRemainingAccounts.push(
|
|
...mintInfos.map((mintInfo) => mintInfo.firstBank()),
|
|
);
|
|
healthRemainingAccounts.push(
|
|
...mintInfos.map((mintInfo) => mintInfo.oracle),
|
|
);
|
|
for (const mangoAccount of mangoAccounts) {
|
|
healthRemainingAccounts.push(
|
|
...mangoAccount.serum3
|
|
.filter((serum3Account) => serum3Account.marketIndex !== 65535)
|
|
.map((serum3Account) => serum3Account.openOrders),
|
|
);
|
|
}
|
|
for (const mangoAccount of mangoAccounts) {
|
|
healthRemainingAccounts.push(
|
|
...mangoAccount.perps
|
|
.filter((perp) => perp.marketIndex !== 65535)
|
|
.map(
|
|
(perp) =>
|
|
Array.from(group.perpMarketsMap.values()).filter(
|
|
(perpMarket) => perpMarket.perpMarketIndex === perp.marketIndex,
|
|
)[0].publicKey,
|
|
),
|
|
);
|
|
}
|
|
|
|
return healthRemainingAccounts;
|
|
}
|
|
}
|