group_helper/telegram/routines/database.go

package routines

import (
	"git.dragon-labs.de/alphyron/group_helper/logic"
	"git.dragon-labs.de/alphyron/group_helper/models"
	"git.dragon-labs.de/alphyron/group_helper/telegram"
	tgbotapi "github.com/go-telegram-bot-api/telegram-bot-api"
)

type databaseRoutine struct {
	groupHelperService logic.GroupHelperService
}

func NewDatabaseRoutine(groupHelperService logic.GroupHelperService) telegram.Routine {
	return &databaseRoutine{groupHelperService: groupHelperService}
}

func (d databaseRoutine) Update(botAPI *tgbotapi.BotAPI, update *tgbotapi.Update, group *models.Group) error {

	if update.Message == nil { // ignore any non-Message Updates
		return nil
	}

	if update.Message.Chat.IsPrivate() || update.Message.Chat.IsChannel() {
		return nil
	}

	if update.Message.NewChatMembers != nil {
		for _, newUser := range *update.Message.NewChatMembers {
			user, err := d.groupHelperService.GetUserByID(int64(newUser.ID))

			if err != nil {
				return err
			}

			go d.groupHelperService.UserJoinGroup(user, group)
		}
		config := tgbotapi.ChatConfig{
			ChatID: update.Message.Chat.ID,
		}
		size, _ := botAPI.GetChatMembersCount(config)
		group.Size = size - 1
		_, err := d.groupHelperService.UpdateGroup(group)
		return err
	}

	if update.Message.LeftChatMember != nil {
		user, err := d.groupHelperService.GetUserByID(int64(update.Message.LeftChatMember.ID))

		if err != nil {
			return err
		}

		go d.groupHelperService.UserLeaveGroup(user, group)
		config := tgbotapi.ChatConfig{
			ChatID: update.Message.Chat.ID,
		}
		size, _ := botAPI.GetChatMembersCount(config)
		group.Size = size - 1
		_, err = d.groupHelperService.UpdateGroup(group)
		return err
	}

	user, err := d.groupHelperService.GetUserByID(int64(update.Message.From.ID))

	if err != nil {
		return err
	}

	_, err = d.groupHelperService.UserJoinGroup(user, group)
	return err
}