2022-03-31 23:59:39 -07:00
|
|
|
import { Provider, Wallet } from '@project-serum/anchor';
|
|
|
|
import { Connection, Keypair, PublicKey } from '@solana/web3.js';
|
2022-03-31 04:55:59 -07:00
|
|
|
import fs from 'fs';
|
|
|
|
import { MangoClient } from './client';
|
2022-03-31 23:59:39 -07:00
|
|
|
import {
|
|
|
|
closeMangoAccount,
|
|
|
|
createGroup,
|
|
|
|
createMangoAccount,
|
|
|
|
deposit,
|
|
|
|
getBank,
|
|
|
|
getBankForGroupAndMint,
|
2022-04-02 02:52:28 -07:00
|
|
|
getBanksForGroup,
|
2022-03-31 23:59:39 -07:00
|
|
|
getGroupForAdmin,
|
|
|
|
getMangoAccount,
|
|
|
|
getMangoAccountsForGroupAndOwner,
|
|
|
|
registerToken,
|
|
|
|
withdraw,
|
|
|
|
} from './instructions';
|
2022-03-23 07:39:11 -07:00
|
|
|
|
2022-04-02 02:52:28 -07:00
|
|
|
async function registerBank(
|
|
|
|
client: MangoClient,
|
|
|
|
group: any,
|
|
|
|
admin: Keypair,
|
|
|
|
mint: PublicKey,
|
|
|
|
oracle: PublicKey,
|
|
|
|
payer: Keypair,
|
|
|
|
tokenIndex: number,
|
|
|
|
) {
|
|
|
|
let banks = await getBankForGroupAndMint(client, group.publicKey, mint);
|
|
|
|
let bank;
|
|
|
|
if (banks.length > 0) {
|
|
|
|
bank = banks[0];
|
|
|
|
console.log(`Found bank ${bank.publicKey.toBase58()}`);
|
|
|
|
} else {
|
|
|
|
await registerToken(
|
|
|
|
client,
|
|
|
|
group.publicKey,
|
|
|
|
admin.publicKey,
|
|
|
|
mint,
|
|
|
|
oracle,
|
|
|
|
payer,
|
|
|
|
tokenIndex,
|
|
|
|
);
|
|
|
|
banks = await getBankForGroupAndMint(client, group.publicKey, mint);
|
|
|
|
bank = banks[0];
|
|
|
|
console.log(`Registered token ${bank.publicKey.toBase58()}`);
|
|
|
|
}
|
|
|
|
return bank;
|
|
|
|
}
|
|
|
|
|
2022-03-23 07:39:11 -07:00
|
|
|
async function main() {
|
2022-04-01 00:11:08 -07:00
|
|
|
//
|
|
|
|
// Setup
|
|
|
|
//
|
2022-03-23 07:39:11 -07:00
|
|
|
const options = Provider.defaultOptions();
|
2022-03-31 00:10:06 -07:00
|
|
|
const connection = new Connection(
|
|
|
|
'https://mango.devnet.rpcpool.com',
|
|
|
|
options,
|
|
|
|
);
|
|
|
|
|
2022-03-31 02:15:39 -07:00
|
|
|
const admin = Keypair.fromSecretKey(
|
2022-04-02 02:52:28 -07:00
|
|
|
Buffer.from(
|
|
|
|
JSON.parse(fs.readFileSync(process.env.ADMIN_KEYPAIR!, 'utf-8')),
|
|
|
|
),
|
2022-03-31 02:15:39 -07:00
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
const adminWallet = new Wallet(admin);
|
2022-04-02 02:52:28 -07:00
|
|
|
console.log(`admin ${adminWallet.publicKey.toBase58()}`);
|
|
|
|
const adminProvider = new Provider(connection, adminWallet, options);
|
|
|
|
const adminClient = await MangoClient.connect(adminProvider, true);
|
2022-03-31 02:15:39 -07:00
|
|
|
|
|
|
|
const payer = Keypair.fromSecretKey(
|
|
|
|
Buffer.from(
|
2022-04-01 00:04:26 -07:00
|
|
|
JSON.parse(fs.readFileSync(process.env.PAYER_KEYPAIR!, 'utf-8')),
|
2022-03-31 02:15:39 -07:00
|
|
|
),
|
2022-03-31 00:10:06 -07:00
|
|
|
);
|
2022-04-02 02:52:28 -07:00
|
|
|
console.log(`payer ${payer.publicKey.toBase58()}`);
|
2022-03-31 02:28:47 -07:00
|
|
|
//
|
2022-03-31 23:59:39 -07:00
|
|
|
// Find existing or create a new group
|
2022-03-31 02:28:47 -07:00
|
|
|
//
|
2022-04-02 02:52:28 -07:00
|
|
|
let groups = await getGroupForAdmin(adminClient, admin.publicKey);
|
2022-03-31 00:10:06 -07:00
|
|
|
let group;
|
2022-03-31 23:59:39 -07:00
|
|
|
if (groups.length > 0) {
|
|
|
|
group = groups[0];
|
|
|
|
console.log(`Found group ${group.publicKey.toBase58()}`);
|
2022-03-31 00:10:06 -07:00
|
|
|
} else {
|
2022-04-02 02:52:28 -07:00
|
|
|
await createGroup(adminClient, admin.publicKey, payer);
|
|
|
|
let groups = await getGroupForAdmin(adminClient, admin.publicKey);
|
2022-03-31 23:59:39 -07:00
|
|
|
group = groups[0];
|
|
|
|
console.log(`Created group ${group.publicKey.toBase58()}`);
|
2022-03-31 00:10:06 -07:00
|
|
|
}
|
|
|
|
|
2022-03-31 02:28:47 -07:00
|
|
|
//
|
2022-03-31 23:59:39 -07:00
|
|
|
// Find existing or register a new token
|
2022-03-31 02:28:47 -07:00
|
|
|
//
|
2022-04-02 02:52:28 -07:00
|
|
|
// TODO: replace with 4zMMC9srt5Ri5X14GAgXhaHii3GnPAEERYPJgZJDncDU,
|
|
|
|
// see https://developers.circle.com/docs/usdc-on-testnet#usdc-on-solana-testnet
|
|
|
|
const usdcDevnetMint = new PublicKey(
|
|
|
|
'8FRFC6MoGGkMFQwngccyu69VnYbzykGeez7ignHVAFSN',
|
|
|
|
);
|
|
|
|
// TODO: replace with a usdc devnet oracle
|
|
|
|
const usdtDevnetOracle = new PublicKey(
|
|
|
|
'38xoQ4oeJCBrcVvca2cGk7iV1dAfrmTR1kmhSCJQ8Jto',
|
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
const btcDevnetMint = new PublicKey(
|
2022-03-31 06:58:12 -07:00
|
|
|
'3UNBZ6o52WTWwjac2kPUb4FyodhU1vFkRJheu1Sh2TvU',
|
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
const btcDevnetOracle = new PublicKey(
|
2022-03-31 06:58:12 -07:00
|
|
|
'HovQMDrbAgAYPCmHVSrezcSmkMtXSSUsLDFANExrZh2J',
|
|
|
|
);
|
2022-04-02 02:52:28 -07:00
|
|
|
let btcBank = await registerBank(
|
|
|
|
adminClient,
|
|
|
|
group,
|
|
|
|
admin,
|
2022-03-31 23:59:39 -07:00
|
|
|
btcDevnetMint,
|
2022-04-02 02:52:28 -07:00
|
|
|
btcDevnetOracle,
|
|
|
|
payer,
|
|
|
|
0,
|
|
|
|
);
|
|
|
|
let usdcBank = await registerBank(
|
|
|
|
adminClient,
|
|
|
|
group,
|
|
|
|
admin,
|
|
|
|
usdcDevnetMint,
|
|
|
|
usdtDevnetOracle,
|
|
|
|
payer,
|
|
|
|
1,
|
2022-03-31 23:59:39 -07:00
|
|
|
);
|
2022-04-02 02:52:28 -07:00
|
|
|
|
|
|
|
//
|
|
|
|
// User operations
|
|
|
|
//
|
|
|
|
|
|
|
|
const user = Keypair.fromSecretKey(
|
|
|
|
Buffer.from(
|
|
|
|
JSON.parse(fs.readFileSync(process.env.USER_KEYPAIR!, 'utf-8')),
|
|
|
|
),
|
|
|
|
);
|
|
|
|
const userWallet = new Wallet(user);
|
|
|
|
const userProvider = new Provider(connection, userWallet, options);
|
|
|
|
const userClient = await MangoClient.connect(userProvider, true);
|
|
|
|
console.log(`user ${userWallet.publicKey.toBase58()}`);
|
2022-03-31 04:30:05 -07:00
|
|
|
|
|
|
|
//
|
2022-03-31 23:59:39 -07:00
|
|
|
// Create mango account
|
2022-03-31 04:30:05 -07:00
|
|
|
//
|
2022-03-31 23:59:39 -07:00
|
|
|
let accounts = await getMangoAccountsForGroupAndOwner(
|
2022-04-02 02:52:28 -07:00
|
|
|
userClient,
|
2022-03-31 23:59:39 -07:00
|
|
|
group.publicKey,
|
2022-04-02 02:52:28 -07:00
|
|
|
user.publicKey,
|
2022-03-31 06:58:12 -07:00
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
let account;
|
|
|
|
if (accounts.length > 0) {
|
|
|
|
account = accounts[0];
|
|
|
|
console.log(`Found mango account ${account.publicKey.toBase58()}`);
|
|
|
|
} else {
|
2022-04-02 02:52:28 -07:00
|
|
|
await createMangoAccount(
|
|
|
|
userClient,
|
|
|
|
group.publicKey,
|
|
|
|
user.publicKey,
|
|
|
|
payer,
|
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
accounts = await getMangoAccountsForGroupAndOwner(
|
2022-04-02 02:52:28 -07:00
|
|
|
userClient,
|
2022-03-31 23:59:39 -07:00
|
|
|
group.publicKey,
|
2022-04-02 02:52:28 -07:00
|
|
|
user.publicKey,
|
2022-03-31 23:59:39 -07:00
|
|
|
);
|
|
|
|
account = accounts[0];
|
|
|
|
console.log(`Created mango account ${account.publicKey.toBase58()}`);
|
|
|
|
}
|
2022-03-31 06:58:12 -07:00
|
|
|
|
|
|
|
// deposit
|
2022-03-31 23:59:39 -07:00
|
|
|
console.log(`Depositing...1000`);
|
|
|
|
await deposit(
|
2022-04-02 02:52:28 -07:00
|
|
|
userClient,
|
2022-03-31 23:59:39 -07:00
|
|
|
group.publicKey,
|
|
|
|
account.publicKey,
|
2022-04-02 02:52:28 -07:00
|
|
|
btcBank.publicKey,
|
|
|
|
btcBank.vault,
|
2022-03-31 23:59:39 -07:00
|
|
|
// BTC token account
|
2022-04-02 02:52:28 -07:00
|
|
|
new PublicKey('DS2vYFVtQbbJDowCG4NEM9KGQ8TJpxKo5efBQj96eCPS'),
|
2022-03-31 23:59:39 -07:00
|
|
|
btcDevnetOracle,
|
2022-04-02 02:52:28 -07:00
|
|
|
user.publicKey,
|
2022-03-31 23:59:39 -07:00
|
|
|
1000,
|
2022-03-31 06:58:12 -07:00
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
|
|
|
|
// withdraw
|
|
|
|
console.log(`Witdrawing...500`);
|
|
|
|
await withdraw(
|
2022-04-02 02:52:28 -07:00
|
|
|
userClient,
|
2022-03-31 23:59:39 -07:00
|
|
|
group.publicKey,
|
|
|
|
account.publicKey,
|
2022-04-02 02:52:28 -07:00
|
|
|
btcBank.publicKey,
|
|
|
|
btcBank.vault,
|
2022-03-31 23:59:39 -07:00
|
|
|
// BTC token account
|
2022-04-02 02:52:28 -07:00
|
|
|
new PublicKey('DS2vYFVtQbbJDowCG4NEM9KGQ8TJpxKo5efBQj96eCPS'),
|
2022-03-31 23:59:39 -07:00
|
|
|
btcDevnetOracle,
|
2022-04-02 02:52:28 -07:00
|
|
|
user.publicKey,
|
2022-03-31 23:59:39 -07:00
|
|
|
500,
|
|
|
|
false,
|
2022-03-31 06:58:12 -07:00
|
|
|
);
|
|
|
|
|
2022-03-31 23:59:39 -07:00
|
|
|
// log
|
2022-04-02 02:52:28 -07:00
|
|
|
const freshBank = await getBank(userClient, btcBank.publicKey);
|
2022-03-31 23:59:39 -07:00
|
|
|
console.log(freshBank.toString());
|
2022-03-31 06:58:12 -07:00
|
|
|
|
2022-04-02 02:52:28 -07:00
|
|
|
const freshAccount = await getMangoAccount(userClient, account.publicKey);
|
2022-03-31 23:59:39 -07:00
|
|
|
console.log(
|
|
|
|
`Mango account ${freshAccount.getNativeDeposit(
|
|
|
|
freshBank,
|
|
|
|
)} Deposits for bank ${freshBank.tokenIndex}`,
|
2022-03-31 06:58:12 -07:00
|
|
|
);
|
|
|
|
|
2022-04-01 00:11:08 -07:00
|
|
|
// close mango account
|
2022-04-02 02:52:28 -07:00
|
|
|
await closeMangoAccount(userClient, account.publicKey, user.publicKey);
|
2022-03-31 23:59:39 -07:00
|
|
|
accounts = await getMangoAccountsForGroupAndOwner(
|
2022-04-02 02:52:28 -07:00
|
|
|
userClient,
|
2022-03-31 23:59:39 -07:00
|
|
|
group.publicKey,
|
2022-04-02 02:52:28 -07:00
|
|
|
user.publicKey,
|
2022-03-31 06:58:12 -07:00
|
|
|
);
|
2022-03-31 23:59:39 -07:00
|
|
|
if (accounts.length === 0) {
|
|
|
|
console.log(`Closed account ${account.publicKey}`);
|
2022-03-31 04:30:05 -07:00
|
|
|
}
|
2022-03-31 23:59:39 -07:00
|
|
|
|
|
|
|
process.exit(0);
|
2022-03-23 07:39:11 -07:00
|
|
|
}
|
|
|
|
|
|
|
|
main();
|