all

src/bots/botManager.ts

@@ -8,6 +8,7 @@ import { createLogger } from "../utils/logger";
 import { messageManager } from "./utils/messageManager";
 import { handleLanguageSelection, handleLanguageChange } from "./handlers/languageHandlers";
 import { getBotIdFromToken, saveBotTokenMapping, isValidBotToken } from "../utils/botUtils";
+import { groupCheckMiddleware } from "./middleware/groupCheckMiddleware";
 // import { balanceUpdateService } from "./services/BalanceUpdateService";
 
 const logger = createLogger('BotManager');
@@ -42,6 +43,10 @@ export interface BotData {
   crypto_wallet_address: string;
   admin_contact: string;
  
+  // Group Join Settings
+  join_group_required: boolean;
+  group_channel_username: string;
+ 
   settings: Record<string, any>;
   state: Record<string, any>;
   suffix_email: string;
@@ -86,6 +91,9 @@ export const initializeBot = async (botToken: string, botData?: BotData) => {
       return next();
     });
     
+    // Add group check middleware
+    bot.use(groupCheckMiddleware);
+    
     // Setup command handlers
     setupCommandHandlers(bot);
     

src/bots/db.sql

@@ -131,6 +131,10 @@ CREATE TABLE IF NOT EXISTS public.bots (
   crypto_wallet_address VARCHAR(255),
   admin_contact VARCHAR(255),
   
+  -- Group Join Settings
+  join_group_required BOOLEAN DEFAULT false,
+  group_channel_username VARCHAR(255),
+  
   settings JSONB DEFAULT '{}',
   state JSONB DEFAULT '{}',
   suffix_email VARCHAR(255) DEFAULT 'saerosms.com',

src/bots/middleware/groupCheckMiddleware.ts

@@ -0,0 +1,52 @@
+import { BotContext } from '../types/botTypes';
+import { createLogger } from '../../utils/logger';
+import { messageManager } from '../utils/messageManager';
+
+const logger = createLogger('GroupCheckMiddleware');
+
+export const groupCheckMiddleware = async (ctx: BotContext, next: () => Promise<void>) => {
+  try {
+    // Skip check for non-message updates
+    if (!ctx.message || !ctx.from) {
+      return next();
+    }
+
+    const botData = ctx.botData;
+    if (!botData?.join_group_required || !botData?.group_channel_username) {
+      return next();
+    }
+
+    // Skip check for admin commands
+    if ('text' in ctx.message && ctx.message.text?.startsWith('/')) {
+      return next();
+    }
+
+    try {
+      // Check if user is a member of the required group
+      const chatMember = await ctx.telegram.getChatMember(
+        botData.group_channel_username,
+        ctx.from.id
+      );
+
+      if (['member', 'administrator', 'creator'].includes(chatMember.status)) {
+        return next();
+      }
+
+      // User is not a member of the required group
+      const message = messageManager.getMessage('join_group_required');
+
+      await ctx.reply(message, {
+        parse_mode: 'HTML',
+        link_preview_options: { is_disabled: true }
+      });
+
+    } catch (error: any) {
+      logger.error(`Error checking group membership: ${error.message}`);
+      // If there's an error checking membership, allow the user to proceed
+      return next();
+    }
+  } catch (error: any) {
+    logger.error(`Error in group check middleware: ${error.message}`);
+    return next();
+  }
+}; 

src/bots/migrations/add_group_join_settings.sql

@@ -0,0 +1,8 @@
+-- Add group join settings columns to bots table
+ALTER TABLE public.bots
+ADD COLUMN IF NOT EXISTS join_group_required BOOLEAN NOT NULL DEFAULT false,
+ADD COLUMN IF NOT EXISTS group_channel_username VARCHAR(255);
+
+-- Add comment to explain the columns
+COMMENT ON COLUMN public.bots.join_group_required IS 'Whether users must join a group to use the bot';
+COMMENT ON COLUMN public.bots.group_channel_username IS 'The username of the required group/channel (without @ symbol)'; 

src/bots/utils/messageUtils.ts

@@ -187,4 +187,11 @@ export const getAffordableProductsMessage = (balance: number, ctx: BotContext) =
 export const getNoAffordableProductsMessage = (balance: number, ctx: BotContext) => {
   return messageManager.getMessage('no_affordable_products')
     .replace('{balance}', formatPrice(ctx, balance));
+};
+
+export const getJoinGroupRequiredMessage = (group: string) => {
+  return {
+    ar: `⚠️ يجب عليك الانضمام إلى المجموعة أولاً لاستخدام البوت\n\n📢 انضم إلى المجموعة: @${group}`,
+    en: `⚠️ You must join the group first to use the bot\n\n📢 Join the group: @${group}`
+  };
 };