106 lines
3.4 KiB
JavaScript
106 lines
3.4 KiB
JavaScript
import db from '../config/database.js';
|
|
|
|
export default class User {
|
|
static async create(telegramId, username) {
|
|
try {
|
|
// First check if user exists
|
|
const existingUser = await this.getById(telegramId);
|
|
if (existingUser) {
|
|
return existingUser.id;
|
|
}
|
|
|
|
// Begin transaction
|
|
await db.runAsync('BEGIN TRANSACTION');
|
|
|
|
// Create new user
|
|
const result = await db.runAsync(
|
|
'INSERT INTO users (telegram_id, username) VALUES (?, ?)',
|
|
[telegramId.toString(), username]
|
|
);
|
|
|
|
// Commit transaction
|
|
await db.runAsync('COMMIT');
|
|
|
|
return result.lastID;
|
|
} catch (error) {
|
|
// Rollback on error
|
|
await db.runAsync('ROLLBACK');
|
|
console.error('Error creating user:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
static async getById(telegramId) {
|
|
try {
|
|
return await db.getAsync(
|
|
'SELECT * FROM users WHERE telegram_id = ?',
|
|
[telegramId.toString()]
|
|
);
|
|
} catch (error) {
|
|
console.error('Error getting user:', error);
|
|
throw error;
|
|
}
|
|
}
|
|
|
|
static async getUserStats(telegramId) {
|
|
try {
|
|
return await db.getAsync(`
|
|
SELECT
|
|
u.*,
|
|
COUNT(DISTINCT p.id) as purchase_count,
|
|
COALESCE(SUM(p.total_price), 0) as total_spent,
|
|
COUNT(DISTINCT cw.id) as crypto_wallet_count,
|
|
COUNT(DISTINCT cw2.id) as archived_wallet_count
|
|
FROM users u
|
|
LEFT JOIN purchases p ON u.id = p.user_id
|
|
LEFT JOIN crypto_wallets cw ON u.id = cw.user_id AND cw.wallet_type NOT LIKE '%_%'
|
|
LEFT JOIN crypto_wallets cw2 ON u.id = cw2.user_id AND cw2.wallet_type LIKE '%_%'
|
|
WHERE u.telegram_id = ?
|
|
GROUP BY u.id
|
|
`, [telegramId.toString()]);
|
|
} catch (error) {
|
|
console.error('Error getting user stats:', error);
|
|
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;
|
|
}
|
|
}
|
|
} |