95 lines
2.4 KiB
Go
95 lines
2.4 KiB
Go
package repository
|
|
|
|
import (
|
|
"database/sql"
|
|
"errors"
|
|
|
|
"github.com/daniil/homelab-api/internal/model"
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
var ErrUserNotFound = errors.New("user not found")
|
|
var ErrUserExists = errors.New("user already exists")
|
|
|
|
type UserRepository struct {
|
|
db *sqlx.DB
|
|
}
|
|
|
|
func NewUserRepository(db *sqlx.DB) *UserRepository {
|
|
return &UserRepository{db: db}
|
|
}
|
|
|
|
func (r *UserRepository) Create(user *model.User) error {
|
|
query := `
|
|
INSERT INTO users (email, username, password_hash, email_verified)
|
|
VALUES ($1, $2, $3, $4)
|
|
RETURNING id, created_at, updated_at`
|
|
|
|
err := r.db.QueryRow(query, user.Email, user.Username, user.PasswordHash, user.EmailVerified).
|
|
Scan(&user.ID, &user.CreatedAt, &user.UpdatedAt)
|
|
|
|
if err != nil {
|
|
if isUniqueViolation(err) {
|
|
return ErrUserExists
|
|
}
|
|
return err
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (r *UserRepository) GetByID(id int64) (*model.User, error) {
|
|
var user model.User
|
|
query := `SELECT id, email, username, password_hash, email_verified, created_at, updated_at FROM users WHERE id = $1`
|
|
|
|
if err := r.db.Get(&user, query, id); err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, ErrUserNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepository) GetByEmail(email string) (*model.User, error) {
|
|
var user model.User
|
|
query := `SELECT id, email, username, password_hash, email_verified, created_at, updated_at FROM users WHERE email = $1`
|
|
|
|
if err := r.db.Get(&user, query, email); err != nil {
|
|
if errors.Is(err, sql.ErrNoRows) {
|
|
return nil, ErrUserNotFound
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &user, nil
|
|
}
|
|
|
|
func (r *UserRepository) Update(user *model.User) error {
|
|
query := `
|
|
UPDATE users
|
|
SET username = $2, updated_at = CURRENT_TIMESTAMP
|
|
WHERE id = $1
|
|
RETURNING updated_at`
|
|
|
|
return r.db.QueryRow(query, user.ID, user.Username).Scan(&user.UpdatedAt)
|
|
}
|
|
|
|
func (r *UserRepository) UpdatePassword(id int64, passwordHash string) error {
|
|
query := `UPDATE users SET password_hash = $2, updated_at = CURRENT_TIMESTAMP WHERE id = $1`
|
|
_, err := r.db.Exec(query, id, passwordHash)
|
|
return err
|
|
}
|
|
|
|
func (r *UserRepository) SetEmailVerified(id int64) error {
|
|
query := `UPDATE users SET email_verified = TRUE, updated_at = CURRENT_TIMESTAMP WHERE id = $1`
|
|
_, err := r.db.Exec(query, id)
|
|
return err
|
|
}
|
|
|
|
func isUniqueViolation(err error) bool {
|
|
return err != nil && (err.Error() == "pq: duplicate key value violates unique constraint \"users_email_key\"" ||
|
|
err.Error() == "UNIQUE constraint failed: users.email")
|
|
}
|