Telegram-Market/telegram-shop
1
1
Fork 0
Code Issues 9 Pull Requests Actions Packages Projects 1 Releases Wiki Activity

user deletion/blocking

Browse Source
This commit is contained in:
Artyom Ashirov 2024-11-15 02:26:13 +03:00
parent b45f7daa6f
commit 4251f1a0bd
6 changed files with 278 additions and 111 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
src/config/database.js
View File

@ -94,6 +94,7 @@ const initDb = async () => {
country TEXT,
city TEXT,
district TEXT,
status INTEGER DEFAULT 0,
created_at DATETIME DEFAULT CURRENT_TIMESTAMP
)
`);

4
src/handlers/adminLocationHandler.js
View File

@ -215,7 +215,7 @@ export default class AdminLocationHandler {
const keyboard = {
inline_keyboard: locations.map(loc => [{
text: `${loc.country} > ${loc.city} > ${loc.district} (P:${loc.product_count} C:${loc.category_count})`,
callback_data: `confirm_delete_${loc.country}_${loc.city}_${loc.district}`
callback_data: `confirm_delete_location_${loc.country}_${loc.city}_${loc.district}`
}])
};
@ -239,7 +239,7 @@ export default class AdminLocationHandler {
async handleConfirmDelete(callbackQuery) {
const chatId = callbackQuery.message.chat.id;
const [country, city, district] = callbackQuery.data
.replace('confirm_delete_', '')
.replace('confirm_delete_location_', '')
.split('_');
try {

74
src/handlers/adminUserHandler.js
View File

@ -265,7 +265,7 @@ ${purchases.map(p => ` • ${p.product_name} x${p.quantity} - $${p.total_price}
const chatId = callbackQuery.message.chat.id;
try {
await User.delete(userId);
await User.updateUserStatus(userId, 1);
const keyboard = {
inline_keyboard: [
@ -273,6 +273,12 @@ ${purchases.map(p => ` • ${p.product_name} x${p.quantity} - $${p.total_price}
]
};
try {
await this.bot.sendMessage(userId, '⚠️Your account has been deleted by administrator');
} catch (e) {
// ignore if we can't notify user
}
await this.bot.editMessageText(
`✅ User ${userId} has been successfully deleted.`,
{
@ -287,6 +293,72 @@ ${purchases.map(p => ` • ${p.product_name} x${p.quantity} - $${p.total_price}
}
}
async handleBlockUser(callbackQuery) {
if (!this.isAdmin(callbackQuery.from.id)) return;
const userId = callbackQuery.data.replace('block_user_', '');
const chatId = callbackQuery.message.chat.id;
try {
const keyboard = {
inline_keyboard: [
[
{text: '✅ Confirm Block', callback_data: `confirm_block_user_${userId}`},
{text: '❌ Cancel', callback_data: `view_user_${userId}`}
]
]
};
await this.bot.editMessageText(
`⚠️ Are you sure you want to block user ${userId}?`,
{
chat_id: chatId,
message_id: callbackQuery.message.message_id,
reply_markup: keyboard,
parse_mode: 'HTML'
}
);
} catch (error) {
console.error('Error in handleBlockUser:', error);
await this.bot.sendMessage(chatId, 'Error processing block request. Please try again.');
}
}
async handleConfirmBlock(callbackQuery) {
if (!this.isAdmin(callbackQuery.from.id)) return;
const userId = callbackQuery.data.replace('confirm_block_user_', '');
const chatId = callbackQuery.message.chat.id;
try {
await User.updateUserStatus(userId, 2);
const keyboard = {
inline_keyboard: [
[{text: '« Back to User List', callback_data: 'admin_users'}]
]
};
try {
await this.bot.sendMessage(userId, '⚠️Your account has been blocked by administrator');
} catch (e) {
// ignore if we can't notify user
}
await this.bot.editMessageText(
`✅ User ${userId} has been successfully blocked.`,
{
chat_id: chatId,
message_id: callbackQuery.message.message_id,
reply_markup: keyboard
}
);
} catch (error) {
console.error('Error in handleConfirmBlock:', error);
await this.bot.sendMessage(chatId, 'Error blocking user. Please try again.');
}
}
async handleEditUserBalance(callbackQuery) {
if (!this.isAdmin(callbackQuery.from.id)) return;

26
src/handlers/userHandler.js
View File

@ -1,11 +1,35 @@
import db from '../config/database.js';
import User from '../models/User.js';
import config from "../config/config.js";
export default class UserHandler {
constructor(bot) {
this.bot = bot;
}
async canUseBot(msg) {
const userId = msg.from.id;
const user = await User.getById(userId);
const keyboard = {
inline_keyboard: [
[{text: "Contact support", url: config.SUPPORT_LINK}]
]
};
switch (user?.status) {
case 0:
return true;
case 1:
await this.bot.sendMessage(userId, '⚠️Your account has been deleted by administrator', {reply_markup: keyboard});
return false;
case 2:
await this.bot.sendMessage(userId, '⚠️Your account has been blocked by administrator', {reply_markup: keyboard});
return false;
default:
return true
}
}
async showProfile(msg) {
const chatId = msg.chat.id;
const userId = msg.from.id;

32
src/index.js
View File

@ -43,6 +43,13 @@ const adminProductHandler = new AdminProductHandler(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) {
@ -64,6 +71,16 @@ bot.onText(/\/admin/, async (msg) => {
bot.on('message', async (msg) => {
if (!msg.text) return;
if (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)) {
@ -126,6 +143,13 @@ 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') {
@ -211,7 +235,7 @@ bot.on('callback_query', async (callbackQuery) => {
} else if (action === 'delete_location') {
logDebug(action, 'handleDeleteLocation');
await adminLocationHandler.handleDeleteLocation(callbackQuery);
} else if (action.startsWith('confirm_delete_')) {
} else if (action.startsWith('confirm_delete_location_')) {
logDebug(action, 'handleConfirmDelete');
await adminLocationHandler.handleConfirmDelete(callbackQuery);
} else if (action === 'admin_menu') {
@ -264,9 +288,15 @@ bot.on('callback_query', async (callbackQuery) => {
} else if (action.startsWith('delete_user_')) {
logDebug(action, 'handleDeleteUser');
await adminUserHandler.handleDeleteUser(callbackQuery);
} else if (action.startsWith('block_user_')) {
logDebug(action, 'handleDeleteUser');
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);

40
src/models/User.js
View File

@ -63,4 +63,44 @@ export default class User {
throw error;
}
}
static async updateUserStatus(telegramId, status) {
// statuses
// 0 - active
// 1 - deleted
// 2 - blocked
try {
await db.runAsync('BEGIN TRANSACTION');
// Update user status
await db.runAsync('UPDATE users SET status = ? WHERE telegram_id = ?', [status, telegramId.toString()]);
// Commit transaction
await db.runAsync('COMMIT');
} catch (e) {
await db.runAsync("ROLLBACK");
console.error('Error deleting user:', error);
throw error;
}
}
static async delete(telegramId) {
try {
await db.runAsync('BEGIN TRANSACTION');
// Delete user and his data
await db.runAsync('DELETE FROM users WHERE telegram_id = ?', [telegramId.toString()]);
await db.runAsync('DELETE FROM transactions WHERE user_id = ?', [telegramId.toString()]);
await db.runAsync('DELETE FROM purchases WHERE user_id = ?', [telegramId.toString()]);
await db.runAsync('DELETE FROM crypto_wallets WHERE user_id = ?', [telegramId.toString()]);
// Commit transaction
await db.runAsync('COMMIT');
} catch (e) {
await db.runAsync("ROLLBACK");
console.error('Error deleting user:', error);
throw error;
}
}
}