Implement leave set and get message command

This commit is contained in:
Alphyron 2020-04-25 23:32:11 +02:00
parent dd157bfd5a
commit f6316ada97
2 changed files with 91 additions and 0 deletions

View File

@ -0,0 +1,90 @@
package commands
import (
"git.dragon-labs.de/alphyron/group_helper/cli"
"git.dragon-labs.de/alphyron/group_helper/logic"
tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
"strings"
)
type leaveMessage struct {
groupHelperService logic.GroupHelperService
}
func NewLeaveMessage(groupHelperService logic.GroupHelperService) cli.Command {
return &leaveMessage{
groupHelperService: groupHelperService,
}
}
func (j leaveMessage) GetUsage() string {
return "/leave (set/get) [message]"
}
func (j leaveMessage) GetCommand() string {
return "/leave"
}
func (j leaveMessage) GetDescription() string {
return "Get or set the leave message"
}
func (j leaveMessage) ExecuteCommand(api *tgbotapi.BotAPI, update *tgbotapi.Update) (bool, error) {
message := update.Message.Text
parts := strings.Split(message, " ")
if len(parts) == 0 {
_, err := api.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "Wrong usage of this Command:\n"+j.GetUsage()))
return err != nil, err
}
args := strings.Split(message, " ")
group, err := j.groupHelperService.GetGroupByID(update.Message.Chat.ID)
if err != nil {
return false, err
}
switch args[1] {
case "get":
msg := tgbotapi.NewMessage(update.Message.Chat.ID, group.UserLeaveMessage)
api.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "Current LeaveMessage (decoded)"))
_, err = api.Send(msg)
return err != nil, err
case "set":
if len(parts) <= 2 {
_, err := api.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "Wrong usage of this Command:\n"+j.GetUsage()))
return err != nil, err
}
newLeaveMessage := strings.Join(args[2:], " ")
group.UserLeaveMessage = newLeaveMessage
group, err = j.groupHelperService.UpdateGroup(group)
if err != nil {
return false, err
}
//TODO Create a replace for placeholder
msg := tgbotapi.NewMessage(update.Message.Chat.ID, group.UserJoinMessage)
msg.ParseMode = "Markdown"
api.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "Example LeaveMessage"))
_, err = api.Send(msg)
return err != nil, err
default:
_, err := api.Send(tgbotapi.NewMessage(update.Message.Chat.ID, "Wrong usage of this Command:\n"+j.GetUsage()))
return err != nil, err
}
}
func (j leaveMessage) AllowChatType(chat *tgbotapi.Chat) bool {
return chat.IsGroup() || chat.IsSuperGroup()
}
func (j leaveMessage) AllowMember(member *tgbotapi.ChatMember) bool {
return member.IsAdministrator() || member.IsCreator()
}
func (j leaveMessage) AllowEveryMember() bool {
return false
}

View File

@ -78,6 +78,7 @@ func InitialCommandManager(bot *tgbotapi.BotAPI, service logic.GroupHelperServic
cm.RegisterCommand(commands.NewInfoCommand())
cm.RegisterCommand(commands.NewHelpCommand(cm))
cm.RegisterCommand(commands.NewJoinMessage(service))
cm.RegisterCommand(commands.NewLeaveMessage(service))
return cm
}