126 lines
3.4 KiB
TypeScript
126 lines
3.4 KiB
TypeScript
import { AnchorProvider, Wallet } from '@coral-xyz/anchor';
|
|
import { Cluster, Connection, Keypair, PublicKey } from '@solana/web3.js';
|
|
import fs from 'fs';
|
|
import { Group } from '../src/accounts/group';
|
|
import { MangoAccount } from '../src/accounts/mangoAccount';
|
|
import { MangoClient } from '../src/client';
|
|
import { MANGO_V4_ID } from '../src/constants';
|
|
|
|
const CLUSTER_URL =
|
|
process.env.CLUSTER_URL_OVERRIDE || process.env.MB_CLUSTER_URL;
|
|
const USER_KEYPAIR =
|
|
process.env.USER_KEYPAIR_OVERRIDE || process.env.MB_PAYER_KEYPAIR;
|
|
const MANGO_ACCOUNT_PK = process.env.MANGO_ACCOUNT_PK;
|
|
const CLUSTER: Cluster =
|
|
(process.env.CLUSTER_OVERRIDE as Cluster) || 'mainnet-beta';
|
|
|
|
async function settlePnlForMangoAccount(
|
|
client: MangoClient,
|
|
group: Group,
|
|
mangoAccount: MangoAccount,
|
|
settler: MangoAccount,
|
|
): Promise<void> {
|
|
for (const pp of mangoAccount.perpActive()) {
|
|
// settle only a specific market
|
|
// if (pp.marketIndex != 2) continue;
|
|
const pm = group.getPerpMarketByMarketIndex(pp.marketIndex);
|
|
const upnlUi = pp.getUnsettledPnlUi(pm);
|
|
if (upnlUi > 0) {
|
|
const c = await pm.getSettlePnlCandidates(
|
|
client,
|
|
group,
|
|
undefined,
|
|
'negative',
|
|
);
|
|
const sig = await client.perpSettlePnl(
|
|
group,
|
|
mangoAccount,
|
|
c[0].account,
|
|
settler!,
|
|
pm.perpMarketIndex,
|
|
);
|
|
console.log(
|
|
`Settled ${mangoAccount.publicKey}'s pnl for ${pm.name} at ${sig}`,
|
|
);
|
|
} else {
|
|
const c = await pm.getSettlePnlCandidates(
|
|
client,
|
|
group,
|
|
undefined,
|
|
'positive',
|
|
);
|
|
const sig = await client.perpSettlePnl(
|
|
group,
|
|
c[0].account,
|
|
mangoAccount,
|
|
settler!,
|
|
pm.perpMarketIndex,
|
|
);
|
|
console.log(
|
|
`Settled ${mangoAccount.publicKey}'s pnl for ${pm.name} at ${sig}`,
|
|
);
|
|
}
|
|
}
|
|
}
|
|
|
|
async function main(): Promise<void> {
|
|
const options = AnchorProvider.defaultOptions();
|
|
const connection = new Connection(CLUSTER_URL!, options);
|
|
const kp = Keypair.fromSecretKey(
|
|
Buffer.from(JSON.parse(fs.readFileSync(USER_KEYPAIR!, 'utf-8'))),
|
|
);
|
|
const wallet = new Wallet(kp);
|
|
const provider = new AnchorProvider(connection, wallet, options);
|
|
const client = MangoClient.connect(provider, CLUSTER, MANGO_V4_ID[CLUSTER], {
|
|
idsSource: 'api',
|
|
});
|
|
|
|
const group = await client.getGroup(
|
|
new PublicKey('78b8f4cGCwmZ9ysPFMWLaLTkkaYnUjwMJYStWe5RTSSX'),
|
|
);
|
|
|
|
let mangoAccounts = await client.getAllMangoAccounts(group, true);
|
|
|
|
const settler = await client.getMangoAccountForOwner(
|
|
group,
|
|
wallet.publicKey,
|
|
0,
|
|
true,
|
|
);
|
|
|
|
if (MANGO_ACCOUNT_PK) {
|
|
const mangoAccount = await client.getMangoAccount(
|
|
new PublicKey(MANGO_ACCOUNT_PK),
|
|
);
|
|
await settlePnlForMangoAccount(client, group, mangoAccount, settler!);
|
|
|
|
process.exit();
|
|
}
|
|
|
|
mangoAccounts = mangoAccounts.filter(
|
|
(a) =>
|
|
a.perpActive().length > 0 &&
|
|
a
|
|
.perpActive()
|
|
.some(
|
|
(pp) =>
|
|
pp.getBasePositionUi(
|
|
group.getPerpMarketByMarketIndex(pp.marketIndex),
|
|
) == 0 &&
|
|
Math.abs(
|
|
pp.getUnsettledFundingUi(
|
|
group.getPerpMarketByMarketIndex(pp.marketIndex),
|
|
),
|
|
) > 0,
|
|
) === true,
|
|
);
|
|
|
|
for (const mangoAccount of mangoAccounts) {
|
|
await settlePnlForMangoAccount(client, group, mangoAccount, settler!);
|
|
}
|
|
|
|
process.exit();
|
|
}
|
|
|
|
main();
|