Initial commit

2025-06-08 04:32:16 +03:00
parent c1ad1d3925
commit dd0296d323
100 changed files with 9773 additions and 462 deletions
--- a/internal/moon/chatMemberUpdated.go
+++ b/internal/moon/chatMemberUpdated.go
@@ -6,12 +6,11 @@ import (
 	"time"

 	"github.com/NicoNex/echotron/v3"
-	"github.com/akyoto/cache"
 )

 // newChatMember parses one of types ChatMemberUpdated event.
 // https://core.telegram.org/bots/api#chatmemberupdated
-func newChatMember(msg *echotron.Update) {
+func chatMemberUpdated(msg *echotron.Update) {
 	Log.Infof(
 		"User %s %s (username = %s, id = %d) in chat %s %s (%d) now %s was %s",
 		msg.ChatMember.NewChatMember.User.FirstName,
@@ -25,31 +24,20 @@ func newChatMember(msg *echotron.Update) {
 		msg.ChatMember.OldChatMember.Status,
 	)

-	// If there is no such chatid, means that no such cache too.
-	if len(ChatList) == 0 || !slices.Contains(ChatList, msg.Message.Chat.ID) {
-		Log.Debugf(
-			"Creating caches for new %s %s (%d)",
-			msg.ChatMember.Chat.Type,
-			msg.ChatMember.Chat.Title,
-			msg.ChatMember.Chat.ID,
-		)
+	Log.Debugf(
+		"Add %s %s (%d) to list of known chats",
+		msg.ChatMember.Chat.Type,
+		msg.ChatMember.Chat.Title,
+		msg.ChatMember.Chat.ID,
+	)

-		NewMembers[msg.Message.Chat.ID] = cache.New(1 * time.Minute)
-		AppearedMembers[msg.Message.Chat.ID] = cache.New(1 * time.Minute)
-		SquashedMembers[msg.Message.Chat.ID] = cache.New(60 * time.Minute)
-
-		Log.Debugf(
-			"Add %s %s (%d) to list of known chats",
-			msg.ChatMember.Chat.Type,
-			msg.ChatMember.Chat.Title,
-			msg.ChatMember.Chat.ID,
-		)
-
-		ChatList = append(ChatList, msg.Message.Chat.ID)
-	}
+	ChatList = append(ChatList, msg.ChatMember.Chat.ID)
+	slices.Sort(ChatList)
+	ChatList = slices.Compact(ChatList)

 	// Check some other attributes.
-	if msg.ChatMember.OldChatMember.Status == "left" && msg.ChatMember.NewChatMember.Status == "member" {
+	switch {
+	case msg.ChatMember.OldChatMember.Status == "left" && msg.ChatMember.NewChatMember.Status == "member":
 		Log.Infof(
 			"Getting info about user %s %s (username = %s, id = %d) in chat %s %s (%d) from LeftMembers DB",
 			msg.ChatMember.NewChatMember.User.FirstName,
@@ -70,7 +58,7 @@ func newChatMember(msg *echotron.Update) {
 		// Yes, this member definitely left chat at some point, we have no question to him/her.
 		if leftTimestampString != "" {
 			Log.Infof(
-				"User %s %s (username = %s, id = %d) in chat %s %s (%d) found in from LeftMembers DB ",
+				"User %s %s (username = %s, id = %d) in chat %s %s (%d) found in from LeftMembers DB",
 				msg.ChatMember.NewChatMember.User.FirstName,
 				msg.ChatMember.NewChatMember.User.LastName,
 				msg.ChatMember.NewChatMember.User.Username,
@@ -84,7 +72,7 @@ func newChatMember(msg *echotron.Update) {
 		}

 		Log.Infof(
-			"User %s %s (username = %s, id = %d) in chat %s %s (%d) not found in LeftMembers DB user %s %s (username = %s, id = %d) in chat %s %s (%d)",
+			"User %s %s (username = %s, id = %d) in chat %s %s (%d) not found in LeftMembers DB",
 			msg.ChatMember.NewChatMember.User.FirstName,
 			msg.ChatMember.NewChatMember.User.LastName,
 			msg.ChatMember.NewChatMember.User.Username,
@@ -150,7 +138,72 @@ func newChatMember(msg *echotron.Update) {
 			msg.ChatMember.Chat.ID,
 		)

-		AppearedMembers[msg.ChatMember.Chat.ID].Set(msg.ChatMember.NewChatMember.User.ID, time.Now().Unix(), 1*time.Minute)
+		key := fmt.Sprintf("%d+%d", msg.ChatMember.Chat.ID, msg.ChatMember.NewChatMember.User.ID)
+		AppearedMembers.Set(key, time.Now().Unix(), 1*time.Minute)
+		PendingCASMembers.Set(key, time.Now().Unix(), 30*time.Minute)
+
+	case msg.ChatMember.OldChatMember.Status == "member" && msg.ChatMember.NewChatMember.Status == "kicked":
+		Log.Infof(
+			"User %s %s (username = %s, id = %d) in chat %s %s (%d) was banned",
+			msg.ChatMember.NewChatMember.User.FirstName,
+			msg.ChatMember.NewChatMember.User.LastName,
+			msg.ChatMember.NewChatMember.User.Username,
+			msg.ChatMember.NewChatMember.User.ID,
+			msg.ChatMember.Chat.Type,
+			msg.ChatMember.Chat.Title,
+			msg.ChatMember.Chat.ID,
+		)
+
+		key := fmt.Sprintf("%d+%d", msg.ChatMember.Chat.ID, msg.ChatMember.NewChatMember.User.ID)
+
+		Log.Debugf(
+			"Put user %s %s (username = %s, id = %d) in %s %s (%d) into SquashedMembers to debounce ban.",
+			msg.ChatMember.NewChatMember.User.FirstName,
+			msg.ChatMember.NewChatMember.User.LastName,
+			msg.ChatMember.NewChatMember.User.Username,
+			msg.ChatMember.NewChatMember.User.ID,
+			msg.ChatMember.Chat.Type,
+			msg.ChatMember.Chat.Title,
+			msg.ChatMember.Chat.ID,
+		)
+
+		SquashedMembers.Set(key, true, 30*time.Second)
+
+		Log.Debugf(
+			"Storing info about user %s %s (username = %s, id = %d) in chat %s %s (%d) to BannedMembers DB",
+			msg.ChatMember.NewChatMember.User.FirstName,
+			msg.ChatMember.NewChatMember.User.LastName,
+			msg.ChatMember.NewChatMember.User.Username,
+			msg.ChatMember.NewChatMember.User.ID,
+			msg.ChatMember.Chat.Type,
+			msg.ChatMember.Chat.Title,
+			msg.ChatMember.Chat.ID,
+		)
+
+		if err := Cfg.SaveKeyValue(
+			"BannedMembers",
+			fmt.Sprintf("%d", msg.ChatMember.Chat.ID),
+			fmt.Sprintf("%d", msg.ChatMember.NewChatMember.User.ID),
+			fmt.Sprintf("%d", time.Now().Unix()),
+		); err != nil {
+			Log.Error(err)
+		} else {
+			Log.Infof("Info about banned userid %d saved to BannedMembers DB", msg.ChatMember.Chat.ID)
+		}
+
+	default:
+		Log.Infof(
+			"User %s %s (username = %s, id = %d) in chat %s %s (%d) now %s was %s",
+			msg.ChatMember.NewChatMember.User.FirstName,
+			msg.ChatMember.NewChatMember.User.LastName,
+			msg.ChatMember.NewChatMember.User.Username,
+			msg.ChatMember.NewChatMember.User.ID,
+			msg.ChatMember.Chat.Type,
+			msg.ChatMember.Chat.Title,
+			msg.ChatMember.Chat.ID,
+			msg.ChatMember.NewChatMember.Status,
+			msg.ChatMember.OldChatMember.Status,
+		)
 	}
 }