Open-IM-Server/pkg/common/db/relation/group_member_model.go

// Copyright © 2023 OpenIM. 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.

package relation

import (
	"context"

	"gorm.io/gorm"

	"github.com/OpenIMSDK/protocol/constant"
	"github.com/OpenIMSDK/tools/ormutil"
	"github.com/OpenIMSDK/tools/utils"

	"github.com/openimsdk/open-im-server/v3/pkg/common/db/table/relation"
)

var _ relation.GroupMemberModelInterface = (*GroupMemberGorm)(nil)

type GroupMemberGorm struct {
	*MetaDB
}

func NewGroupMemberDB(db *gorm.DB) relation.GroupMemberModelInterface {
	return &GroupMemberGorm{NewMetaDB(db, &relation.GroupMemberModel{})}
}

func (g *GroupMemberGorm) NewTx(tx any) relation.GroupMemberModelInterface {
	return &GroupMemberGorm{NewMetaDB(tx.(*gorm.DB), &relation.GroupMemberModel{})}
}

func (g *GroupMemberGorm) Create(ctx context.Context, groupMemberList []*relation.GroupMemberModel) (err error) {
	return utils.Wrap(g.db(ctx).Create(&groupMemberList).Error, "")
}

func (g *GroupMemberGorm) Delete(ctx context.Context, groupID string, userIDs []string) (err error) {
	return utils.Wrap(
		g.db(ctx).Where("group_id = ? and user_id in (?)", groupID, userIDs).Delete(&relation.GroupMemberModel{}).Error,
		"",
	)
}

func (g *GroupMemberGorm) DeleteGroup(ctx context.Context, groupIDs []string) (err error) {
	return utils.Wrap(g.db(ctx).Where("group_id in (?)", groupIDs).Delete(&relation.GroupMemberModel{}).Error, "")
}

func (g *GroupMemberGorm) Update(ctx context.Context, groupID string, userID string, data map[string]any) (err error) {
	return utils.Wrap(g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(data).Error, "")
}

func (g *GroupMemberGorm) UpdateRoleLevel(
	ctx context.Context,
	groupID string,
	userID string,
	roleLevel int32,
) (rowsAffected int64, err error) {
	db := g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Updates(map[string]any{
		"role_level": roleLevel,
	})
	return db.RowsAffected, utils.Wrap(db.Error, "")
}

func (g *GroupMemberGorm) Find(
	ctx context.Context,
	groupIDs []string,
	userIDs []string,
	roleLevels []int32,
) (groupMembers []*relation.GroupMemberModel, err error) {
	db := g.db(ctx)
	if len(groupIDs) > 0 {
		db = db.Where("group_id in (?)", groupIDs)
	}
	if len(userIDs) > 0 {
		db = db.Where("user_id in (?)", userIDs)
	}
	if len(roleLevels) > 0 {
		db = db.Where("role_level in (?)", roleLevels)
	}
	return groupMembers, utils.Wrap(db.Find(&groupMembers).Error, "")
}

func (g *GroupMemberGorm) Take(
	ctx context.Context,
	groupID string,
	userID string,
) (groupMember *relation.GroupMemberModel, err error) {
	groupMember = &relation.GroupMemberModel{}
	return groupMember, utils.Wrap(
		g.db(ctx).Where("group_id = ? and user_id = ?", groupID, userID).Take(groupMember).Error,
		"",
	)
}

func (g *GroupMemberGorm) TakeOwner(
	ctx context.Context,
	groupID string,
) (groupMember *relation.GroupMemberModel, err error) {
	groupMember = &relation.GroupMemberModel{}
	return groupMember, utils.Wrap(
		g.db(ctx).Where("group_id = ? and role_level = ?", groupID, constant.GroupOwner).Take(groupMember).Error,
		"",
	)
}

func (g *GroupMemberGorm) SearchMember(
	ctx context.Context,
	keyword string,
	groupIDs []string,
	userIDs []string,
	roleLevels []int32,
	pageNumber, showNumber int32,
) (total uint32, groupList []*relation.GroupMemberModel, err error) {
	db := g.db(ctx)
	ormutil.GormIn(&db, "group_id", groupIDs)
	ormutil.GormIn(&db, "user_id", userIDs)
	ormutil.GormIn(&db, "role_level", roleLevels)
	return ormutil.GormSearch[relation.GroupMemberModel](db, []string{"nickname"}, keyword, pageNumber, showNumber)
}

func (g *GroupMemberGorm) MapGroupMemberNum(
	ctx context.Context,
	groupIDs []string,
) (count map[string]uint32, err error) {
	return ormutil.MapCount(g.db(ctx).Where("group_id in (?)", groupIDs), "group_id")
}

func (g *GroupMemberGorm) FindJoinUserID(
	ctx context.Context,
	groupIDs []string,
) (groupUsers map[string][]string, err error) {
	var groupMembers []*relation.GroupMemberModel
	if err := g.db(ctx).Select("group_id, user_id").Where("group_id in (?)", groupIDs).Find(&groupMembers).Error; err != nil {
		return nil, utils.Wrap(err, "")
	}
	groupUsers = make(map[string][]string)
	for _, item := range groupMembers {
		v, ok := groupUsers[item.GroupID]
		if !ok {
			groupUsers[item.GroupID] = []string{item.UserID}
		} else {
			groupUsers[item.GroupID] = append(v, item.UserID)
		}
	}
	return groupUsers, nil
}

func (g *GroupMemberGorm) FindMemberUserID(ctx context.Context, groupID string) (userIDs []string, err error) {
	return userIDs, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Pluck("user_id", &userIDs).Error, "")
}

func (g *GroupMemberGorm) FindUserJoinedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
	return groupIDs, utils.Wrap(g.db(ctx).Where("user_id = ?", userID).Pluck("group_id", &groupIDs).Error, "")
}

func (g *GroupMemberGorm) TakeGroupMemberNum(ctx context.Context, groupID string) (count int64, err error) {
	return count, utils.Wrap(g.db(ctx).Where("group_id = ?", groupID).Count(&count).Error, "")
}

func (g *GroupMemberGorm) FindUsersJoinedGroupID(ctx context.Context, userIDs []string) (map[string][]string, error) {
	var groupMembers []*relation.GroupMemberModel
	err := g.db(ctx).Select("group_id, user_id").Where("user_id IN (?)", userIDs).Find(&groupMembers).Error
	if err != nil {
		return nil, err
	}
	result := make(map[string][]string)
	for _, groupMember := range groupMembers {
		v, ok := result[groupMember.UserID]
		if !ok {
			result[groupMember.UserID] = []string{groupMember.GroupID}
		} else {
			result[groupMember.UserID] = append(v, groupMember.GroupID)
		}
	}
	return result, nil
}

func (g *GroupMemberGorm) FindUserManagedGroupID(ctx context.Context, userID string) (groupIDs []string, err error) {
	return groupIDs, utils.Wrap(
		g.db(ctx).
			Model(&relation.GroupMemberModel{}).
			Where("user_id = ? and (role_level = ? or role_level = ?)", userID, constant.GroupOwner, constant.GroupAdmin).
			Pluck("group_id", &groupIDs).
			Error,
		"",
	)
}