Impl CreateInvitation

Author: gedorinku

api/entries.validator.pb.go

@@ -121,9 +121,6 @@
           "type": "integer",
           "format": "int64"
         },
-        "token": {
-          "type": "string"
-        },
         "email": {
           "type": "string"
         }

api/invitations.pb.go

@@ -8,6 +8,8 @@ option go_package = "github.com/ProgrammingLab/prolab-accounts/api;api_pb";
 import "google/api/annotations.proto";
 import "google/protobuf/empty.proto";
 
+import "users.proto";
+
 service InvitationService {
   rpc ListInvitations (ListInvitationsRequest) returns (ListInvitationsResponse) {
     option (google.api.http) = {
@@ -34,8 +36,7 @@ service InvitationService {
 
 message Invitation {
   uint32 invitation_id = 1;
-  string token = 2;
-  string email= 3; 
+  string email= 2;
 }
 
 message ListInvitationsRequest {

api/invitations.swagger.json

@@ -15,6 +15,7 @@ import (
 	entrystore "github.com/ProgrammingLab/prolab-accounts/infra/store/entry"
 	feedstore "github.com/ProgrammingLab/prolab-accounts/infra/store/feed"
 	heartbeatstore "github.com/ProgrammingLab/prolab-accounts/infra/store/heartbeat"
+	invitationstore "github.com/ProgrammingLab/prolab-accounts/infra/store/invitation"
 	profilestore "github.com/ProgrammingLab/prolab-accounts/infra/store/profile"
 	rolestore "github.com/ProgrammingLab/prolab-accounts/infra/store/role"
 	sessionstore "github.com/ProgrammingLab/prolab-accounts/infra/store/session"
@@ -33,6 +34,7 @@ type StoreComponent interface {
 	RoleStore(ctx context.Context) store.RoleStore
 	HeartbeatStore(ctx context.Context) store.HeartbeatStore
 	DepartmentStore(ctx context.Context) store.DepartmentStore
+	InvitationStore(ctx context.Context) store.InvitationStore
 }
 
 // NewStoreComponent returns new store component
@@ -169,3 +171,7 @@ func (s *storeComponentImpl) DepartmentStore(ctx context.Context) store.Departme
 func (s *storeComponentImpl) HeartbeatStore(ctx context.Context) store.HeartbeatStore {
 	return heartbeatstore.NewHeartbeatStore(ctx, s.client, s.cfg)
 }
+
+func (s *storeComponentImpl) InvitationStore(ctx context.Context) store.InvitationStore {
+	return invitationstore.NewInvitationStore(ctx, s.db)
+}

api/protos/invitations.proto

@@ -2,14 +2,20 @@ package server
 
 import (
 	"context"
+	"database/sql"
 
 	"github.com/golang/protobuf/ptypes/empty"
 	"github.com/izumin5210/grapi/pkg/grapiserver"
+	"github.com/pkg/errors"
 	"google.golang.org/grpc/codes"
 	"google.golang.org/grpc/status"
 
 	api_pb "github.com/ProgrammingLab/prolab-accounts/api"
 	"github.com/ProgrammingLab/prolab-accounts/app/di"
+	"github.com/ProgrammingLab/prolab-accounts/app/interceptor"
+	"github.com/ProgrammingLab/prolab-accounts/app/util"
+	"github.com/ProgrammingLab/prolab-accounts/infra/record"
+	"github.com/ProgrammingLab/prolab-accounts/model"
 )
 
 // InvitationServiceServer is a composite interface of api_pb.InvitationServiceServer and grapiserver.Server.
@@ -26,6 +32,11 @@ func NewInvitationServiceServer(store di.StoreComponent, cli di.ClientComponent)
 	}
 }
 
+var (
+	// ErrEmailAlreadyInUse is returned when email is already in use
+	ErrEmailAlreadyInUse = status.Error(codes.AlreadyExists, "email is already in use")
+)
+
 type invitationServiceServerImpl struct {
 	di.StoreComponent
 	di.ClientComponent
@@ -42,11 +53,61 @@ func (s *invitationServiceServerImpl) GetInvitation(ctx context.Context, req *ap
 }
 
 func (s *invitationServiceServerImpl) CreateInvitation(ctx context.Context, req *api_pb.CreateInvitationRequest) (*api_pb.Invitation, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	admin, err := s.getAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	us := s.UserStore(ctx)
+	_, err = us.GetUserByEmail(req.GetEmail())
+	if err == nil {
+		return nil, ErrEmailAlreadyInUse
+	}
+	if errors.Cause(err) != sql.ErrNoRows {
+		return nil, err
+	}
+
+	is := s.InvitationStore(ctx)
+	inv, err := is.CreateInvitation(model.UserID(admin.ID), req.GetEmail())
+	if err != nil {
+		return nil, err
+	}
+
+	sender := s.EmailSender(ctx)
+	err = sender.SendInvitationEmail(inv)
+	if err != nil {
+		return nil, err
+	}
+
+	return invitationToResponse(inv), nil
 }
 
 func (s *invitationServiceServerImpl) DeleteInvitation(ctx context.Context, req *api_pb.DeleteInvitationRequest) (*empty.Empty, error) {
 	// TODO: Not yet implemented.
 	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
 }
+
+func (s *invitationServiceServerImpl) getAdmin(ctx context.Context) (*record.User, error) {
+	userID, ok := interceptor.GetCurrentUserID(ctx)
+	if !ok {
+		return nil, util.ErrUnauthenticated
+	}
+
+	us := s.UserStore(ctx)
+	u, err := us.GetUserWithPrivate(userID)
+	if err != nil {
+		return nil, err
+	}
+
+	if u.Authority != int(model.Admin) {
+		return nil, util.ErrUnauthenticated
+	}
+	return u, nil
+}
+
+func invitationToResponse(inv *record.Invitation) *api_pb.Invitation {
+	return &api_pb.Invitation{
+		InvitationId: uint32(inv.ID),
+		Email:        inv.Email,
+	}
+}

app/di/store_component.go

@@ -46,6 +46,11 @@ func (s *invitationStoreImpl) CreateInvitation(inviter model.UserID, email strin
 		return nil, err
 	}
 
+	_, err = record.Invitations(record.InvitationWhere.Email.EQ(email)).DeleteAll(s.ctx, s.db)
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
 	inv := &record.Invitation{
 		Code:      code,
 		Email:     email,

app/server/invitations_server.go

@@ -60,6 +60,18 @@ func (s *userStoreImpl) GetPublicUserByName(name string) (*record.User, error) {
 	return u, nil
 }
 
+func (s *userStoreImpl) GetUserByEmail(email string) (*record.User, error) {
+	mods := []qm.QueryMod{
+		record.UserWhere.Email.EQ(email),
+	}
+	u, err := record.Users(mods...).One(s.ctx, s.db)
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	return u, nil
+}
+
 func (s *userStoreImpl) GetUserWithPrivate(userID model.UserID) (*record.User, error) {
 	mods := []qm.QueryMod{
 		qm.Load("Profile.Role"),

infra/store/invitation/invitation_store.go

@@ -9,6 +9,7 @@ import (
 type UserStore interface {
 	CreateUser(user *record.User) error
 	GetPublicUserByName(name string) (*record.User, error)
+	GetUserByEmail(email string) (*record.User, error)
 	GetUserWithPrivate(userID model.UserID) (*record.User, error)
 	ListPublicUsers(minUserID model.UserID, limit int) ([]*record.User, model.UserID, error)
 	UpdateFullName(userID model.UserID, fullName string) (*record.User, error)