resolver_team_member.go

// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
// See LICENSE.txt for license information.

package api4

import (
	"context"
	"strings"

	"github.com/mattermost/mattermost-server/v6/model"
)

// teamMember is an internal graphQL wrapper struct to add resolver methods.
type teamMember struct {
	model.TeamMember
}

// match with api4.getTeam
func (tm *teamMember) Team(ctx context.Context) (*model.Team, error) {
	return getGraphQLTeam(ctx, tm.TeamId)
}

// match with api4.getUser
func (tm *teamMember) User(ctx context.Context) (*user, error) {
	return getGraphQLUser(ctx, tm.UserId)
}

// match with api4.getCategoriesForTeamForUser
func (tm *teamMember) SidebarCategories(ctx context.Context) ([]*model.SidebarCategoryWithChannels, error) {
	c, err := getCtx(ctx)
	if err != nil {
		return nil, err
	}

	if !c.App.SessionHasPermissionToUser(*c.AppContext.Session(), tm.UserId) {
		c.SetPermissionError(model.PermissionEditOtherUsers)
		return nil, c.Err
	}

	categories, appErr := c.App.GetSidebarCategories(tm.UserId, tm.TeamId)
	if appErr != nil {
		return nil, appErr
	}

	// TODO: look into optimizing this.
	// create map
	orderMap := make(map[string]*model.SidebarCategoryWithChannels, len(categories.Categories))
	for _, category := range categories.Categories {
		orderMap[category.Id] = category
	}

	// create a new slice based on the order
	res := make([]*model.SidebarCategoryWithChannels, 0, len(categories.Categories))
	for _, categoryId := range categories.Order {
		res = append(res, orderMap[categoryId])
	}

	return res, nil
}

// match with api4.getRolesByNames
func (tm *teamMember) Roles_(ctx context.Context) ([]*model.Role, error) {
	c, err := getCtx(ctx)
	if err != nil {
		return nil, err
	}

	return getGraphQLRoles(c, strings.Fields(tm.Roles))
}