Impl InvitationsServiceServer

Author: gedorinku

app/server/invitations_server.go

@@ -43,13 +43,34 @@ type invitationServiceServerImpl struct {
 }
 
 func (s *invitationServiceServerImpl) ListInvitations(ctx context.Context, req *api_pb.ListInvitationsRequest) (*api_pb.ListInvitationsResponse, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	_, err := s.getAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	is := s.InvitationStore(ctx)
+	invs, err := is.ListInvitations()
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+
+	resp := &api_pb.ListInvitationsResponse{
+		Invitations: invitationsToResponse(invs),
+	}
+	return resp, nil
 }
 
 func (s *invitationServiceServerImpl) GetInvitation(ctx context.Context, req *api_pb.GetInvitationRequest) (*api_pb.Invitation, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	is := s.InvitationStore(ctx)
+	inv, err := is.GetInvitation(req.GetToken())
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return nil, util.ErrNotFound
+		}
+		return nil, err
+	}
+
+	return invitationToResponse(inv), nil
 }
 
 func (s *invitationServiceServerImpl) CreateInvitation(ctx context.Context, req *api_pb.CreateInvitationRequest) (*api_pb.Invitation, error) {
@@ -83,8 +104,21 @@ func (s *invitationServiceServerImpl) CreateInvitation(ctx context.Context, req
 }
 
 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!")
+	_, err := s.getAdmin(ctx)
+	if err != nil {
+		return nil, err
+	}
+
+	is := s.InvitationStore(ctx)
+	err = is.DeleteInvitation(int64(req.GetInvitationId()))
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return nil, util.ErrNotFound
+		}
+		return nil, err
+	}
+
+	return &empty.Empty{}, nil
 }
 
 func (s *invitationServiceServerImpl) getAdmin(ctx context.Context) (*record.User, error) {
@@ -105,6 +139,15 @@ func (s *invitationServiceServerImpl) getAdmin(ctx context.Context) (*record.Use
 	return u, nil
 }
 
+func invitationsToResponse(invs []*record.Invitation) []*api_pb.Invitation {
+	resp := make([]*api_pb.Invitation, 0, len(invs))
+	for _, i := range invs {
+		resp = append(resp, invitationToResponse(i))
+	}
+
+	return resp
+}
+
 func invitationToResponse(inv *record.Invitation) *api_pb.Invitation {
 	return &api_pb.Invitation{
 		InvitationId: uint32(inv.ID),

infra/store/invitation/invitation_store.go

@@ -46,8 +46,12 @@ func (s *invitationStoreImpl) ListInvitations() ([]*record.Invitation, error) {
 	return invs, nil
 }
 
-func (s *invitationStoreImpl) GetInvitation(id int64) (*record.Invitation, error) {
-	inv, err := record.FindInvitation(s.ctx, s.db, id)
+func (s *invitationStoreImpl) GetInvitation(code string) (*record.Invitation, error) {
+	mods := []qm.QueryMod{
+		record.InvitationWhere.Code.EQ(code),
+	}
+
+	inv, err := record.Invitations(mods...).One(s.ctx, s.db)
 	if err != nil {
 		return nil, errors.WithStack(err)
 	}
@@ -79,7 +83,10 @@ func (s *invitationStoreImpl) CreateInvitation(inviter model.UserID, email strin
 }
 
 func (s *invitationStoreImpl) DeleteInvitation(id int64) error {
-	_, err := record.Invitations(record.InvitationWhere.ID.EQ(id)).DeleteAll(s.ctx, s.db)
+	n, err := record.Invitations(record.InvitationWhere.ID.EQ(id)).DeleteAll(s.ctx, s.db)
+	if n == 0 {
+		return errors.WithStack(sql.ErrNoRows)
+	}
 	return errors.WithStack(err)
 }
 

infra/store/invitation_store.go

@@ -8,7 +8,7 @@ import (
 // InvitationStore provides invitations
 type InvitationStore interface {
 	ListInvitations() ([]*record.Invitation, error)
-	GetInvitation(id int64) (*record.Invitation, error)
+	GetInvitation(code string) (*record.Invitation, error)
 	CreateInvitation(inviter model.UserID, email string) (*record.Invitation, error)
 	DeleteInvitation(id int64) error
 }