eleksir/bunny-bot
0
Fork 0
Code Pull Requests Projects Activity

Initial commit

Browse Source
This commit is contained in:
eleksir
2025-06-08 04:32:16 +03:00
parent c1ad1d3925
commit 2cf4cbe344
100 changed files with 9767 additions and 462 deletions
Download Patch File Download Diff File Expand all files Collapse all files

33
internal/moon/animation.go
View File

@@ -1,16 +1,27 @@
package moon
import "github.com/NicoNex/echotron/v3"
import (
"fmt"
"github.com/NicoNex/echotron/v3"
)
func animation(msg *echotron.Update) {
Log.Infof(
"User %s %s (username = %s, id = %d) sent animation to chat %s %s (%d)",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
)
if Cfg.LogChats {
ChatLog(
msg.Message.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) sent animation to chat %s %s (%d)",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
),
)
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

9
internal/moon/casChatAPI.go
View File

@@ -41,7 +41,9 @@ func CasCheckID(id int64) (bool, error) {
// Client should not return with http status code other than 200.
if resp.StatusCode > 200 {
return result, fmt.Errorf("api.cas.chat returns non-200 http status code: %d %s", resp.StatusCode, resp.Status) //nolint: err113
return result, fmt.Errorf(
"api.cas.chat returns non-200 http status code: %d %s",
resp.StatusCode, resp.Status) //nolint: err113
}
body, err := io.ReadAll(resp.Body)
@@ -65,7 +67,8 @@ func CasCheckID(id int64) (bool, error) {
// Не удалось преобразовать map-ку в json.
if err != nil {
return result, fmt.Errorf("unable to parse back to json response from api.cas.chat parsed by hjson: %w", err)
return result, fmt.Errorf(
"unable to parse back to json response from api.cas.chat parsed by hjson: %w", err)
}
if err := json.Unmarshal(tmpjson, &notBanned); err == nil {
@@ -93,3 +96,5 @@ func CasCheckID(id int64) (bool, error) {
return result, fmt.Errorf("unable to parse response from api.cas.chat: %s", string(body)) //nolint: err113
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

107
internal/moon/chatMemberUpdated.go
View File

@@ -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,
)
}
}

49
internal/moon/chatlog.go Normal file
View File

@@ -0,0 +1,49 @@
package moon
import (
"fmt"
"os"
"path/filepath"
"strconv"
"time"
"github.com/rs/zerolog"
"github.com/rs/zerolog/log"
)
// ChatLog logs message to separate file, placed in basedir(Cfg.LogFile).
func ChatLog(chatid int64, message string) {
var (
filename string
)
if Cfg.LogFile != "" {
filename = fmt.Sprintf("%s/%s", filepath.Dir(Cfg.LogFile), strconv.FormatInt(chatid, 10))
} else {
filename = os.DevNull
}
f, err := os.OpenFile(filename, os.O_APPEND|os.O_WRONLY|os.O_CREATE, 0644)
if err != nil {
fmt.Printf("Unable to open file %s for logging: %s", filename, err)
// TODO: Should we try harder?
return
}
defer f.Close()
log.Logger = log.Output(
zerolog.ConsoleWriter{
Out: f,
NoColor: true,
TimeFormat: time.DateTime,
PartsOrder: []string{"time", "message"},
},
)
log.Info().Msg(message)
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

45
internal/moon/editedMessage.go Normal file
View File

@@ -0,0 +1,45 @@
package moon
import (
"fmt"
"github.com/NicoNex/echotron/v3"
)
// editedMessage handles editedMessage type of update.
func editedMessage(msg *echotron.Update) {
// Edited message can contain all kinds of edited attributes that telegram allows to change in message.
// Not sure is we should handle all of them.
if Cfg.LogChats {
if msg.EditedMessage.Text != "" {
ChatLog(
msg.EditedMessage.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) in %s %s (%d) edited message: %s",
msg.EditedMessage.From.FirstName,
msg.EditedMessage.From.LastName,
msg.EditedMessage.From.Username,
msg.EditedMessage.From.ID,
msg.EditedMessage.Chat.Type,
msg.EditedMessage.Chat.Title,
msg.EditedMessage.Chat.ID,
msg.EditedMessage.Text,
),
)
} else {
ChatLog(
msg.EditedMessage.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) in %s %s (%d) edited message",
msg.EditedMessage.From.FirstName,
msg.EditedMessage.From.LastName,
msg.EditedMessage.From.Username,
msg.EditedMessage.From.ID,
msg.EditedMessage.Chat.Type,
msg.EditedMessage.Chat.Title,
msg.EditedMessage.Chat.ID,
),
)
}
}
}

