73 lines
2.9 KiB
TypeScript
73 lines
2.9 KiB
TypeScript
|
|
import { NextRequest, NextResponse } from 'next/server';
|
|
import { db } from '@/lib/db';
|
|
import { CryptoEngine } from '@/lib/crypto-engine';
|
|
import { ethers } from 'ethers';
|
|
import { PublicKey } from '@solana/web3.js';
|
|
|
|
export async function GET() {
|
|
try {
|
|
// 1. Fetch platform addresses from settings
|
|
const settingsRes = await db.query("SELECT key, value FROM system_settings WHERE key IN ('sol_platform_address', 'evm_platform_address', 'tron_platform_address', 'btc_platform_address')");
|
|
const settings: Record<string, string> = {};
|
|
settingsRes.rows.forEach(r => settings[r.key] = r.value);
|
|
|
|
const addresses = {
|
|
EVM: settings.evm_platform_address || "0x70997970C51812dc3A010C7d01b50e0d17dc79C8",
|
|
SOLANA: settings.sol_platform_address || "Ajr4nKieZJVu9q2d1eVF9pQPRCLoZ6v4tapB3iQn2SyQ",
|
|
TRON: settings.tron_platform_address || "TY795B6FmDNV4Xm5U6G1rP9yvX7S9rK6G1P", // Valid TRON address format
|
|
BITCOIN: settings.btc_platform_address || "17V95B6FmDNV4Xm5U6G1rP9yvX7S9rK6G1P" // Valid BTC address format
|
|
};
|
|
|
|
// 2. Fetch balances for each network from config
|
|
const cryptoConfig = require('@/lib/crypto-config.json');
|
|
const balances: any = {};
|
|
|
|
for (const netConfig of cryptoConfig.networks) {
|
|
const net = netConfig.id;
|
|
console.log(`[Treasury] Fetching balances for ${net}...`);
|
|
const engine = new CryptoEngine(net);
|
|
const addr = (addresses as any)[['POLYGON', 'BSC', 'ETH'].includes(net) ? 'EVM' : net];
|
|
|
|
if (!addr) {
|
|
console.warn(`[Treasury] No address found for network ${net}`);
|
|
continue;
|
|
}
|
|
|
|
try {
|
|
const tokenBalances: Record<string, string> = {};
|
|
let nativeBalance = "0.00";
|
|
let nativeSymbol = net;
|
|
|
|
for (const token of netConfig.tokens) {
|
|
const balance = await engine.getBalance(addr, token.symbol);
|
|
if (token.address === 'NATIVE') {
|
|
nativeBalance = balance;
|
|
nativeSymbol = token.symbol;
|
|
} else {
|
|
tokenBalances[token.symbol] = balance;
|
|
}
|
|
}
|
|
|
|
balances[net] = {
|
|
address: addr,
|
|
native: nativeBalance,
|
|
nativeSymbol: nativeSymbol,
|
|
tokens: tokenBalances
|
|
};
|
|
} catch (err) {
|
|
console.error(`[Treasury Balance Error] ${net}:`, err);
|
|
balances[net] = { address: addr, native: "Error", tokens: {} };
|
|
}
|
|
}
|
|
|
|
return NextResponse.json({
|
|
success: true,
|
|
balances
|
|
});
|
|
|
|
} catch (error: any) {
|
|
return NextResponse.json({ error: error.message }, { status: 500 });
|
|
}
|
|
}
|