Impl UpdateUserBlog

Author: gedorinku

api/entries.validator.pb.go

@@ -2,9 +2,11 @@ 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"
 
@@ -46,22 +48,9 @@ func (s *userBlogServiceServerImpl) CreateUserBlog(ctx context.Context, req *api
 	}
 
 	blog := req.GetBlog()
-	var feedURL string
-	if req.GetAutoDetectFeed() {
-		fs := s.FeedStore(ctx)
-		u, err := fs.GetFeedURL(blog.GetUrl())
-		if err != nil {
-			return nil, ErrFeedURLDetectAutomatically
-		}
-		feedURL = u
-	} else {
-		u := blog.GetFeedUrl()
-		fs := s.FeedStore(ctx)
-		err := fs.IsValidFeedURL(u)
-		if err != nil {
-			return nil, ErrInvalidFeedURL
-		}
-		feedURL = u
+	feedURL, err := getFeedURL(ctx, s, req)
+	if err != nil {
+		return nil, err
 	}
 
 	b := &record.Blog{
@@ -71,7 +60,7 @@ func (s *userBlogServiceServerImpl) CreateUserBlog(ctx context.Context, req *api
 	}
 
 	bs := s.UserBlogStore(ctx)
-	err := bs.CreateUserBlog(b)
+	err = bs.CreateUserBlog(b)
 	if err != nil {
 		return nil, err
 	}
@@ -80,15 +69,66 @@ func (s *userBlogServiceServerImpl) CreateUserBlog(ctx context.Context, req *api
 }
 
 func (s *userBlogServiceServerImpl) UpdateUserBlog(ctx context.Context, req *api_pb.UpdateUserBlogRequest) (*api_pb.Blog, 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
+	}
+
+	blog := req.GetBlog()
+	feedURL, err := getFeedURL(ctx, s, req)
+	if err != nil {
+		return nil, err
+	}
+
+	b := &record.Blog{
+		ID:      int64(blog.GetBlogId()),
+		URL:     blog.GetUrl(),
+		FeedURL: feedURL,
+		UserID:  int64(userID),
+	}
+
+	bs := s.UserBlogStore(ctx)
+	err = bs.UpdateUserBlog(b)
+	if err != nil {
+		if errors.Cause(err) == sql.ErrNoRows {
+			return nil, util.ErrNotFound
+		}
+		return nil, err
+	}
+
+	return blogToResponse(b), nil
 }
 
 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!")
 }
 
+type blogRequest interface {
+	GetBlog() *api_pb.Blog
+	GetAutoDetectFeed() bool
+}
+
+func getFeedURL(ctx context.Context, s di.StoreComponent, req blogRequest) (string, error) {
+	blog := req.GetBlog()
+	if req.GetAutoDetectFeed() {
+		fs := s.FeedStore(ctx)
+		u, err := fs.GetFeedURL(blog.GetUrl())
+		if err != nil {
+			return "", ErrFeedURLDetectAutomatically
+		}
+		return u, nil
+	}
+
+	u := blog.GetFeedUrl()
+	fs := s.FeedStore(ctx)
+	err := fs.IsValidFeedURL(u)
+	if err != nil {
+		return "", ErrInvalidFeedURL
+	}
+	return u, nil
+}
+
 func blogToResponse(blog *record.Blog) *api_pb.Blog {
 	return &api_pb.Blog{
 		BlogId:  uint32(blog.ID),

api/oauth.validator.pb.go

@@ -10,7 +10,9 @@ import (
 
 var (
 	// ErrUnauthenticated represents unauthenticated error
-	ErrUnauthenticated = status.Error(codes.Unauthenticated, "Unauthenticated")
+	ErrUnauthenticated = status.Error(codes.Unauthenticated, "unauthenticated")
+	// ErrNotFound represents not found error
+	ErrNotFound = status.Error(codes.NotFound, "not found")
 )
 
 // CodeFromHTTPStatus converts corresponding HTTP response status into the gRPC error code.

api/users.validator.pb.go

@@ -7,6 +7,7 @@ import (
 	"github.com/pkg/errors"
 	"github.com/volatiletech/sqlboiler/boil"
 
+	"github.com/ProgrammingLab/prolab-accounts/app/util"
 	"github.com/ProgrammingLab/prolab-accounts/infra/record"
 	"github.com/ProgrammingLab/prolab-accounts/infra/store"
 )
@@ -35,7 +36,39 @@ func (s *userBlogStoreImpl) CreateUserBlog(blog *record.Blog) error {
 }
 
 func (s *userBlogStoreImpl) UpdateUserBlog(blog *record.Blog) error {
-	panic("not implemented")
+	tx, err := s.db.Begin()
+	if err != nil {
+		return errors.WithStack(err)
+	}
+	defer func() {
+		if err = util.ErrorFromRecover(recover()); err != nil {
+			_ = tx.Rollback()
+		}
+	}()
+
+	exists, err := record.FindBlog(s.ctx, tx, blog.ID)
+	if err != nil {
+		_ = tx.Rollback()
+		return errors.WithStack(err)
+	}
+	if exists.UserID != blog.UserID {
+		_ = tx.Rollback()
+		return sql.ErrNoRows
+	}
+
+	_, err = blog.Update(s.ctx, tx, boil.Infer())
+	if err != nil {
+		_ = tx.Rollback()
+		return errors.WithStack(err)
+	}
+
+	err = tx.Commit()
+	if err != nil {
+		_ = tx.Rollback()
+		return errors.WithStack(err)
+	}
+
+	return nil
 }
 
 func (s *userBlogStoreImpl) DeleteUserBlog(blogID int64) error {