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 2579cd7c94
16 changed files with 306 additions and 69 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
internal/moon/animation.go
View File

@@ -3,14 +3,18 @@ package moon
import "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 {
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,
)
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

2
internal/moon/casChatAPI.go
View File

@@ -93,3 +93,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: */

6
internal/moon/chatMemberUpdated.go
View File

@@ -34,9 +34,9 @@ func newChatMember(msg *echotron.Update) {
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)
NewMembers[msg.ChatMember.Chat.ID] = cache.New(1 * time.Minute)
AppearedMembers[msg.ChatMember.Chat.ID] = cache.New(1 * time.Minute)
SquashedMembers[msg.ChatMember.Chat.ID] = cache.New(60 * time.Minute)
Log.Debugf(
"Add %s %s (%d) to list of known chats",

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

@@ -0,0 +1,45 @@
package moon
import (
"fmt"
"log"
"os"
)
func ChatLog(file, text string) {
var (
isOpened = false
)
for _, l := range chatlogs {
if l.Name == file {
isOpened = true
break
}
}
if !isOpened {
// open log file
logFile, err := os.OpenFile(file, os.O_APPEND|os.O_RDWR|os.O_CREATE, 0644)
if err != nil {
fmt.Printf("Unable to open log file %s: %s", file, err)
}
l := Chatlog{Name: file, File: logFile}
chatlogs = append(chatlogs, l)
}
for _, l := range chatlogs {
if l.Name == file {
log.SetOutput(l.File)
// optional: log date-time, filename, and line number
log.SetFlags(log.Lshortfile | log.LstdFlags)
log.Println(text)
break
}
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

2
internal/moon/globals.go
View File

@@ -28,6 +28,8 @@ var (
AppearedMembers map[int64]*cache.Cache
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.
// TODO: Remove items on chatMemberUpdated event.
if err := Cfg.SaveKeyValue(
"LeftMembers",
fmt.Sprintf("%d", msg.Message.Chat.ID),

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

@@ -0,0 +1,36 @@
package moon
import "github.com/NicoNex/echotron/v3"
func photo(msg *echotron.Update) {
if Cfg.LogChats {
if msg.Message.Text != "" {
Log.Infof(
"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 {
Log.Infof(
"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: */

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

@@ -0,0 +1,53 @@
package moon
import "github.com/NicoNex/echotron/v3"
func reaction(msg *echotron.Update) {
if Cfg.LogChats {
var reactionString string
for _, item := range msg.MessageReaction.NewReaction {
if item.Type == "Emoji" {
reactionString += item.Emoji
} else if item.Type == "CustomEmoji" {
reactionString += item.CustomEmoji
}
}
if reactionString != "" {
Log.Infof(
"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 {
if item.Type == "Emoji" {
reactionString += item.Emoji
} else if item.Type == "CustomEmoji" {
reactionString += item.CustomEmoji
}
}
Log.Infof(
"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: */

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

@@ -0,0 +1,18 @@
package moon
import (
"github.com/NicoNex/echotron/v3"
"github.com/davecgh/go-spew/spew"
)
func reactionCount(msg *echotron.Update) {
Log.Infof(
"Reaction count event in %s %s (%d): %",
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: */

2
internal/moon/squash.go
View File

@@ -91,3 +91,5 @@ func squash(chatid int64, chattype, chattitle string, userid int64, firstname, l
}
}
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */

18
internal/moon/telegram.go
View File

@@ -72,6 +72,24 @@ func telegramMsgParser(msg *echotron.Update) {
return
}
if msg.MessageReaction != nil {
reaction(msg)
return
}
if msg.MessageReactionCount != nil {
reactionCount(msg)
return
}
if msg.Message != nil && msg.Message.Photo != nil {
photo(msg)
return
}
Log.Debug(spew.Sdump(msg))
}

82
internal/moon/textMessage.go
View File

@@ -1,62 +1,42 @@
package moon
import (
"fmt"
"time"
"github.com/NicoNex/echotron/v3"
)
func textMessage(msg *echotron.Update) {
if msg.Message.ReplyToMessage == nil {
// 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,
)
}
// 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)
if Cfg.LogChats {
if msg.Message.ReplyToMessage == nil {
// 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 {
Log.Infof("Info about new userid %d saved to new users db", msg.Message.From.ID)
// 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,
)
}
}
}

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 {
Name string
File *os.File
}
/* vim: set ft=go noet ai ts=4 sw=4 sts=4: */