369 lines
16 KiB
JavaScript
369 lines
16 KiB
JavaScript
import TelegramBot from 'node-telegram-bot-api';
|
|
import config from './config/config.js';
|
|
import UserHandler from './handlers/userHandler.js';
|
|
import UserProductHandler from './handlers/userProductHandler.js';
|
|
import UserWalletsHandler from './handlers/userWalletsHandler.js';
|
|
import UserLocationHandler from './handlers/userLocationHandler.js';
|
|
import AdminHandler from './handlers/adminHandler.js';
|
|
import AdminUserHandler from './handlers/adminUserHandler.js';
|
|
import AdminLocationHandler from './handlers/adminLocationHandler.js';
|
|
import AdminProductHandler from './handlers/adminProductHandler.js';
|
|
import ErrorHandler from './utils/errorHandler.js';
|
|
import User from './models/User.js';
|
|
import AdminUserLocationHandler from "./handlers/adminUserLocationHandler.js";
|
|
import AdminDumpHandler from "./handlers/adminDumpHandler.js";
|
|
|
|
// Debug logging function
|
|
const logDebug = (action, functionName) => {
|
|
console.log(`[DEBUG] Button Press: ${action}`);
|
|
console.log(`[DEBUG] Calling Function: ${functionName}`);
|
|
};
|
|
|
|
const initBot = () => {
|
|
try {
|
|
const bot = new TelegramBot(config.BOT_TOKEN, {polling: true});
|
|
console.log('Bot initialized successfully');
|
|
return bot;
|
|
} catch (error) {
|
|
console.error('Failed to initialize bot:', error);
|
|
process.exit(1);
|
|
}
|
|
};
|
|
|
|
const bot = initBot();
|
|
const userHandler = new UserHandler(bot);
|
|
const userProductHandler = new UserProductHandler(bot);
|
|
const userWalletsHandler = new UserWalletsHandler(bot);
|
|
const userLocationHandler = new UserLocationHandler(bot);
|
|
const adminHandler = new AdminHandler(bot);
|
|
const adminUserHandler = new AdminUserHandler(bot);
|
|
const adminLocationHandler = new AdminLocationHandler(bot);
|
|
const adminUserLocationHandler = new AdminUserLocationHandler(bot);
|
|
const adminProductHandler = new AdminProductHandler(bot);
|
|
const adminDumpHandler = new AdminDumpHandler(bot);
|
|
|
|
// Start command - Create user profile
|
|
bot.onText(/\/start/, async (msg) => {
|
|
logDebug('/start', 'handleStart');
|
|
|
|
const canUse = await userHandler.canUseBot(msg);
|
|
|
|
if (!canUse) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
await userHandler.handleStart(msg);
|
|
} catch (error) {
|
|
await ErrorHandler.handleError(bot, msg.chat.id, error, 'start command');
|
|
}
|
|
});
|
|
|
|
// Admin command
|
|
bot.onText(/\/admin/, async (msg) => {
|
|
logDebug('/admin', 'handleAdminCommand');
|
|
try {
|
|
await adminHandler.handleAdminCommand(msg);
|
|
} catch (error) {
|
|
await ErrorHandler.handleError(bot, msg.chat.id, error, 'admin command');
|
|
}
|
|
});
|
|
|
|
// Handle user menu buttons
|
|
bot.on('message', async (msg) => {
|
|
if (msg.text && msg.text.toLowerCase() === '/start') {
|
|
return;
|
|
}
|
|
|
|
const canUse = await userHandler.canUseBot(msg);
|
|
|
|
if (!canUse) {
|
|
return;
|
|
}
|
|
|
|
try {
|
|
// Check for admin location input
|
|
if (await adminLocationHandler.handleLocationInput(msg)) {
|
|
return;
|
|
}
|
|
|
|
// Check for admin category input
|
|
if (await adminProductHandler.handleCategoryInput(msg)) {
|
|
return;
|
|
}
|
|
|
|
// Check for admin subcategory input
|
|
if (await adminProductHandler.handleSubcategoryInput(msg)) {
|
|
return;
|
|
}
|
|
|
|
// Check for product import
|
|
if (await adminProductHandler.handleProductImport(msg)) {
|
|
return;
|
|
}
|
|
|
|
// Check for product edition
|
|
if (await adminProductHandler.handleProductEditImport(msg)) {
|
|
return;
|
|
}
|
|
|
|
// Check for database dump import
|
|
if (await adminDumpHandler.handleDumpImport(msg)) {
|
|
return;
|
|
}
|
|
|
|
logDebug(msg.text, 'handleMessage');
|
|
|
|
switch (msg.text) {
|
|
case '📦 Products':
|
|
await userProductHandler.showProducts(msg);
|
|
break;
|
|
case '👤 Profile':
|
|
await userHandler.showProfile(msg);
|
|
break;
|
|
case '💰 Wallets':
|
|
await userWalletsHandler.showBalance(msg);
|
|
break;
|
|
case '🛍 Purchases':
|
|
await userProductHandler.showPurchases(msg);
|
|
break;
|
|
case '📦 Manage Products':
|
|
if (adminHandler.isAdmin(msg.from.id)) {
|
|
await adminProductHandler.handleProductManagement(msg);
|
|
}
|
|
break;
|
|
case '👥 Manage Users':
|
|
if (adminHandler.isAdmin(msg.from.id)) {
|
|
await adminUserHandler.handleUserList(msg);
|
|
}
|
|
break;
|
|
case '📍 Manage Locations':
|
|
if (adminHandler.isAdmin(msg.from.id)) {
|
|
await adminLocationHandler.handleViewLocations(msg);
|
|
}
|
|
break;
|
|
case '💾 Database Backup':
|
|
if (adminHandler.isAdmin(msg.from.id)) {
|
|
await adminDumpHandler.handleDump(msg);
|
|
}
|
|
break;
|
|
}
|
|
} catch (error) {
|
|
await ErrorHandler.handleError(bot, msg.chat.id, error, 'message handler');
|
|
}
|
|
});
|
|
|
|
// Handle callback queries
|
|
bot.on('callback_query', async (callbackQuery) => {
|
|
const action = callbackQuery.data;
|
|
const msg = callbackQuery.message;
|
|
|
|
const canUse = await userHandler.canUseBot(callbackQuery);
|
|
|
|
if (!canUse) {
|
|
await bot.answerCallbackQuery(callbackQuery.id);
|
|
return;
|
|
}
|
|
|
|
try {
|
|
// Profile and location management
|
|
if (action === 'set_location') {
|
|
logDebug(action, 'handleSetLocation');
|
|
await userLocationHandler.handleSetLocation(callbackQuery);
|
|
} else if (action.startsWith('set_country_')) {
|
|
logDebug(action, 'handleSetCountry');
|
|
await userLocationHandler.handleSetCountry(callbackQuery);
|
|
} else if (action.startsWith('set_city_')) {
|
|
logDebug(action, 'handleSetCity');
|
|
await userLocationHandler.handleSetCity(callbackQuery);
|
|
} else if (action.startsWith('set_district_')) {
|
|
logDebug(action, 'handleSetDistrict');
|
|
await userLocationHandler.handleSetDistrict(callbackQuery);
|
|
} else if (action === 'back_to_profile') {
|
|
logDebug(action, 'handleBackToProfile');
|
|
await userHandler.handleBackToProfile(callbackQuery);
|
|
} else if (action === 'back_to_balance') {
|
|
logDebug(action, 'handleBackToBalance');
|
|
await userWalletsHandler.handleBackToBalance(callbackQuery);
|
|
}
|
|
// Wallet management
|
|
else if (action === 'add_wallet') {
|
|
logDebug(action, 'handleAddWallet');
|
|
await userWalletsHandler.handleAddWallet(callbackQuery);
|
|
} else if (action === 'top_up_wallet') {
|
|
logDebug(action, 'handleTopUpWallet');
|
|
await userWalletsHandler.handleTopUpWallet(callbackQuery);
|
|
} else if (action === 'wallet_history') {
|
|
logDebug(action, 'handleWalletHistory');
|
|
await userWalletsHandler.handleWalletHistory(callbackQuery);
|
|
} else if (action === 'view_archived_wallets') {
|
|
logDebug(action, 'handleViewArchivedWallets');
|
|
await userWalletsHandler.handleViewArchivedWallets(callbackQuery);
|
|
} else if (action === 'refresh_balance') {
|
|
logDebug(action, 'handleRefreshBalance');
|
|
await userWalletsHandler.handleRefreshBalance(callbackQuery);
|
|
}
|
|
// Wallet generation
|
|
else if (action.startsWith('generate_wallet_')) {
|
|
logDebug(action, 'handleGenerateWallet');
|
|
await userWalletsHandler.handleGenerateWallet(callbackQuery);
|
|
}
|
|
// Shop navigation
|
|
else if (action === 'shop_start') {
|
|
logDebug(action, 'showProducts');
|
|
await userProductHandler.showProducts(msg);
|
|
} else if (action.startsWith('shop_country_')) {
|
|
logDebug(action, 'handleCountrySelection');
|
|
await userProductHandler.handleCountrySelection(callbackQuery);
|
|
} else if (action.startsWith('shop_city_')) {
|
|
logDebug(action, 'handleCitySelection');
|
|
await userProductHandler.handleCitySelection(callbackQuery);
|
|
} else if (action.startsWith('shop_district_')) {
|
|
logDebug(action, 'handleDistrictSelection');
|
|
await userProductHandler.handleDistrictSelection(callbackQuery);
|
|
} else if (action.startsWith('shop_category_')) {
|
|
logDebug(action, 'handleCategorySelection');
|
|
await userProductHandler.handleCategorySelection(callbackQuery);
|
|
} else if (action.startsWith('shop_subcategory_')) {
|
|
logDebug(action, 'handleSubcategorySelection');
|
|
await userProductHandler.handleSubcategorySelection(callbackQuery);
|
|
} else if (action.startsWith('shop_product_')) {
|
|
logDebug(action, 'handleProductSelection');
|
|
await userProductHandler.handleProductSelection(callbackQuery);
|
|
} else if (action.startsWith('increase_quantity_')) {
|
|
logDebug(action, 'handleIncreaseQuantity');
|
|
await userProductHandler.handleIncreaseQuantity(callbackQuery);
|
|
} else if (action.startsWith('decrease_quantity_')) {
|
|
logDebug(action, 'handleDecreaseQuantity');
|
|
await userProductHandler.handleDecreaseQuantity(callbackQuery);
|
|
} else if (action.startsWith('buy_product_')) {
|
|
logDebug(action, 'handleBuyProduct');
|
|
await userProductHandler.handleBuyProduct(callbackQuery);
|
|
} else if (action.startsWith('pay_with_')) {
|
|
logDebug(action, 'handlePay');
|
|
await userProductHandler.handlePay(callbackQuery);
|
|
}
|
|
// Admin location management
|
|
else if (action === 'add_location') {
|
|
logDebug(action, 'handleAddLocation');
|
|
await adminLocationHandler.handleAddLocation(callbackQuery);
|
|
} else if (action === 'view_locations') {
|
|
logDebug(action, 'handleViewLocations');
|
|
await adminLocationHandler.handleViewLocations(callbackQuery);
|
|
} else if (action === 'delete_location') {
|
|
logDebug(action, 'handleDeleteLocation');
|
|
await adminLocationHandler.handleDeleteLocation(callbackQuery);
|
|
} else if (action.startsWith('confirm_delete_location_')) {
|
|
logDebug(action, 'handleConfirmDelete');
|
|
await adminLocationHandler.handleConfirmDelete(callbackQuery);
|
|
} else if (action === 'admin_menu') {
|
|
logDebug(action, 'backToMenu');
|
|
await adminLocationHandler.backToMenu(callbackQuery);
|
|
}
|
|
|
|
// Admin product management
|
|
else if (action === 'manage_products') {
|
|
logDebug(action, 'handleProductManagement');
|
|
await adminProductHandler.handleProductManagement(callbackQuery);
|
|
} else if (action.startsWith('prod_country_')) {
|
|
logDebug(action, 'handleCountrySelection');
|
|
await adminProductHandler.handleCountrySelection(callbackQuery);
|
|
} else if (action.startsWith('prod_city_')) {
|
|
logDebug(action, 'handleCitySelection');
|
|
await adminProductHandler.handleCitySelection(callbackQuery);
|
|
} else if (action.startsWith('prod_district_')) {
|
|
logDebug(action, 'handleDistrictSelection');
|
|
await adminProductHandler.handleDistrictSelection(callbackQuery);
|
|
} else if (action.startsWith('add_category_')) {
|
|
logDebug(action, 'handleAddCategory');
|
|
await adminProductHandler.handleAddCategory(callbackQuery);
|
|
} else if (action.startsWith('edit_category_')) {
|
|
logDebug(action, 'handleEditCategory');
|
|
await adminProductHandler.handleEditCategory(callbackQuery);
|
|
} else if (action.startsWith('prod_category_')) {
|
|
logDebug(action, 'handleCategorySelection');
|
|
await adminProductHandler.handleCategorySelection(callbackQuery);
|
|
} else if (action.startsWith('add_subcategory_')) {
|
|
logDebug(action, 'handleAddSubcategory');
|
|
await adminProductHandler.handleAddSubcategory(callbackQuery);
|
|
} else if (action.startsWith('prod_subcategory_')) {
|
|
logDebug(action, 'handleSubcategorySelection');
|
|
await adminProductHandler.handleSubcategorySelection(callbackQuery);
|
|
} else if (action.startsWith('list_products_')) {
|
|
logDebug(action, 'handleProductListPage');
|
|
await adminProductHandler.handleProductListPage(callbackQuery);
|
|
} else if (action.startsWith('add_product_')) {
|
|
logDebug(action, 'handleAddProduct');
|
|
await adminProductHandler.handleAddProduct(callbackQuery);
|
|
} else if (action.startsWith('view_product_')) {
|
|
logDebug(action, 'handleViewProduct');
|
|
await adminProductHandler.handleViewProduct(callbackQuery);
|
|
} else if (action.startsWith('edit_product_')) {
|
|
logDebug(action, 'handleProductEdit');
|
|
await adminProductHandler.handleProductEdit(callbackQuery)
|
|
} else if (action.startsWith('delete_product_')) {
|
|
logDebug(action, 'handleProductDelete');
|
|
await adminProductHandler.handleProductDelete(callbackQuery);
|
|
} else if (action.startsWith('confirm_delete_product_')) {
|
|
logDebug(action, 'handleConfirmDelete');
|
|
await adminProductHandler.handleConfirmDelete(callbackQuery);
|
|
}
|
|
// Admin user management
|
|
else if (action.startsWith('view_user_')) {
|
|
logDebug(action, 'handleViewUser');
|
|
await adminUserHandler.handleViewUser(callbackQuery);
|
|
} else if (action.startsWith('list_users_')) {
|
|
logDebug(action, 'handleUserListPage');
|
|
await adminUserHandler.handleUserListPage(callbackQuery);
|
|
} else if (action.startsWith('delete_user_')) {
|
|
logDebug(action, 'handleDeleteUser');
|
|
await adminUserHandler.handleDeleteUser(callbackQuery);
|
|
} else if (action.startsWith('block_user_')) {
|
|
logDebug(action, 'handleBlockUser');
|
|
await adminUserHandler.handleBlockUser(callbackQuery);
|
|
} else if (action.startsWith('confirm_delete_user_')) {
|
|
logDebug(action, 'handleConfirmDelete');
|
|
await adminUserHandler.handleConfirmDelete(callbackQuery);
|
|
} else if (action.startsWith('confirm_block_user_')) {
|
|
logDebug(action, 'handleConfirmBlock');
|
|
await adminUserHandler.handleConfirmBlock(callbackQuery);
|
|
} else if (action.startsWith('edit_user_balance_')) {
|
|
logDebug(action, 'handleEditUserBalance');
|
|
await adminUserHandler.handleEditUserBalance(callbackQuery);
|
|
}
|
|
// Admin users location management
|
|
else if (action.startsWith('edit_user_location_')) {
|
|
logDebug(action, 'handleEditUserLocation');
|
|
await adminUserLocationHandler.handleEditUserLocation(callbackQuery);
|
|
} else if (action.startsWith('edit_user_country_')) {
|
|
logDebug(action, 'handleEditUserCountry');
|
|
await adminUserLocationHandler.handleEditUserCountry(callbackQuery);
|
|
} else if (action.startsWith('edit_user_city_')) {
|
|
logDebug(action, 'handleEditUserCity');
|
|
await adminUserLocationHandler.handleEditUserCity(callbackQuery);
|
|
} else if (action.startsWith('edit_user_district_')) {
|
|
logDebug(action, 'handleEditUserDistrict');
|
|
await adminUserLocationHandler.handleEditUserDistrict(callbackQuery)
|
|
}
|
|
// Dump manage
|
|
else if (action === "export_database") {
|
|
await adminDumpHandler.handleExportDatabase(callbackQuery);
|
|
return;
|
|
} else if (action === "import_database") {
|
|
await adminDumpHandler.handleImportDatabase(callbackQuery);
|
|
}
|
|
|
|
await bot.answerCallbackQuery(callbackQuery.id);
|
|
} catch (error) {
|
|
await ErrorHandler.handleError(bot, msg.chat.id, error, 'callback query');
|
|
}
|
|
});
|
|
|
|
// Error handling
|
|
bot.on('polling_error', ErrorHandler.handlePollingError);
|
|
|
|
process.on('unhandledRejection', (error) => {
|
|
console.error('Unhandled promise rejection:', error);
|
|
});
|
|
|
|
console.log('Bot is running...'); |