141 lines
4.8 KiB
TypeScript
141 lines
4.8 KiB
TypeScript
import { ponder } from "ponder:registry";
|
|
import { positions, stats, STATS_ID, TAX_RATES } from "ponder:schema";
|
|
import {
|
|
ensureStatsExists,
|
|
getStakeTotalSupply,
|
|
parseRingBuffer,
|
|
refreshOutstandingStake,
|
|
serializeRingBuffer,
|
|
updateHourlyData,
|
|
RING_BUFFER_SEGMENTS,
|
|
} from "./helpers/stats";
|
|
|
|
const ZERO = 0n;
|
|
|
|
async function getKraikenTotalSupply(context: any) {
|
|
return context.client.readContract({
|
|
abi: context.contracts.Kraiken.abi,
|
|
address: context.contracts.Kraiken.address,
|
|
functionName: "totalSupply",
|
|
});
|
|
}
|
|
|
|
function toShareRatio(share: bigint, stakeTotalSupply: bigint): number {
|
|
if (stakeTotalSupply === 0n) return 0;
|
|
return Number(share) / Number(stakeTotalSupply);
|
|
}
|
|
|
|
ponder.on("Stake:PositionCreated", async ({ event, context }: any) => {
|
|
await ensureStatsExists(context, event.block.timestamp);
|
|
|
|
const stakeTotalSupply = await getStakeTotalSupply(context);
|
|
const shareRatio = toShareRatio(event.args.share, stakeTotalSupply);
|
|
const totalSupplyInit = await getKraikenTotalSupply(context);
|
|
|
|
await context.db.insert(positions).values({
|
|
id: event.args.positionId.toString(),
|
|
owner: event.args.owner as `0x${string}`,
|
|
share: shareRatio,
|
|
taxRate: TAX_RATES[Number(event.args.taxRate)] || 0,
|
|
kraikenDeposit: event.args.kraikenDeposit,
|
|
stakeDeposit: event.args.kraikenDeposit,
|
|
taxPaid: ZERO,
|
|
snatched: 0,
|
|
creationTime: event.block.timestamp,
|
|
lastTaxTime: event.block.timestamp,
|
|
status: "Active",
|
|
createdAt: event.block.timestamp,
|
|
totalSupplyInit,
|
|
totalSupplyEnd: null,
|
|
payout: ZERO,
|
|
});
|
|
|
|
await refreshOutstandingStake(context);
|
|
});
|
|
|
|
ponder.on("Stake:PositionRemoved", async ({ event, context }: any) => {
|
|
await ensureStatsExists(context, event.block.timestamp);
|
|
|
|
const positionId = event.args.positionId.toString();
|
|
const position = await context.db.find(positions, { id: positionId });
|
|
if (!position) return;
|
|
|
|
const totalSupplyEnd = await getKraikenTotalSupply(context);
|
|
|
|
await context.db.update(positions, { id: positionId }).set({
|
|
status: "Closed",
|
|
closedAt: event.block.timestamp,
|
|
totalSupplyEnd,
|
|
payout: (position.payout ?? ZERO) + event.args.kraikenPayout,
|
|
kraikenDeposit: ZERO,
|
|
stakeDeposit: ZERO,
|
|
});
|
|
|
|
await refreshOutstandingStake(context);
|
|
await updateHourlyData(context, event.block.timestamp);
|
|
});
|
|
|
|
ponder.on("Stake:PositionShrunk", async ({ event, context }: any) => {
|
|
await ensureStatsExists(context, event.block.timestamp);
|
|
|
|
const positionId = event.args.positionId.toString();
|
|
const position = await context.db.find(positions, { id: positionId });
|
|
if (!position) return;
|
|
|
|
const stakeTotalSupply = await getStakeTotalSupply(context);
|
|
const shareRatio = toShareRatio(event.args.newShares, stakeTotalSupply);
|
|
|
|
await context.db.update(positions, { id: positionId }).set({
|
|
share: shareRatio,
|
|
kraikenDeposit: BigInt(position.kraikenDeposit ?? ZERO) - event.args.kraikenPayout,
|
|
stakeDeposit: BigInt(position.stakeDeposit ?? ZERO) - event.args.kraikenPayout,
|
|
snatched: position.snatched + 1,
|
|
payout: BigInt(position.payout ?? ZERO) + event.args.kraikenPayout,
|
|
});
|
|
|
|
await refreshOutstandingStake(context);
|
|
await updateHourlyData(context, event.block.timestamp);
|
|
});
|
|
|
|
ponder.on("Stake:PositionTaxPaid", async ({ event, context }: any) => {
|
|
await ensureStatsExists(context, event.block.timestamp);
|
|
await updateHourlyData(context, event.block.timestamp);
|
|
|
|
const positionId = event.args.positionId.toString();
|
|
const position = await context.db.find(positions, { id: positionId });
|
|
if (!position) return;
|
|
|
|
const stakeTotalSupply = await getStakeTotalSupply(context);
|
|
const shareRatio = toShareRatio(event.args.newShares, stakeTotalSupply);
|
|
|
|
await context.db.update(positions, { id: positionId }).set({
|
|
taxPaid: BigInt(position.taxPaid ?? ZERO) + event.args.taxPaid,
|
|
share: shareRatio,
|
|
taxRate: TAX_RATES[Number(event.args.taxRate)] || position.taxRate,
|
|
lastTaxTime: event.block.timestamp,
|
|
});
|
|
|
|
const statsData = await context.db.find(stats, { id: STATS_ID });
|
|
if (statsData) {
|
|
const ringBuffer = parseRingBuffer(statsData.ringBuffer as string[]);
|
|
const pointer = statsData.ringBufferPointer ?? 0;
|
|
const baseIndex = pointer * RING_BUFFER_SEGMENTS;
|
|
|
|
ringBuffer[baseIndex + 3] = ringBuffer[baseIndex + 3] + event.args.taxPaid;
|
|
|
|
await context.db.update(stats, { id: STATS_ID }).set({
|
|
ringBuffer: serializeRingBuffer(ringBuffer),
|
|
totalTaxPaid: statsData.totalTaxPaid + event.args.taxPaid,
|
|
});
|
|
}
|
|
|
|
await refreshOutstandingStake(context);
|
|
await updateHourlyData(context, event.block.timestamp);
|
|
});
|
|
|
|
ponder.on("Stake:PositionRateHiked", async ({ event, context }: any) => {
|
|
const positionId = event.args.positionId.toString();
|
|
await context.db.update(positions, { id: positionId }).set({
|
|
taxRate: TAX_RATES[Number(event.args.newTaxRate)] || 0,
|
|
});
|
|
});
|