init app
This commit is contained in:
@@ -0,0 +1,6 @@
|
||||
package entities
|
||||
|
||||
type User struct {
|
||||
ChatID string `sql:"chat_id"`
|
||||
UserID string `sql:"user_id"`
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
@@ -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
|
||||
}
|
||||
@@ -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)
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user