12
internal/moon/globals.go
View File

@@ -2,6 +2,7 @@ package moon
import (
"os"
"time"
"github.com/NicoNex/echotron/v3"
"github.com/akyoto/cache"
@@ -19,15 +20,20 @@ var (
DB = make(map[string]*pebble.DB)
// SquashedMembers struct with banned members.
SquashedMembers map[int64]*cache.Cache
SquashedMembers = cache.New(1 * time.Minute)
// NewMembers struct with new members appeared via message with new_chat_members field set.
NewMembers map[int64]*cache.Cache
NewMembers = cache.New(1 * time.Minute)
// AppearedMembers struct with new members appeared via ChatMemberUpdated event.
AppearedMembers map[int64]*cache.Cache
AppearedMembers = cache.New(60 * time.Minute)
// PendingCASMembers contains newly comed members that pass first cas check, but still waiting second cas check.
PendingCASMembers = cache.New(60 * time.Minute)
ChatList = make([]int64, 0)
chatlogs []Chatlog
)
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

2
internal/moon/leftChatMember.go
View File

@@ -19,6 +19,8 @@ func leftChatMember(msg *echotron.Update) {
msg.Message.Chat.ID,
)
// Save people that left chat to pebbledb. To preserve their state.
// Left folks can eventually return and in chat_mamber_updated event their state will be left->member.
if err := Cfg.SaveKeyValue(
"LeftMembers",
fmt.Sprintf("%d", msg.Message.Chat.ID),

41
internal/moon/newChatMembers.go
View File

@@ -1,15 +1,27 @@
package moon
import (
"fmt"
"slices"
"time"
"github.com/NicoNex/echotron/v3"
cache "github.com/akyoto/cache"
)
// newChatMembers parses NewChatMember event.
func newChatMembers(msg *echotron.Update) {
// If there is no such chatid, means that no such cache too.
Log.Debugf(
"Add %s %s (%d) to list of known chats",
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
)
ChatList = append(ChatList, msg.Message.Chat.ID)
slices.Sort(ChatList)
ChatList = slices.Compact(ChatList)
for _, user := range msg.Message.NewChatMembers {
Log.Infof(
"User %s %s (username = %s, id = %d) joined %s %s (%d)",
@@ -22,29 +34,6 @@ func newChatMembers(msg *echotron.Update) {
msg.Message.Chat.ID,
)
// If there is no such chatid, means that no such cache too.
if !slices.Contains(ChatList, msg.Message.Chat.ID) {
Log.Debugf(
"Creating caches for new %s %s (%d)",
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.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.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
)
ChatList = append(ChatList, msg.Message.Chat.ID)
}
casBanned, err := CasCheckID(user.ID)
if err != nil {
@@ -99,7 +88,9 @@ func newChatMembers(msg *echotron.Update) {
msg.Message.Chat.ID,
)
NewMembers[msg.Message.Chat.ID].Set(user.ID, time.Now().Unix(), 1*time.Minute)
key := fmt.Sprintf("%d+%d", msg.Message.Chat.ID, user.ID)
NewMembers.Set(key, time.Now().Unix(), 1*time.Minute)
PendingCASMembers.Set(key, time.Now().Unix(), 30*time.Minute)
}
}

46
internal/moon/photo.go Normal file
View File

@@ -0,0 +1,46 @@
package moon
import (
"fmt"
"github.com/NicoNex/echotron/v3"
)
func photo(msg *echotron.Update) {
if Cfg.LogChats {
if msg.Message.Text != "" {
ChatLog(
msg.Message.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) sent %d photos to %s %s (%d) with text: %s",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
len(msg.Message.Photo),
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
msg.Message.Text,
),
)
} else {
ChatLog(
msg.Message.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) sent %d photos to %s %s (%d)",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
len(msg.Message.Photo),
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
),
)
}
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

65
internal/moon/reaction.go Normal file
View File

@@ -0,0 +1,65 @@
package moon
import (
"fmt"
"github.com/NicoNex/echotron/v3"
)
func reaction(msg *echotron.Update) {
if Cfg.LogChats {
var reactionString string
for _, item := range msg.MessageReaction.NewReaction {
switch item.Type {
case "Emoji":
reactionString += item.Emoji
case "CustomEmoji":
reactionString += item.CustomEmoji
}
}
if reactionString != "" {
ChatLog(
msg.MessageReaction.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) set reaction %s to chat %s %s (%d)",
msg.MessageReaction.User.FirstName,
msg.MessageReaction.User.LastName,
msg.MessageReaction.User.Username,
msg.MessageReaction.User.ID,
reactionString,
msg.MessageReaction.Chat.Type,
msg.MessageReaction.Chat.Title,
msg.MessageReaction.Chat.ID,
),
)
} else {
for _, item := range msg.MessageReaction.OldReaction {
switch item.Type {
case "Emoji":
reactionString += item.Emoji
case "CustomEmoji":
reactionString += item.CustomEmoji
}
}
ChatLog(
msg.MessageReaction.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) removed reaction %s to chat %s %s (%d)",
msg.MessageReaction.User.FirstName,
msg.MessageReaction.User.LastName,
msg.MessageReaction.User.Username,
msg.MessageReaction.User.ID,
reactionString,
msg.MessageReaction.Chat.Type,
msg.MessageReaction.Chat.Title,
msg.MessageReaction.Chat.ID,
),
)
}
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

