add admin, add groupDB and prepare to groups events. many refactors

.env.example

@@ -1,3 +1,4 @@
 BOT_TOKEN=
+ADMIN_USER_ID=0
 DEFAULT_LOCALE=uk-UA
 BOT_PREFIX=🤖

cmd/bot/main.go

@@ -35,7 +35,7 @@ func main() {
  }
  }(database)
 
- err = bot.InitBot(cfg.TelegramLoggerBotToken, translator, database)
+ err = bot.InitBot(cfg.TelegramLoggerBotToken, cfg.TelegramAdminUserID, translator, database)
  if err != nil {
  log.Error().Err(err).Send()
  panic(err)

cmd/config.go

@@ -8,10 +8,10 @@ import (
 )
 
 type Config struct {
- TelegramLoggerBotToken  string `mapstructure:"bot_token"`
- TelegramLoggerBotChatID int64 `mapstructure:"notifications_chat_id"`
- DefaultLocale  string `mapstructure:"default_locale"`
- BotPrefix  string `mapstructure:"bot_prefix"`
+ TelegramLoggerBotToken string `mapstructure:"bot_token"`
+ TelegramAdminUserID int64 `mapstructure:"admin_user_id"`
+ DefaultLocale string `mapstructure:"default_locale"`
+ BotPrefix string `mapstructure:"bot_prefix"`
 }
 
 func LoadConfig(path, configFile, configType string) (config Config, err error) {
@@ -20,7 +20,7 @@ func LoadConfig(path, configFile, configType string) (config Config, err error)
  viper.SetConfigType(configType)
 
  viper.SetDefault("bot_token", "")
- viper.SetDefault("notifications_chat_id", 0)
+ viper.SetDefault("admin_user_id", 0)
  viper.SetDefault("default_locale", "uk-UA")
  viper.SetDefault("bot_prefix", "🤖")
 

internal/bot/bot.go

@@ -12,13 +12,13 @@ import (
 )
 
 type bot struct {
- bot *tgbotapi.BotAPI
- db *botDB
- tl *translations.Translator
+ bot *tgbotapi.BotAPI
+ db *botDB
+ tl *translations.Translator
+ adminID int64
 }
 
-func InitBot(token string, translator *translations.Translator, db *db.DB) error {
-
+func InitBot(token string, adminID int64, translator *translations.Translator, db *db.DB) error {
  botAPI, err := tgbotapi.NewBotAPI(token)
  if err != nil {
  return err
@@ -27,9 +27,10 @@ func InitBot(token string, translator *translations.Translator, db *db.DB) error
  dbState := newBotDB(db)
 
  botState := &bot{
- bot: botAPI,
- db: dbState,
- tl: translator,
+ bot: botAPI,
+ db: dbState,
+ tl: translator,
+ adminID: adminID,
  }
  botState.InitUpdates()
 
@@ -52,6 +53,13 @@ func (b *bot) InitUpdates() {
  }
 
  if update.Message != nil {
+ if update.Message.MigrateToChatID != 0 && b.db.groupDB().get() == update.Message.Chat.ID {
+ err := b.db.groupDB().set(update.Message.MigrateToChatID)
+ if err != nil {
+ log.Error().Err(err).Send()
+ }
+ }
+
  if update.Message.IsCommand() {
  switch update.Message.Command() {
  case startCommand:
@@ -62,11 +70,21 @@ func (b *bot) InitUpdates() {
  b.BreakCommand(update)
  case cancelCommand:
  b.CancelCommand(update)
+ case getID:
+ b.GetID(update)
+ case setGroup:
+ b.SetGroup(update)
+ case event:
+ b.Event(update)
  }
 
  continue
  }
 
+ if !update.FromChat().IsPrivate() {
+ continue
+ }
+
  fromQueue, err := b.db.queueDB().get(update.Message.From.ID)
  if err != nil {
  log.Error().Err(err).Send()
@@ -95,7 +113,7 @@ func (b *bot) InitUpdates() {
  }
 
  msg := tgbotapi.NewMessage(
- update.Message.Chat.ID,
+ update.Message.From.ID,
  "🤖 Ваше повідомлення було отримано, якщо у вас є що додати - напишіть.\n\nСкоро до вас доєднається оператор технічної підтримки",
  )
  _, err = b.bot.Send(msg)
@@ -135,7 +153,7 @@ func (b *bot) InitUpdates() {
  log.Error().Err(fmt.Errorf("whoomSend is empty")).Send()
  }
 
- msg := tgbotapi.NewCopyMessage(*whoomSend, update.Message.Chat.ID, update.Message.MessageID)
+ msg := tgbotapi.NewCopyMessage(*whoomSend, update.Message.From.ID, update.Message.MessageID)
  if update.Message.ReplyToMessage != nil {
  replyToID, err := b.db.messagesIDsDB().get(update.Message.ReplyToMessage.MessageID)
  if err != nil && err != badger.ErrKeyNotFound {

internal/bot/commands.go

@@ -1,6 +1,8 @@
 package bot
 
 import (
+ "fmt"
+
  tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api/v5"
  "github.com/samarets/support-bot/internal/log"
 )
@@ -10,10 +12,14 @@ const (
  connectCommand = "connect"
  breakCommand = "break"
  cancelCommand = "cancel"
+ getID = "getid"
+
+ setGroup = "set_group"
+ event = "event"
 )
 
 func (b *bot) StartCommand(update tgbotapi.Update) {
- if update.Message == nil {
+ if !update.FromChat().IsPrivate() {
  return
  }
 
@@ -213,3 +219,49 @@ func (b *bot) CancelCommand(update tgbotapi.Update) {
  return
  }
 }
+
+func (b *bot) GetID(update tgbotapi.Update) {
+ msg := tgbotapi.NewMessage(update.Message.Chat.ID, fmt.Sprintf("🤖 %d", update.SentFrom().ID))
+ _, err := b.bot.Send(msg)
+ if err != nil {
+ log.Error().Err(err).Send()
+ return
+ }
+}
+
+func (b *bot) SetGroup(update tgbotapi.Update) {
+ user := update.SentFrom()
+ if user.ID != b.adminID {
+ return
+ }
+
+ err := b.db.groupDB().set(update.FromChat().ID)
+ if err != nil {
+ log.Error().Err(err).Send()
+ return
+ }
+
+ msg := tgbotapi.NewMessage(
+ update.FromChat().ID,
+ b.tl.GetMessage(b.db.languageDB().get(user.ID), "channel_saved"),
+ )
+ _, err = b.bot.Send(msg)
+ if err != nil {
+ log.Error().Err(err).Send()
+ return
+ }
+}
+
+func (b *bot) Event(update tgbotapi.Update) {
+ groupID := b.db.groupDB().get()
+
+ msg := tgbotapi.NewMessage(
+ groupID,
+ "event",
+ )
+ _, err := b.bot.Send(msg)
+ if err != nil {
+ log.Error().Err(err).Send()
+ return
+ }
+}

internal/bot/db.go

@@ -16,6 +16,7 @@ const (
  buffer = "buffer"
  messagesIDs = "messagesIDs"
  language = "language"
+ group = "group"
 )
 
 type botDB struct {
@@ -201,6 +202,36 @@ func (db *languageDB) get(userID int64) string {
  return lang
 }
 
+type groupDB struct {
+ *botDB
+}
+
+func (db *botDB) groupDB() *groupDB {
+ return &groupDB{
+ botDB: db,
+ }
+}
+
+func (db *groupDB) set(groupID int64) error {
+ return db.db.Set([]byte(group), groupID)
+}
+
+func (db *groupDB) get() int64 {
+ var groupID int64
+ err := db.db.Get([]byte(group), &groupID)
+ if err != nil {
+ switch err {
+ case badger.ErrKeyNotFound:
+ return 0
+ default:
+ log.Error().Err(err).Send()
+ return 0
+ }
+ }
+
+ return groupID
+}
+
 func mergePrefixDB(prefix string, id interface{}) []byte {
  return []byte(fmt.Sprintf("%s-%d", prefix, id))
 }

locales/en-US/admin.json

@@ -1,3 +1,4 @@
 {
+ "channel_saved": "The message distribution channel has been saved",
  "user_connected": "You have been attached to a user [{{ .Name }}](tg:https://user?id={{ .ID }})\nID: {{ .ID }}\nUser question:"
 }

locales/uk-UA/admin.json

@@ -1,3 +1,4 @@
 {
+ "channel_saved": "Канал для поширення повідомлень був збережений",
  "user_connected": "Ви були доєднані до користувача [{{ .Name }}](tg:https://user?id={{ .ID }})\nID: {{ .ID }}\nПитання користувача:"
 }