mirror of
https://github.com/muety/wakapi.git
synced 2023-08-10 21:12:56 +03:00
108 lines
2.5 KiB
Go
108 lines
2.5 KiB
Go
package repositories
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/muety/wakapi/models"
|
|
"gorm.io/gorm"
|
|
)
|
|
|
|
type UserRepository struct {
|
|
db *gorm.DB
|
|
}
|
|
|
|
func NewUserRepository(db *gorm.DB) *UserRepository {
|
|
return &UserRepository{db: db}
|
|
}
|
|
|
|
func (r *UserRepository) GetById(userId string) (*models.User, error) {
|
|
u := &models.User{}
|
|
if err := r.db.Where(&models.User{ID: userId}).First(u).Error; err != nil {
|
|
return u, err
|
|
}
|
|
return u, nil
|
|
}
|
|
|
|
func (r *UserRepository) GetByApiKey(key string) (*models.User, error) {
|
|
u := &models.User{}
|
|
if err := r.db.Where(&models.User{ApiKey: key}).First(u).Error; err != nil {
|
|
return u, err
|
|
}
|
|
return u, nil
|
|
}
|
|
|
|
func (r *UserRepository) GetAll() ([]*models.User, error) {
|
|
var users []*models.User
|
|
if err := r.db.
|
|
Where(&models.User{}).
|
|
Find(&users).Error; err != nil {
|
|
return nil, err
|
|
}
|
|
return users, nil
|
|
}
|
|
|
|
func (r *UserRepository) Count() (int64, error) {
|
|
var count int64
|
|
if err := r.db.
|
|
Model(&models.User{}).
|
|
Count(&count).Error; err != nil {
|
|
return 0, err
|
|
}
|
|
return count, nil
|
|
}
|
|
|
|
func (r *UserRepository) InsertOrGet(user *models.User) (*models.User, bool, error) {
|
|
result := r.db.FirstOrCreate(user, &models.User{ID: user.ID})
|
|
if err := result.Error; err != nil {
|
|
return nil, false, err
|
|
}
|
|
|
|
if result.RowsAffected == 1 {
|
|
return user, true, nil
|
|
}
|
|
|
|
return user, false, nil
|
|
}
|
|
|
|
func (r *UserRepository) Update(user *models.User) (*models.User, error) {
|
|
updateMap := map[string]interface{}{
|
|
"api_key": user.ApiKey,
|
|
"password": user.Password,
|
|
"last_logged_in_at": user.LastLoggedInAt,
|
|
"share_data_max_days": user.ShareDataMaxDays,
|
|
"share_editors": user.ShareEditors,
|
|
"share_languages": user.ShareLanguages,
|
|
"share_oss": user.ShareOSs,
|
|
"share_projects": user.ShareProjects,
|
|
"share_machines": user.ShareMachines,
|
|
"wakatime_api_key": user.WakatimeApiKey,
|
|
}
|
|
|
|
result := r.db.Model(user).Updates(updateMap)
|
|
if err := result.Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if result.RowsAffected != 1 {
|
|
return nil, errors.New("nothing updated")
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (r *UserRepository) UpdateField(user *models.User, key string, value interface{}) (*models.User, error) {
|
|
result := r.db.Model(user).Update(key, value)
|
|
if err := result.Error; err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
if result.RowsAffected != 1 {
|
|
return nil, errors.New("nothing updated")
|
|
}
|
|
|
|
return user, nil
|
|
}
|
|
|
|
func (r *UserRepository) Delete(user *models.User) error {
|
|
return r.db.Delete(user).Error
|
|
}
|