23
internal/moon/reactionCount.go Normal file
View File

@@ -0,0 +1,23 @@
package moon
import (
"fmt"
"github.com/NicoNex/echotron/v3"
"github.com/davecgh/go-spew/spew"
)
func reactionCount(msg *echotron.Update) {
ChatLog(
msg.MessageReaction.Chat.ID,
fmt.Sprintf(
"Reaction count event in %s %s (%d): %s",
msg.MessageReaction.Chat.Type,
msg.MessageReaction.Chat.Title,
msg.MessageReaction.Chat.ID,
spew.Sdump(msg),
),
)
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

387
internal/moon/runAndBan.go
View File

@@ -1,163 +1,160 @@
package moon
import (
"fmt"
"slices"
"strconv"
"strings"
"time"
)
func RunAndBan() {
for {
// If user have both events, he is not spammer. So wipe him from unchecked events list, just for sure.
for _, chatid := range ChatList {
Log.Infof("RunAndBan by chatid: %d", chatid)
var (
ListAppearedMembers, ListNewMembers []string
empty string
)
var ListAppearedMembers, ListNewMembers []int64
AppearedMembers.Range(
func(k, _ any) bool {
ListAppearedMembers = append(ListAppearedMembers, k.(string))
AppearedMembers[chatid].Range(
func(k, _ any) bool {
ListAppearedMembers = append(ListAppearedMembers, k.(int64))
return true
},
)
return true
},
)
NewMembers.Range(
func(k, _ any) bool {
ListNewMembers = append(ListNewMembers, k.(string))
NewMembers[chatid].Range(
func(k, _ any) bool {
ListNewMembers = append(ListNewMembers, k.(int64))
return true
},
)
return true
},
)
for _, AppearedMember := range ListAppearedMembers {
for _, NewMember := range ListNewMembers {
if AppearedMember == NewMember {
// TODO: Store also additional attributes in temporary database.
Log.Infof(
"Skip banning user %s %s (username = %s, id = %d) in %s %s (%d), because it has both events newMember and updateMember",
"",
"",
"",
NewMember,
"",
"",
chatid,
)
Log.Infof(
"Deleting user %s %s (username = %s, id = %d) in %s %s (%d), from AppearedMember DB",
"",
"",
"",
NewMember,
"",
"",
chatid,
)
AppearedMembers[chatid].Delete(AppearedMember)
Log.Infof(
"Deleting user %s %s (username = %s, id = %d) in %s %s (%d), from NewMember DB",
"",
"",
"",
NewMember,
"",
"",
chatid,
)
NewMembers[chatid].Delete(NewMember)
}
for _, AppearedMember := range ListAppearedMembers {
for _, NewMember := range ListNewMembers {
if AppearedMember == NewMember {
AppearedMembers.Delete(AppearedMember)
NewMembers.Delete(NewMember)
}
}
}
currentTime := time.Now().Unix()
// We would like to process only events that definitely older than 10 seconds!
for _, chatid := range ChatList {
var (
ListAppearedMembers, ListNewMembers []int64
)
ListAppearedMembers = ListAppearedMembers[:0]
ListNewMembers = ListNewMembers[:0]
ListAppearedMembers = slices.Clip(ListAppearedMembers)
ListNewMembers = slices.Clip(ListNewMembers)
AppearedMembers[chatid].Range(
func(k, v interface{}) bool {
ListAppearedMembers = append(ListAppearedMembers, k.(int64))
AppearedMembers.Range(
func(k, _ any) bool {
ListAppearedMembers = append(ListAppearedMembers, k.(string))
return true
},
)
return true
},
)
NewMembers[chatid].Range(
func(k, v interface{}) bool {
ListNewMembers = append(ListNewMembers, k.(int64))
NewMembers.Range(
func(k, _ any) bool {
ListNewMembers = append(ListNewMembers, k.(string))
return true
},
)
return true
},
)
for _, AppearedMember := range ListAppearedMembers {
eventTimestamp, found := AppearedMembers[chatid].Get(AppearedMember)
for _, AppearedMember := range ListAppearedMembers {
eventTimestamp, found := AppearedMembers.Get(AppearedMember)
if !found {
continue
}
eventTimeDelta := currentTime - eventTimestamp.(int64)
// Looks like it is spammer!
if eventTimeDelta > 10 {
// TODO: Store also additional attributes in temporary database.
Log.Infof(
"Ban user %s %s (username = %s, id = %d) in chat %s %s (%d) that have no newChatMember message since chatMemberUpdated for more than 10 seconds",
"",
"",
"",
AppearedMember,
"",
"",
chatid,
)
squash(
chatid,
"",
"",
AppearedMember,
"",
"",
"",
)
AppearedMembers[chatid].Delete(AppearedMember)
}
if !found {
continue
}
for _, NewMember := range ListNewMembers {
eventTimestamp, found := NewMembers[chatid].Get(NewMember)
eventTimeDelta := currentTime - eventTimestamp.(int64)
// Spammer?
if eventTimeDelta > 10 {
item := strings.Split(AppearedMember, "+")
chatid, err := strconv.ParseInt(item[0], 10, 64)
if err != nil {
Log.Errorf("Got broken key in AppearedMembers cache: %s", AppearedMember)
if !found {
continue
}
eventTimeDelta := currentTime - eventTimestamp.(int64)
userid, err := strconv.ParseInt(item[1], 10, 64)
// Totally new members should not have chatMemberUpdated events.
if eventTimeDelta > 15 {
// TODO: Store also additional attributes in temporary database.
Log.Debugf("Clean newChatMember event record older than 10 seconds for user user %s %s (username = %s, id = %d) in chat %s %s (%d)",
"",
"",
"",
NewMember,
"",
"",
chatid,
)
if err != nil {
Log.Errorf("Got broken key in AppearedMembers cache: %s", AppearedMember)
NewMembers[chatid].Delete(NewMember)
continue
}
Log.Infof(
"Ban user %s %s (username = %s, id = %d) in chat %s %s (%d) that have no newChatMember message since chatMemberUpdated for more than 10 seconds",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
squash(
chatid,
empty,
empty,
userid,
empty,
empty,
empty,
)
AppearedMembers.Delete(AppearedMember)
}
}
for _, NewMember := range ListNewMembers {
eventTimestamp, found := NewMembers.Get(NewMember)
if !found {
continue
}
eventTimeDelta := currentTime - eventTimestamp.(int64)
// Spammer?
if eventTimeDelta > 10 {
item := strings.Split(NewMember, "+")
chatid, err := strconv.ParseInt(item[0], 10, 64)
if err != nil {
Log.Errorf("Got broken key in AppearedMembers cache: %s", NewMember)
continue
}
userid, err := strconv.ParseInt(item[1], 10, 64)
if err != nil {
Log.Errorf("Got broken key in AppearedMembers cache: %s", NewMember)
continue
}
Log.Debugf("Clean newChatMember event record older than 10 seconds for user user %s %s (username = %s, id = %d) in chat %s %s (%d)",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
NewMembers.Delete(NewMember)
}
}
@@ -165,4 +162,142 @@ func RunAndBan() {
}
}
func PeriodicCASCheck() {
for {
var (
ListPendingMembers []string
empty string
)
PendingCASMembers.Range(
func(k, _ any) bool {
ListPendingMembers = append(ListPendingMembers, k.(string))
return true
},
)
currentTime := time.Now().Unix()
for _, PendingMember := range ListPendingMembers {
eventTimestamp, found := PendingCASMembers.Get(PendingMember)
if !found {
continue
}
eventTimeDelta := currentTime - eventTimestamp.(int64)
// It's time to check for second time this id if it was banned in CAS.
if eventTimeDelta > 300 {
item := strings.Split(PendingMember, "+")
chatid, err := strconv.ParseInt(item[0], 10, 64)
if err != nil {
Log.Errorf("Got broken key in PendingMembers cache: %s", PendingMember)
continue
}
userid, err := strconv.ParseInt(item[1], 10, 64)
if err != nil {
Log.Errorf("Got broken key in PendingMembers cache: %s", PendingMember)
continue
}
value := Cfg.GetValue(
"BannedMembers",
fmt.Sprintf("%d", chatid),
fmt.Sprintf("%d", userid),
)
if value != "" {
Log.Debugf("User %s %s (username = %s, id = %d) in %s %s (%d) is already banned, skip cas check.",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
Log.Debugf(
"Remove user %s %s (username = %s, id = %d) in %s %s (%d) from PendingCASMembers",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
key := fmt.Sprintf("%d+%d", chatid, userid)
PendingCASMembers.Delete(key)
return
}
Log.Debugf("Checking user %s %s (username = %s, id = %d) in %s %s (%d) for second time in CAS",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
banned, err := CasCheckID(userid)
if err != nil {
Log.Error(err)
}
if banned {
Log.Infof(
"Ban user %s %s (username = %s, id = %d) in %s %s (%d) by CAS blacklist",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
squash(
userid,
empty,
empty,
userid,
empty,
empty,
empty,
)
}
Log.Debugf(
"Remove user %s %s (username = %s, id = %d) in %s %s (%d) from PendingCASMembers",
empty,
empty,
empty,
userid,
empty,
empty,
chatid,
)
PendingCASMembers.Delete(PendingMember)
}
}
time.Sleep(10 * time.Minute)
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

63
internal/moon/squash.go
View File

@@ -1,11 +1,11 @@
package moon
import (
"fmt"
"slices"
"time"
"github.com/NicoNex/echotron/v3"
"github.com/akyoto/cache"
)
// TODO: refactor to return bool, error
@@ -16,18 +16,13 @@ func squash(chatid int64, chattype, chattitle string, userid int64, firstname, l
// If there is no such chatid, means that no such cache too.
if len(ChatList) == 0 || !slices.Contains(ChatList, chatid) {
Log.Debugf("Creating caches for new %s %s (%d)", chattype, chattitle, chatid)
NewMembers[chatid] = cache.New(1 * time.Minute)
AppearedMembers[chatid] = cache.New(1 * time.Minute)
SquashedMembers[chatid] = cache.New(5 * time.Minute)
Log.Debugf("Add %s %s (%d) to list of known chats", chattype, chattitle, chatid)
ChatList = append(ChatList, chatid)
}
if found, exist := SquashedMembers[chatid].Get(userid); exist && found.(bool) {
key := fmt.Sprintf("%d+%d", chatid, userid)
if _, exist := SquashedMembers.Get(key); exist {
Log.Infof(
"Skip banning user %s %s (username = %s, id = %d) in %s %s (%d), already banned",
firstname,
@@ -85,9 +80,55 @@ func squash(chatid int64, chattype, chattitle string, userid int64, firstname, l
chatid,
)
SquashedMembers[chatid].Set(userid, true, 30*time.Second)
NewMembers[chatid].Delete(userid)
AppearedMembers[chatid].Delete(userid)
key := fmt.Sprintf("%d+%d", chatid, userid)
Log.Debugf(
"Put user %s %s (username = %s, id = %d) in %s %s (%d) into SquashedMembers to debounce ban.",
firstname,
lastname,
username,
userid,
chattype,
chattitle,
chatid,
)
SquashedMembers.Set(key, true, 30*time.Second)
Log.Debugf(
"Remove user %s %s (username = %s, id = %d) in %s %s (%d) from NewMembers.",
firstname,
lastname,
username,
userid,
chattype,
chattitle,
chatid,
)
NewMembers.Delete(key)
Log.Debugf(
"Remove user %s %s (username = %s, id = %d) in %s %s (%d) from AppearedMembers.",
firstname,
lastname,
username,
userid,
chattype,
chattitle,
chatid,
)
AppearedMembers.Delete(key)
if err := Cfg.SaveKeyValue(
"BannedMembers",
fmt.Sprintf("%d", chatid),
fmt.Sprintf("%d", userid),
fmt.Sprintf("%d", time.Now().Unix()),
); err != nil {
Log.Error(err)
} else {
Log.Infof("Info about banned userid %d saved to BannedMembers db", userid)
}
}
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

43
internal/moon/telegram.go
View File

@@ -37,42 +37,39 @@ func Telegram(cfg MyConfig) {
// telegramMsgParser parses events generated by echotron.
func telegramMsgParser(msg *echotron.Update) {
if msg.Message != nil && msg.Message.Text != "" {
switch {
case msg.Message != nil && msg.Message.Text != "":
textMessage(msg)
return
}
case msg.EditedMessage != nil:
editedMessage(msg)
// ChatMember updated event. It goes to chatMemberUpdated.go.
if msg.ChatMember != nil {
if msg.ChatMember.NewChatMember.User != nil {
newChatMember(msg)
}
return
}
case msg.ChatMember != nil:
chatMemberUpdated(msg)
// New members entered to chat.
if msg.Message != nil && msg.Message.NewChatMembers != nil {
case msg.Message != nil && msg.Message.NewChatMembers != nil:
newChatMembers(msg)
return
}
// Chat member leaves chat.
if msg.Message != nil && msg.Message.LeftChatMember != nil {
case msg.Message != nil && msg.Message.LeftChatMember != nil:
leftChatMember(msg)
return
}
if msg.Message != nil && msg.Message.Animation != nil {
case msg.Message != nil && msg.Message.Animation != nil:
animation(msg)
return
}
case msg.MessageReaction != nil:
reaction(msg)
Log.Debug(spew.Sdump(msg))
case msg.MessageReactionCount != nil:
reactionCount(msg)
case msg.Message != nil && msg.Message.Photo != nil:
photo(msg)
default:
Log.Debug(spew.Sdump(msg))
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

88
internal/moon/textMessage.go
View File

@@ -2,61 +2,53 @@ package moon
import (
"fmt"
"time"
"github.com/NicoNex/echotron/v3"
)
// textMessage handles text message events.
func textMessage(msg *echotron.Update) {
if msg.Message.ReplyToMessage == nil {
// Not sure if we must handle all kinds of these events.
if Cfg.LogChats {
switch {
// Just log it, it seems later here should be ban/mute command set.
Log.Infof(
"User %s %s (username = %s, id = %d) in %s %s (%d) says: %s",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
msg.Message.Text,
)
} else {
// Just log it.
Log.Infof(
"User %s %s (username = %s, id = %d) in %s %s (%d) replies to message %d by %s %s (username = %s, id = %d): %s",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
msg.Message.ReplyToMessage.ID,
msg.Message.ReplyToMessage.From.FirstName,
msg.Message.ReplyToMessage.From.LastName,
msg.Message.ReplyToMessage.From.Username,
msg.Message.ReplyToMessage.From.ID,
msg.Message.Text,
)
}
case msg.Message != nil && msg.Message.ReplyToMessage == nil:
ChatLog(
msg.Message.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) in %s %s (%d) says: %s",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
msg.Message.Text,
),
)
case msg.Message != nil && msg.Message.ReplyToMessage != nil:
ChatLog(
msg.Message.Chat.ID,
fmt.Sprintf(
"User %s %s (username = %s, id = %d) in %s %s (%d) replies to message %d by %s %s (username = %s, id = %d): %s",
msg.Message.From.FirstName,
msg.Message.From.LastName,
msg.Message.From.Username,
msg.Message.From.ID,
msg.Message.Chat.Type,
msg.Message.Chat.Title,
msg.Message.Chat.ID,
msg.Message.ReplyToMessage.ID,
msg.Message.ReplyToMessage.From.FirstName,
msg.Message.ReplyToMessage.From.LastName,
msg.Message.ReplyToMessage.From.Username,
msg.Message.ReplyToMessage.From.ID,
msg.Message.Text,
),
)
// Update database of new members :)
// Try to get timestamp when this member joined to chat
if joinTimestampString := Cfg.GetValue(
"NewMembers",
fmt.Sprintf("%d", msg.Message.Chat.ID),
fmt.Sprintf("%d", msg.Message.From.ID),
); joinTimestampString == "" {
if err := Cfg.SaveKeyValue(
"NewMembers",
fmt.Sprintf("%d", msg.Message.Chat.ID),
fmt.Sprintf("%d", msg.Message.From.ID),
fmt.Sprintf("%d", time.Now().Unix()),
); err != nil {
Log.Error(err)
} else {
Log.Infof("Info about new userid %d saved to new users db", msg.Message.From.ID)
}
}
}

7
internal/moon/types.go
View File

@@ -1,6 +1,7 @@
package moon
import (
"os"
"time"
)
@@ -10,6 +11,7 @@ type MyConfig struct {
Token string `json:"token,omitempty"`
LogLevel string `json:"loglevel,omitempty"`
LogFile string `json:"logfile,omitempty"`
LogChats bool `json:"logchats,omitempty"`
CSign string `json:"csign,omitempty"`
DataDir string `json:"datadir,omitempty"`
}
@@ -30,4 +32,9 @@ type CasTrue struct {
} `json:"result"`
}
type Chatlog struct {
LogFilePath string
File *os.File
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */