Skip to content

Commit

Permalink
Merge branch 'master' into PLT-5376
Browse files Browse the repository at this point in the history
  • Loading branch information
andreistanciu24 committed Jan 31, 2017
2 parents 8835db3 + 67739cb commit 8963390
Show file tree
Hide file tree
Showing 15 changed files with 735 additions and 62 deletions.
2 changes: 1 addition & 1 deletion api/team_test.go
Expand Up @@ -45,7 +45,7 @@ func TestCreateTeam(t *testing.T) {

rteam.Data.(*model.Team).Id = ""
if _, err := Client.CreateTeam(rteam.Data.(*model.Team)); err != nil {
if err.Message != "A team with that domain already exists" {
if err.Message != "A team with that name already exists" {
t.Fatal(err)
}
}
Expand Down
1 change: 1 addition & 0 deletions api4/api.go
Expand Up @@ -138,6 +138,7 @@ func InitApi(full bool) {
BaseRoutes.Webrtc = BaseRoutes.ApiRoot.PathPrefix("/webrtc").Subrouter()

InitUser()
InitTeam()

// REMOVE CONDITION WHEN APIv3 REMOVED
if full {
Expand Down
42 changes: 42 additions & 0 deletions api4/team.go
@@ -0,0 +1,42 @@
// Copyright (c) 2017 Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package api4

import (
"net/http"

l4g "github.com/alecthomas/log4go"
"github.com/mattermost/platform/app"
"github.com/mattermost/platform/model"
"github.com/mattermost/platform/utils"
)

func InitTeam() {
l4g.Debug(utils.T("api.team.init.debug"))

BaseRoutes.Teams.Handle("", ApiSessionRequired(createTeam)).Methods("POST")

}

func createTeam(c *Context, w http.ResponseWriter, r *http.Request) {
team := model.TeamFromJson(r.Body)
if team == nil {
c.SetInvalidParam("team")
return
}

if !app.SessionHasPermissionTo(c.Session, model.PERMISSION_CREATE_TEAM) {
c.Err = model.NewAppError("createTeam", "api.team.is_team_creation_allowed.disabled.app_error", nil, "", http.StatusForbidden)
return
}

rteam, err := app.CreateTeamWithUser(team, c.Session.UserId)
if err != nil {
c.Err = err
return
}

w.WriteHeader(http.StatusCreated)
w.Write([]byte(rteam.ToJson()))
}
75 changes: 75 additions & 0 deletions api4/team_test.go
@@ -0,0 +1,75 @@
// Copyright (c) 2017 Mattermost, Inc. All Rights Reserved.
// See License.txt for license information.

package api4

import (
"net/http"
"strconv"
"testing"

"github.com/mattermost/platform/model"
"github.com/mattermost/platform/utils"
)

func TestCreateTeam(t *testing.T) {
th := Setup().InitBasic()
Client := th.Client

team := &model.Team{Name: GenerateTestUsername(), DisplayName: "Some Team", Type: model.TEAM_OPEN}
rteam, resp := Client.CreateTeam(team)
CheckNoError(t, resp)

if rteam.Name != team.Name {
t.Fatal("names did not match")
}

if rteam.DisplayName != team.DisplayName {
t.Fatal("display names did not match")
}

if rteam.Type != team.Type {
t.Fatal("types did not match")
}

_, resp = Client.CreateTeam(rteam)
CheckBadRequestStatus(t, resp)

rteam.Id = ""
_, resp = Client.CreateTeam(rteam)
CheckErrorMessage(t, resp, "A team with that name already exists")
CheckBadRequestStatus(t, resp)

rteam.Name = ""
_, resp = Client.CreateTeam(rteam)
CheckErrorMessage(t, resp, "Name must be 2 or more lowercase alphanumeric characters")
CheckBadRequestStatus(t, resp)

if r, err := Client.DoApiPost("/teams", "garbage"); err == nil {
t.Fatal("should have errored")
} else {
if r.StatusCode != http.StatusBadRequest {
t.Log("actual: " + strconv.Itoa(r.StatusCode))
t.Log("expected: " + strconv.Itoa(http.StatusBadRequest))
t.Fatal("wrong status code")
}
}

Client.Logout()

_, resp = Client.CreateTeam(rteam)
CheckUnauthorizedStatus(t, resp)

// Update permission
enableTeamCreation := utils.Cfg.TeamSettings.EnableTeamCreation
defer func() {
utils.Cfg.TeamSettings.EnableTeamCreation = enableTeamCreation
}()
utils.Cfg.TeamSettings.EnableTeamCreation = false
utils.SetDefaultRolesBasedOnConfig()

th.LoginBasic()
_, resp = Client.CreateTeam(team)
CheckForbiddenStatus(t, resp)

}
175 changes: 174 additions & 1 deletion app/import.go
Expand Up @@ -15,6 +15,7 @@ import (
l4g "github.com/alecthomas/log4go"
"github.com/mattermost/platform/model"
"github.com/mattermost/platform/utils"
"net/http"
)

// Import Data Models
Expand All @@ -23,6 +24,7 @@ type LineImportData struct {
Type string `json:"type"`
Team *TeamImportData `json:"team"`
Channel *ChannelImportData `json:"channel"`
User *UserImportData `json:"user"`
}

type TeamImportData struct {
Expand All @@ -42,6 +44,19 @@ type ChannelImportData struct {
Purpose *string `json:"purpose"`
}

type UserImportData struct {
Username *string `json:"username"`
Email *string `json:"email"`
AuthService *string `json:"auth_service"`
AuthData *string `json:"auth_data"`
Nickname *string `json:"nickname"`
FirstName *string `json:"first_name"`
LastName *string `json:"last_name"`
Position *string `json:"position"`
Roles *string `json:"roles"`
Locale *string `json:"locale"`
}

//
// -- Bulk Import Functions --
// These functions import data directly into the database. Security and permission checks are bypassed but validity is
Expand Down Expand Up @@ -86,6 +101,12 @@ func ImportLine(line LineImportData, dryRun bool) *model.AppError {
} else {
return ImportChannel(line.Channel, dryRun)
}
case line.Type == "user":
if line.User == nil {
return model.NewAppError("BulkImport", "app.import.import_line.null_user.error", nil, "", http.StatusBadRequest)
} else {
return ImportUser(line.User, dryRun)
}
default:
return model.NewLocAppError("BulkImport", "app.import.import_line.unknown_line_type.error", map[string]interface{}{"Type": line.Type}, "")
}
Expand Down Expand Up @@ -251,6 +272,158 @@ func validateChannelImportData(data *ChannelImportData) *model.AppError {
return nil
}

func ImportUser(data *UserImportData, dryRun bool) *model.AppError {
if err := validateUserImportData(data); err != nil {
return err
}

// If this is a Dry Run, do not continue any further.
if dryRun {
return nil
}

var user *model.User
if result := <-Srv.Store.User().GetByUsername(*data.Username); result.Err == nil {
user = result.Data.(*model.User)
} else {
user = &model.User{}
}

user.Username = *data.Username
user.Email = *data.Email

var password string
var authService string
var authData *string

if data.AuthService != nil {
authService = *data.AuthService
}

// AuthData and Password are mutually exclusive.
if data.AuthData != nil {
authData = data.AuthData
password = ""
} else {
// If no Auth Data is specified, we must generate a password.
password = model.NewId()
authData = nil
}

user.Password = password
user.AuthService = authService
user.AuthData = authData

// Automatically assume all emails are verified.
emailVerified := true
user.EmailVerified = emailVerified

if data.Nickname != nil {
user.Nickname = *data.Nickname
}

if data.FirstName != nil {
user.FirstName = *data.FirstName
}

if data.LastName != nil {
user.LastName = *data.LastName
}

if data.Position != nil {
user.Position = *data.Position
}

if data.Locale != nil {
user.Locale = *data.Locale
} else {
user.Locale = *utils.Cfg.LocalizationSettings.DefaultClientLocale
}

var roles string
if data.Roles != nil {
roles = *data.Roles
} else if len(user.Roles) == 0 {
// Set SYSTEM_USER roles on newly created users by default.
roles = model.ROLE_SYSTEM_USER.Id
}
user.Roles = roles

if user.Id == "" {
if _, err := createUser(user); err != nil {
return err
}
} else {
if _, err := UpdateUser(user, utils.GetSiteURL(), false); err != nil {
return err
}
if _, err := UpdateUserRoles(user.Id, roles); err != nil {
return err
}
if len(password) > 0 {
if err := UpdatePassword(user, password); err != nil {
return err
}
} else {
if res := <-Srv.Store.User().UpdateAuthData(user.Id, authService, authData, user.Email, false); res.Err != nil {
return res.Err
}
}
if emailVerified {
if err := VerifyUserEmail(user.Id); err != nil {
return err
}
}
}

return nil
}

func validateUserImportData(data *UserImportData) *model.AppError {

if data.Username == nil {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.username_missing.error", nil, "", http.StatusBadRequest)
} else if !model.IsValidUsername(*data.Username) {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.username_invalid.error", nil, "", http.StatusBadRequest)
}

if data.Email == nil {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.email_missing.error", nil, "", http.StatusBadRequest)
} else if len(*data.Email) == 0 || len(*data.Email) > model.USER_EMAIL_MAX_LENGTH {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.email_length.error", nil, "", http.StatusBadRequest)
}

if data.AuthService != nil && len(*data.AuthService) == 0 {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.auth_service_length.error", nil, "", http.StatusBadRequest)
}

if data.AuthData != nil && len(*data.AuthData) > model.USER_AUTH_DATA_MAX_LENGTH {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.auth_data_length.error", nil, "", http.StatusBadRequest)
}

if data.Nickname != nil && utf8.RuneCountInString(*data.Nickname) > model.USER_NICKNAME_MAX_RUNES {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.nickname_length.error", nil, "", http.StatusBadRequest)
}

if data.FirstName != nil && utf8.RuneCountInString(*data.FirstName) > model.USER_FIRST_NAME_MAX_RUNES {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.first_name_length.error", nil, "", http.StatusBadRequest)
}

if data.LastName != nil && utf8.RuneCountInString(*data.LastName) > model.USER_LAST_NAME_MAX_RUNES {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.last_name_length.error", nil, "", http.StatusBadRequest)
}

if data.Position != nil && utf8.RuneCountInString(*data.Position) > model.USER_POSITION_MAX_RUNES {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.position_length.error", nil, "", http.StatusBadRequest)
}

if data.Roles != nil && !model.IsValidUserRoles(*data.Roles) {
return model.NewAppError("BulkImport", "app.import.validate_user_import_data.roles_invalid.error", nil, "", http.StatusBadRequest)
}

return nil
}

//
// -- Old SlackImport Functions --
// Import functions are sutible for entering posts and users into the database without
Expand Down Expand Up @@ -288,7 +461,7 @@ func ImportPost(post *model.Post) {
}
}

func ImportUser(team *model.Team, user *model.User) *model.User {
func OldImportUser(team *model.Team, user *model.User) *model.User {
user.MakeNonNil()

user.Roles = model.ROLE_SYSTEM_USER.Id
Expand Down

0 comments on commit 8963390

Please sign in to comment.