Skip to content

Implement /amm explorer for liquidity pools #312

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
talktosam2003 wants to merge 1 commit into
base: master
Choose a base branch
from
Open

Implement /amm explorer for liquidity pools #312

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
133 changes: 132 additions & 1 deletion packages/bot/src/adapters/discord.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,6 +6,7 @@ export class DiscordAdapter {
private client: Client;
private userChannels: Map<string, string> = new Map(); // userId -> channelId
private token: string;
private backendUrl: string = process.env.BACKEND_URL || 'http://localhost:2333';

constructor(token: string) {
this.token = token;
Expand Down Expand Up @@ -38,13 +39,19 @@ export class DiscordAdapter {
);
}

if (message.content === "!help") {
await message.reply(
"**Commands:**\n`!start`, `!help`, `!sponsor`, `!trustline`, `!amm`"
);
}

if (message.content === "!sponsor") {
const userId = message.author.id;
await message.reply("⏳ Requesting account sponsorship...");

try {
const response = await fetch(
`${BACKEND_URL}/api/account/${userId}/sponsor`,
`${this.backendUrl}/api/account/${userId}/sponsor`,
{
method: "POST",
headers: { "Content-Type": "application/json" },
Expand Down Expand Up @@ -99,6 +106,130 @@ export class DiscordAdapter {
await message.reply(`❌ Error: ${error instanceof Error ? error.message : String(error)}`);
}
}

if (message.content.startsWith('!amm')) {
const args = message.content.split(' ').slice(1);
if (args.length < 1) {
let helpMsg = '🔍 **AMM Explorer**\n\n';
helpMsg += 'Use this command to search for liquidity pools and view their metrics.\n\n';
helpMsg += '**Usage:**\n';
helpMsg += '• `!amm search <assetA> <assetB>`\n';
helpMsg += '• `!amm stats <poolId>`\n\n';
helpMsg += '**Examples:**\n';
helpMsg += '• `!amm search XLM USDC`\n';
helpMsg += '• `!amm search XLM yXLM:GBSH...`\n';
helpMsg += '• `!amm stats 65f...a1b`';

return message.reply(helpMsg);
}

const subCommand = args[0].toLowerCase();
const horizonUrl = process.env.STELLAR_HORIZON_URL || 'https://horizon.stellar.org';

try {
if (subCommand === 'search') {
if (args.length < 3) {
return message.reply('❌ Usage: `!amm search <assetA> <assetB>`');
}
const assetA = args[1];
const assetB = args[2];
await message.reply(`🔍 Searching for **${assetA}/${assetB}** pools...`);

const formatAsset = (a: string) => {
const upper = a.toUpperCase();
if (upper === 'XLM' || upper === 'NATIVE') return 'native';
if (a.includes(':')) {
const [code, issuer] = a.split(':');
return `${code.toUpperCase()}:${issuer}`;
}
return a;
};

const response = await fetch(`${horizonUrl}/liquidity_pools?assets=${formatAsset(assetA)},${formatAsset(assetB)}`);
const data = await response.json() as any;
const pools = data._embedded?.records || [];

if (pools.length === 0) {
return message.reply(`❌ No liquidity pools found for **${assetA}/${assetB}**.`);
}

let responseMsg = `✅ **Found ${pools.length} pool(s) for ${assetA}/${assetB}:**\n\n`;
pools.slice(0, 5).forEach((p: any) => {
const resA = p.reserves[0];
const resB = p.reserves[1];
const shortId = `${p.id.slice(0, 8)}...${p.id.slice(-8)}`;

// APR calculation
const reserveA = parseFloat(resA.amount);
const reserveB = parseFloat(resB.amount);
const volumeEntry = p.volume ? p.volume[Object.keys(p.volume)[0]] : null;
const volume24h = volumeEntry ? parseFloat(volumeEntry.base_volume) + parseFloat(volumeEntry.counter_volume) : 0;
const totalLiquidity = reserveA + reserveB;
const apr = totalLiquidity > 0 ? ((volume24h * 0.003 * 365) / totalLiquidity * 100).toFixed(2) : '0.00';

responseMsg += `🔹 **Pool** \`${shortId}\`\n`;
responseMsg += `💰 **Reserves:**\n`;
responseMsg += ` • ${reserveA.toLocaleString()} ${resA.asset.split(':')[0]}\n`;
responseMsg += ` • ${reserveB.toLocaleString()} ${resB.asset.split(':')[0]}\n`;
responseMsg += `📊 **Fee:** ${(p.fee_bp / 100).toFixed(2)}% | **APR:** ${apr}%\n`;
responseMsg += `🔗 \`!amm stats ${p.id}\`\n\n`;
});

if (pools.length > 5) {
responseMsg += `*...and ${pools.length - 5} more.*`;
}

await message.reply(responseMsg);
} else if (subCommand === 'stats') {
if (args.length < 2) {
return message.reply('❌ Usage: `!amm stats <poolId>`');
}
const poolId = args[1];
await message.reply(`📊 Fetching stats for pool \`${poolId.slice(0, 8)}...\``);

const response = await fetch(`${horizonUrl}/liquidity_pools/${poolId}`);
if (response.status === 404) {
return message.reply('❌ Liquidity pool not found.');
}
const p = await response.json() as any;

const resA = p.reserves[0];
const resB = p.reserves[1];

// APR calculation
const reserveA = parseFloat(resA.amount);
const reserveB = parseFloat(resB.amount);
const volumeEntry = p.volume ? p.volume[Object.keys(p.volume)[0]] : null;
const volume24h = volumeEntry ? parseFloat(volumeEntry.base_volume) + parseFloat(volumeEntry.counter_volume) : 0;
const totalLiquidity = reserveA + reserveB;
const apr = totalLiquidity > 0 ? ((volume24h * 0.003 * 365) / totalLiquidity * 100).toFixed(2) : '0.00';

let responseMsg = `📊 **Liquidity Pool Metrics**\n\n`;
responseMsg += `🆔 **ID:** \`${p.id}\`\n\n`;

responseMsg += `**Assets:**\n`;
responseMsg += `• ${resA.asset}\n`;
responseMsg += `• ${resB.asset}\n\n`;

responseMsg += `**Reserves:**\n`;
responseMsg += `• **${reserveA.toLocaleString()}** ${resA.asset.split(':')[0]}\n`;
responseMsg += `• **${reserveB.toLocaleString()}** ${resB.asset.split(':')[0]}\n\n`;

responseMsg += `**Statistics:**\n`;
responseMsg += `• **Shares:** ${parseFloat(p.total_shares).toLocaleString()}\n`;
responseMsg += `• **Trustlines:** ${p.total_trustlines}\n`;
responseMsg += `• **Fee:** ${(p.fee_bp / 100).toFixed(2)}%\n`;
responseMsg += `• **APR:** ${apr}%\n`;

await message.reply(responseMsg);
} else {
await message.reply('❓ Unknown subcommand. Use `search` or `stats`.');
}
} catch (error) {
console.error('AMM Explorer error:', error);
await message.reply(`❌ **Error:** ${error instanceof Error ? error.message : 'Unknown error'}`);
}
}
});

await this.client.login(token);
Expand Down
126 changes: 125 additions & 1 deletion packages/bot/src/adapters/telegram.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,131 @@ export class TelegramAdapter {
this.bot = new Telegraf(this.token);

this.bot.start((ctx) => ctx.reply('Welcome to Chen Pilot! I am your AI-powered Stellar DeFi assistant.'));
this.bot.help((ctx) => ctx.reply('Commands: /start, /balance, /swap, /trustline'));
this.bot.help((ctx) => ctx.reply('Commands: /start, /balance, /swap, /trustline, /amm'));

this.bot.command('amm', async (ctx) => {
const args = ctx.message.text.split(' ').slice(1);
if (args.length < 1) {
let helpMsg = '🔍 <b>AMM Explorer</b>\n\n';
helpMsg += 'Use this command to search for liquidity pools and view their metrics.\n\n';
helpMsg += '<b>Usage:</b>\n';
helpMsg += '• <code>/amm search &lt;assetA&gt; &lt;assetB&gt;</code>\n';
helpMsg += '• <code>/amm stats &lt;poolId&gt;</code>\n\n';
helpMsg += '<b>Examples:</b>\n';
helpMsg += '• <code>/amm search XLM USDC</code>\n';
helpMsg += '• <code>/amm search XLM yXLM:GBSH...</code>\n';
helpMsg += '• <code>/amm stats 65f...a1b</code>';

return ctx.reply(helpMsg, { parse_mode: 'HTML' });
}

const subCommand = args[0].toLowerCase();
const horizonUrl = process.env.STELLAR_HORIZON_URL || 'https://horizon.stellar.org';

try {
if (subCommand === 'search') {
if (args.length < 3) {
return ctx.reply('❌ Usage: <code>/amm search &lt;assetA&gt; &lt;assetB&gt;</code>', { parse_mode: 'HTML' });
}
const assetA = args[1];
const assetB = args[2];
const loadingMsg = await ctx.reply(`🔍 Searching for <b>${assetA}/${assetB}</b> pools...`, { parse_mode: 'HTML' });

const formatAsset = (a: string) => {
const upper = a.toUpperCase();
if (upper === 'XLM' || upper === 'NATIVE') return 'native';
if (a.includes(':')) {
const [code, issuer] = a.split(':');
return `${code.toUpperCase()}:${issuer}`;
}
return a;
};

const response = await fetch(`${horizonUrl}/liquidity_pools?assets=${formatAsset(assetA)},${formatAsset(assetB)}`);
const data = await response.json() as any;
const pools = data._embedded?.records || [];

if (pools.length === 0) {
return ctx.reply(`❌ No liquidity pools found for <b>${assetA}/${assetB}</b>.`, { parse_mode: 'HTML' });
}

let message = `✅ <b>Found ${pools.length} pool(s) for ${assetA}/${assetB}:</b>\n\n`;
pools.slice(0, 5).forEach((p: any) => {
const resA = p.reserves[0];
const resB = p.reserves[1];
const shortId = `${p.id.slice(0, 8)}...${p.id.slice(-8)}`;

// APR calculation
const reserveA = parseFloat(resA.amount);
const reserveB = parseFloat(resB.amount);
const volumeEntry = p.volume ? p.volume[Object.keys(p.volume)[0]] : null;
const volume24h = volumeEntry ? parseFloat(volumeEntry.base_volume) + parseFloat(volumeEntry.counter_volume) : 0;
const totalLiquidity = reserveA + reserveB;
const apr = totalLiquidity > 0 ? ((volume24h * 0.003 * 365) / totalLiquidity * 100).toFixed(2) : '0.00';

message += `🔹 <b>Pool</b> <code>${shortId}</code>\n`;
message += `💰 <b>Reserves:</b>\n`;
message += ` • ${reserveA.toLocaleString()} ${resA.asset.split(':')[0]}\n`;
message += ` • ${reserveB.toLocaleString()} ${resB.asset.split(':')[0]}\n`;
message += `📊 <b>Fee:</b> ${(p.fee_bp / 100).toFixed(2)}% | <b>APR:</b> ${apr}%\n`;
message += `🔗 <code>/amm stats ${p.id}</code>\n\n`;
});

if (pools.length > 5) {
message += `<i>...and ${pools.length - 5} more.</i>`;
}

await ctx.reply(message, { parse_mode: 'HTML' });
} else if (subCommand === 'stats') {
if (args.length < 2) {
return ctx.reply('❌ Usage: <code>/amm stats &lt;poolId&gt;</code>', { parse_mode: 'HTML' });
}
const poolId = args[1];
await ctx.reply(`📊 Fetching stats for pool <code>${poolId.slice(0, 8)}...</code>`, { parse_mode: 'HTML' });

const response = await fetch(`${horizonUrl}/liquidity_pools/${poolId}`);
if (response.status === 404) {
return ctx.reply('❌ Liquidity pool not found.');
}
const p = await response.json() as any;

const resA = p.reserves[0];
const resB = p.reserves[1];

// APR calculation
const reserveA = parseFloat(resA.amount);
const reserveB = parseFloat(resB.amount);
const volumeEntry = p.volume ? p.volume[Object.keys(p.volume)[0]] : null;
const volume24h = volumeEntry ? parseFloat(volumeEntry.base_volume) + parseFloat(volumeEntry.counter_volume) : 0;
const totalLiquidity = reserveA + reserveB;
const apr = totalLiquidity > 0 ? ((volume24h * 0.003 * 365) / totalLiquidity * 100).toFixed(2) : '0.00';

let message = `📊 <b>Liquidity Pool Metrics</b>\n\n`;
message += `🆔 <b>ID:</b> <code>${p.id}</code>\n\n`;

message += `<b>Assets:</b>\n`;
message += `• ${resA.asset}\n`;
message += `• ${resB.asset}\n\n`;

message += `<b>Reserves:</b>\n`;
message += `• <b>${reserveA.toLocaleString()}</b> ${resA.asset.split(':')[0]}\n`;
message += `• <b>${reserveB.toLocaleString()}</b> ${resB.asset.split(':')[0]}\n\n`;

message += `<b>Statistics:</b>\n`;
message += `• <b>Shares:</b> ${parseFloat(p.total_shares).toLocaleString()}\n`;
message += `• <b>Trustlines:</b> ${p.total_trustlines}\n`;
message += `• <b>Fee:</b> ${(p.fee_bp / 100).toFixed(2)}%\n`;
message += `• <b>APR:</b> ${apr}%\n`;

await ctx.reply(message, { parse_mode: 'HTML' });
} else {
await ctx.reply('❓ Unknown subcommand. Use <b>search</b> or <b>stats</b>.', { parse_mode: 'HTML' });
}
} catch (error) {
console.error('AMM Explorer error:', error);
await ctx.reply(`❌ <b>Error:</b> ${error instanceof Error ? error.message : 'Unknown error'}`, { parse_mode: 'HTML' });
}
});

this.bot.command('trustline', async (ctx) => {
const args = ctx.message.text.split(' ').slice(1);
Expand Down
3 changes: 3 additions & 0 deletions src/Agents/registry/ToolAutoDiscovery.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -50,6 +50,9 @@ export class ToolAutoDiscovery {
const { strategyRegistryTool } = await import("../tools/strategyRegistry");
toolRegistry.register(strategyRegistryTool);

const { ammExplorerTool } = await import("../tools/ammExplorer");
toolRegistry.register(ammExplorerTool);

// todo
// await this.discoverToolsFromDirectory();

Expand Down
Loading