75 lines
2.1 KiB
Go
75 lines
2.1 KiB
Go
package repository
|
|
|
|
import (
|
|
"users_management/m/model/entity"
|
|
|
|
"github.com/google/uuid"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UsersRepo interface {
|
|
Post(user entity.User) error
|
|
GetRoleByDepartment(departmentName string) (entity.Role, error)
|
|
GetUserByUsername(username string) (entity.User, error)
|
|
GetUserByNomorInduk(nomorInduk string) (entity.User, error) // Add this
|
|
CreateUserFromExternal(user entity.User) error // Add this
|
|
UpdateUserRole(nomorInduk string, roleID uuid.UUID) error // Add this
|
|
GetAllUsers() ([]entity.User, error) // Add this
|
|
}
|
|
|
|
type usersRepo struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewUsersRepo(db *gorm.DB) UsersRepo {
|
|
return &usersRepo{
|
|
db: db,
|
|
}
|
|
}
|
|
|
|
func (r *usersRepo) Post(user entity.User) error {
|
|
err := r.db.Create(&user).Error
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func (r *usersRepo) GetRoleByDepartment(departmentName string) (entity.Role, error) {
|
|
var role entity.Role
|
|
err := r.db.Where("name = ?", departmentName).First(&role).Error
|
|
if err != nil {
|
|
return role, err
|
|
}
|
|
return role, nil
|
|
}
|
|
|
|
func (r *usersRepo) GetUserByUsername(username string) (entity.User, error) {
|
|
var users entity.User
|
|
err := r.db.Where("username = ?", username).Preload("Role").First(&users).Error
|
|
if err != nil {
|
|
return users, err
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
// New methods for RBAC
|
|
func (r *usersRepo) GetUserByNomorInduk(nomorInduk string) (entity.User, error) {
|
|
var user entity.User
|
|
err := r.db.Where("nomor_induk = ?", nomorInduk).Preload("Role").First(&user).Error
|
|
return user, err
|
|
}
|
|
|
|
func (r *usersRepo) CreateUserFromExternal(user entity.User) error {
|
|
return r.db.Create(&user).Error
|
|
}
|
|
|
|
func (r *usersRepo) UpdateUserRole(nomorInduk string, roleID uuid.UUID) error {
|
|
return r.db.Model(&entity.User{}).Where("nomor_induk = ?", nomorInduk).Update("role_id", roleID).Error
|
|
}
|
|
|
|
func (r *usersRepo) GetAllUsers() ([]entity.User, error) {
|
|
var users []entity.User
|
|
err := r.db.Preload("Role").Find(&users).Error
|
|
return users, err
|
|
} |