ec1f89217a
Full Mattermost server source with integrated Community Enterprise features. Includes vendor directory for offline/air-gapped builds. Structure: - enterprise-impl/: Enterprise feature implementations - enterprise-community/: Init files that register implementations - enterprise/: Bridge imports (community_imports.go) - vendor/: All dependencies for offline builds Build (online): go build ./cmd/mattermost Build (offline/air-gapped): go build -mod=vendor ./cmd/mattermost 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
321 lines
9.6 KiB
Go
321 lines
9.6 KiB
Go
// Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
|
|
// See LICENSE.txt for license information.
|
|
|
|
package slashcommands
|
|
|
|
import (
|
|
"strings"
|
|
|
|
"github.com/mattermost/mattermost/server/public/model"
|
|
"github.com/mattermost/mattermost/server/public/shared/i18n"
|
|
"github.com/mattermost/mattermost/server/public/shared/mlog"
|
|
"github.com/mattermost/mattermost/server/public/shared/request"
|
|
"github.com/mattermost/mattermost/server/v8/channels/app"
|
|
)
|
|
|
|
type InviteProvider struct {
|
|
}
|
|
|
|
const (
|
|
CmdInvite = "invite"
|
|
)
|
|
|
|
type UserError int64
|
|
|
|
const (
|
|
NoError UserError = iota
|
|
UserInChannel
|
|
UserNotInTeam
|
|
IsConstrained
|
|
Unknown
|
|
)
|
|
|
|
func init() {
|
|
app.RegisterCommandProvider(&InviteProvider{})
|
|
}
|
|
|
|
func (*InviteProvider) GetTrigger() string {
|
|
return CmdInvite
|
|
}
|
|
|
|
func (*InviteProvider) GetCommand(a *app.App, T i18n.TranslateFunc) *model.Command {
|
|
return &model.Command{
|
|
Trigger: CmdInvite,
|
|
AutoComplete: true,
|
|
AutoCompleteDesc: T("api.command_invite.desc"),
|
|
AutoCompleteHint: T("api.command_invite.hint"),
|
|
DisplayName: T("api.command_invite.name"),
|
|
}
|
|
}
|
|
|
|
func (i *InviteProvider) DoCommand(a *app.App, rctx request.CTX, args *model.CommandArgs, message string) *model.CommandResponse {
|
|
return &model.CommandResponse{
|
|
Text: i.doCommand(a, rctx, args, message),
|
|
ResponseType: model.CommandResponseTypeEphemeral,
|
|
}
|
|
}
|
|
|
|
func (i *InviteProvider) doCommand(a *app.App, rctx request.CTX, args *model.CommandArgs, message string) string {
|
|
if message == "" {
|
|
return args.T("api.command_invite.missing_message.app_error")
|
|
}
|
|
|
|
resps := &[]string{}
|
|
|
|
targetUsers, targetChannels, resp := i.parseMessage(a, rctx, args, resps, message)
|
|
if resp != "" {
|
|
return resp
|
|
}
|
|
|
|
// Verify that the inviter has permissions to invite users to the every channel.
|
|
targetChannels = i.checkPermissions(a, rctx, args, resps, targetUsers[0], targetChannels)
|
|
|
|
// track errors returned for various users.
|
|
differentChannels := make(map[string][]string)
|
|
nonTeamUsers := make(map[string][]string)
|
|
channelConstrained := make([]string, 0, 1)
|
|
usersInChannel := make([]string, 0, 1)
|
|
errorUsers := make([]string, 0, 1)
|
|
|
|
for _, targetChannel := range targetChannels {
|
|
var targetTeamDisplay string
|
|
for _, targetUser := range targetUsers {
|
|
userError := i.addUserToChannel(a, rctx, args, targetUser, targetChannel)
|
|
if userError == NoError {
|
|
if args.ChannelId != targetChannel.Id {
|
|
differentChannels[targetChannel.Name] = append(differentChannels[targetChannel.Name], targetUser.Username)
|
|
}
|
|
} else if userError == UserNotInTeam {
|
|
if targetTeamDisplay == "" {
|
|
targetTeam, err := a.GetTeam(targetChannel.TeamId)
|
|
if err != nil {
|
|
targetTeamDisplay = "unknown"
|
|
} else {
|
|
targetTeamDisplay = targetTeam.DisplayName
|
|
}
|
|
}
|
|
nonTeamUsers[targetTeamDisplay] = append(nonTeamUsers[targetTeamDisplay], targetUser.Username)
|
|
} else if userError == IsConstrained {
|
|
channelConstrained = append(channelConstrained, targetUser.Username)
|
|
} else if userError == UserInChannel {
|
|
usersInChannel = append(usersInChannel, targetUser.Username)
|
|
} else {
|
|
errorUsers = append(errorUsers, targetUser.Username)
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(usersInChannel) > 0 {
|
|
if len(usersInChannel) > 10 {
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.user_already_in_channel.overflow", map[string]any{
|
|
"FirstUser": "@" + usersInChannel[0],
|
|
"Others": len(usersInChannel) - 1,
|
|
}),
|
|
)
|
|
} else {
|
|
usersString := map[string]any{
|
|
"User": "@" + strings.Join(usersInChannel, ", @"),
|
|
}
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.user_already_in_channel.app_error", len(usersInChannel), usersString),
|
|
)
|
|
}
|
|
}
|
|
|
|
if len(differentChannels) > 0 {
|
|
for k, v := range differentChannels {
|
|
if len(v) > 10 {
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.successOverflow", map[string]any{
|
|
"FirstUser": "@" + v[0],
|
|
"Others": len(v) - 1,
|
|
"Channel": k,
|
|
}),
|
|
)
|
|
} else {
|
|
usersString := map[string]any{
|
|
"User": "@" + strings.Join(v, ", @"),
|
|
"Channel": k,
|
|
}
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.success", usersString),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(nonTeamUsers) > 0 {
|
|
for k, v := range nonTeamUsers {
|
|
if len(v) > 10 {
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.user_not_in_team.messageOverflow", map[string]any{
|
|
"FirstUser": "@" + v[0],
|
|
"Others": len(v) - 1,
|
|
"Team": k,
|
|
}),
|
|
)
|
|
} else {
|
|
usersString := map[string]any{
|
|
"Users": "@" + strings.Join(v, ", @"),
|
|
"Team": k,
|
|
}
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.user_not_in_team.app_error", usersString),
|
|
)
|
|
}
|
|
}
|
|
}
|
|
|
|
if len(channelConstrained) > 0 {
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.channel_constrained_user_denied"),
|
|
)
|
|
}
|
|
|
|
if len(errorUsers) > 0 {
|
|
*resps = append(*resps,
|
|
args.T("api.command_invite.fail.app_error"),
|
|
)
|
|
}
|
|
|
|
if len(*resps) > 0 {
|
|
return strings.Join(*resps, "\n")
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func (i *InviteProvider) getUsersFromMentionName(a *app.App, mentionName string) ([]*model.User, *model.Group) {
|
|
userProfile, err := a.Srv().Store().User().GetByUsername(mentionName)
|
|
if err == nil && userProfile.DeleteAt == 0 {
|
|
return []*model.User{userProfile}, nil
|
|
}
|
|
|
|
group, appErr := a.GetGroupByName(mentionName, model.GroupSearchOpts{FilterAllowReference: true})
|
|
if appErr != nil || group == nil {
|
|
return nil, nil
|
|
}
|
|
|
|
members, appErr := a.GetGroupMemberUsers(group.Id)
|
|
if appErr != nil {
|
|
return nil, nil
|
|
}
|
|
|
|
return members, group
|
|
}
|
|
|
|
func (i *InviteProvider) parseMessage(a *app.App, rctx request.CTX, args *model.CommandArgs, resps *[]string, message string) ([]*model.User, []*model.Channel, string) {
|
|
splitMessage := strings.Split(message, " ")
|
|
|
|
targetUsers := make([]*model.User, 0, 1)
|
|
targetChannels := make([]*model.Channel, 0)
|
|
|
|
for j, msg := range splitMessage {
|
|
if msg == "" {
|
|
continue
|
|
}
|
|
|
|
if msg[0] == '@' || (msg[0] != '~' && j == 0) {
|
|
targetMentionName := strings.TrimPrefix(msg, "@")
|
|
users, _ := i.getUsersFromMentionName(a, targetMentionName)
|
|
if len(users) == 0 {
|
|
*resps = append(*resps, args.T("api.command_invite.missing_user.app_error", map[string]any{
|
|
"User": targetMentionName,
|
|
}))
|
|
continue
|
|
}
|
|
targetUsers = append(targetUsers, users...)
|
|
} else {
|
|
targetChannelName := strings.TrimPrefix(msg, "~")
|
|
channelToJoin, err := a.GetChannelByName(rctx, targetChannelName, args.TeamId, false)
|
|
if err != nil {
|
|
*resps = append(*resps, args.T("api.command_invite.channel.error", map[string]any{
|
|
"Channel": targetChannelName,
|
|
}))
|
|
continue
|
|
}
|
|
targetChannels = append(targetChannels, channelToJoin)
|
|
}
|
|
}
|
|
|
|
if len(targetUsers) == 0 {
|
|
if len(*resps) != 0 {
|
|
return nil, nil, strings.Join(*resps, "\n")
|
|
}
|
|
return nil, nil, args.T("api.command_invite.missing_message.app_error")
|
|
}
|
|
|
|
if len(targetChannels) == 0 {
|
|
if len(*resps) != 0 {
|
|
return nil, nil, strings.Join(*resps, "\n")
|
|
}
|
|
|
|
channelToJoin, err := a.GetChannel(rctx, args.ChannelId)
|
|
if err != nil {
|
|
return nil, nil, args.T("api.command_invite.channel.app_error")
|
|
}
|
|
targetChannels = append(targetChannels, channelToJoin)
|
|
}
|
|
|
|
return targetUsers, targetChannels, ""
|
|
}
|
|
|
|
func (i *InviteProvider) checkPermissions(a *app.App, rctx request.CTX, args *model.CommandArgs, resps *[]string, targetUser *model.User, targetChannels []*model.Channel) []*model.Channel {
|
|
var err *model.AppError
|
|
validChannels := make([]*model.Channel, 0, len(targetChannels))
|
|
for _, targetChannel := range targetChannels {
|
|
switch targetChannel.Type {
|
|
case model.ChannelTypeOpen:
|
|
if !a.HasPermissionToChannel(rctx, args.UserId, targetChannel.Id, model.PermissionManagePublicChannelMembers) {
|
|
*resps = append(*resps, args.T("api.command_invite.permission.app_error", map[string]any{
|
|
"User": targetUser.Username,
|
|
"Channel": targetChannel.Name,
|
|
}))
|
|
continue
|
|
}
|
|
case model.ChannelTypePrivate:
|
|
if !a.HasPermissionToChannel(rctx, args.UserId, targetChannel.Id, model.PermissionManagePrivateChannelMembers) {
|
|
if _, err = a.GetChannelMember(rctx, targetChannel.Id, args.UserId); err == nil {
|
|
// User doing the inviting is a member of the channel.
|
|
*resps = append(*resps, args.T("api.command_invite.permission.app_error", map[string]any{
|
|
"User": targetUser.Username,
|
|
"Channel": targetChannel.Name,
|
|
}))
|
|
continue
|
|
}
|
|
// User doing the inviting is *not* a member of the channel.
|
|
*resps = append(*resps, args.T("api.command_invite.private_channel.app_error", map[string]any{
|
|
"Channel": targetChannel.Name,
|
|
}))
|
|
continue
|
|
}
|
|
default:
|
|
*resps = append(*resps, args.T("api.command_invite.directchannel.app_error"))
|
|
continue
|
|
}
|
|
validChannels = append(validChannels, targetChannel)
|
|
}
|
|
return validChannels
|
|
}
|
|
|
|
func (i *InviteProvider) addUserToChannel(a *app.App, rctx request.CTX, args *model.CommandArgs, userProfile *model.User, channelToJoin *model.Channel) UserError {
|
|
// Check if user is already in the channel
|
|
_, err := a.GetChannelMember(rctx, channelToJoin.Id, userProfile.Id)
|
|
if err == nil {
|
|
return UserInChannel
|
|
}
|
|
|
|
if _, err = a.AddChannelMember(rctx, userProfile.Id, channelToJoin, app.ChannelMemberOpts{UserRequestorID: args.UserId}); err != nil {
|
|
if err.Id == "api.channel.add_members.user_denied" {
|
|
return IsConstrained
|
|
} else if err.Id == "app.team.get_member.missing.app_error" ||
|
|
err.Id == "api.channel.add_user.to.channel.failed.deleted.app_error" {
|
|
return UserNotInTeam
|
|
}
|
|
rctx.Logger().Warn("addUserToChannel had unexpected error.", mlog.String("UserId", userProfile.Id), mlog.Err(err))
|
|
return Unknown
|
|
}
|
|
|
|
return NoError
|
|
}
|