78 lines
2.0 KiB
Go
78 lines
2.0 KiB
Go
package repositories
|
|
|
|
import (
|
|
"fmt"
|
|
"rijig/model"
|
|
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserProfilRepository interface {
|
|
FindByID(userID string) (*model.User, error)
|
|
FindAll(page, limit int) ([]model.User, error)
|
|
Update(user *model.User) error
|
|
UpdateAvatar(userID, avatarURL string) error
|
|
UpdatePassword(userID string, newPassword string) error
|
|
}
|
|
|
|
type userProfilRepository struct {
|
|
DB *gorm.DB
|
|
}
|
|
|
|
func NewUserProfilRepository(db *gorm.DB) UserProfilRepository {
|
|
return &userProfilRepository{DB: db}
|
|
}
|
|
|
|
func (r *userProfilRepository) FindByID(userID string) (*model.User, error) {
|
|
var user model.User
|
|
err := r.DB.Preload("Role").Where("id = ?", userID).First(&user).Error
|
|
if err != nil {
|
|
if err == gorm.ErrRecordNotFound {
|
|
return nil, fmt.Errorf("user with ID %s not found", userID)
|
|
}
|
|
return nil, fmt.Errorf("error finding user with ID %s: %v", userID, err)
|
|
}
|
|
|
|
if user.Role == nil {
|
|
return nil, fmt.Errorf("role not found for user ID %s", userID)
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *userProfilRepository) FindAll(page, limit int) ([]model.User, error) {
|
|
var users []model.User
|
|
offset := (page - 1) * limit
|
|
err := r.DB.Preload("Role").Offset(offset).Limit(limit).Find(&users).Error
|
|
if err != nil {
|
|
return nil, fmt.Errorf("error finding all users: %v", err)
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
func (r *userProfilRepository) Update(user *model.User) error {
|
|
err := r.DB.Save(user).Error
|
|
if err != nil {
|
|
return fmt.Errorf("error updating user: %v", err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *userProfilRepository) UpdateAvatar(userID, avatarURL string) error {
|
|
var user model.User
|
|
err := r.DB.Model(&user).Where("id = ?", userID).Update("avatar", avatarURL).Error
|
|
if err != nil {
|
|
return fmt.Errorf("error updating avatar for user ID %s: %v", userID, err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *userProfilRepository) UpdatePassword(userID string, newPassword string) error {
|
|
var user model.User
|
|
err := r.DB.Model(&user).Where("id = ?", userID).Update("password", newPassword).Error
|
|
if err != nil {
|
|
return fmt.Errorf("error updating password for user ID %s: %v", userID, err)
|
|
}
|
|
return nil
|
|
}
|