feat: Add unread notifications field for user profile

swibly · Nov 5, 2024 · c0ca4c3 · c0ca4c3
1 parent f91be80
commit c0ca4c3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 4 deletions.
diff --git a/internal/model/dto/user.go b/internal/model/dto/user.go
@@ -98,7 +98,8 @@ type UserProfile struct {
 	Country  string `json:"country"`
 	Language string `json:"language"`
 
-	Permissions []string `gorm:"-" json:"permissions"`
+	Permissions         []string `gorm:"-" json:"permissions"`
+	UnreadNotifications int64    `gorm:"-" json:"unread_notifications"`
 
 	ProfilePicture string `json:"pfp"`
 }

diff --git a/internal/service/repository/user.go b/internal/service/repository/user.go
@@ -13,8 +13,9 @@ import (
 type userRepository struct {
 	db *gorm.DB
 
-	followRepo     FollowRepository
-	permissionRepo PermissionRepository
+	followRepo       FollowRepository
+	permissionRepo   PermissionRepository
+	notificationRepo NotificationRepository
 }
 
 type UserRepository interface {
@@ -31,7 +32,7 @@ type UserRepository interface {
 }
 
 func NewUserRepository() UserRepository {
-	return &userRepository{db: db.Postgres, followRepo: NewFollowRepository(), permissionRepo: NewPermissionRepository()}
+	return &userRepository{db: db.Postgres, followRepo: NewFollowRepository(), permissionRepo: NewPermissionRepository(), notificationRepo: NewNotificationRepository()}
 }
 
 func (u userRepository) Create(createModel *model.User) error {
@@ -81,6 +82,12 @@ func (u userRepository) Get(searchModel *model.User) (*dto.UserProfile, error) {
 		}
 	}
 
+	if count, err := u.notificationRepo.GetUnreadCount(user.ID); err != nil {
+		return nil, err
+	} else {
+		user.UnreadNotifications = count
+	}
+
 	return user, nil
 }