feat(invite): more progress on completing invite form and actually creating a real user, #2353

pkg/api/api.go

@@ -45,6 +45,7 @@ func Register(r *macaron.Macaron) {
 
 	// invited
 	r.Get("/api/user/invite/:code", wrap(GetInviteInfoByCode))
+	r.Post("/api/user/invite/complete", bind(dtos.CompleteInviteForm{}), wrap(CompleteInvite))
 
 	// reset password
 	r.Get("/user/password/send-reset-email", Index)
@@ -97,7 +98,7 @@ func Register(r *macaron.Macaron) {
 			// invites
 			r.Get("/invites", wrap(GetPendingOrgInvites))
 			r.Post("/invites", bind(dtos.AddInviteForm{}), wrap(AddOrgInvite))
-			r.Patch("/invites/:id/revoke", wrap(RevokeInvite))
+			r.Patch("/invites/:code/revoke", wrap(RevokeInvite))
 		}, regOrgAdmin)
 
 		// create new org

pkg/api/dtos/invite.go

@@ -14,3 +14,12 @@ type InviteInfo struct {
 	Name     string `json:"name"`
 	Username string `json:"username"`
 }
+
+type CompleteInviteForm struct {
+	InviteCode      string `json:"inviteCode"`
+	Email           string `json:"email" binding:"Required"`
+	Name            string `json:"name"`
+	Username        string `json:"username"`
+	Password        string `json:"password"`
+	ConfirmPassword string `json:"confirmPassword"`
+}

pkg/api/org_invite.go

@@ -3,6 +3,8 @@ package api
 import (
 	"github.com/grafana/grafana/pkg/api/dtos"
 	"github.com/grafana/grafana/pkg/bus"
+	"github.com/grafana/grafana/pkg/events"
+	"github.com/grafana/grafana/pkg/metrics"
 	"github.com/grafana/grafana/pkg/middleware"
 	m "github.com/grafana/grafana/pkg/models"
 	"github.com/grafana/grafana/pkg/setting"
@@ -85,8 +87,7 @@ func AddOrgInvite(c *middleware.Context, inviteDto dtos.AddInviteForm) Response
 
 func RevokeInvite(c *middleware.Context) Response {
 	cmd := m.UpdateTempUserStatusCommand{
-		Id:     c.ParamsInt64(":id"),
-		OrgId:  c.OrgId,
+		Code:   c.Params(":code"),
 		Status: m.TmpUserRevoked,
 	}
 
@@ -98,7 +99,7 @@ func RevokeInvite(c *middleware.Context) Response {
 }
 
 func GetInviteInfoByCode(c *middleware.Context) Response {
-	query := m.GetTempUsersByCodeQuery{Code: c.Params(":code")}
+	query := m.GetTempUserByCodeQuery{Code: c.Params(":code")}
 
 	if err := bus.Dispatch(&query); err != nil {
 		if err == m.ErrTempUserNotFound {
@@ -115,3 +116,48 @@ func GetInviteInfoByCode(c *middleware.Context) Response {
 
 	return Json(200, &info)
 }
+
+func CompleteInvite(c *middleware.Context, completeInvite dtos.CompleteInviteForm) Response {
+	query := m.GetTempUserByCodeQuery{Code: completeInvite.InviteCode}
+
+	if err := bus.Dispatch(&query); err != nil {
+		if err == m.ErrTempUserNotFound {
+			return ApiError(404, "Invite not found", nil)
+		}
+		return ApiError(500, "Failed to get invite", err)
+	}
+
+	invite := query.Result
+
+	cmd := m.CreateUserCommand{
+		Email:    completeInvite.Email,
+		Name:     completeInvite.Name,
+		Login:    completeInvite.Username,
+		Password: completeInvite.Password,
+	}
+
+	if err := bus.Dispatch(&cmd); err != nil {
+		return ApiError(500, "failed to create user", err)
+	}
+
+	user := cmd.Result
+
+	bus.Publish(&events.UserSignedUp{
+		Id:    user.Id,
+		Name:  user.Name,
+		Email: user.Email,
+		Login: user.Login,
+	})
+
+	// update temp user status
+	updateTmpUserCmd := m.UpdateTempUserStatusCommand{Code: invite.Code, Status: m.TmpUserCompleted}
+	if err := bus.Dispatch(&updateTmpUserCmd); err != nil {
+		return ApiError(500, "Failed to update invite status", err)
+	}
+
+	loginUserWithUser(&user, c)
+
+	metrics.M_Api_User_SignUp.Inc(1)
+
+	return ApiSuccess("User created and logged in")
+}

pkg/models/temp_user.go

@@ -56,8 +56,7 @@ type CreateTempUserCommand struct {
 }
 
 type UpdateTempUserStatusCommand struct {
-	Id     int64
-	OrgId  int64
+	Code   string
 	Status TempUserStatus
 }
 
@@ -68,7 +67,7 @@ type GetTempUsersForOrgQuery struct {
 	Result []*TempUserDTO
 }
 
-type GetTempUsersByCodeQuery struct {
+type GetTempUserByCodeQuery struct {
 	Code string
 
 	Result *TempUser

pkg/services/sqlstore/temp_user.go

@@ -12,13 +12,13 @@ func init() {
 	bus.AddHandler("sql", CreateTempUser)
 	bus.AddHandler("sql", GetTempUsersForOrg)
 	bus.AddHandler("sql", UpdateTempUserStatus)
-	bus.AddHandler("sql", GetTempUsersByCode)
+	bus.AddHandler("sql", GetTempUserByCode)
 }
 
 func UpdateTempUserStatus(cmd *m.UpdateTempUserStatusCommand) error {
 	return inTransaction(func(sess *xorm.Session) error {
-		var rawSql = "UPDATE temp_user SET status=? WHERE id=? and org_id=?"
-		_, err := sess.Exec(rawSql, string(cmd.Status), cmd.Id, cmd.OrgId)
+		var rawSql = "UPDATE temp_user SET status=? WHERE code=?"
+		_, err := sess.Exec(rawSql, string(cmd.Status), cmd.Code)
 		return err
 	})
 }
@@ -70,7 +70,7 @@ func GetTempUsersForOrg(query *m.GetTempUsersForOrgQuery) error {
 	return err
 }
 
-func GetTempUsersByCode(query *m.GetTempUsersByCodeQuery) error {
+func GetTempUserByCode(query *m.GetTempUserByCodeQuery) error {
 	var user m.TempUser
 	has, err := x.Table("temp_user").Where("code=?", query.Code).Get(&user)
 

pkg/services/sqlstore/temp_user_test.go

@@ -17,6 +17,7 @@ func TestTempUserCommandsAndQueries(t *testing.T) {
 			cmd := m.CreateTempUserCommand{
 				OrgId:  2256,
 				Name:   "hello",
+				Code:   "asd",
 				Email:  "e@as.co",
 				Status: m.TmpUserInvitePending,
 			}
@@ -32,7 +33,7 @@ func TestTempUserCommandsAndQueries(t *testing.T) {
 			})
 
 			Convey("Should be able update status", func() {
-				cmd2 := m.UpdateTempUserStatusCommand{OrgId: 2256, Status: m.TmpUserRevoked, Id: cmd.Result.Id}
+				cmd2 := m.UpdateTempUserStatusCommand{Code: "asd", Status: m.TmpUserRevoked}
 				err := UpdateTempUserStatus(&cmd2)
 				So(err, ShouldBeNil)
 			})