blockworks-foundation
/
mango-v4
mirror of https://github.com/blockworks-foundation/mango-v4.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

deposit and withdraw examples 

Signed-off-by: microwavedcola1 <microwavedcola@gmail.com>
This commit is contained in:
microwavedcola1 2022-04-07 18:58:42 +02:00
parent 70e39f6333
commit dc26190a81
5 changed files with 143 additions and 42 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
ts/accounts/types/bank.ts
View File

@ -55,8 +55,8 @@ export class Bank {
constructor(
public publicKey: PublicKey,
group: PublicKey,
mint: PublicKey,
public group: PublicKey,
public mint: PublicKey,
public vault: PublicKey,
public oracle: PublicKey,
depositIndex: I80F48Dto,

6
ts/accounts/types/mangoAccount.ts
View File

@ -48,9 +48,9 @@ export class MangoAccount {
constructor(
public publicKey: PublicKey,
group: PublicKey,
owner: PublicKey,
delegate: PublicKey,
public group: PublicKey,
public owner: PublicKey,
public delegate: PublicKey,
tokens: { values: TokenAccountDto[] },
serum3: { values: Serum3AccountDto[] },
perps: unknown,

167
ts/client.ts
View File

@ -1,11 +1,18 @@
import { Program, Provider } from '@project-serum/anchor';
import * as spl from '@solana/spl-token';
import { PublicKey, TransactionSignature } from '@solana/web3.js';
import { Bank, getBanksForGroup } from './accounts/types/bank';
import {
Bank,
getBanksForGroup,
getMintInfoForTokenIndex,
} from './accounts/types/bank';
import { getGroupForAdmin, Group } from './accounts/types/group';
import {
createMangoAccount,
deposit,
getMangoAccountsForGroupAndOwner,
MangoAccount,
withdraw,
} from './accounts/types/mangoAccount';
import { IDL, MangoV4 } from './mango_v4';
@ -16,6 +23,93 @@ export const MANGO_V4_ID = new PublicKey(
export class MangoClient {
constructor(public program: Program<MangoV4>, public devnet?: boolean) {}
/// public
public async getGroup(adminPk: PublicKey): Promise<Group> {
return await getGroupForAdmin(this, adminPk);
}
public async getBanksForGroup(group: Group): Promise<Bank[]> {
return await getBanksForGroup(this, group.publicKey);
}
public async createMangoAccount(
group: Group,
ownerPk: PublicKey,
accountNumber: number,
): Promise<TransactionSignature> {
return createMangoAccount(this, group.publicKey, ownerPk, accountNumber);
}
public async getMangoAccount(
group: Group,
ownerPk: PublicKey,
): Promise<MangoAccount[]> {
return await getMangoAccountsForGroupAndOwner(
this,
group.publicKey,
ownerPk,
);
}
public async deposit(
group: Group,
mangoAccount: MangoAccount,
bank: Bank,
amount: number,
) {
const tokenAccountPk = await spl.getAssociatedTokenAddress(
bank.mint,
mangoAccount.owner,
);
const healthRemainingAccounts: PublicKey[] =
await this.buildHealthRemainingAccounts(group, mangoAccount, bank);
return await deposit(
this,
group.publicKey,
mangoAccount.publicKey,
bank.publicKey,
bank.vault,
tokenAccountPk,
mangoAccount.owner,
healthRemainingAccounts,
amount,
);
}
public async withdraw(
group: Group,
mangoAccount: MangoAccount,
bank: Bank,
amount: number,
allowBorrow: boolean,
) {
const tokenAccountPk = await spl.getAssociatedTokenAddress(
bank.mint,
mangoAccount.owner,
);
const healthRemainingAccounts: PublicKey[] =
await this.buildHealthRemainingAccounts(group, mangoAccount, bank);
return await withdraw(
this,
group.publicKey,
mangoAccount.publicKey,
bank.publicKey,
bank.vault,
tokenAccountPk,
mangoAccount.owner,
healthRemainingAccounts,
amount,
allowBorrow,
);
}
/// static
static async connect(
provider: Provider,
devnet?: boolean,
@ -55,42 +149,45 @@ export class MangoClient {
);
}
public async getGroup(adminPk: PublicKey): Promise<Group> {
return await getGroupForAdmin(this, adminPk);
}
/// private
public async getBanksForGroup(group: Group): Promise<Bank[]> {
return await getBanksForGroup(this, group.publicKey);
}
public async createMangoAccount(
private async buildHealthRemainingAccounts(
group: Group,
ownerPk: PublicKey,
accountNumber: number,
): Promise<TransactionSignature> {
return createMangoAccount(this, group.publicKey, ownerPk, accountNumber);
}
mangoAccount: MangoAccount,
bank: Bank,
) {
const healthRemainingAccounts: PublicKey[] = [];
{
const tokenIndices = mangoAccount.tokens
.filter((token) => token.tokenIndex !== 65535)
.map((token) => token.tokenIndex);
tokenIndices.push(bank.tokenIndex);
public async getMangoAccount(
group: Group,
ownerPk: PublicKey,
): Promise<MangoAccount[]> {
return await getMangoAccountsForGroupAndOwner(
this,
group.publicKey,
ownerPk,
);
const mintInfos = await Promise.all(
[...new Set(tokenIndices)].map(async (tokenIndex) =>
getMintInfoForTokenIndex(this, group.publicKey, tokenIndex),
),
);
healthRemainingAccounts.push(
...mintInfos.flatMap((mintinfos) => {
return mintinfos.flatMap((mintinfo) => {
return mintinfo.bank;
});
}),
);
healthRemainingAccounts.push(
...mintInfos.flatMap((mintinfos) => {
return mintinfos.flatMap((mintinfo) => {
return mintinfo.oracle;
});
}),
);
healthRemainingAccounts.push(
...mangoAccount.serum3
.filter((serum3Account) => serum3Account.marketIndex !== 65535)
.map((serum3Account) => serum3Account.openOrders),
);
}
return healthRemainingAccounts;
}
// public async deposit(group: Group, mangoAccount: MangoAccount, bank: Bank) {
// return await deposit(
// this,
// group.publicKey,
// mangoAccount.publicKey,
// bank.publicKey,
// bank.vault,
// tokenAccountPk,
// ownerPk,
// );
// }
}

2
ts/example0.ts
View File

@ -280,7 +280,7 @@ async function main() {
1000,
);
console.log(`Witdrawing...500`);
console.log(`Withdrawing...500`);
await withdraw(
userClient,
group.publicKey,

6
ts/example1.ts
View File

@ -44,7 +44,11 @@ async function main() {
mangoAccount = mangoAccounts[0];
console.log(`MangoAccount ${mangoAccount.publicKey}`);
process.exit(0);
console.log(`Depositing...1000`);
await client.deposit(group, mangoAccount, banks[0], 1000);
console.log(`Withdrawing...500`);
await client.withdraw(group, mangoAccount, banks[0], 500, false);
}
main();