user deletion/blocking

src/models/User.js

@@ -1,50 +1,50 @@
 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;
-      }
+    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');
+            // 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]
-      );
+            // Create new user
+            const result = await db.runAsync(
+                'INSERT INTO users (telegram_id, username) VALUES (?, ?)',
+                [telegramId.toString(), username]
+            );
 
-      // Commit transaction
-      await db.runAsync('COMMIT');
+            // 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;
+            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 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(`
+    static async getUserStats(telegramId) {
+        try {
+            return await db.getAsync(`
         SELECT 
           u.*,
           COUNT(DISTINCT p.id) as purchase_count,
@@ -58,9 +58,49 @@ export default class User {
         WHERE u.telegram_id = ?
         GROUP BY u.id
       `, [telegramId.toString()]);
-    } catch (error) {
-      console.error('Error getting user stats:', error);
-      throw error;
+        } 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;
+        }
     }
-  }
 }