github.com/spline-fu/mattermost-server@v4.10.10+incompatible/app/command_invite.go (about)

     1  // Copyright (c) 2015-present Mattermost, Inc. All Rights Reserved.
     2  // See License.txt for license information.
     3  
     4  package app
     5  
     6  import (
     7  	"strings"
     8  
     9  	"github.com/mattermost/mattermost-server/mlog"
    10  	"github.com/mattermost/mattermost-server/model"
    11  	goi18n "github.com/nicksnyder/go-i18n/i18n"
    12  )
    13  
    14  type InviteProvider struct {
    15  }
    16  
    17  const (
    18  	CMD_INVITE = "invite"
    19  )
    20  
    21  func init() {
    22  	RegisterCommandProvider(&InviteProvider{})
    23  }
    24  
    25  func (me *InviteProvider) GetTrigger() string {
    26  	return CMD_INVITE
    27  }
    28  
    29  func (me *InviteProvider) GetCommand(a *App, T goi18n.TranslateFunc) *model.Command {
    30  	return &model.Command{
    31  		Trigger:          CMD_INVITE,
    32  		AutoComplete:     true,
    33  		AutoCompleteDesc: T("api.command_invite.desc"),
    34  		AutoCompleteHint: T("api.command_invite.hint"),
    35  		DisplayName:      T("api.command_invite.name"),
    36  	}
    37  }
    38  
    39  func (me *InviteProvider) DoCommand(a *App, args *model.CommandArgs, message string) *model.CommandResponse {
    40  	if message == "" {
    41  		return &model.CommandResponse{Text: args.T("api.command_invite.missing_message.app_error"), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    42  	}
    43  
    44  	splitMessage := strings.SplitN(message, " ", 2)
    45  	targetUsername := splitMessage[0]
    46  	targetUsername = strings.TrimPrefix(targetUsername, "@")
    47  
    48  	var userProfile *model.User
    49  	if result := <-a.Srv.Store.User().GetByUsername(targetUsername); result.Err != nil {
    50  		mlog.Error(result.Err.Error())
    51  		return &model.CommandResponse{Text: args.T("api.command_invite.missing_user.app_error"), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    52  	} else {
    53  		userProfile = result.Data.(*model.User)
    54  	}
    55  
    56  	var channelToJoin *model.Channel
    57  	var err *model.AppError
    58  	// User set a channel to add the invited user
    59  	if len(splitMessage) > 1 && splitMessage[1] != "" {
    60  		targetChannelName := strings.TrimPrefix(strings.TrimSpace(splitMessage[1]), "~")
    61  
    62  		if channelToJoin, err = a.GetChannelByName(targetChannelName, args.TeamId); err != nil {
    63  			return &model.CommandResponse{Text: args.T("api.command_invite.channel.error", map[string]interface{}{"Channel": targetChannelName}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    64  		}
    65  	} else {
    66  		channelToJoin, err = a.GetChannel(args.ChannelId)
    67  		if err != nil {
    68  			return &model.CommandResponse{Text: args.T("api.command_invite.channel.app_error"), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    69  		}
    70  	}
    71  
    72  	// Check if is a Direct Channel
    73  	if channelToJoin.Type == model.CHANNEL_DIRECT || channelToJoin.Type == model.CHANNEL_GROUP {
    74  		return &model.CommandResponse{Text: args.T("api.command_invite.directchannel.app_error"), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    75  	}
    76  
    77  	// Check Permissions
    78  	if channelToJoin.Type == model.CHANNEL_OPEN && !a.SessionHasPermissionToChannel(args.Session, channelToJoin.Id, model.PERMISSION_MANAGE_PUBLIC_CHANNEL_MEMBERS) {
    79  		return &model.CommandResponse{Text: args.T("api.command_invite.permission.app_error", map[string]interface{}{"User": userProfile.Username, "Channel": channelToJoin.Name}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    80  	}
    81  
    82  	// Check if the user who wants to add another is trying to add in a pvt channel, but does not have permission
    83  	// but is in the channel
    84  	_, err = a.GetChannelMember(channelToJoin.Id, args.UserId)
    85  	if channelToJoin.Type == model.CHANNEL_PRIVATE && !a.SessionHasPermissionToChannel(args.Session, channelToJoin.Id, model.PERMISSION_MANAGE_PRIVATE_CHANNEL_MEMBERS) && err == nil {
    86  		return &model.CommandResponse{Text: args.T("api.command_invite.permission.app_error", map[string]interface{}{"User": userProfile.Username, "Channel": channelToJoin.Name}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    87  	}
    88  
    89  	// In this case just check if is a pvt channel and user has permission
    90  	if channelToJoin.Type == model.CHANNEL_PRIVATE && !a.SessionHasPermissionToChannel(args.Session, channelToJoin.Id, model.PERMISSION_MANAGE_PRIVATE_CHANNEL_MEMBERS) {
    91  		return &model.CommandResponse{Text: args.T("api.command_invite.private_channel.app_error", map[string]interface{}{"Channel": channelToJoin.Name}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    92  	}
    93  
    94  	// Check if user is already in the channel
    95  	_, err = a.GetChannelMember(channelToJoin.Id, userProfile.Id)
    96  	if err == nil {
    97  		return &model.CommandResponse{Text: args.T("api.command_invite.user_already_in_channel.app_error", map[string]interface{}{"User": userProfile.Username}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
    98  	}
    99  
   100  	if _, err := a.AddChannelMember(userProfile.Id, channelToJoin, args.Session.UserId, ""); err != nil {
   101  		return &model.CommandResponse{Text: args.T("api.command_invite.fail.app_error"), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
   102  	}
   103  
   104  	if args.ChannelId != channelToJoin.Id {
   105  		return &model.CommandResponse{Text: args.T("api.command_invite.success", map[string]interface{}{"User": userProfile.Username, "Channel": channelToJoin.Name}), ResponseType: model.COMMAND_RESPONSE_TYPE_EPHEMERAL}
   106  	}
   107  
   108  	return &model.CommandResponse{}
   109  }