Add new user handler for new member events and update message handlers to support replies. Introduce configuration for new member sampling and enhance message templates accordingly.

Author: GPSxtreme

.changeset/real-jobs-fold.md

@@ -0,0 +1,5 @@
+---
+"@iqai/mcp-telegram": patch
+---
+
+added new user handler & updated message handlers to user reply_to on messages

src/config.ts

@@ -84,6 +84,7 @@ const envSchema = z.object({
 	SAMPLING_ENABLE_LOCATION: z.coerce.boolean().default(false),
 	SAMPLING_ENABLE_CONTACT: z.coerce.boolean().default(false),
 	SAMPLING_ENABLE_POLL: z.coerce.boolean().default(false),
+	SAMPLING_ENABLE_NEW_MEMBER: z.coerce.boolean().default(true),
 
 	// Response behavior
 	SAMPLING_MAX_TOKENS: z.coerce.number().default(1000),
@@ -133,6 +134,7 @@ export const samplingConfig = {
 		[MessageType.LOCATION]: env.SAMPLING_ENABLE_LOCATION,
 		[MessageType.CONTACT]: env.SAMPLING_ENABLE_CONTACT,
 		[MessageType.POLL]: env.SAMPLING_ENABLE_POLL,
+		[MessageType.NEW_MEMBER]: env.SAMPLING_ENABLE_NEW_MEMBER,
 	},
 
 	// Response behavior
@@ -249,5 +251,15 @@ message_id: {messageId}
 topic_id: {topicId}
 message_type: {messageType}
 content: {content}`,
+
+		[TemplateType.NEW_MEMBER]: `NEW MEMBER JOINED:
+chat_id: {chatId}
+message_id: {messageId}
+number_of_new_members: {numberOfNewMembers}
+new_member_id: {newMemberId}
+new_member_username: {newMemberUsername}
+new_member_first_name: {newMemberFirstName}
+new_member_last_name: {newMemberLastName}
+added_by_user_id: {addedByUserId}`,
 	},
 } as const;

src/sampling/handler.ts

@@ -18,6 +18,7 @@ import {
 	handleLocationMessage,
 	handleContactMessage,
 	handlePollMessage,
+	handleNewMemberMessage,
 } from "./message-handlers.js";
 
 export class SamplingHandler {
@@ -127,6 +128,22 @@ export class SamplingHandler {
 		this.bot.on(message("poll"), async (ctx: Context) => {
 			await this.processMessage(ctx, MessageType.POLL, handlePollMessage);
 		});
+
+		// New members joined
+		this.bot.on("new_chat_members", async (ctx: Context) => {
+			// Directly handle without text validation
+			if (!this.validator.shouldProcessMessage(ctx, MessageType.NEW_MEMBER))
+				return;
+			if (
+				!this.rateLimiter.checkRateLimit(ctx.from?.id || 0, ctx.chat?.id || 0)
+			)
+				return;
+
+			const templateData = handleNewMemberMessage(ctx);
+			if (!templateData) return;
+
+			await this.handleMessage(ctx, MessageType.NEW_MEMBER, templateData);
+		});
 	}
 
 	private async handleMessage(
@@ -217,6 +234,7 @@ export class SamplingHandler {
 					request.chatId,
 					responseText,
 					request.templateData.topicId,
+					request.messageId,
 				);
 			}
 
@@ -231,6 +249,7 @@ export class SamplingHandler {
 					request.chatId,
 					"Sorry, I encountered an error while processing your request. Please try again.",
 					request.templateData.topicId,
+					request.messageId,
 				);
 			}
 		}

src/sampling/message-handlers.ts

@@ -172,3 +172,31 @@ export function handlePollMessage(ctx: Context): MessageTemplateData | null {
 		messageType: MessageType.POLL,
 	};
 }
+
+export function handleNewMemberMessage(
+	ctx: Context,
+): MessageTemplateData | null {
+	const msg = ctx.message;
+	if (!msg || !("new_chat_members" in msg) || !ctx.chat) return null;
+
+	const newMembers = msg.new_chat_members;
+	if (!newMembers || newMembers.length === 0) return null;
+
+	const firstNewMember = newMembers[0];
+
+	return {
+		content: `[New member joined] ${firstNewMember.username || firstNewMember.first_name}`,
+		userId: msg.from?.id ?? firstNewMember.id,
+		chatId: ctx.chat.id,
+		isDM: false,
+		messageId: msg.message_id,
+		topicId: msg.message_thread_id,
+		messageType: MessageType.NEW_MEMBER,
+		newMemberId: firstNewMember.id,
+		newMemberUsername: firstNewMember.username,
+		newMemberFirstName: firstNewMember.first_name,
+		newMemberLastName: firstNewMember.last_name,
+		addedByUserId: msg.from?.id,
+		numberOfNewMembers: newMembers.length,
+	};
+}

src/sampling/types.ts

@@ -10,6 +10,7 @@ export enum MessageType {
 	LOCATION = "location",
 	CONTACT = "contact",
 	POLL = "poll",
+	NEW_MEMBER = "new_member",
 }
 
 export enum TemplateType {
@@ -23,6 +24,7 @@ export enum TemplateType {
 	CONTACT = "contact",
 	POLL = "poll",
 	FALLBACK = "fallback",
+	NEW_MEMBER = "new_member",
 }
 
 export interface SamplingRequest {
@@ -67,6 +69,12 @@ export interface MessageTemplateData {
 	phoneNumber?: string;
 	pollQuestion?: string;
 	pollOptions?: string;
+	newMemberId?: number;
+	newMemberUsername?: string;
+	newMemberFirstName?: string;
+	newMemberLastName?: string;
+	addedByUserId?: number;
+	numberOfNewMembers?: number;
 	[key: string]: string | number | boolean | undefined;
 }
 

src/sampling/validators.ts

@@ -66,8 +66,12 @@ export class MessageValidator {
 		// Check if it's a DM
 		const isDM = chatId === userId;
 
-		// For groups: check mention requirement
+		// For groups: check mention requirement except for system events like NEW_MEMBER
 		if (!isDM && samplingConfig.mentionOnly) {
+			// Allow NEW_MEMBER without mention
+			if ((messageType as string) === "new_member") {
+				return true;
+			}
 			if ("text" in msg && msg.text) {
 				return this.isMentioned(msg.text, msg.entities);
 			}

src/services/telegram-service.ts

@@ -31,11 +31,33 @@ export class TelegramService {
 		chatId: string | number,
 		text: string,
 		topicId?: number,
+		replyToMessageId?: number,
 	): Promise<MessageInfo> {
 		try {
-			const message = await this.bot.telegram.sendMessage(chatId, text, {
-				message_thread_id: topicId,
-			});
+			const options: {
+				message_thread_id?: number;
+				reply_parameters?: {
+					message_id: number;
+					allow_sending_without_reply?: boolean;
+				};
+			} = {};
+
+			if (typeof topicId === "number") {
+				options.message_thread_id = topicId;
+			}
+
+			if (typeof replyToMessageId === "number") {
+				options.reply_parameters = {
+					message_id: replyToMessageId,
+					allow_sending_without_reply: true,
+				};
+			}
+
+			const message = await this.bot.telegram.sendMessage(
+				chatId,
+				text,
+				options,
+			);
 
 			return {
 				messageId: message.message_id,