108 lines
4.0 KiB
TypeScript
108 lines
4.0 KiB
TypeScript
|
|
import Koa from "koa";
|
|
import Router from "koa-router";
|
|
import mongo from "koa-mongo";
|
|
import bodyParser from "koa-bodyparser";
|
|
import cors from "@koa/cors";
|
|
import * as cron from "node-cron";
|
|
import {MongoClient, ObjectId} from "mongodb";
|
|
|
|
import { MangoClient, IDS } from '@blockworks-foundation/mango-client';
|
|
import { Connection, PublicKey } from '@solana/web3.js';
|
|
|
|
import { UserError } from './errors';
|
|
import { sendLogsToDiscord } from './logger';
|
|
import { initiateTelegramBot, generateTelegramCode, validateMarginAccount, validatePhoneNumber, validateEmail, reduceMangoGroups, sendAlert } from './utils';
|
|
import config from './environment';
|
|
|
|
const MESSAGE = 'Your collateral ratio is at or below @ratio@% \n';
|
|
|
|
const app = new Koa();
|
|
const router = new Router;
|
|
|
|
const cluster = 'mainnet-beta';
|
|
const client = new MangoClient();
|
|
const clusterIds = IDS[cluster];
|
|
const connection = new Connection(IDS.cluster_urls[cluster], 'singleGossip');
|
|
const dexProgramId = new PublicKey(clusterIds.dex_program_id);
|
|
|
|
app.use(cors());
|
|
app.use(bodyParser());
|
|
app.use(mongo({ uri: config.dbConnectionString }, { useUnifiedTopology: true }));
|
|
|
|
initiateTelegramBot();
|
|
|
|
router.post('/alerts', async(ctx, next) => {
|
|
try {
|
|
const alert = ctx.request.body;
|
|
await validateMarginAccount(client, connection, dexProgramId, alert);
|
|
if (alert.alertProvider == 'sms') {
|
|
const phoneNumber = `+${alert.phoneNumber.code}${alert.phoneNumber.phone}`;
|
|
await validatePhoneNumber(phoneNumber);
|
|
ctx.body = { status: 'success' };
|
|
} else if (alert.alertProvider == 'mail') {
|
|
validateEmail(alert.email);
|
|
ctx.body = { status: 'success' };
|
|
} else if (alert.alertProvider == 'tg') {
|
|
const code = generateTelegramCode();
|
|
alert.tgCode = code;
|
|
ctx.body = { code };
|
|
} else {
|
|
throw new UserError('Invalid alert provider');
|
|
}
|
|
alert.open = true;
|
|
alert.timestamp = new Date();
|
|
ctx.db.collection('alerts').insertOne(alert);
|
|
} catch (e) {
|
|
let errorMessage = 'Something went wrong';
|
|
if (e.name == 'UserError') {
|
|
errorMessage = e.message;
|
|
} else {
|
|
sendLogsToDiscord(null, e);
|
|
}
|
|
ctx.throw(400, errorMessage);
|
|
}
|
|
await next();
|
|
});
|
|
|
|
app.use(router.allowedMethods());
|
|
app.use(router.routes());
|
|
|
|
app.listen(config.port, () => {
|
|
const readyMessage = `> Server ready on http://localhost:${config.port}`;
|
|
console.log(readyMessage)
|
|
sendLogsToDiscord(readyMessage, null);
|
|
});
|
|
|
|
cron.schedule("1 * * * *", async () => {
|
|
try {
|
|
const mongoConnection = await MongoClient.connect(config.dbConnectionString, { useUnifiedTopology: true });
|
|
const db = mongoConnection.db(config.db);
|
|
const alerts = await db.collection('alerts').find({open: true}).toArray();
|
|
const uniqueMangoGroupPks: string[] = [...new Set(alerts.map(alert => alert.mangoGroupPk))];
|
|
const mangoGroups:any = await reduceMangoGroups(client, connection, uniqueMangoGroupPks);
|
|
alerts.forEach(async (alert) => {
|
|
const marginAccountPk = new PublicKey(alert.marginAccountPk);
|
|
const marginAccount = await client.getMarginAccount(connection, marginAccountPk, dexProgramId);
|
|
const collateralRatio = marginAccount.getCollateralRatio(mangoGroups[alert.mangoGroupPk]['mangoGroup'], mangoGroups[alert.mangoGroupPk]['prices']);
|
|
if (collateralRatio <= alert.collateralRatioThresh) {
|
|
let message = MESSAGE.replace('@ratio@', alert.collateralRatioThresh);
|
|
message += marginAccount.toPrettyString(
|
|
mangoGroups[alert.mangoGroupPk]['mangoGroup'],
|
|
mangoGroups[alert.mangoGroupPk]['prices']
|
|
);
|
|
message += '\nVisit https://trade.mango.markets/'
|
|
const alertSent = sendAlert(alert, message);
|
|
if (alertSent) {
|
|
db.collection('alerts').updateOne({ _id: new ObjectId(alert._id) }, { '$set': { open: false } });
|
|
}
|
|
}
|
|
});
|
|
const expiryTime = new Date( Date.now() - (1000 * 60 * 1) ); // 15 Minutes
|
|
console.log(expiryTime);
|
|
db.collection('alerts').deleteMany({ tgChatId: { $exists: false }, timestamp: { '$lt': expiryTime } });
|
|
} catch (e) {
|
|
sendLogsToDiscord(null, e);
|
|
}
|
|
});
|