123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116 |
- require('dotenv').config();
- const express = require('express');
- const cors = require('cors');
- const path = require('path');
- const TelegramBot = require('node-telegram-bot-api');
- const fs = require('fs');
- const moment = require('moment');
- const schedule = require('node-schedule');
- const {
- pool,
- testConnection
- } = require('./config/database');
- // const initDatabase = require('./config/initDb');
- const Group = require('./models/Group');
- const Transaction = require('./models/Transaction');
- // 日志格式化函数
- function formatLog(data) {
- const separator = '-'.repeat(30);
- let logMessage = `${separator}\n`;
- if (typeof data === 'object') {
- logMessage += Object.entries(data)
- .map(([key, value]) => `${key}: ${value}`)
- .join('\n');
- } else {
- logMessage += data;
- }
- logMessage += `\n${separator}\n`;
- return logMessage;
- }
- const app = express();
- // 初始化数据存储
- let data = {
- deposits: [], // 入款记录
- withdrawals: [], // 下发记录
- lastUpdate: null,
- allowedGroups: [] // 允许使用的群组ID
- };
- // 创建机器人实例
- const bot = new TelegramBot(process.env.BOT_TOKEN, {
- polling: true
- });
- // 中间件
- app.use(cors());
- app.use(express.json());
- app.use(express.urlencoded({ extended: true }));
- app.use('/admin/views', express.static(path.join(__dirname, 'views')));
- // 路由
- app.get('/', (req, res) => {
- res.sendFile(path.join(__dirname, 'views', 'login.html'));
- });
- app.use('/api/users', require('./routes/userRoutes'));
- app.use('/api/groups', require('./routes/groupRoutes'));
- app.use('/api/transactions', require('./routes/transactionRoutes'));
- app.use('/api/statistics', require('./routes/statisticsRoutes'));
- app.use('/api/settings', require('./routes/settingsRoutes'));
- // 检查群组权限
- function isGroupAllowed(chatId) {
- const chatIdStr = chatId.toString();
- return data.allowedGroups.includes(chatIdStr) ||
- data.allowedGroups.includes(chatIdStr.replace('-', ''));
- }
- // 前置处理
- // 检查用户是否有权限操作机器人
- async function checkUserPermission(chatId, userId) {
- try {
- const [group] = await pool.query(
- 'SELECT creator_id, COALESCE(operators, "[]") as operators FROM groups WHERE group_id = ?',
- [chatId.toString()]
- );
- if (!group || !group[0]) {
- console.log(`权限检查失败 - 群组不存在: ${chatId}`);
- return false;
- }
- const groupInfo = group[0];
- const userIdStr = userId.toString();
- const isCreator = groupInfo.creator_id === userIdStr;
-
- let operators = [];
- try {
- if (groupInfo.operators) {
- operators = JSON.parse(groupInfo.operators);
- if (!Array.isArray(operators)) {
- operators = [];
- }
- }
- } catch (e) {
- console.error('解析操作人列表失败:', e);
- operators = [];
- }
- // console.log(groupInfo.operators)
- // console.log(operators);
- // console.log(userIdStr);
- const isOperator = operators.some(op => op.operator_id == userIdStr);
-
- // 只在权限检查失败时输出详细日志
- if (!isCreator && !isOperator) {
- console.log(`权限检查失败 - 用户ID: ${userIdStr}, 群组ID: ${chatId}, 创建者ID: ${groupInfo.creator_id}, 操作人数量: ${operators.length}`);
- }
-
- return isCreator || isOperator;
- } catch (error) {
- console.error('检查用户权限失败:', error);
- return false;
- }
- }
- // 处理消息发送
- async function sendMessage(chatId, text, options = {}) {
- try {
- // 如果包含内联键盘,验证URL
- if (options.reply_markup && options.reply_markup.inline_keyboard) {
- const keyboard = generateInlineKeyboard(chatId);
- if (!keyboard) {
- // 如果键盘无效,发送不带键盘的消息
- return await bot.sendMessage(chatId, text, {
- parse_mode: 'HTML'
- });
- }
- options.reply_markup = keyboard;
- }
- return await bot.sendMessage(chatId, text, {
- ...options,
- parse_mode: 'HTML'
- });
- } catch (error) {
- console.error('发送消息失败:', error);
- if (error.message.includes('bot was kicked from the group chat')) {
- const index = data.allowedGroups.indexOf(chatId.toString());
- if (index > -1) {
- data.allowedGroups.splice(index, 1);
- saveData();
- console.log(`群组 ${chatId} 已被移除出允许列表`);
- }
- }
- return null;
- }
- }
- // - 1.快捷命令
- bot.on('message', async (msg) => {
- if (!isGroupAllowed(msg.chat.id)) return;
- const text = msg.text?.trim();
- if (!text) return;
- console.error(msg);
- // 如果是TRX地址,直接处理,跳过权限检查
- if (/^T[A-Za-z0-9]{33}$/.test(text)) {
- try {
- const groupId = msg.chat.id.toString();
-
- // 检查地址是否已存在
- const [existingAddress] = await pool.query(
- 'SELECT * FROM trx_addresses WHERE address = ? AND group_id = ?',
- [text, groupId]
- );
- if (existingAddress && existingAddress.length > 0) {
- // 更新使用次数和最后出现时间
- await pool.query(`
- UPDATE trx_addresses
- SET usage_count = usage_count + 1,
- last_seen_time = CURRENT_TIMESTAMP
- WHERE address = ? AND group_id = ?
- `, [text, groupId]);
- const newCount = existingAddress[0].usage_count + 1;
- await sendMessage(msg.chat.id, `此地址累计发送第${newCount}次`);
- console.log(`TRX地址使用次数更新 - 群组: ${msg.chat.title}, 地址: ${text}, 次数: ${newCount}, 时间: ${new Date().toLocaleString()}`);
- } else {
- // 插入新地址记录
- await pool.query(`
- INSERT INTO trx_addresses (address, group_id)
- VALUES (?, ?)
- `, [text, groupId]);
- await sendMessage(msg.chat.id, '此地址累计发送第1次');
- console.log(`新TRX地址记录 - 群组: ${msg.chat.title}, 地址: ${text}, 时间: ${new Date().toLocaleString()}`);
- }
- } catch (error) {
- console.error('处理TRX地址失败:', error);
- await sendMessage(msg.chat.id, '处理地址失败,请稍后重试');
- }
- return;
- }
- // 0. 检查用户权限
- const hasPermission = await checkUserPermission(msg.chat.id, msg.from.id);
- if (!hasPermission) {
- // 如果不是创建人或操作人,直接返回
- return;
- }
- // 1. 处理入款命令
- if (text.startsWith('+')) {
- let amount, exchangeRate, feeRate;
- const parts = text.substring(1).split('/');
- amount = parseFloat(parts[0]);
-
- // 如果指定了汇率,则使用指定的汇率
- if (parts.length > 1) {
- exchangeRate = parseFloat(parts[1]);
- }
-
- // 如果指定了费率,则使用指定的费率
- if (parts.length > 2) {
- feeRate = parseFloat(parts[2]);
- }
- if (!isNaN(amount)) {
- const transactionData = {
- groupId: msg.chat.id.toString(),
- groupName: msg.chat.title || '未命名群组',
- amount: amount,
- type: 'deposit',
- exchangeRate: exchangeRate,
- feeRate: feeRate,
- operatorId: msg.from.id
- };
- // console.log(transactionData);
- try {
- const result = await Transaction.deposit(transactionData);
- if (result.success) {
- const billMessage = await generateBillMessage(msg.chat.id);
- if (billMessage) {
- await sendMessage(msg.chat.id, billMessage, {
- reply_markup: generateInlineKeyboard(msg.chat.id)
- });
- console.log(`入款成功 - 群组: ${msg.chat.title}, 金额: ${amount}, 时间: ${new Date().toLocaleString()}`);
- } else {
- await sendMessage(msg.chat.id, '入款成功,但获取账单信息失败');
- console.log(`入款成功(无账单) - 群组: ${msg.chat.title}, 金额: ${amount}, 时间: ${new Date().toLocaleString()}`);
- }
- } else {
- await sendMessage(msg.chat.id, result.message || '入款失败');
- console.log(`入款失败 - 群组: ${msg.chat.title}, 金额: ${amount}, 原因: ${result.message}, 时间: ${new Date().toLocaleString()}`);
- }
- } catch (error) {
- console.error('快捷入款失败:', error);
- await sendMessage(msg.chat.id, '记录入款失败,请稍后重试');
- }
- }
- }
- // 1.1 处理入款修正命令
- else if (text.startsWith('-') && !text.includes('下发')) {
- let amount, exchangeRate, feeRate;
- const parts = text.substring(1).split('/');
- amount = parseFloat(parts[0]);
-
- // 如果指定了汇率,则使用指定的汇率
- if (parts.length > 1) {
- exchangeRate = parseFloat(parts[1]);
- }
-
- // 如果指定了费率,则使用指定的费率
- if (parts.length > 2) {
- feeRate = parseFloat(parts[2]);
- }
- if (!isNaN(amount)) {
- const transactionData = {
- groupId: msg.chat.id.toString(),
- groupName: msg.chat.title || '未命名群组',
- amount: -amount,
- type: 'deposit',
- exchangeRate: exchangeRate,
- feeRate: feeRate,
- operatorId: msg.from.id
- };
- try {
- const result = await Transaction.deposit(transactionData);
- if (result.success) {
- const billMessage = await generateBillMessage(msg.chat.id);
- if (billMessage) {
- await sendMessage(msg.chat.id, billMessage, {
- reply_markup: generateInlineKeyboard(msg.chat.id)
- });
- console.log(`入款修正成功 - 群组: ${msg.chat.title}, 金额: -${amount}, 时间: ${new Date().toLocaleString()}`);
- } else {
- await sendMessage(msg.chat.id, '入款修正成功,但获取账单信息失败');
- console.log(`入款修正成功(无账单) - 群组: ${msg.chat.title}, 金额: -${amount}, 时间: ${new Date().toLocaleString()}`);
- }
- } else {
- await sendMessage(msg.chat.id, result.message || '入款修正失败');
- console.log(`入款修正失败 - 群组: ${msg.chat.title}, 金额: -${amount}, 原因: ${result.message}, 时间: ${new Date().toLocaleString()}`);
- }
- } catch (error) {
- console.error('快捷入款修正失败:', error);
- await sendMessage(msg.chat.id, '记录入款修正失败,请稍后重试');
- }
- }
- }
- // 2. 处理下发命令
- else if (text.startsWith('下发')) {
- let amount, exchangeRate, feeRate;
- const parts = text.replace(/[^0-9./-]/g, '').split('/');
- amount = parseFloat(parts[0]);
-
- // 如果指定了汇率,则使用指定的汇率
- if (parts.length > 1) {
- exchangeRate = parseFloat(parts[1]);
- }
-
- // 如果指定了费率,则使用指定的费率
- if (parts.length > 2) {
- feeRate = parseFloat(parts[2]);
- }
- if (!isNaN(amount)) {
- // 获取当前账单信息
- const [records] = await pool.query(`
- SELECT t.*,
- COALESCE(t.fee_rate, g.in_fee_rate) as fee_rate,
- COALESCE(t.exchange_rate, g.in_exchange_rate) as exchange_rate,
- t.totalDeposit as total_deposit,
- t.totalWithdrawal as total_withdrawal,
- t.depositFee as deposit_fee,
- t.withdrawalFee as withdrawal_fee,
- t.totalUDeposit as total_u_deposit,
- t.totalUWithdrawal as total_u_withdrawal
- FROM transactions t
- LEFT JOIN groups g ON t.group_id = g.group_id
- WHERE t.group_id = ?
- AND DATE(t.time) = CURDATE()
- ORDER BY t.time DESC
- LIMIT 1
- `, [msg.chat.id.toString()]);
- if (!records || records.length === 0) {
- await sendMessage(msg.chat.id, '暂无交易记录,无法计算下发金额');
- return;
- }
- const latestRecord = records[0];
- const totalUDeposit = parseFloat(latestRecord.total_u_deposit) || 0;
- const totalUWithdrawal = parseFloat(latestRecord.total_u_withdrawal) || 0;
- // 直接使用输入的金额,不再限制
- const actualAmount = amount;
- if (actualAmount <= 0) {
- await sendMessage(msg.chat.id, '下发金额必须大于0');
- return;
- }
- const transactionData = {
- groupId: msg.chat.id.toString(),
- groupName: msg.chat.title || '未命名群组',
- amount: actualAmount,
- type: 'distribute',
- exchangeRate: exchangeRate,
- feeRate: feeRate,
- operatorId: msg.from.id
- };
- try {
- const result = await Transaction.distribute(transactionData);
- if (result.success) {
- const billMessage = await generateBillMessage(msg.chat.id);
- if (billMessage) {
- await sendMessage(msg.chat.id, billMessage, {
- reply_markup: generateInlineKeyboard(msg.chat.id)
- });
- console.log(`下发成功 - 群组: ${msg.chat.title}, 金额: ${actualAmount}, 时间: ${new Date().toLocaleString()}`);
- } else {
- await sendMessage(msg.chat.id, '下发成功,但获取账单信息失败');
- console.log(`下发成功(无账单) - 群组: ${msg.chat.title}, 金额: ${actualAmount}, 时间: ${new Date().toLocaleString()}`);
- }
- } else {
- await sendMessage(msg.chat.id, result.message || '下发失败');
- console.log(`下发失败 - 群组: ${msg.chat.title}, 金额: ${actualAmount}, 原因: ${result.message}, 时间: ${new Date().toLocaleString()}`);
- }
- } catch (error) {
- console.error('快捷下发失败:', error);
- await sendMessage(msg.chat.id, '记录下发失败,请稍后重试');
- }
- }
- }
- // 2.1 处理下发修正命令
- else if (text.startsWith('下发-')) {
- let amount, exchangeRate, feeRate;
- const parts = text.replace(/[^0-9./-]/g, '').split('/');
- amount = parseFloat(parts[0]);
-
- // 如果指定了汇率,则使用指定的汇率
- if (parts.length > 1) {
- exchangeRate = parseFloat(parts[1]);
- }
-
- // 如果指定了费率,则使用指定的费率
- if (parts.length > 2) {
- feeRate = parseFloat(parts[2]);
- }
- if (!isNaN(amount)) {
- // 获取当前账单信息
- const [records] = await pool.query(`
- SELECT t.*,
- COALESCE(t.fee_rate, g.in_fee_rate) as fee_rate,
- COALESCE(t.exchange_rate, g.in_exchange_rate) as exchange_rate,
- t.totalDeposit as total_deposit,
- t.totalWithdrawal as total_withdrawal,
- t.depositFee as deposit_fee,
- t.withdrawalFee as withdrawal_fee,
- t.totalUDeposit as total_u_deposit,
- t.totalUWithdrawal as total_u_withdrawal
- FROM transactions t
- LEFT JOIN groups g ON t.group_id = g.group_id
- WHERE t.group_id = ?
- AND DATE(t.time) = CURDATE()
- ORDER BY t.time DESC
- LIMIT 1
- `, [msg.chat.id.toString()]);
- if (!records || records.length === 0) {
- await sendMessage(msg.chat.id, '暂无交易记录,无法计算下发修正金额');
- return;
- }
- const latestRecord = records[0];
- const totalUDeposit = parseFloat(latestRecord.total_u_deposit) || 0;
- const totalUWithdrawal = parseFloat(latestRecord.total_u_withdrawal) || 0;
- const remainingAmount = totalUDeposit - totalUWithdrawal;
- // 计算实际修正金额
- const actualAmount = Math.min(amount, totalUWithdrawal);
- if (actualAmount <= 0) {
- await sendMessage(msg.chat.id, '没有可修正的下发金额');
- return;
- }
- const transactionData = {
- groupId: msg.chat.id.toString(),
- groupName: msg.chat.title || '未命名群组',
- amount: -actualAmount,
- type: 'distribute',
- exchangeRate: exchangeRate,
- feeRate: feeRate,
- operatorId: msg.from.id
- };
- try {
- const result = await Transaction.distribute(transactionData);
- if (result.success) {
- const billMessage = await generateBillMessage(msg.chat.id);
- if (billMessage) {
- await sendMessage(msg.chat.id, billMessage, {
- reply_markup: generateInlineKeyboard(msg.chat.id)
- });
- console.log(`下发修正成功 - 群组: ${msg.chat.title}, 金额: -${actualAmount}, 时间: ${new Date().toLocaleString()}`);
- } else {
- await sendMessage(msg.chat.id, '下发修正成功,但获取账单信息失败');
- console.log(`下发修正成功(无账单) - 群组: ${msg.chat.title}, 金额: -${actualAmount}, 时间: ${new Date().toLocaleString()}`);
- }
- } else {
- await sendMessage(msg.chat.id, result.message || '下发修正失败');
- console.log(`下发修正失败 - 群组: ${msg.chat.title}, 金额: -${actualAmount}, 原因: ${result.message}, 时间: ${new Date().toLocaleString()}`);
- }
- } catch (error) {
- console.error('快捷下发修正失败:', error);
- await sendMessage(msg.chat.id, '记录下发修正失败,请稍后重试');
- }
- }
- }
- // 3. 处理设置费率命令
- else if (text.startsWith('设置费率')) {
- const feeRate = parseFloat(text.replace('设置费率', '').trim());
-
- if (!isNaN(feeRate) && feeRate >= 0 && feeRate <= 100) {
- try {
- // 更新群组的入款和出款费率
- await pool.query(`
- UPDATE groups
- SET in_fee_rate = ?,
- out_fee_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [feeRate, feeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `费率${feeRate}%已设置成功`);
- console.log(`费率设置成功 - 群组: ${msg.chat.title}, 费率: ${feeRate}%, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置费率失败:', error);
- await sendMessage(msg.chat.id, '设置费率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '费率设置失败,请输入0-100之间的数字');
- }
- }
- // 3.1 处理设置入款费率命令
- else if (text.startsWith('设置入款费率')) {
- const feeRate = parseFloat(text.replace('设置入款费率', '').trim());
-
- if (!isNaN(feeRate) && feeRate >= 0 && feeRate <= 100) {
- try {
- // 只更新群组的入款费率
- await pool.query(`
- UPDATE groups
- SET in_fee_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [feeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `入款费率${feeRate}%已设置成功`);
- console.log(`入款费率设置成功 - 群组: ${msg.chat.title}, 费率: ${feeRate}%, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置入款费率失败:', error);
- await sendMessage(msg.chat.id, '设置入款费率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '入款费率设置失败,请输入0-100之间的数字');
- }
- }
- // 3.2 处理设置出款费率命令
- else if (text.startsWith('设置出款费率')) {
- const feeRate = parseFloat(text.replace('设置出款费率', '').trim());
-
- if (!isNaN(feeRate) && feeRate >= 0 && feeRate <= 100) {
- try {
- // 只更新群组的出款费率
- await pool.query(`
- UPDATE groups
- SET out_fee_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [feeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `出款费率${feeRate}%已设置成功`);
- console.log(`出款费率设置成功 - 群组: ${msg.chat.title}, 费率: ${feeRate}%, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置出款费率失败:', error);
- await sendMessage(msg.chat.id, '设置出款费率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '出款费率设置失败,请输入0-100之间的数字');
- }
- }
- // 4. 处理设置汇率命令
- else if (text.startsWith('设置汇率')) {
- const exchangeRate = parseFloat(text.replace('设置汇率', '').trim());
-
- if (!isNaN(exchangeRate) && exchangeRate > 0) {
- try {
- // 更新群组的入款和出款汇率
- await pool.query(`
- UPDATE groups
- SET in_exchange_rate = ?,
- out_exchange_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [exchangeRate, exchangeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `汇率${exchangeRate}已设置成功`);
- console.log(`汇率设置成功 - 群组: ${msg.chat.title}, 汇率: ${exchangeRate}, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置汇率失败:', error);
- await sendMessage(msg.chat.id, '设置汇率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '汇率设置失败,请输入大于0的数字');
- }
- }
- // 4.1 处理设置入款汇率命令
- else if (text.startsWith('设置入款汇率')) {
- const exchangeRate = parseFloat(text.replace('设置入款汇率', '').trim());
-
- if (!isNaN(exchangeRate) && exchangeRate > 0) {
- try {
- // 只更新群组的入款汇率
- await pool.query(`
- UPDATE groups
- SET in_exchange_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [exchangeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `入款汇率${exchangeRate}已设置成功`);
- console.log(`入款汇率设置成功 - 群组: ${msg.chat.title}, 汇率: ${exchangeRate}, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置入款汇率失败:', error);
- await sendMessage(msg.chat.id, '设置入款汇率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '入款汇率设置失败,请输入大于0的数字');
- }
- }
- // 4.2 处理设置出款汇率命令
- else if (text.startsWith('设置出款汇率')) {
- const exchangeRate = parseFloat(text.replace('设置出款汇率', '').trim());
-
- if (!isNaN(exchangeRate) && exchangeRate > 0) {
- try {
- // 只更新群组的出款汇率
- await pool.query(`
- UPDATE groups
- SET out_exchange_rate = ?,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [exchangeRate, msg.chat.id.toString()]);
- await sendMessage(msg.chat.id, `出款汇率${exchangeRate}已设置成功`);
- console.log(`出款汇率设置成功 - 群组: ${msg.chat.title}, 汇率: ${exchangeRate}, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置出款汇率失败:', error);
- await sendMessage(msg.chat.id, '设置出款汇率失败,请稍后重试');
- }
- } else {
- await sendMessage(msg.chat.id, '出款汇率设置失败,请输入大于0的数字');
- }
- }
- // 5. 处理设置操作人命令
- else if (text.startsWith('设置操作人')) {
- try {
- const groupId = msg.chat.id.toString();
- // const mentionedUser = msg.entities?.find(e => e.type === 'mention');
-
- if (!msg.reply_to_message.from.id) {
- // 如果没有@用户,回复原消息并提示设置用户名
- await bot.sendMessage(msg.chat.id, '请通过回复要添加操作人的消息设置', {
- reply_to_message_id: msg.message_id
- });
- return;
- }
- // 获取被@的用户名
- const username = msg.reply_to_message.from.first_name;//text.slice(mentionedUser.offset + 1, mentionedUser.offset + mentionedUser.length);
-
- // 获取群组信息
- const [group] = await pool.query(
- 'SELECT creator_id, COALESCE(operators, "[]") as operators FROM groups WHERE group_id = ?',
- [groupId]
- );
- if (!group || !group[0]) {
- await sendMessage(msg.chat.id, '群组信息不存在');
- return;
- }
- const groupInfo = group[0];
- const userId = msg.from.id.toString();
-
- // 检查操作权限(群组创建者或现有操作人)
- const isCreator = groupInfo.creator_id === userId;
- let operators = [];
-
- try {
- if (groupInfo.operators) {
- operators = JSON.parse(groupInfo.operators);
- if (!Array.isArray(operators)) {
- operators = [];
- }
- }
- } catch (e) {
- console.error('解析操作人列表失败:', e);
- operators = [];
- }
-
- const isOperator = operators.some(op => op.operator_id === userId);
- if (!isCreator && !isOperator) {
- await sendMessage(msg.chat.id, '您没有权限设置操作人');
- return;
- }
- // 获取被@用户的ID
- const [user] = await pool.query(
- 'SELECT id FROM users WHERE username = ?',
- [username]
- );
- let newOperatorId=msg.reply_to_message.from.id;
- let newOperatorUid=msg.reply_to_message.from.username;
- if (!user || !user[0]) {
- // 如果用户不存在,创建新用户
- try {
- const [result] = await pool.query(
- 'INSERT INTO users (id,uid,username, password, role) VALUES (?,?,?, ?, ?)',
- [newOperatorId,newOperatorUid,username, '', 'user']
- );
- // newOperatorId = result.insertId.toString();
- console.log(`创建新用户成功 - 用户名: ${username}, ID: ${newOperatorId}`);
- } catch (error) {
- console.error('创建新用户失败:', error);
- await sendMessage(msg.chat.id, '创建用户失败,请稍后重试');
- return;
- }
- } else {
- // newOperatorId = user[0].id.toString();
- }
- // 检查是否已经是操作人
- if (operators.some(op => op.operator_id === newOperatorId)) {
- await sendMessage(msg.chat.id, '该用户已经是操作人');
- return;
- }
- // 添加新操作人
- operators.push({
- operator_id: newOperatorId,
- operator_username: username,
- operator_uid:newOperatorUid,
- added_by: userId,
- added_at: new Date().toISOString()
- });
- // 更新群组
- await pool.query(
- 'UPDATE groups SET operators = ? WHERE group_id = ?',
- [JSON.stringify(operators), groupId]
- );
- await sendMessage(msg.chat.id, `已成功设置 @${username} 为操作人`);
- console.log(`设置操作人成功 - 群组: ${msg.chat.title}, 新操作人: ${username}, 设置者: ${msg.from.username || msg.from.first_name}, 时间: ${new Date().toLocaleString()}`);
- } catch (error) {
- console.error('设置操作人失败:', error);
- if (error.code === 'ER_BAD_FIELD_ERROR') {
- await sendMessage(msg.chat.id, '系统正在升级,请稍后再试');
- } else {
- await sendMessage(msg.chat.id, '设置操作人失败,请稍后重试');
- }
- }
- }
- });
- // - 2.查看账单
- // 生成账单消息
- async function generateBillMessage(chatId) {
- try {
- console.log(`开始生成账单消息 - 群组ID: ${chatId}`);
- // 获取群组的最后加入时间和费率信息
- const [groupInfo] = await pool.query(
- 'SELECT last_join_time, in_fee_rate, in_exchange_rate, out_fee_rate, out_exchange_rate FROM groups WHERE group_id = ?',
- [chatId.toString()]
- );
- if (!groupInfo || groupInfo.length === 0) {
- console.log(`群组信息不存在 - 群组ID: ${chatId}`);
- return '暂无交易记录';
- }
- console.log(`获取到群组信息:`, groupInfo[0]);
- const lastJoinTime = groupInfo[0].last_join_time;
- const inFeeRate = parseFloat(groupInfo[0].in_fee_rate) || 0;
- const inExchangeRate = parseFloat(groupInfo[0].in_exchange_rate) || 0;
- const outFeeRate = parseFloat(groupInfo[0].out_fee_rate) || 0;
- const outExchangeRate = parseFloat(groupInfo[0].out_exchange_rate) || 0;
- // 获取机器人加入后的交易记录
- const [records] = await pool.query(`
- SELECT t.*,
- COALESCE(t.fee_rate, g.in_fee_rate) as fee_rate,
- COALESCE(t.exchange_rate, g.in_exchange_rate) as exchange_rate,
- t.totalDeposit as total_deposit,
- t.totalWithdrawal as total_withdrawal,
- t.depositFee as deposit_fee,
- t.withdrawalFee as withdrawal_fee,
- t.totalUDeposit as total_u_deposit,
- t.totalUWithdrawal as total_u_withdrawal
- FROM transactions t
- LEFT JOIN groups g ON t.group_id = g.group_id
- WHERE t.group_id = ?
- AND DATE(t.time) = CURDATE()
- ORDER BY t.time DESC
- `, [chatId.toString()]);
- if (!records || records.length === 0) {
- console.log(`今日无交易记录 - 群组ID: ${chatId}`);
- return '暂无交易记录';
- }
- console.log(`获取到交易记录数量: ${records.length}`);
- // 将记录按类型分类:入款和下发
- const deposits = records.filter(r => r.type === 'deposit');
- const withdrawals = records.filter(r => r.type === 'withdrawal');
- // 获取最新一条记录的总金额数据
- const latestRecord = records[0];
- console.log(`最新交易记录:`, latestRecord);
- const totalDeposit = parseFloat(latestRecord.total_deposit) || 0;
- const totalWithdrawal = parseFloat(latestRecord.total_withdrawal) || 0;
- const depositFee = parseFloat(latestRecord.deposit_fee) || 0;
- const withdrawalFee = parseFloat(latestRecord.withdrawal_fee) || 0;
- const totalUDeposit = parseFloat(latestRecord.total_u_deposit) || 0;
- const totalUWithdrawal = parseFloat(latestRecord.total_u_withdrawal) || 0;
- console.log(`金额数据:`, {
- totalDeposit,
- totalWithdrawal,
- depositFee,
- withdrawalFee,
- totalUDeposit,
- totalUWithdrawal
- });
- // 获取当前日期
- const today = new Date();
- const version = 'v29';
- const dateStr = today.toISOString().split('T')[0].replace(/-/g, '-');
- let message = `当前版本:<b>${dateStr}:${version}</b>\n\n`;
- // 添加入款记录
- if (deposits.length > 0) {
- message += `<b>入款笔数</b>:<code>${deposits.length}</code>\n`;
- deposits.forEach(deposit => {
- message += `<code>${moment(deposit.time).format('HH:mm:ss')} ${parseFloat(deposit.amount).toFixed(2)}</code>\n`;
- });
- message += '\n';
- } else {
- message += `<b>入款笔数</b>:<code>0</code>\n\n`;
- }
- // // 添加出款记录
- // if (withdrawals.length > 0) {
- // message += `<b>出款笔数</b>:<code>${withdrawals.length}</code>\n`;
- // withdrawals.forEach(withdrawal => {
- // message += `<code>${moment(withdrawal.time).format('HH:mm:ss')} ${parseFloat(withdrawal.amount).toFixed(2)}</code>\n`;
- // });
- // message += '\n';
- // } else {
- // message += `<b>出款笔数</b>:<code>0</code>\n\n`;
- // }
- // 添加费率信息
- message += `<b>入款费率</b>:<code>${inFeeRate}%</code>\n`;
- message += `<b>入款汇率</b>:<code>${inExchangeRate}</code>\n`;
- message += `<b>入款总额</b>:<code>${totalDeposit.toFixed(2)}</code>\n`;
- message += `<b>入款合计</b>:<code>${totalDeposit.toFixed(2)}|${totalUDeposit.toFixed(2)}U</code>\n\n`;
- // message += `<b>出款费率</b>:<code>${outFeeRate}%</code>\n`;
- // message += `<b>出款汇率</b>:<code>${outExchangeRate}</code>\n`;
- // message += `<b>出款总额</b>:<code>${totalWithdrawal.toFixed(2)}</code>\n`;
- // message += `<b>出款合计</b>:<code>${(totalWithdrawal - withdrawalFee).toFixed(2)}|${totalUWithdrawal.toFixed(2)}U</code>\n\n`;
- // 添加余额信息
- const remainingAmount = totalUDeposit - totalUWithdrawal;
- message += `<b>应下发</b>:<code>${(remainingAmount < 0 ? 0 : remainingAmount).toFixed(2)}U</code>\n`;
- message += `<b>已下发</b>:<code>${totalUWithdrawal.toFixed(2)}U</code>\n`;
- message += `<b>未下发</b>:<code>${(remainingAmount < 0 ? 0 : remainingAmount).toFixed(2)}U</code>`;
- console.log(`账单消息生成成功 - 群组ID: ${chatId}`);
- return message;
- } catch (error) {
- console.error('生成账单消息失败:', error);
- console.error('错误详情:', {
- message: error.message,
- stack: error.stack,
- chatId: chatId
- });
- return '获取账单信息失败,请稍后重试';
- }
- }
- // 生成内联键盘
- function generateInlineKeyboard(chatId) {
- const keyboard = {
- inline_keyboard: [
- [{
- text: '点击跳转完整账单',
- callback_data: `bill_page_${chatId}`
- }],
- [{
- text: '24小时商务对接',
- callback_data: 'business_contact'
- }]
- ]
- };
- return keyboard;
- }
- // 处理内联按钮回调
- bot.on('callback_query', async (callbackQuery) => {
- const chatId = callbackQuery.message.chat.id;
- const data = callbackQuery.data;
- try {
- if (data.startsWith('bill_page_')) {
- const groupId = data.split('_')[2];
- console.log('https://jfpay.top/admin/views/statistics_bill.html?groupId='+groupId)
- await bot.sendMessage(chatId, `点击查看完整账单:[完整账单](https://jfpay.top/admin/views/statistics_bill.html?groupId=${groupId})`, {parse_mode: 'Markdown'});
- } else if (data === 'business_contact') {
- await bot.sendMessage(chatId, `24小时商务对接:[点击跳转](https://t.me/yyyyaaaa123_bot)`, {parse_mode: 'Markdown'})
- }
- } catch (error) {
- console.log(error)
- console.error(formatLog('处理内联按钮回调失败', error));
- await bot.answerCallbackQuery(callbackQuery.id, {
- text: '操作失败,请稍后重试',
- show_alert: true
- });
- }
- });
- bot.onText(/\/bill/, async (msg) => {
- const billMessage = await generateBillMessage(msg.chat.id);
- sendMessage(msg.chat.id, billMessage, {
- reply_markup: generateInlineKeyboard(msg.chat.id)
- });
- });
- // - 3.帮助命令
- bot.onText(/\/help/, (msg) => {
- const helpMessage = `
- 🤖 <b>机器人使用指南</b>
- <b>📝 快捷命令</b>
- • <code>+数字</code> - 快速记录入款
- • <code>-数字</code> - 快速记录入款修正
- • <code>下发数字</code> - 快速记录下发
- • <code>下发-数字</code> - 快速记录下发修正
- <b>⚙️ 设置命令</b>
- • <code>设置费率数字</code> - 同时设置入款和出款费率
- • <code>设置入款费率数字</code> - 设置入款费率
- • <code>设置出款费率数字</code> - 设置出款费率
- • <code>设置汇率数字</code> - 同时设置入款和出款汇率
- • <code>设置入款汇率数字</code> - 设置入款汇率
- • <code>设置出款汇率数字</code> - 设置出款汇率
- • <code>回复消息"设置操作人"</code> - 将回复的用户设为群组操作人
- <b>📊 查询命令</b>
- • <code>/bill</code> - 查看当前账单
- • <code>/help</code> - 显示此帮助信息
- <b>💡 使用提示</b>
- • 所有金额输入请使用数字
- • 账单信息实时更新
- • 如需帮助请联系管理员
- `;
- sendMessage(msg.chat.id, helpMessage);
- });
- // - 4.群组信息更新
- // 处理机器人成员状态更新(包括被邀请到新群组)
- bot.on('my_chat_member', async (msg) => {
- try {
- const chatId = msg.chat.id.toString();
- const chatTitle = msg.chat.title || '未命名群组';
- const creatorId = msg.from.id.toString();
- const newStatus = msg.new_chat_member.status;
- // 只处理机器人被添加到群组的情况
- if (newStatus === 'member' || newStatus === 'administrator') {
- // 检查群组是否已存在
- const [existingGroup] = await pool.query(
- 'SELECT * FROM groups WHERE group_id = ?',
- [chatId]
- );
- if (!existingGroup || existingGroup.length === 0) {
- // 创建新群组记录
- await pool.query(`
- INSERT INTO groups (
- group_id,
- group_name,
- creator_id,
- in_fee_rate,
- out_fee_rate,
- in_exchange_rate,
- out_exchange_rate,
- last_join_time,
- created_at,
- updated_at,
- operators
- ) VALUES (?, ?, ?, 0, 0, 1, 1, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, CURRENT_TIMESTAMP, '[]')
- `, [chatId, chatTitle, creatorId]);
- // 将群组添加到允许列表
- if (!data.allowedGroups.includes(chatId)) {
- data.allowedGroups.push(chatId);
- saveData();
- }
- // 发送欢迎消息
- await sendMessage(msg.chat.id, `感谢您添加我进入群组!\n\n我已初始化群组账单系统,您可以使用以下命令开始使用:\n\n• <code>/help</code> 查看使用指南\n• <code>/bill</code> 查看当前账单`);
-
- console.log(`新群组初始化成功 - 群组: ${chatTitle}, ID: ${chatId}, 创建者: ${creatorId}, 时间: ${new Date().toLocaleString()}`);
- } else {
- // 更新群组信息
- await pool.query(`
- UPDATE groups
- SET group_name = ?,
- last_join_time = CURRENT_TIMESTAMP,
- updated_at = CURRENT_TIMESTAMP
- WHERE group_id = ?
- `, [chatTitle, chatId]);
- // 确保群组在允许列表中
- if (!data.allowedGroups.includes(chatId)) {
- data.allowedGroups.push(chatId);
- saveData();
- }
- await sendMessage(msg.chat.id, `我已重新加入群组!\n\n您可以使用 <code>/help</code> 查看使用指南`);
- console.log(`群组信息更新成功 - 群组: ${chatTitle}, ID: ${chatId}, 时间: ${new Date().toLocaleString()}`);
- }
- }
- } catch (error) {
- console.error('处理机器人成员状态更新失败:', error);
- await sendMessage(msg.chat.id, '初始化群组失败,请稍后重试');
- }
- });
- // 保存数据
- function saveData() {
- try {
- fs.writeFileSync(process.env.DB_FILE, JSON.stringify(data, null, 2));
- } catch (error) {
- console.error(formatLog('Error saving data', error));
- }
- }
- // 加载数据
- function loadData() {
- try {
- if (fs.existsSync(process.env.DB_FILE)) {
- const savedData = JSON.parse(fs.readFileSync(process.env.DB_FILE));
- data = {
- ...data,
- ...savedData
- };
- }
- } catch (error) {
- console.error(formatLog('Error loading data', error));
- }
- }
- // 测试数据库连接并初始化
- testConnection().then(() => {
- // 设置柬埔寨时间00:00重置任务(东七区)
- schedule.scheduleJob({second:0, minute: 0,hour: 0, tz: 'Asia/Bangkok' }, async () => {
- try {
- // 重置所有群组的费率和汇率
- await pool.query(`
- UPDATE groups
- SET in_fee_rate = 0,
- out_fee_rate = 0,
- in_exchange_rate = 1,
- out_exchange_rate = 1,
- updated_at = CURRENT_TIMESTAMP
- `);
-
- console.log('每日重置完成 - 时间:', new Date().toLocaleString());
-
- // 向所有群组发送重置通知
- const [groups] = await pool.query('SELECT group_id FROM groups');
- for (const group of groups) {
- try {
- await sendMessage(group.group_id, '⚠️ 系统已重置\n\n请重新设置费率和汇率后继续使用。\n\n使用以下命令设置:\n• 设置费率数字\n• 设置汇率数字');
- } catch (error) {
- console.error(`发送重置通知失败 - 群组ID: ${group.group_id}`, error);
- }
- }
- } catch (error) {
- console.error('每日重置失败:', error);
- }
- });
-
- console.log('已设置每日重置任务 - 东七区时间 00:00');
- }).then(() => {
- // 加载数据
- loadData();
- // 启动服务器
- const PORT = process.env.PORT || 3000;
- app.listen(PORT, () => {
- console.log(formatLog({
- PORT: PORT
- }));
- console.log('机器人已准备就绪!');
- });
- }).catch(error => {
- console.error(formatLog('启动失败', error));
- process.exit(1);
- });
- // 导入公共路由
- const publicRoutes = require('./routes/public');
- // 注册公共路由
- app.use('/api/public', publicRoutes);
- // 错误处理中间件
- app.use((err, req, res, next) => {
- console.error(err.stack);
- res.status(500).json({ message: '服务器错误' });
- });
- // 404 处理
- app.use((req, res) => {
- res.status(404).json({ message: '未找到请求的资源' });
- });
- module.exports = app;
|