txnbot - rewrite and lint (#53)
resolves #46 Co-authored-by: johba <johba@harb.eth> Reviewed-on: https://codeberg.org/johba/harb/pulls/53
This commit is contained in:
johba
parent
dc61771dfc
commit
514be62cbb
19 changed files with 736 additions and 429 deletions
388
services/txnBot/src/service.ts
Normal file
388
services/txnBot/src/service.ts
Normal file
|
|
@ -0,0 +1,388 @@
|
|||
import path from 'path';
|
||||
import { fileURLToPath } from 'url';
|
||||
import dotenv from 'dotenv';
|
||||
import { ethers, Contract, Wallet, JsonRpcProvider, TransactionResponse } from 'ethers';
|
||||
import express, { Request, Response, NextFunction } from 'express';
|
||||
import { decodePositionId } from 'kraiken-lib/ids';
|
||||
import { isPositionDelinquent } from 'kraiken-lib/staking';
|
||||
import { logger } from './logger.js';
|
||||
import { hasRecenterAccess, readRecenterAccess, type RecenterAccessReader } from './recenterAccess.js';
|
||||
import { Position, EnvConfig, RecenterAccessStatus, RecenterEligibility, RecenterResult, GraphQLResponse } from './types.js';
|
||||
|
||||
const __filename = fileURLToPath(import.meta.url);
|
||||
const __dirname = path.dirname(__filename);
|
||||
|
||||
const dotenvPath = process.env.TXN_BOT_ENV_FILE ? path.resolve(process.env.TXN_BOT_ENV_FILE) : path.resolve(__dirname, '..', '.env');
|
||||
|
||||
dotenv.config({ path: dotenvPath });
|
||||
|
||||
const ACTIVE_POSITIONS_QUERY = `
|
||||
query ActivePositions {
|
||||
positionss(where: { status: "Active" }, limit: 1000) {
|
||||
items {
|
||||
id
|
||||
share
|
||||
lastTaxTime
|
||||
taxRate
|
||||
status
|
||||
}
|
||||
}
|
||||
}
|
||||
`;
|
||||
|
||||
const PROVIDER_URL = process.env.PROVIDER_URL;
|
||||
const PRIVATE_KEY = process.env.PRIVATE_KEY;
|
||||
const LM_CONTRACT_ADDRESS = process.env.LM_CONTRACT_ADDRESS;
|
||||
const STAKE_CONTRACT_ADDRESS = process.env.STAKE_CONTRACT_ADDRESS;
|
||||
const GRAPHQL_ENDPOINT = process.env.GRAPHQL_ENDPOINT;
|
||||
const ENVIRONMENT = process.env.ENVIRONMENT || 'UNSPECIFIED';
|
||||
|
||||
const requiredEnv: Partial<EnvConfig> = {
|
||||
PROVIDER_URL,
|
||||
PRIVATE_KEY,
|
||||
LM_CONTRACT_ADDRESS,
|
||||
STAKE_CONTRACT_ADDRESS,
|
||||
GRAPHQL_ENDPOINT,
|
||||
};
|
||||
|
||||
for (const [key, value] of Object.entries(requiredEnv)) {
|
||||
if (!value) {
|
||||
throw new Error(`Missing required environment variable: ${key}`);
|
||||
}
|
||||
}
|
||||
|
||||
const LM_ABI = [
|
||||
{ type: 'function', name: 'recenter', inputs: [], outputs: [], stateMutability: 'nonpayable' },
|
||||
{ type: 'function', name: 'feeDestination', inputs: [], outputs: [{ type: 'address' }], stateMutability: 'view' },
|
||||
{ type: 'function', name: 'recenterAccess', inputs: [], outputs: [{ type: 'address' }], stateMutability: 'view' },
|
||||
];
|
||||
const STAKE_ABI = [
|
||||
{
|
||||
inputs: [{ internalType: 'uint256', name: 'positionId', type: 'uint256' }],
|
||||
name: 'payTax',
|
||||
outputs: [],
|
||||
stateMutability: 'nonpayable',
|
||||
type: 'function',
|
||||
},
|
||||
];
|
||||
|
||||
const provider: JsonRpcProvider = new ethers.JsonRpcProvider(PROVIDER_URL);
|
||||
|
||||
const wallet: Wallet = new ethers.Wallet(PRIVATE_KEY as string, provider);
|
||||
const liquidityManager: Contract = new ethers.Contract(LM_CONTRACT_ADDRESS as string, LM_ABI, wallet);
|
||||
const stakeContract: Contract = new ethers.Contract(STAKE_CONTRACT_ADDRESS as string, STAKE_ABI, wallet);
|
||||
const recenterAccessReader: RecenterAccessReader = {
|
||||
recenterAccess: async (): Promise<string> => {
|
||||
const method = liquidityManager.getFunction('recenterAccess');
|
||||
return (await method()) as string;
|
||||
},
|
||||
};
|
||||
const walletAddress: string = ethers.getAddress(wallet.address);
|
||||
const ZERO_ADDRESS: string = ethers.ZeroAddress;
|
||||
|
||||
const startTime = new Date();
|
||||
let lastRecenterTime: Date | null = null;
|
||||
let lastLiquidationTime: Date | null = null;
|
||||
let lastRecenterTx: string | null = null;
|
||||
let lastRecenterAccessStatus: RecenterAccessStatus | null = null;
|
||||
let lastRecenterEligibility: RecenterEligibility = {
|
||||
checkedAtMs: 0,
|
||||
canRecenter: false,
|
||||
reason: null,
|
||||
error: null,
|
||||
};
|
||||
|
||||
async function fetchActivePositions(): Promise<Position[]> {
|
||||
const response = await fetch(GRAPHQL_ENDPOINT as string, {
|
||||
method: 'POST',
|
||||
headers: { 'Content-Type': 'application/json' },
|
||||
body: JSON.stringify({ query: ACTIVE_POSITIONS_QUERY }),
|
||||
});
|
||||
|
||||
if (!response.ok) {
|
||||
throw new Error(`GraphQL request failed with status ${response.status}`);
|
||||
}
|
||||
|
||||
const payload = (await response.json()) as GraphQLResponse;
|
||||
if (payload.errors && payload.errors.length > 0) {
|
||||
const messages = payload.errors.map(error => error.message).join('; ');
|
||||
throw new Error(`GraphQL responded with errors: ${messages}`);
|
||||
}
|
||||
|
||||
return payload.data?.positionss?.items ?? [];
|
||||
}
|
||||
|
||||
async function checkFunds(): Promise<string> {
|
||||
const balance = await provider.getBalance(wallet.address);
|
||||
return ethers.formatEther(balance);
|
||||
}
|
||||
|
||||
function extractRevertReason(error: unknown): string | null {
|
||||
const err = error as {
|
||||
shortMessage?: string;
|
||||
reason?: string;
|
||||
info?: { error?: { message?: string } };
|
||||
error?: { message?: string };
|
||||
message?: string;
|
||||
};
|
||||
|
||||
const candidates = [err?.shortMessage, err?.reason, err?.info?.error?.message, err?.error?.message, err?.message];
|
||||
|
||||
for (const candidate of candidates) {
|
||||
if (typeof candidate !== 'string' || candidate.length === 0) {
|
||||
continue;
|
||||
}
|
||||
const match = candidate.match(/execution reverted(?: due to)?(?::|: )?(.*)/i);
|
||||
if (match && match[1]) {
|
||||
return match[1].trim();
|
||||
}
|
||||
if (!candidate.toLowerCase().startsWith('execution reverted')) {
|
||||
return candidate.trim();
|
||||
}
|
||||
}
|
||||
return null;
|
||||
}
|
||||
|
||||
async function getRecenterAccessStatus(forceRefresh = false): Promise<RecenterAccessStatus> {
|
||||
const now = Date.now();
|
||||
if (
|
||||
!forceRefresh &&
|
||||
lastRecenterAccessStatus &&
|
||||
lastRecenterAccessStatus.checkedAtMs &&
|
||||
now - lastRecenterAccessStatus.checkedAtMs < 30000
|
||||
) {
|
||||
return lastRecenterAccessStatus;
|
||||
}
|
||||
|
||||
let recenterAddress: string | null = null;
|
||||
let hasAccess: boolean | null = null;
|
||||
let slotHex: string | null = null;
|
||||
let errorMessage: string | null = null;
|
||||
|
||||
try {
|
||||
const address = await readRecenterAccess(recenterAccessReader, ZERO_ADDRESS);
|
||||
recenterAddress = address;
|
||||
hasAccess = hasRecenterAccess(address, walletAddress, ZERO_ADDRESS);
|
||||
slotHex = 'recenterAccess()';
|
||||
} catch (error) {
|
||||
const err = error as { shortMessage?: string; message?: string };
|
||||
errorMessage = err?.shortMessage || err?.message || 'unknown error';
|
||||
recenterAddress = null;
|
||||
}
|
||||
|
||||
lastRecenterAccessStatus = {
|
||||
hasAccess,
|
||||
recenterAccessAddress: recenterAddress,
|
||||
slot: slotHex,
|
||||
checkedAtMs: now,
|
||||
error: errorMessage,
|
||||
};
|
||||
|
||||
return lastRecenterAccessStatus;
|
||||
}
|
||||
|
||||
async function evaluateRecenterOpportunity(): Promise<RecenterEligibility> {
|
||||
const now = Date.now();
|
||||
const accessStatus = await getRecenterAccessStatus(true);
|
||||
|
||||
if (accessStatus.error && accessStatus.hasAccess === null) {
|
||||
lastRecenterEligibility = {
|
||||
checkedAtMs: now,
|
||||
canRecenter: false,
|
||||
reason: 'Failed to determine recenter access.',
|
||||
error: accessStatus.error,
|
||||
};
|
||||
return lastRecenterEligibility;
|
||||
}
|
||||
|
||||
if (accessStatus.hasAccess === false) {
|
||||
lastRecenterEligibility = {
|
||||
checkedAtMs: now,
|
||||
canRecenter: false,
|
||||
reason: 'txnBot is not the authorized recenter caller.',
|
||||
error: null,
|
||||
};
|
||||
return lastRecenterEligibility;
|
||||
}
|
||||
|
||||
try {
|
||||
await liquidityManager.recenter.estimateGas();
|
||||
lastRecenterEligibility = {
|
||||
checkedAtMs: now,
|
||||
canRecenter: true,
|
||||
reason: null,
|
||||
error: null,
|
||||
};
|
||||
} catch (error) {
|
||||
const err = error as { shortMessage?: string; message?: string };
|
||||
lastRecenterEligibility = {
|
||||
checkedAtMs: now,
|
||||
canRecenter: false,
|
||||
reason: extractRevertReason(error) || 'recenter not currently executable.',
|
||||
error: err?.shortMessage || err?.message || null,
|
||||
};
|
||||
}
|
||||
|
||||
return lastRecenterEligibility;
|
||||
}
|
||||
|
||||
async function attemptRecenter(): Promise<RecenterResult> {
|
||||
const eligibility = await evaluateRecenterOpportunity();
|
||||
if (!eligibility.canRecenter) {
|
||||
return {
|
||||
executed: false,
|
||||
message: eligibility.reason || 'Liquidity manager denied recenter.',
|
||||
eligibility,
|
||||
};
|
||||
}
|
||||
|
||||
const tx: TransactionResponse = (await liquidityManager.recenter()) as TransactionResponse;
|
||||
lastRecenterTx = tx.hash;
|
||||
await tx.wait();
|
||||
lastRecenterTime = new Date();
|
||||
return {
|
||||
executed: true,
|
||||
txHash: tx.hash,
|
||||
message: 'recenter transaction submitted',
|
||||
eligibility: lastRecenterEligibility,
|
||||
};
|
||||
}
|
||||
|
||||
async function checkPosition(position: Position): Promise<number> {
|
||||
const taxRate = Number(position.taxRate);
|
||||
const lastTaxTime = Number(position.lastTaxTime);
|
||||
if (isPositionDelinquent(lastTaxTime, taxRate)) {
|
||||
const positionId = decodePositionId(position.id);
|
||||
logger.info(`Calling payTax on ${positionId}`);
|
||||
const tx: TransactionResponse = (await stakeContract.payTax(positionId)) as TransactionResponse;
|
||||
await tx.wait();
|
||||
lastLiquidationTime = new Date();
|
||||
return 1;
|
||||
} else {
|
||||
return 0;
|
||||
}
|
||||
}
|
||||
|
||||
function formatDuration(ms: number): string {
|
||||
let seconds = Math.floor(ms / 1000);
|
||||
let minutes = Math.floor(seconds / 60);
|
||||
let hours = Math.floor(minutes / 60);
|
||||
const days = Math.floor(hours / 24);
|
||||
|
||||
seconds = seconds % 60;
|
||||
minutes = minutes % 60;
|
||||
hours = hours % 24;
|
||||
|
||||
return `${days} days, ${hours} hours, ${minutes} minutes`;
|
||||
}
|
||||
|
||||
async function liquidityLoop(): Promise<void> {
|
||||
try {
|
||||
const result = await attemptRecenter();
|
||||
if (result.executed) {
|
||||
logger.info(`recenter called successfully. tx=${result.txHash}`);
|
||||
} else {
|
||||
const reason = result.message ? `Reason: ${result.message}` : 'Reason: unavailable';
|
||||
logger.info(`Recenter skipped. ${reason} - ${new Date().toISOString()}`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error('Error in liquidity loop', { error });
|
||||
}
|
||||
}
|
||||
|
||||
async function liquidationLoop(): Promise<void> {
|
||||
let counter = 0;
|
||||
try {
|
||||
const positions = await fetchActivePositions();
|
||||
for (const position of positions) {
|
||||
counter = counter + (await checkPosition(position));
|
||||
}
|
||||
if (counter === 0) {
|
||||
logger.info(`No tax can be claimed at the moment. - ${new Date().toISOString()}`);
|
||||
}
|
||||
} catch (error) {
|
||||
logger.error('Error in liquidation loop', { error });
|
||||
}
|
||||
}
|
||||
|
||||
async function main(): Promise<void> {
|
||||
logger.info(`txnBot service started for environment ${ENVIRONMENT}`);
|
||||
|
||||
await liquidityLoop();
|
||||
await liquidationLoop();
|
||||
setInterval(liquidityLoop, 3 * 60000); // 3 minute
|
||||
setInterval(liquidationLoop, 20 * 60000); // 20 minutes
|
||||
}
|
||||
|
||||
main().catch(async error => {
|
||||
logger.error('Fatal error', { error });
|
||||
});
|
||||
|
||||
const app = express();
|
||||
const PORT = process.env.PORT || 3000;
|
||||
|
||||
app.use((req: Request, res: Response, next: NextFunction) => {
|
||||
res.setHeader('Access-Control-Allow-Origin', '*');
|
||||
res.setHeader('Access-Control-Allow-Methods', 'GET,POST,OPTIONS');
|
||||
res.setHeader('Access-Control-Allow-Headers', 'Content-Type');
|
||||
if (req.method === 'OPTIONS') {
|
||||
return res.sendStatus(204);
|
||||
}
|
||||
return next();
|
||||
});
|
||||
|
||||
app.get('/status', async (_req: Request, res: Response) => {
|
||||
try {
|
||||
const balance = await checkFunds();
|
||||
const uptime = formatDuration(new Date().getTime() - startTime.getTime());
|
||||
const recenterAccessStatus = await getRecenterAccessStatus();
|
||||
const recenterEligibility = lastRecenterEligibility;
|
||||
const status = {
|
||||
balance: `${balance} ETH`,
|
||||
uptime: uptime,
|
||||
lastRecenterTime: lastRecenterTime ? lastRecenterTime.toISOString() : 'Never',
|
||||
lastLiquidationTime: lastLiquidationTime ? lastLiquidationTime.toISOString() : 'Never',
|
||||
lastRecenterTx,
|
||||
recenterAccess: {
|
||||
hasAccess: recenterAccessStatus?.hasAccess ?? null,
|
||||
grantedTo: recenterAccessStatus?.recenterAccessAddress ?? null,
|
||||
slot: recenterAccessStatus?.slot ?? null,
|
||||
checkedAt: recenterAccessStatus?.checkedAtMs ? new Date(recenterAccessStatus.checkedAtMs).toISOString() : null,
|
||||
error: recenterAccessStatus?.error ?? null,
|
||||
},
|
||||
recenterEligibility: {
|
||||
canRecenter: recenterEligibility?.canRecenter ?? null,
|
||||
reason: recenterEligibility?.reason ?? null,
|
||||
checkedAt: recenterEligibility?.checkedAtMs ? new Date(recenterEligibility.checkedAtMs).toISOString() : null,
|
||||
error: recenterEligibility?.error ?? null,
|
||||
},
|
||||
};
|
||||
|
||||
if (parseFloat(balance) < 0.1) {
|
||||
res.status(500).send(`Low Ethereum Balance: ${balance} ETH`);
|
||||
} else {
|
||||
res.status(200).json(status);
|
||||
}
|
||||
} catch (error) {
|
||||
const err = error as { message?: string };
|
||||
res.status(500).send(`Error checking funds: ${err.message}`);
|
||||
}
|
||||
});
|
||||
|
||||
app.post('/recenter', async (_req: Request, res: Response) => {
|
||||
try {
|
||||
const result = await attemptRecenter();
|
||||
if (!result.executed) {
|
||||
return res.status(202).json(result);
|
||||
}
|
||||
return res.status(200).json(result);
|
||||
} catch (error) {
|
||||
const err = error as { message?: string };
|
||||
logger.error('Manual recenter failed', { error });
|
||||
return res.status(500).json({ error: err.message || 'recenter failed' });
|
||||
}
|
||||
});
|
||||
|
||||
app.listen(PORT, () => {
|
||||
logger.info(`HTTP server running on port ${PORT}`);
|
||||
});
|
||||
Loading…
Add table
Add a link
Reference in a new issue