group_helper/cli/commands/leavemessage.go

package commands

import (
	"git.dragon-labs.de/alphyron/group_helper/cli"
	"git.dragon-labs.de/alphyron/group_helper/logic"
	message2 "git.dragon-labs.de/alphyron/group_helper/message"
	"git.dragon-labs.de/alphyron/group_helper/models"
	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, group *models.Group) (bool, error) {
	message := update.Message.Text
	parts := strings.Split(message, " ")

	if len(parts) == 1 {
		_, 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, " ")

	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
		}

		msg := tgbotapi.NewMessage(update.Message.Chat.ID, message2.ReplaceMessage(group.UserLeaveMessage, update.Message.From, group))
		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
}