51 lines
1.6 KiB
TypeScript
51 lines
1.6 KiB
TypeScript
import { PublicKey } from '@solana/web3.js';
|
|
import { MangoClient } from '../client';
|
|
import { Bank } from './bank';
|
|
import { PerpMarket } from './perp';
|
|
import { Serum3Market } from './serum3';
|
|
|
|
export class Group {
|
|
static from(publicKey: PublicKey, obj: { admin: PublicKey }): Group {
|
|
return new Group(publicKey, obj.admin, new Map(), new Map(), new Map());
|
|
}
|
|
|
|
constructor(
|
|
public publicKey: PublicKey,
|
|
public admin: PublicKey,
|
|
public banksMap: Map<string, Bank>,
|
|
public serum3MarketsMap: Map<string, Serum3Market>,
|
|
public perpMarketsMap: Map<string, PerpMarket>,
|
|
) {}
|
|
|
|
public findBank(tokenIndex: number): Bank | undefined {
|
|
return Array.from(this.banksMap.values()).find(
|
|
(bank) => bank.tokenIndex === tokenIndex,
|
|
);
|
|
}
|
|
|
|
public async reload(client: MangoClient) {
|
|
await this.reloadBanks(client);
|
|
await this.reloadSerum3Markets(client);
|
|
await this.reloadPerpMarkets(client);
|
|
}
|
|
|
|
public async reloadBanks(client: MangoClient) {
|
|
const banks = await client.getBanksForGroup(this);
|
|
this.banksMap = new Map(banks.map((bank) => [bank.name, bank]));
|
|
}
|
|
|
|
public async reloadSerum3Markets(client: MangoClient) {
|
|
const serum3Markets = await client.serum3GetMarket(this);
|
|
this.serum3MarketsMap = new Map(
|
|
serum3Markets.map((serum3Market) => [serum3Market.name, serum3Market]),
|
|
);
|
|
}
|
|
|
|
public async reloadPerpMarkets(client: MangoClient) {
|
|
const perpMarkets = await client.perpGetMarket(this);
|
|
this.perpMarketsMap = new Map(
|
|
perpMarkets.map((perpMarket) => [perpMarket.name, perpMarket]),
|
|
);
|
|
}
|
|
}
|