255 lines
6.4 KiB
TypeScript
255 lines
6.4 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 { 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,
|
|
altAddress: PublicKey,
|
|
addresses: PublicKey[],
|
|
): Promise<void> {
|
|
await group.reloadAll(client);
|
|
const alt = await client.program.provider.connection.getAddressLookupTable(
|
|
altAddress,
|
|
);
|
|
|
|
addresses = addresses.filter(
|
|
(newAddress) =>
|
|
alt.value?.state.addresses &&
|
|
alt.value?.state.addresses.findIndex((addressInAlt) =>
|
|
addressInAlt.equals(newAddress),
|
|
) === -1,
|
|
);
|
|
if (addresses.length === 0) {
|
|
return;
|
|
}
|
|
const extendIx = AddressLookupTableProgram.extendLookupTable({
|
|
lookupTable: alt.value!.key,
|
|
payer: payer.publicKey,
|
|
authority: payer.publicKey,
|
|
addresses,
|
|
});
|
|
|
|
console.log(
|
|
`Extending ${altAddress} with ${nick} ${
|
|
addresses.length
|
|
} addresses - ${addresses.join(', ')}`,
|
|
);
|
|
|
|
if (DRY_RUN) {
|
|
return;
|
|
}
|
|
|
|
const extendTx = await buildVersionedTx(
|
|
client.program.provider as AnchorProvider,
|
|
[extendIx],
|
|
);
|
|
const sig = await client.program.provider.connection.sendTransaction(
|
|
extendTx,
|
|
);
|
|
console.log(`https://explorer.solana.com/tx/${sig}`);
|
|
}
|
|
|
|
async function run(): 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 addressess
|
|
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',
|
|
);
|
|
// bank mints
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'token mints',
|
|
altAddress1,
|
|
Array.from(group.banksMapByMint.values())
|
|
.flat()
|
|
.map((bank) => [bank.mint])
|
|
.flat(),
|
|
);
|
|
// bank mint infos
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'mint infos',
|
|
altAddress1,
|
|
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,
|
|
Array.from(group.serum3MarketsMapByMarketIndex.values())
|
|
.flat()
|
|
.map((serum3Market) => serum3Market.publicKey),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets',
|
|
altAddress1,
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.publicKey),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets bids',
|
|
altAddress1,
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.bidsAddress),
|
|
);
|
|
await extendTable(
|
|
client,
|
|
group,
|
|
payer,
|
|
'serum3 external markets asks',
|
|
altAddress1,
|
|
Array.from(group.serum3ExternalMarketsMap.values())
|
|
.flat()
|
|
.map((serum3ExternalMarket) => serum3ExternalMarket.asksAddress),
|
|
);
|
|
} catch (error) {
|
|
console.log(error);
|
|
}
|
|
}
|
|
|
|
run();
|