Feature/get post (#49)

* Get post feature

- feature added
- untested
- missing: video's thumbnail,
author's avatar url, can_comment

Author: thanhphanphu18@gmail.com

* get_post fix & test

- update the get post router
- fix the form media url function
(missing "post")
- add the cleanup token daemon
- tested - 1000 | OK | get by creator

Author: thanhphanphu18@gmail.com

Author: thanhpp

cmd/laclongquan/start.go

@@ -69,11 +69,12 @@ func start(configPath string) {
 	}
 	logger.Info("http server OK")
 
+	// ------------- Daemons ---------------
 	mainCtx := context.Background()
 	daemonMan := booting.NewDaemonManeger(mainCtx)
 
 	logger.Info("starting daemons....")
-	daemonMan.Start(httpDaemon)
+	daemonMan.Start(httpDaemon, authSrv.DeleteExpiredDaemons())
 	booting.WaitSignals(mainCtx)
 	daemonMan.Stop()
 

internal/laclongquan/application/application.go

@@ -32,6 +32,7 @@ func NewApplication(
 		PostHandler: NewPostHandler(
 			postRepo,
 			saveDir,
+			likeRepo,
 			commentRepo,
 			relationRepo,
 			userRepo,

internal/laclongquan/application/posthandler.go

@@ -12,6 +12,7 @@ import (
 type PostHandler struct {
 	fac          entity.PostFactory
 	repo         repository.PostRepository
+	likeRepo     repository.LikeRepository
 	commentRepo  repository.CommentRepository
 	relationRepo repository.RelationRepository
 	userRepo     repository.UserRepository
@@ -20,13 +21,15 @@ type PostHandler struct {
 }
 
 func NewPostHandler(repo repository.PostRepository, saveDir string,
+	likeRepo repository.LikeRepository,
 	commentRepo repository.CommentRepository,
 	relationRepo repository.RelationRepository,
 	userRepo repository.UserRepository,
 ) PostHandler {
 	return PostHandler{
 		fac:          entity.NewPostFactory(),
 		repo:         repo,
+		likeRepo:     likeRepo,
 		commentRepo:  commentRepo,
 		relationRepo: relationRepo,
 		userRepo:     userRepo,

internal/laclongquan/application/posthandler_getpost.go

@@ -0,0 +1,83 @@
+package application
+
+import (
+	"context"
+
+	"github.com/thanhpp/zola/internal/laclongquan/domain/entity"
+)
+
+type GetPostResult struct {
+	Post         *entity.Post
+	Author       *entity.User
+	LikeCount    int
+	CommentCount int
+	IsLiked      bool
+	CanEdit      bool
+	CanComment   bool
+}
+
+func (p PostHandler) GetPost(ctx context.Context, userID, postID string) (*GetPostResult, error) {
+	user, err := p.userRepo.GetByID(ctx, userID)
+	if err != nil {
+		return nil, err
+	}
+
+	post, err := p.repo.GetByID(ctx, postID)
+	if err != nil {
+		return nil, err
+	}
+
+	// get the relation between the user and the post creator
+	var relation *entity.Relation
+	if userID != post.Creator() {
+		relation, err = p.relationRepo.GetRelationBetween(ctx, userID, post.Creator())
+		if err != nil {
+			return nil, err
+		}
+
+		if relation.IsBlock() {
+			return nil, ErrAlreadyBlocked
+		}
+	}
+
+	// authorization
+	if err := post.CanUserGetPost(user, relation); err != nil {
+		return nil, err
+	}
+
+	// get the author of the post
+	author, err := p.userRepo.GetByID(ctx, post.Creator())
+	if err != nil {
+		return nil, err
+	}
+
+	// like count
+	likeCount, err := p.likeRepo.Count(ctx, postID)
+	if err != nil {
+		return nil, err
+	}
+
+	// comment count
+	commentCount, err := p.commentRepo.CountByPostID(ctx, postID)
+	if err != nil {
+		return nil, err
+	}
+
+	// form response
+	var (
+		canEdit bool
+		// FIXME: canComment bool
+	)
+	if err := post.CanUserEditPost(user); err == nil {
+		canEdit = true
+	}
+
+	return &GetPostResult{
+		Post:         post,
+		Author:       author,
+		CommentCount: commentCount,
+		LikeCount:    likeCount,
+		IsLiked:      p.likeRepo.IsLiked(ctx, userID, postID),
+		CanEdit:      canEdit,
+	}, nil
+}

internal/laclongquan/domain/entity/post.go

@@ -1,17 +1,25 @@
 package entity
 
 import (
+	"errors"
 	"strings"
+	"time"
 
 	"github.com/google/uuid"
 )
 
+var (
+	ErrNilUser = errors.New("nil user")
+)
+
 type Post struct {
-	id      uuid.UUID
-	creator uuid.UUID
-	content string
-	status  PostStatus
-	media   []Media
+	id        uuid.UUID
+	creator   uuid.UUID
+	content   string
+	status    PostStatus
+	media     []Media
+	createdAt time.Time
+	updatedAt time.Time
 }
 
 func (p Post) ID() string {
@@ -75,6 +83,14 @@ func (p *Post) AddMedia(m Media) error {
 	return nil
 }
 
+func (p Post) CreatedAt() int64 {
+	return p.createdAt.Unix()
+}
+
+func (p Post) UpdatedAt() int64 {
+	return p.updatedAt.Unix()
+}
+
 func (p Post) CanUserGetMedia(user *User, relation *Relation, mediaID string) (*Media, error) {
 	if user.IsLocked() {
 		return nil, ErrLockedUser
@@ -84,7 +100,7 @@ func (p Post) CanUserGetMedia(user *User, relation *Relation, mediaID string) (*
 		return nil, ErrLockedPost
 	}
 
-	if user.ID().String() != p.Creator() && relation == nil || (relation != nil && relation.IsFriend()) {
+	if (user.ID().String() != p.Creator() && relation == nil) || (relation != nil && relation.IsFriend()) {
 		return nil, ErrPermissionDenied
 	}
 
@@ -97,6 +113,42 @@ func (p Post) CanUserGetMedia(user *User, relation *Relation, mediaID string) (*
 	return nil, ErrPostNotContainsMedia
 }
 
+func (p Post) CanUserGetPost(user *User, relation *Relation) error {
+	if user.IsLocked() {
+		return ErrLockedUser
+	}
+
+	if p.IsLocked() {
+		return ErrLockedUser
+	}
+
+	if (user.ID().String() != p.Creator() && relation == nil) || (relation != nil && relation.IsFriend()) {
+		return ErrPermissionDenied
+	}
+
+	return nil
+}
+
+func (p Post) CanUserEditPost(user *User) error {
+	if user == nil {
+		return ErrNilUser
+	}
+
+	if user.IsLocked() {
+		return ErrLockedUser
+	}
+
+	if p.IsLocked() {
+		return ErrLockedPost
+	}
+
+	if p.Creator() != user.ID().String() {
+		return ErrNotCreator
+	}
+
+	return nil
+}
+
 func (p *Post) RemoveMedia(ids ...string) ([]*Media, error) {
 	var (
 		deleted    = make([]*Media, 0, len(ids))

internal/laclongquan/domain/entity/post_database.go

@@ -2,6 +2,7 @@ package entity
 
 import (
 	"os"
+	"time"
 
 	"github.com/google/uuid"
 )
@@ -30,7 +31,7 @@ func NewMediaFromDB(id, owner, mediaType, path string) (*Media, error) {
 	}, nil
 }
 
-func NewPostFromDB(id, creator, status, content string, media []Media) (*Post, error) {
+func NewPostFromDB(id, creator, status, content string, media []Media, createdAt, updatedAt time.Time) (*Post, error) {
 	postID, err := uuid.Parse(id)
 	if err != nil {
 		return nil, err
@@ -42,10 +43,12 @@ func NewPostFromDB(id, creator, status, content string, media []Media) (*Post, e
 	}
 
 	return &Post{
-		id:      postID,
-		creator: creatorID,
-		status:  PostStatus(status),
-		content: content,
-		media:   media,
+		id:        postID,
+		creator:   creatorID,
+		status:    PostStatus(status),
+		content:   content,
+		media:     media,
+		createdAt: createdAt,
+		updatedAt: updatedAt,
 	}, nil
 }

internal/laclongquan/domain/repository/commentrepo.go

@@ -14,7 +14,11 @@ var (
 type CommentUpdateFunc func(ctx context.Context, comment *entity.Comment) (*entity.Comment, error)
 
 type CommentRepository interface {
+	// read
 	GetByIDAndPostID(ctx context.Context, commentID, postID string) (*entity.Comment, error)
+	CountByPostID(ctx context.Context, postID string) (int, error)
+
+	// write
 	Create(ctx context.Context, comment *entity.Comment) error
 	Update(ctx context.Context, commentID, postID string, fn CommentUpdateFunc) error
 	Delete(ctx context.Context, comment *entity.Comment) error

internal/laclongquan/domain/repository/likerepo.go

@@ -7,6 +7,10 @@ import (
 )
 
 type LikeRepository interface {
-	CreateOrDelete(ctx context.Context, like *entity.Like) error
+	// read
 	Count(ctx context.Context, postID string) (int, error)
+	IsLiked(ctx context.Context, userID, postID string) bool
+
+	// write
+	CreateOrDelete(ctx context.Context, like *entity.Like) error
 }

internal/laclongquan/infrastructure/adapter/gormdb/commentdb.go

@@ -77,6 +77,17 @@ func (c commentGorm) getByPostIDCommentID(ctx context.Context, tx *gorm.DB, post
 	return c.unmarshal(cmtDB, post, user), nil
 }
 
+func (c commentGorm) CountByPostID(ctx context.Context, postID string) (int, error) {
+	var commentCount int64
+	if err := c.db.WithContext(ctx).Model(c.cmtModel).
+		Where("post_uuid = ?", postID).Count(&commentCount).
+		Error; err != nil {
+		return 0, err
+	}
+
+	return int(commentCount), nil
+}
+
 func (c commentGorm) Create(ctx context.Context, comment *entity.Comment) error {
 	return c.db.WithContext(ctx).Model(c.cmtModel).
 		Create(c.marshal(comment)).Error

internal/laclongquan/infrastructure/adapter/gormdb/likedb.go

@@ -33,6 +33,30 @@ func (l likeGorm) unmarshal(likeDB *LikeDB) *entity.Like {
 	}
 }
 
+func (l likeGorm) Count(ctx context.Context, postID string) (int, error) {
+	var count = new(int64)
+
+	if err := l.db.WithContext(ctx).Model(l.model).
+		Where("post_uuid = ?", postID).
+		Count(count).Error; err != nil {
+		return 0, err
+	}
+
+	return int(*count), nil
+}
+
+func (l likeGorm) IsLiked(ctx context.Context, userID, postID string) bool {
+	var count = new(int64)
+
+	if err := l.db.WithContext(ctx).Model(l.model).
+		Where("post_uuid = ? AND creator_uuid = ?", postID, userID).
+		Count(count).Error; err != nil {
+		return false
+	}
+
+	return *count > 0
+}
+
 func (l likeGorm) CreateOrDelete(ctx context.Context, like *entity.Like) error {
 	likeDB := l.marshal(like)
 
@@ -51,15 +75,3 @@ func (l likeGorm) CreateOrDelete(ctx context.Context, like *entity.Like) error {
 
 	return nil
 }
-
-func (l likeGorm) Count(ctx context.Context, postID string) (int, error) {
-	var count = new(int64)
-
-	if err := l.db.WithContext(ctx).Model(l.model).
-		Where("post_uuid = ?", postID).
-		Count(count).Error; err != nil {
-		return 0, err
-	}
-
-	return int(*count), nil
-}