messages.addChatUser_handler.go

// Copyright 2022 Teamgram Authors
//  All rights reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//   http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
//
// Author: teamgramio (teamgram.io@gmail.com)
//

package core

import (
	"math/rand"

	"github.com/teamgram/proto/mtproto"
	msgpb "github.com/teamgram/teamgram-server/app/messenger/msg/msg/msg"
	chatpb "github.com/teamgram/teamgram-server/app/service/biz/chat/chat"
	userpb "github.com/teamgram/teamgram-server/app/service/biz/user/user"
)

// MessagesAddChatUser
// messages.addChatUser#f24753e3 chat_id:long user_id:InputUser fwd_limit:int = Updates;
func (c *ChatsCore) MessagesAddChatUser(in *mtproto.TLMessagesAddChatUser) (*mtproto.Updates, error) {
	var (
		err       error
		addUser   = mtproto.FromInputUser(c.MD.UserId, in.UserId)
		chat      *mtproto.MutableChat
		inviterId int64
		isBot     = false
	)

	if !addUser.IsUser() || addUser.PeerId == 0 {
		err = mtproto.ErrPeerIdInvalid
		c.Logger.Errorf("messages.addChatUser - error: %v", err)
		return nil, err
	}

	if !c.MD.IsAdmin {
		inviterId = c.MD.UserId

		// 400	USERS_TOO_MUCH	The maximum number of users has been exceeded (to create a chat, for example)
		// 400	USER_ALREADY_PARTICIPANT	The user is already in the group
		// 400	USER_ID_INVALID	The provided user ID is invalid
		// 403	USER_NOT_MUTUAL_CONTACT	The provided user is not a mutual contact
		// 403	USER_PRIVACY_RESTRICTED	The user's privacy settings do not allow you to do this
		users, _ := c.svcCtx.Dao.UserClient.UserGetMutableUsers(c.ctx, &userpb.TLUserGetMutableUsers{
			Id: []int64{inviterId, addUser.PeerId},
		})

		me, _ := users.GetImmutableUser(inviterId)
		added, _ := users.GetImmutableUser(addUser.PeerId)
		if me == nil || added == nil {
			err = mtproto.ErrPeerIdInvalid
			c.Logger.Errorf("messages.addChatUser - error: %v", err)
			return nil, err
		}

		if added.IsBot() && added.BotNochats() {
			err = mtproto.ErrBotGroupsBlocked
			c.Logger.Errorf("messages.addChatUser - error: %v", err)
			return nil, err
		}

		rules, _ := c.svcCtx.Dao.UserClient.UserGetPrivacy(c.ctx, &userpb.TLUserGetPrivacy{
			UserId:  addUser.PeerId,
			KeyType: userpb.CHAT_INVITE,
		})
		if len(rules.Datas) > 0 {
			// return true
			allowAddChat := userpb.CheckPrivacyIsAllow(
				addUser.PeerId,
				rules.Datas,
				inviterId,
				func(id, checkId int64) bool {
					contact, _ := c.svcCtx.Dao.UserClient.UserCheckContact(c.ctx, &userpb.TLUserCheckContact{
						UserId: id,
						Id:     checkId,
					})
					return mtproto.FromBool(contact)
				},
				func(checkId int64, idList []int64) bool {
					chatIdList, _ := mtproto.SplitChatAndChannelIdList(idList)
					return c.svcCtx.Dao.ChatClient.CheckParticipantIsExist(c.ctx, checkId, chatIdList)
				})
			if !allowAddChat {
				err = mtproto.ErrUserPrivacyRestricted
				c.Logger.Errorf("not allow addChat: %v", err)
				return nil, err
			}
		}

		isBot = added.IsBot()
	} else {
		inviterId = 0
	}

	chat, err = c.svcCtx.Dao.ChatClient.Client().ChatAddChatUser(c.ctx, &chatpb.TLChatAddChatUser{
		ChatId:    in.ChatId,
		InviterId: inviterId,
		UserId:    addUser.PeerId,
		IsBot:     isBot,
	})
	//request.ChatId, md.UserId, peer.PeerId)
	if err != nil {
		c.Logger.Errorf("addChatUser error: %v", err)
		return nil, err
	}

	fromId := c.MD.UserId
	if c.MD.IsAdmin {
		fromId = chat.Creator()
	}

	rUpdates, err := c.svcCtx.Dao.MsgClient.MsgSendMessage(c.ctx, &msgpb.TLMsgSendMessage{
		UserId:    fromId,
		AuthKeyId: c.MD.AuthId,
		PeerType:  mtproto.PEER_CHAT,
		PeerId:    in.ChatId,
		Message: msgpb.MakeTLOutboxMessage(&msgpb.OutboxMessage{
			NoWebpage:    true,
			Background:   false,
			RandomId:     rand.Int63(),
			Message:      chat.MakeMessageService(fromId, mtproto.MakeMessageActionChatAddUser(addUser.PeerId)),
			ScheduleDate: nil,
		}).To_OutboxMessage(),
	})

	if err != nil {
		c.Logger.Errorf("addChatUser error: %v", err)
		return nil, err
	}

	return rUpdates, nil
}