Pins to selected chats

Closes #6

lib/chatter/chatterinterface/chatterinterface.go

@@ -17,6 +17,7 @@ type ChatterInterface interface {
 	GetAllPrivateChats() ([]dbmapping.Chat, bool)
 	GetLeaguePrivateChats() ([]dbmapping.Chat, bool)
 	GetAllGroupChats() ([]dbmapping.Chat, bool)
+	GetGroupChatsByIDs(chatsIDs string) ([]dbmapping.Chat, bool)
 
 	UpdateChatTitle(chatRaw *dbmapping.Chat, newTitle string) (*dbmapping.Chat, bool)
 	UpdateChatTelegramID(update *tgbotapi.Update) (*dbmapping.Chat, bool)

lib/chatter/getters.go

@@ -6,6 +6,8 @@ package chatter
 import (
 	"github.com/go-telegram-bot-api/telegram-bot-api"
 	"lab.pztrn.name/fat0troll/i2_bot/lib/dbmapping"
+	"strconv"
+	"strings"
 	"time"
 )
 
@@ -140,3 +142,34 @@ func (ct *Chatter) GetAllGroupChats() ([]dbmapping.Chat, bool) {
 
 	return groupChats, true
 }
+
+// GetGroupChatsByIDs returns group chats with selected IDs
+func (ct *Chatter) GetGroupChatsByIDs(chatsIDs string) ([]dbmapping.Chat, bool) {
+	groupChats := []dbmapping.Chat{}
+
+	queryIDs := make([]int, 0)
+	queryIDsStr := strings.Split(chatsIDs, ",")
+	for i := range queryIDsStr {
+		id, _ := strconv.Atoi(queryIDsStr[i])
+		if id != 0 {
+			queryIDs = append(queryIDs, id)
+		}
+	}
+
+	finalQueryIDs := ""
+	for i := range queryIDs {
+		finalQueryIDs += strconv.Itoa(queryIDs[i])
+		if i < len(queryIDs)-1 {
+			finalQueryIDs += ","
+		}
+	}
+	c.Log.Debug("Chat query IDs: " + finalQueryIDs)
+
+	err := c.Db.Select(&groupChats, "SELECT * FROM chats WHERE chat_type IN ('group', 'supergroup') AND id IN ("+finalQueryIDs+")")
+	if err != nil {
+		c.Log.Error(err)
+		return groupChats, false
+	}
+
+	return groupChats, true
+}

lib/pinner/exported.go

@@ -26,5 +26,5 @@ func New(ac *appcontext.Context) {
 func (p *Pinner) Init() {
 	c.Log.Info("Initializing Pinner...")
 
-	c.Cron.AddFunc("0 55 */2 * * *", p.PinBattleAlert)
+	c.Cron.AddFunc("0 55 4-23/2 * * *", p.PinBattleAlert)
 }

lib/pinner/pinner.go

@@ -5,6 +5,8 @@ package pinner
 
 import (
 	"github.com/go-telegram-bot-api/telegram-bot-api"
+	"strconv"
+	"strings"
 )
 
 // PinMessageToAllChats pins message to all groups where bot exist
@@ -70,6 +72,77 @@ func (p *Pinner) PinMessageToAllChats(update *tgbotapi.Update) string {
 	return "ok"
 }
 
+// PinMessageToSomeChats pins message to selected groups where bot exist
+func (p *Pinner) PinMessageToSomeChats(update *tgbotapi.Update) string {
+	commandArgs := update.Message.CommandArguments()
+	commandArgsList := strings.Split(commandArgs, " ")
+	if len(commandArgsList) < 2 {
+		return "fail"
+	}
+
+	chatsToPin := commandArgsList[0]
+	messageToPin := commandArgsList[1]
+
+	if messageToPin == "" {
+		return "fail"
+	}
+
+	groupChats, ok := c.Chatter.GetGroupChatsByIDs(chatsToPin)
+	if !ok {
+		return "fail"
+	}
+	c.Log.Debug("Got " + strconv.Itoa(len(groupChats)) + " group chats...")
+
+	for i := range groupChats {
+		if groupChats[i].ChatType == "supergroup" {
+			message := messageToPin + "\n\n"
+			message += "© " + update.Message.From.FirstName + " " + update.Message.From.LastName
+			message += " (@" + update.Message.From.UserName + ")"
+
+			msg := tgbotapi.NewMessage(groupChats[i].TelegramID, message)
+			msg.ParseMode = "Markdown"
+
+			pinnableMessage, err := c.Bot.Send(msg)
+			if err != nil {
+				c.Log.Error(err.Error())
+
+				message := "*Ваше сообщение не отправлено.*\n\n"
+				message += "Обычно это связано с тем, что нарушена разметка Markdown. "
+				message += "К примеру, если вы хотели использовать нижнее\\_подчёркивание, то печатать его надо так — \\\\_. То же самое касается всех управляющих разметкой символов в Markdown в случае, если вы их хотите использовать как текст, а не как управляющий символ Markdown."
+
+				msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
+				msg.ParseMode = "Markdown"
+
+				c.Bot.Send(msg)
+
+				return "fail"
+			}
+
+			pinChatMessageConfig := tgbotapi.PinChatMessageConfig{
+				ChatID:              pinnableMessage.Chat.ID,
+				MessageID:           pinnableMessage.MessageID,
+				DisableNotification: true,
+			}
+
+			_, err = c.Bot.PinChatMessage(pinChatMessageConfig)
+			if err != nil {
+				c.Log.Error(err.Error())
+			}
+		}
+	}
+
+	message := "*Ваше сообщение отправлено и запинено во все чаты, где сидит бот.*\n\n"
+	message += "Текст отправленного сообщения:\n\n"
+	message += messageToPin
+
+	msg := tgbotapi.NewMessage(update.Message.Chat.ID, message)
+	msg.ParseMode = "Markdown"
+
+	c.Bot.Send(msg)
+
+	return "ok"
+}
+
 // PinBattleAlert pins to all squads 'battle alert' at :55 of every even hour
 // Even hours are in Moscow timezone
 func (p *Pinner) PinBattleAlert() {

lib/pinner/pinnerinterface/pinnerinterface.go

@@ -12,5 +12,7 @@ type PinnerInterface interface {
 	Init()
 
 	PinBattleAlert()
+
+	PinMessageToSomeChats(update *tgbotapi.Update) string
 	PinMessageToAllChats(update *tgbotapi.Update) string
 }

lib/router/private_request.go

@@ -80,7 +80,7 @@ func (r *Router) routePrivateRequest(update *tgbotapi.Update, playerRaw *dbmappi
 				}
 
 				return c.Talkers.AnyMessageUnauthorized(update)
-			case update.Message.Command() == "group_chats":
+			case update.Message.Command() == "chats":
 				if c.Users.PlayerBetterThan(playerRaw, "admin") {
 					c.Chatter.GroupsList(update)
 					return "ok"
@@ -96,6 +96,12 @@ func (r *Router) routePrivateRequest(update *tgbotapi.Update, playerRaw *dbmappi
 
 				return c.Talkers.AnyMessageUnauthorized(update)
 			case update.Message.Command() == "pin":
+				if c.Users.PlayerBetterThan(playerRaw, "admin") {
+					return c.Pinner.PinMessageToSomeChats(update)
+				}
+
+				return c.Talkers.AnyMessageUnauthorized(update)
+			case update.Message.Command() == "pin_all":
 				if c.Users.PlayerBetterThan(playerRaw, "admin") {
 					return c.Pinner.PinMessageToAllChats(update)
 				}

lib/talkers/help.go

@@ -20,9 +20,10 @@ func (t *Talkers) HelpMessage(update *tgbotapi.Update, playerRaw *dbmapping.Play
 	if c.Users.PlayerBetterThan(playerRaw, "admin") {
 		message += "+ /send\\_all _текст_ — отправить сообщение всем пользователям бота\n"
 		message += "+ /send\\_league _текст_ — отправить сообщение всем пользователям бота, у которых профиль лиги Инстинкт\n"
-		message += "+ /group\\_chats — получить список групп, в которых работает бот.\n"
+		message += "+ /chats — получить список групп, в которых работает бот.\n"
 		message += "+ /squads — получить список отрядов.\n"
-		message += "+ /pin _текст_ — отправить сообщение во все группы, где находится бот. Сообщение будет автоматически запинено.\n"
+		message += "+ /pin _номера чатов_ _текст_ — отправить сообщение в чаты с номерами. Сообщение будет автоматичекси запинено. Пример: \"/pin 2,3,5 привет мир\". Внимание: между номерами чатов ставятся запятые без пробелов! Всё, что идёт после второго пробела в команде — сообщение\n"
+		message += "+ /pin\\_all _текст_ — отправить сообщение во все группы, где находится бот. Сообщение будет автоматически запинено.\n"
 		message += "+ /users —  просмотреть зарегистрированных пользователей бота\n"
 	}
 	message += "+ /help – выводит данное сообщение\n"