This commit is contained in:
2024-11-12 02:32:06 +07:00
commit 077437d33c
18 changed files with 432 additions and 0 deletions
+6
View File
@@ -0,0 +1,6 @@
package entities
type User struct {
ChatID string `sql:"chat_id"`
UserID string `sql:"user_id"`
}
+16
View File
@@ -0,0 +1,16 @@
package messenger
//go:generate mockgen -source=$GOFILE -destination=mocks/$GOFILE -package=mocks
import "context"
type Message struct {
ChatID string
UserID string
Text string
}
type IMessenger interface {
GetMessage(ctx context.Context) (*Message, error)
SendMessage(ctx context.Context, msg *Message) error
}
@@ -0,0 +1,64 @@
package telegram
import (
"context"
"fmt"
"log"
"strconv"
"git.3crabs.ru/save_my_money/smm_tg/internal/modules/messenger"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
)
type messengerTelegram struct {
bot *tgbotapi.BotAPI
updates tgbotapi.UpdatesChannel
}
func NewMessengerTelegram(token string) (messenger.IMessenger, error) {
bot, err := tgbotapi.NewBotAPI(token)
if err != nil {
return nil, err
}
log.Printf("Authorized on account %s", bot.Self.UserName)
u := tgbotapi.NewUpdate(0)
u.Timeout = 60
updates := bot.GetUpdatesChan(u)
return &messengerTelegram{
bot: bot,
updates: updates,
}, nil
}
func (m *messengerTelegram) GetMessage(ctx context.Context) (*messenger.Message, error) {
tgMsg := <-m.updates
log.Printf("get grom tg: %+v", tgMsg)
chatID := int64(0)
if tgMsg.Message != nil && tgMsg.Message.Chat != nil {
chatID = tgMsg.Message.Chat.ID
}
if chatID == 0 {
return nil, fmt.Errorf("get chat id error %+v", tgMsg)
}
msg := &messenger.Message{
ChatID: fmt.Sprintf("%d", chatID),
UserID: tgMsg.Message.From.UserName,
Text: tgMsg.Message.Text,
}
log.Println("get:", msg)
return msg, nil
}
func (m *messengerTelegram) SendMessage(ctx context.Context, msg *messenger.Message) error {
chatID, err := strconv.ParseInt(msg.ChatID, 10, 64)
if err != nil {
return err
}
tgMsg := tgbotapi.NewMessage(chatID, msg.Text)
if _, err = m.bot.Send(tgMsg); err != nil {
return err
}
log.Println("send:", msg)
return nil
}
+35
View File
@@ -0,0 +1,35 @@
package bot_ping
import (
"context"
"strings"
"git.3crabs.ru/save_my_money/smm_tg/internal/modules/messenger"
"git.3crabs.ru/save_my_money/smm_tg/internal/services/bot"
)
type botPing struct {
messenger messenger.IMessenger
}
func NewBotPing(
messenger messenger.IMessenger,
) bot.IBot {
return &botPing{
messenger: messenger,
}
}
func (bot *botPing) Process(ctx context.Context, msg *messenger.Message) error {
if !strings.Contains(msg.Text, "/ping") {
return nil
}
bot.messenger.SendMessage(
ctx,
&messenger.Message{
ChatID: msg.ChatID,
Text: "pong",
},
)
return nil
}
+11
View File
@@ -0,0 +1,11 @@
package bot
import (
"context"
"git.3crabs.ru/save_my_money/smm_tg/internal/modules/messenger"
)
type IBot interface {
Process(ctx context.Context, msg *messenger.Message) error
}
+31
View File
@@ -0,0 +1,31 @@
package listener
import (
"context"
"log"
"git.3crabs.ru/save_my_money/smm_tg/internal/modules/messenger"
"git.3crabs.ru/save_my_money/smm_tg/internal/services/bot"
)
type ListenerService struct {
Messenger messenger.IMessenger
Bots []bot.IBot
}
func (s *ListenerService) Run(ctx context.Context) error {
for {
msg, err := s.Messenger.GetMessage(ctx)
if err != nil {
if err == context.Canceled {
return nil
}
return err
}
for _, b := range s.Bots {
if err := b.Process(ctx, msg); err != nil {
log.Println(err)
}
}
}
}