PLT-6196: Websocket message when added to a team. (#5968)

2 files changed, 16 insertions, 3 deletions

diff --git a/app/team.go b/app/team.go
index 91601fbb2..8ad5dd466 100644
--- a/app/team.go
+++ b/app/team.go
@@ -436,11 +436,18 @@ func AddTeamMember(teamId, userId, siteURL string) (*model.TeamMember, *model.Ap
 		return nil, err
 	}
 
-	if teamMember, err := GetTeamMember(teamId, userId); err != nil {
+	var teamMember *model.TeamMember
+	var err *model.AppError
+	if teamMember, err = GetTeamMember(teamId, userId); err != nil {
 		return nil, err
-	} else {
-		return teamMember, nil
 	}
+
+	message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil)
+	message.Add("team_id", teamId)
+	message.Add("user_id", userId)
+	Publish(message)
+
+	return teamMember, nil
 }
 
 func AddTeamMembers(teamId string, userIds []string, siteURL string) ([]*model.TeamMember, *model.AppError) {
@@ -456,6 +463,11 @@ func AddTeamMembers(teamId string, userIds []string, siteURL string) ([]*model.T
 		} else {
 			members = append(members, teamMember)
 		}
+
+		message := model.NewWebSocketEvent(model.WEBSOCKET_EVENT_ADDED_TO_TEAM, "", "", userId, nil)
+		message.Add("team_id", teamId)
+		message.Add("user_id", userId)
+		Publish(message)
 	}
 
 	return members, nil
diff --git a/model/websocket_message.go b/model/websocket_message.go
index 6c47eb948..c364f2613 100644
--- a/model/websocket_message.go
+++ b/model/websocket_message.go
@@ -18,6 +18,7 @@ const (
 	WEBSOCKET_EVENT_DIRECT_ADDED       = "direct_added"
 	WEBSOCKET_EVENT_GROUP_ADDED        = "group_added"
 	WEBSOCKET_EVENT_NEW_USER           = "new_user"
+	WEBSOCKET_EVENT_ADDED_TO_TEAM      = "added_to_team"
 	WEBSOCKET_EVENT_LEAVE_TEAM         = "leave_team"
 	WEBSOCKET_EVENT_UPDATE_TEAM        = "update_team"
 	WEBSOCKET_EVENT_USER_ADDED         = "user_added"