qwen-alpha 1.0.15 → 1.0.17

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "qwen-alpha",
-  "version": "1.0.15",
+  "version": "1.0.17",
   "description": "Telegram bot for Qwen Code integration — AI-powered code review, bug detection, and code generation",
   "main": "src/index.js",
   "bin": {

package/src/bot/handlers/admin.js

@@ -50,6 +50,8 @@ async function adminHandler(ctx) {
 **Управление пользователями:**
 /admin ban <user_id> — забанить
 /admin unban <user_id> — разбанить
+/admin lock — заблокировать бота для всех кроме админов
+/admin unlock — разблокировать бота
 
 **Сессии:**
 /admin sessions list — список сессий
@@ -128,7 +130,7 @@ async function adminHandler(ctx) {
         await ctx.reply('❌ Usage: /admin unban <user_id>');
         return;
       }
-      
+
       const success = userService.unban(targetId);
       if (success) {
         await ctx.reply(`✅ Пользователь ${targetId} разбанен.`);
@@ -138,7 +140,31 @@ async function adminHandler(ctx) {
       }
       break;
     }
-    
+
+    case 'lock': {
+      // Блокировка всех пользователей кроме админов
+      const settings = storeManager.get('settings');
+      const data = settings.getData();
+      data.locked = true;
+      settings.setData(data);
+      
+      await ctx.reply('🔒 **Бот заблокирован для всех пользователей кроме админов.**\n\nИспользуйте /admin unlock для разблокировки.', { parse_mode: 'Markdown' });
+      logger.info({ userId }, 'Bot locked for all users except admins');
+      break;
+    }
+
+    case 'unlock': {
+      // Разблокировка бота
+      const settings = storeManager.get('settings');
+      const data = settings.getData();
+      data.locked = false;
+      settings.setData(data);
+      
+      await ctx.reply('🔓 **Бот разблокирован.**\n\nВсе пользователи могут снова использовать бота.', { parse_mode: 'Markdown' });
+      logger.info({ userId }, 'Bot unlocked for all users');
+      break;
+    }
+
     case 'sessions': {
       const subCommand = args[2];
       

package/src/bot/handlers/message.js

@@ -55,26 +55,22 @@ async function messageHandler(ctx) {
     let contextMessages = [];
     let session = ctx.state.session;
 
-    // Если сессии нет в группе - создаём новую
-    if (!session && !isPrivate) {
-      session = sessionService.create({
-        userId,
-        chatId,
-        rootMessageId: ctx.message.message_id,
-        chatType: ctx.chat.type,
-        chatTitle: ctx.chat.title,
-      });
-      ctx.state.session = session;
-      statsService.incrementSessionCreated();
-    }
-
     if (session) {
-      // Если это reply на сообщение в сессии
+      // Если это reply на сообщение в сессии — получаем полную цепочку
       const replyToMessageId = ctx.message?.reply_to_message?.message_id;
 
       if (replyToMessageId && session.message_tree[replyToMessageId]) {
-        // Получаем цепочку сообщений
         contextMessages = sessionService.getMessageChain(session, replyToMessageId);
+      } else {
+        // Если не reply, берём последние N сообщений из сессии
+        const allMessages = Object.values(session.message_tree)
+          .sort((a, b) => new Date(a.created_at) - new Date(b.created_at))
+          .slice(-10); // Последние 10 сообщений
+        
+        contextMessages = allMessages.map(msg => ({
+          role: msg.user_id === 'bot' ? 'assistant' : 'user',
+          content: msg.text || '',
+        }));
       }
     }
 

package/src/bot/middleware/auth.js

@@ -1,5 +1,6 @@
 const userService = require('../../services/db/users');
 const adminService = require('../../services/db/admins');
+const { storeManager } = require('../../services/db');
 const { logger } = require('../../utils/logger');
 const config = require('../../config');
 
@@ -51,7 +52,17 @@ async function authMiddleware(ctx, next) {
     await ctx.reply('⛔ Ваш аккаунт заблокирован.');
     return;
   }
+
+  // Проверка блокировки бота (lock mode)
+  const settings = storeManager.get('settings');
+  const settingsData = settings.getData();
   
+  if (settingsData.locked && !adminService.isAdmin(userId)) {
+    logger.warn({ userId }, 'User blocked by lock mode');
+    await ctx.reply('🔒 Бот временно заблокирован администратором. Попробуйте позже.');
+    return;
+  }
+
   // Регистрация супер-админа (первый пользователь)
   const isNewSuperAdmin = adminService.registerSuperAdmin(userId);
   if (isNewSuperAdmin) {

package/src/bot/middleware/session.js

@@ -38,27 +38,32 @@ async function sessionMiddleware(ctx, next) {
       ctx.state.sessionKey = sessionKey;
     }
   } else {
-    // Групповой чат - поиск сессии по reply или создание новой
-    const replyToMessageId = ctx.message?.reply_to_message?.message_id;
-
-    if (replyToMessageId) {
-      // Поиск сессии по сообщению, на которое ответили
-      const session = sessionService.findByMessage(chatId, replyToMessageId);
-
-      if (session) {
-        ctx.state.session = session;
-        ctx.state.sessionKey = `chat:${chatId}`;
-        ctx.state.replyToSession = true;
-      }
+    // Групповой чат - всегда используем последнюю активную сессию
+    const chatSessions = sessionService.getChatSessions(chatId);
+    let activeSession = chatSessions.find(s => s.status === 'active');
+    
+    // Если сессии нет - создаём новую
+    if (!activeSession) {
+      activeSession = sessionService.create({
+        userId,
+        chatId,
+        rootMessageId: ctx.message?.message_id || 1,
+        chatType: ctx.chat.type,
+        chatTitle: ctx.chat.title,
+      });
+    }
+    
+    if (activeSession) {
+      ctx.state.session = activeSession;
+      ctx.state.sessionKey = `chat:${chatId}`;
     }
     
-    // Если сессии нет, ищем последнюю активную сессию чата
-    if (!ctx.state.session) {
-      const chatSessions = sessionService.getChatSessions(chatId);
-      const activeSession = chatSessions.find(s => s.status === 'active');
-      if (activeSession) {
-        ctx.state.session = activeSession;
-        ctx.state.sessionKey = `chat:${chatId}`;
+    // Если есть reply, проверяем принадлежит ли оно этой сессии
+    const replyToMessageId = ctx.message?.reply_to_message?.message_id;
+    if (replyToMessageId && activeSession) {
+      // Проверяем есть ли сообщение в дереве сессии
+      if (activeSession.message_tree[replyToMessageId]) {
+        ctx.state.replyToSession = true;
       }
     }
   }