Updated TG group add limitation feature.

Author: azep-ninja

packages/client-telegram/src/messageManager.ts

@@ -166,12 +166,12 @@ export class MessageManager {
         this.bot = bot;
         this.runtime = runtime;
 
-        this.initializeTeamMemberUsernames().catch(error =>
+        this._initializeTeamMemberUsernames().catch(error =>
             elizaLogger.error("Error initializing team member usernames:", error)
         );
     }
 
-    private async initializeTeamMemberUsernames(): Promise<void> {
+    private async _initializeTeamMemberUsernames(): Promise<void> {
         if (!this.runtime.character.clientConfig?.telegram?.isPartOfTeam) return;
 
         const teamAgentIds = this.runtime.character.clientConfig.telegram.teamAgentIds || [];
@@ -189,7 +189,7 @@ export class MessageManager {
         }
     }
 
-    private getTeamMemberUsername(id: string): string | undefined {
+    private _getTeamMemberUsername(id: string): string | undefined {
         return this.teamMemberUsernames.get(id);
     }
 
@@ -749,7 +749,7 @@ export class MessageManager {
             );
 
             const mentionedTeamMember = otherTeamMembers.find(id => {
-                const username = this.getTeamMemberUsername(id);
+                const username = this._getTeamMemberUsername(id);
                 return username && messageText?.includes(`@${username}`);
             });
 

packages/client-telegram/src/telegramClient.ts

@@ -1,4 +1,5 @@
 import { Context, Telegraf } from "telegraf";
+import { message } from 'telegraf/filters';
 import { IAgentRuntime, elizaLogger } from "@ai16z/eliza";
 import { MessageManager } from "./messageManager.ts";
 import { getOrCreateRecommenderInBe } from "./getOrCreateRecommenderInBe.ts";
@@ -47,11 +48,56 @@ export class TelegramClient {
         this.messageManager.bot = this.bot;
     }
 
+    private async isGroupAuthorized(ctx: Context): Promise<boolean> {
+        const config = this.runtime.character.clientConfig?.telegram;
+        if (ctx.from?.id === ctx.botInfo?.id) {
+            return false;
+        }
+
+        if (!config?.shouldOnlyJoinInAllowedGroups) {
+            return true;
+        }
+
+        const allowedGroups = config.allowedGroupIds || [];
+        const currentGroupId = ctx.chat.id.toString();
+
+        if (!allowedGroups.includes(currentGroupId)) {
+            elizaLogger.info(`Unauthorized group detected: ${currentGroupId}`);
+            try {
+                await ctx.reply("Not authorized. Leaving.");
+                await ctx.leaveChat();
+            } catch (error) {
+                elizaLogger.error(`Error leaving unauthorized group ${currentGroupId}:`, error);
+            }
+            return false;
+        }
+
+        return true;
+    }
+
     private setupMessageHandlers(): void {
         elizaLogger.log("Setting up message handler...");
 
+        this.bot.on(message('new_chat_members'), async (ctx) => {
+            try {
+                const newMembers = ctx.message.new_chat_members;
+                const isBotAdded = newMembers.some(member => member.id === ctx.botInfo.id);
+
+                if (isBotAdded && !(await this.isGroupAuthorized(ctx))) {
+                    return;
+                }
+            } catch (error) {
+                elizaLogger.error("Error handling new chat members:", error);
+            }
+        });
+
         this.bot.on("message", async (ctx) => {
             try {
+                // Check group authorization first
+                if (!(await this.isGroupAuthorized(ctx))) {
+                    return;
+                }
+
                 if (this.tgTrader) {
                     const userId = ctx.from?.id.toString();
                     const username =
@@ -76,12 +122,18 @@ export class TelegramClient {
                         );
                     }
                 }
+
                 await this.messageManager.handleMessage(ctx);
             } catch (error) {
                 elizaLogger.error("❌ Error handling message:", error);
-                await ctx.reply(
-                    "An error occurred while processing your message."
-                );
+                // Don't try to reply if we've left the group or been kicked
+                if (error?.response?.error_code !== 403) {
+                    try {
+                        await ctx.reply("An error occurred while processing your message.");
+                    } catch (replyError) {
+                        elizaLogger.error("Failed to send error message:", replyError);
+                    }
+                }
             }
         });
 

packages/core/src/types.ts

@@ -724,6 +724,8 @@ export type Character = {
             shouldIgnoreBotMessages?: boolean;
             shouldIgnoreDirectMessages?: boolean;
             shouldRespondOnlyToMentions?: boolean;
+            shouldOnlyJoinInAllowedGroups?: boolean;
+            allowedGroupIds?: string[];
             messageSimilarityThreshold?: number;
             isPartOfTeam?: boolean;
             teamAgentIds?: string[];