348 lines
9.2 KiB
TypeScript
348 lines
9.2 KiB
TypeScript
import { AnchorProvider, Wallet } from '@coral-xyz/anchor';
|
|
import {
|
|
ASSOCIATED_TOKEN_PROGRAM_ID,
|
|
NATIVE_MINT,
|
|
TOKEN_PROGRAM_ID,
|
|
} from '@solana/spl-token';
|
|
import {
|
|
AddressLookupTableProgram,
|
|
Cluster,
|
|
ComputeBudgetProgram,
|
|
Connection,
|
|
Keypair,
|
|
PublicKey,
|
|
SYSVAR_INSTRUCTIONS_PUBKEY,
|
|
SystemProgram,
|
|
} from '@solana/web3.js';
|
|
import fs from 'fs';
|
|
import chunk from 'lodash/chunk';
|
|
import { Group } from '../src/accounts/group';
|
|
import { MangoClient } from '../src/client';
|
|
import {
|
|
MANGO_V4_ID,
|
|
MANGO_V4_MAIN_GROUP,
|
|
OPENBOOK_PROGRAM_ID,
|
|
} from '../src/constants';
|
|
import { buildVersionedTx } from '../src/utils';
|
|
|
|
const { MB_CLUSTER_URL, MB_PAYER3_KEYPAIR, DRY_RUN } = process.env;
|
|
const CLUSTER: Cluster = (process.env.CLUSTER as Cluster) || 'mainnet-beta';
|
|
|
|
// eslint-disable-next-line no-inner-declarations
|
|
async function extendTable(
|
|
client: MangoClient,
|
|
group: Group,
|
|
payer: Keypair,
|
|
nick: string,
|
|
altAddresses: PublicKey[],
|
|
addressesToAdd: PublicKey[],
|
|
): Promise<void> {
|
|
await group.reloadAll(client);
|
|
|
|
let addressesAlreadyIndexed: PublicKey[] = [];
|
|
for (const altAddr of altAddresses) {
|
|
const alt = await client.program.provider.connection.getAddressLookupTable(
|
|
altAddr,
|
|
);
|
|
if (alt.value?.state.addresses) {
|
|
addressesAlreadyIndexed = addressesAlreadyIndexed.concat(
|
|
alt.value?.state.addresses,
|
|
);
|
|
}
|
|
}
|
|
|
|
addressesToAdd = addressesToAdd.filter(
|
|
(newAddress) =>
|
|
addressesAlreadyIndexed.findIndex((addressInAlt) =>
|
|
addressInAlt.equals(newAddress),
|
|
) === -1,
|
|
);
|
|
if (addressesToAdd.length === 0) {
|
|
return;
|
|
}
|
|
|
|
let altIndex = 0;
|
|
for (const chunk_ of chunk(addressesToAdd, 20)) {
|
|
let alt;
|
|
while (altIndex < altAddresses.length) {
|
|
alt = await client.program.provider.connection.getAddressLookupTable(
|
|
altAddresses[altIndex],
|
|
);
|
|
if (alt.value?.state.addresses.length < 234) {
|
|
break;
|
|
} else {
|
|
if (altIndex == altAddresses.length - 1) {
|
|
console.log(
|
|
`...need to create a new alt, all existing ones are full`,
|
|
);
|
|
process.exit(-1);
|
|
}
|
|
console.log(
|
|
`...using a new alt ${altAddresses[altIndex + 1]}, ${
|
|
altAddresses[altIndex]
|
|
} is almost full`,
|
|
);
|
|
}
|
|
altIndex++;
|
|
}
|
|
|
|
console.log(
|
|
`Extending ${altAddresses[altIndex]} with ${nick} ${
|
|
chunk_.length
|
|
} addresses - ${chunk_.join(', ')}`,
|
|
);
|
|
|
|
const extendIx = AddressLookupTableProgram.extendLookupTable({
|
|
lookupTable: alt.value!.key,
|
|
payer: payer.publicKey,
|
|
authority: payer.publicKey,
|
|
addresses: chunk_,
|
|
});
|
|
const extendTx = await buildVersionedTx(
|
|
client.program.provider as AnchorProvider,
|
|
[extendIx],
|
|
);
|
|
|
|
if (DRY_RUN) {
|
|
continue;
|
|
}
|
|
const sig = await client.program.provider.connection.sendTransaction(
|
|
extendTx,
|
|
);
|
|
console.log(`https://explorer.solana.com/tx/${sig}`);
|
|
}
|
|
}
|
|
|
|
async function createANewAlt() {
|
|
const options = AnchorProvider.defaultOptions();
|
|
const connection = new Connection(MB_CLUSTER_URL!, options);
|
|
const payer = Keypair.fromSecretKey(
|
|
Buffer.from(JSON.parse(fs.readFileSync(MB_PAYER3_KEYPAIR!, 'utf-8'))),
|
|
);
|
|
const payerWallet = new Wallet(payer);
|
|
const userProvider = new AnchorProvider(connection, payerWallet, options);
|
|
const client = await MangoClient.connect(
|
|
userProvider,
|
|
CLUSTER,
|
|
MANGO_V4_ID[CLUSTER],
|
|
{
|
|
idsSource: 'get-program-accounts',
|
|
},
|
|
);
|
|
|
|
const createIx = AddressLookupTableProgram.createLookupTable({
|
|
authority: payer.publicKey,
|
|
payer: payer.publicKey,
|
|
recentSlot: await connection.getSlot('finalized'),
|
|
});
|
|
const createTx = await buildVersionedTx(
|
|
client.program.provider as AnchorProvider,
|
|
[createIx[0]],
|
|
);
|
|
const sig = await connection.sendTransaction(createTx);
|
|
console.log(
|
|
`...created ALT ${createIx[1]} https://explorer.solana.com/tx/${sig}`,
|
|
);
|
|
}
|
|
|
|
async function populateExistingAlts(): Promise<void> {
|
|
try {
|
|
const options = AnchorProvider.defaultOptions();
|
|
const connection = new Connection(MB_CLUSTER_URL!, options);
|
|
const payer = Keypair.fromSecretKey(
|
|
Buffer.from(JSON.parse(fs.readFileSync(MB_PAYER3_KEYPAIR!, 'utf-8'))),
|
|
);
|
|
const payerWallet = new Wallet(payer);
|
|
const userProvider = new AnchorProvider(connection, payerWallet, options);
|
|
const client = await MangoClient.connect(
|
|
userProvider,
|
|
CLUSTER,
|
|
MANGO_V4_ID[CLUSTER],
|
|
{
|
|
idsSource: 'get-program-accounts',
|
|
},
|
|
);
|
|
const group = await client.getGroup(MANGO_V4_MAIN_GROUP);
|
|
|
|
//
|
|
// Table 0 - liquidation relevant accounts
|
|
//
|
|
// const altAddress0 = group.addressLookupTablesList[0].key;
|
|
const altAddress0 = new PublicKey(
|
|
'AgCBUZ6UMWqPLftTxeAqpQxtrfiCyL2HgRfmmM6QTfCj',
|
|
);
|
|
// group and insurance vault
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'group',
|
|
[altAddress0],
|
|
[group.publicKey, group.insuranceVault],
|
|
);
|
|
// Banks + vaults + oracles
|
|
// Split into 3 ixs since we end up with RangeError: encoding overruns Uint8Array otherwise
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'token banks',
|
|
[altAddress0],
|
|
Array.from(group.banksMapByMint.values())
|
|
.flat()
|
|
.map((bank) => bank.publicKey),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'token bank oracles',
|
|
[altAddress0],
|
|
Array.from(group.banksMapByMint.values())
|
|
.flat()
|
|
.map((bank) => bank.oracle),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'token bank vaults',
|
|
[altAddress0],
|
|
Array.from(group.banksMapByMint.values())
|
|
.flat()
|
|
.map((bank) => bank.vault),
|
|
);
|
|
// Perps + oracles
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'perp markets and perp oracles',
|
|
[altAddress0],
|
|
Array.from(group.perpMarketsMapByMarketIndex.values())
|
|
.flat()
|
|
.map((perpMarket) => [perpMarket.publicKey, perpMarket.oracle])
|
|
.flat(),
|
|
);
|
|
// Well known addresses
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'well known addresses',
|
|
[altAddress0],
|
|
[
|
|
// Solana specific
|
|
SystemProgram.programId,
|
|
TOKEN_PROGRAM_ID,
|
|
ASSOCIATED_TOKEN_PROGRAM_ID,
|
|
NATIVE_MINT,
|
|
SYSVAR_INSTRUCTIONS_PUBKEY,
|
|
ComputeBudgetProgram.programId,
|
|
// Misc.
|
|
OPENBOOK_PROGRAM_ID['mainnet-beta'],
|
|
],
|
|
);
|
|
|
|
//
|
|
// Table 1 - everything else
|
|
//
|
|
// const altAddress1 = group.addressLookupTablesList[1].key;
|
|
const altAddress1 = new PublicKey(
|
|
'FGZCgVhVGqzfWnmJFP9Hx4BvGvnFApEp1dM2whzXvg1Z',
|
|
);
|
|
const altAddress2 = new PublicKey(
|
|
'FsruqicZDGnCnm7dRthjL5eFrTmaNRkkomxhPJQP2kdu',
|
|
);
|
|
const altAddress3 = new PublicKey(
|
|
'2JAg3Rm6TmQ3gSYgUCCyZ9bCQKThD9jxHCN6U2ByTPMb',
|
|
);
|
|
// bank mints
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'token mints',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.banksMapByMint.values())
|
|
.flat()
|
|
.map((bank) => [bank.mint])
|
|
.flat(),
|
|
);
|
|
// bank mint infos
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'mint infos',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.mintInfosMapByMint.values())
|
|
.flat()
|
|
.map((mintInto) => [mintInto.publicKey])
|
|
.flat(),
|
|
);
|
|
// serum3
|
|
// Split into 4 ixs since we end up with RangeError: encoding overruns Uint8Array otherwise
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 markets',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.serum3MarketsMapByMarketIndex.values())
|
|
.flat()
|
|
.map((serum3Market) => serum3Market.publicKey),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.publicKey),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets bids',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.bidsAddress),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets asks',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.asksAddress),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'perp market event queues, bids, and asks',
|
|
[altAddress1, altAddress2, altAddress3],
|
|
Array.from(group.perpMarketsMapByMarketIndex.values())
|
|
.flat()
|
|
.map((perpMarket) => [
|
|
perpMarket.eventQueue,
|
|
perpMarket.bids,
|
|
perpMarket.asks,
|
|
])
|
|
.flat(),
|
|
);
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
}
|
|
|
|
// createANewAlt();
|
|
populateExistingAlts();
|