From 46de48a28b47320f219817ffcbe527f22891e157 Mon Sep 17 00:00:00 2001
From: Omar Polo <op@omarpolo.com>
Date: Tue, 30 Mar 2021 09:57:15 +0000
Subject: [PATCH] implement my_chat_member and chat_member

---
 admin.go   | 55 ++++++++++++++++++++++++++++++++++++++++++++++++++++++
 bot.go     | 28 +++++++++++++++++++++++++++
 telebot.go | 10 ++++++++++
 3 files changed, 93 insertions(+)

diff --git a/admin.go b/admin.go
index 7d0cfd7..64a4df1 100644
--- a/admin.go
+++ b/admin.go
@@ -6,6 +6,61 @@ import (
 	"time"
 )
 
+// ChatInviteLink object represents an invite for a chat.
+type ChatInviteLink struct {
+	// The invite link.
+	InviteLink string `json:"invite_link"`
+
+	// The creator of the link.
+	Creator *User `json:"creator"`
+
+	// If the link is primary.
+	IsPrimary bool `json:"is_primary"`
+
+	// If the link is revoked.
+	IsRevoked bool `json:"is_revoked"`
+
+	// (Optional) Point in time when the link will expire, use
+	// ChatInviteLink.ExpireTime() to get time.Time
+	Unixtime int64 `json:"expire_date,omitempty"`
+
+	// (Optional) Maximum number of users that can be members of
+	// the chat simultaneously.
+	MemberLimit int `json:"member_limit,omitempty"`
+}
+
+// ExpireTime returns the moment of the link expiration in local time.
+func (c *ChatInviteLink) ExpireTime() time.Time {
+	return time.Unix(c.Unixtime, 0)
+}
+
+// ChatMemberUpdated object represents changes in the status of a chat member.
+type ChatMemberUpdated struct {
+	// Chat where the user belongs to.
+	Chat Chat `json:"chat"`
+
+	// From which user the action was triggered.
+	From User `json:"user"`
+
+	// Unixtime, use ChatMemberUpdated.Time() to get time.Time
+	Unixtime int64 `json:"date"`
+
+	// Previous information about the chat member.
+	OldChatMember *ChatMember `json:"old_chat_member"`
+
+	// New information about the chat member.
+	NewChatMember *ChatMember `json:"new_chat_member"`
+
+	// (Optional) ChatInviteLink which was used by the user to
+	// join the chat; for joining by invite link events only.
+	ChatInviteLink *ChatInviteLink `json:"chat_invite_link"`
+}
+
+// Time returns the moment of the change in local time.
+func (c *ChatMemberUpdated) Time() time.Time {
+	return time.Unix(c.Unixtime, 0)
+}
+
 // Rights is a list of privileges available to chat members.
 type Rights struct {
 	CanBeEdited         bool `json:"can_be_edited"`
diff --git a/bot.go b/bot.go
index 08d4d5a..ddbf350 100644
--- a/bot.go
+++ b/bot.go
@@ -129,6 +129,8 @@ type Update struct {
 	PreCheckoutQuery   *PreCheckoutQuery   `json:"pre_checkout_query,omitempty"`
 	Poll               *Poll               `json:"poll,omitempty"`
 	PollAnswer         *PollAnswer         `json:"poll_answer,omitempty"`
+	MyChatMember       *ChatMemberUpdated  `json:"my_chat_member,omitempty"`
+	ChatMember         *ChatMemberUpdated  `json:"chat_member,omitempty"`
 }
 
 // Command represents a bot command.
@@ -491,6 +493,32 @@ func (b *Bot) ProcessUpdate(upd Update) {
 		return
 	}
 
+	if upd.MyChatMember != nil {
+		if handler, ok := b.handlers[OnMyChatMember]; ok {
+			handler, ok := handler.(func(*ChatMemberUpdated))
+			if !ok {
+				panic("telebot: my chat member handler is bad")
+			}
+
+			b.runHandler(func() { handler(upd.MyChatMember) })
+		}
+
+		return
+	}
+
+	if upd.ChatMember != nil {
+		if handler, ok := b.handlers[OnChatMember]; ok {
+			handler, ok := handler.(func(*ChatMemberUpdated))
+			if !ok {
+				panic("telebot: my chat member handler is bad")
+			}
+
+			b.runHandler(func() { handler(upd.MyChatMember) })
+		}
+
+		return
+	}
+
 }
 
 func (b *Bot) handle(end string, m *Message) bool {
diff --git a/telebot.go b/telebot.go
index a9003ad..5cb25fb 100644
--- a/telebot.go
+++ b/telebot.go
@@ -120,6 +120,16 @@ const (
 	// Handler: func(*PollAnswer)
 	OnPollAnswer = "\apoll_answer"
 
+	// Will fire on MyChatMember
+	//
+	// Handler: func(*ChatMemberUpdated)
+	OnMyChatMember = "\amy_chat_member"
+
+	// Will fire on ChatMember
+	//
+	// Handler: func(*ChatMemberUpdated)
+	OnChatMember = "\achat_member"
+
 	// Will fire on VoiceChatStarted
 	//
 	// Handler: func(*Message)