Impl DeleteUserBlog

Author: gedorinku

app/server/user_blogs_server.go

@@ -15,6 +15,7 @@ import (
 	"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"
 )
 
 // UserBlogServiceServer is a composite interface of api_pb.UserBlogServiceServer and grapiserver.Server.
@@ -88,6 +89,11 @@ func (s *userBlogServiceServerImpl) UpdateUserBlog(ctx context.Context, req *api
 	}
 
 	bs := s.UserBlogStore(ctx)
+
+	if err := s.canWrite(ctx, userID, b.ID); err != nil {
+		return nil, err
+	}
+
 	err = bs.UpdateUserBlog(b)
 	if err != nil {
 		if errors.Cause(err) == sql.ErrNoRows {
@@ -100,8 +106,39 @@ func (s *userBlogServiceServerImpl) UpdateUserBlog(ctx context.Context, req *api
 }
 
 func (s *userBlogServiceServerImpl) DeleteUserBlog(ctx context.Context, req *api_pb.DeleteUserBlogRequest) (*empty.Empty, error) {
-	// TODO: Not yet implemented.
-	return nil, status.Error(codes.Unimplemented, "TODO: You should implement it!")
+	userID, ok := interceptor.GetCurrentUserID(ctx)
+	if !ok {
+		return nil, util.ErrUnauthenticated
+	}
+
+	blogID := int64(req.GetBlogId())
+	bs := s.UserBlogStore(ctx)
+
+	if err := s.canWrite(ctx, userID, blogID); err != nil {
+		return nil, err
+	}
+
+	err := bs.DeleteUserBlog(blogID)
+	if err != nil {
+		return nil, err
+	}
+
+	return &empty.Empty{}, nil
+}
+
+func (s *userBlogServiceServerImpl) canWrite(ctx context.Context, userID model.UserID, blogID int64) error {
+	bs := s.UserBlogStore(ctx)
+	b, err := bs.GetUserBlog(int64(blogID))
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return util.ErrNotFound
+		}
+		return err
+	}
+	if b.UserID != int64(userID) {
+		return util.ErrNotFound
+	}
+	return nil
 }
 
 type blogRequest interface {

infra/store/user_blog/user_blog_store.go

@@ -6,6 +6,7 @@ import (
 
 	"github.com/pkg/errors"
 	"github.com/volatiletech/sqlboiler/boil"
+	"github.com/volatiletech/sqlboiler/queries/qm"
 
 	"github.com/ProgrammingLab/prolab-accounts/app/util"
 	"github.com/ProgrammingLab/prolab-accounts/infra/record"
@@ -25,6 +26,14 @@ func NewUserBlogStore(ctx context.Context, db *sql.DB) store.UserBlogStore {
 	}
 }
 
+func (s *userBlogStoreImpl) GetUserBlog(blogID int64) (*record.Blog, error) {
+	b, err := record.FindBlog(s.ctx, s.db, int64(blogID))
+	if err != nil {
+		return nil, errors.WithStack(err)
+	}
+	return b, nil
+}
+
 func (s *userBlogStoreImpl) CreateUserBlog(blog *record.Blog) error {
 	blog.ID = 0
 	err := blog.Insert(s.ctx, s.db, boil.Infer())
@@ -72,5 +81,10 @@ func (s *userBlogStoreImpl) UpdateUserBlog(blog *record.Blog) error {
 }
 
 func (s *userBlogStoreImpl) DeleteUserBlog(blogID int64) error {
-	panic("not implemented")
+	_, err := record.Blogs(qm.Where("id = ?", blogID)).DeleteAll(s.ctx, s.db)
+	if err != nil {
+		return errors.WithStack(err)
+	}
+
+	return nil
 }

infra/store/user_blog_store.go

@@ -6,6 +6,7 @@ import (
 
 // UserBlogStore accesses users data
 type UserBlogStore interface {
+	GetUserBlog(blogID int64) (*record.Blog, error)
 	CreateUserBlog(blog *record.Blog) error
 	UpdateUserBlog(blog *record.Blog) error
 	DeleteUserBlog(blogID int64) error